c20d8e878a4eec04ebaa26ab892efa1790aea431
[simantics/platform.git] / bundles / org.simantics.acorn / src / org / simantics / acorn / HeadState.java
1 package org.simantics.acorn;
2
3 import java.io.ByteArrayInputStream;
4 import java.io.File;
5 import java.io.IOException;
6 import java.io.OutputStream;
7 import java.nio.file.Files;
8 import java.nio.file.Path;
9 import java.security.MessageDigest;
10 import java.security.NoSuchAlgorithmException;
11 import java.util.ArrayList;
12 import java.util.Arrays;
13
14 import org.simantics.databoard.Bindings;
15 import org.simantics.databoard.binding.mutable.MutableVariant;
16 import org.simantics.databoard.serialization.Serializer;
17 import org.simantics.databoard.util.binary.BinaryMemory;
18
19 public class HeadState {
20
21     public int headChangeSetId = 0;
22     public long transactionId = 1;
23     public long reservedIds = 3;
24
25     public ArrayList<String> clusters = new ArrayList<>();
26     public ArrayList<String> files = new ArrayList<>();
27     public ArrayList<String> stream = new ArrayList<>();
28     public ArrayList<String> cs = new ArrayList<>();
29 //    public ArrayList<String> ccs = new ArrayList<String>();
30
31     public static HeadState load(Path directory) throws InvalidHeadStateException {
32         Path f = directory.resolve("head.state");
33         
34         try {
35             byte[] bytes = Files.readAllBytes(f);
36             MessageDigest sha1 = MessageDigest.getInstance("SHA-1");
37             int digestLength = sha1.getDigestLength();
38             
39             sha1.update(bytes, digestLength, bytes.length - digestLength);
40             byte[] newChecksum = sha1.digest();
41             if (!Arrays.equals(newChecksum, Arrays.copyOfRange(bytes, 0, digestLength))) {
42                 throw new InvalidHeadStateException(
43                         "Checksum " + Arrays.toString(newChecksum) + " does not match excpected "
44                                 + Arrays.toString(Arrays.copyOfRange(bytes, 0, digestLength)) + " for " + f.toAbsolutePath());
45             }
46             
47             HeadState object = (HeadState) org.simantics.databoard.Files.readFile(new ByteArrayInputStream(bytes, digestLength, bytes.length - digestLength), Bindings.getBindingUnchecked(HeadState.class));
48             return object;
49             
50         } catch (IOException i) {
51             return new HeadState();
52 //            throw new InvalidHeadStateException(i);
53         } catch (NoSuchAlgorithmException e) {
54             throw new Error("SHA-1 Algorithm not found", e);
55         } catch (Throwable t) {
56             throw new InvalidHeadStateException(t);
57         }
58     }
59     
60     public void save(Path directory) throws IOException {
61         Path f = directory.resolve("head.state");
62         try {
63             BinaryMemory rf = new BinaryMemory(4096);
64             try {
65                 MutableVariant v = new MutableVariant(Bindings.getBindingUnchecked(HeadState.class), this);
66                 Serializer s = Bindings.getSerializerUnchecked( Bindings.VARIANT );
67                 s.serialize(rf, v);
68             } finally {
69                 rf.close();
70             }
71             
72             byte[] bytes = rf.toByteBuffer().array();
73             
74             MessageDigest sha1 = MessageDigest.getInstance("SHA-1");
75             sha1.update(bytes);
76             byte[] checksum = sha1.digest();
77             
78             try (OutputStream out = Files.newOutputStream(f)) {
79                 out.write(checksum);
80                 out.write(bytes);
81             }
82             FileIO.syncPath(f);
83         } catch (NoSuchAlgorithmException e) {
84             throw new Error("SHA-1 digest not found, should not happen", e);
85         }
86     }
87
88     public static void validateHeadStateIntegrity(Path headState) throws InvalidHeadStateException, IOException {
89         try {
90             byte[] bytes = Files.readAllBytes(headState);
91             MessageDigest sha1 = MessageDigest.getInstance("SHA-1");
92             int digestLength = sha1.getDigestLength();
93             sha1.update(bytes, digestLength, bytes.length - digestLength);
94             byte[] newChecksum = sha1.digest();
95             if (!Arrays.equals(newChecksum, Arrays.copyOfRange(bytes, 0, digestLength))) {
96                 throw new InvalidHeadStateException(
97                         "Checksum " + Arrays.toString(newChecksum) + " does not match excpected "
98                                 + Arrays.toString(Arrays.copyOfRange(bytes, 0, digestLength)) + " for " + headState.toAbsolutePath());
99             }
100         } catch (NoSuchAlgorithmException e) {
101             throw new Error("SHA-1 digest not found, should not happen", e);
102         }
103     }
104 }