flowscope 0.25.2

//! Pluggable L7 message parsers.
//!
//! Two trait families:
//!
//! - [`SessionParser`] — for **stream-based** protocols (HTTP/1, TLS,
//!   DNS-over-TCP). One parser per session; receives bytes via
//!   `feed_initiator` / `feed_responder`; returns a `Vec` of typed
//!   messages every call. Pair with `netring::SessionStream` to get
//!   an async stream of L7 events.
//!
//! - [`DatagramParser`] — for **packet-based** protocols (DNS-over-UDP,
//!   syslog, NTP, SNMP). Receives one L4 payload at a time. Pair with
//!   `netring::DatagramStream`.
//!
//! Both trait shapes return owned `Vec<Message>` rather than borrowed
//! iterators or `SmallVec` to keep the public API stable across
//! versions of `smallvec` etc. The per-call allocation is amortized
//! across many bytes worth of work.
//!
//! # SessionParser vs `Reassembler`
//!
//! [`crate::Reassembler`] is the lower-level hook: one instance per
//! `(flow, side)`, receives raw TCP segments and yields the ordered
//! byte stream as [`crate::StreamChunks`] (data interleaved with gap
//! marks). `SessionParser` is the higher-level
//! abstraction: one instance per flow, two `feed_*` methods,
//! returns typed messages directly. Pick whichever fits your
//! integration:
//!
//! | Concern                       | `Reassembler`           | `SessionParser`             |
//! |-------------------------------|-------------------------|------------------------------|
//! | Granularity                   | per (flow, side)        | per flow                     |
//! | Output                        | `StreamChunks` (bytes + gaps) | typed messages         |
//! | Cross-direction state         | painful                 | natural                      |
//! | UDP support                   | no                      | use [`DatagramParser`]       |
//!
//! # Running parsers
//!
//! - [`SessionDriver`] / [`DatagramDriver`] — one parser type over a
//!   whole capture, yielding an ordered stream of [`SessionEvent`]s
//!   (lifecycle, messages, parser closes, anomalies). What
//!   `netring`'s `SessionStream` / `DatagramStream` wrap.
//! - [`crate::driver::Driver`] — several parsers (by port, by
//!   signature, or on every flow) sharing one flow table, with typed
//!   per-parser drain handles.
//!
//! Both are built on the same engine, so they behave identically:
//! one flow table, one reassembler per flow side (shared by every
//! parser), and these per-parser rules:
//!
//! - A parser that reports [`SessionParser::is_poisoned`] or
//!   [`SessionParser::is_done`] after a call is **closed** (a parser
//!   close event with [`EndReason::ParseError`] /
//!   [`EndReason::ParserDone`], plus an
//!   [`AnomalyKind::SessionParseError`] for poison when anomalies are
//!   on). It is never fed again for that flow — and never re-created:
//!   the flow itself stays tracked until its transport end. A parser
//!   close never ends the flow; `Ended` / `Closed` always carries a
//!   transport reason (`Fin`, `Rst`, `IdleTimeout`, `Evicted`,
//!   `ForceClosed`).
//! - The one reporting difference: a parser still open when its flow
//!   ends. `SessionDriver` / `DatagramDriver` say so with `Closed`
//!   alone; the typed `Driver` also emits one `Event::ParserClosed`
//!   per slot carrying the transport reason
//!   ([`EndReason::is_transport`]) right before `Ended`.
//! - Missing bytes are reported through [`SessionParser::on_gap`].
//!   The default answer, [`GapResponse::StopSide`], stops feeding
//!   **that side** only (a parser side-stopped event with
//!   [`EndReason::StreamGap`]); the other side keeps being parsed.
//!   [`GapResponse::Continue`] resynchronises, [`GapResponse::Stop`]
//!   closes the whole parser.
//! - A reassembly stop on one side (per-side cap under
//!   [`crate::OverflowPolicy::DropFlow`], memcap) stops that side
//!   with [`EndReason::BufferOverflow`].
//! - When both sides are stopped the parser closes once (detail
//!   `"both sides stopped"`).
//! - At flow end each side still being read gets
//!   [`SessionParser::fin_initiator`] / [`SessionParser::fin_responder`]
//!   (graceful end, or that side sent a FIN) or `rst_*`; a stopped
//!   side gets neither.
//! - Datagram parsers only see the transports they declare
//!   ([`DatagramParser::transports`], default [`Transports::UDP`]):
//!   a UDP parser never gets ICMP messages and vice versa.
//!
//! # Example
//!
//! ```
//! use flowscope::{FlowSide, SessionParser, Timestamp};
//!
//! #[derive(Default, Clone)]
//! struct LineParser {
//!     init_buf: Vec<u8>,
//!     resp_buf: Vec<u8>,
//! }
//!
//! impl SessionParser for LineParser {
//!     type Message = (FlowSide, String);
//!
//!     fn feed_initiator(&mut self, bytes: &[u8], _ts: Timestamp, out: &mut Vec<Self::Message>) {
//!         feed(&mut self.init_buf, bytes, FlowSide::Initiator, out);
//!     }
//!     fn feed_responder(&mut self, bytes: &[u8], _ts: Timestamp, out: &mut Vec<Self::Message>) {
//!         feed(&mut self.resp_buf, bytes, FlowSide::Responder, out);
//!     }
//! }
//!
//! fn feed(buf: &mut Vec<u8>, bytes: &[u8], side: FlowSide, out: &mut Vec<(FlowSide, String)>) {
//!     buf.extend_from_slice(bytes);
//!     while let Some(nl) = buf.iter().position(|&b| b == b'\n') {
//!         let line = String::from_utf8_lossy(&buf[..nl]).into_owned();
//!         out.push((side, line));
//!         buf.drain(..=nl);
//!     }
//! }
//! ```

use crate::{
    event::{AnomalyKind, EndReason, FlowSide, FlowStats},
    extractor::{L4Proto, Orientation},
    parser_kind::ParserKind,
    timestamp::Timestamp,
};

