zond_engine/protocols/
mod.rs

1// Copyright (c) 2026 Erik Lening (hollowpointer) and Contributors
2//
3// This Source Code Form is subject to the terms of the Mozilla Public License, v. 2.0.
4// If a copy of the MPL was not distributed with this file, You can obtain one at
5// https://mozilla.org/MPL/2.0/.
6
7pub mod arp;
8pub mod dns;
9pub mod ethernet;
10pub mod icmp;
11pub mod ip;
12pub mod mdns;
13pub mod ndp;
14pub mod tcp;
15pub mod udp;
16pub mod utils;
17
18use pnet::packet::ethernet::{EtherTypes, EthernetPacket};
19use pnet::datalink::MacAddr;
20use std::net::{IpAddr, Ipv4Addr, Ipv6Addr};
21use crate::core::models::ip::range::Ipv4Range;
22use crate::core::models::ip::set::IpSet;
23
24type Bytes = Vec<u8>;
25type PacketIter = Box<dyn Iterator<Item = (Bytes, IpAddr)> + Send>;
26
27pub fn eth_packet_iter(
28    local_mac: &MacAddr,
29    src_v4: &Option<Ipv4Addr>,
30    link_local: &Option<Ipv6Addr>,
31    ip_set: &IpSet,
32) -> anyhow::Result<PacketIter> {
33    let arp_iter = src_v4
34        .as_ref()
35        .map(|v4| create_arp_iter(local_mac, v4, ip_set))
36        .transpose()?
37        .into_iter()
38        .flatten();
39
40    let icmp_iter = link_local
41        .as_ref()
42        .map(|v6| create_icmpv6_packets(local_mac, v6))
43        .transpose()?
44        .into_iter()
45        .flatten();
46
47    Ok(Box::new(arp_iter.chain(icmp_iter)))
48}
49
50pub fn create_arp_iter(local_mac: &MacAddr, src_ip: &Ipv4Addr, ip_set: &IpSet) -> anyhow::Result<PacketIter> {
51    let local_mac = *local_mac;
52    let src_ip = *src_ip;
53    let dst_mac = MacAddr::broadcast();
54
55    let ranges: Vec<Ipv4Range> = ip_set.v4().to_vec();
56
57    let iter = ranges
58        .into_iter()
59        .flat_map(|range| {
60            let start: u32 = range.start_addr.into();
61            let end: u32 = range.end_addr.into();
62            (start..=end).map(Ipv4Addr::from)
63        })
64        .map(move |dst_addr| {
65            let packet = arp::create_packet(&local_mac, dst_mac, &src_ip, dst_addr)
66                .expect("Failed to create ARP packet");
67
68            (packet, IpAddr::V4(dst_addr))
69        });
70
71    Ok(Box::new(iter))
72}
73
74fn create_icmpv6_packets(local_mac: &MacAddr, link_local_addr: &Ipv6Addr) -> anyhow::Result<PacketIter> {
75    let packet: Vec<u8> = icmp::create_all_nodes_echo_request_v6(local_mac, link_local_addr)?;
76
77    let iter = std::iter::once((packet, IpAddr::V6(*link_local_addr)));
78
79    Ok(Box::new(iter))
80}
81
82pub fn get_ip_addr_from_eth(frame: &EthernetPacket) -> anyhow::Result<IpAddr> {
83    match frame.get_ethertype() {
84        EtherTypes::Arp => Ok(IpAddr::V4(arp::get_ipv4_addr_from_eth(frame)?)),
85        EtherTypes::Ipv4 => Ok(IpAddr::V4(ip::get_ipv4_addr_from_eth(frame)?)),
86        EtherTypes::Ipv6 => Ok(IpAddr::V6(ip::get_ipv6_src_addr_from_eth(frame)?)),
87        _ => Err(anyhow::anyhow!(
88            "Unsupported EtherType: {:?}",
89            frame.get_ethertype()
90        )),
91    }
92}
zond_engine/protocols/mod.rs

zond_engine/protocols/
mod.rs
