srp/src/utils.rs

   1 use digest::{Digest, Output};
   2 use num_bigint::BigUint;
   3
   4 use crate::types::SrpGroup;
   5
   6 // u = H(PAD(A) | PAD(B))
   7 pub fn compute_u<D: Digest>(a_pub: &[u8], b_pub: &[u8]) -> BigUint {
   8     let mut u = D::new();
   9     u.update(a_pub);
  10     u.update(b_pub);
  11     BigUint::from_bytes_be(&u.finalize())
  12 }
  13
  14 // k = H(N | PAD(g))
  15 pub fn compute_k<D: Digest>(params: &SrpGroup) -> BigUint {
  16     let n = params.n.to_bytes_be();
  17     let g_bytes = params.g.to_bytes_be();
  18     let mut buf = vec![0u8; n.len()];
  19     let l = n.len() - g_bytes.len();
  20     buf[l..].copy_from_slice(&g_bytes);
  21
  22     let mut d = D::new();
  23     d.update(&n);
  24     d.update(&buf);
  25     BigUint::from_bytes_be(d.finalize().as_slice())
  26 }
  27
  28 // M1 = H(H(N) XOR H(g) | H(U) | s | A | B | K)
  29 pub fn compute_m1<D: Digest>(
  30     params: &SrpGroup,
  31     identity_hash: &[u8],
  32     salt: &[u8],
  33     a_pub: &[u8],
  34     b_pub: &[u8],
  35     key: &[u8],
  36 ) -> Output<D> {
  37     let mut d_n = D::new();
  38     d_n.update(params.n.to_bytes_be());
  39     let h_n = d_n.finalize();
  40
  41     let mut d_g = D::new();
  42     d_g.update(params.g.to_bytes_be());
  43     let h_g = d_g.finalize();
  44
  45     let ng_xor: Vec<u8> = h_n.iter().zip(h_g.iter()).map(|(n, g)| n ^ g).collect();
  46
  47     let mut d = D::new();
  48     d.update(ng_xor);
  49     d.update(identity_hash);
  50     d.update(salt);
  51     d.update(a_pub);
  52     d.update(b_pub);
  53     d.update(key);
  54     d.finalize()
  55 }
  56
  57 // M2 = H(A, M1, K)
  58 pub fn compute_m2<D: Digest>(a_pub: &[u8], m1: &Output<D>, key: &[u8]) -> Output<D> {
  59     let mut d = D::new();
  60     d.update(a_pub);
  61     d.update(m1);
  62     d.update(key);
  63     d.finalize()
  64 }