#[cfg(all(feature = "extractors", feature = "reassembler"))]
pub(crate) mod core;
#[cfg(all(feature = "extractors", feature = "reassembler"))]
mod driver;
#[cfg(all(feature = "extractors", feature = "reassembler"))]
pub(crate) mod engine;

#[cfg(all(feature = "extractors", feature = "reassembler"))]
pub use driver::{DatagramDriver, SessionDriver};

/// Default per-side buffer cap for [`BufferedFrameDrain`] /
/// [`AccumulatingSessionParser`]. 64 KiB matches the TCP
/// receive-buffer scale.
pub const DEFAULT_FRAME_DRAIN_MAX_BUFFER: usize = 64 * 1024;

/// Reasons [`BufferedFrameDrain`] / [`AccumulatingSessionParser`]
/// poison themselves.
#[derive(Debug, Clone, PartialEq, Eq)]
#[non_exhaustive]
pub enum FrameDrainError {
    /// Buffer reached its `max_buffer` ceiling before the parser
    /// drained a complete message. Indicates protocol desync —
    /// the parser can't make progress.
    BufferFull,
    /// The `parse_one` closure returned `Some((msg, 0))` — a
    /// zero-byte advance. Reserved for closure-author bugs; the
    /// drain stops to avoid an infinite loop and the parser is
    /// poisoned.
    ZeroByteAdvance,
}

impl std::fmt::Display for FrameDrainError {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            FrameDrainError::BufferFull => f.write_str("parser buffer exceeded max_buffer cap"),
            FrameDrainError::ZeroByteAdvance => {
                f.write_str("parse_one returned Some((_, 0)) — zero-byte advance")
            }
        }
    }
}

impl std::error::Error for FrameDrainError {}

/// Buffered drain helper for custom parsers.
///
/// Encapsulates the "accumulate bytes, repeatedly call a parser
/// closure, drain consumed prefix, retain partial" pattern that
/// every custom [`SessionParser`] writes. Catches the off-by-one
/// bugs around drain offsets.
///
/// Typical use inside a [`SessionParser`] impl:
///
/// ```rust
/// # use flowscope::session::BufferedFrameDrain;
/// # use flowscope::{FlowSide, SessionParser, Timestamp};
/// # #[derive(Debug, Clone)] struct Msg;
/// # fn parse_one(b: &[u8]) -> Option<(Msg, usize)> { None }
/// #[derive(Default, Clone)]
/// struct MyParser {
///     init: BufferedFrameDrain<Msg>,
///     resp: BufferedFrameDrain<Msg>,
/// }
///
/// impl SessionParser for MyParser {
///     type Message = Msg;
///     fn feed_initiator(&mut self, b: &[u8], _: Timestamp, out: &mut Vec<Msg>) {
///         let _ = self.init.extend(b);
///         self.init.drain_with(parse_one);
///         out.append(&mut self.init.take_messages());
///     }
///     fn feed_responder(&mut self, b: &[u8], _: Timestamp, out: &mut Vec<Msg>) {
///         let _ = self.resp.extend(b);
///         self.resp.drain_with(parse_one);
///         out.append(&mut self.resp.take_messages());
///     }
/// }
/// ```
///
/// New in 0.10.0 (plan 106).
#[derive(Debug)]
pub struct BufferedFrameDrain<M> {
    buf: Vec<u8>,
    out: Vec<M>,
    max_buffer: usize,
    poisoned: Option<FrameDrainError>,
}

impl<M> Default for BufferedFrameDrain<M> {
    fn default() -> Self {
        Self::new()
    }
}

impl<M> Clone for BufferedFrameDrain<M> {
    /// Clones the configuration (`max_buffer`) but resets the
    /// buffer / messages / poison state — typical use is to
    /// rebuild the helper per session.
    fn clone(&self) -> Self {
        Self {
            buf: Vec::new(),
            out: Vec::new(),
            max_buffer: self.max_buffer,
            poisoned: None,
        }
    }
}

impl<M> BufferedFrameDrain<M> {
    /// Construct with the [default buffer
    /// cap](DEFAULT_FRAME_DRAIN_MAX_BUFFER).
    pub fn new() -> Self {
        Self::with_max_buffer(DEFAULT_FRAME_DRAIN_MAX_BUFFER)
    }

    /// Construct with a custom per-side buffer cap.
    pub fn with_max_buffer(max_buffer: usize) -> Self {
        Self {
            buf: Vec::new(),
            out: Vec::new(),
            max_buffer,
            poisoned: None,
        }
    }

    /// Append `bytes` to the buffer.
    ///
    /// Returns `Err(FrameDrainError::BufferFull)` if the new
    /// length would exceed `max_buffer`. The helper sets its
    /// internal poison flag (queryable via [`Self::is_poisoned`]).
    /// `bytes` are still appended up to the cap before poisoning,
    /// so consumers that surface poison to the driver get the
    /// last bit of data they can use.
    pub fn extend(&mut self, bytes: &[u8]) -> Result<(), FrameDrainError> {
        if self.poisoned.is_some() {
            return Err(self.poisoned.clone().unwrap());
        }
        let new_len = self.buf.len().saturating_add(bytes.len());
        if new_len > self.max_buffer {
            let room = self.max_buffer.saturating_sub(self.buf.len());
            self.buf.extend_from_slice(&bytes[..room.min(bytes.len())]);
            self.poisoned = Some(FrameDrainError::BufferFull);
            return Err(FrameDrainError::BufferFull);
        }
        self.buf.extend_from_slice(bytes);
        Ok(())
    }

