packet_builder/
udp.rs

1use pnet::packet::udp::{MutableUdpPacket};
2use pnet::packet::udp::ipv4_checksum as ipv4_udp_checksum;
3use std::net::Ipv4Addr;
4use L4Checksum;
5
6impl <'p>L4Checksum for MutableUdpPacket<'p> {
7  fn checksum_ipv4(&mut self, source: &Ipv4Addr, destination: &Ipv4Addr) -> () {
8    self.set_checksum(ipv4_udp_checksum(&self.to_immutable(), source, destination));
9  }
10}
11
12#[macro_export]
13macro_rules! udp {
14   ({$($func:ident => $value:expr), *}, $payload_pkt:expr, $protocol:expr, $buf:expr) => {{
15      const UDP_HEADER_LEN: usize = 8;
16      let total_len = UDP_HEADER_LEN + $payload_pkt.packet().len();
17      let buf_len = $buf.len();
18      let mut pkt = pnet::packet::udp::MutableUdpPacket::new(&mut $buf[buf_len - total_len..]).unwrap();
19      pkt.set_length(total_len as u16);
20      pkt.set_destination(53);
21      pkt.set_source(12345);
22      $(
23        pkt.$func($value);
24      )*
25      (pkt, pnet::packet::ip::IpNextHeaderProtocols::Udp)
26   }};
27}
28
29#[cfg(test)]
30mod tests {
31   use pnet::packet::Packet;
32   use ::payload;
33   use payload::PayloadData;
34   use udp;
35
36   #[test]
37   fn macro_udp_basic() {
38      let mut buf = [0; 13];
39      let (pkt, proto) = udp!({set_source => 53, set_destination => 5353},
40        payload!({"hello".to_string().into_bytes()}, buf).0, None, buf);
41      assert_eq!(proto, pnet::packet::ip::IpNextHeaderProtocols::Udp);
42
43      let buf_expected = vec![0; 13];
44      let mut pkt_expected = pnet::packet::udp::MutableUdpPacket::owned(buf_expected).unwrap();
45      pkt_expected.set_destination(5353); 
46      pkt_expected.set_source(53); 
47      pkt_expected.set_length(13 as u16);
48      pkt_expected.set_payload(&"hello".to_string().into_bytes()); 
49      assert_eq!(pkt_expected.packet(), pkt.packet());
50   }
51}
52
packet_builder/udp.rs

packet_builder/
udp.rs
