nftnl 0.9.2

use std::ptr;

use super::{Expression, Rule};
use nftnl_sys::{self as sys, libc};

bitflags::bitflags! {
    #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
    pub struct States: u32 {
        const INVALID = 1;
        const ESTABLISHED = 2;
        const RELATED = 4;
        const NEW = 8;
        const UNTRACKED = 64;
    }
}

bitflags::bitflags! {
    #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
    pub struct ConntrackStatus: u32 {
        const EXPECTED = 1;
        const SEEN_REPLY = 2;
        const ASSURED = 4;
        const CONFIRMED = 8;
        const SRC_NAT = 16;
        const DST_NAT = 32;
        const SEQ_ADJUST = 64;
        const SRC_NAT_DONE = 128;
        const DST_NAT_DONE = 256;
        const DYING = 512;
        const FIXED_TIMEOUT = 1024;
        const TEMPLATE = 2048;
        const UNTRACKED = 4096;
        const HELPER = 8192;
        const OFFLOAD = 16384;
        const HW_OFFLOAD = 32768;
    }
}

pub enum Conntrack {
    State,
    Status,
    Mark { set: bool },
}

impl Conntrack {
    fn raw_key(&self) -> u32 {
        match *self {
            Conntrack::State => libc::NFT_CT_STATE as u32,
            Conntrack::Status => libc::NFT_CT_STATUS as u32,
            Conntrack::Mark { .. } => libc::NFT_CT_MARK as u32,
        }
    }
}

impl Expression for Conntrack {
    fn to_expr(&self, _rule: &Rule) -> ptr::NonNull<sys::nftnl_expr> {
        let expr = try_alloc!(unsafe { sys::nftnl_expr_alloc(c"ct".as_ptr()) });

        if let Conntrack::Mark { set: true } = self {
            unsafe {
                sys::nftnl_expr_set_u32(
                    expr.as_ptr(),
                    sys::NFTNL_EXPR_CT_SREG as u16,
                    libc::NFT_REG_1 as u32,
                )
            };
        } else {
            unsafe {
                sys::nftnl_expr_set_u32(
                    expr.as_ptr(),
                    sys::NFTNL_EXPR_CT_DREG as u16,
                    libc::NFT_REG_1 as u32,
                )
            };
        }

        unsafe {
            sys::nftnl_expr_set_u32(expr.as_ptr(), sys::NFTNL_EXPR_CT_KEY as u16, self.raw_key());
        }

        expr
    }
}

#[macro_export]
macro_rules! nft_expr_ct {
    (state) => {
        $crate::expr::Conntrack::State
    };
    (status) => {
        $crate::expr::Conntrack::Status
    };
    (mark set) => {
        $crate::expr::Conntrack::Mark { set: true }
    };
    (mark) => {
        $crate::expr::Conntrack::Mark { set: false }
    };
}