blob: 3ec6468031c465d88fc417118b7044afae89a49f (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
package org.openslx.filetransfer.util;
import java.nio.ByteBuffer;
import java.util.Collection;
import java.util.List;
import org.openslx.filetransfer.FileRange;
public class FileChunk
{
public static final int CHUNK_SIZE_MIB = 16;
public static final int CHUNK_SIZE = CHUNK_SIZE_MIB * ( 1024 * 1024 );
public final FileRange range;
public final byte[] sha1sum;
private int failCount = 0;
public FileChunk( long startOffset, long endOffset, byte[] sha1sum )
{
this.range = new FileRange( startOffset, endOffset );
this.sha1sum = sha1sum;
}
/**
* Signal that transferring this chunk seems to have failed (checksum
* mismatch).
*
* @return Number of times the transfer failed now
*/
public synchronized int incFailed()
{
return ++failCount;
}
public int getChunkIndex()
{
return (int) ( range.startOffset / CHUNK_SIZE );
}
@Override
public String toString()
{
return "[Chunk " + getChunkIndex() + " (" + range.startOffset + "-" + range.endOffset + "), fails: " + failCount + "]";
}
//
public static int fileSizeToChunkCount( long fileSize )
{
return (int) ( ( fileSize + CHUNK_SIZE - 1 ) / CHUNK_SIZE );
}
public static void createChunkList( Collection<FileChunk> list, long fileSize, List<ByteBuffer> sha1Sums )
{
if ( fileSize < 0 )
throw new IllegalArgumentException( "fileSize cannot be negative" );
if ( !list.isEmpty() )
throw new IllegalArgumentException( "Passed list is not empty" );
long chunkCount = fileSizeToChunkCount( fileSize );
if ( sha1Sums != null ) {
if ( sha1Sums.size() != chunkCount )
throw new IllegalArgumentException(
"Passed a sha1sum list, but hash count in list doesn't match expected chunk count" );
long offset = 0;
for ( ByteBuffer sha1sum : sha1Sums ) { // Do this as we don't know how efficient List.get(index) is...
long end = offset + CHUNK_SIZE;
if ( end > fileSize )
end = fileSize;
list.add( new FileChunk( offset, end, sha1sum.array() ) );
offset = end;
}
return;
}
long offset = 0;
while ( offset < fileSize ) { // ...otherwise we could share this code
long end = offset + CHUNK_SIZE;
if ( end > fileSize )
end = fileSize;
list.add( new FileChunk( offset, end, null ) );
offset = end;
}
}
public boolean hasSha1Sum()
{
return sha1sum != null && sha1sum.length == 20;
}
}
|