rmqtt-conf 0.3.5

//! Network listener configuration for TCP, TLS, WebSocket, WSS, and QUIC protocols.
//!
//! Defines the configuration structures for each listener type, including
//! connection limits, keepalive settings, QoS restrictions, TLS certificate
//! paths, and proxy protocol support.

use std::net::SocketAddr;
use std::num::{NonZeroU16, NonZeroU32};
use std::ops::Deref;
use std::str::FromStr;
use std::sync::Arc;
use std::time::Duration;

use serde::de::{self, Deserializer};
use serde::Deserialize;

use rmqtt_codec::types::QoS;

use super::{deserialize_addr, deserialize_duration, to_duration, Bytesize};

type HashMap<K, V> = std::collections::HashMap<K, V, ahash::RandomState>;

type Port = u16;

/// Collection of all configured network listeners, organized by protocol.
///
/// Contains separate maps for TCP, TLS (TCP+SSL), WebSocket, WebSocket Secure,
/// and QUIC listeners, each keyed by port number. Parsed from raw TOML config
/// during the server initialization phase.
#[derive(Debug, Clone, Deserialize, Default)]
pub struct Listeners {
    #[serde(rename = "tcp")]
    #[serde(default)]
    _tcps: HashMap<String, ListenerInner>,

    #[serde(rename = "tls")]
    #[serde(default)]
    _tlss: HashMap<String, ListenerInner>,

    #[serde(rename = "ws")]
    #[serde(default)]
    _wss: HashMap<String, ListenerInner>,

    #[serde(rename = "wss")]
    #[serde(default)]
    _wsss: HashMap<String, ListenerInner>,

    #[serde(rename = "quic")]
    #[serde(default)]
    _quics: HashMap<String, ListenerInner>,

    #[serde(default, skip)]
    pub tcps: HashMap<Port, Listener>,
    #[serde(default, skip)]
    pub tlss: HashMap<Port, Listener>,
    #[serde(default, skip)]
    pub wss: HashMap<Port, Listener>,
    #[serde(default, skip)]
    pub wsss: HashMap<Port, Listener>,
    #[serde(default, skip)]
    pub quics: HashMap<Port, Listener>,
}

impl Listeners {
    #[inline]
    pub(crate) fn init(&mut self) {
        for (name, mut inner) in self._tcps.drain() {
            if inner.enable {
                inner.name = format!("{name}/tcp");
                self.tcps.insert(inner.addr.port(), Listener::new(inner));
            }
        }

        for (name, mut inner) in self._tlss.drain() {
            if inner.enable {
                inner.name = format!("{name}/tls");
                self.tlss.insert(inner.addr.port(), Listener::new(inner));
            }
        }

        for (name, mut inner) in self._wss.drain() {
            if inner.enable {
                inner.name = format!("{name}/ws");
                self.wss.insert(inner.addr.port(), Listener::new(inner));
            }
        }

        for (name, mut inner) in self._wsss.drain() {
            if inner.enable {
                inner.name = format!("{name}/wss");
                self.wsss.insert(inner.addr.port(), Listener::new(inner));
            }
        }

        for (name, mut inner) in self._quics.drain() {
            if inner.enable {
                inner.name = format!("{name}/quic");
                self.quics.insert(inner.addr.port(), Listener::new(inner));
            }
        }
    }

    /// Returns a reference to the TCP listener for the given port, if configured.
    #[inline]
    pub fn tcp(&self, port: u16) -> Option<Listener> {
        self.tcps.get(&port).cloned()
    }

    /// Returns the TLS (TCP+SSL) listener for the given port, if configured.
    #[inline]
    pub fn tls(&self, port: u16) -> Option<Listener> {
        self.tlss.get(&port).cloned()
    }

    /// Returns the WebSocket listener for the given port, if configured.
    #[inline]
    pub fn ws(&self, port: u16) -> Option<Listener> {
        self.wss.get(&port).cloned()
    }

    /// Returns the secure WebSocket (WSS) listener for the given port, if configured.
    #[inline]
    pub fn wss(&self, port: u16) -> Option<Listener> {
        self.wsss.get(&port).cloned()
    }

    /// Returns the QUIC listener for the given port, if configured.
    #[inline]
    pub fn quic(&self, port: u16) -> Option<Listener> {
        self.quics.get(&port).cloned()
    }

