srp/src/server.rs

   1 //! SRP server implementation
   2 //!
   3 //! # Usage
   4 //! First receive user's username and public value `a_pub`, retrieve from a
   5 //! database `UserRecord` for a given username, generate `b` (e.g. 512 bits
   6 //! long) and initialize SRP server instance:
   7 //!
   8 //! ```ignore
   9 //! use srp::groups::G_2048;
  10 //!
  11 //! let (username, a_pub) = conn.receive_handshake();
  12 //! let user = db.retrieve_user_record(username);
  13 //! let b = [0u8; 64];
  14 //! rng.fill_bytes(&mut b);
  15 //! let server = SrpServer::<Sha256>::new(&user, &a_pub, &b, &G_2048)?;
  16 //! ```
  17 //!
  18 //! Next send to user `b_pub` and `salt` from user record:
  19 //!
  20 //! ```ignore
  21 //! let b_pub = server.get_b_pub();
  22 //! conn.reply_to_handshake(&user.salt, b_pub);
  23 //! ```
  24 //!
  25 //! And finally recieve user proof, verify it and send server proof in the
  26 //! reply:
  27 //!
  28 //! ```ignore
  29 //! let user_proof = conn.receive_proof();
  30 //! let server_proof = server.verify(user_proof)?;
  31 //! conn.send_proof(server_proof);
  32 //! ```
  33 //!
  34 //! To get the shared secret use `get_key` method. As alternative to using
  35 //! `verify` method it's also possible to use this key for authentificated
  36 //! encryption.
  37 use std::marker::PhantomData;
  38
  39 use digest::Digest;
  40 use generic_array::GenericArray;
  41 use num::{BigUint, Zero};
  42
  43 use crate::tools::powm;
  44 use crate::types::{SrpAuthError, SrpGroup};
  45
  46 /// Data provided by users upon registration, usually stored in the database.
  47 pub struct UserRecord<'a> {
  48     pub username: &'a [u8],
  49     pub salt: &'a [u8],
  50     /// Password verifier
  51     pub verifier: &'a [u8],
  52 }
  53
  54 /// SRP server state
  55 pub struct SrpServer<D: Digest> {
  56     b: BigUint,
  57     a_pub: BigUint,
  58     b_pub: BigUint,
  59
  60     key: GenericArray<u8, D::OutputSize>,
  61
  62     d: PhantomData<D>,
  63 }
  64
  65 impl<D: Digest> SrpServer<D> {
  66     /// Create new server state.
  67     pub fn new(
  68         user: &UserRecord<'_>,
  69         a_pub: &[u8],
  70         b: &[u8],
  71         params: &SrpGroup,
  72     ) -> Result<Self, SrpAuthError> {
  73         let a_pub = BigUint::from_bytes_be(a_pub);
  74         // Safeguard against malicious A
  75         if &a_pub % &params.n == BigUint::zero() {
  76             return Err(SrpAuthError {
  77                 description: "Malicious a_pub value",
  78             });
  79         }
  80         let v = BigUint::from_bytes_be(user.verifier);
  81         let b = BigUint::from_bytes_be(b) % &params.n;
  82         let k = params.compute_k::<D>();
  83         // kv + g^b
  84         let interm = (k * &v) % &params.n;
  85         let b_pub = (interm + &params.powm(&b)) % &params.n;
  86         // H(A || B)
  87         let u = {
  88             let mut d = D::new();
  89             d.input(&a_pub.to_bytes_be());
  90             d.input(&b_pub.to_bytes_be());
  91             d.result()
  92         };
  93         let d = Default::default();
  94         //(Av^u) ^ b
  95         let key = {
  96             let u = BigUint::from_bytes_be(&u);
  97             let t = (&a_pub * powm(&v, &u, &params.n)) % &params.n;
  98             let s = powm(&t, &b, &params.n);
  99             D::digest(&s.to_bytes_be())
 100         };
 101         Ok(Self {
 102             b,
 103             a_pub,
 104             b_pub,
 105             key,
 106             d,
 107         })
 108     }
 109
 110     /// Get private `b` value. (see `new_with_b` documentation)
 111     pub fn get_b(&self) -> Vec<u8> {
 112         self.b.to_bytes_be()
 113     }
 114
 115     /// Get public `b_pub` value for sending to the user.
 116     pub fn get_b_pub(&self) -> Vec<u8> {
 117         self.b_pub.to_bytes_be()
 118     }
 119
 120     /// Get shared secret between user and the server. (do not forget to verify
 121     /// that keys are the same!)
 122     pub fn get_key(&self) -> GenericArray<u8, D::OutputSize> {
 123         self.key.clone()
 124     }
 125
 126     /// Process user proof of having the same shared secret and compute
 127     /// server proof for sending to the user.
 128     pub fn verify(
 129         &self,
 130         user_proof: &[u8],
 131     ) -> Result<GenericArray<u8, D::OutputSize>, SrpAuthError> {
 132         // M = H(A, B, K)
 133         let mut d = D::new();
 134         d.input(&self.a_pub.to_bytes_be());
 135         d.input(&self.b_pub.to_bytes_be());
 136         d.input(&self.key);
 137
 138         if user_proof == d.result().as_slice() {
 139             // H(A, M, K)
 140             let mut d = D::new();
 141             d.input(&self.a_pub.to_bytes_be());
 142             d.input(user_proof);
 143             d.input(&self.key);
 144             Ok(d.result())
 145         } else {
 146             Err(SrpAuthError {
 147                 description: "Incorrect user proof",
 148             })
 149         }
 150     }
 151 }