diff options
Diffstat (limited to 'lib/srp_test.go')
| -rw-r--r-- | lib/srp_test.go | 66 |
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 + } +} |