pktmon 0.6.2

A library for capturing network packets on Windows using the PktMon service
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

use std::{
    sync::{
        Arc, Mutex,
        atomic::{AtomicBool, Ordering},
    },
    thread,
    time::Duration,
};

use pktmon::{
    Capture,
    filter::{PktMonFilter, TransportProtocol},
};

fn main() {
    colog::default_builder()
        .filter_level(log::LevelFilter::Trace)
        .init();

    let mut sniffer = Capture::new().unwrap();

    sniffer
        .add_filter(PktMonFilter {
            name: "UDP Traffic".to_string(),
            transport_protocol: Some(TransportProtocol::UDP),

            ..PktMonFilter::default()
        })
        .unwrap();

    sniffer.start().unwrap();

    let sniffer = Arc::new(Mutex::new(sniffer));
    let running = Arc::new(AtomicBool::new(true));

    let join_handle = {
        let running = running.clone();
        let sniffer = sniffer.clone();
        thread::spawn(move || {
            loop {
                let sniffer = sniffer.lock().unwrap();
                if let Ok(packet) = sniffer.next_packet_timeout(Duration::from_secs(1)) {
                    println!("{:?}", packet.payload);
                }

                if !running.load(Ordering::Relaxed) {
                    break;
                }
            }
        })
    };

    {
        let running = running.clone();
        let sniffer = sniffer.clone();
        std::thread::sleep(std::time::Duration::from_secs(5));

        running.store(false, Ordering::Relaxed);
        let mut sniffer = sniffer.lock().unwrap();
        sniffer.stop().unwrap();
    }

    join_handle.join().unwrap();

    let sniffer = Arc::try_unwrap(sniffer).unwrap().into_inner().unwrap();
    sniffer.unload().unwrap();
}