    /// Looks up a listener by port across all protocol types.
    ///
    /// Searches TCP, TLS, WebSocket, WSS, and QUIC in order and returns the
    /// first match found.
    #[inline]
    pub fn get(&self, port: u16) -> Option<Listener> {
        if let Some(l) = self.tcp(port) {
            return Some(l);
        }
        if let Some(l) = self.tls(port) {
            return Some(l);
        }
        if let Some(l) = self.ws(port) {
            return Some(l);
        }
        if let Some(l) = self.wss(port) {
            return Some(l);
        }
        if let Some(l) = self.quic(port) {
            return Some(l);
        }
        None
    }

    #[inline]
    pub(crate) fn set_default(&mut self) {
        let inner = Listener::default();
        self.tcps.insert(inner.addr.port(), inner);
    }
}

/// A single listener configuration, wrapping an `Arc<ListenerInner>`.
///
/// Provides thread-safe shared access to the underlying listener settings
/// via dereferencing. Created during the initialization phase.
#[derive(Debug, Clone, Default)]
pub struct Listener {
    inner: Arc<ListenerInner>,
}

impl Listener {
    #[inline]
    fn new(inner: ListenerInner) -> Self {
        Self { inner: Arc::new(inner) }
    }
}

impl Deref for Listener {
    type Target = ListenerInner;
    fn deref(&self) -> &Self::Target {
        &self.inner
    }
}

/// Detailed configuration for a single network listener.
///
/// Contains all tunable parameters including connection limits, keepalive
/// intervals, QoS levels, TLS certificate paths, rate limiting, and
/// MQTT session expiry settings.
#[derive(Debug, Clone, Deserialize)]
pub struct ListenerInner {
    #[serde(default)]
    pub name: String,
    #[serde(default = "ListenerInner::enable_default")]
    pub enable: bool,
    #[serde(deserialize_with = "deserialize_addr")]
    pub addr: SocketAddr,
    // #[serde(default = "ListenerInner::workers_default")]
    // pub workers: usize,
    #[serde(default = "ListenerInner::max_connections_default")]
    pub max_connections: usize,
    #[serde(default = "ListenerInner::max_handshaking_limit_default")]
    pub max_handshaking_limit: usize,
    #[serde(default = "ListenerInner::max_packet_size_default")]
    pub max_packet_size: Bytesize,
    #[serde(default = "ListenerInner::backlog_default")]
    pub backlog: i32,
    #[serde(default = "ListenerInner::nodelay_default")]
    pub nodelay: bool,
    #[serde(default = "ListenerInner::reuseaddr_default")]
    pub reuseaddr: Option<bool>,
    #[serde(default = "ListenerInner::reuseport_default")]
    pub reuseport: Option<bool>,
    #[serde(default = "ListenerInner::allow_anonymous_default")]
    pub allow_anonymous: bool,
    #[serde(default = "ListenerInner::min_keepalive_default")]
    pub min_keepalive: u16,
    #[serde(default = "ListenerInner::max_keepalive_default")]
    pub max_keepalive: u16,
    #[serde(default = "ListenerInner::allow_zero_keepalive_default")]
    pub allow_zero_keepalive: bool,
    #[serde(default = "ListenerInner::keepalive_backoff_default")]
    pub keepalive_backoff: f32,
    #[serde(default = "ListenerInner::max_inflight_default")]
    pub max_inflight: NonZeroU16,
    #[serde(default = "ListenerInner::handshake_timeout_default", deserialize_with = "deserialize_duration")]
    pub handshake_timeout: Duration,
    #[serde(default = "ListenerInner::max_mqueue_len_default")]
    pub max_mqueue_len: usize,
    #[serde(
        default = "ListenerInner::mqueue_rate_limit_default",
        deserialize_with = "ListenerInner::deserialize_mqueue_rate_limit"
    )]
    pub mqueue_rate_limit: (NonZeroU32, Duration),

    #[serde(default = "ListenerInner::max_clientid_len_default")]
    pub max_clientid_len: usize,

    #[serde(
        default = "ListenerInner::max_qos_allowed_default",
        deserialize_with = "ListenerInner::deserialize_max_qos_allowed"
    )]
    pub max_qos_allowed: QoS,

    #[serde(default = "ListenerInner::max_topic_levels_default")]
    pub max_topic_levels: usize,

    #[serde(
        default = "ListenerInner::session_expiry_interval_default",
        deserialize_with = "deserialize_duration"
    )]
    pub session_expiry_interval: Duration,

    #[serde(default, deserialize_with = "deserialize_duration")]
    pub max_session_expiry_interval: Duration,

    #[serde(
        default = "ListenerInner::message_retry_interval_default",
        deserialize_with = "deserialize_duration"
    )]
    pub message_retry_interval: Duration,

    #[serde(
        default = "ListenerInner::message_expiry_interval_default",
        deserialize_with = "ListenerInner::deserialize_message_expiry_interval"
    )]
    pub message_expiry_interval: Duration,

    #[serde(default = "ListenerInner::max_subscriptions_default")]
    pub max_subscriptions: usize,

    #[serde(default)]
    pub max_topic_aliases: u16,

    #[serde(default = "ListenerInner::cross_certificate_default")]
    pub cross_certificate: bool,
    pub cert: Option<String>,
    pub key: Option<String>,
    pub client_ca_certs: Option<String>,

    #[serde(default)]
    pub limit_subscription: bool,
    #[serde(default)]
    pub delayed_publish: bool,
    #[serde(default)]
    pub proxy_protocol: bool,
    #[serde(
        default = "ListenerInner::proxy_protocol_timeout_default",
        deserialize_with = "deserialize_duration"
    )]
    pub proxy_protocol_timeout: Duration,

    #[serde(default)]
    pub cert_cn_as_username: bool,
    #[serde(default)]
    pub cert_subject_dn_as_username: bool,
    #[serde(default)]
    pub collect_cert_info: bool,

    //QUIC max_idle_timeout
    #[serde(default = "ListenerInner::idle_timeout_default", deserialize_with = "deserialize_duration")]
    pub idle_timeout: Duration,
}