    /// Repeatedly call `parse_one` and drain the consumed prefix.
    /// Pushes each parsed message into the internal `out` queue
    /// (drain via [`Self::take_messages`]).
    ///
    /// `parse_one(buf) -> Option<(M, usize)>` semantics:
    /// - `Some((msg, n))` — a complete message; advance `n` bytes.
    /// - `None` — need more bytes.
    /// - `Some((_, 0))` is treated as poison (zero-byte advance
    ///   would loop forever).
    pub fn drain_with<F>(&mut self, mut parse_one: F)
    where
        F: FnMut(&[u8]) -> Option<(M, usize)>,
    {
        while self.poisoned.is_none() {
            match parse_one(&self.buf) {
                Some((msg, 0)) => {
                    self.out.push(msg);
                    self.poisoned = Some(FrameDrainError::ZeroByteAdvance);
                    return;
                }
                Some((msg, n)) => {
                    self.out.push(msg);
                    if n >= self.buf.len() {
                        self.buf.clear();
                    } else {
                        self.buf.drain(..n);
                    }
                }
                None => return,
            }
        }
    }

    /// Take the accumulated messages, clearing the queue.
    pub fn take_messages(&mut self) -> Vec<M> {
        std::mem::take(&mut self.out)
    }

    /// Bytes currently held in the buffer.
    pub fn buffered_len(&self) -> usize {
        self.buf.len()
    }

    /// `true` if the helper has poisoned itself (`BufferFull` or
    /// `ZeroByteAdvance`). Once set, [`Self::extend`] is a no-op.
    pub fn is_poisoned(&self) -> bool {
        self.poisoned.is_some()
    }

    /// Poison reason, if any.
    pub fn poison_reason(&self) -> Option<&FrameDrainError> {
        self.poisoned.as_ref()
    }
}

/// Convenience [`SessionParser`] impl over a `parse_one` closure
/// of shape `Fn(&[u8]) -> Option<(M, usize)>`.
///
/// Wraps two [`BufferedFrameDrain`]s (one per side) and exposes
/// the canonical "init + resp + drain-loop" pattern through one
/// constructor call. Reduces ~25 LoC of boilerplate per custom
/// parser to:
///
/// ```rust
/// # use flowscope::session::AccumulatingSessionParser;
/// # use flowscope::ParserKind;
/// # #[derive(Debug, Clone)] struct Msg;
/// # fn parse_one(b: &[u8]) -> Option<(Msg, usize)> { None }
/// let parser = AccumulatingSessionParser::new(ParserKind::Other("my-protocol"), parse_one);
/// ```
///
/// The closure must be `Clone + Send + 'static` so the parser
/// can be cloned per session (via the [`SessionParserFactory`]
/// blanket impl). Capturing closures usually satisfy `Clone` if
/// the captured values do.
///
/// New in 0.10.0 (plan 106).
pub struct AccumulatingSessionParser<F, M>
where
    F: Fn(&[u8]) -> Option<(M, usize)> + Clone + Send + 'static,
    M: Send + std::fmt::Debug + 'static,
{
    parser_kind: ParserKind,
    parse_one: F,
    max_buffer: usize,
    init: BufferedFrameDrain<M>,
    resp: BufferedFrameDrain<M>,
}

impl<F, M> std::fmt::Debug for AccumulatingSessionParser<F, M>
where
    F: Fn(&[u8]) -> Option<(M, usize)> + Clone + Send + 'static,
    M: Send + std::fmt::Debug + 'static,
{
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("AccumulatingSessionParser")
            .field("parser_kind", &self.parser_kind)
            .field("init_buffered", &self.init.buffered_len())
            .field("resp_buffered", &self.resp.buffered_len())
            .field(
                "poisoned",
                &(self.init.is_poisoned() || self.resp.is_poisoned()),
            )
            .finish()
    }
}

impl<F, M> Clone for AccumulatingSessionParser<F, M>
where
    F: Fn(&[u8]) -> Option<(M, usize)> + Clone + Send + 'static,
    M: Send + std::fmt::Debug + 'static,
{
    /// Clones the configuration + parse closure; resets per-session
    /// buffer state. Use for per-session reuse via the
    /// [`SessionParserFactory`] blanket impl.
    fn clone(&self) -> Self {
        Self {
            parser_kind: self.parser_kind,
            parse_one: self.parse_one.clone(),
            max_buffer: self.max_buffer,
            init: BufferedFrameDrain::with_max_buffer(self.max_buffer),
            resp: BufferedFrameDrain::with_max_buffer(self.max_buffer),
        }
    }
}

impl<F, M> AccumulatingSessionParser<F, M>
where
    F: Fn(&[u8]) -> Option<(M, usize)> + Clone + Send + 'static,
    M: Send + std::fmt::Debug + 'static,
{
    /// Construct with the [default per-side buffer
    /// cap](DEFAULT_FRAME_DRAIN_MAX_BUFFER).
    pub fn new(parser_kind: ParserKind, parse_one: F) -> Self {
        Self::with_max_buffer(parser_kind, parse_one, DEFAULT_FRAME_DRAIN_MAX_BUFFER)
    }

    /// Construct with a custom per-side buffer cap.
    pub fn with_max_buffer(parser_kind: ParserKind, parse_one: F, max_buffer: usize) -> Self {
        Self {
            parser_kind,
            parse_one,
            max_buffer,
            init: BufferedFrameDrain::with_max_buffer(max_buffer),
            resp: BufferedFrameDrain::with_max_buffer(max_buffer),
        }
    }
}

impl<F, M> SessionParser for AccumulatingSessionParser<F, M>
where
    F: Fn(&[u8]) -> Option<(M, usize)> + Clone + Send + 'static,
    M: Send + std::fmt::Debug + 'static,
{
    type Message = M;

    fn feed_initiator(&mut self, bytes: &[u8], _ts: Timestamp, out: &mut Vec<M>) {
        if self.init.extend(bytes).is_err() {
            out.append(&mut self.init.take_messages());
            return;
        }
        let parse_one = self.parse_one.clone();
        self.init.drain_with(parse_one);
        out.append(&mut self.init.take_messages());
    }

    fn feed_responder(&mut self, bytes: &[u8], _ts: Timestamp, out: &mut Vec<M>) {
        if self.resp.extend(bytes).is_err() {
            out.append(&mut self.resp.take_messages());
            return;
        }
        let parse_one = self.parse_one.clone();
        self.resp.drain_with(parse_one);
        out.append(&mut self.resp.take_messages());
    }

    fn parser_kind(&self) -> ParserKind {
        self.parser_kind
    }

    fn is_poisoned(&self) -> bool {
        self.init.is_poisoned() || self.resp.is_poisoned()
    }

    fn poison_reason(&self) -> Option<&str> {
        // First poisoned side wins.
        let err = self.init.poison_reason().or(self.resp.poison_reason())?;
        Some(match err {
            FrameDrainError::BufferFull => "buffer cap exceeded",
            FrameDrainError::ZeroByteAdvance => "parse_one returned zero-byte advance",
        })
    }
}

