quiche-h3 0.0.2

//! Server wiring: [`H3QuicheAcceptor`] over `tokio_quiche::listen()` (design
//! §7.1). One acceptor wraps exactly one `QuicConnectionStream` (one socket).
//!
//! `accept()` must never serialize handshakes: awaiting establishment inline
//! would let one slow/stalled TLS handshake block acceptance of every other
//! pending connection (a head-of-line-blocking / DoS hazard). Instead the
//! acceptor drives a bounded number of handshakes concurrently through a
//! [`FuturesUnordered`] and returns whichever completes first. Per-connection
//! setup failures and per-packet stream item errors are surfaced **log-only**
//! and skipped; `accept()`'s `Err` is reserved for listener/socket-fatal
//! conditions (§7.1, §8.4).

use std::num::NonZeroUsize;
use std::sync::Arc;

use bytes::Bytes;
use futures::future::BoxFuture;
use futures::stream::FuturesUnordered;
use futures::{FutureExt, StreamExt};
use tokio::net::UdpSocket;
use tokio_quiche::metrics::DefaultMetrics;
use tokio_quiche::settings::{CertificateKind, Hooks, QuicSettings, TlsCertificatePaths};
use tokio_quiche::{ConnectionParams, QuicConnectionStream};

use crate::buffer::PKT_BUF_LEN;
use crate::driver::{DriverBufferConfig, QuicheDriver, BYTE_CHANNEL_DEPTH};
use crate::endpoint::{EndpointShared, H3QuicheEndpoint};
use crate::stream::Connection;
use crate::Error;

/// Default per-listener cap on concurrently-progressing handshakes (§7.1). The
/// incoming branch of the accept loop is not polled once this many handshakes
/// are in flight, bounding the bridge-owned worker/future set under a flood.
pub const DEFAULT_MAX_IN_FLIGHT_HANDSHAKES: usize = 256;

/// Provisional accept-queue depth for peer-initiated bidi/uni streams (§5.2).
const DEFAULT_ACCEPT_BIDI_CAP: usize = 128;
const DEFAULT_ACCEPT_UNI_CAP: usize = 128;

/// Owned, cloneable server configuration. Owns everything `tokio_quiche::listen`
/// borrows (TLS cert/key **paths**, `QuicSettings`, `Hooks`) so the acceptor is
/// self-contained, plus the accept-queue depths and per-listener handshake cap
/// (§7.1).
///
/// **Construction (CO-C):** construct via [`Default`] + functional-update syntax
/// (e.g. `H3QuicheServerConfig { cert_path, ..Default::default() }`). New fields
/// (like the SF-4/SF-5 buffer knobs below) are added additively with defaults, so
/// **`..Default::default()` / `Default::default()` construction keeps compiling**
/// across additions (a caller that instead uses an *exhaustive* field literal must
/// name any new field). We deliberately do **not** mark this `#[non_exhaustive]`:
/// that would forbid struct-literal/FRU construction downstream entirely (forcing
/// a mutate-after-default/builder style), a heavier break than additive fields —
/// see Docs.md/§12.
#[derive(Clone)]
pub struct H3QuicheServerConfig {
    /// Path to the PEM X.509 certificate chain.
    pub cert_path: String,
    /// Path to the PEM private key.
    pub key_path: String,
    /// QUIC transport settings (ALPN defaults to `[b"h3"]`).
    pub settings: QuicSettings,
    /// Connection lifecycle hooks.
    pub hooks: Hooks,
    /// Bound on the peer-initiated bidi accept queue.
    pub accept_bidi_cap: usize,
    /// Bound on the peer-initiated uni accept queue.
    pub accept_uni_cap: usize,
    /// Cap on concurrently-progressing handshakes per listener (§7.1).
    pub max_in_flight_handshakes: NonZeroUsize,
    /// Per-recv byte-channel depth per accepted connection (SF-4). Defaults to
    /// `BYTE_CHANNEL_DEPTH` (64); the per-stream in-flight memory bound is
    /// `recv_channel_depth × MAX_CHUNK`. **Trade-off**: lowering it saves memory
    /// at the cost of per-stream throughput/buffering.
    pub recv_channel_depth: usize,
    /// Outbound packet-buffer size in bytes per accepted connection (SF-5).
    /// Defaults to `PKT_BUF_LEN` (64 KiB). **Do NOT shrink below a full GSO
    /// batch without a datapath assessment** (§5, §12): it can regress egress
    /// batching/throughput.
    pub packet_buffer_size: usize,
    /// Optional aggregate cap (bytes) on buffered outbound send data admitted to
    /// each accepted connection's worker (SF-6). `None` (default) leaves the send
    /// path unbounded, preserving historical behavior. A finite cap bounds
    /// resident admitted send bytes to at most `cap + one admission unit`, so a
    /// slow/stalled peer cannot grow send-side memory without limit. Front-end
    /// writes past the cap park (async backpressure) rather than being dropped or
    /// reordered (§12 S3).
    pub max_buffered_send_bytes: Option<usize>,
}

