77 lines
2.5 KiB
Rust
77 lines
2.5 KiB
Rust
use crate::error::Error;
|
|
#[cfg(feature = "logging")]
|
|
use crate::log::warn;
|
|
use crate::msgs::enums::{ContentType, HandshakeType};
|
|
use crate::msgs::message::MessagePayload;
|
|
|
|
/// For a Message $m, and a HandshakePayload enum member $payload_type,
|
|
/// return Ok(payload) if $m is both a handshake message and one that
|
|
/// has the given $payload_type. If not, return Err(rustls::Error) quoting
|
|
/// $handshake_type as the expected handshake type.
|
|
macro_rules! require_handshake_msg(
|
|
( $m:expr, $handshake_type:path, $payload_type:path ) => (
|
|
match &$m.payload {
|
|
MessagePayload::Handshake { parsed: $crate::msgs::handshake::HandshakeMessagePayload {
|
|
payload: $payload_type(hm),
|
|
..
|
|
}, .. } => Ok(hm),
|
|
payload => Err($crate::check::inappropriate_handshake_message(
|
|
payload,
|
|
&[$crate::msgs::enums::ContentType::Handshake],
|
|
&[$handshake_type]))
|
|
}
|
|
)
|
|
);
|
|
|
|
/// Like require_handshake_msg, but moves the payload out of $m.
|
|
#[cfg(feature = "tls12")]
|
|
macro_rules! require_handshake_msg_move(
|
|
( $m:expr, $handshake_type:path, $payload_type:path ) => (
|
|
match $m.payload {
|
|
MessagePayload::Handshake { parsed: $crate::msgs::handshake::HandshakeMessagePayload {
|
|
payload: $payload_type(hm),
|
|
..
|
|
}, .. } => Ok(hm),
|
|
payload =>
|
|
Err($crate::check::inappropriate_handshake_message(
|
|
&payload,
|
|
&[$crate::msgs::enums::ContentType::Handshake],
|
|
&[$handshake_type]))
|
|
}
|
|
)
|
|
);
|
|
|
|
pub(crate) fn inappropriate_message(
|
|
payload: &MessagePayload,
|
|
content_types: &[ContentType],
|
|
) -> Error {
|
|
warn!(
|
|
"Received a {:?} message while expecting {:?}",
|
|
payload.content_type(),
|
|
content_types
|
|
);
|
|
Error::InappropriateMessage {
|
|
expect_types: content_types.to_vec(),
|
|
got_type: payload.content_type(),
|
|
}
|
|
}
|
|
|
|
pub(crate) fn inappropriate_handshake_message(
|
|
payload: &MessagePayload,
|
|
content_types: &[ContentType],
|
|
handshake_types: &[HandshakeType],
|
|
) -> Error {
|
|
match payload {
|
|
MessagePayload::Handshake { parsed, .. } => {
|
|
warn!(
|
|
"Received a {:?} handshake message while expecting {:?}",
|
|
parsed.typ, handshake_types
|
|
);
|
|
Error::InappropriateHandshakeMessage {
|
|
expect_types: handshake_types.to_vec(),
|
|
got_type: parsed.typ,
|
|
}
|
|
}
|
|
payload => inappropriate_message(payload, content_types),
|
|
}
|
|
}
|