bgpkit_parser/parser/bmp/
mod.rs

1/*!
2Provides parsing for BMP and OpenBMP binary-formatted messages.
3*/
4use crate::parser::bmp::error::ParserBmpError;
5use crate::parser::bmp::messages::*;
6pub use crate::parser::bmp::openbmp::parse_openbmp_header;
7use crate::utils::ReadUtils;
8use bytes::Bytes;
9
10pub mod error;
11pub mod messages;
12pub mod openbmp;
13
14/// Parse OpenBMP `raw_bmp` message.
15///
16/// An OpenBMP `raw_bmp` message contains a [OpenBmpHeader][OpenBmpHeader] and a [BmpMessage].
17///
18/// [OpenBmpHeader]: crate::parser::bmp::openbmp::OpenBmpHeader
19pub fn parse_openbmp_msg(mut data: Bytes) -> Result<BmpMessage, ParserBmpError> {
20    let _header = parse_openbmp_header(&mut data)?;
21    parse_bmp_msg(&mut data)
22}
23
24/// Parse a BMP message.
25pub fn parse_bmp_msg(data: &mut Bytes) -> Result<BmpMessage, ParserBmpError> {
26    let common_header = parse_bmp_common_header(data)?;
27
28    let content_length = common_header.msg_len as usize - 6;
29    data.has_n_remaining(content_length)?;
30    let mut content = data.split_to(content_length);
31
32    // if total_len>common_header.msg_len {
33    //     // truncated message
34    //     let diff = total_len - common_header.msg_len;
35    //     let bytes_left = reader.get_ref().len() as u32 - reader.position() as u32;
36    //     buffer = Vec::with_capacity((bytes_left - diff) as usize);
37    // } else if total_len == common_header.msg_len {
38    // } else {
39    //     return Err(ParserBmpError::CorruptedBmpMessage)
40    // };
41    // reader.read_exact(&mut buffer)?;
42
43    match &common_header.msg_type {
44        BmpMsgType::RouteMonitoring => {
45            let per_peer_header = parse_per_peer_header(&mut content)?;
46            let msg = parse_route_monitoring(&mut content, &per_peer_header.asn_length())?;
47            Ok(BmpMessage {
48                common_header,
49                per_peer_header: Some(per_peer_header),
50                message_body: BmpMessageBody::RouteMonitoring(msg),
51            })
52        }
53        BmpMsgType::RouteMirroringMessage => {
54            let per_peer_header = parse_per_peer_header(&mut content)?;
55            let msg = parse_route_mirroring(&mut content, &per_peer_header.asn_length())?;
56            Ok(BmpMessage {
57                common_header,
58                per_peer_header: Some(per_peer_header),
59                message_body: BmpMessageBody::RouteMirroring(msg),
60            })
61        }
62        BmpMsgType::StatisticsReport => {
63            let per_peer_header = parse_per_peer_header(&mut content)?;
64            let msg = parse_stats_report(&mut content)?;
65            Ok(BmpMessage {
66                common_header,
67                per_peer_header: Some(per_peer_header),
68                message_body: BmpMessageBody::StatsReport(msg),
69            })
70        }
71        BmpMsgType::PeerDownNotification => {
72            let per_peer_header = parse_per_peer_header(&mut content)?;
73            let msg = parse_peer_down_notification(&mut content)?;
74            Ok(BmpMessage {
75                common_header,
76                per_peer_header: Some(per_peer_header),
77                message_body: BmpMessageBody::PeerDownNotification(msg),
78            })
79        }
80        BmpMsgType::PeerUpNotification => {
81            let per_peer_header = parse_per_peer_header(&mut content)?;
82            let msg = parse_peer_up_notification(
83                &mut content,
84                &per_peer_header.afi(),
85                &per_peer_header.asn_length(),
86            )?;
87            Ok(BmpMessage {
88                common_header,
89                per_peer_header: Some(per_peer_header),
90                message_body: BmpMessageBody::PeerUpNotification(msg),
91            })
92        }
93        BmpMsgType::InitiationMessage => {
94            let msg = parse_initiation_message(&mut content)?;
95            Ok(BmpMessage {
96                common_header,
97                per_peer_header: None,
98                message_body: BmpMessageBody::InitiationMessage(msg),
99            })
100        }
101        BmpMsgType::TerminationMessage => {
102            let msg = parse_termination_message(&mut content)?;
103            Ok(BmpMessage {
104                common_header,
105                per_peer_header: None,
106                message_body: BmpMessageBody::TerminationMessage(msg),
107            })
108        }
109    }
110}
111
112#[cfg(test)]
113#[allow(unused_variables)]
114mod tests {
115    use super::*;
116    use crate::parser::bmp::openbmp::parse_openbmp_header;
117
118    #[test]
119    fn test_peer_down_notification() {
120        let input = "4f424d500107006400000033800c6184b9c2000c602cbf4f072f3ae149d23486024bc3dadfc4000a69732d63632d626d7031c677060bdd020a9e92be000200de2e3180df3369000000000000000000000000000c726f7574652d76696577733500000001030000003302000000000000000000000000000000000000000000003fda060e00000da30000000061523c36000c0e1c0200000a";
121        let decoded = hex::decode(input).unwrap();
122        let mut data = Bytes::from(decoded);
123        let _header = parse_openbmp_header(&mut data).unwrap();
124        let _msg = parse_bmp_msg(&mut data).unwrap();
125    }
126
127    #[test]
128    fn test_route_monitoring() {
129        let input = "4f424d500107005c000000b0800c618881530002f643fef880938d19e9d632c815d1e95a87e1000a69732d61682d626d7031eb4de4e596b282c6a995b067df4abc8cc342f19200000000000000000000000000046c696e780000000103000000b00000c00000000000000000200107f800040000000000001aae000400001aae5474800e02dddf5d00000000ffffffffffffffffffffffffffffffff00800200000069400101005002001602050000192f00001aae0000232a000328eb00032caec008181aae42681aae44581aae464f1aae59d91aae866543000000900e002c00020120200107f800040000000000001aae0004fe8000000000000082711ffffe7f29f100302a0fca8000010a";
130        let decoded = hex::decode(input).unwrap();
131        let mut data = Bytes::from(decoded);
132        let _header = parse_openbmp_header(&mut data).unwrap();
133        let _msg = parse_bmp_msg(&mut data).unwrap();
134    }
135
136    #[test]
137    fn parse_peer_up_notification() {}
138}
bgpkit_parser/parser/bmp/mod.rs

bgpkit_parser/parser/bmp/
mod.rs
