ts_disco_protocol 0.2.0

tailscale disco protocol definition
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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127

use core::{
    cmp::Ordering,
    hash::Hash,
    net::{IpAddr, Ipv6Addr, SocketAddr, SocketAddrV6},
};

use zerocopy::NetworkEndian;

/// An endpoint included in a [`CallMeMaybe`][crate::CallMeMaybe] message: a socket address
/// on which this node believes it's reachable.
///
/// All addresses are encoded as IPv6: IPv4 is mapped.
#[derive(
    Debug,
    Copy,
    Clone,
    PartialEq,
    Eq,
    Hash,
    zerocopy::Immutable,
    zerocopy::FromBytes,
    zerocopy::IntoBytes,
    zerocopy::Unaligned,
    zerocopy::KnownLayout,
)]
#[repr(C, packed)]
pub struct Endpoint {
    addr: [zerocopy::U16<NetworkEndian>; 8],
    port: zerocopy::U16<NetworkEndian>,
}

impl Endpoint {
    /// Report the address part of this endpoint.
    ///
    /// Does not unwrap IPv4-mapped IPv6: this is just the literal value in the endpoint.
    pub const fn addr_v6(&self) -> Ipv6Addr {
        Ipv6Addr::new(
            self.addr[0].get(),
            self.addr[1].get(),
            self.addr[2].get(),
            self.addr[3].get(),
            self.addr[4].get(),
            self.addr[5].get(),
            self.addr[6].get(),
            self.addr[7].get(),
        )
    }

    /// Report the address part of this endpoint with any IPv4-in-IPv6 mapping unwrapped.
    pub const fn addr(&self) -> IpAddr {
        let addr = self.addr_v6();

        match addr.to_ipv4_mapped() {
            Some(v4) => IpAddr::V4(v4),
            None => IpAddr::V6(addr),
        }
    }

    /// Report the port part of this endpoint.
    pub const fn port(&self) -> u16 {
        self.port.get()
    }

    /// Return this endpoint as a [`SocketAddrV6`].
    pub const fn socket_addr_v6(&self) -> SocketAddrV6 {
        SocketAddrV6::new(self.addr_v6(), self.port(), 0, 0)
    }

    /// Return this endpoint as a [`SocketAddr`] with any IPv4-in-IPv6 mapping unwrapped.
    pub const fn socket_addr(&self) -> SocketAddr {
        SocketAddr::new(self.addr(), self.port())
    }

    /// Construct a new value from a socket addr.
    ///
    /// Applies IPv4 to IPv6 mapping.
    pub const fn from_socket_addr(sa: SocketAddr) -> Self {
        let ip = match sa.ip() {
            IpAddr::V4(sa) => sa.to_ipv6_mapped(),
            IpAddr::V6(sa) => sa,
        };

        Self {
            addr: zerocopy::transmute!(ip.segments()),
            port: zerocopy::U16::new(sa.port()),
        }
    }
}

impl PartialOrd for Endpoint {
    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
        Some(self.cmp(other))
    }
}

impl Ord for Endpoint {
    fn cmp(&self, other: &Self) -> Ordering {
        self.socket_addr().cmp(&other.socket_addr())
    }
}

impl From<Endpoint> for SocketAddrV6 {
    fn from(value: Endpoint) -> Self {
        value.socket_addr_v6()
    }
}

impl From<Endpoint> for SocketAddr {
    fn from(value: Endpoint) -> Self {
        value.socket_addr()
    }
}

impl From<SocketAddrV6> for Endpoint {
    fn from(value: SocketAddrV6) -> Self {
        Self {
            addr: zerocopy::transmute!(value.ip().segments()),
            port: value.port().into(),
        }
    }
}

impl From<SocketAddr> for Endpoint {
    fn from(value: SocketAddr) -> Self {
        Self::from_socket_addr(value)
    }
}