/// Convenience [`DatagramParser`] impl over a `parse_one` closure
/// of shape `Fn(&[u8]) -> Option<M>`.
///
/// One UDP packet, one optional message. The closure receives the
/// raw payload; return `Some(message)` to emit one event,
/// `None` to drop the packet silently.
///
/// New in 0.10.0 (plan 106).
pub struct PerDatagramParser<F, M>
where
    F: Fn(&[u8]) -> Option<M> + Clone + Send + 'static,
    M: Send + std::fmt::Debug + 'static,
{
    parser_kind: ParserKind,
    parse_one: F,
}

impl<F, M> std::fmt::Debug for PerDatagramParser<F, M>
where
    F: Fn(&[u8]) -> Option<M> + Clone + Send + 'static,
    M: Send + std::fmt::Debug + 'static,
{
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("PerDatagramParser")
            .field("parser_kind", &self.parser_kind)
            .finish()
    }
}

impl<F, M> Clone for PerDatagramParser<F, M>
where
    F: Fn(&[u8]) -> Option<M> + Clone + Send + 'static,
    M: Send + std::fmt::Debug + 'static,
{
    fn clone(&self) -> Self {
        Self {
            parser_kind: self.parser_kind,
            parse_one: self.parse_one.clone(),
        }
    }
}

impl<F, M> PerDatagramParser<F, M>
where
    F: Fn(&[u8]) -> Option<M> + Clone + Send + 'static,
    M: Send + std::fmt::Debug + 'static,
{
    pub fn new(parser_kind: ParserKind, parse_one: F) -> Self {
        Self {
            parser_kind,
            parse_one,
        }
    }
}

impl<F, M> DatagramParser for PerDatagramParser<F, M>
where
    F: Fn(&[u8]) -> Option<M> + Clone + Send + 'static,
    M: Send + std::fmt::Debug + 'static,
{
    type Message = M;

    fn parse(&mut self, payload: &[u8], _side: FlowSide, _ts: Timestamp, out: &mut Vec<M>) {
        if let Some(msg) = (self.parse_one)(payload) {
            out.push(msg);
        }
    }

    fn parser_kind(&self) -> ParserKind {
        self.parser_kind
    }
}

