pnet_layers 0.1.0

A scapy like wrapper around the pnet crate to easily parse, craft or manipulate network packets in Rust.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

use crate::{ArpMut, EtherMut, LayerMut, LayerMutable, VlanMut, helper::tcp::TcpPacket};
use pnet::{
    packet::arp::{ArpHardwareTypes, ArpOperations},
    util::MacAddr,
};
use std::net::Ipv4Addr;

/// A helper struct to define the different parameters to craft a ARP packet including the Ethernet and VLAN layers
pub struct ArpPacket {
    eth_src: MacAddr,
    eth_dst: MacAddr,
    vlan_id: Option<u16>,
    ipv4_src: Ipv4Addr,
    ipv4_dst: Ipv4Addr,
}

impl ArpPacket {
    /// craft a `ArpPacket` from a `TcpPacket`
    #[must_use]
    pub fn from_tcp(target: &TcpPacket) -> ArpPacket {
        ArpPacket {
            eth_src: target.eth_src,
            eth_dst: target.eth_dst,
            vlan_id: target.vlan_id,
            ipv4_dst: target.ipv4_dst,
            ipv4_src: target.ipv4_src,
        }
    }
    /// craft an Arp Replay packet
    #[must_use]
    pub fn reply(&self) -> Option<EtherMut> {
        let mut ether = EtherMut::new();
        {
            let mut eth = ether.modify()?;
            eth.set_source(self.eth_src);
            eth.set_destination(self.eth_dst);
        }

        if let Some(vlan_id) = self.vlan_id {
            let mut vlan = VlanMut::new();
            {
                let mut pkt = vlan.modify()?;
                pkt.set_vlan_identifier(vlan_id);
            }
            ether.add(LayerMut::Vlan(vlan));
        }

        let mut arp = ArpMut::new();
        {
            let mut pkt = arp.modify()?;
            pkt.set_protocol_type(pnet::packet::ethernet::EtherTypes::Ipv4);
            pkt.set_hardware_type(ArpHardwareTypes::Ethernet);
            pkt.set_hw_addr_len(6);
            pkt.set_proto_addr_len(4);
            pkt.set_operation(ArpOperations::Reply);
            pkt.set_sender_hw_addr(self.eth_src);
            pkt.set_sender_proto_addr(self.ipv4_src);
            pkt.set_target_proto_addr(self.ipv4_dst);
            pkt.set_target_hw_addr(self.eth_dst);
        }
        ether.add(LayerMut::Arp(arp));
        Some(ether)
    }
}