impl Default for H3QuicheServerConfig {
    fn default() -> Self {
        Self {
            cert_path: String::new(),
            key_path: String::new(),
            settings: QuicSettings::default(),
            hooks: Hooks::default(),
            accept_bidi_cap: DEFAULT_ACCEPT_BIDI_CAP,
            accept_uni_cap: DEFAULT_ACCEPT_UNI_CAP,
            max_in_flight_handshakes: NonZeroUsize::new(DEFAULT_MAX_IN_FLIGHT_HANDSHAKES)
                .expect("DEFAULT_MAX_IN_FLIGHT_HANDSHAKES is non-zero"),
            recv_channel_depth: BYTE_CHANNEL_DEPTH,
            packet_buffer_size: PKT_BUF_LEN,
            max_buffered_send_bytes: None,
        }
    }
}

/// A standalone HTTP/3-over-quiche acceptor: one per bound socket. Its
/// [`accept`](H3QuicheAcceptor::accept) yields the crate's front-end
/// [`Connection<Bytes>`], which implements [`h3::quic::Connection<Bytes>`]. The
/// `h3_util::H3Acceptor` conformance is provided by the h3-util `quiche_h3`
/// wrapper (Phase 9), keeping this crate free of a circular h3-util dependency
/// (design §10).
pub struct H3QuicheAcceptor {
    stream: QuicConnectionStream<DefaultMetrics>,
    handshakes: FuturesUnordered<BoxFuture<'static, Result<Connection<Bytes>, Error>>>,
    max_in_flight_handshakes: NonZeroUsize,
    accept_bidi_cap: usize,
    accept_uni_cap: usize,
    /// Per-connection buffer sizing (SF-4/SF-5) applied to every accepted
    /// connection's driver; sourced from [`H3QuicheServerConfig`].
    buffers: DriverBufferConfig,
    incoming_done: bool,
    /// Endpoint registry shared by every acceptor from the same `bind()` call
    /// and by the [`H3QuicheEndpoint`] handles it hands out (§5.1). Registration
    /// at accept time and `close()` linearize under this one lock (the admission
    /// fence), so a single endpoint governs shutdown across all its sockets.
    shared: Arc<EndpointShared>,
}

