netget-tool 0.3.5

Network protocol stubbing toolkit: ~280 protocol servers and clients whose responses come from an LLM, a script, a static rule or a person.
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

//! Protocol binding defaults
//!
//! Provides a flexible system for protocols to specify default binding parameters.
//! Each protocol can define defaults for MAC address, network interface, host, and port.

/// Default binding parameters for a protocol
///
/// Protocols can return `Some(BindingDefaults)` from `default_binding()` to opt into
/// the flexible binding system. Protocols that return `None` use the legacy listen_addr system.
///
/// User-provided values override protocol defaults.
#[derive(Debug, Clone)]
pub struct BindingDefaults {
    /// MAC address for Layer 2 protocols (e.g., ARP spoofing with specific MAC)
    pub mac_address: Option<String>,

    /// Network interface for raw protocols (e.g., "lo", "eth0", "en0")
    pub interface: Option<String>,

    /// Host address (IPv4, IPv6, or hostname) for socket-based protocols
    /// Examples: "127.0.0.1", "0.0.0.0", "::", "localhost"
    pub host: Option<String>,

    /// Port number for socket-based protocols
    /// Use Some(0) for automatic port assignment
    pub port: Option<u16>,
}

impl BindingDefaults {
    /// Create binding defaults for port-based protocols (TCP, UDP, HTTP, DNS, etc.)
    ///
    /// # Arguments
    /// * `default_host` - Host to bind (e.g., "127.0.0.1", "0.0.0.0")
    /// * `default_port` - Port to bind (use 0 for automatic assignment)
    ///
    /// # Example
    /// ```ignore
    /// // TCP: bind to loopback by default, auto-assign port
    /// BindingDefaults::port_based("127.0.0.1", 0)
    /// ```
    pub fn port_based(default_host: &str, default_port: u16) -> Self {
        Self {
            mac_address: None,
            interface: None,
            host: Some(default_host.to_string()),
            port: Some(default_port),
        }
    }

    /// Create binding defaults for interface-based protocols (ICMP, ARP, DataLink, etc.)
    ///
    /// # Arguments
    /// * `default_interface` - Interface to bind (e.g., "lo", "eth0")
    ///
    /// # Example
    /// ```ignore
    /// // ICMP: bind to loopback interface by default
    /// BindingDefaults::interface_based("lo")
    /// ```
    pub fn interface_based(default_interface: &str) -> Self {
        Self {
            mac_address: None,
            interface: Some(default_interface.to_string()),
            host: None,
            port: None,
        }
    }

    /// Apply user-provided values, falling back to protocol defaults
    ///
    /// User values take precedence over defaults. Returns the final binding
    /// configuration with all fields resolved.
    ///
    /// # Arguments
    /// * `user_mac` - User-provided MAC address (overrides default)
    /// * `user_interface` - User-provided interface (overrides default)
    /// * `user_host` - User-provided host (overrides default)
    /// * `user_port` - User-provided port (overrides default)
    ///
    /// # Returns
    /// Tuple of (final_mac, final_interface, final_host, final_port)
    pub fn apply(
        &self,
        user_mac: Option<String>,
        user_interface: Option<String>,
        user_host: Option<String>,
        user_port: Option<u16>,
    ) -> (Option<String>, Option<String>, Option<String>, Option<u16>) {
        (
            user_mac.or_else(|| self.mac_address.clone()),
            user_interface.or_else(|| self.interface.clone()),
            user_host.or_else(|| self.host.clone()),
            user_port.or(self.port),
        )
    }
}