lib: add srp ephemeral functions

1 files changed, 66 insertions, 0 deletions

diff --git a/lib/srp_test.go b/lib/srp_test.go
index a31e658..3e3c8d6 100644
--- a/lib/srp_test.go
+++ b/lib/srp_test.go
@@ -68,6 +68,42 @@ func TestNewSRPUser(t *testing.T) {
 	}
 }
 
+func TestSRPUserEphemeralKey(t *testing.T) {
+	n := StripSpaceChars(
+		`ffffffffffffffffc90fdaa22168c234c4c6628b80dc1cd129024
+                 e088a67cc74020bbea63b139b22514a08798e3404ddef9519b3cd
+                 3a431b302b0a6df25f14374fe1356d6d51c245e485b576625e7ec
+                 6f44c42e9a637ed6b0bff5cb6f406b7edee386bfb5a899fa5ae9f
+                 24117c4b1fe649286651ece45b3dc2007cb8a163bf0598da48361
+                 c55d39a69163fa8fd24cf5f83655d23dca3ad961c62f356208552
+                 bb9ed529077096966d670c354e4abc9804f1746c08ca237327fff
+                 fffffffffffff`)
+	g := "2"
+	k := "3"
+	ident := "s@ricketyspace.net"
+	pass := "d59d6c93af0f37f272d924979"
+	user, err := NewSRPUser(n, g, k, ident, pass)
+	if err != nil {
+		t.Errorf("Error: %v\n", err)
+		return
+	}
+
+	user.EphemeralKeyGen()
+	if user.b.Cmp(big.NewInt(0)) != 1 {
+		t.Errorf("Error: b is <= 0")
+		return
+	}
+	pub, err := user.EphemeralKeyPub()
+	if err != nil {
+		t.Errorf("Error: %v\n", err)
+		return
+	}
+	if pub.Cmp(big.NewInt(0)) != 1 {
+		t.Errorf("Error: pub is <= 0")
+		return
+	}
+}
+
 func TestNewSRPClientSession(t *testing.T) {
 	n := StripSpaceChars(
 		`ffffffffffffffffc90fdaa22168c234c4c6628b80dc1cd129024
@@ -116,3 +152,33 @@ func TestNewSRPClientSession(t *testing.T) {
 		return
 	}
 }
+
+func TestSRPClientSessionEphemeralKeyPub(t *testing.T) {
+	n := StripSpaceChars(
+		`ffffffffffffffffc90fdaa22168c234c4c6628b80dc1cd129024
+                 e088a67cc74020bbea63b139b22514a08798e3404ddef9519b3cd
+                 3a431b302b0a6df25f14374fe1356d6d51c245e485b576625e7ec
+                 6f44c42e9a637ed6b0bff5cb6f406b7edee386bfb5a899fa5ae9f
+                 24117c4b1fe649286651ece45b3dc2007cb8a163bf0598da48361
+                 c55d39a69163fa8fd24cf5f83655d23dca3ad961c62f356208552
+                 bb9ed529077096966d670c354e4abc9804f1746c08ca237327fff
+                 fffffffffffff`)
+	g := "2"
+	k := "3"
+	ident := "s@ricketyspace.net"
+	session, err := NewSRPClientSession(n, g, k, ident)
+	if err != nil {
+		t.Errorf("Error: %v\n", err)
+		return
+	}
+
+	pub, err := session.EphemeralKeyPub()
+	if err != nil {
+		t.Errorf("Error: %v\n", err)
+		return
+	}
+	if pub.Cmp(big.NewInt(0)) != 1 {
+		t.Errorf("Error: pub is <= 0")
+		return
+	}
+}