/// Parses a stream-oriented L7 protocol session. One instance per
/// flow; both directions feed through the same parser, allowing
/// state to interleave.
///
/// Implementors are owned by the per-flow slot; sync (no `await`).
/// Backpressure flows from the consuming `Stream` back to the
/// kernel ring once the per-flow message buffer fills up — see
/// the `netring::SessionStream` adapter.
///
/// # Per-flow rich state
///
/// For consumers that maintain per-flow user state updated by BOTH
/// the reassembler and the parser — TCP rich stats, application-
/// level counters, middleware state machines — keep that state on
/// [`crate::FlowEntry::user`] (typed via the `S` parameter on
/// the internal session engine) and update it from your event
/// loop after `track()`. The pattern is documented in
/// `docs/recipes.md` → "Per-flow user state via the consumer
/// loop". Avoid piping `&mut S` through `feed_*` — it would
/// ripple a generic parameter through every shipped parser.
pub trait SessionParser: Send + 'static {
    /// L7 message produced by this parser.
    ///
    /// - `Send + 'static` so messages can cross task boundaries.
    /// - `Debug` is required for the per-message `tracing::trace!`
    ///   event under the `tracing` feature (filter via
    ///   `EnvFilter::new("flowscope.message=warn")` to suppress).
    ///   Almost every Rust type derives `Debug` anyway, and the
    ///   bound is trivial to add for those that don't.
    type Message: Send + std::fmt::Debug + 'static;

    /// Feed the next chunk of bytes from the **initiator** side.
    /// `ts` is the observed time of the packet carrying these bytes.
    /// Push any complete messages parsed during this call into
    /// `out`.
    ///
    /// **0.11 break (plan 119):** signature changed from
    /// `-> Vec<Self::Message>` to taking `out: &mut Vec<…>`.
    /// Same idiom as `httparse::Request::parse` etc.
    fn feed_initiator(&mut self, bytes: &[u8], ts: Timestamp, out: &mut Vec<Self::Message>);

    /// Feed the next chunk of bytes from the **responder** side.
    fn feed_responder(&mut self, bytes: &[u8], ts: Timestamp, out: &mut Vec<Self::Message>);

    /// The initiator side ended cleanly: called once at flow end when
    /// the flow ended gracefully ([`EndReason::is_graceful`]: FIN,
    /// idle timeout, force close) or the initiator sent a FIN — after
    /// its final bytes were fed. Not called for a side the parser
    /// stopped reading ([`GapResponse::StopSide`], reassembly stop).
    /// Default: no-op.
    fn fin_initiator(&mut self, _out: &mut Vec<Self::Message>) {}

    /// The responder side ended cleanly; see
    /// [`fin_initiator`](Self::fin_initiator).
    fn fin_responder(&mut self, _out: &mut Vec<Self::Message>) {}

    /// The initiator side ended abruptly: called once at flow end
    /// instead of [`fin_initiator`](Self::fin_initiator) when the flow
    /// was not ended gracefully (RST, eviction) and this side sent no
    /// FIN. Not called for a stopped side. Default: no-op.
    fn rst_initiator(&mut self) {}

    /// The responder side ended abruptly; see
    /// [`rst_initiator`](Self::rst_initiator).
    fn rst_responder(&mut self) {}

    /// Periodic time hook. The driver calls this on every `sweep` /
    /// `finish` with the sweep's `now`, for every still-live parser.
    /// Lets stateful parsers emit time-driven messages (timeouts,
    /// unanswered requests). Emitted messages are attributed to
    /// [`FlowSide::Initiator`]. Default: no-op.
    fn on_tick(&mut self, _now: Timestamp, _out: &mut Vec<Self::Message>) {}

    /// Bytes are missing from `side`'s stream: `missing` bytes the
    /// reassembler never saw (capture loss, asymmetric routing, an
    /// out-of-order hole that expired, or bytes dropped by the
    /// sliding-window buffer cap). The next `feed_*` call for that
    /// side resumes **after** the hole.
    ///
    /// Return [`GapResponse::Continue`] if the parser can cope (it
    /// resynchronises on message boundaries, or only counts bytes).
    /// The default, [`GapResponse::StopSide`], stops feeding **this
    /// side** (a framed parser fed a spliced stream would otherwise
    /// mis-parse silently, as in Suricata, where app-layer parsers
    /// that don't declare gap support stop on a gap) while the other
    /// side keeps being parsed and still gets its `fin_*` / `rst_*`.
    /// [`GapResponse::Stop`] closes the whole parser
    /// ([`EndReason::StreamGap`]) — for protocols whose state spans
    /// both directions (HTTP/2's HPACK).
    ///
    /// A stopped side gets no further `feed_*`, `on_gap` or
    /// `fin_*` / `rst_*` call: flush anything worth keeping here.
    /// Messages pushed into `out` are emitted before the stop.
    fn on_gap(
        &mut self,
        _side: FlowSide,
        _missing: u64,
        _ts: Timestamp,
        _out: &mut Vec<Self::Message>,
    ) -> GapResponse {
        GapResponse::StopSide
    }

    /// True after the parser has hit an unrecoverable error and
    /// can no longer make progress. The engines check this after
    /// every `feed_*` / `on_gap` / `on_tick` call and close the parser
    /// on `true` — it is not fed again for that flow, and the flow
    /// itself stays tracked. Default: `false` (parser never poisons).
    ///
    /// Parsers that want to drop a malformed message and keep
    /// going should NOT use this — just don't push the message
    /// into the returned `Vec`. Reserve poison for cases where
    /// internal state is corrupted past recovery (desynced framing,
    /// invalid magic bytes that won't appear later, etc.).
    ///
    /// The reassembly-side counterpart is
    /// [`crate::Reassembler::stop_reason`].
    fn is_poisoned(&self) -> bool {
        false
    }

    /// Optional human-readable description of why the parser
    /// poisoned. Consulted only when [`is_poisoned`](Self::is_poisoned)
    /// returns `true`. Default: `None`.
    ///
    /// The driver truncates to ~256 bytes when forwarding the
    /// resulting `FlowAnomaly` ([`AnomalyKind::SessionParseError`]).
    fn poison_reason(&self) -> Option<&str> {
        None
    }

    /// Symmetric "I'm done — close this flow cleanly" signal.
    /// Default: `false` (parser never self-terminates).
    ///
    /// Returning `true` tells the engine this parser has no more
    /// useful work to extract. The engine closes the parser with
    /// [`crate::EndReason::ParserDone`] right after emitting the
    /// messages of the same `feed_*` / `on_tick` call and stops
    /// feeding it; the flow itself stays tracked until its transport
    /// end.
    ///
    /// Reserve for protocols where **nothing** further on the
    /// connection can matter: a session-end sentinel after which the
    /// peers only close (e.g. an SMTP `QUIT` / `221` exchange), or a
    /// protocol switch to something this parser cannot read. Not for
    /// "one exchange complete" — DNS-over-TCP and HTTP/1.1 reuse the
    /// connection, and a parser done after the first pair would lose
    /// every later one. Do **not** use this to give up on bad input —
    /// that's [`is_poisoned`](Self::is_poisoned), which routes through
    /// [`crate::EndReason::ParseError`].
    ///
    /// Should be idempotent: once `is_done()` returns `true`, it
    /// should keep returning `true` for the lifetime of the parser.
    /// [`is_poisoned`](Self::is_poisoned) takes precedence — a
    /// parser that's both `is_done` and `is_poisoned` surfaces as
    /// `ParseError`, not `ParserDone`.
    fn is_done(&self) -> bool {
        false
    }

    /// Typed identity of this parser, surfaced on
    /// [`crate::driver::Event::ParserClosed::parser_kind`] and
    /// [`crate::driver::SlotHandle::parser_kind`]. Lifted from
    /// `&'static str` to the [`ParserKind`] enum in 0.20 (#109);
    /// originally added in 0.5.0.
    ///
    /// Built-in parsers return their dedicated variant
    /// ([`ParserKind::Http1`], [`ParserKind::DnsUdp`], …).
    /// Downstream parsers either add a [`ParserKind::Other`] with a
    /// stable, label-safe slug (`"crate-name/protocol"`) or leave
    /// the default.
    ///
    /// - Default: [`ParserKind::Unspecified`] (no kind set).
    /// - `as_str()` yields the metric-label slug (allocation-free).
    fn parser_kind(&self) -> ParserKind {
        ParserKind::Unspecified
    }
}

/// Builds a fresh [`SessionParser`] per session. Modeled on
/// [`crate::ReassemblerFactory`].
///
/// Most parsers can skip implementing this manually: any parser
/// that's `SessionParser + Default + Clone` automatically becomes
/// a factory via the blanket impl below.
pub trait SessionParserFactory<K>: Send + 'static {
    type Parser: SessionParser;
    fn new_parser(&mut self, key: &K) -> Self::Parser;
}