impl H3QuicheAcceptor {
    /// Bind the given sockets and return one acceptor per socket
    /// (`tokio_quiche::listen` yields one `QuicConnectionStream` per socket).
    /// Validation failures — an empty socket set or unreadable/missing TLS
    /// cert/key files — surface as `Err(Error)` **here**, before any accept loop
    /// runs. Per-connection handshake failures never surface here; they are
    /// log-only (§8.4).
    pub fn bind(
        sockets: impl IntoIterator<Item = UdpSocket>,
        config: &H3QuicheServerConfig,
    ) -> Result<Vec<Self>, Error> {
        let sockets: Vec<UdpSocket> = sockets.into_iter().collect();
        if sockets.is_empty() {
            return Err("quiche-h3: H3QuicheAcceptor::bind requires at least one socket".into());
        }

        // Validate the TLS material is a readable file and the ALPN is set
        // before handing paths to quiche (which otherwise surfaces the failure
        // only asynchronously per-conn).
        crate::ensure_readable_file(&config.cert_path, "TLS certificate")?;
        crate::ensure_readable_file(&config.key_path, "TLS private key")?;
        crate::ensure_nonempty_alpn(&config.settings, "server config")?;

        // `ConnectionParams<'_>` borrows the owned cert/key paths; `listen` only
        // borrows `params` for the duration of the call, so this temporary is
        // sufficient.
        let params = ConnectionParams::new_server(
            config.settings.clone(),
            TlsCertificatePaths {
                cert: &config.cert_path,
                private_key: &config.key_path,
                kind: CertificateKind::X509,
            },
            config.hooks.clone(),
        );

        let streams = tokio_quiche::listen(sockets, params, DefaultMetrics)
            .map_err(|e| -> Error { Box::new(e) })?;

        // One endpoint registry governs every acceptor produced by this
        // `bind()` call (FR-002/§5.1): all per-socket acceptors and the
        // `H3QuicheEndpoint` handles they hand out share this single `Arc`, so a
        // shutdown fences admission and drains workers across all sockets.
        let shared = EndpointShared::new();

        Ok(streams
            .into_iter()
            .map(|stream| Self {
                stream,
                handshakes: FuturesUnordered::new(),
                max_in_flight_handshakes: config.max_in_flight_handshakes,
                accept_bidi_cap: config.accept_bidi_cap,
                accept_uni_cap: config.accept_uni_cap,
                buffers: DriverBufferConfig {
                    recv_channel_depth: config.recv_channel_depth,
                    packet_buffer_size: config.packet_buffer_size,
                    max_buffered_send_bytes: config.max_buffered_send_bytes,
                },
                incoming_done: false,
                shared: Arc::clone(&shared),
            })
            .collect())
    }

    /// Return a cloneable [`H3QuicheEndpoint`] handle over this acceptor's shared
    /// endpoint registry (§5.2). The handle drives graceful shutdown
    /// ([`close`](H3QuicheEndpoint::close)) and drain observation
    /// ([`wait_idle`](H3QuicheEndpoint::wait_idle)); it shares state with every
    /// acceptor from the same `bind()` call and outlives the acceptor(s), so it
    /// still reaches live workers after the acceptor is dropped.
    ///
    /// # Same-port rebind after shutdown
    ///
    /// [`wait_idle`](H3QuicheEndpoint::wait_idle) resolves when every bridge
    /// worker has ended, but the underlying UDP socket is owned by
    /// tokio-quiche's router task, which releases its socket handles only when it
    /// is next polled after the acceptor is dropped. Rebinding the **same** UDP
    /// port immediately after `wait_idle()` therefore commonly fails its first
    /// attempt and succeeds on the next. Spike S1 (`tests/endpoint_shutdown.rs`)
    /// measured this residual on Linux loopback: when the rebind is attempted at
    /// the tightest window (immediately after `wait_idle()` resolves) the first
    /// attempt essentially always needed one retry, and a **single** backoff
    /// retry always sufficed (worst observed: **2 attempts**; retry latency ≈ one
    /// backoff interval). Giving the router task even a few hundred microseconds
    /// of unrelated work first drops the first-attempt-failure rate to ~5–25%,
    /// but the bound is the same. Treat this as an observed-on-Linux sample, not a
    /// cross-platform guarantee: the robust contract is simply *"a same-port
    /// rebind may need a short bounded retry,"* so callers that must rebind the
    /// exact port should use one, e.g.:
    ///
    /// ```no_run
    /// # use std::net::SocketAddr;
    /// # use tokio::net::UdpSocket;
    /// # async fn rebind(addr: SocketAddr) -> std::io::Result<UdpSocket> {
    /// let mut last_err = None;
    /// for _ in 0..20 {
    ///     match UdpSocket::bind(addr).await {
    ///         Ok(sock) => return Ok(sock),
    ///         Err(e) => {
    ///             last_err = Some(e);
    ///             tokio::time::sleep(std::time::Duration::from_millis(5)).await;
    ///         }
    ///     }
    /// }
    /// Err(last_err.unwrap())
    /// # }
    /// ```
    ///
    /// Note also that **all** acceptors from one `bind()` call (and any live
    /// connection workers) must be dropped/drained before the FD is released.
    pub fn endpoint(&self) -> H3QuicheEndpoint {
        H3QuicheEndpoint::new(Arc::clone(&self.shared))
    }

