1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
use crate::{Batch, Chain, Hook, MsgType, Policy, Table};
use std::ffi::CString;
use std::rc::Rc;
/// A helper trait over [`crate::Chain`].
pub trait ChainMethods {
/// Creates a new Chain instance from a [`crate::Hook`] over a [`crate::Table`].
fn from_hook(hook: Hook, table: Rc<Table>) -> Self
where Self: std::marker::Sized;
/// Adds a [`crate::Policy`] to the current Chain.
fn verdict(self, policy: Policy) -> Self;
fn add_to_batch(self, batch: &mut Batch) -> Self;
}
impl ChainMethods for Chain {
fn from_hook(hook: Hook, table: Rc<Table>) -> Self {
let chain_name = match hook {
Hook::PreRouting => "prerouting",
Hook::Out => "out",
Hook::PostRouting => "postrouting",
Hook::Forward => "forward",
Hook::In => "in",
};
let chain_name = CString::new(chain_name).unwrap();
let mut chain = Chain::new(&chain_name, table);
chain.set_hook(hook, 0);
chain
}
fn verdict(mut self, policy: Policy) -> Self {
self.set_policy(policy);
self
}
fn add_to_batch(self, batch: &mut Batch) -> Self {
batch.add(&self, MsgType::Add);
self
}
}
|