impl<K, P> SessionParserFactory<K> for P
where
    P: SessionParser + Default + Clone,
{
    type Parser = P;
    fn new_parser(&mut self, _key: &K) -> P {
        self.clone()
    }
}

/// Parses a packet-oriented L7 protocol. One instance per flow;
/// receives one L4 payload at a time along with which side sent it.
pub trait DatagramParser: Send + 'static {
    /// L7 message produced by this parser. Same `Debug` bound as
    /// [`SessionParser::Message`].
    type Message: Send + std::fmt::Debug + 'static;

    /// Parse one L4 payload. `side` is the direction relative to
    /// the flow's initiator; `ts` is the observed time of the
    /// datagram. Push any complete messages decoded into `out`.
    ///
    /// **0.11 break (plan 119):** signature changed from
    /// `-> Vec<Self::Message>` to taking `out: &mut Vec<…>`.
    fn parse(
        &mut self,
        payload: &[u8],
        side: FlowSide,
        ts: Timestamp,
        out: &mut Vec<Self::Message>,
    );

    /// Periodic time hook — see [`SessionParser::on_tick`]. The
    /// driver calls this on every `sweep` / `finish`. Default: no-op.
    fn on_tick(&mut self, _now: Timestamp, _out: &mut Vec<Self::Message>) {}

    /// True after the parser has hit an unrecoverable error. See
    /// [`SessionParser::is_poisoned`] for the contract.
    fn is_poisoned(&self) -> bool {
        false
    }

    /// Optional reason for poison. See
    /// [`SessionParser::poison_reason`].
    fn poison_reason(&self) -> Option<&str> {
        None
    }

    /// Symmetric "I'm done — close this flow cleanly" signal,
    /// mirroring [`SessionParser::is_done`]. Default: `false`.
    fn is_done(&self) -> bool {
        false
    }

    /// See [`SessionParser::parser_kind`]. Default
    /// [`ParserKind::Unspecified`].
    fn parser_kind(&self) -> ParserKind {
        ParserKind::Unspecified
    }

    /// Which transports this parser reads. The engines hand it only
    /// datagrams of these transports: a UDP parser never sees an
    /// ICMP message and an ICMP parser never sees a UDP payload.
    /// Default [`Transports::UDP`]. New in 0.25.0.
    fn transports(&self) -> Transports {
        Transports::UDP
    }
}

bitflags::bitflags! {
    /// Transports a [`DatagramParser`] reads (see
    /// [`DatagramParser::transports`]). New in 0.25.0.
    #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
    pub struct Transports: u8 {
        /// UDP payloads.
        const UDP = 1;
        /// Whole ICMPv4 messages.
        const ICMP = 1 << 1;
        /// Whole ICMPv6 messages.
        const ICMPV6 = 1 << 2;
        /// SCTP packets (the whole SCTP packet: common header and chunks).
        const SCTP = 1 << 3;
        /// Any other non-TCP IP protocol (the whole L4 payload).
        const OTHER = 1 << 4;
    }
}

impl Transports {
    /// ICMPv4 and ICMPv6.
    pub const ICMP_ANY: Transports = Transports::ICMP.union(Transports::ICMPV6);

    /// Whether a datagram of this L4 protocol is admitted. Unknown
    /// L4 (`None`) counts as [`Transports::OTHER`].
    pub fn admits(self, l4: Option<crate::L4Proto>) -> bool {
        use crate::L4Proto;
        let bit = match l4 {
            Some(L4Proto::Udp) => Transports::UDP,
            Some(L4Proto::Icmp) => Transports::ICMP,
            Some(L4Proto::IcmpV6) => Transports::ICMPV6,
            Some(L4Proto::Sctp) => Transports::SCTP,
            Some(L4Proto::Tcp) => return false,
            _ => Transports::OTHER,
        };
        self.contains(bit)
    }
}

/// Builds a fresh [`DatagramParser`] per session.
pub trait DatagramParserFactory<K>: Send + 'static {
    type Parser: DatagramParser;
    fn new_parser(&mut self, key: &K) -> Self::Parser;

    /// Transports the parsers read (see
    /// [`DatagramParser::transports`]). Default [`Transports::UDP`];
    /// factories that clone a template delegate to it.
    fn transports(&self) -> Transports {
        Transports::UDP
    }
}

impl<K, P> DatagramParserFactory<K> for P
where
    P: DatagramParser + Default + Clone,
{
    type Parser = P;
    fn new_parser(&mut self, _key: &K) -> P {
        self.clone()
    }
    fn transports(&self) -> Transports {
        DatagramParser::transports(self)
    }
}

/// Answer to [`SessionParser::on_gap`].
#[derive(Debug, Clone, Copy, PartialEq, Eq, Default)]
#[non_exhaustive]
pub enum GapResponse {
    /// Keep feeding this parser; bytes after the gap follow.
    Continue,
    /// **Default.** Stop feeding the side with the gap; the other
    /// side keeps being parsed. Reported as a parser side stop
    /// ([`SessionEvent::ParserSideStopped`] /
    /// [`crate::driver::Event::ParserSideStopped`]). When both sides
    /// are stopped the parser is closed. New in 0.25.0.
    #[default]
    StopSide,
    /// Close the parser for this flow ([`EndReason::StreamGap`]).
    Stop,
}

