SRP::Authentication::Session holds the per session data

1 files changed, 4 insertions, 4 deletions

diff --git a/lib/srp/client.rb b/lib/srp/client.rb
index 1be2461..667d5ba 100644
--- a/lib/srp/client.rb
+++ b/lib/srp/client.rb
@@ -19,10 +19,10 @@ module SRP
       x = calculate_x(username, password, salt)
       a = bigrand(32).hex
       aa = modpow(GENERATOR, a, PRIME_N) # A = g^a (mod N)
-      bb = server.initialize_auth(aa)
-      u = calculate_u(aa, bb, PRIME_N)
-      client_s = calculate_client_s(x, a, bb, u)
-      server.authenticate(calculate_m(aa,bb,client_s))
+      session = server.initialize_auth(aa)
+      u = calculate_u(aa, session.bb, PRIME_N)
+      client_s = calculate_client_s(x, a, session.bb, u)
+      server.authenticate(calculate_m(aa,session.bb,client_s), session)
     end
 
     protected