use super::{Expression, Rule};
use crate::set::Set;
use nftnl_sys::{self as sys, libc};
use std::{ffi::CString, ptr};
pub struct Lookup {
set_name: CString,
set_id: u32,
}
impl Lookup {
pub fn new<K>(set: &Set<'_, K>) -> Self {
Lookup {
set_name: set.get_name().to_owned(),
set_id: set.get_id(),
}
}
}
impl Expression for Lookup {
fn to_expr(&self, _rule: &Rule) -> ptr::NonNull<sys::nftnl_expr> {
unsafe {
let expr = try_alloc!(sys::nftnl_expr_alloc(c"lookup".as_ptr()));
sys::nftnl_expr_set_u32(
expr.as_ptr(),
sys::NFTNL_EXPR_LOOKUP_SREG as u16,
libc::NFT_REG_1 as u32,
);
sys::nftnl_expr_set_str(
expr.as_ptr(),
sys::NFTNL_EXPR_LOOKUP_SET as u16,
self.set_name.as_ptr(),
);
sys::nftnl_expr_set_u32(
expr.as_ptr(),
sys::NFTNL_EXPR_LOOKUP_SET_ID as u16,
self.set_id,
);
expr
}
}
}
#[macro_export]
macro_rules! nft_expr_lookup {
($set:expr) => {
$crate::expr::Lookup::new($set)
};
}