summaryrefslogtreecommitdiffstats
path: root/src/main/java/org/openslx/imagemaster/crcchecker/CRCFile.java
blob: b557adae25d73da648c1c8bf20128107a02f1a9c (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
package org.openslx.imagemaster.crcchecker;

import java.io.DataInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.zip.CRC32;

/**
 * Represents a crcfile
 * @author nils
 *
 */
public class CRCFile
{
	private File file;
	private List<Integer> crcSums = null;
	
	CRCFile(String filename) {
		this.file = new File( filename );
	}
	
	/**
	 * Checks if this crc file is valid.
	 * (If the crc over the file is equal to the first crc sum.)
	 * @return Whether the crc file is valid
	 * @throws IOException If the file could not be read or could not be found
	 */
	public boolean isValid() throws IOException {
		FileInputStream fis = new FileInputStream( file );
		DataInputStream dis = new DataInputStream( fis );
		
		int crcSum = dis.readInt();
		
		CRC32 crcCalc = new CRC32();
		
		byte[] bytes = new byte[(int) file.length() - Integer.SIZE/8];	// byte array with length of the file minus the first crc sum (=4byte)
		fis.read( bytes );
		crcCalc.update( bytes );
		
		dis.close();
		
		if (crcSum == Integer.reverseBytes( (int) crcCalc.getValue() ) ) return true;
		else return false;
	}
	
	/**
	 * Get a specified crcSum for a block number
	 * @param blockNumber
	 * @return The crcSum or 0 if the block number is invalid
	 * @throws IOException 
	 */
	public int getCRCSum(int blockNumber) throws IOException {
		if (crcSums == null) {
			// the crcSums were not read yet
			DataInputStream dis = new DataInputStream( new FileInputStream( file ) );
			crcSums = new ArrayList<>();
			for (int i = 0; i < file.length()/4; i++) {
				int s = dis.readInt();
				if (i > 0) crcSums.add( s );	// skip the first crcSum because it's the sum over the crcFile
			}
			dis.close();
		}
		
		if (blockNumber < 0) return 0;
		if (blockNumber > crcSums.size() - 1) return 0; 
		
		return crcSums.get( blockNumber );
	}		
}