    /// Accept the next established connection (§7.1). Returns `Ok(Some(conn))`
    /// for a ready connection, `Ok(None)` once the listener has ended and all
    /// in-flight handshakes have drained, or `Err` only for a listener/socket
    /// fatal. Per-connection setup failures and per-packet stream item errors
    /// are logged and skipped — one peer can never fail `accept` for everyone.
    pub async fn accept(&mut self) -> Result<Option<Connection<Bytes>>, Error> {
        loop {
            // Register the accept-wake waiter BEFORE observing `closing`, so a
            // `close()` racing this check cannot be lost (§5.5 lost-wakeup
            // discipline). Enabling the `Notified` future arms it without
            // awaiting.
            let accept_wake = self.shared.accept_wake.notified();
            tokio::pin!(accept_wake);
            accept_wake.as_mut().enable();

            // Once `close()` is observed the acceptor stops admitting: it neither
            // starts new workers nor yields freshly-established connections
            // (§5.4). It still drains in-flight handshakes so their workers are
            // registered and force-closed, then reports end-of-stream.
            let closing = self.shared.is_closing();
            let admission_done = self.incoming_done || closing;

            if admission_done && self.handshakes.is_empty() {
                return Ok(None);
            }

            tokio::select! {
                biased;

                // 0. `close()` woke a parked `accept()`: fall through to re-read
                //    `closing` at the top of the loop (never returns early). Only
                //    armed until `closing` latches — afterwards there is nothing
                //    more to wake for.
                _ = &mut accept_wake, if !closing => {
                    continue;
                }

                // 1. A started handshake finished. Decoupled from how many other
                //    handshakes are still in flight, so one stalled peer cannot
                //    block delivery of a ready connection.
                Some(res) = self.handshakes.next(), if !self.handshakes.is_empty() => {
                    match res {
                        Ok(conn) => {
                            // A handshake that completed once `close()` has been
                            // observed must be dropped, not yielded (§5.5): its
                            // worker was registered and is force-closed by the
                            // `close()` snapshot. Dropping the `Connection`
                            // releases its strong `cmd_tx`; the peer may then
                            // observe a teardown close rather than the exact
                            // `(code, reason)` (acceptable, Spec AS-1.1/P3.2).
                            //
                            // Re-read `closing` under the endpoint lock HERE,
                            // rather than trusting the stale snapshot taken at the
                            // top of the loop: `close()` can linearize on another
                            // thread between that snapshot and this `select!`
                            // poll, and `biased` polling means the ready-handshake
                            // arm can win the same poll in which the `accept_wake`
                            // arm saw `Pending`. The fresh read makes this the
                            // true accept/yield linearization point (§5.4, FR-006).
                            if closing || self.shared.is_closing() {
                                drop(conn);
                                continue;
                            }
                            return Ok(Some(conn));
                        }
                        Err(_e) => {
                            // Per-connection pre-handshake failure. `accept` has
                            // no per-connection error channel, so surface it
                            // log-only and keep serving others (§8.4).
                            #[cfg(feature = "tracing")]
                            tracing::debug!(
                                error = %_e,
                                "quiche-h3: connection setup failed before handshake"
                            );
                            continue;
                        }
                    }
                }

                // 2. A new incoming connection: START its handshake but DO NOT
                //    await it here. Disabled once admission is done (listener
                //    ended or endpoint closing) or while the in-flight set is
                //    full, so backpressure is exerted by not polling the stream
                //    again until a slot frees.
                iqc = self.stream.next(),
                    if !admission_done
                        && self.handshakes.len() < self.max_in_flight_handshakes.get() =>
                {
                    let Some(item) = iqc else {
                        self.incoming_done = true;
                        continue;
                    };
                    match item {
                        Ok(iqc) => {
                            let (mut driver, handles) = QuicheDriver::<Bytes>::with_buffers(
                                true,
                                self.accept_bidi_cap,
                                self.accept_uni_cap,
                                self.buffers,
                            );
                            // Admission fence (§5.1): register under the endpoint
                            // lock at the true linearization point. If `close()`
                            // landed between the top-of-loop check and here,
                            // `try_register` returns `None` and we drop the
                            // nascent connection WITHOUT starting a worker (§5.4).
                            match crate::endpoint::try_register(&self.shared, &handles.cmd_tx) {
                                None => {
                                    // Dropping `iqc`/`driver`/`handles` abandons
                                    // the connection; no worker is spawned.
                                    continue;
                                }
                                Some(reg) => {
                                    // Move the deregistration guard into the
                                    // driver so it drops at worker exit (§5.5),
                                    // then start (synchronous; the returned
                                    // `QuicConnection` is metadata only, §2.3 T2).
                                    driver.set_conn_registration(reg);
                                    let _qconn = iqc.start(driver);
                                    self.handshakes.push(
                                        handles
                                            .into_established_connection()
                                            .map(|res| res.map_err(|e| -> Error { Box::new(e) }))
                                            .boxed(),
                                    );
                                }
                            }
                        }
                        Err(_e) => {
                            // A stream item `Err` is a per-packet/per-attempt
                            // failure on the pinned 0.19.1 surface, not listener
                            // termination (§7.1 T4). Log and continue.
                            #[cfg(feature = "tracing")]
                            tracing::debug!(
                                error = %_e,
                                "quiche-h3: rejected initial connection packet; listener continues"
                            );
                            continue;
                        }
                    }
                }
            }
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn bind_rejects_empty_socket_set() {
        let config = H3QuicheServerConfig::default();
        // `H3QuicheAcceptor` is not `Debug`, so inspect the `Result` by hand
        // rather than via `expect_err`.
        match H3QuicheAcceptor::bind(Vec::<UdpSocket>::new(), &config) {
            Ok(_) => panic!("empty socket set must be rejected"),
            Err(e) => assert!(e.to_string().contains("at least one socket")),
        }
    }

    #[tokio::test]
    async fn bind_rejects_missing_cert() {
        // A real ephemeral socket is required to reach the TLS-material check
        // (the empty-set guard runs first); no handshake is performed.
        let sock = UdpSocket::bind("127.0.0.1:0").await.expect("bind udp");
        let config = H3QuicheServerConfig {
            cert_path: "/nonexistent/quiche-h3/missing.crt".to_string(),
            key_path: "/nonexistent/quiche-h3/missing.key".to_string(),
            ..H3QuicheServerConfig::default()
        };
        let err = match H3QuicheAcceptor::bind([sock], &config) {
            Ok(_) => panic!("missing cert path must be rejected"),
            Err(e) => e,
        };
        assert!(err.to_string().contains("certificate"));
    }

    #[test]
    fn default_handshake_cap_is_256() {
        assert_eq!(DEFAULT_MAX_IN_FLIGHT_HANDSHAKES, 256);
        assert_eq!(
            H3QuicheServerConfig::default()
                .max_in_flight_handshakes
                .get(),
            256
        );
    }

    /// SF-4/SF-5 (SC-006): the server config defaults to the historical buffer
    /// sizes so out-of-the-box behavior is unchanged; overrides are honored.
    #[test]
    fn server_config_buffer_defaults_and_overrides() {
        let def = H3QuicheServerConfig::default();
        assert_eq!(def.recv_channel_depth, BYTE_CHANNEL_DEPTH);
        assert_eq!(def.packet_buffer_size, PKT_BUF_LEN);

        let custom = H3QuicheServerConfig {
            recv_channel_depth: 16,
            packet_buffer_size: 8192,
            ..H3QuicheServerConfig::default()
        };
        assert_eq!(custom.recv_channel_depth, 16);
        assert_eq!(custom.packet_buffer_size, 8192);
    }

    /// SF-6 (SC-007): the aggregate send-byte cap defaults to `None` (unbounded,
    /// behavior unchanged) and a configured cap is preserved on the config.
    #[test]
    fn server_config_send_cap_defaults_none_and_overrides() {
        let def = H3QuicheServerConfig::default();
        assert_eq!(def.max_buffered_send_bytes, None);

        let custom = H3QuicheServerConfig {
            max_buffered_send_bytes: Some(1 << 20),
            ..H3QuicheServerConfig::default()
        };
        assert_eq!(custom.max_buffered_send_bytes, Some(1 << 20));
    }
}