ws-async 0.1.0

Asynchronous WebSocket Library
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

use hyper::header::HeaderValue;
use hyper::{Request, Response};
use rand::distributions::Alphanumeric;
use rand::{thread_rng, Rng};
use sha1::{Digest, Sha1};
use std::convert::TryFrom;
use tokio::task::{self, JoinError};

pub const SEC_WEBSOCKET_ACCEPT: &str = "Sec-WebSocket-Accept";
pub const SEC_WEBSOCKET_KEY: &str = "Sec-WebSocket-Key";

unsafe fn encode_value(bytes: &[u8]) -> HeaderValue {
    match HeaderValue::try_from(base64::encode(&bytes)) {
        Ok(hv) => hv,
        Err(_) => unreachable!(),
    }
}

pub async fn generate() -> Result<HeaderValue, JoinError> {
    task::spawn_blocking(|| {
        let mut rng = thread_rng();
        let chars: String = std::iter::repeat(())
            .map(|()| rng.sample(Alphanumeric))
            .take(16)
            .collect();

        unsafe { encode_value(chars.as_bytes()) }
    })
    .await
}

pub async fn accept(key: &HeaderValue) -> Result<HeaderValue, JoinError> {
    const GUID: &[u8] = b"258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
    let bytes = [key.as_bytes(), GUID].concat();

    task::spawn_blocking(move || {
        let mut sha1 = Sha1::default();
        sha1.input(bytes);

        unsafe { encode_value(&sha1.result()) }
    })
    .await
}