Skip to main content

ts_control/
node.rs

1//! The parsed domain [`Node`] model: a tailnet node decoded from the wire (`tailcfg.Node`).
2//!
3//! [`Node`] is the owned, validated form the rest of the fork reasons about (addresses, keys, caps,
4//! accepted routes, peerAPI/VIP services), built from the borrow-bound `ts_control_serde::Node` via
5//! the [`From`] impl. It also carries the route/exit-node/funnel predicates ([`Node::is_subnet_route`],
6//! [`Node::routes_to_install`], [`Node::can_funnel`]) and the [`ExitNodeSelector`] resolution.
7//!
8//! Fail-closed: route, funnel, and service-host gates all deny on a missing/malformed input.
9
10use core::net::{IpAddr, Ipv4Addr, Ipv6Addr, SocketAddr};
11use std::collections::BTreeMap;
12
13use chrono::{DateTime, Utc};
14use ts_capabilityversion::CapabilityVersion;
15use ts_keys::{DiscoPublicKey, MachinePublicKey, NodePublicKey};
16
17use crate::dns::Resolver;
18
19/// An owned node-capability map (`Node.CapMap` in Go: `map[NodeCapability][]RawMessage`).
20///
21/// Keys are capability names or URLs (e.g. `"funnel"`, `"https"`, or
22/// `"https://tailscale.com/cap/funnel-ports?ports=443,8443"`); values are the raw JSON-encoded
23/// argument blobs for that capability (often empty). Stored *owned* because the wire form
24/// ([`ts_control_serde::Node::cap_map`]) borrows from the decode buffer, whereas the domain
25/// [`Node`] outlives it. Funnel gating only inspects the keys (see [`Node::can_funnel`] and
26/// [`Node::check_funnel_port`]); the values are retained for capabilities that carry argument data.
27pub type NodeCapMap = BTreeMap<String, Vec<String>>;
28
29/// Whether `addr` falls in a range Tailscale assigns to nodes: the CGNAT range for IPv4
30/// (`100.64.0.0/10`, excluding the ChromeOS VM carve-out `100.115.92.0/23`) and the Tailscale
31/// ULA for IPv6 (`fd7a:115c:a1e0::/48`).
32///
33/// Mirrors `tsaddr.IsTailscaleIP` in the Go client. Used to tell a peer's own node addresses
34/// (always single Tailscale IPs) apart from the larger subnet routes it advertises.
35pub fn is_tailscale_ip(addr: IpAddr) -> bool {
36    match addr {
37        IpAddr::V4(v4) => {
38            let cgnat = ipnet::Ipv4Net::new(Ipv4Addr::new(100, 64, 0, 0), 10).unwrap();
39            let chromeos = ipnet::Ipv4Net::new(Ipv4Addr::new(100, 115, 92, 0), 23).unwrap();
40            cgnat.contains(&v4) && !chromeos.contains(&v4)
41        }
42        IpAddr::V6(v6) => {
43            let ula = ipnet::Ipv6Net::new(Ipv6Addr::new(0xfd7a, 0x115c, 0xa1e0, 0, 0, 0, 0, 0), 48)
44                .unwrap();
45            ula.contains(&v6)
46        }
47    }
48}
49
50/// The unique id of a node.
51pub type Id = i64;
52
53/// The stable ID of a node.
54#[derive(
55    Debug, Clone, PartialEq, Eq, Hash, PartialOrd, Ord, serde::Serialize, serde::Deserialize,
56)]
57pub struct StableId(pub String);
58
59/// How this node selects which peer to use as its exit node (`--exit-node` in the Go client).
60///
61/// Mirrors the Go client's `--exit-node`, which accepts a tailnet IP, a MagicDNS name, or a stable
62/// node ID, and resolves it to a `StableNodeID` (`resolveExitNodeIPLocked`). We keep the selector
63/// *unresolved* and re-run [`ExitNodeSelector::resolve`] against the live peer set on every route
64/// rebuild, so an IP- or name-based selection follows the peer as the netmap changes (e.g. the
65/// exit node re-registers under a new stable id).
66///
67/// A selector can be parsed from a string with [`str::parse`]/[`FromStr`](core::str::FromStr),
68/// auto-detecting the variant the way the Go CLI's `--exit-node` does: a value that parses as an IP
69/// address becomes [`ExitNodeSelector::Ip`], anything else becomes [`ExitNodeSelector::Name`].
70/// Stable-id selection is available only by constructing [`ExitNodeSelector::StableId`] directly
71/// (it is not auto-detected, since a stable id is otherwise indistinguishable from a hostname).
72#[derive(Debug, Clone, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
73pub enum ExitNodeSelector {
74    /// Select the peer with this exact stable node id.
75    StableId(StableId),
76    /// Select the peer whose tailnet address is this IP.
77    Ip(IpAddr),
78    /// Select the peer matching this bare hostname or MagicDNS name (case-insensitive, optional
79    /// trailing dot), as per [`Node::matches_name`].
80    Name(String),
81}
82
83impl core::str::FromStr for ExitNodeSelector {
84    type Err = core::convert::Infallible;
85
86    /// Parse a selector from a string, auto-detecting IP vs. name (matching the Go CLI's
87    /// `--exit-node`). Parsing never fails: a non-IP string is taken as a MagicDNS name.
88    fn from_str(s: &str) -> Result<Self, Self::Err> {
89        Ok(match s.parse::<IpAddr>() {
90            Ok(ip) => ExitNodeSelector::Ip(ip),
91            Err(_) => ExitNodeSelector::Name(s.to_owned()),
92        })
93    }
94}
95
96impl ExitNodeSelector {
97    /// Resolve this selector to the stable id of the matching peer, if any, given the current set
98    /// of peers.
99    ///
100    /// Resolution is **deterministic**: if a selector somehow matches more than one peer (e.g. two
101    /// peers sharing a MagicDNS name during a transient netmap state), the peer with the smallest
102    /// [`StableId`] is chosen. This matters because both the outbound route table and the inbound
103    /// source filter resolve independently; a deterministic tiebreak guarantees they pick the
104    /// *same* peer, preserving the cryptokey-routing coupling that prevents source-spoofing.
105    ///
106    /// Returns `None` when no peer matches (a stale/typo'd selector). Callers treat `None` as
107    /// fail-closed: no peer is granted a default route, so internet-bound traffic is dropped.
108    pub fn resolve<'a>(&self, peers: impl Iterator<Item = &'a Node>) -> Option<StableId> {
109        peers
110            .filter(|node| match self {
111                ExitNodeSelector::StableId(id) => &node.stable_id == id,
112                ExitNodeSelector::Ip(ip) => node.tailnet_address.contains(*ip),
113                ExitNodeSelector::Name(name) => node.matches_name(name),
114            })
115            .map(|node| &node.stable_id)
116            .min()
117            .cloned()
118    }
119}
120
121/// A node in a tailnet.
122#[derive(Debug, Clone, PartialEq, Eq, Hash)]
123pub struct Node {
124    /// The node's id.
125    pub id: Id,
126    /// The node's stable id.
127    pub stable_id: StableId,
128
129    /// This node's hostname.
130    pub hostname: String,
131
132    /// The integer id of the user that owns this node (`Node.User` in Go). `0` when control sends
133    /// no owner (e.g. tagged/ACL nodes have no human owner). Join against the netmap's
134    /// `UserProfiles` table (accumulated by the runtime's peer tracker) to resolve a login/display
135    /// name — see the runtime `WhoIs` lookup.
136    pub user_id: ts_control_serde::UserId,
137
138    /// The tailnet this node belongs to.
139    pub tailnet: Option<String>,
140
141    /// The tags assigned to this node.
142    pub tags: Vec<String>,
143
144    /// Every prefix control assigned this node (`tailcfg.Node.Addresses`), in wire order.
145    ///
146    /// Normally one IPv4 `/32` and one IPv6 `/128`, but the wire field is a variable-length list:
147    /// an IPv6-off tailnet assigns only the v4 prefix, and nothing in the protocol stops control
148    /// assigning more than one prefix of a family.
149    ///
150    /// [`tailnet_address`](Self::tailnet_address) is the *identity* projection of this list — the
151    /// first prefix of each family — and is what the overlay, MagicDNS and exit-node selection
152    /// reason about. The whole list is retained because [`is_router`](Self::is_router) has to ask
153    /// "is this prefix one of my own?" of **all** of them, exactly as Go's `tailcfg.Node.IsRouter`
154    /// does. Keep the two consistent when building a `Node` by hand.
155    pub addresses: Vec<ipnet::IpNet>,
156
157    /// The address of the node in the tailnet: the first prefix of each family in
158    /// [`addresses`](Self::addresses), with an unspecified placeholder for a family the tailnet
159    /// does not assign.
160    pub tailnet_address: TailnetAddress,
161
162    /// The node's [`NodePublicKey`].
163    pub node_key: NodePublicKey,
164    /// The node key's expiration.
165    pub node_key_expiry: Option<DateTime<Utc>>,
166
167    /// Whether this node's key is expired (`tailcfg.Node.Expired`).
168    ///
169    /// Two writers, exactly as upstream. Control may send it on the wire, and the client sets it
170    /// itself — only ever `false` → `true` — when
171    /// [`node_key_expiry`](Self::node_key_expiry) has passed, so the decision is made against a
172    /// clock corrected for control skew rather than the raw local one. See
173    /// [`ExpiryManager::flag_expired_peer`](crate::ExpiryManager::flag_expired_peer), which is what
174    /// sets it and which also clears this node's endpoints and home DERP and breaks its
175    /// [`node_key`](Self::node_key).
176    ///
177    /// An expired peer is **kept** in the netmap, not dropped: that is what lets a caller answer
178    /// [`PEER_KEY_EXPIRED`](crate::PEER_KEY_EXPIRED) rather than "no such peer". Distinct from
179    /// [`key_expired`](Self::key_expired), which recomputes the answer from the raw local clock and
180    /// is what the **self**-node re-auth decision reads.
181    pub expired: bool,
182
183    /// Whether control reports this node currently connected to the coordination server
184    /// (`tailcfg.Node.Online`, a tri-state `*bool`). `None` = unknown / no permission to know /
185    /// never been online — **do not collapse to `false`** (that would fabricate an offline status
186    /// control never asserted). Updated by full nodes AND by the delta channels (a
187    /// [`PeerChange::online`], or the `MapResponse.online_change` map).
188    pub online: Option<bool>,
189    /// When control last saw this node online (`tailcfg.Node.LastSeen`). Per Go, only meaningful
190    /// while `online` is not `Some(true)` ("not updated when Online is true"). `None` = unknown /
191    /// never online.
192    pub last_seen: Option<DateTime<Utc>>,
193
194    /// Marshalled TKA node-key signature (`tailcfg.Node.KeySignature`); empty when control sends
195    /// none. Verified against a TKA `Authority` at the peer-trust chokepoint WHEN tailnet-lock
196    /// enforcement is active.
197    pub key_signature: Vec<u8>,
198
199    /// The node's [`MachinePublicKey`], if known.
200    pub machine_key: Option<MachinePublicKey>,
201    /// The node's [`DiscoPublicKey`], if known.
202    pub disco_key: Option<DiscoPublicKey>,
203
204    /// Whether control marked this node as peerAPI-only and outside tailnet lock's coverage
205    /// (`tailcfg.Node.UnsignedPeerAPIOnly`).
206    ///
207    /// Such a node carries no node-key signature. Upstream Go treats that as deliberate: it exempts
208    /// the node from tailnet-lock verification and, in exchange, gives it **no network access** —
209    /// only this node's peerAPI.
210    ///
211    /// **This fork does not implement that admission exemption yet.** While a tailnet-lock authority
212    /// with a **non-empty trusted-key set** is active, the runtime's peer-admission gate
213    /// (`ts_runtime`'s `PeerTracker::tka_snapshot_admits`) drops *every* peer with an empty
214    /// [`key_signature`](Self::key_signature), this flag included — so such a peer is not admitted
215    /// to the peer db at all and gets no peerAPI access either. That is stricter than Go (the safe
216    /// direction); the carve-out is tracked as a parity gap in `docs/PARITY_ROADMAP.md`.
217    ///
218    /// The trusted-key qualifier is not hypothetical hedging, it names the one case where the gate
219    /// does not enforce: an authority whose trusted-key set is *empty* admits every peer, signed or
220    /// not. A verified chain can never produce that state (genesis rejects an empty key set and the
221    /// last key cannot be removed), so it means a `ts_tka` invariant was violated — and the gate
222    /// prefers admitting everyone (logged at `error!`) over black-holing the whole netmap. In that
223    /// state this flag changes nothing either, because nothing is being enforced against.
224    ///
225    /// The *routes* half of upstream's treatment **is** implemented here. Because the node is
226    /// outside the lock, a (possibly malicious) control server must not be able to grant it
227    /// network access via advertised routes, so [`accepted_routes`](Self::accepted_routes) is
228    /// clamped to the node's own [`addresses`](Self::addresses) when this is set. See the `From`
229    /// impl on this type, which mirrors Go's `upgradeNode` in `control/controlclient/map.go`.
230    ///
231    /// The clamp is **unconditional** — it does not depend on tailnet lock being enabled locally,
232    /// because the point is that an unsigned peer is by definition outside the lock's coverage.
233    ///
234    /// The clamp alone is not enough, because it closes only one of the two doors control has.
235    /// Control can leave the node's `AllowedIPs` at its own addresses — which the clamp permits,
236    /// those *are* its addresses — and write those same addresses into the **packet filter** as an
237    /// allowed source instead. Upstream's answer is to reject the filter outright: a filter that
238    /// grants an unsigned peer network access is treated as invalid ("the server is either broken
239    /// or malicious") and ignored wholesale. That is
240    /// [`ts_packetfilter::permits_unlocked_nodes`], driven by `ts_runtime`'s packet-filter updater
241    /// on every netmap that moves either the filter or the peer set — Go's
242    /// `packetFilterPermitsUnlockedNodes` / `nodeBackend.unlockedNodesPermitted` in
243    /// `ipn/ipnlocal/local.go`.
244    ///
245    /// The **capability** half is still unported: there is no per-peer capability map in this
246    /// domain model, only the node-attribute [`cap_map`](Self::cap_map). When it is ported, note
247    /// that upstream does **not** withhold every capability. `capsAllowedForUnsignedPeer`
248    /// (`ipn/ipnlocal/node_backend.go`) keeps `tailcfg.PeerCapabilityIngress` when the peer has it
249    /// and drops the rest, on upstream's own reasoning that "Tailscale Funnel ingress nodes are
250    /// unsigned by design, and the capability only permits ingress requests over the PeerAPI, which
251    /// unsigned peers can already reach". Withholding it too would refuse Funnel ingress from real
252    /// Tailscale nodes, so the carve-out travels with the port rather than after it.
253    pub unsigned_peer_api_only: bool,
254
255    /// The routes this node accepts traffic for.
256    ///
257    /// Clamped to [`addresses`](Self::addresses) when
258    /// [`unsigned_peer_api_only`](Self::unsigned_peer_api_only) is set.
259    pub accepted_routes: Vec<ipnet::IpNet>,
260    /// The underlay addresses this node is reachable on (`Endpoints` in Go).
261    pub underlay_addresses: Vec<SocketAddr>,
262
263    /// The node's advertised SSH host public keys, in known_hosts format (Go
264    /// `tailcfg.Hostinfo.SSHHostKeys`, surfaced by tsnet as `ipnstate.PeerStatus.SSH_HostKeys`).
265    /// Used by `tailscale ssh` to pin a peer's host key (TOFU). Empty when control advertised none
266    /// (the wire `Hostinfo.sshHostKeys` was absent), never fabricated. Projected from
267    /// [`ts_control_serde::HostInfo::ssh_host_keys`].
268    pub ssh_host_keys: Vec<String>,
269
270    /// The DERP region for this node, if known.
271    pub derp_region: Option<ts_derp::RegionId>,
272
273    /// This node's advertised capability version (`Node.Cap` in Go). Old control servers may not
274    /// send it, in which case it defaults to [`CapabilityVersion::default`]. Used to gate features
275    /// that require a minimum peer capability, e.g. exit-node DNS proxying (`peerCanProxyDNS`).
276    pub cap: CapabilityVersion,
277
278    /// This node's capability map (`Node.CapMap` in Go). Keys are capability names/URLs; values are
279    /// the raw JSON argument blobs (often empty). Threaded from the wire
280    /// ([`ts_control_serde::Node::cap_map`]) as an owned copy. Used to gate node-level features such
281    /// as Funnel ingress ([`Node::can_funnel`], [`Node::check_funnel_port`]).
282    pub cap_map: NodeCapMap,
283
284    /// The peerAPI port this node advertises over IPv4 (`peerapi4` service), if any.
285    ///
286    /// Derived from `HostInfo.Services`. `None` means the peer advertises no IPv4 peerAPI, so it
287    /// cannot be reached for peerAPI DoH (DNS-over-HTTPS) exit-node delegation.
288    pub peerapi_port: Option<u16>,
289
290    /// Whether this peer advertises the `peerapi-dns-proxy` service (Go `PeerAPIDNSProxy`),
291    /// indicating it will proxy DNS lookups for other nodes when used as an exit node.
292    pub peerapi_dns_proxy: bool,
293
294    /// Whether this is a non-Tailscale WireGuard-only peer (`IsWireGuardOnly` in Go). Such peers
295    /// cannot run a peerAPI DoH server, so exit-node DNS for them comes from
296    /// [`Node::exit_node_dns_resolvers`] instead.
297    pub is_wireguard_only: bool,
298
299    /// DNS resolvers to use when this WireGuard-only peer is selected as an exit node
300    /// (`ExitNodeDNSResolvers` in Go). Only meaningful when [`Node::is_wireguard_only`] is set.
301    /// Encrypted-transport resolvers are dropped (see `Resolver::from_serde`).
302    pub exit_node_dns_resolvers: Vec<Resolver>,
303
304    /// Whether this node advertises itself as a **peer relay** (Go `Hostinfo.PeerRelay`): it runs a
305    /// UDP relay server other peers can allocate relay endpoints on. This fork is a relay client
306    /// only and never sets this for itself; it is parsed off peers so a relay candidate can be
307    /// recognized. Actually *using* a relay path (the Geneve data path + allocation handshake) is
308    /// not yet implemented — see the crate docs.
309    pub peer_relay: bool,
310
311    /// Per-service virtual IP addresses of the Tailscale VIP services this node *hosts*, keyed by
312    /// `svc:<label>` service name. Parsed from the `service-host`
313    /// ([`ts_control_serde::NODE_ATTR_SERVICE_HOST`]) node-capability value
314    /// (`tailcfg.ServiceIPMappings`). These VIPs are control-assigned and also injected into the
315    /// node's `AllowedIPs`; the application netstack must accept packets for them so a
316    /// `Device::listen_service`-bound listener can answer. Empty when the
317    /// node hosts no VIP services (the common case). Per-service IP lists are deduplicated, source
318    /// order otherwise preserved. Use [`Node::service_addresses`] for the flattened set (netstack
319    /// accept list) and [`Node::service_addresses_for`] for a specific service's VIPs.
320    pub service_vips: alloc::collections::BTreeMap<String, Vec<IpAddr>>,
321}
322
323impl Node {
324    /// The fully-qualified domain name of the node.
325    ///
326    /// This is a string of the form `$HOST.$TAILNET_DOMAIN.`. For tailnets controlled by
327    /// Tailscale's control plane, this usually means `$HOST.tail1234.ts.net.`
328    ///
329    /// The `trailing_dot` parameter specifies whether to include the trailing dot in the
330    /// fqdn. This is included by the definition of FQDN, and is the way the Go codebase
331    /// formats this field, but the parameter is included to allow turning it off for use
332    /// in contexts that expect it to be absent.
333    pub fn fqdn(&self, trailing_dot: bool) -> String {
334        let dot = if trailing_dot { "." } else { "" };
335        match &self.tailnet {
336            Some(tailnet) => format!("{}.{tailnet}{dot}", self.hostname),
337            None => format!("{}{dot}", self.hostname),
338        }
339    }
340
341    /// Whether this node's key has expired as of `now`, mirroring Go's
342    /// `netmap.NetworkMap.SelfKeyExpiry` + the `!expiry.IsZero() && expiry.Before(now)` check in
343    /// `ipnlocal`. A node with no expiry ([`Node::node_key_expiry`] is `None`, the Go "zero value =
344    /// does not expire") is never expired.
345    ///
346    /// Like Go, this fork is **reactive**: it reports expiry rather than auto-rotating in the
347    /// background (Go transitions to `NeedsLogin` on expiry and re-registers via stored auth-key or
348    /// interactive login). A caller observing `true` should re-register
349    /// (`crate::tokio::register`) — supplying `RegisterRequest::old_node_key` (the prior key) and
350    /// a fresh `node_key` when rotating the key, or the same key to merely refresh.
351    pub fn key_expired(&self, now: DateTime<Utc>) -> bool {
352        match self.node_key_expiry {
353            None => false,
354            Some(expiry) => expiry < now,
355        }
356    }
357
358    /// The instant this node's key expires (`Node.KeyExpiry` in Go), or `None` if it never expires.
359    /// A caller can schedule a re-evaluation/re-auth at this time.
360    pub fn key_expiry(&self) -> Option<DateTime<Utc>> {
361        self.node_key_expiry
362    }
363
364    /// Whether this node advertises itself as a peer relay (Go `Hostinfo.PeerRelay`): it runs a UDP
365    /// relay server other peers may allocate relay endpoints on. Recognizing a relay candidate;
366    /// actually traversing a relay path is not yet implemented in this fork.
367    pub fn is_peer_relay(&self) -> bool {
368        self.peer_relay
369    }
370
371    /// The key-expiry instant as **Unix seconds**, or `None` if the key never expires. Provided for
372    /// callers (e.g. the root crate) that don't depend on `chrono`.
373    pub fn key_expiry_unix(&self) -> Option<i64> {
374        self.node_key_expiry.map(|t| t.timestamp())
375    }
376
377    /// Whether the key has expired as of `now_unix_secs` (Unix seconds). Equivalent to
378    /// [`key_expired`](Self::key_expired) for `chrono`-free callers. A key with no expiry is never
379    /// expired.
380    pub fn key_expired_at_unix(&self, now_unix_secs: i64) -> bool {
381        match self.key_expiry_unix() {
382            None => false,
383            Some(expiry) => expiry < now_unix_secs,
384        }
385    }
386
387    /// The fully-qualified domain name of the node, only returning `Some` if the tailnet
388    /// component is present.
389    ///
390    /// See [`Node::fqdn`].
391    pub fn fqdn_opt(&self, trailing_dot: bool) -> Option<String> {
392        let dot = if trailing_dot { "." } else { "" };
393        let tailnet = self.tailnet.as_deref()?;
394
395        Some(format!("{}.{tailnet}{dot}", self.hostname))
396    }
397
398    /// Report whether this node matches the given `name`.
399    ///
400    /// `name` is checked for equality with both this node's bare hostname and its fqdn. A
401    /// trailing `.` may be present. Matching is case-insensitive (DNS names are
402    /// case-insensitive), so this agrees with the canonicalized MagicDNS-name index used for
403    /// peer lookups.
404    pub fn matches_name(&self, name: &str) -> bool {
405        // Strip an optional trailing root dot, then chop our `.tailnet` suffix off the end (if it
406        // matches, case-insensitively) and compare the remainder to our hostname. If the tailnet
407        // suffix doesn't match, the final case-insensitive compare against our bare hostname fails
408        // naturally; if `name` was just the hostname, nothing is chopped and we compare directly.
409
410        let name = name.strip_suffix('.').unwrap_or(name);
411
412        let name = if let Some(tailnet) = &self.tailnet {
413            name.get(name.len().saturating_sub(tailnet.len())..)
414                .filter(|suffix| suffix.eq_ignore_ascii_case(tailnet))
415                .and_then(|_| name.get(..name.len() - tailnet.len()))
416                .and_then(|name| name.strip_suffix('.'))
417                .unwrap_or(name)
418        } else {
419            name
420        };
421
422        name.eq_ignore_ascii_case(&self.hostname)
423    }
424
425    /// Report whether this node is a **router**: it routes addresses besides its own. An exit
426    /// node, a subnet router and an app connector are all routers.
427    ///
428    /// Mirrors Go's `tailcfg.Node.IsRouter` (`tailcfg/tailcfg.go`, added upstream in `8d830599b`),
429    /// which is `true` when any prefix in `AllowedIPs` is not also one of the node's own
430    /// `Addresses`. It is a *derived predicate*, not a wire field: control sends nothing new for
431    /// it, so there is no interop surface here and no capability version to gate on.
432    ///
433    /// Deliberately **not** [`Node::is_subnet_route`] folded over [`Node::accepted_routes`]. That
434    /// predicate also excuses any single Tailscale-range IP (`100.64.0.0/10` /
435    /// `fd7a:115c:a1e0::/48`) so route installation never mistakes another peer's address for an
436    /// advertised subnet; Go's `IsRouter` makes no such exception — a `/32` that is not *this*
437    /// node's own address still makes it a router. The two must stay separate.
438    ///
439    /// The comparison is against [`Node::addresses`] — *every* prefix control assigned this node,
440    /// as Go's `slices.Contains(n.Addresses, r)` is — and not against the first-prefix-per-family
441    /// pair in [`Node::tailnet_address`]. A node control handed two prefixes of one family would
442    /// otherwise have the second read as a routed address and be misreported as a router.
443    pub fn is_router(&self) -> bool {
444        self.accepted_routes
445            .iter()
446            .any(|route| !self.addresses.contains(route))
447    }
448
449    /// Report whether `route` is an advertised *subnet* route (as opposed to one of this node's
450    /// own tailnet addresses).
451    ///
452    /// Mirrors `cidrIsSubnet` in the Go client (`wgengine/wgcfg/nmcfg/nmcfg.go`). A route is *not*
453    /// a subnet route (i.e. it's a self-address) when it is a single host IP that is either a
454    /// Tailscale-assigned IP or exactly one of this node's [`TailnetAddress`] addresses. Everything
455    /// else — multi-IP CIDRs, and single IPs outside the Tailscale ranges — is a subnet route.
456    ///
457    /// The default route (`0.0.0.0/0` / `::/0`) is treated as a subnet route here; exit-node
458    /// handling is a separate concern.
459    pub fn is_subnet_route(&self, route: &ipnet::IpNet) -> bool {
460        let host_prefix = match route {
461            ipnet::IpNet::V4(_) => 32,
462            ipnet::IpNet::V6(_) => 128,
463        };
464
465        if route.prefix_len() != host_prefix {
466            // Any multi-IP CIDR (including the default route) is a subnet route.
467            return true;
468        }
469
470        let addr = route.addr();
471        !(is_tailscale_ip(addr) || self.tailnet_address.contains(addr))
472    }
473
474    /// The routes that should be installed for this peer, given whether this node accepts
475    /// advertised subnet routes (`--accept-routes` / `RouteAll` in the Go client) and which peer
476    /// (if any) is the selected exit node (`--exit-node` / `ExitNodeID` in the Go client).
477    ///
478    /// This node's own addresses (the peer's `/32` and `/128`) are always installed so the peer
479    /// itself stays reachable. Larger advertised subnet routes are only installed when
480    /// `accept_routes` is set; otherwise they are dropped (fail-closed). The same filtered set
481    /// governs both outbound routing to the peer and inbound source validation, exactly as
482    /// WireGuard cryptokey routing couples them in the Go client.
483    ///
484    /// The default route (`0.0.0.0/0` / `::/0`) is installed *only* for the peer whose
485    /// [`StableId`] equals `exit_node`, mirroring `nmcfg.go`'s `if allowedIP.Bits()==0 &&
486    /// peer.StableID()!=exitNode { skip }`. Exit-node use is gated behind this separate, explicit
487    /// preference (`ExitNodeID`, not `RouteAll`): conflating the two would let enabling
488    /// subnet-route acceptance silently route every packet through any peer advertising a default
489    /// route — unacceptable for a fail-closed privacy posture. When `exit_node` is `None` (the
490    /// default) no peer ever receives a `/0`, so internet-bound traffic has no overlay route and is
491    /// dropped by the userspace netstack (fail-closed, no leak). Longest-prefix-match means a peer
492    /// selected as the exit node still loses more-specific destinations to other peers; only
493    /// residual default-route traffic egresses through it.
494    pub fn routes_to_install<'a>(
495        &'a self,
496        accept_routes: bool,
497        exit_node: Option<&StableId>,
498    ) -> impl Iterator<Item = &'a ipnet::IpNet> + 'a {
499        // Computed eagerly so the returned iterator doesn't borrow `exit_node`.
500        let is_selected_exit = exit_node == Some(&self.stable_id);
501        self.accepted_routes.iter().filter(move |route| {
502            if route.prefix_len() == 0 {
503                // Default route: installed only when this peer is the selected exit node. Both the
504                // outbound route table and the inbound source filter call this, so the exit peer
505                // may legitimately source arbitrary internet IPs on return traffic — and only it.
506                return is_selected_exit;
507            }
508            accept_routes || !self.is_subnet_route(route)
509        })
510    }
511
512    /// The capability version at and above which a peer can proxy DNS for nodes using it as an exit
513    /// node (Go `tailcfg.CapabilityVersion` `peerCanProxyDNS`, introduced 2022-01-12 at V26).
514    const PEER_CAN_PROXY_DNS: CapabilityVersion = CapabilityVersion::V26;
515
516    /// The base URL of this peer's IPv4 peerAPI DoH endpoint for exit-node DNS proxying, if it can
517    /// proxy DNS. Returns e.g. `http://100.64.0.5:8080/dns-query`.
518    ///
519    /// Mirrors Go `peerAPIBase(...)+"/dns-query"` gated by `exitNodeCanProxyDNS`: a peer can proxy
520    /// DNS when it advertises an IPv4 peerAPI port **and** either advertises the explicit
521    /// `peerapi-dns-proxy` service or is new enough ([`Node::cap`] ≥ `PEER_CAN_PROXY_DNS`). A
522    /// WireGuard-only peer never runs a peerAPI, so it returns `None` here (its exit-node DNS comes
523    /// from [`Node::exit_node_dns_resolvers`] instead).
524    ///
525    /// IPv4-only by deliberate design: the tailnet dataplane in this fork binds IPv4 only, so we
526    /// never form a peerAPI URL on the peer's IPv6 address.
527    ///
528    /// `None` for an [`expired`](Self::expired) peer — see [`Node::peerapi_addr`].
529    pub fn peerapi_doh_url(&self) -> Option<String> {
530        self.peerapi_doh_addr()
531            .map(|addr| format!("http://{addr}/dns-query"))
532    }
533
534    /// The IPv4 socket address (`<tailnet-ipv4>:<peerapi-port>`) of this peer's peerAPI DoH endpoint
535    /// for exit-node DNS proxying, if it can proxy DNS. Same gate as [`Node::peerapi_doh_url`]; this
536    /// is the form the DoH *client* dials (over the overlay netstack) when delegating recursive
537    /// resolution to a selected exit node. `SocketAddr`'s `Display` is `ip:port`, so
538    /// `peerapi_doh_url` formats to `http://<ip>:<port>/dns-query` over this.
539    pub fn peerapi_doh_addr(&self) -> Option<SocketAddr> {
540        if self.is_wireguard_only || self.expired {
541            return None;
542        }
543        let port = self.peerapi_port?;
544        if !(self.peerapi_dns_proxy || self.cap >= Self::PEER_CAN_PROXY_DNS) {
545            return None;
546        }
547        Some(SocketAddr::new(
548            IpAddr::V4(self.tailnet_address.ipv4.addr()),
549            port,
550        ))
551    }
552
553    /// The IPv4 peerAPI socket address (`<tailnet-ipv4>:<peerapi4-port>`) of this node, if it
554    /// advertises an IPv4 peerAPI. Unlike [`Node::peerapi_doh_addr`], this is **not** gated on the
555    /// DNS-proxy capability: it is the general base for any peerAPI request to this node (e.g. a
556    /// Taildrop `PUT /v0/put/<name>` upload), mirroring Go's `peerAPIBase`/`peerAPIPorts`.
557    ///
558    /// IPv4-only by this fork's deliberate design (the tailnet dataplane binds IPv4 only, so we never
559    /// form a peerAPI URL on the peer's IPv6 address). Returns `None` for a WireGuard-only peer (which
560    /// runs no peerAPI) or a peer advertising no IPv4 peerAPI port.
561    ///
562    /// Also `None` for an [`expired`](Self::expired) peer: Go refuses a peerAPI dial to one with
563    /// [`PEER_KEY_EXPIRED`](crate::PEER_KEY_EXPIRED) (`LocalBackend.pingPeerAPI`), and this is the
564    /// chokepoint every peerAPI dial in this fork resolves its destination through. Callers that
565    /// want to *report* the refusal rather than silently skip the peer should test
566    /// [`expired`](Self::expired) first.
567    pub fn peerapi_addr(&self) -> Option<SocketAddr> {
568        if self.is_wireguard_only || self.expired {
569            return None;
570        }
571        let port = self.peerapi_port?;
572        Some(SocketAddr::new(
573            IpAddr::V4(self.tailnet_address.ipv4.addr()),
574            port,
575        ))
576    }
577
578    /// The node attribute granting HTTPS (TLS cert provisioning) for this node (Go
579    /// `tailcfg.CapabilityHTTPS`). One of the two caps [`Node::can_funnel`] requires.
580    const CAP_HTTPS: &'static str = "https";
581
582    /// The node attribute granting the ability to host Funnel ingress (Go `tailcfg.NodeAttrFunnel`).
583    /// The other cap [`Node::can_funnel`] requires.
584    const NODE_ATTR_FUNNEL: &'static str = "funnel";
585
586    /// The capability URL whose `?ports=` query enumerates the ports Funnel may listen on (Go
587    /// `tailcfg.CapabilityFunnelPorts`). The allowed ports live entirely in the *key's* query
588    /// string, not the cap value.
589    const CAP_FUNNEL_PORTS: &'static str = "https://tailscale.com/cap/funnel-ports";
590
591    /// Report whether the cap map contains `cap` as a key (Go `NodeCapMap.Contains` / `HasCap`).
592    pub fn has_node_attr(&self, cap: &str) -> bool {
593        self.cap_map.contains_key(cap)
594    }
595
596    /// Report whether this node is permitted to host Tailscale Funnel ingress.
597    ///
598    /// Mirrors Go `ipn.NodeCanFunnel`: the node must advertise BOTH `CapabilityHTTPS` (`"https"`)
599    /// AND `NodeAttrFunnel` (`"funnel"`) in its cap map. Fail-closed: a missing cap denies.
600    pub fn can_funnel(&self) -> bool {
601        self.has_node_attr(Self::CAP_HTTPS) && self.has_node_attr(Self::NODE_ATTR_FUNNEL)
602    }
603
604    /// The capability control grants the **self** node when Taildrop is enabled for the tailnet (Go
605    /// `tailcfg.CapabilityFileSharing`). Gates [`Node::can_share_files`].
606    const CAP_FILE_SHARING: &'static str = "https://tailscale.com/cap/file-sharing";
607
608    /// The capability marking a **peer** as an explicit Taildrop send target even across owners (Go
609    /// `tailcfg.PeerCapabilityFileSharingTarget`). Checked by [`Node::is_file_sharing_target`].
610    const CAP_FILE_SHARING_TARGET: &'static str = "tailscale.com/cap/file-sharing-target";
611
612    /// Report whether this node may send Taildrop files — i.e. the admin has enabled file sharing for
613    /// the tailnet (Go `self.CapMap().Contains(CapabilityFileSharing)`). Applied to the **self** node
614    /// as the node-level gate in `FileTargets`; fail-closed when the cap is absent.
615    pub fn can_share_files(&self) -> bool {
616        self.has_node_attr(Self::CAP_FILE_SHARING)
617    }
618
619    /// Report whether this **peer** is an explicit Taildrop send target via ACL caps (Go
620    /// `PeerHasCap(p, PeerCapabilityFileSharingTarget)`) — the cross-owner path that lets a peer owned
621    /// by a different user still be a valid target.
622    pub fn is_file_sharing_target(&self) -> bool {
623        self.has_node_attr(Self::CAP_FILE_SHARING_TARGET)
624    }
625
626    /// The node attribute control sets on a node whose **subdomains** all resolve to the node
627    /// itself (Go `tailcfg/nodecap`'s `NodeAttrDNSSubdomainResolve`). Read by
628    /// [`Node::resolves_subdomains`].
629    const NODE_ATTR_DNS_SUBDOMAIN_RESOLVE: &'static str = "dns-subdomain-resolve";
630
631    /// Report whether every subdomain of this node's MagicDNS name resolves to this node's
632    /// addresses — `foo.<node>` and `bar.foo.<node>` alike.
633    ///
634    /// Go's resolver (`net/dns/resolver/tsdns.go`) learns the same thing two ways — a
635    /// `Config.SubdomainHosts` set of FQDNs beside its `Hosts` map, and a `SubdomainHost` predicate
636    /// on its MagicDNS host index — and on a lookup miss walks the queried name's parents,
637    /// answering from the first parent either one accepts. Here the attribute on the node *is* that
638    /// predicate, read where the parent walk finds the node.
639    ///
640    /// Being a plain per-node attribute, it needs no capability version: a node control has not set
641    /// it on is unaffected, and its subdomains stay `NXDOMAIN`.
642    pub fn resolves_subdomains(&self) -> bool {
643        self.has_node_attr(Self::NODE_ATTR_DNS_SUBDOMAIN_RESOLVE)
644    }
645
646    /// The node attribute control sets to stop the DNS forwarder re-asking a truncated upstream
647    /// answer over TCP (Go `tailcfg/nodecap`'s `NodeAttrDNSForwarderDisableTCPRetries`, surfaced in
648    /// `control/controlknobs` as `Knobs.DisableDNSForwarderTCPRetries`). Read by
649    /// [`Node::disable_dns_forwarder_tcp_retries`].
650    const NODE_ATTR_DNS_FORWARDER_DISABLE_TCP_RETRIES: &'static str =
651        "dns-forwarder-disable-tcp-retries";
652
653    /// Report whether control has told this node **not** to retry a truncated forwarded DNS answer
654    /// over TCP.
655    ///
656    /// The retry is on by default and this attribute is its *off* switch — so, unlike every other
657    /// attribute here, the fail-closed reading is the one that ignores it: a node control has not
658    /// set it on keeps retrying, which is what a stub resolver on this node needs for a name whose
659    /// answer does not fit a datagram. Go reads it the same way round
660    /// (`skipTCP := skipTCPRetry() || (f.controlKnobs != nil &&
661    /// f.controlKnobs.DisableDNSForwarderTCPRetries.Load())`, net/dns/resolver/forwarder.go).
662    ///
663    /// Upstream dates a client's understanding of the attribute to capability version 75
664    /// ([`ts_capabilityversion::CapabilityVersion::V75`]), which this tree's `CURRENT` is well
665    /// above, so control will send it to this node when the tailnet sets it.
666    pub fn disable_dns_forwarder_tcp_retries(&self) -> bool {
667        self.has_node_attr(Self::NODE_ATTR_DNS_FORWARDER_DISABLE_TCP_RETRIES)
668    }
669
670    /// The node attribute by which control asks this node to keep its periodic STUN sweep running
671    /// even while the datapath is idle (Go `tailcfg/nodecap`'s `NodeAttrDebugForceBackgroundSTUN`,
672    /// surfaced in `control/controlknobs` as `Knobs.ForceBackgroundSTUN`). Read by
673    /// [`force_background_stun`](Self::force_background_stun).
674    const NODE_ATTR_DEBUG_FORCE_BACKGROUND_STUN: &'static str = "debug-always-stun";
675
676    /// Report whether control has asked this node to keep STUNning in the background regardless of
677    /// datapath activity.
678    ///
679    /// This is the single override on the idle stop condition in Go magicsock's
680    /// `shouldDoPeriodicReSTUNLocked`: once the datapath has been idle longer than the session-active
681    /// timeout the periodic sweep stops, *unless* `c.controlKnobs.ForceBackgroundSTUN` is set, in
682    /// which case it keeps running. It overrides nothing else — a node with no peers still does not
683    /// STUN, with or without the attribute, because that arm returns before the idle arm is reached.
684    ///
685    /// Read off the **self** node's cap map, like every other control knob. Absent (the normal case)
686    /// means "let the idle stop apply", which is the quiet default; the attribute is a debugging
687    /// escape hatch control sets deliberately, so there is nothing to fail closed to here.
688    pub fn force_background_stun(&self) -> bool {
689        self.has_node_attr(Self::NODE_ATTR_DEBUG_FORCE_BACKGROUND_STUN)
690    }
691
692    /// The node attribute by which control asks this node to collapse its per-peer CGNAT host
693    /// routes into the single `100.64.0.0/10` (Go `tailcfg/nodecap`'s `OneCGNATEnable`).
694    ///
695    /// Note the query string: the key is the literal `one-cgnat?v=true`, not `one-cgnat`. The
696    /// attribute is a tri-state carried as two mutually exclusive keys rather than as a key with a
697    /// value, so the lookup is on the whole literal.
698    const NODE_ATTR_ONE_CGNAT_ENABLE: &'static str = "one-cgnat?v=true";
699
700    /// The node attribute by which control asks this node to keep one host route **per peer** no
701    /// matter how many peers there are (Go `tailcfg/nodecap`'s `OneCGNATDisable`). The other half
702    /// of [`NODE_ATTR_ONE_CGNAT_ENABLE`](Self::NODE_ATTR_ONE_CGNAT_ENABLE)'s tri-state.
703    const NODE_ATTR_ONE_CGNAT_DISABLE: &'static str = "one-cgnat?v=false";
704
705    /// Control's tri-state instruction about collapsing this node's per-peer CGNAT host routes
706    /// into the single `100.64.0.0/10`, read off the **self** node's cap map.
707    ///
708    /// Mirrors Go `ipn/ipnlocal`'s read of `nodecap.OneCGNATEnable` / `nodecap.OneCGNATDisable`
709    /// into `controlknobs.Knobs.OneCGNAT`, which is an `opt.Bool` and not a `bool` precisely so the
710    /// third state exists:
711    ///
712    /// * `Some(true)` — `one-cgnat?v=true`: always collapse.
713    /// * `Some(false)` — `one-cgnat?v=false`: never collapse, one `/32` per peer however many
714    ///   peers there are.
715    /// * `None` — neither attribute present: control has no opinion, and the consumer's own
716    ///   peer-count threshold decides (Go `net/routemanager`'s `cgnatThreshold`).
717    ///
718    /// A node holding BOTH attributes reads as `Some(true)`: the enabling attribute is checked
719    /// first and wins. Control setting both is a policy conflict rather than a state upstream
720    /// specifies, and collapsing is the safe way to break the tie — the `/10` is a superset of the
721    /// `/32`s it replaces, so no peer becomes unreachable, whereas honouring the disabling
722    /// attribute on a tailnet large enough for control to have set the enabling one is exactly the
723    /// unbounded host route table the threshold exists to prevent.
724    pub fn one_cgnat(&self) -> Option<bool> {
725        if self.has_node_attr(Self::NODE_ATTR_ONE_CGNAT_ENABLE) {
726            Some(true)
727        } else if self.has_node_attr(Self::NODE_ATTR_ONE_CGNAT_DISABLE) {
728            Some(false)
729        } else {
730            None
731        }
732    }
733
734    /// The node attribute by which control asks this node to stop processing netmap updates through
735    /// the delta (incremental) path (Go `tailcfg/nodecap`'s `DisableDeltaUpdates`, read into
736    /// `controlknobs.Knobs.DisableDeltaUpdates`). Read off the **self** node's cap map by
737    /// [`delta_updates_disabled`](Self::delta_updates_disabled).
738    ///
739    /// Upstream documents the intent on the knob itself: the client "should not process updates via
740    /// the delta update mechanism and should instead treat all netmap changes as 'full' ones as
741    /// tailscaled did in 1.48.x and earlier". It is control's escape hatch for a delta-encoding bug
742    /// on *either* side of the map protocol — control emitting bad patches, or a client applying
743    /// them wrongly — without waiting for a client release to ship.
744    const NODE_ATTR_DISABLE_DELTA_UPDATES: &'static str = "disable-delta-updates";
745
746    /// Report whether control has asked this node to decline the incremental netmap path and treat
747    /// every netmap change as a full one.
748    ///
749    /// Mirrors the first statement of Go `control/controlclient/map.go`'s `tryHandleIncrementally`:
750    /// `if ms.controlKnobs != nil && ms.controlKnobs.DisableDeltaUpdates.Load() { return false }`.
751    /// Returning `false` there does **not** reject the response and does not drop the mutations it
752    /// carries — it declines the incremental arm so the full netmap rebuild handles the very same
753    /// response. A consumer of this method owes the same shape: fall back, never drop.
754    ///
755    /// Absent attribute ⇒ `false` ⇒ the delta path, which is the default and the overwhelmingly
756    /// common case. Being a plain per-node attribute it needs no capability version: a node control
757    /// has not set it on is unaffected.
758    pub fn delta_updates_disabled(&self) -> bool {
759        self.has_node_attr(Self::NODE_ATTR_DISABLE_DELTA_UPDATES)
760    }
761
762    /// The node attribute by which control tells this node to stop sending disco heartbeats to its
763    /// peers (Go `tailcfg/nodecap`'s `SilentDisco`, read into `controlknobs.Knobs.SilentDisco` and
764    /// handed to magicsock by `ipn/ipnlocal`'s `b.MagicConn().SetSilentDisco(...)`). Read off the
765    /// **self** node's cap map by [`silent_disco`](Self::silent_disco).
766    ///
767    /// Upstream's own summary of the attribute is one sentence — it "makes the client suppress
768    /// disco heartbeats to its peers" — and the node it is set on is the node that goes quiet, so
769    /// it is the *self* node's cap map that decides, never the peer's.
770    const NODE_ATTR_SILENT_DISCO: &'static str = "silent-disco";
771
772    /// Report whether control has asked this node to stop heartbeating its peers' confirmed direct
773    /// paths.
774    ///
775    /// Mirrors Go magicsock `Conn.debugFlagsLocked`'s `heartbeatDisabled`, which the netmap push
776    /// (`endpoint.updateFromNode`) stamps onto every endpoint. With it set, the periodic
777    /// keep-the-best-path-alive ping is not sent — and, as the compensating half, an inbound packet
778    /// from the current best address extends that path's trust directly, because with no heartbeat
779    /// there is nothing else keeping it trusted.
780    ///
781    /// Absent attribute ⇒ `false` ⇒ the existing heartbeat cadence, which is the default and the
782    /// overwhelmingly common case. Being a plain per-node attribute it needs no capability version:
783    /// a node control has not set it on is unaffected.
784    ///
785    /// Go additionally ORs in a `TS_DEBUG_ENABLE_SILENT_DISCO` envknob at the same place. This tree
786    /// has no envknob layer at all, so the control attribute is the whole input here.
787    pub fn silent_disco(&self) -> bool {
788        self.has_node_attr(Self::NODE_ATTR_SILENT_DISCO)
789    }
790
791    /// The node attribute by which control tells this node that the network it sits on tolerates
792    /// nothing but TCP on port 443 (Go `tailcfg/nodecap`'s `OnlyTCP443`, read off `nm.SelfNode` in
793    /// `ipn/ipnlocal` and handed to magicsock by `b.MagicConn().SetOnlyTCP443(...)`). Read off the
794    /// **self** node's cap map by [`only_tcp_443`](Self::only_tcp_443).
795    ///
796    /// Upstream's own words: the client "should not attempt to generate any outbound traffic that
797    /// isn't TCP on port 443", which "thus implies all traffic is over DERP". The node it is set on
798    /// is the node that goes quiet, so it is the *self* node's cap map that decides, never the
799    /// peer's.
800    const NODE_ATTR_ONLY_TCP_443: &'static str = "only-tcp-443";
801
802    /// Report whether control has told this node to emit nothing but TCP/443 — no UDP at all, so
803    /// every tailnet packet rides DERP.
804    ///
805    /// This is the single predicate behind all of upstream's TCP-443-only behaviour: magicsock's
806    /// UDP send chokepoint refuses silently, netcheck's UDP (STUN) probes refuse with
807    /// `errors.ErrUnsupported`, the portmapper is disabled and the peer-relay client is switched
808    /// off. Consumers in this tree read it here rather than each re-deriving it from the cap map.
809    ///
810    /// Absent attribute ⇒ `false` ⇒ UDP as before, which is the default and the overwhelmingly
811    /// common case. Being a plain per-node attribute it needs no capability version: a node control
812    /// has not set it on is unaffected. It is read live off each netmap, never latched at start-up,
813    /// because upstream's setter (`Conn.SetOnlyTCP443`) is live — control withdrawing the attribute
814    /// has to restore UDP without a restart.
815    pub fn only_tcp_443(&self) -> bool {
816        self.has_node_attr(Self::NODE_ATTR_ONLY_TCP_443)
817    }
818
819    /// Report whether `wanted_port` is allowed for Funnel on this node.
820    ///
821    /// Mirrors Go `ipn.CheckFunnelPort`: scan the cap-map keys for one prefixed by
822    /// `Node::CAP_FUNNEL_PORTS`, URL-parse that key, read its `ports` query parameter, and match
823    /// `wanted_port` against the comma-separated list of single ports and `first-last` ranges. The
824    /// port list lives in the *key*, never the value. Fail-closed: no matching cap, an empty or
825    /// unparseable `ports` query, or a key whose non-query part isn't exactly the funnel-ports URL
826    /// all deny.
827    pub fn check_funnel_port(&self, wanted_port: u16) -> bool {
828        // Extract the `ports=` list from the first cap-map key that is the funnel-ports URL with a
829        // non-empty `ports` query. Returns `None` (deny) if the key is unparseable, the query is
830        // missing/empty, or the URL (sans query) isn't exactly the funnel-ports cap.
831        let parse_attr = |attr: &str| -> Option<String> {
832            let mut url = url::Url::parse(attr).ok()?;
833            let ports = url
834                .query_pairs()
835                .find(|(k, _)| k == "ports")
836                .map(|(_, v)| v.into_owned())?;
837            if ports.is_empty() {
838                return None;
839            }
840            url.set_query(None);
841            // Go compares `u.String()` against the bare cap; `url`'s serializer keeps a trailing
842            // `/` only if present in the input, and the funnel-ports cap has none, so a direct
843            // string compare matches Go's behavior.
844            if url.as_str() != Self::CAP_FUNNEL_PORTS {
845                return None;
846            }
847            Some(ports)
848        };
849
850        let Some(ports_str) = self
851            .cap_map
852            .keys()
853            .filter(|attr| attr.starts_with(Self::CAP_FUNNEL_PORTS))
854            .find_map(|attr| parse_attr(attr))
855        else {
856            return false;
857        };
858
859        let wanted = wanted_port.to_string();
860        for ps in ports_str.split(',') {
861            if ps.is_empty() {
862                continue;
863            }
864            match ps.split_once('-') {
865                None => {
866                    if ps == wanted {
867                        return true;
868                    }
869                }
870                Some((first, last)) => {
871                    let (Ok(fp), Ok(lp)) = (first.parse::<u16>(), last.parse::<u16>()) else {
872                        continue;
873                    };
874                    if fp <= wanted_port && wanted_port <= lp {
875                        return true;
876                    }
877                }
878            }
879        }
880        false
881    }
882
883    /// Report whether this node is permitted to host Tailscale VIP services.
884    ///
885    /// Mirrors the Go grant model: possession of the `service-host`
886    /// ([`ts_control_serde::NODE_ATTR_SERVICE_HOST`]) node-capability **and** at least one assigned
887    /// VIP address. Go additionally requires the host to be tagged
888    /// (`ErrUntaggedServiceHost`); that tag gate is enforced at
889    /// `Device::listen_service` using [`Node::tags`]. Fail-closed: no cap
890    /// or no assigned VIP denies.
891    pub fn is_service_host(&self) -> bool {
892        self.has_node_attr(ts_control_serde::NODE_ATTR_SERVICE_HOST)
893            && !self.service_vips.is_empty()
894    }
895
896    /// The control-assigned VIP addresses for one named service (`svc:<label>`), or an empty slice
897    /// if this node does not host that service. This is the exact per-service mapping (so a
898    /// multi-service co-host binds the right VIP for each service).
899    pub fn service_addresses_for(&self, service: &str) -> &[IpAddr] {
900        self.service_vips
901            .get(service)
902            .map(Vec::as_slice)
903            .unwrap_or(&[])
904    }
905
906    /// The flattened, deduplicated set of every VIP address this node hosts across all services.
907    /// Used to widen the netstack's accepted-address set so any hosted-service listener is
908    /// reachable. Per-service binding uses [`Node::service_addresses_for`] instead.
909    pub fn service_addresses(&self) -> Vec<IpAddr> {
910        let mut seen = alloc::collections::BTreeSet::new();
911        let mut out = Vec::new();
912        for addr in self.service_vips.values().flatten() {
913            if seen.insert(*addr) {
914                out.push(*addr);
915            }
916        }
917        out
918    }
919}
920
921/// Validate a Tailscale VIP service name (`tailcfg.ServiceName.Validate`): it must carry the
922/// `svc:` prefix ([`ts_control_serde::SERVICE_NAME_PREFIX`]) followed by a valid DNS label
923/// (1–63 chars, ASCII alphanumeric or `-`, not starting/ending with `-`). Returns the bare label on
924/// success. Fail-closed: anything malformed is rejected so a listener can never bind for a bogus
925/// service name.
926pub fn validate_service_name(name: &str) -> Option<&str> {
927    let label = name.strip_prefix(ts_control_serde::SERVICE_NAME_PREFIX)?;
928    if label.is_empty() || label.len() > 63 {
929        return None;
930    }
931    if label.starts_with('-') || label.ends_with('-') {
932        return None;
933    }
934    if label
935        .bytes()
936        .all(|b| b.is_ascii_alphanumeric() || b == b'-')
937    {
938        Some(label)
939    } else {
940        None
941    }
942}
943
944/// Parse the per-service VIP map this node hosts from the `service-host` node-capability value(s).
945/// Each value is the raw JSON text of a [`ts_control_serde::ServiceIpMappings`] object (svc-name ->
946/// VIP IPs); unparseable values are skipped (fail-closed: a malformed mapping contributes no VIPs).
947/// Per-service IP lists are deduplicated, source order otherwise preserved.
948fn service_vips_from_cap_map(
949    cap_map: &NodeCapMap,
950) -> alloc::collections::BTreeMap<String, Vec<IpAddr>> {
951    let mut out: alloc::collections::BTreeMap<String, Vec<IpAddr>> =
952        alloc::collections::BTreeMap::new();
953    let Some(values) = cap_map.get(ts_control_serde::NODE_ATTR_SERVICE_HOST) else {
954        return out;
955    };
956
957    for raw in values {
958        let Ok(mappings) = serde_json::from_str::<ts_control_serde::ServiceIpMappings>(raw) else {
959            continue;
960        };
961        for (name, addrs) in &mappings.0 {
962            let entry = out.entry((*name).to_string()).or_default();
963            for addr in addrs {
964                if !entry.contains(addr) {
965                    entry.push(*addr);
966                }
967            }
968        }
969    }
970    out
971}
972
973/// Collect a wire ([`ts_control_serde`]) node cap map into an owned [`NodeCapMap`].
974///
975/// Keys are copied as owned strings; each value's raw JSON text is preserved verbatim. The wire map
976/// borrows from the decode buffer, so an owned copy is required to outlive it on the domain
977/// [`Node`].
978fn cap_map_from_serde(wire: &ts_nodecapability::Map<'_>) -> NodeCapMap {
979    wire.iter()
980        .map(|(&key, values)| {
981            let owned_values = values.0.iter().map(|v| v.get().to_owned()).collect();
982            (key.to_owned(), owned_values)
983        })
984        .collect()
985}
986
987/// Extract the advertised IPv4 peerAPI port and whether the explicit `peerapi-dns-proxy` service is
988/// advertised, from a peer's `HostInfo.Services` list.
989fn peerapi_from_services(
990    services: Option<&[ts_control_serde::Service<'_>]>,
991) -> (Option<u16>, bool) {
992    use ts_control_serde::ServiceProto;
993
994    let Some(services) = services else {
995        return (None, false);
996    };
997    let mut port = None;
998    let mut dns_proxy = false;
999    for svc in services {
1000        match svc.proto {
1001            ServiceProto::PeerApi4 => port = Some(svc.port),
1002            ServiceProto::PeerApiDnsProxy => dns_proxy = true,
1003            _ => {}
1004        }
1005    }
1006    (port, dns_proxy)
1007}
1008
1009/// Addresses for a node within a tailnet.
1010#[derive(Debug, Clone, PartialEq, Eq, Hash)]
1011pub struct TailnetAddress {
1012    /// The IPv4 address of the node in the tailnet.
1013    pub ipv4: ipnet::Ipv4Net,
1014    /// The IPv6 address of the node in the tailnet.
1015    pub ipv6: ipnet::Ipv6Net,
1016}
1017
1018impl TailnetAddress {
1019    /// Report whether `addr` matches either address in this [`TailnetAddress`].
1020    pub fn contains(&self, addr: IpAddr) -> bool {
1021        match addr {
1022            IpAddr::V4(a) => self.ipv4.addr() == a,
1023            IpAddr::V6(a) => self.ipv6.addr() == a,
1024        }
1025    }
1026}
1027
1028impl From<&ts_control_serde::Node<'_>> for Node {
1029    fn from(value: &ts_control_serde::Node) -> Self {
1030        let fqdn_without_trailing_dot = value.name.strip_suffix('.').unwrap_or(&value.name);
1031
1032        let (hostname, tailnet) = match fqdn_without_trailing_dot.split_once('.') {
1033            Some((hostname, tailnet)) => (hostname, Some(tailnet.to_owned())),
1034            None => (fqdn_without_trailing_dot, None),
1035        };
1036
1037        let (peerapi_port, peerapi_dns_proxy) =
1038            peerapi_from_services(value.host_info.services.as_deref());
1039
1040        let cap_map = cap_map_from_serde(&value.cap_map);
1041        let service_vips = service_vips_from_cap_map(&cap_map);
1042
1043        // `addresses` is a variable-length `Vec<IpNet>` on the wire (Go `[]netip.Prefix`), not a
1044        // fixed (v4, v6) pair: an IPv6-off tailnet assigns only a v4 prefix. The whole list is kept
1045        // verbatim on `Node::addresses` (Go's `Node.Addresses`, which `IsRouter` tests routes
1046        // against); `tailnet_address` is the identity projection. Pick the first of each
1047        // family. The v4 prefix is the node's tailnet identity (always present on a normal node);
1048        // if somehow absent we fall back to the unspecified `0.0.0.0/32` rather than panicking.
1049        // The v6 prefix is optional — when the tailnet is IPv4-only there is none, and the overlay
1050        // never reads `ipv6` in that mode (gated on `enable_ipv6`); we synthesize the unspecified
1051        // `::/128` placeholder so the domain `TailnetAddress` stays infallible.
1052        let ipv4 = value
1053            .addresses
1054            .iter()
1055            .find_map(|p| match p {
1056                ipnet::IpNet::V4(n) => Some(*n),
1057                ipnet::IpNet::V6(_) => None,
1058            })
1059            .unwrap_or_else(|| ipnet::Ipv4Net::new(core::net::Ipv4Addr::UNSPECIFIED, 32).unwrap());
1060        let ipv6 = value
1061            .addresses
1062            .iter()
1063            .find_map(|p| match p {
1064                ipnet::IpNet::V6(n) => Some(*n),
1065                ipnet::IpNet::V4(_) => None,
1066            })
1067            .unwrap_or_else(|| ipnet::Ipv6Net::new(core::net::Ipv6Addr::UNSPECIFIED, 128).unwrap());
1068
1069        Self {
1070            id: value.id,
1071            stable_id: StableId(value.stable_id.0.to_string()),
1072
1073            hostname: hostname.to_owned(),
1074            user_id: value.user,
1075            tailnet,
1076
1077            tags: value
1078                .tags
1079                .as_ref()
1080                .map(|x| x.iter().map(|x| x.to_string()).collect())
1081                .unwrap_or_default(),
1082
1083            addresses: value.addresses.clone(),
1084            tailnet_address: TailnetAddress { ipv4, ipv6 },
1085            node_key: value.key,
1086            node_key_expiry: value.key_expiry,
1087            // Control's own verdict, carried verbatim; `ExpiryManager` only ever raises it.
1088            expired: value.expired,
1089            online: value.online,
1090            last_seen: value.last_seen,
1091            key_signature: value.key_signature.to_vec(),
1092            machine_key: value.machine,
1093            disco_key: value.disco_key,
1094
1095            unsigned_peer_api_only: value.unsigned_peer_api_only,
1096
1097            // Per capver-112, `AllowedIPs` null/absent means "same as `addresses`". Fall back to the
1098            // node's own assigned prefixes verbatim (whatever families the wire carried), not a
1099            // synthesized v4+v6 pair.
1100            //
1101            // `UnsignedPeerAPIOnly` clamps the result back to `addresses` whatever control sent,
1102            // mirroring Go's `upgradeNode` (`control/controlclient/map.go`): such a node is outside
1103            // tailnet lock's coverage, so a possibly-malicious control server must not be able to
1104            // grant it network access by handing it advertised routes (in the limit, `0.0.0.0/0`).
1105            // Unconditional, exactly as upstream — it does not depend on tailnet lock being
1106            // enabled here.
1107            accepted_routes: if value.unsigned_peer_api_only {
1108                value.addresses.clone()
1109            } else {
1110                value
1111                    .allowed_ips
1112                    .clone()
1113                    .unwrap_or_else(|| value.addresses.clone())
1114            },
1115            underlay_addresses: value.endpoints.clone(),
1116
1117            // legacy_derp_string is still in practical use as of 3/2026
1118            #[allow(deprecated)]
1119            derp_region: value
1120                .home_derp
1121                .or(value.legacy_derp_string)
1122                .or_else(|| value.host_info.net_info.as_ref()?.preferred_derp)
1123                .map(|x| ts_derp::RegionId(x.into())),
1124
1125            cap: value.cap,
1126            cap_map,
1127            peerapi_port,
1128            peerapi_dns_proxy,
1129            is_wireguard_only: value.is_wireguard_only,
1130            exit_node_dns_resolvers: value
1131                .exit_node_dns_resolvers
1132                .iter()
1133                .filter_map(Resolver::from_serde)
1134                .collect(),
1135            peer_relay: value.host_info.peer_relay,
1136            // Project the advertised SSH host keys (Go `Hostinfo.SSHHostKeys`), mapping the
1137            // borrowed `Option<Vec<&str>>` to owned `Vec<String>`; absent ⇒ empty (never
1138            // fabricated), matching how `services`/`peer_relay` above are projected from host_info.
1139            ssh_host_keys: value
1140                .host_info
1141                .ssh_host_keys
1142                .as_ref()
1143                .map(|keys| keys.iter().map(|k| k.to_string()).collect())
1144                .unwrap_or_default(),
1145            service_vips,
1146        }
1147    }
1148}
1149
1150/// An incremental update to a single already-known peer [`Node`], carried in
1151/// [`MapResponse::peers_changed_patch`][ts_control_serde::MapResponse::peers_changed_patch].
1152///
1153/// Control sends a patch (rather than a full node in `peers_changed`) when only a peer's
1154/// reachability changes mid-session — most importantly its UDP `endpoints`
1155/// and home [`derp_region`][PeerChange::derp_region] when an idle peer re-establishes connectivity.
1156/// Every field is `Option`: a patch sets only the fields it carries and leaves the rest of the
1157/// target node unchanged (see `PeerTracker::apply_peer_update` for the merge). Owned counterpart
1158/// of the borrow-bound [`ts_control_serde::PeerChange`]; the fields that map onto a domain
1159/// [`Node`] field are retained, including control's `online`/`last_seen` liveness deltas — the
1160/// dominant channel by which peer online transitions are delivered (see [`Node::online`]).
1161#[derive(Debug, Clone, PartialEq, Eq)]
1162pub struct PeerChange {
1163    /// The [`Node::id`] of the peer being mutated. If no peer with this id is in the current
1164    /// netmap, the patch is ignored (the wire contract — a patch never creates a node).
1165    pub id: Id,
1166    /// If `Some`, the peer's new home DERP region.
1167    pub derp_region: Option<ts_derp::RegionId>,
1168    /// If `Some`, the peer's new advertised capability version.
1169    pub cap: Option<CapabilityVersion>,
1170    /// If `Some`, the peer's new capability map (replaces the prior map wholesale).
1171    pub cap_map: Option<NodeCapMap>,
1172    /// If `Some`, the peer's new UDP underlay endpoints (`Endpoints` in Go; replaces the prior
1173    /// set). This is the field that lets magicsock re-handshake a peer that moved.
1174    pub underlay_addresses: Option<Vec<SocketAddr>>,
1175    /// If `Some`, the peer's new WireGuard public key (key rotation).
1176    pub node_key: Option<NodePublicKey>,
1177    /// If `Some`, the marshalled TKA signature over the new node key. Re-verified at the
1178    /// peer-trust chokepoint when tailnet-lock enforcement is active.
1179    pub key_signature: Option<Vec<u8>>,
1180    /// If `Some`, the peer's new disco public key.
1181    pub disco_key: Option<DiscoPublicKey>,
1182    /// If `Some`, the peer's new node-key expiry (`KeyExpiry` in Go). Maps to
1183    /// [`Node::node_key_expiry`]; carried so an expiry-only patch isn't lost until the next full
1184    /// resync.
1185    pub node_key_expiry: Option<DateTime<Utc>>,
1186    /// If `Some`, the peer's new online status (`PeerChange.Online`). `None` here means "this patch
1187    /// did not touch online", **not** "offline" — the merge sets [`Node::online`] only when present.
1188    pub online: Option<bool>,
1189    /// If `Some`, the peer's new last-seen time (`PeerChange.LastSeen`). Maps to [`Node::last_seen`].
1190    pub last_seen: Option<DateTime<Utc>>,
1191}
1192
1193impl From<&ts_control_serde::PeerChange<'_>> for PeerChange {
1194    fn from(value: &ts_control_serde::PeerChange) -> Self {
1195        Self {
1196            id: value.node_id,
1197            derp_region: value.derp_region.map(|x| ts_derp::RegionId(x.into())),
1198            cap: value.cap,
1199            cap_map: value.cap_map.as_ref().map(cap_map_from_serde),
1200            underlay_addresses: value.endpoints.clone(),
1201            node_key: value.key,
1202            key_signature: value.key_signature.map(|s| s.to_vec()),
1203            disco_key: value.disco_key,
1204            node_key_expiry: value.key_expiry,
1205            online: value.online,
1206            last_seen: value.last_seen,
1207        }
1208    }
1209}
1210
1211/// Identity of the user that owns a [`Node`], resolved from the netmap's `UserProfiles` table
1212/// (Go `tailcfg.UserProfile`). Owned counterpart of the borrow-bound
1213/// [`ts_control_serde::UserProfile`]. Keyed by [`UserProfile::id`] (== [`Node::user_id`]).
1214///
1215/// Mostly display-friendly text ([`login_name`](Self::login_name),
1216/// [`display_name`](Self::display_name)), plus [`groups`](Self::groups) — the one attribute here an
1217/// embedder can *authorise* on, because it is the one a node cannot re-derive from anything else
1218/// control sends.
1219#[derive(Debug, Clone, PartialEq, Eq)]
1220pub struct UserProfile {
1221    /// The integer id of the Tailscale user this profile describes (matches [`Node::user_id`]).
1222    pub id: ts_control_serde::UserId,
1223    /// An email-ish login name for display (e.g. `alice@example.com` / `alice@github`). May be
1224    /// empty if control sent none.
1225    pub login_name: String,
1226    /// The user's display name (e.g. `Alice Smith`), if the IdP provided one.
1227    pub display_name: Option<String>,
1228    /// The groups that contain this user and that the coordination server was configured to report
1229    /// to this node (Go `tailcfg.UserProfile.Groups`): SCIM groups (e.g.
1230    /// `engineering@example.com`) or tailnet-policy group names (e.g. `group:eng`).
1231    ///
1232    /// Carried in the order control sent it (control sorts it when it loads the profile from
1233    /// storage). **Empty** when control reported no groups — including every control server older
1234    /// than the field, which omits it entirely. An empty list therefore means "control told this
1235    /// node nothing", not "this user is in no group": treat it as no grant, never as a denial you
1236    /// can act on.
1237    pub groups: Vec<String>,
1238}
1239
1240impl From<&ts_control_serde::UserProfile<'_>> for UserProfile {
1241    fn from(value: &ts_control_serde::UserProfile) -> Self {
1242        Self {
1243            id: value.id,
1244            login_name: value.login_name.to_string(),
1245            display_name: value.display_name.as_deref().map(str::to_string),
1246            groups: value.groups.iter().map(|g| g.to_string()).collect(),
1247        }
1248    }
1249}
1250
1251impl UserProfile {
1252    /// The best human-facing label for this user: the login name when present, else the display
1253    /// name, else `None`. This is what a `WhoIs` surfaces as the owning user.
1254    pub fn best_label(&self) -> Option<String> {
1255        if !self.login_name.is_empty() {
1256            Some(self.login_name.clone())
1257        } else {
1258            self.display_name.clone()
1259        }
1260    }
1261}
1262
1263#[cfg(test)]
1264pub(crate) mod tests {
1265    use super::*;
1266
1267    /// The wire `Node.User` id must be carried onto the domain `Node.user_id` by the `From` impl
1268    /// (the field the runtime joins against the netmap `UserProfiles` table for `WhoIs.user`).
1269    /// Guards against the `From` impl wiring the wrong serde field or dropping it.
1270    #[test]
1271    fn from_wire_node_carries_user_id() {
1272        let mut wire = ts_control_serde::Node {
1273            user: 4242,
1274            ..Default::default()
1275        };
1276        wire.name = "host.tail.ts.net.".into();
1277        let domain: Node = (&wire).into();
1278        assert_eq!(domain.user_id, 4242);
1279
1280        // Default (no owner / tagged node) stays 0.
1281        let tagged = ts_control_serde::Node::default();
1282        assert_eq!(Node::from(&tagged).user_id, 0);
1283    }
1284
1285    /// The wire `Hostinfo.sshHostKeys` must be projected onto the domain `Node.ssh_host_keys`
1286    /// (the field `tailscale ssh` reads via `StatusNode` to pin a peer's host key). Present →
1287    /// carried verbatim; absent → empty (never fabricated).
1288    #[test]
1289    fn from_wire_node_carries_ssh_host_keys() {
1290        let wire = ts_control_serde::Node {
1291            host_info: ts_control_serde::HostInfo {
1292                ssh_host_keys: Some(vec![
1293                    "ssh-ed25519 AAAAC3Nz host",
1294                    "ecdsa-sha2-nistp256 AAAAE2Vj host",
1295                ]),
1296                ..Default::default()
1297            },
1298            ..Default::default()
1299        };
1300        let domain: Node = (&wire).into();
1301        assert_eq!(
1302            domain.ssh_host_keys,
1303            vec![
1304                "ssh-ed25519 AAAAC3Nz host".to_string(),
1305                "ecdsa-sha2-nistp256 AAAAE2Vj host".to_string(),
1306            ]
1307        );
1308
1309        // Absent on the wire → empty Vec, not fabricated.
1310        let bare = ts_control_serde::Node::default();
1311        assert!(Node::from(&bare).ssh_host_keys.is_empty());
1312    }
1313
1314    /// A node from an **IPv4-only** tailnet (IPv6-off control plane / Headscale) carries a
1315    /// single-element `addresses` list. This used to fail deserialization ("invalid length 1,
1316    /// expected a tuple of size 2") when `addresses` was a fixed 2-tuple; it must now parse and
1317    /// derive the v4 identity, with the unused v6 a synthesized placeholder.
1318    #[test]
1319    fn from_wire_node_ipv4_only_addresses() {
1320        let wire = ts_control_serde::Node {
1321            addresses: vec!["100.64.0.5/32".parse().unwrap()],
1322            ..Default::default()
1323        };
1324        let domain: Node = (&wire).into();
1325        assert_eq!(
1326            domain.tailnet_address.ipv4,
1327            "100.64.0.5/32".parse().unwrap()
1328        );
1329        // No v6 on the wire → unspecified placeholder (never read in IPv4-only mode).
1330        assert_eq!(
1331            domain.tailnet_address.ipv6,
1332            ipnet::Ipv6Net::new(core::net::Ipv6Addr::UNSPECIFIED, 128).unwrap()
1333        );
1334        // AllowedIPs absent → falls back to the node's own assigned prefixes (just the v4 here).
1335        assert_eq!(
1336            domain.accepted_routes,
1337            vec!["100.64.0.5/32".parse::<ipnet::IpNet>().unwrap()]
1338        );
1339    }
1340
1341    /// A dual-stack node carries both families (any order); the domain picks the first of each.
1342    #[test]
1343    fn from_wire_node_dual_stack_addresses() {
1344        let wire = ts_control_serde::Node {
1345            addresses: vec![
1346                "100.64.0.7/32".parse().unwrap(),
1347                "fd7a:115c:a1e0::7/128".parse().unwrap(),
1348            ],
1349            ..Default::default()
1350        };
1351        let domain: Node = (&wire).into();
1352        assert_eq!(
1353            domain.tailnet_address.ipv4,
1354            "100.64.0.7/32".parse().unwrap()
1355        );
1356        assert_eq!(
1357            domain.tailnet_address.ipv6,
1358            "fd7a:115c:a1e0::7/128".parse().unwrap()
1359        );
1360    }
1361
1362    /// A wire peer that owns `100.64.0.9/32` and is handed `route` plus the default route in its
1363    /// `AllowedIPs`. `unsigned` sets `UnsignedPeerAPIOnly`; everything else is identical between
1364    /// the two, so the only variable in the test below is that flag.
1365    fn wire_peer_advertising(
1366        stable_id: &'static str,
1367        route: &str,
1368        unsigned: bool,
1369    ) -> ts_control_serde::Node<'static> {
1370        ts_control_serde::Node {
1371            stable_id: ts_control_serde::StableNodeId(stable_id),
1372            addresses: vec!["100.64.0.9/32".parse().unwrap()],
1373            allowed_ips: Some(vec![
1374                "100.64.0.9/32".parse().unwrap(),
1375                route.parse().unwrap(),
1376                "0.0.0.0/0".parse().unwrap(),
1377            ]),
1378            unsigned_peer_api_only: unsigned,
1379            ..Default::default()
1380        }
1381    }
1382
1383    /// `UnsignedPeerAPIOnly` must clamp a peer's accepted routes back to its own addresses, so a
1384    /// control server cannot grant an unsigned (lock-exempt) peer network access via advertised
1385    /// routes. Mirrors Go's `upgradeNode` in `control/controlclient/map.go`.
1386    ///
1387    /// The signed peer is the control: it advertises the **same** route and the same default route,
1388    /// and keeps both. Without it this test would still pass if the `From` impl simply dropped every
1389    /// advertised route.
1390    #[test]
1391    fn from_wire_unsigned_peer_api_only_clamps_routes_to_own_addresses() {
1392        let own: ipnet::IpNet = "100.64.0.9/32".parse().unwrap();
1393        let subnet: ipnet::IpNet = "192.0.2.0/24".parse().unwrap();
1394        let default_route: ipnet::IpNet = "0.0.0.0/0".parse().unwrap();
1395
1396        let unsigned: Node = (&wire_peer_advertising("nUnsigned", "192.0.2.0/24", true)).into();
1397        let signed: Node = (&wire_peer_advertising("nSigned", "192.0.2.0/24", false)).into();
1398
1399        // The flag is carried onto the domain node, not silently dropped.
1400        assert!(unsigned.unsigned_peer_api_only);
1401        assert!(!signed.unsigned_peer_api_only);
1402
1403        // Unsigned: clamped to its own addresses. The advertised subnet and the default route are
1404        // both gone, whatever control sent.
1405        assert_eq!(unsigned.accepted_routes, vec![own]);
1406
1407        // Signed: the identical advertisement survives verbatim.
1408        assert_eq!(
1409            signed.accepted_routes,
1410            vec![own, subnet, default_route],
1411            "the clamp must be specific to UnsignedPeerAPIOnly, not a blanket route drop"
1412        );
1413
1414        // Consequences the rest of the fork reads. `is_router` reports the unsigned peer routes
1415        // nothing but itself...
1416        assert!(!unsigned.is_router());
1417        assert!(signed.is_router());
1418
1419        // ...and no route-install policy can resurrect the advertisement: even with
1420        // `--accept-routes` on AND the peer selected as the exit node — the most permissive input
1421        // `routes_to_install` accepts — the unsigned peer yields only its own address.
1422        let installed: Vec<_> = unsigned
1423            .routes_to_install(true, Some(&unsigned.stable_id))
1424            .copied()
1425            .collect();
1426        assert_eq!(installed, vec![own]);
1427
1428        // The same permissive inputs against the signed peer do install the subnet and the /0,
1429        // proving the difference is the flag and not the policy arguments.
1430        let installed_signed: Vec<_> = signed
1431            .routes_to_install(true, Some(&signed.stable_id))
1432            .copied()
1433            .collect();
1434        assert_eq!(installed_signed, vec![own, subnet, default_route]);
1435    }
1436
1437    /// The wire default (`UnsignedPeerAPIOnly` absent) must leave `AllowedIPs` untouched, including
1438    /// the capver-112 "null AllowedIPs means the node's own addresses" fallback. Guards against the
1439    /// clamp being applied on the wrong branch.
1440    #[test]
1441    fn from_wire_default_is_not_clamped() {
1442        let wire = ts_control_serde::Node {
1443            addresses: vec!["100.64.0.9/32".parse().unwrap()],
1444            allowed_ips: Some(vec!["198.51.100.0/24".parse().unwrap()]),
1445            ..Default::default()
1446        };
1447        assert!(!wire.unsigned_peer_api_only);
1448        let domain: Node = (&wire).into();
1449        assert_eq!(
1450            domain.accepted_routes,
1451            vec!["198.51.100.0/24".parse::<ipnet::IpNet>().unwrap()]
1452        );
1453    }
1454
1455    /// An unsigned peer with **no** `AllowedIPs` on the wire still lands on its own addresses (the
1456    /// clamp and the capver-112 fallback agree), and a multi-prefix unsigned peer keeps *all* of
1457    /// its assigned prefixes — the clamp is to `Addresses`, not to the v4/v6 identity pair.
1458    #[test]
1459    fn from_wire_unsigned_peer_clamp_keeps_every_assigned_prefix() {
1460        let wire = ts_control_serde::Node {
1461            addresses: vec![
1462                "100.64.0.9/32".parse().unwrap(),
1463                "fd7a:115c:a1e0::9/128".parse().unwrap(),
1464            ],
1465            allowed_ips: None,
1466            unsigned_peer_api_only: true,
1467            ..Default::default()
1468        };
1469        let domain: Node = (&wire).into();
1470        assert_eq!(
1471            domain.accepted_routes,
1472            vec![
1473                "100.64.0.9/32".parse::<ipnet::IpNet>().unwrap(),
1474                "fd7a:115c:a1e0::9/128".parse::<ipnet::IpNet>().unwrap(),
1475            ]
1476        );
1477        assert!(!domain.is_router());
1478    }
1479
1480    /// The deserialization regression itself: a MapResponse-style Node JSON with a 1-element
1481    /// `Addresses` array must parse (this is the exact shape the dev-Headscale sends).
1482    #[test]
1483    fn deserialize_node_with_single_address() {
1484        let json = r#"{
1485            "ID": 1,
1486            "StableID": "n1",
1487            "Name": "host.tail.ts.net.",
1488            "User": 1,
1489            "Addresses": ["100.64.0.9/32"],
1490            "Key": "nodekey:0000000000000000000000000000000000000000000000000000000000000000",
1491            "Machine": null,
1492            "DiscoKey": null,
1493            "AllowedIPs": null,
1494            "Endpoints": []
1495        }"#;
1496        let wire: ts_control_serde::Node = serde_json::from_str(json).expect("1-addr node parses");
1497        assert_eq!(wire.addresses.len(), 1);
1498        let domain: Node = (&wire).into();
1499        assert_eq!(
1500            domain.tailnet_address.ipv4,
1501            "100.64.0.9/32".parse().unwrap()
1502        );
1503    }
1504
1505    #[test]
1506    fn key_expiry_semantics() {
1507        let now: DateTime<Utc> = "2026-06-05T00:00:00Z".parse().unwrap();
1508        let past: DateTime<Utc> = "2020-01-01T00:00:00Z".parse().unwrap();
1509        let future: DateTime<Utc> = "2099-01-01T00:00:00Z".parse().unwrap();
1510
1511        let mut n = node("h", Some("t.ts.net"));
1512
1513        // No expiry set => never expired (Go zero-value semantics).
1514        n.node_key_expiry = None;
1515        assert!(!n.key_expired(now));
1516        assert_eq!(n.key_expiry(), None);
1517
1518        // Future expiry => not yet expired.
1519        n.node_key_expiry = Some(future);
1520        assert!(!n.key_expired(now));
1521        assert_eq!(n.key_expiry(), Some(future));
1522
1523        // Past expiry => expired.
1524        n.node_key_expiry = Some(past);
1525        assert!(n.key_expired(now));
1526    }
1527
1528    #[test]
1529    fn key_expiry_unix_agrees_with_chrono() {
1530        // The chrono-free variants (`key_expired_at_unix` / `key_expiry_unix`) must agree with the
1531        // chrono variants for the same none/future/past cases (Unix seconds of the same instants).
1532        let now: DateTime<Utc> = "2026-06-05T00:00:00Z".parse().unwrap();
1533        let past: DateTime<Utc> = "2020-01-01T00:00:00Z".parse().unwrap();
1534        let future: DateTime<Utc> = "2099-01-01T00:00:00Z".parse().unwrap();
1535        let now_unix = now.timestamp();
1536
1537        let mut n = node("h", Some("t.ts.net"));
1538
1539        // No expiry => never expired; the unix accessor reports `None`.
1540        n.node_key_expiry = None;
1541        assert_eq!(n.key_expired(now), n.key_expired_at_unix(now_unix));
1542        assert!(!n.key_expired_at_unix(now_unix));
1543        assert_eq!(n.key_expiry_unix(), None);
1544
1545        // Future expiry => not yet expired; unix accessor matches the chrono timestamp.
1546        n.node_key_expiry = Some(future);
1547        assert_eq!(n.key_expired(now), n.key_expired_at_unix(now_unix));
1548        assert!(!n.key_expired_at_unix(now_unix));
1549        assert_eq!(n.key_expiry_unix(), Some(future.timestamp()));
1550
1551        // Past expiry => expired; unix accessor matches the chrono timestamp.
1552        n.node_key_expiry = Some(past);
1553        assert_eq!(n.key_expired(now), n.key_expired_at_unix(now_unix));
1554        assert!(n.key_expired_at_unix(now_unix));
1555        assert_eq!(n.key_expiry_unix(), Some(past.timestamp()));
1556    }
1557
1558    #[test]
1559    fn key_expiry_boundary_is_not_expired() {
1560        // A key whose expiry exactly equals `now` is NOT expired: the code uses strict `<`, matching
1561        // Go's `Before`. Both the chrono and chrono-free variants must agree at the boundary.
1562        let now: DateTime<Utc> = "2026-06-05T00:00:00Z".parse().unwrap();
1563        let now_unix = now.timestamp();
1564
1565        let mut n = node("h", Some("t.ts.net"));
1566        n.node_key_expiry = Some(now);
1567
1568        assert!(!n.key_expired(now));
1569        assert!(!n.key_expired_at_unix(now_unix));
1570    }
1571
1572    #[test]
1573    fn is_peer_relay_returns_field() {
1574        let mut n = node("h", Some("t.ts.net"));
1575
1576        n.peer_relay = true;
1577        assert!(n.is_peer_relay());
1578
1579        n.peer_relay = false;
1580        assert!(!n.is_peer_relay());
1581    }
1582
1583    /// A minimal well-formed peer, shared with the `expiry` module's tests so both reason about
1584    /// the same node shape.
1585    pub(crate) fn test_node() -> Node {
1586        node("h", Some("t.ts.net"))
1587    }
1588
1589    fn node(hostname: &str, tailnet: Option<&str>) -> Node {
1590        Node {
1591            id: 1,
1592            stable_id: StableId("n1".to_string()),
1593            hostname: hostname.to_string(),
1594            user_id: 0,
1595            tailnet: tailnet.map(str::to_string),
1596            tags: vec![],
1597            addresses: vec![
1598                "100.64.0.1/32".parse().unwrap(),
1599                "fd7a::1/128".parse().unwrap(),
1600            ],
1601            tailnet_address: TailnetAddress {
1602                ipv4: "100.64.0.1/32".parse().unwrap(),
1603                ipv6: "fd7a::1/128".parse().unwrap(),
1604            },
1605            node_key: [0u8; 32].into(),
1606            node_key_expiry: None,
1607            expired: false,
1608            online: None,
1609            last_seen: None,
1610            key_signature: vec![],
1611            machine_key: None,
1612            disco_key: None,
1613            accepted_routes: vec![],
1614            underlay_addresses: vec![],
1615            derp_region: None,
1616            cap: CapabilityVersion::default(),
1617            cap_map: NodeCapMap::new(),
1618            peerapi_port: None,
1619            peerapi_dns_proxy: false,
1620            is_wireguard_only: false,
1621            exit_node_dns_resolvers: vec![],
1622            peer_relay: false,
1623            ssh_host_keys: vec![],
1624            service_vips: Default::default(),
1625            unsigned_peer_api_only: false,
1626        }
1627    }
1628
1629    #[test]
1630    fn matches_name_is_case_and_trailing_dot_insensitive() {
1631        let n = node("MyHost", Some("tail-scale.ts.net"));
1632
1633        // bare hostname, any case
1634        assert!(n.matches_name("myhost"));
1635        assert!(n.matches_name("MYHOST"));
1636        assert!(n.matches_name("MyHost"));
1637
1638        // fqdn, any case, with and without trailing dot
1639        assert!(n.matches_name("myhost.tail-scale.ts.net"));
1640        assert!(n.matches_name("MYHOST.TAIL-SCALE.TS.NET"));
1641        assert!(n.matches_name("myhost.tail-scale.ts.net."));
1642        assert!(n.matches_name("MyHost.Tail-Scale.TS.NET."));
1643
1644        // wrong host / wrong tailnet must not match
1645        assert!(!n.matches_name("other"));
1646        assert!(!n.matches_name("myhost.other.ts.net"));
1647    }
1648
1649    #[test]
1650    fn matches_name_no_tailnet() {
1651        let n = node("solo", None);
1652        assert!(n.matches_name("solo"));
1653        assert!(n.matches_name("SOLO."));
1654        assert!(!n.matches_name("solo.ts.net"));
1655    }
1656
1657    #[test]
1658    fn is_tailscale_ip_ranges() {
1659        // CGNAT v4
1660        assert!(is_tailscale_ip("100.64.0.1".parse().unwrap()));
1661        assert!(is_tailscale_ip("100.127.255.254".parse().unwrap()));
1662        // ChromeOS carve-out is excluded
1663        assert!(!is_tailscale_ip("100.115.92.5".parse().unwrap()));
1664        // outside CGNAT
1665        assert!(!is_tailscale_ip("10.0.0.1".parse().unwrap()));
1666        assert!(!is_tailscale_ip("100.128.0.1".parse().unwrap()));
1667        // Tailscale ULA v6
1668        assert!(is_tailscale_ip("fd7a:115c:a1e0::1".parse().unwrap()));
1669        assert!(!is_tailscale_ip("fd00::1".parse().unwrap()));
1670    }
1671
1672    /// Taildrop SSRF guard (defense-in-depth). `Device::send_file` rejects an upload destination
1673    /// unless `is_tailscale_ip(peer.peerapi_addr().ip())` holds. `Device::send_file` itself needs a
1674    /// live runtime (it goes through `self.channel()`), so it can't be unit-tested here; instead we
1675    /// test the exact composition the guard relies on — `is_tailscale_ip ∘ peerapi_addr` — against a
1676    /// `Node` whose `tailnet_address.ipv4` has been corrupted to a non-CGNAT (public) address. A
1677    /// well-formed peer always has a CGNAT 100.64.0.0/10 address, but the guard exists to catch a
1678    /// malformed/hostile node; this proves it would reject one.
1679    #[test]
1680    fn taildrop_ssrf_guard_rejects_non_cgnat_peerapi_addr() {
1681        let mut n = node("evil", Some("ts.net"));
1682        // Corrupt the peer to a public, non-CGNAT address and advertise a peerAPI port so
1683        // `peerapi_addr` returns `Some(_)`.
1684        n.tailnet_address.ipv4 = "1.2.3.4/32".parse().unwrap();
1685        n.peerapi_port = Some(443);
1686
1687        let addr = n
1688            .peerapi_addr()
1689            .expect("peerapi_addr yields Some with a port set");
1690        assert_eq!(addr.ip(), Ipv4Addr::new(1, 2, 3, 4));
1691        // The guard `if !is_tailscale_ip(dst.ip()) { return Err(BadRequest) }` WOULD reject this.
1692        assert!(
1693            !is_tailscale_ip(addr.ip()),
1694            "SSRF guard must reject a peer whose peerAPI addr is not a Tailscale CGNAT IP"
1695        );
1696
1697        // Conversely, a well-formed CGNAT peer passes the guard.
1698        let mut good = node("friend", Some("ts.net"));
1699        good.peerapi_port = Some(443);
1700        let good_addr = good.peerapi_addr().expect("peerapi_addr yields Some");
1701        assert!(is_tailscale_ip(good_addr.ip()));
1702    }
1703
1704    /// Ported from upstream's `TestNodeIsRouter` (`tailcfg/tailcfg_test.go`, `8d830599b`): a node
1705    /// is a router exactly when its `AllowedIPs` reach past its own `Addresses`. The absent case
1706    /// (a plain node advertising only its own addresses) is asserted alongside the present one,
1707    /// since "no routes besides my own" is the answer that must not drift.
1708    #[test]
1709    fn is_router_reports_routes_beyond_own_addresses() {
1710        let v4: ipnet::Ipv4Net = "100.64.0.1/32".parse().unwrap();
1711        let v6: ipnet::Ipv6Net = "fd7a:115c:a1e0::1/128".parse().unwrap();
1712        let self4 = ipnet::IpNet::V4(v4);
1713        let self6 = ipnet::IpNet::V6(v6);
1714
1715        let cases: &[(&str, Vec<ipnet::IpNet>, bool)] = &[
1716            ("empty", vec![], false),
1717            ("plain-ipv4", vec![self4], false),
1718            ("plain-ipv6", vec![self6], false),
1719            ("plain-ipv4-ipv6", vec![self4, self6], false),
1720            ("duplicates", vec![self4, self4], false),
1721            (
1722                "exit-node-ipv4",
1723                vec![self4, "0.0.0.0/0".parse().unwrap()],
1724                true,
1725            ),
1726            ("exit-node-ipv6", vec![self6, "::/0".parse().unwrap()], true),
1727            (
1728                "exit-node-ipv4-ipv6",
1729                vec![
1730                    self4,
1731                    self6,
1732                    "0.0.0.0/0".parse().unwrap(),
1733                    "::/0".parse().unwrap(),
1734                ],
1735                true,
1736            ),
1737            (
1738                "subnet-router-ipv4",
1739                vec![self4, "192.0.2.0/24".parse().unwrap()],
1740                true,
1741            ),
1742            (
1743                "subnet-router-ipv6",
1744                vec![self6, "2001:db8::/32".parse().unwrap()],
1745                true,
1746            ),
1747            (
1748                "subnet-router-ipv4-ipv6",
1749                vec![
1750                    self4,
1751                    self6,
1752                    "192.0.2.0/24".parse().unwrap(),
1753                    "2001:db8::/32".parse().unwrap(),
1754                ],
1755                true,
1756            ),
1757            // Go's `IsRouter` has no Tailscale-range exception: another peer's /32 is still a
1758            // routed address. This is where it parts ways with `is_subnet_route`.
1759            (
1760                "other-tailnet-host",
1761                vec![self4, "100.64.5.5/32".parse().unwrap()],
1762                true,
1763            ),
1764        ];
1765
1766        for (name, allowed, want) in cases {
1767            let mut n = node("host", Some("ts.net"));
1768            n.addresses = vec![self4, self6];
1769            n.tailnet_address = TailnetAddress { ipv4: v4, ipv6: v6 };
1770            n.accepted_routes = allowed.clone();
1771            assert_eq!(n.is_router(), *want, "{name}");
1772        }
1773    }
1774
1775    /// Go's `IsRouter` tests each `AllowedIPs` prefix against the node's **whole** `Addresses`
1776    /// slice, so every prefix control assigned is "its own". The wire field is a variable-length
1777    /// list, not a v4/v6 pair, so a tailnet may hand a node more than one prefix of a family; such
1778    /// a node must not be reported as a router on account of the extra one — which comparing only
1779    /// against the first-of-family `tailnet_address` pair does. Runs through the production `From`
1780    /// impl so the retention of the full list is pinned along with the predicate.
1781    #[test]
1782    fn is_router_tests_every_assigned_address_not_only_the_first_of_each_family() {
1783        let second4: ipnet::IpNet = "100.64.0.9/32".parse().unwrap();
1784        let second6: ipnet::IpNet = "fd7a:115c:a1e0::9/128".parse().unwrap();
1785        let wire = ts_control_serde::Node {
1786            addresses: vec![
1787                "100.64.0.1/32".parse().unwrap(),
1788                second4,
1789                "fd7a:115c:a1e0::1/128".parse().unwrap(),
1790                second6,
1791            ],
1792            ..Default::default()
1793        };
1794        let domain: Node = (&wire).into();
1795
1796        // The identity projection is still the first prefix of each family...
1797        assert_eq!(
1798            domain.tailnet_address.ipv4,
1799            "100.64.0.1/32".parse().unwrap()
1800        );
1801        // ...but every assigned prefix is retained, and (AllowedIPs absent ⇒ routes are exactly
1802        // the addresses) none of them makes the node a router.
1803        assert_eq!(domain.addresses, wire.addresses);
1804        assert!(
1805            !domain.is_router(),
1806            "a node whose routes are exactly its own assigned prefixes is not a router"
1807        );
1808
1809        // Either second-of-family address on its own is still not a routed prefix.
1810        for extra in [second4, second6] {
1811            let mut n = domain.clone();
1812            n.accepted_routes = vec![extra];
1813            assert!(
1814                !n.is_router(),
1815                "{extra} is one of this node's own addresses"
1816            );
1817        }
1818
1819        // The predicate still fires for a route that does reach past every assigned address.
1820        let mut router = domain.clone();
1821        router.accepted_routes.push("192.0.2.0/24".parse().unwrap());
1822        assert!(router.is_router(), "a real subnet route makes it a router");
1823    }
1824
1825    #[test]
1826    fn is_subnet_route_distinguishes_self_from_subnet() {
1827        let n = node("host", Some("ts.net"));
1828
1829        // The node's own /32 and /128 are self-addresses, not subnet routes.
1830        assert!(!n.is_subnet_route(&"100.64.0.1/32".parse().unwrap()));
1831        assert!(!n.is_subnet_route(&"fd7a::1/128".parse().unwrap()));
1832        // A different single Tailscale IP is still a self-address (Tailscale-assigned host).
1833        assert!(!n.is_subnet_route(&"100.64.5.5/32".parse().unwrap()));
1834        // A LAN /24 the node advertises is a subnet route.
1835        assert!(n.is_subnet_route(&"192.168.1.0/24".parse().unwrap()));
1836        // A single non-Tailscale host IP counts as a subnet route.
1837        assert!(n.is_subnet_route(&"8.8.8.8/32".parse().unwrap()));
1838        // The default route is treated as a subnet route.
1839        assert!(n.is_subnet_route(&"0.0.0.0/0".parse().unwrap()));
1840        assert!(n.is_subnet_route(&"::/0".parse().unwrap()));
1841    }
1842
1843    #[test]
1844    fn routes_to_install_gates_subnets_on_accept_routes() {
1845        let mut n = node("host", Some("ts.net"));
1846        let self4: ipnet::IpNet = "100.64.0.1/32".parse().unwrap();
1847        let self6: ipnet::IpNet = "fd7a::1/128".parse().unwrap();
1848        let subnet: ipnet::IpNet = "192.168.1.0/24".parse().unwrap();
1849        n.accepted_routes = vec![self4, self6, subnet];
1850
1851        // accept_routes off: only the self addresses are installed.
1852        let off: Vec<_> = n.routes_to_install(false, None).copied().collect();
1853        assert_eq!(off, vec![self4, self6]);
1854
1855        // accept_routes on: the advertised subnet is installed too.
1856        let on: Vec<_> = n.routes_to_install(true, None).copied().collect();
1857        assert_eq!(on, vec![self4, self6, subnet]);
1858    }
1859
1860    #[test]
1861    fn routes_to_install_default_route_only_for_selected_exit_node() {
1862        let mut n = node("host", Some("ts.net"));
1863        n.stable_id = StableId("exit1".to_string());
1864        let self4: ipnet::IpNet = "100.64.0.1/32".parse().unwrap();
1865        let default4: ipnet::IpNet = "0.0.0.0/0".parse().unwrap();
1866        let default6: ipnet::IpNet = "::/0".parse().unwrap();
1867        n.accepted_routes = vec![self4, default4, default6];
1868
1869        // No exit node selected: default routes are excluded even with accept_routes on
1870        // (fail-closed — internet-bound traffic has no overlay route and is dropped).
1871        let none_off: Vec<_> = n.routes_to_install(false, None).copied().collect();
1872        assert_eq!(none_off, vec![self4]);
1873        let none_on: Vec<_> = n.routes_to_install(true, None).copied().collect();
1874        assert_eq!(none_on, vec![self4]);
1875
1876        // A *different* peer selected as exit node: this peer still gets no default route.
1877        let other = StableId("exit2".to_string());
1878        let other_sel: Vec<_> = n.routes_to_install(false, Some(&other)).copied().collect();
1879        assert_eq!(other_sel, vec![self4]);
1880
1881        // This peer selected as the exit node: its default routes are installed.
1882        let me = StableId("exit1".to_string());
1883        let sel: Vec<_> = n.routes_to_install(false, Some(&me)).copied().collect();
1884        assert_eq!(sel, vec![self4, default4, default6]);
1885    }
1886
1887    fn exit_node_with(id: &str, ipv4: &str, hostname: &str, tailnet: Option<&str>) -> Node {
1888        let mut n = node(hostname, tailnet);
1889        n.stable_id = StableId(id.to_string());
1890        n.tailnet_address.ipv4 = format!("{ipv4}/32").parse().unwrap();
1891        n
1892    }
1893
1894    #[test]
1895    fn exit_node_selector_resolves_by_id_ip_and_name() {
1896        let a = exit_node_with("nA", "100.64.0.5", "alpha", Some("ts.net"));
1897        let b = exit_node_with("nB", "100.64.0.6", "beta", Some("ts.net"));
1898        let peers = [a, b];
1899        let it = || peers.iter();
1900
1901        // By stable id.
1902        assert_eq!(
1903            ExitNodeSelector::StableId(StableId("nB".into())).resolve(it()),
1904            Some(StableId("nB".into()))
1905        );
1906        // By tailnet IP.
1907        assert_eq!(
1908            ExitNodeSelector::Ip("100.64.0.5".parse().unwrap()).resolve(it()),
1909            Some(StableId("nA".into()))
1910        );
1911        // By MagicDNS name (fqdn, case-insensitive).
1912        assert_eq!(
1913            ExitNodeSelector::Name("BETA.ts.net".into()).resolve(it()),
1914            Some(StableId("nB".into()))
1915        );
1916        // By bare hostname.
1917        assert_eq!(
1918            ExitNodeSelector::Name("alpha".into()).resolve(it()),
1919            Some(StableId("nA".into()))
1920        );
1921        // Unresolvable selector => None (fail-closed at the call site).
1922        assert_eq!(
1923            ExitNodeSelector::Ip("100.64.0.99".parse().unwrap()).resolve(it()),
1924            None
1925        );
1926        assert_eq!(ExitNodeSelector::Name("ghost".into()).resolve(it()), None);
1927    }
1928
1929    #[test]
1930    fn exit_node_selector_resolution_is_deterministic_on_ties() {
1931        // Two peers sharing a name (transient netmap state): the smallest stable id wins, so the
1932        // outbound table and inbound source filter — which resolve independently — agree.
1933        let a = exit_node_with("nZ", "100.64.0.5", "dup", Some("ts.net"));
1934        let b = exit_node_with("nA", "100.64.0.6", "dup", Some("ts.net"));
1935        let peers = [a, b];
1936
1937        assert_eq!(
1938            ExitNodeSelector::Name("dup".into()).resolve(peers.iter()),
1939            Some(StableId("nA".into())),
1940            "smallest stable id wins the tie"
1941        );
1942        // Order of iteration must not change the result.
1943        assert_eq!(
1944            ExitNodeSelector::Name("dup".into()).resolve(peers.iter().rev()),
1945            Some(StableId("nA".into()))
1946        );
1947    }
1948
1949    #[test]
1950    fn peerapi_doh_url_requires_port_and_capability() {
1951        let mut n = node("exit", Some("ts.net"));
1952        n.tailnet_address.ipv4 = "100.64.0.5/32".parse().unwrap();
1953
1954        // No peerAPI port advertised: cannot proxy DNS.
1955        n.peerapi_port = None;
1956        n.cap = CapabilityVersion::V130;
1957        assert_eq!(n.peerapi_doh_url(), None);
1958
1959        // Port advertised but capability too old and no explicit service: cannot proxy.
1960        n.peerapi_port = Some(8080);
1961        n.cap = CapabilityVersion::V25;
1962        n.peerapi_dns_proxy = false;
1963        assert_eq!(n.peerapi_doh_url(), None);
1964
1965        // Port + new-enough capability: yields the DoH URL on the IPv4 address.
1966        n.cap = CapabilityVersion::V26;
1967        assert_eq!(
1968            n.peerapi_doh_url().as_deref(),
1969            Some("http://100.64.0.5:8080/dns-query")
1970        );
1971
1972        // Port + explicit peerapi-dns-proxy service, even with an old capability.
1973        n.cap = CapabilityVersion::V25;
1974        n.peerapi_dns_proxy = true;
1975        assert_eq!(
1976            n.peerapi_doh_url().as_deref(),
1977            Some("http://100.64.0.5:8080/dns-query")
1978        );
1979
1980        // WireGuard-only peers never run a peerAPI: no DoH URL even with a port.
1981        n.is_wireguard_only = true;
1982        assert_eq!(n.peerapi_doh_url(), None);
1983    }
1984
1985    #[test]
1986    fn peerapi_doh_addr_matches_url_gate() {
1987        let mut n = node("exit", Some("ts.net"));
1988        n.tailnet_address.ipv4 = "100.64.0.5/32".parse().unwrap();
1989        n.peerapi_port = Some(8080);
1990        n.cap = CapabilityVersion::V26;
1991
1992        // The addr form the DoH client dials is the same gated endpoint as the URL.
1993        assert_eq!(
1994            n.peerapi_doh_addr(),
1995            Some("100.64.0.5:8080".parse().unwrap())
1996        );
1997        // And it composes into exactly the URL form.
1998        assert_eq!(
1999            n.peerapi_doh_url().as_deref(),
2000            Some("http://100.64.0.5:8080/dns-query")
2001        );
2002
2003        // Gated off the same way: no port => no addr.
2004        n.peerapi_port = None;
2005        assert_eq!(n.peerapi_doh_addr(), None);
2006    }
2007
2008    #[test]
2009    fn peerapi_addr_returns_addr_when_advertised() {
2010        let mut n = node("peer", Some("ts.net"));
2011        n.tailnet_address.ipv4 = "100.64.0.5/32".parse().unwrap();
2012        n.peerapi_port = Some(8089);
2013
2014        // Not gated on the DNS-proxy capability: a plain advertised peerAPI port is enough.
2015        assert_eq!(n.peerapi_addr(), Some("100.64.0.5:8089".parse().unwrap()));
2016    }
2017
2018    #[test]
2019    fn peerapi_addr_none_when_no_port() {
2020        let mut n = node("peer", Some("ts.net"));
2021        n.tailnet_address.ipv4 = "100.64.0.5/32".parse().unwrap();
2022        n.peerapi_port = None;
2023
2024        assert_eq!(n.peerapi_addr(), None);
2025    }
2026
2027    #[test]
2028    fn peerapi_addr_none_for_wireguard_only() {
2029        let mut n = node("peer", Some("ts.net"));
2030        n.tailnet_address.ipv4 = "100.64.0.5/32".parse().unwrap();
2031        n.peerapi_port = Some(8089);
2032        n.is_wireguard_only = true;
2033
2034        // WireGuard-only peers run no peerAPI, even with a port set.
2035        assert_eq!(n.peerapi_addr(), None);
2036    }
2037
2038    #[test]
2039    fn can_share_files_gated_on_self_capability() {
2040        let mut n = node("self", Some("ts.net"));
2041        assert!(
2042            !n.can_share_files(),
2043            "no cap → file sharing not enabled (fail-closed)"
2044        );
2045        n.cap_map
2046            .insert("https://tailscale.com/cap/file-sharing".to_string(), vec![]);
2047        assert!(n.can_share_files(), "the file-sharing cap enables it");
2048    }
2049
2050    #[test]
2051    fn is_file_sharing_target_gated_on_peer_capability() {
2052        let mut n = node("peer", Some("ts.net"));
2053        assert!(
2054            !n.is_file_sharing_target(),
2055            "no cap → not an explicit target"
2056        );
2057        n.cap_map
2058            .insert("tailscale.com/cap/file-sharing-target".to_string(), vec![]);
2059        assert!(
2060            n.is_file_sharing_target(),
2061            "the file-sharing-target cap marks a cross-owner target"
2062        );
2063    }
2064
2065    #[test]
2066    fn resolves_subdomains_gated_on_the_node_attribute() {
2067        let mut n = node("peer", Some("ts.net"));
2068        assert!(
2069            !n.resolves_subdomains(),
2070            "no attribute → not a subdomain host: control has to opt the node in"
2071        );
2072        n.cap_map
2073            .insert("dns-subdomain-resolve".to_string(), vec![]);
2074        assert!(
2075            n.resolves_subdomains(),
2076            "the dns-subdomain-resolve attribute makes this node a subdomain host"
2077        );
2078    }
2079
2080    /// The DNS forwarder's TCP retry is ON by default, so the attribute has to be read as the *off*
2081    /// switch it is: absent means retry. Getting the polarity backwards would silently disable the
2082    /// retry on every tailnet that never set the attribute — the failure the retry exists to remove.
2083    #[test]
2084    fn dns_forwarder_tcp_retries_disabled_only_by_the_node_attribute() {
2085        let mut n = node("peer", Some("ts.net"));
2086        assert!(
2087            !n.disable_dns_forwarder_tcp_retries(),
2088            "no attribute → the retry stays on: this is the off switch, not the on switch"
2089        );
2090        n.cap_map
2091            .insert("dns-forwarder-disable-tcp-retries".to_string(), vec![]);
2092        assert!(
2093            n.disable_dns_forwarder_tcp_retries(),
2094            "the dns-forwarder-disable-tcp-retries attribute turns the TCP retry off"
2095        );
2096    }
2097
2098    /// The debug-always-stun attribute is control's only way to keep the periodic STUN sweep running
2099    /// once the datapath has gone idle, so its key has to be the literal Go sends
2100    /// (`NodeAttrDebugForceBackgroundSTUN`). Absent is the quiet default.
2101    #[test]
2102    fn force_background_stun_gated_on_the_node_attribute() {
2103        let mut n = node("self", Some("ts.net"));
2104        assert!(
2105            !n.force_background_stun(),
2106            "no attribute → the idle stop applies, which is the quiet default"
2107        );
2108        n.cap_map.insert("debug-always-stun".to_string(), vec![]);
2109        assert!(
2110            n.force_background_stun(),
2111            "the debug-always-stun attribute keeps the background sweep running"
2112        );
2113    }
2114
2115    #[test]
2116    fn one_cgnat_is_a_tri_state_read_off_the_query_string_keys() {
2117        let mut n = node("self", Some("ts.net"));
2118        assert_eq!(
2119            n.one_cgnat(),
2120            None,
2121            "neither attribute → control has no opinion, the threshold decides"
2122        );
2123
2124        // The key carries a query string; the bare `one-cgnat` is not the attribute and must not
2125        // be mistaken for either half of the tri-state.
2126        n.cap_map.insert("one-cgnat".to_string(), vec![]);
2127        assert_eq!(
2128            n.one_cgnat(),
2129            None,
2130            "a bare `one-cgnat` key is not one of the two attributes control sets"
2131        );
2132
2133        n.cap_map.insert("one-cgnat?v=false".to_string(), vec![]);
2134        assert_eq!(
2135            n.one_cgnat(),
2136            Some(false),
2137            "`one-cgnat?v=false` forces one route per peer"
2138        );
2139
2140        n.cap_map.insert("one-cgnat?v=true".to_string(), vec![]);
2141        assert_eq!(
2142            n.one_cgnat(),
2143            Some(true),
2144            "a node holding both attributes collapses: the enabling attribute is checked first"
2145        );
2146
2147        n.cap_map.remove("one-cgnat?v=false");
2148        assert_eq!(
2149            n.one_cgnat(),
2150            Some(true),
2151            "`one-cgnat?v=true` alone collapses"
2152        );
2153    }
2154
2155    #[test]
2156    fn delta_updates_disabled_reads_the_disable_delta_updates_attribute() {
2157        let mut n = node("self", Some("ts.net"));
2158        assert!(
2159            !n.delta_updates_disabled(),
2160            "absent attribute → the incremental path, which is the default"
2161        );
2162
2163        // The attribute is the bare key; the presence of the key is the whole signal, and its
2164        // value (control sends an empty one) is never read.
2165        n.cap_map
2166            .insert("disable-delta-updates".to_string(), vec![]);
2167        assert!(
2168            n.delta_updates_disabled(),
2169            "control granted the attribute → decline the incremental path"
2170        );
2171
2172        n.cap_map.remove("disable-delta-updates");
2173        assert!(
2174            !n.delta_updates_disabled(),
2175            "control withdrawing the attribute returns the node to the incremental path"
2176        );
2177    }
2178
2179    #[test]
2180    fn silent_disco_reads_the_silent_disco_attribute() {
2181        let mut n = node("self", Some("ts.net"));
2182        assert!(
2183            !n.silent_disco(),
2184            "absent attribute → keep heartbeating, which is the default"
2185        );
2186
2187        // The attribute is the bare key; the presence of the key is the whole signal, and its
2188        // value (control sends an empty one) is never read.
2189        n.cap_map.insert("silent-disco".to_string(), vec![]);
2190        assert!(
2191            n.silent_disco(),
2192            "control granted the attribute → stop heartbeating peers"
2193        );
2194
2195        n.cap_map.remove("silent-disco");
2196        assert!(
2197            !n.silent_disco(),
2198            "control withdrawing the attribute returns the node to the heartbeat cadence"
2199        );
2200    }
2201
2202    /// Control's `only-tcp-443` attribute is read off the self node's cap map, and — because
2203    /// upstream's `SetOnlyTCP443` is live rather than start-time — withdrawing it reads back as
2204    /// `false` on the very next netmap.
2205    #[test]
2206    fn only_tcp_443_reads_the_only_tcp_443_attribute() {
2207        let mut n = node("self", Some("ts.net"));
2208        assert!(
2209            !n.only_tcp_443(),
2210            "absent attribute → UDP as before, which is the default"
2211        );
2212
2213        // The attribute is the bare key; the presence of the key is the whole signal, and its
2214        // value (control sends an empty one) is never read.
2215        n.cap_map.insert("only-tcp-443".to_string(), vec![]);
2216        assert!(
2217            n.only_tcp_443(),
2218            "control granted the attribute → this node emits nothing but TCP/443"
2219        );
2220
2221        n.cap_map.remove("only-tcp-443");
2222        assert!(
2223            !n.only_tcp_443(),
2224            "control withdrawing the attribute must restore UDP, with no restart"
2225        );
2226    }
2227
2228    #[test]
2229    fn peerapi_from_services_extracts_v4_port_and_dns_proxy_flag() {
2230        use ts_control_serde::{Service, ServiceProto};
2231
2232        let services = [
2233            Service {
2234                proto: ServiceProto::PeerApi4,
2235                port: 8080,
2236                description: "peerapi".into(),
2237            },
2238            Service {
2239                proto: ServiceProto::PeerApi6,
2240                port: 9090,
2241                description: "peerapi6".into(),
2242            },
2243            Service {
2244                proto: ServiceProto::PeerApiDnsProxy,
2245                port: 1,
2246                description: "dns".into(),
2247            },
2248        ];
2249        let (port, dns_proxy) = peerapi_from_services(Some(&services));
2250        assert_eq!(port, Some(8080), "only the IPv4 peerAPI port is taken");
2251        assert!(dns_proxy);
2252
2253        // No services at all.
2254        assert_eq!(peerapi_from_services(None), (None, false));
2255    }
2256
2257    #[test]
2258    fn exit_node_selector_parses_ip_vs_name() {
2259        assert_eq!(
2260            "100.64.0.5".parse::<ExitNodeSelector>().unwrap(),
2261            ExitNodeSelector::Ip("100.64.0.5".parse().unwrap())
2262        );
2263        assert_eq!(
2264            "fd7a::5".parse::<ExitNodeSelector>().unwrap(),
2265            ExitNodeSelector::Ip("fd7a::5".parse().unwrap())
2266        );
2267        assert_eq!(
2268            "my-exit.ts.net".parse::<ExitNodeSelector>().unwrap(),
2269            ExitNodeSelector::Name("my-exit.ts.net".into())
2270        );
2271    }
2272}