r-lanlib 0.11.0

A library crate for performing network scanning operations on any local area network (LAN)
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
100
101
102
103
104
105

use std::str::FromStr;
use std::sync::Arc;

use crate::wire::mocks::MockPacketSender;

use super::*;

#[test]
fn new() {
    let source_ip = Ipv4Addr::from_str("192.168.1.1").unwrap();
    let source_mac = MacAddr::default();
    let source_port = 54321;
    let sender: Arc<Mutex<dyn Sender>> =
        Arc::new(Mutex::new(MockPacketSender::new()));

    let heart_beat = HeartBeat::builder()
        .source_mac(source_mac)
        .source_ipv4(source_ip)
        .source_port(source_port)
        .packet_sender(sender)
        .build()
        .unwrap();

    assert_eq!(heart_beat.source_mac, source_mac);
    assert_eq!(heart_beat.source_ipv4, source_ip);
    assert_eq!(heart_beat.source_port, source_port);
}

#[test]
fn sends_heartbeat_packets() {
    let source_ip = Ipv4Addr::from_str("192.168.1.1").unwrap();
    let source_mac = MacAddr::default();
    let source_port = 54321;

    let mut packet_sender = MockPacketSender::new();

    let heartbeat_packet = HeartbeatPacketBuilder::default()
        .source_ip(source_ip)
        .source_mac(source_mac)
        .source_port(source_port)
        .build()
        .unwrap();

    let expected_packet = heartbeat_packet.to_raw();

    packet_sender
        .expect_send()
        .withf(move |p| p == expected_packet)
        .returning(|_| Ok(()));

    let sender = Arc::new(Mutex::new(packet_sender));

    let heart_beat = HeartBeat::builder()
        .source_mac(source_mac)
        .source_ipv4(source_ip)
        .source_port(source_port)
        .packet_sender(sender)
        .build()
        .unwrap();

    heart_beat.beat().unwrap();
}

#[test]
fn sends_heartbeat_packets_in_thread() {
    let source_ip = Ipv4Addr::from_str("192.168.1.1").unwrap();
    let source_mac = MacAddr::default();
    let source_port = 54321;

    let mut packet_sender = MockPacketSender::new();

    let heartbeat_packet = HeartbeatPacketBuilder::default()
        .source_ip(source_ip)
        .source_mac(source_mac)
        .source_port(source_port)
        .build()
        .unwrap();

    let expected_packet = heartbeat_packet.to_raw();

    packet_sender
        .expect_send()
        .withf(move |p| p == expected_packet)
        .returning(|_| Ok(()));

    let sender = Arc::new(Mutex::new(packet_sender));

    let heart_beat = HeartBeat::builder()
        .source_mac(source_mac)
        .source_ipv4(source_ip)
        .source_port(source_port)
        .packet_sender(sender)
        .build()
        .unwrap();

    let (stop_tx, stop_rx) = mpsc::channel();

    let handle = heart_beat.start_in_thread(stop_rx).unwrap();

    thread::sleep(Duration::from_millis(2000));

    stop_tx.send(()).unwrap();

    handle.join().unwrap().unwrap();
}