qsocket 0.1.3

QSocket library.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

use crate::QSocketError;
use spake2::{Ed25519Group, Identity, Password, Spake2};
use std::io::{Read, Write};

pub enum PakeMode {
    Client,
    Server,
}

pub fn init_pake_handshake<S>(
    mode: PakeMode,
    mut sock: S,
    password: String,
) -> Result<Vec<u8>, QSocketError>
where
    S: Read + Write + std::marker::Send,
{
    let pass_digest = md5::compute(password.clone()).to_vec();
    #[allow(unused_assignments)]
    let mut state: Option<Spake2<Ed25519Group>> = None;
    #[allow(unused_assignments)]
    let mut outbound_msg: Vec<u8> = Vec::new();

    match mode {
        PakeMode::Client => {
            let (s, o) = Spake2::<Ed25519Group>::start_a(
                &Password::new(pass_digest.clone()),
                &Identity::new(&pass_digest[..4]),
                &Identity::new(&pass_digest[..4]),
            );
            state = Some(s);
            outbound_msg = o;
        }
        PakeMode::Server => {
            let (s, o) = Spake2::<Ed25519Group>::start_b(
                &Password::new(pass_digest.clone()),
                &Identity::new(&pass_digest[..4]),
                &Identity::new(&pass_digest[..4]),
            );
            state = Some(s);
            outbound_msg = o;
        }
    }

    sock.write_all(&outbound_msg)?;
    let mut inbound_msg = vec![0; 1024];
    let n = sock.read(&mut inbound_msg)?;
    let key = state.unwrap().finish(&inbound_msg[..n]);
    if key.is_err() {
        return Err(QSocketError::PakeError);
    }
    Ok(key.unwrap())
}