/// Output of [`SessionDriver`] / [`DatagramDriver`]: flow lifecycle,
/// parser messages, parser closes and anomalies, in the order they
/// happened. `K` is the flow key, `M` the parser's message type.
///
/// For one flow the order is: `Started`, then any number of
/// `Application` / `FlowAnomaly` / `Tick` / `ParserSideStopped`,
/// possibly one `ParserClosed` (the parser gave up or finished early
/// — the flow goes on), then `Closed`. Messages a parser flushes at
/// flow end (`fin_*`) come before `Closed`.
#[derive(Debug, Clone)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[cfg_attr(feature = "serde", serde(tag = "type", rename_all = "snake_case"))]
#[cfg_attr(
    feature = "serde",
    serde(bound(
        serialize = "K: serde::Serialize, M: serde::Serialize",
        deserialize = "K: serde::de::DeserializeOwned, M: serde::de::DeserializeOwned"
    ))
)]
#[non_exhaustive]
pub enum SessionEvent<K, M> {
    /// First packet of a new flow.
    Started {
        key: K,
        /// Logical role of the side that sent the first packet
        /// (arrival-order relative; see `orientation` for the
        /// race-stable axis).
        side: FlowSide,
        /// Canonical, address-sorted direction of the first packet.
        orientation: Orientation,
        ts: Timestamp,
        l4: Option<L4Proto>,
    },
    /// The parser emitted a message.
    Application {
        key: K,
        /// Side whose bytes produced the message (`on_tick` output is
        /// attributed to the initiator).
        side: FlowSide,
        /// Canonical direction matching `side`.
        orientation: Orientation,
        message: M,
        /// Timestamp of the packet (or sweep) that produced it.
        ts: Timestamp,
        /// [`SessionParser::parser_kind`] of the producing parser.
        parser_kind: ParserKind,
    },
    /// The parser was closed before its flow ended — poisoned
    /// ([`EndReason::ParseError`]), done ([`EndReason::ParserDone`]),
    /// a gap answered with [`GapResponse::Stop`]
    /// ([`EndReason::StreamGap`]), or both sides stopped (detail
    /// `"both sides stopped"`, reason of the last side stop). The flow
    /// stays tracked and ends later with `Closed`; the parser is not
    /// fed again for it. (A parser still open when its flow ends gets
    /// no `ParserClosed`: `Closed` says so — so `reason.is_parser()`
    /// always holds here. The typed [`crate::driver::Driver`] differs:
    /// it emits an `Event::ParserClosed` with the transport reason per
    /// slot right before `Ended`; [`EndReason::is_transport`] /
    /// [`EndReason::is_parser`] tell the two apart.)
    ParserClosed {
        key: K,
        parser_kind: ParserKind,
        reason: EndReason,
        /// Human-readable detail: the parser's `poison_reason()`
        /// (truncated to 256 bytes), the gap size, or the stop reason.
        detail: Option<String>,
        ts: Timestamp,
    },
    /// The parser stopped reading one side of the flow — a gap it
    /// cannot bridge ([`EndReason::StreamGap`],
    /// [`GapResponse::StopSide`]) or a reassembly limit on that side
    /// ([`EndReason::BufferOverflow`]). The other side keeps being
    /// parsed; when both sides are stopped a `ParserClosed` follows.
    /// New in 0.25.0.
    ParserSideStopped {
        key: K,
        parser_kind: ParserKind,
        side: FlowSide,
        reason: EndReason,
        /// The gap size or the reassembly stop reason.
        detail: Option<String>,
        ts: Timestamp,
    },
    /// The flow ended (transport reason: FIN / RST / idle / eviction /
    /// force-close). `stats` includes the reassembly diagnostics.
    Closed {
        key: K,
        reason: EndReason,
        stats: FlowStats,
        l4: Option<L4Proto>,
        ts: Timestamp,
    },
    /// Per-flow anomaly (reassembly, parser poison). Only when the
    /// driver was built with anomalies enabled.
    FlowAnomaly {
        key: K,
        kind: AnomalyKind,
        ts: Timestamp,
    },
    /// Tracker-global anomaly (eviction pressure, memcap). Only when
    /// anomalies are enabled.
    TrackerAnomaly { kind: AnomalyKind, ts: Timestamp },
    /// Periodic stats snapshot, when
    /// [`crate::FlowTrackerConfig::flow_tick_interval`] is set.
    Tick {
        key: K,
        stats: FlowStats,
        ts: Timestamp,
    },
}

impl<K, M> SessionEvent<K, M> {
    /// The session form of a flow-tracker event, as the session
    /// engines emit it: `Started` and anomalies / ticks map 1:1,
    /// `Ended` becomes `Closed` (stamped with the flow's last packet).
    /// Per-packet `Packet` / `Established` / `StateChange` have no
    /// session form (`None`). Lets a caller moving from a flow stream to
    /// a session stream keep the flow events it already queued.
    pub fn from_flow_event(ev: crate::event::FlowEvent<K>) -> Option<Self> {
        use crate::event::FlowEvent;
        Some(match ev {
            FlowEvent::Started {
                key,
                side,
                orientation,
                ts,
                l4,
            } => SessionEvent::Started {
                key,
                side,
                orientation,
                ts,
                l4,
            },
            FlowEvent::Ended {
                key,
                reason,
                stats,
                l4,
                ..
            } => {
                let ts = stats.last_seen;
                SessionEvent::Closed {
                    key,
                    reason,
                    stats,
                    l4,
                    ts,
                }
            }
            FlowEvent::FlowAnomaly { key, kind, ts } => SessionEvent::FlowAnomaly { key, kind, ts },
            FlowEvent::TrackerAnomaly { kind, ts } => SessionEvent::TrackerAnomaly { kind, ts },
            FlowEvent::Tick { key, stats, ts } => SessionEvent::Tick { key, stats, ts },
            FlowEvent::Packet { .. }
            | FlowEvent::Established { .. }
            | FlowEvent::StateChange { .. } => return None,
        })
    }

    /// The flow key, when the variant has one.
    pub fn key(&self) -> Option<&K> {
        match self {
            SessionEvent::Started { key, .. }
            | SessionEvent::Application { key, .. }
            | SessionEvent::ParserClosed { key, .. }
            | SessionEvent::ParserSideStopped { key, .. }
            | SessionEvent::Closed { key, .. }
            | SessionEvent::FlowAnomaly { key, .. }
            | SessionEvent::Tick { key, .. } => Some(key),
            SessionEvent::TrackerAnomaly { .. } => None,
        }
    }

