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

mod common;
use std::{net::UdpSocket, time::Duration};

use push_packet::{
    Tap,
    rules::{Action, Rule},
};
use serial_test::serial;

use crate::common::VethHarness;

#[serial]
#[test]
#[ignore]
fn route_packets() -> Result<(), Box<dyn std::error::Error>> {
    VethHarness::run(|harness| {
        let mut tap = Tap::builder(harness.veth_2())
            .rule(
                Rule::builder()
                    .source_cidr(harness.veth_1_ip())
                    .action(Action::Route),
            )
            .build()?;
        let (_tx, mut rx) = tap.route_channel()?;

        let src = format!("{}:3000", harness.veth_1_ip());
        let dst = format!("{}:3000", harness.veth_2_ip());
        let sender = harness.spawn_in_namespapce(move || {
            std::thread::sleep(Duration::from_millis(10));
            let sock = UdpSocket::bind(src).unwrap();
            for packet in ["test1", "test2"] {
                sock.send_to(packet.as_bytes(), &dst).unwrap();
            }
        });

        let mut count = 0;
        while count < 2 {
            if rx.try_recv().is_ok() {
                count += 1;
            }
        }
        sender.join().expect("sender panicked");
        Ok(())
    })
}