summaryrefslogtreecommitdiffstats
path: root/lib/dh_test.go
diff options
context:
space:
mode:
authorsiddharth <s@ricketyspace.net>2021-11-21 18:45:53 -0500
committersiddharth <s@ricketyspace.net>2021-11-21 18:45:53 -0500
commit2b52444043b26a01cc1eb82c456e325916e25194 (patch)
tree6f84867fe3131aa6fb497faaaea0659c1b1923e2 /lib/dh_test.go
parent1748a57ad36b3cd9cb0522ee2963b7183f4ee0b4 (diff)
lib: implement diffie-hellman
Diffstat (limited to 'lib/dh_test.go')
-rw-r--r--lib/dh_test.go36
1 files changed, 36 insertions, 0 deletions
diff --git a/lib/dh_test.go b/lib/dh_test.go
new file mode 100644
index 0000000..f313d82
--- /dev/null
+++ b/lib/dh_test.go
@@ -0,0 +1,36 @@
+// Copyright © 2021 siddharth <s@ricketyspace.net>
+// SPDX-License-Identifier: ISC
+
+package lib
+
+import "testing"
+
+func TestDH(t *testing.T) {
+ p := `ffffffffffffffffc90fdaa22168c234c4c6628b80dc1cd129024
+e088a67cc74020bbea63b139b22514a08798e3404ddef9519b3cd
+3a431b302b0a6df25f14374fe1356d6d51c245e485b576625e7ec
+6f44c42e9a637ed6b0bff5cb6f406b7edee386bfb5a899fa5ae9f
+24117c4b1fe649286651ece45b3dc2007cb8a163bf0598da48361
+c55d39a69163fa8fd24cf5f83655d23dca3ad961c62f356208552
+bb9ed529077096966d670c354e4abc9804f1746c08ca237327fff
+fffffffffffff`
+ g := "2"
+
+ bobDH, ok := NewDH(p, g)
+ if !ok {
+ t.Errorf("Error: Unable to initialize Bob's DH\n")
+ return
+ }
+
+ aliceDH, ok := NewDH(p, g)
+ if !ok {
+ t.Errorf("Error: Unable to initialize Alice's DH\n")
+ return
+ }
+
+ bobSK := bobDH.SharedSecret(aliceDH.Pub())
+ aliceSK := aliceDH.SharedSecret(bobDH.Pub())
+ if bobSK.Cmp(aliceSK) != 0 {
+ t.Errorf("Error: Shared Secret for Bob and Alice is not the same\n")
+ }
+}