impl Default for ListenerInner {
    fn default() -> Self {
        Self {
            name: "external/tcp".into(),
            enable: ListenerInner::enable_default(),
            addr: ListenerInner::addr_default(),
            // workers: ListenerInner::workers_default(),
            max_connections: ListenerInner::max_connections_default(),
            max_handshaking_limit: ListenerInner::max_handshaking_limit_default(),
            max_packet_size: ListenerInner::max_packet_size_default(),
            reuseaddr: ListenerInner::reuseaddr_default(),
            reuseport: ListenerInner::reuseport_default(),
            backlog: ListenerInner::backlog_default(),
            nodelay: ListenerInner::nodelay_default(),

            allow_anonymous: ListenerInner::allow_anonymous_default(),
            min_keepalive: ListenerInner::min_keepalive_default(),
            max_keepalive: ListenerInner::max_keepalive_default(),
            allow_zero_keepalive: ListenerInner::allow_zero_keepalive_default(),
            keepalive_backoff: ListenerInner::keepalive_backoff_default(),
            max_inflight: ListenerInner::max_inflight_default(),
            handshake_timeout: ListenerInner::handshake_timeout_default(),
            max_mqueue_len: ListenerInner::max_mqueue_len_default(),
            mqueue_rate_limit: ListenerInner::mqueue_rate_limit_default(),
            max_clientid_len: ListenerInner::max_clientid_len_default(),
            max_qos_allowed: ListenerInner::max_qos_allowed_default(),
            max_topic_levels: ListenerInner::max_topic_levels_default(),
            session_expiry_interval: ListenerInner::session_expiry_interval_default(),
            max_session_expiry_interval: Duration::ZERO,
            message_retry_interval: ListenerInner::message_retry_interval_default(),
            message_expiry_interval: ListenerInner::message_expiry_interval_default(),
            max_subscriptions: ListenerInner::max_subscriptions_default(),
            max_topic_aliases: 0,
            cross_certificate: ListenerInner::cross_certificate_default(),
            cert: None,
            key: None,
            client_ca_certs: None,
            limit_subscription: false,
            delayed_publish: false,
            proxy_protocol: false,
            proxy_protocol_timeout: ListenerInner::proxy_protocol_timeout_default(),

            cert_cn_as_username: false,
            cert_subject_dn_as_username: false,
            collect_cert_info: false,
            idle_timeout: ListenerInner::idle_timeout_default(),
        }
    }
}

impl ListenerInner {
    fn enable_default() -> bool {
        true
    }
    #[inline]
    fn addr_default() -> SocketAddr {
        ([0, 0, 0, 0], 1883).into()
    }
    // #[inline]
    // fn workers_default() -> usize {
    //     8
    // }
    #[inline]
    fn max_connections_default() -> usize {
        1024000
    }
    #[inline]
    fn max_handshaking_limit_default() -> usize {
        500
    }
    #[inline]
    fn max_packet_size_default() -> Bytesize {
        Bytesize(1024 * 1024)
    }
    #[inline]
    fn reuseaddr_default() -> Option<bool> {
        Some(true)
    }
    #[inline]
    fn reuseport_default() -> Option<bool> {
        None
    }
    #[inline]
    fn backlog_default() -> i32 {
        1024
    }
    #[inline]
    fn nodelay_default() -> bool {
        false
    }

