1use crate::enums::{ContentType, HandshakeType};
2use crate::error::Error;
3use crate::log::warn;
4use crate::msgs::message::MessagePayload;
5
6/// For a Message $m, and a HandshakePayload enum member $payload_type,
7/// return Ok(payload) if $m is both a handshake message and one that
8/// has the given $payload_type. If not, return Err(rustls::Error) quoting
9/// $handshake_type as the expected handshake type.
10macro_rules! require_handshake_msg(
11 ( $m:expr, $handshake_type:path, $payload_type:path ) => (
12 match &$m.payload {
13 MessagePayload::Handshake { parsed: $crate::msgs::handshake::HandshakeMessagePayload {
14 payload: $payload_type(hm),
15 ..
16 }, .. } => Ok(hm),
17 payload => Err($crate::check::inappropriate_handshake_message(
18 payload,
19 &[$crate::ContentType::Handshake],
20 &[$handshake_type]))
21 }
22 )
23);
24
25/// Like require_handshake_msg, but moves the payload out of $m.
26macro_rules! require_handshake_msg_move(
27 ( $m:expr, $handshake_type:path, $payload_type:path ) => (
28 match $m.payload {
29 MessagePayload::Handshake { parsed: $crate::msgs::handshake::HandshakeMessagePayload {
30 payload: $payload_type(hm),
31 ..
32 }, .. } => Ok(hm),
33 payload =>
34 Err($crate::check::inappropriate_handshake_message(
35 &payload,
36 &[$crate::ContentType::Handshake],
37 &[$handshake_type]))
38 }
39 )
40);
41
42pub(crate) fn inappropriate_message(
43 payload: &MessagePayload<'_>,
44 content_types: &[ContentType],
45) -> Error {
46 warn!(
47 "Received a {:?} message while expecting {:?}",
48 payload.content_type(),
49 content_types
50 );
51 Error::InappropriateMessage {
52 expect_types: content_types.to_vec(),
53 got_type: payload.content_type(),
54 }
55}
56
57pub(crate) fn inappropriate_handshake_message(
58 payload: &MessagePayload<'_>,
59 content_types: &[ContentType],
60 handshake_types: &[HandshakeType],
61) -> Error {
62 match payload {
63 MessagePayload::Handshake { parsed: &HandshakeMessagePayload<'_>, .. } => {
64 warn!(
65 "Received a {:?} handshake message while expecting {:?}",
66 parsed.typ, handshake_types
67 );
68 Error::InappropriateHandshakeMessage {
69 expect_types: handshake_types.to_vec(),
70 got_type: parsed.typ,
71 }
72 }
73 payload: &MessagePayload<'_> => inappropriate_message(payload, content_types),
74 }
75}
76