use std::fmt::{Display, Formatter};
#[derive(Debug, PartialEq, Eq)]
pub enum HandshakeError {
HttpVersion,
HttpMethod,
HttpSatusCode,
HttpHost,
Upgrade,
Connection,
SecWebSocketKey,
SecWebSocketAccept,
SecWebSocketVersion,
NotEnoughData,
NotEnoughCapacity,
Httparse(httparse::Error),
Manual(&'static str),
}
impl Display for HandshakeError {
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
use HandshakeError::*;
match self {
HttpVersion => write!(f, "Illegal http version"),
HttpMethod => write!(f, "Illegal http method"),
HttpSatusCode => write!(f, "Illegal http status code"),
HttpHost => write!(f, "Missing http host header"),
Upgrade => write!(f, "Missing or illegal upgrade header"),
Connection => write!(f, "Missing or illegal connection header"),
SecWebSocketKey => {
write!(f, "Missing sec-websocket-key header")
}
SecWebSocketAccept => {
write!(f, "Missing or illegal sec-websocket-accept header")
}
SecWebSocketVersion => {
write!(f, "Missing or illegal sec-websocket-version")
}
NotEnoughData => write!(f, "Not enough data to parse"),
NotEnoughCapacity => write!(f, "Not enough space to write to"),
Httparse(e) => write!(f, "Http parse error: {}", e),
Manual(s) => write!(f, "Manual error: {}", s),
}
}
}
impl From<httparse::Error> for HandshakeError {
fn from(e: httparse::Error) -> Self { HandshakeError::Httparse(e) }
}
impl std::error::Error for HandshakeError {
fn source(&self) -> Option<&(dyn std::error::Error + 'static)> {
if let HandshakeError::Httparse(e) = self {
Some(e)
} else {
None
}
}
}