    #[inline]
    fn allow_anonymous_default() -> bool {
        false
    }
    #[inline]
    fn min_keepalive_default() -> u16 {
        0
    }
    #[inline]
    fn max_keepalive_default() -> u16 {
        u16::MAX
    }
    #[inline]
    fn allow_zero_keepalive_default() -> bool {
        true
    }
    #[inline]
    fn keepalive_backoff_default() -> f32 {
        0.75
    }
    #[inline]
    fn max_inflight_default() -> NonZeroU16 {
        if let Some(max_inflight) = NonZeroU16::new(16) {
            max_inflight
        } else {
            unreachable!()
        }
    }
    #[inline]
    fn handshake_timeout_default() -> Duration {
        Duration::from_secs(15)
    }
    #[inline]
    fn max_mqueue_len_default() -> usize {
        1000
    }
    #[inline]
    fn mqueue_rate_limit_default() -> (NonZeroU32, Duration) {
        (NonZeroU32::MAX, Duration::from_secs(1))
    }
    #[inline]
    fn max_clientid_len_default() -> usize {
        65535
    }
    #[inline]
    fn max_qos_allowed_default() -> QoS {
        QoS::ExactlyOnce
    }
    #[inline]
    fn max_topic_levels_default() -> usize {
        0
    }
    #[inline]
    fn session_expiry_interval_default() -> Duration {
        Duration::from_secs(7200)
    }
    #[inline]
    fn message_retry_interval_default() -> Duration {
        Duration::from_secs(30)
    }
    #[inline]
    fn message_expiry_interval_default() -> Duration {
        Duration::from_secs(300)
    }
    #[inline]
    fn max_subscriptions_default() -> usize {
        0
    }

    /// Returns the handshake timeout in milliseconds as a `u16`, capped at `u16::MAX`.
    #[inline]
    pub fn handshake_timeout(&self) -> u16 {
        let millis = self.handshake_timeout.as_millis();
        if millis > 0xffff {
            0xffff
        } else {
            millis as u16
        }
    }

    #[inline]
    fn deserialize_mqueue_rate_limit<'de, D>(deserializer: D) -> Result<(NonZeroU32, Duration), D::Error>
    where
        D: Deserializer<'de>,
    {
        let v = String::deserialize(deserializer)?;
        let pair: Vec<&str> = v.split(',').collect();
        if pair.len() == 2 {
            let burst = NonZeroU32::from_str(pair[0])
                .map_err(|e| de::Error::custom(format!("mqueue_rate_limit, burst format error, {e}")))?;
            let replenish_n_per = to_duration(pair[1]);
            if replenish_n_per.as_millis() == 0 {
                return Err(de::Error::custom(format!(
                    "mqueue_rate_limit, value format error, {}",
                    pair.join(",")
                )));
            }
            Ok((burst, replenish_n_per))
        } else {
            Err(de::Error::custom(format!("mqueue_rate_limit, value format error, {}", pair.join(","))))
        }
    }
    #[inline]
    fn deserialize_max_qos_allowed<'de, D>(deserializer: D) -> Result<QoS, D::Error>
    where
        D: Deserializer<'de>,
    {
        let qos = match u8::deserialize(deserializer)? {
            0 => QoS::AtMostOnce,
            1 => QoS::AtLeastOnce,
            2 => QoS::ExactlyOnce,
            _ => return Err(de::Error::custom("QoS configuration error, only values (0,1,2) are supported")),
        };
        Ok(qos)
    }

    #[inline]
    fn deserialize_message_expiry_interval<'de, D>(deserializer: D) -> Result<Duration, D::Error>
    where
        D: Deserializer<'de>,
    {
        let v = String::deserialize(deserializer)?;
        let mut d = to_duration(&v);
        if d.is_zero() {
            d = Duration::from_secs(u32::MAX as u64);
        }
        Ok(d)
    }

    #[inline]
    fn cross_certificate_default() -> bool {
        false
    }
    #[inline]
    fn proxy_protocol_timeout_default() -> Duration {
        Duration::from_secs(5)
    }

    #[inline]
    fn idle_timeout_default() -> Duration {
        Duration::from_secs(90)
    }
}