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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99

use ipnet::IpNet;
use push_packet_common::engine::linear::{
    FLAG_DESTINATION_CIDR, FLAG_DESTINATION_PORT, FLAG_PROTOCOL, FLAG_SOURCE_CIDR,
    FLAG_SOURCE_PORT, Ipv4Rule, Ipv6Rule, RuleExt,
};

use crate::rules::Rule;

impl From<&Rule> for Ipv4Rule {
    fn from(value: &Rule) -> Self {
        let Rule {
            action,
            protocol,
            source_cidr,
            source_port,
            destination_cidr,
            destination_port,
        } = value;
        let mut rule = Ipv4Rule::default();
        let (action, take) = (*action).into_common_action();
        rule.common.action = action;
        if let Some(take) = take {
            rule.common.take = take;
        }
        if let Some(protocol) = protocol {
            rule.common.protocol = *protocol;
            rule.set_flag(FLAG_PROTOCOL);
        }
        if let Some(IpNet::V4(source_cidr)) = source_cidr {
            rule.source_cidr = source_cidr.addr().into();
            rule.common.source_prefix_len = source_cidr.prefix_len();
            rule.set_flag(FLAG_SOURCE_CIDR);
        }

        if let Some(source_port) = source_port {
            rule.common.source_port_min = *source_port.start();
            rule.common.source_port_max = *source_port.end();
            rule.set_flag(FLAG_SOURCE_PORT);
        }
        if let Some(IpNet::V4(destination_cidr)) = destination_cidr {
            rule.destination_cidr = destination_cidr.addr().into();
            rule.common.destination_prefix_len = destination_cidr.prefix_len();
            rule.set_flag(FLAG_DESTINATION_CIDR);
        }

        if let Some(destination_port) = destination_port {
            rule.common.destination_port_min = *destination_port.start();
            rule.common.destination_port_max = *destination_port.end();
            rule.set_flag(FLAG_DESTINATION_PORT);
        }
        rule
    }
}

impl From<&Rule> for Ipv6Rule {
    fn from(value: &Rule) -> Self {
        let Rule {
            action,
            protocol,
            source_cidr,
            source_port,
            destination_cidr,
            destination_port,
        } = value;
        let mut rule = Ipv6Rule::default();
        let (action, take) = action.into_common_action();
        rule.common.action = action;
        if let Some(take) = take {
            rule.common.take = take;
        }
        if let Some(protocol) = protocol {
            rule.common.protocol = *protocol;
            rule.set_flag(FLAG_PROTOCOL);
        }
        if let Some(IpNet::V6(source_cidr)) = source_cidr {
            rule.source_cidr = source_cidr.addr().octets();
            rule.common.source_prefix_len = source_cidr.prefix_len();
            rule.set_flag(FLAG_SOURCE_CIDR);
        }

        if let Some(source_port) = source_port {
            rule.common.source_port_min = *source_port.start();
            rule.common.source_port_max = *source_port.end();
            rule.set_flag(FLAG_SOURCE_PORT);
        }
        if let Some(IpNet::V6(destination_cidr)) = destination_cidr {
            rule.destination_cidr = destination_cidr.addr().octets();
            rule.common.destination_prefix_len = destination_cidr.prefix_len();
            rule.set_flag(FLAG_DESTINATION_CIDR);
        }

        if let Some(destination_port) = destination_port {
            rule.common.destination_port_min = *destination_port.start();
            rule.common.destination_port_max = *destination_port.end();
            rule.set_flag(FLAG_DESTINATION_PORT);
        }
        rule
    }
}