via 2.0.0-gm.42

An async multi-threaded web framework for people who appreciate simplicity.
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

use base64::engine::{Engine, general_purpose::STANDARD as base64};

#[cfg(feature = "aws-lc-rs")]
use aws_lc_rs::digest::{Context, SHA1_FOR_LEGACY_USE_ONLY};

#[cfg(feature = "ring")]
use ring::digest::{Context, SHA1_FOR_LEGACY_USE_ONLY};

use crate::ws::error::UpgradeError;

const WS_ACCEPT_GUID: &[u8] = b"258EAFA5-E914-47DA-95CA-C5AB0DC85B11";

pub struct Base64EncodedDigest([u8; 28]);

pub fn sha1(input: &[u8]) -> Result<Base64EncodedDigest, UpgradeError> {
    let mut hasher = Context::new(&SHA1_FOR_LEGACY_USE_ONLY);
    let mut buf = [0; 28];

    if input.is_ascii() {
        hasher.update(input);
        hasher.update(WS_ACCEPT_GUID);
        if base64.encode_slice(hasher.finish(), &mut buf).is_ok() {
            return Ok(Base64EncodedDigest(buf));
        }
    }

    Err(UpgradeError::Other)
}

impl Base64EncodedDigest {
    #[inline(always)]
    pub fn as_str(&self) -> &str {
        // Safety: Base64 is guaranteed to be ASCII and therefore, valid UTF-8.
        unsafe { str::from_utf8_unchecked(&self.0) }
    }
}