summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--lib/srp.go30
-rw-r--r--lib/srp_test.go49
2 files changed, 79 insertions, 0 deletions
diff --git a/lib/srp.go b/lib/srp.go
index 20208a6..491565c 100644
--- a/lib/srp.go
+++ b/lib/srp.go
@@ -78,6 +78,8 @@ type SRPClientSession struct {
// Multipier parameter. Client and server agree upon the value
// of N.
k *big.Int
+ // Hashing object for H() function.
+ h Sha256
// User's email address
ident string
// Scrambling parameter.
@@ -131,3 +133,31 @@ func NewSRPUser(n, g, k, ident, pass string) (*SRPUser, error) {
return user, nil
}
+
+func NewSRPClientSession(n, g, k, ident string) (*SRPClientSession, error) {
+ var ok bool
+
+ session := new(SRPClientSession)
+ session.n, ok = new(big.Int).SetString(StripSpaceChars(n), 16)
+ if !ok {
+ return nil, CPError{"n is invalid"}
+ }
+ session.g, ok = new(big.Int).SetString(StripSpaceChars(g), 16)
+ if !ok {
+ return nil, CPError{"g is invalid"}
+ }
+ session.k, ok = new(big.Int).SetString(StripSpaceChars(k), 16)
+ if !ok {
+ return nil, CPError{"k is invalid"}
+ }
+ session.ident = ident
+
+ // Initialize hashing object.
+ session.h = Sha256{}
+ session.h.Init([]uint32{})
+
+ // Generate secret ephemeral value.
+ session.a = big.NewInt(RandomInt(1, 10000000))
+
+ return session, nil
+}
diff --git a/lib/srp_test.go b/lib/srp_test.go
index d7e3b0d..a31e658 100644
--- a/lib/srp_test.go
+++ b/lib/srp_test.go
@@ -67,3 +67,52 @@ func TestNewSRPUser(t *testing.T) {
return
}
}
+
+func TestNewSRPClientSession(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
+ }
+
+ // Check n.
+ bigN, _ := new(big.Int).SetString(StripSpaceChars(n), 16)
+ if session.n.Cmp(bigN) != 0 {
+ t.Error("Error: n not set correctly\n")
+ return
+ }
+ // Check g.
+ bigG, _ := new(big.Int).SetString(StripSpaceChars(g), 16)
+ if session.g.Cmp(bigG) != 0 {
+ t.Error("Error: g not set correctly\n")
+ return
+ }
+ // Check k.
+ bigK, _ := new(big.Int).SetString(StripSpaceChars(k), 16)
+ if session.k.Cmp(bigK) != 0 {
+ t.Error("Error: k not set correctly\n")
+ return
+ }
+ // Check ident.
+ if session.ident != ident {
+ t.Error("Error: user not set correctly\n")
+ return
+ }
+ // Check a.
+ if session.a.Cmp(big.NewInt(1)) < 0 {
+ t.Error("Error: a not set correctly\n")
+ return
+ }
+}