pub mod activated;

pub mod inactive;

#[cfg(all(not(windows), feature = "capture-stream"))]

#[cfg_attr(docsrs, doc(cfg(all(not(windows), feature = "capture-stream"))))]

pub mod selectable;

use std::{

    ffi::CString,

    fmt,

    marker::PhantomData,

    mem::MaybeUninit,

    ptr::{self, NonNull},

    sync::Arc,

};

#[cfg(windows)]

use windows_sys::Win32::Foundation::HANDLE;

use crate::{Error, packet::PacketHeader, raw};

/// Phantom type representing an inactive capture handle.

pub enum Inactive {}

/// Phantom type representing an active capture handle.

pub enum Active {}

/// Phantom type representing an offline capture handle, from a pcap dump file.

/// Implements `Activated` because it behaves nearly the same as a live handle.

pub enum Offline {}

/// Phantom type representing a dead capture handle.  This can be use to create

/// new save files that are not generated from an active capture.

/// Implements `Activated` because it behaves nearly the same as a live handle.

pub enum Dead {}

/// `Capture`s can be in different states at different times, and in these states they

/// may or may not have particular capabilities. This trait is implemented by phantom

/// types which allows us to punt these invariants to the type system to avoid runtime

/// errors.

pub trait Activated: State {}

impl Activated for Active {}

impl Activated for Offline {}

impl Activated for Dead {}

/// `Capture`s can be in different states at different times, and in these states they

/// may or may not have particular capabilities. This trait is implemented by phantom

/// types which allows us to punt these invariants to the type system to avoid runtime

/// errors.

pub trait State {}

impl State for Inactive {}

impl State for Active {}

impl State for Offline {}

impl State for Dead {}

/// This is a pcap capture handle which is an abstraction over the `pcap_t` provided by pcap.

/// There are many ways to instantiate and interact with a pcap handle, so phantom types are

/// used to express these behaviors.

///

/// **`Capture<Inactive>`** is created via `Capture::from_device()`. This handle is inactive,

/// so you cannot (yet) obtain packets from it. However, you can configure things like the

/// buffer size, snaplen, timeout, and promiscuity before you activate it.

///

/// **`Capture<Active>`** is created by calling `.open()` on a `Capture<Inactive>`. This

/// activates the capture handle, allowing you to get packets with `.next_packet()` or apply filters

/// with `.filter()`.

///

/// **`Capture<Offline>`** is created via `Capture::from_file()`. This allows you to read a

/// pcap format dump file as if you were opening an interface -- very useful for testing or

/// analysis.

///

/// **`Capture<Dead>`** is created via `Capture::dead()`. This allows you to create a pcap

/// format dump file without needing an active capture.

///

/// # Example:

///

/// ```no_run

/// # use pcap::{Capture, Device};

/// let mut cap = Capture::from_device(Device::lookup().unwrap().unwrap()) // open the "default" interface

///               .unwrap() // assume the device exists and we are authorized to open it

///               .open() // activate the handle

///               .unwrap(); // assume activation worked

///

/// while let Ok(packet) = cap.next_packet() {

///     println!("received packet! {:?}", packet);

/// }

/// ```

pub struct Capture<T: State + ?Sized> {

    nonblock: bool,

    warning: Option<Warning>,

    #[cfg(windows)]

    min_to_copy: Option<i32>,

    // Holds the header that `next_packet` converts when libpcap lays it out differently from

    // `PacketHeader`.

    header: MaybeUninit<PacketHeader>,

    handle: Arc<PcapHandle>,

    _marker: PhantomData<T>,

}

struct PcapHandle {

    handle: NonNull<raw::pcap_t>,

}

impl PcapHandle {

}

// `PcapHandle` is safe to Send as it encapsulates the entire lifetime of `raw::pcap_t *`

// `PcapHandle` is only Sync under special circumstances when used in thread-safe functions such as

// the `pcap_breakloop` function. The Sync correctness is left to the wrapping structure to provide.

unsafe impl Send for PcapHandle {}

impl Drop for PcapHandle {

}

unsafe impl<T: State + ?Sized> Send for Capture<T> {}

impl<T: State + ?Sized> Capture<T> {

    // `Capture` is not safe to implement Sync as the libpcap functions it uses are not promised to

    // have thread-safe access to the same `raw::pcap_t *` from multiple threads.

    #[allow(clippy::arc_with_non_send_sync)]

    {

            };

            ))

    /// Set the minumum amount of data received by the kernel in a single call.

    ///

    /// Note that this value is set to 0 when the capture is set to immediate mode. You should not

    /// call `min_to_copy` on captures in immediate mode if you want them to stay in immediate mode.

    #[cfg(windows)]

    pub fn min_to_copy(self, to: i32) -> Capture<T> {

        unsafe {

            raw::pcap_setmintocopy(self.handle.as_ptr(), to as _);

        }

        self

    }

    /// Get handle to the Capture context's internal Win32 event semaphore.

    ///

    /// Setting this event will cause a blocking capture call to unblock and return.

    ///

    /// # Example

    /// The _winevt_ example demonstrates how to use the event semaphore to send command requests

    /// to a capture loop running in a separate thread.

    ///

    /// # Safety

    ///

    /// The caller must ensure that the `Capture` context outlives the returned `HANDLE` since it is

    /// a kernel object owned by the `Capture`'s pcap context.

    #[cfg(windows)]

    pub unsafe fn get_event(&self) -> HANDLE {

        unsafe { raw::pcap_getevent(self.handle.as_ptr()) }

    }

    /// Turn one of libpcap's status codes into an error, with the message it left behind.

    /// Whether the capture is reading a savefile rather than an interface.

}

