lib: add Mac and MacVerify for md4

2 files changed, 33 insertions, 0 deletions

diff --git a/lib/md4.go b/lib/md4.go
index 632f6da..eab3073 100644
--- a/lib/md4.go
+++ b/lib/md4.go
@@ -251,3 +251,16 @@ func (md *Md4) Hash() []byte {
 
 	return d
 }
+
+func (md *Md4) Mac(secret, msg []byte) []byte {
+	md.Message(append(secret, msg...))
+	return md.Hash()
+}
+
+func (md *Md4) MacVerify(secret, msg, mac []byte) bool {
+	md.Message(append(secret, msg...))
+	if BytesEqual(md.Hash(), mac) {
+		return true
+	}
+	return false
+}
diff --git a/lib/md4_test.go b/lib/md4_test.go
index 31b7b31..8f832b8 100644
--- a/lib/md4_test.go
+++ b/lib/md4_test.go
@@ -56,3 +56,23 @@ func TestMd4Hash(t *testing.T) {
 	}
 
 }
+
+func TestMd4MacVerify(t *testing.T) {
+	md4 := Md4{}
+	md4.Init([]uint32{})
+
+	sec := StrToBytes("honey")
+	msg := StrToBytes("abc")
+
+	// Test Mac
+	mac := md4.Mac(sec, msg)
+	e := "22fd3b98ca4a901fec432ed2c49d3f83" // Expected Mac
+	if BytesToHexStr(mac) != e {
+		t.Errorf("Error: md5 mac failed %x != %s\n", mac, e)
+	}
+
+	// Test MacVerify.
+	if !md4.MacVerify(sec, msg, mac) {
+		t.Errorf("Error: mac verification failed\n")
+	}
+}