    /// The event's timestamp.
    pub fn timestamp(&self) -> Timestamp {
        match self {
            SessionEvent::Started { ts, .. }
            | SessionEvent::Application { ts, .. }
            | SessionEvent::ParserClosed { ts, .. }
            | SessionEvent::ParserSideStopped { ts, .. }
            | SessionEvent::Closed { ts, .. }
            | SessionEvent::FlowAnomaly { ts, .. }
            | SessionEvent::TrackerAnomaly { ts, .. }
            | SessionEvent::Tick { ts, .. } => *ts,
        }
    }

    /// The anomaly kind, for the two anomaly variants.
    pub fn anomaly_kind(&self) -> Option<&AnomalyKind> {
        match self {
            SessionEvent::FlowAnomaly { kind, .. } | SessionEvent::TrackerAnomaly { kind, .. } => {
                Some(kind)
            }
            _ => None,
        }
    }
}

/// A [`SessionParserFactory`] / [`DatagramParserFactory`] that clones
/// a template parser for every flow. The blanket factory impls need
/// `Default + Clone`; this one only needs `Clone`, for parsers built
/// from configuration.
#[derive(Debug, Clone)]
pub struct TemplateFactory<P>(pub P);

impl<K, P> SessionParserFactory<K> for TemplateFactory<P>
where
    P: SessionParser + Clone,
{
    type Parser = P;
    fn new_parser(&mut self, _key: &K) -> P {
        self.0.clone()
    }
}

impl<K, P> DatagramParserFactory<K> for TemplateFactory<P>
where
    P: DatagramParser + Clone,
{
    type Parser = P;
    fn new_parser(&mut self, _key: &K) -> P {
        self.0.clone()
    }
    fn transports(&self) -> Transports {
        self.0.transports()
    }
}

#[cfg(any(feature = "ftp", feature = "smtp"))]
/// Gap recovery for line-oriented protocols: the bytes right after a
/// gap start mid-line, so they are discarded up to and including the
/// next `\n`. Feed each new chunk through this while `pending` is
/// set; returns the part of `bytes` to parse.
pub(crate) fn skip_partial_line<'a>(pending: &mut bool, bytes: &'a [u8]) -> &'a [u8] {
    if !*pending {
        return bytes;
    }
    match bytes.iter().position(|&b| b == b'\n') {
        Some(p) => {
            *pending = false;
            &bytes[p + 1..]
        }
        None => &[],
    }
}

#[cfg(any(feature = "smb", feature = "modbus"))]
/// Gap recovery for framed binary protocols: when `pending`, drop
/// buffered bytes up to the first offset `find` accepts as a frame
/// start (keeping at most `keep` trailing bytes while none is found).
pub(crate) fn resync_frames(
    pending: &mut bool,
    buf: &mut bytes::BytesMut,
    keep: usize,
    find: impl Fn(&[u8]) -> Option<usize>,
) {
    if !*pending {
        return;
    }
    match find(buf) {
        Some(at) => {
            let _ = buf.split_to(at);
            *pending = false;
        }
        None => {
            let drop = buf.len().saturating_sub(keep);
            let _ = buf.split_to(drop);
        }
    }
}

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

    #[test]
    fn flow_events_map_to_their_session_form() {
        use crate::event::FlowEvent;
        let stats = FlowStats {
            last_seen: Timestamp::new(7, 0),
            ..Default::default()
        };
        let closed: Option<SessionEvent<u8, ()>> =
            SessionEvent::from_flow_event(FlowEvent::Ended {
                key: 1u8,
                reason: EndReason::Fin,
                stats,
                history: Default::default(),
                l4: None,
            });
        assert!(matches!(
            closed,
            Some(SessionEvent::Closed { key: 1, reason: EndReason::Fin, ts, .. }) if ts == Timestamp::new(7, 0)
        ));
        let tick: Option<SessionEvent<u8, ()>> = SessionEvent::from_flow_event(FlowEvent::Tick {
            key: 2u8,
            stats: FlowStats::default(),
            ts: Timestamp::new(8, 0),
        });
        assert!(matches!(tick, Some(SessionEvent::Tick { key: 2, .. })));
    }

    #[derive(Default, Clone)]
    struct CountParser {
        init_bytes: usize,
        resp_bytes: usize,
    }

    impl SessionParser for CountParser {
        type Message = (FlowSide, usize);
        fn feed_initiator(&mut self, b: &[u8], _ts: Timestamp, out: &mut Vec<Self::Message>) {
            self.init_bytes += b.len();
            out.push((FlowSide::Initiator, self.init_bytes));
        }
        fn feed_responder(&mut self, b: &[u8], _ts: Timestamp, out: &mut Vec<Self::Message>) {
            self.resp_bytes += b.len();
            out.push((FlowSide::Responder, self.resp_bytes));
        }
    }

    #[test]
    fn auto_impl_session_parser_factory() {
        // CountParser is Default + Clone + SessionParser → automatic factory.
        let mut f: CountParser = CountParser::default();
        let mut p: CountParser = SessionParserFactory::<u32>::new_parser(&mut f, &7);
        let mut m = Vec::new();
        p.feed_initiator(b"abc", Timestamp::default(), &mut m);
        assert_eq!(m, vec![(FlowSide::Initiator, 3)]);
    }

    #[derive(Default, Clone)]
    struct EchoDgram;
    impl DatagramParser for EchoDgram {
        type Message = (FlowSide, Vec<u8>);
        fn parse(
            &mut self,
            payload: &[u8],
            side: FlowSide,
            _ts: Timestamp,
            out: &mut Vec<Self::Message>,
        ) {
            out.push((side, payload.to_vec()));
        }
    }

    #[test]
    fn auto_impl_datagram_parser_factory() {
        let mut f = EchoDgram;
        let mut p: EchoDgram = DatagramParserFactory::<()>::new_parser(&mut f, &());
        let mut m = Vec::new();
        p.parse(b"hello", FlowSide::Responder, Timestamp::default(), &mut m);
        assert_eq!(m, vec![(FlowSide::Responder, b"hello".to_vec())]);
    }
}