use std::collections::HashMap;
use std::net::{IpAddr, SocketAddr};
use serde::{Deserialize, Serialize};
fn parse_external_advert_addr(raw: &str, bind_port: u16) -> Option<SocketAddr> {
if let Ok(sa) = raw.parse::<SocketAddr>() {
return Some(sa);
}
let ip: IpAddr = raw.parse().ok()?;
Some(SocketAddr::new(ip, bind_port))
}
fn parse_bind_port(raw: &str) -> Option<u16> {
raw.parse::<SocketAddr>().ok().map(|sa| sa.port())
}
const DEFAULT_UDP_BIND_ADDR: &str = "0.0.0.0:2121";
const DEFAULT_UDP_MTU: u16 = 1280;
const DEFAULT_UDP_RECV_BUF: usize = 16 * 1024 * 1024;
const DEFAULT_UDP_SEND_BUF: usize = 8 * 1024 * 1024;
#[derive(Debug, Clone, Default, Serialize, Deserialize)]
#[serde(deny_unknown_fields)]
pub struct UdpConfig {
#[serde(default, skip_serializing_if = "Option::is_none")]
pub bind_addr: Option<String>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub mtu: Option<u16>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub recv_buf_size: Option<usize>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub send_buf_size: Option<usize>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub advertise_on_nostr: Option<bool>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub public: Option<bool>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub external_addr: Option<String>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub outbound_only: Option<bool>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub accept_connections: Option<bool>,
}
impl UdpConfig {
pub fn bind_addr(&self) -> &str {
if self.outbound_only() {
"0.0.0.0:0"
} else {
self.bind_addr.as_deref().unwrap_or(DEFAULT_UDP_BIND_ADDR)
}
}
pub fn mtu(&self) -> u16 {
self.mtu.unwrap_or(DEFAULT_UDP_MTU)
}
pub fn recv_buf_size(&self) -> usize {
self.recv_buf_size.unwrap_or(DEFAULT_UDP_RECV_BUF)
}
pub fn send_buf_size(&self) -> usize {
self.send_buf_size.unwrap_or(DEFAULT_UDP_SEND_BUF)
}
pub fn advertise_on_nostr(&self) -> bool {
if self.outbound_only() {
false
} else {
self.advertise_on_nostr.unwrap_or(false)
}
}
pub fn is_public(&self) -> bool {
self.public.unwrap_or(false)
}
pub fn external_advert_addr(&self) -> Option<SocketAddr> {
let raw = self.external_addr.as_deref()?;
let bind_port = parse_bind_port(self.bind_addr())?;
parse_external_advert_addr(raw, bind_port)
}
pub fn outbound_only(&self) -> bool {
self.outbound_only.unwrap_or(false)
}
pub fn accept_connections(&self) -> bool {
self.accept_connections.unwrap_or(true)
}
}
#[cfg(feature = "sim-transport")]
const DEFAULT_SIM_MTU: u16 = 1280;
#[cfg(feature = "sim-transport")]
const DEFAULT_SIM_NETWORK: &str = "default";
#[cfg(feature = "sim-transport")]
#[derive(Debug, Clone, Default, Serialize, Deserialize)]
#[serde(deny_unknown_fields)]
pub struct SimTransportConfig {
#[serde(default, skip_serializing_if = "Option::is_none")]
pub network: Option<String>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub addr: Option<String>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub mtu: Option<u16>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub auto_connect: Option<bool>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub accept_connections: Option<bool>,
}
#[cfg(feature = "sim-transport")]
impl SimTransportConfig {
pub fn network(&self) -> &str {
self.network.as_deref().unwrap_or(DEFAULT_SIM_NETWORK)
}
pub fn mtu(&self) -> u16 {
self.mtu.unwrap_or(DEFAULT_SIM_MTU)
}
pub fn auto_connect(&self) -> bool {
self.auto_connect.unwrap_or(false)
}
pub fn accept_connections(&self) -> bool {
self.accept_connections.unwrap_or(true)
}
}
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(untagged)]
pub enum TransportInstances<T> {
Single(T),
Named(HashMap<String, T>),
}
impl<T> TransportInstances<T> {
pub fn len(&self) -> usize {
match self {
TransportInstances::Single(_) => 1,
TransportInstances::Named(map) => map.len(),
}
}
pub fn is_empty(&self) -> bool {
match self {
TransportInstances::Single(_) => false,
TransportInstances::Named(map) => map.is_empty(),
}
}
pub fn iter(&self) -> impl Iterator<Item = (Option<&str>, &T)> {
match self {
TransportInstances::Single(config) => vec![(None, config)].into_iter(),
TransportInstances::Named(map) => map
.iter()
.map(|(k, v)| (Some(k.as_str()), v))
.collect::<Vec<_>>()
.into_iter(),
}
}
}
impl<T> Default for TransportInstances<T> {
fn default() -> Self {
TransportInstances::Named(HashMap::new())
}
}
const DEFAULT_ETHERNET_ETHERTYPE: u16 = 0x2121;
const DEFAULT_ETHERNET_RECV_BUF: usize = 2 * 1024 * 1024;
const DEFAULT_ETHERNET_SEND_BUF: usize = 2 * 1024 * 1024;
const DEFAULT_BEACON_INTERVAL_SECS: u64 = 30;
const MIN_BEACON_INTERVAL_SECS: u64 = 10;
#[derive(Debug, Clone, Default, Serialize, Deserialize)]
#[serde(deny_unknown_fields)]
pub struct EthernetConfig {
pub interface: String,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub ethertype: Option<u16>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub mtu: Option<u16>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub recv_buf_size: Option<usize>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub send_buf_size: Option<usize>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub discovery: Option<bool>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub announce: Option<bool>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub auto_connect: Option<bool>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub accept_connections: Option<bool>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub beacon_interval_secs: Option<u64>,
}
impl EthernetConfig {
pub fn ethertype(&self) -> u16 {
self.ethertype.unwrap_or(DEFAULT_ETHERNET_ETHERTYPE)
}
pub fn recv_buf_size(&self) -> usize {
self.recv_buf_size.unwrap_or(DEFAULT_ETHERNET_RECV_BUF)
}
pub fn send_buf_size(&self) -> usize {
self.send_buf_size.unwrap_or(DEFAULT_ETHERNET_SEND_BUF)
}
pub fn discovery(&self) -> bool {
self.discovery.unwrap_or(true)
}
pub fn announce(&self) -> bool {
self.announce.unwrap_or(false)
}
pub fn auto_connect(&self) -> bool {
self.auto_connect.unwrap_or(false)
}
pub fn accept_connections(&self) -> bool {
self.accept_connections.unwrap_or(false)
}
pub fn beacon_interval_secs(&self) -> u64 {
self.beacon_interval_secs
.unwrap_or(DEFAULT_BEACON_INTERVAL_SECS)
.max(MIN_BEACON_INTERVAL_SECS)
}
}
const DEFAULT_TCP_MTU: u16 = 1400;
const DEFAULT_TCP_CONNECT_TIMEOUT_MS: u64 = 5000;
const DEFAULT_TCP_KEEPALIVE_SECS: u64 = 30;
const DEFAULT_TCP_RECV_BUF: usize = 2 * 1024 * 1024;
const DEFAULT_TCP_SEND_BUF: usize = 2 * 1024 * 1024;
const DEFAULT_TCP_MAX_INBOUND: usize = 256;
#[derive(Debug, Clone, Default, Serialize, Deserialize)]
#[serde(deny_unknown_fields)]
pub struct TcpConfig {
#[serde(default, skip_serializing_if = "Option::is_none")]
pub bind_addr: Option<String>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub mtu: Option<u16>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub connect_timeout_ms: Option<u64>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub nodelay: Option<bool>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub keepalive_secs: Option<u64>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub recv_buf_size: Option<usize>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub send_buf_size: Option<usize>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub max_inbound_connections: Option<usize>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub advertise_on_nostr: Option<bool>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub external_addr: Option<String>,
}
impl TcpConfig {
pub fn mtu(&self) -> u16 {
self.mtu.unwrap_or(DEFAULT_TCP_MTU)
}
pub fn connect_timeout_ms(&self) -> u64 {
self.connect_timeout_ms
.unwrap_or(DEFAULT_TCP_CONNECT_TIMEOUT_MS)
}
pub fn nodelay(&self) -> bool {
self.nodelay.unwrap_or(true)
}
pub fn keepalive_secs(&self) -> u64 {
self.keepalive_secs.unwrap_or(DEFAULT_TCP_KEEPALIVE_SECS)
}
pub fn recv_buf_size(&self) -> usize {
self.recv_buf_size.unwrap_or(DEFAULT_TCP_RECV_BUF)
}
pub fn send_buf_size(&self) -> usize {
self.send_buf_size.unwrap_or(DEFAULT_TCP_SEND_BUF)
}
pub fn max_inbound_connections(&self) -> usize {
self.max_inbound_connections
.unwrap_or(DEFAULT_TCP_MAX_INBOUND)
}
pub fn advertise_on_nostr(&self) -> bool {
self.advertise_on_nostr.unwrap_or(false)
}
pub fn external_advert_addr(&self) -> Option<SocketAddr> {
let raw = self.external_addr.as_deref()?;
let bind_port = parse_bind_port(self.bind_addr.as_deref()?)?;
parse_external_advert_addr(raw, bind_port)
}
}
const DEFAULT_TOR_SOCKS5_ADDR: &str = "127.0.0.1:9050";
const DEFAULT_TOR_CONTROL_ADDR: &str = "/run/tor/control";
const DEFAULT_TOR_COOKIE_PATH: &str = "/var/run/tor/control.authcookie";
const DEFAULT_TOR_CONNECT_TIMEOUT_MS: u64 = 120_000;
const DEFAULT_TOR_MTU: u16 = 1400;
const DEFAULT_TOR_MAX_INBOUND: usize = 64;
const DEFAULT_HOSTNAME_FILE: &str = "/var/lib/tor/fips_onion_service/hostname";
const DEFAULT_DIRECTORY_BIND_ADDR: &str = "127.0.0.1:8443";
const DEFAULT_TOR_ADVERTISED_PORT: u16 = 443;
#[derive(Debug, Clone, Default, Serialize, Deserialize)]
#[serde(deny_unknown_fields)]
pub struct TorConfig {
#[serde(default, skip_serializing_if = "Option::is_none")]
pub mode: Option<String>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub socks5_addr: Option<String>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub connect_timeout_ms: Option<u64>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub mtu: Option<u16>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub control_addr: Option<String>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub control_auth: Option<String>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub cookie_path: Option<String>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub max_inbound_connections: Option<usize>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub directory_service: Option<DirectoryServiceConfig>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub advertise_on_nostr: Option<bool>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub advertised_port: Option<u16>,
}
#[derive(Debug, Clone, Default, Serialize, Deserialize)]
#[serde(deny_unknown_fields)]
pub struct DirectoryServiceConfig {
#[serde(default, skip_serializing_if = "Option::is_none")]
pub hostname_file: Option<String>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub bind_addr: Option<String>,
}
impl DirectoryServiceConfig {
pub fn hostname_file(&self) -> &str {
self.hostname_file
.as_deref()
.unwrap_or(DEFAULT_HOSTNAME_FILE)
}
pub fn bind_addr(&self) -> &str {
self.bind_addr
.as_deref()
.unwrap_or(DEFAULT_DIRECTORY_BIND_ADDR)
}
}
impl TorConfig {
pub fn mode(&self) -> &str {
self.mode.as_deref().unwrap_or("socks5")
}
pub fn socks5_addr(&self) -> &str {
self.socks5_addr
.as_deref()
.unwrap_or(DEFAULT_TOR_SOCKS5_ADDR)
}
pub fn control_addr(&self) -> &str {
self.control_addr
.as_deref()
.unwrap_or(DEFAULT_TOR_CONTROL_ADDR)
}
pub fn control_auth(&self) -> &str {
self.control_auth.as_deref().unwrap_or("cookie")
}
pub fn cookie_path(&self) -> &str {
self.cookie_path
.as_deref()
.unwrap_or(DEFAULT_TOR_COOKIE_PATH)
}
pub fn connect_timeout_ms(&self) -> u64 {
self.connect_timeout_ms
.unwrap_or(DEFAULT_TOR_CONNECT_TIMEOUT_MS)
}
pub fn mtu(&self) -> u16 {
self.mtu.unwrap_or(DEFAULT_TOR_MTU)
}
pub fn max_inbound_connections(&self) -> usize {
self.max_inbound_connections
.unwrap_or(DEFAULT_TOR_MAX_INBOUND)
}
pub fn advertise_on_nostr(&self) -> bool {
self.advertise_on_nostr.unwrap_or(false)
}
pub fn advertised_port(&self) -> u16 {
self.advertised_port.unwrap_or(DEFAULT_TOR_ADVERTISED_PORT)
}
}
const DEFAULT_BLE_PSM: u16 = 0x0085;
const DEFAULT_BLE_MTU: u16 = 2048;
const DEFAULT_BLE_MAX_CONNECTIONS: usize = 7;
const DEFAULT_BLE_CONNECT_TIMEOUT_MS: u64 = 10_000;
const DEFAULT_BLE_PROBE_COOLDOWN_SECS: u64 = 30;
#[derive(Debug, Clone, Default, Serialize, Deserialize)]
#[serde(deny_unknown_fields)]
pub struct BleConfig {
#[serde(default, skip_serializing_if = "Option::is_none")]
pub adapter: Option<String>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub psm: Option<u16>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub mtu: Option<u16>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub max_connections: Option<usize>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub connect_timeout_ms: Option<u64>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub advertise: Option<bool>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub scan: Option<bool>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub auto_connect: Option<bool>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub accept_connections: Option<bool>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub probe_cooldown_secs: Option<u64>,
}
impl BleConfig {
pub fn adapter(&self) -> &str {
self.adapter.as_deref().unwrap_or("hci0")
}
pub fn psm(&self) -> u16 {
self.psm.unwrap_or(DEFAULT_BLE_PSM)
}
pub fn mtu(&self) -> u16 {
self.mtu.unwrap_or(DEFAULT_BLE_MTU)
}
pub fn max_connections(&self) -> usize {
self.max_connections.unwrap_or(DEFAULT_BLE_MAX_CONNECTIONS)
}
pub fn connect_timeout_ms(&self) -> u64 {
self.connect_timeout_ms
.unwrap_or(DEFAULT_BLE_CONNECT_TIMEOUT_MS)
}
pub fn advertise(&self) -> bool {
self.advertise.unwrap_or(true)
}
pub fn scan(&self) -> bool {
self.scan.unwrap_or(true)
}
pub fn auto_connect(&self) -> bool {
self.auto_connect.unwrap_or(false)
}
pub fn accept_connections(&self) -> bool {
self.accept_connections.unwrap_or(true)
}
pub fn probe_cooldown_secs(&self) -> u64 {
self.probe_cooldown_secs
.unwrap_or(DEFAULT_BLE_PROBE_COOLDOWN_SECS)
}
}
#[derive(Debug, Clone, Default, Serialize, Deserialize)]
pub struct TransportsConfig {
#[serde(default, skip_serializing_if = "is_transport_empty")]
pub udp: TransportInstances<UdpConfig>,
#[cfg(feature = "sim-transport")]
#[serde(default, skip_serializing_if = "is_transport_empty")]
pub sim: TransportInstances<SimTransportConfig>,
#[serde(default, skip_serializing_if = "is_transport_empty")]
pub ethernet: TransportInstances<EthernetConfig>,
#[serde(default, skip_serializing_if = "is_transport_empty")]
pub tcp: TransportInstances<TcpConfig>,
#[serde(default, skip_serializing_if = "is_transport_empty")]
pub tor: TransportInstances<TorConfig>,
#[serde(default, skip_serializing_if = "is_transport_empty")]
pub ble: TransportInstances<BleConfig>,
}
fn is_transport_empty<T>(instances: &TransportInstances<T>) -> bool {
instances.is_empty()
}
impl TransportsConfig {
pub fn is_empty(&self) -> bool {
self.udp.is_empty()
&& {
#[cfg(feature = "sim-transport")]
{
self.sim.is_empty()
}
#[cfg(not(feature = "sim-transport"))]
{
true
}
}
&& self.ethernet.is_empty()
&& self.tcp.is_empty()
&& self.tor.is_empty()
&& self.ble.is_empty()
}
pub fn merge(&mut self, other: TransportsConfig) {
if !other.udp.is_empty() {
self.udp = other.udp;
}
#[cfg(feature = "sim-transport")]
if !other.sim.is_empty() {
self.sim = other.sim;
}
if !other.ethernet.is_empty() {
self.ethernet = other.ethernet;
}
if !other.tcp.is_empty() {
self.tcp = other.tcp;
}
if !other.tor.is_empty() {
self.tor = other.tor;
}
if !other.ble.is_empty() {
self.ble = other.ble;
}
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn parse_external_addr_accepts_bare_ipv4_with_appended_bind_port() {
let sa = parse_external_advert_addr("198.51.100.1", 2121).unwrap();
assert_eq!(sa.to_string(), "198.51.100.1:2121");
}
#[test]
fn parse_external_addr_accepts_full_ipv4_socket_addr() {
let sa = parse_external_advert_addr("198.51.100.1:443", 2121).unwrap();
assert_eq!(sa.to_string(), "198.51.100.1:443");
}
#[test]
fn parse_external_addr_accepts_bare_ipv6_with_appended_bind_port() {
let sa = parse_external_advert_addr("2001:db8::1", 443).unwrap();
assert_eq!(sa.to_string(), "[2001:db8::1]:443");
}
#[test]
fn parse_external_addr_accepts_bracketed_ipv6_with_explicit_port() {
let sa = parse_external_advert_addr("[2001:db8::1]:8443", 443).unwrap();
assert_eq!(sa.to_string(), "[2001:db8::1]:8443");
}
#[test]
fn parse_external_addr_rejects_garbage() {
assert!(parse_external_advert_addr("not-an-ip", 443).is_none());
assert!(parse_external_advert_addr("", 443).is_none());
}
#[test]
fn udp_external_advert_addr_combines_with_bind_port_default() {
let cfg = UdpConfig {
external_addr: Some("198.51.100.1".to_string()),
..UdpConfig::default()
};
let sa = cfg.external_advert_addr().unwrap();
assert_eq!(sa.to_string(), "198.51.100.1:2121");
}
#[test]
fn udp_external_advert_addr_with_explicit_full_socket_addr_overrides_bind_port() {
let cfg = UdpConfig {
bind_addr: Some("0.0.0.0:2121".to_string()),
external_addr: Some("198.51.100.1:9999".to_string()),
..UdpConfig::default()
};
let sa = cfg.external_advert_addr().unwrap();
assert_eq!(sa.to_string(), "198.51.100.1:9999");
}
#[test]
fn udp_external_advert_addr_returns_none_when_unset() {
let cfg = UdpConfig::default();
assert!(cfg.external_advert_addr().is_none());
}
#[test]
fn tcp_external_advert_addr_requires_bind_port() {
let cfg = TcpConfig {
external_addr: Some("198.51.100.1".to_string()),
..TcpConfig::default()
};
assert!(cfg.external_advert_addr().is_none());
let cfg = TcpConfig {
bind_addr: Some("0.0.0.0:443".to_string()),
external_addr: Some("198.51.100.1".to_string()),
..TcpConfig::default()
};
let sa = cfg.external_advert_addr().unwrap();
assert_eq!(sa.to_string(), "198.51.100.1:443");
}
#[test]
fn tcp_external_advert_addr_with_full_socket_addr_independent_of_bind() {
let cfg = TcpConfig {
bind_addr: Some("0.0.0.0:443".to_string()),
external_addr: Some("198.51.100.1:8443".to_string()),
..TcpConfig::default()
};
let sa = cfg.external_advert_addr().unwrap();
assert_eq!(sa.to_string(), "198.51.100.1:8443");
}
#[test]
fn parse_bind_port_extracts_from_socket_addr_strings() {
assert_eq!(parse_bind_port("0.0.0.0:2121"), Some(2121));
assert_eq!(parse_bind_port("[::]:443"), Some(443));
assert_eq!(parse_bind_port("not-a-socket-addr"), None);
}
}