packet_builder/
icmp.rs

1use pnet::packet::util::checksum as generic_checksum;
2use pnet::packet::Packet;
3use std::net::Ipv4Addr;
4use L4Checksum;
5
6macro_rules! icmp_pkt_macro_generator {
7  ($($name:ident => $icmp_type:ty), *) => {
8    $(
9      #[macro_export]
10      macro_rules! $name {
11        ($args:tt, $payload_pkt:expr, $proto:expr, $buf:expr) => {{
12          icmp!($args, $payload_pkt, $icmp_type, $buf) 
13        }};
14        ($args:tt, $buf:expr) => {{
15          icmp!($args, $icmp_type, $buf) 
16        }}; 
17      }
18    )*
19  };
20}
21
22macro_rules! icmp_checksum_func_gen {
23  ($($icmp_type:ty),*) => {
24    $(
25      impl <'p>L4Checksum for $icmp_type {
26        fn checksum_ipv4(&mut self, _source: &Ipv4Addr, _destination: &Ipv4Addr) {
27          // ICMP checksum is the same as IP
28          self.set_checksum(generic_checksum(&self.packet(), 1)); 
29        }
30      }
31    )*
32  };
33}
34
35icmp_checksum_func_gen!(pnet::packet::icmp::echo_reply::MutableEchoReplyPacket<'p>, 
36                        pnet::packet::icmp::echo_request::MutableEchoRequestPacket<'p>, 
37                        pnet::packet::icmp::destination_unreachable::MutableDestinationUnreachablePacket<'p>, 
38                        pnet::packet::icmp::time_exceeded::MutableTimeExceededPacket<'p>); 
39
40icmp_pkt_macro_generator!(icmp_echo_req => pnet::packet::icmp::echo_request::MutableEchoRequestPacket,
41                          icmp_echo_reply => pnet::packet::icmp::echo_reply::MutableEchoReplyPacket,
42                          icmp_dest_unreach => pnet::packet::icmp::destination_unreachable::MutableDestinationUnreachablePacket,
43                          icmp_time_exceed => pnet::packet::icmp::time_exceeded::MutableTimeExceededPacket);
44
45#[macro_export]
46macro_rules! icmp {
47   ({$($func:ident => $value:expr), *}, $icmp_type:ty, $buf:expr) => {{
48      let total_len = <$icmp_type>::minimum_packet_size();
49      let buf_len = $buf.len();
50      let mut pkt = <$icmp_type>::new(&mut $buf[buf_len - total_len..]).unwrap();
51      pkt.set_icmp_type(IcmpTypes::EchoRequest);
52      $(
53        pkt.$func($value);
54      )*
55      (pkt, pnet::packet::ip::IpNextHeaderProtocols::Icmp)
56   }};
57   ({$($func:ident => $value:expr), *}, $payload_pkt:expr, $icmp_type:ty, $buf:expr) => {{
58      let total_len = <$icmp_type>::minimum_packet_size() + $payload_pkt.packet().len();
59      let buf_len = $buf.len();
60      let mut pkt = <$icmp_type>::new(&mut $buf[buf_len - total_len..]).unwrap();
61      pkt.set_icmp_type(IcmpTypes::EchoRequest);
62      $(
63        pkt.$func($value);
64      )*
65      (pkt, pnet::packet::ip::IpNextHeaderProtocols::Icmp)
66   }};
67}
68
69
70#[cfg(test)]
71mod tests {
72   use pnet::packet::Packet;
73   use ::payload;
74   use payload::PayloadData;
75   use pnet::packet::icmp::{IcmpTypes};
76   use icmp;
77
78   #[test]
79   fn macro_icmp_basic() {
80      let mut buf = [0; 13];
81      let (pkt, proto) = icmp_dest_unreach!({set_icmp_type => IcmpTypes::DestinationUnreachable},
82        payload!({"hello".to_string().into_bytes()}, buf).0, None, buf);
83      assert_eq!(proto, pnet::packet::ip::IpNextHeaderProtocols::Icmp);
84
85      let buf_expected = vec![0; 13];
86      let mut pkt_expected = pnet::packet::icmp::destination_unreachable::MutableDestinationUnreachablePacket::owned(buf_expected).unwrap();
87      pkt_expected.set_icmp_type(IcmpTypes::DestinationUnreachable);
88      pkt_expected.set_payload(&"hello".to_string().into_bytes()); 
89      assert_eq!(pkt_expected.packet(), pkt.packet());
90   }
91}
92
packet_builder/icmp.rs

packet_builder/
icmp.rs
