Merge dd8a681 into 0057c01

tlsh.go

@@ -65,6 +65,22 @@ func (t *Tlsh) String() string {
 	return hex.EncodeToString(t.Binary())
 }
 
+// Parsing the hash of the string type 
+func ParseStringToTlsh(hashString string) (*Tlsh, error) {
+	var code [codeSize]byte
+	hashByte, err := hex.DecodeString(hashString)
+	if err != nil {
+		return &Tlsh{}, err
+	}
+	chechsum := swapByte(hashByte[0])
+	lValue := swapByte(hashByte[1])
+	qRatio := hashByte[2]
+	q1Ratio := (qRatio >> 4) & 0xF
+	q2Ratio := qRatio & 0xF
+	copy(code[:], hashByte[3:])
+	return New(chechsum, lValue, q1Ratio, q2Ratio, qRatio, code), nil
+}
+
 func quartilePoints(buckets [numBuckets]uint) (q1, q2, q3 uint) {
 	var spl, spr uint
 	p1 := uint(effBuckets/4 - 1)

tlsh_test.go

@@ -79,6 +79,19 @@ func TestDiff(t *testing.T) {
 	}
 }
 
+func TestParseStringToTlsh(t *testing.T) {
+	for _, tc := range hashTestCases {
+		if hash, err := ParseStringToTlsh(tc.hash); err != nil || hash.String() != tc.hash {
+			if err != nil {
+				t.Error(err)
+			}
+			if hash.String() != tc.hash {
+				t.Errorf("\noriginal and parsed tlsh have different hash %s vs. %s\n", tc.hash, hash.String())
+			}
+		}
+	}
+}
+
 func BenchmarkPearson(b *testing.B) {
 	var salt = byte(0)
 	var keys = [3]byte{1, 3, 7}