diff --git a/src/common/sec_websocket_key.rs b/src/common/sec_websocket_key.rs index 8f6f7c01..84fa1eee 100644 --- a/src/common/sec_websocket_key.rs +++ b/src/common/sec_websocket_key.rs @@ -1,8 +1,19 @@ +use base64::{engine::general_purpose::STANDARD, Engine}; +use http::HeaderValue; + /// The `Sec-Websocket-Key` header. #[derive(Clone, Debug, PartialEq, Eq, Hash)] -pub struct SecWebsocketKey(pub(super) http::HeaderValue); +pub struct SecWebsocketKey(pub(super) HeaderValue); derive_header! { SecWebsocketKey(_), name: SEC_WEBSOCKET_KEY } + +impl From<[u8; 16]> for SecWebsocketKey { + fn from(bytes: [u8; 16]) -> Self { + let mut value = HeaderValue::from_str(&STANDARD.encode(bytes)).unwrap(); + value.set_sensitive(true); + Self(value) + } +}