#[repr(u32)]

#[derive(Debug, PartialEq, Eq, Clone, Copy)]

/// Timestamp resolution types

///

/// Not all systems and interfaces will necessarily support all of these resolutions when doing

/// live captures; all of them can be requested when reading a safefile.

pub enum Precision {

    /// Use timestamps with microsecond precision. This is the default.

    Micro = 0,

    /// Use timestamps with nanosecond precision.

    Nano = 1,

}

/// What libpcap warned about while activating a capture.

///

/// A warning is not a failure. The capture works, but a requested option could not be applied.

/// [`Capture::warning`] reports the one an activation raised.

#[derive(Debug, PartialEq, Eq, Clone)]

pub struct Warning {

    /// The condition libpcap has a code of its own for

    pub code: WarningCode,

    /// The message libpcap left with it, empty if it left none

    pub message: String,

}

impl Warning {

    /// Read one of the warnings libpcap activates with, along with the message it left behind.

        };

}

impl fmt::Display for Warning {

        } else {

        }

}

/// A list of libpcap's own warning codes.

#[derive(Debug, PartialEq, Eq, Clone, Copy)]

#[non_exhaustive]

pub enum WarningCode {

    /// Promiscuous mode was asked for but the device does not have it

    PromiscuousModeNotSupported,

    /// The device does not support the timestamp type that was set

    TimestampTypeNotSupported,

    /// Something libpcap has no code of its own for

    Unspecified,

}

impl fmt::Display for WarningCode {

            WarningCode::PromiscuousModeNotSupported => {

            }

            WarningCode::TimestampTypeNotSupported => {

            }

        })

}

// GRCOV_EXCL_START

#[cfg(test)]

pub mod testmod {

    use raw::testmod::RAWMTX;

    use super::*;

    use crate::packet::{self, Layout};

    pub struct TestCapture<T: State + ?Sized> {

        pub capture: Capture<T>,

        _close_ctx: raw::__pcap_close::Context,

        _layout_ctx: packet::__layout::Context,

    }

    pub fn test_capture<T: State + ?Sized>(pcap: *mut raw::pcap_t) -> TestCapture<T> {

        // Lock must be acquired by caller.

        assert!(RAWMTX.try_lock().is_err());

        let ctx = raw::pcap_close_context();

        ctx.checkpoint();

        ctx.expect()

            .withf_st(move |ptr| *ptr == pcap)

            .return_once(|_| {});

        let layout_ctx = packet::layout_context();

        layout_ctx.expect().times(..).return_const(Layout::Native);

        TestCapture {

            capture: Capture::<T>::from_handle(NonNull::new(pcap).unwrap()),

            _close_ctx: ctx,

            _layout_ctx: layout_ctx,

        }

    }

}

// GRCOV_EXCL_STOP

#[cfg(test)]

mod tests {

    use crate::{

        capture::testmod::test_capture,

        raw::testmod::{RAWMTX, as_pcap_t},

    };

    use super::*;

    #[test]

    fn test_warning_from_status() {

        let message = CString::new("not quite").unwrap();

        let cases = [

            (

                raw::PCAP_WARNING_PROMISC_NOTSUP,

                WarningCode::PromiscuousModeNotSupported,

            ),

            (

                raw::PCAP_WARNING_TSTAMP_TYPE_NOTSUP,

                WarningCode::TimestampTypeNotSupported,

            ),

            (raw::PCAP_WARNING, WarningCode::Unspecified),

            // A code libpcap has yet to define still warns.

            (99, WarningCode::Unspecified),

        ];

        for (status, code) in cases {

            let warning = unsafe { Warning::from_status(status, message.as_ptr()) };

            assert_eq!(warning.code, code);

            assert_eq!(warning.to_string(), "libpcap warning: not quite");

            // With no message left behind, the code has to describe itself.

            let warning = unsafe { Warning::from_status(status, ptr::null()) };

            assert!(warning.message.is_empty());

            assert_eq!(warning.to_string(), format!("libpcap warning: {code}"));

        }

    }

    #[test]

    fn test_capture_getters() {

        let _m = RAWMTX.lock();

        let mut dummy: isize = 777;

        let pcap = as_pcap_t(&mut dummy);

        let test_capture = test_capture::<Active>(pcap);

        let capture = test_capture.capture;

        assert!(!capture.is_nonblock());

        assert_eq!(capture.as_ptr(), capture.handle.as_ptr());

    }

    #[test]

    #[cfg(windows)]

    fn test_min_to_copy() {

        let _m = RAWMTX.lock();

        let mut dummy: isize = 777;

        let pcap = as_pcap_t(&mut dummy);

        let test_capture = test_capture::<Active>(pcap);

        let capture = test_capture.capture;

        let ctx = raw::pcap_setmintocopy_context();

        ctx.expect()

            .withf_st(move |arg1, _| *arg1 == pcap)

            .return_once(|_, _| 0);

        let _capture = capture.min_to_copy(5);

    }

    #[test]

    #[cfg(windows)]

    fn test_get_event() {

        let _m = RAWMTX.lock();

        let mut dummy: isize = 777;

        let pcap = as_pcap_t(&mut dummy);

        let test_capture = test_capture::<Active>(pcap);

        let capture = test_capture.capture;

        let ctx = raw::pcap_getevent_context();

        ctx.expect()

            .withf_st(move |arg1| *arg1 == pcap)

            .return_once(|_| 5 as HANDLE);

        let handle = unsafe { capture.get_event() };

        assert_eq!(handle, 5 as HANDLE);

    }

    #[test]

    fn test_precision() {

        assert_ne!(Precision::Micro, Precision::Nano);

    }

}