diff src/com/tmate/hgkit/ll/DigestHelper.java @ 42:92c3d0920d58

Real integrity check, with exception. DigestHelper refactored to accomodate new needs
author Artem Tikhomirov <tikhomirov.artem@gmail.com>
date Fri, 14 Jan 2011 04:29:03 +0100
parents 858d1b2458cb
children
line wrap: on
line diff
--- a/src/com/tmate/hgkit/ll/DigestHelper.java	Fri Jan 14 04:14:08 2011 +0100
+++ b/src/com/tmate/hgkit/ll/DigestHelper.java	Fri Jan 14 04:29:03 2011 +0100
@@ -9,12 +9,18 @@
 import java.security.NoSuchAlgorithmException;
 
 /**
- * TODO sha1_binary to give array for Nodeid.equalsTo 
- *
+ * <pre>
+ * DigestHelper dh;
+ * dh.sha1(...).asHexString();
+ *  or 
+ * dh = dh.sha1(...);
+ * nodeid.equalsTo(dh.asBinary());
+ * </pre>
  * @author artem
  */
 public class DigestHelper {
 	private MessageDigest sha1;
+	private byte[] digest;
 
 	public DigestHelper() {
 	}
@@ -32,12 +38,12 @@
 	}
 
 
-	public String sha1(Nodeid nodeid1, Nodeid nodeid2, byte[] data) {
+	public DigestHelper sha1(Nodeid nodeid1, Nodeid nodeid2, byte[] data) {
 		return sha1(nodeid1.cloneData(), nodeid2.cloneData(), data);
 	}
 
 	//  sha1_digest(min(p1,p2) ++ max(p1,p2) ++ final_text)
-	public String sha1(byte[] nodeidParent1, byte[] nodeidParent2, byte[] data) {
+	public DigestHelper sha1(byte[] nodeidParent1, byte[] nodeidParent2, byte[] data) {
 		MessageDigest alg = getSHA1();
 		if ((nodeidParent1[0] & 0x00FF) < (nodeidParent2[0] & 0x00FF)) { 
 			alg.update(nodeidParent1);
@@ -46,21 +52,36 @@
 			alg.update(nodeidParent2);
 			alg.update(nodeidParent1);
 		}
-		byte[] digest = alg.digest(data);
+		digest = alg.digest(data);
 		assert digest.length == 20;
-		return toHexString(digest, 0, 20);
+		return this;
+	}
+	
+	public String asHexString() {
+		if (digest == null) {
+			throw new IllegalStateException("Shall init with sha1() call first");
+		}
+		return toHexString(digest, 0, digest.length);
+	}
+	
+	// by reference, be careful not to modify (or #clone() if needed)
+	public byte[] asBinary() {
+		if (digest == null) {
+			throw new IllegalStateException("Shall init with sha1() call first");
+		}
+		return digest; 
 	}
 
 	// XXX perhaps, digest functions should throw an exception, as it's caller responsibility to deal with eof, etc
-	public byte[] sha1(InputStream is /*ByteBuffer*/) throws IOException {
+	public DigestHelper sha1(InputStream is /*ByteBuffer*/) throws IOException {
 		MessageDigest alg = getSHA1();
 		byte[] buf = new byte[1024];
 		int c;
 		while ((c = is.read(buf)) != -1) {
 			alg.update(buf, 0, c);
 		}
-		byte[] digest = alg.digest();
-		return digest;
+		digest = alg.digest();
+		return this;
 	}
 
 	public static String toHexString(byte[] data, final int offset, final int count) {