rnc 0.1.1

arbitrary TCP and UDP connections and listens
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

use std::error::Error;
use std::fmt;
use std::net::{AddrParseError, IpAddr, Ipv6Addr, SocketAddr};
use std::str::FromStr;

use crate::cli;

/// This enum represents a host parsing error.
///
/// # Variants
///
/// * `AddrParse(AddrParseError)` - An `AddrParseError` occurred.
/// * `NoHostName` - No hostname was specified.
#[derive(Debug)]
pub enum HostParseError {
    AddrParse(AddrParseError),
    NoHostName,
}

/// This function parses the socket address from the command line arguments.
///
/// If the hostname is not specified, and the program is listening, then
/// the hostname is set to `::` (IPv6 unspecified address).
///
/// If the hostname is not specified, and the program is connecting, then
/// an error is returned.
///
/// # Arguments
///
/// * `cli` - The command line arguments.
///
/// # Returns
///
/// * `Result<SocketAddr, HostParseError>` - The parsed socket address.
pub fn parse_socket_addr(cli: &cli::Cli) -> Result<SocketAddr, HostParseError> {
    let port = parse_port(cli.port);

    match &cli.hostname {
        None => {
            if cli.listen {
                Ok(SocketAddr::new(IpAddr::V6(Ipv6Addr::UNSPECIFIED), port))
            } else {
                Err(HostParseError::NoHostName)
            }
        }
        Some(hostname) => Ok(SocketAddr::from_str(&format!("{}:{}", hostname, port))?),
    }
}

/// This function parses the port from the command line arguments.
///
/// If the port is not specified, then the port is set to 31337.
///
/// # Arguments
///
/// * `port` - The port to parse.
///
/// # Returns
///
/// * `u16` - The parsed port.
fn parse_port(port: Option<u16>) -> u16 {
    match port {
        Some(port) => port,
        None => 31337,
    }
}

impl Error for HostParseError {}

impl fmt::Display for HostParseError {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        match self {
            HostParseError::AddrParse(e) => write!(f, "AddrParseError: {}", e),
            HostParseError::NoHostName => write!(f, "No hostname specified"),
        }
    }
}

impl From<AddrParseError> for HostParseError {
    fn from(error: AddrParseError) -> Self {
        HostParseError::AddrParse(error)
    }
}