push-packet 0.1.0

Packet-inspection and routing library for Linux, built on eBPF XDP and AF_XDP with aya.
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

use std::{
    collections::{HashMap, VecDeque},
    net::IpAddr,
    time::{Instant, SystemTime, UNIX_EPOCH},
};

use crate::{
    cli::Args,
    packet::{PacketInfo, ProtoFilter},
};

pub struct State {
    pub packet_info: HashMap<IpAddr, PacketInfo>,
    pub mask_seed: Option<u64>,
    pub scroll: usize,
    pub show_stale: bool,
    pub show_log: bool,
    pub show_tcp: bool,
    pub show_udp: bool,
    pub show_icmp: bool,
    pub show_v4: bool,
    pub show_v6: bool,
    pub take: u32,
    pub log: VecDeque<(Instant, String)>,
}

impl State {
    pub fn new(args: &Args) -> Self {
        let mask_seed = args.mask.then(|| {
            SystemTime::now()
                .duration_since(UNIX_EPOCH)
                .map(|d| d.as_nanos() as u64)
                .unwrap_or(0)
        });
        Self {
            packet_info: HashMap::new(),
            mask_seed,
            scroll: 0,
            show_stale: true,
            show_log: false,
            show_tcp: !args.no_tcp,
            show_udp: !args.no_udp,
            show_icmp: !args.no_icmp,
            show_v4: !args.no_v4,
            show_v6: !args.no_v6,
            take: 100,
            log: VecDeque::new(),
        }
    }

    pub fn proto_filter(&self) -> ProtoFilter {
        ProtoFilter {
            tcp: self.show_tcp,
            udp: self.show_udp,
            icmp: self.show_icmp,
        }
    }
}