kernel/linux: fix handle matching

hack3ric · Jan 10, 2025 · d974721 · d974721
1 parent 6ef171e
commit d974721
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/kernel/linux/mod.rs b/src/kernel/linux/mod.rs
@@ -82,14 +82,18 @@ impl Kernel for Linux {
 
     let handle: Self::Handle = (result.objects.iter())
       .filter_map(|x| {
-        if let NfObject::CmdObject(NfCmd::Add(NfListObject::Rule(rule))) = x {
+        if let NfObject::CmdObject(NfCmd::Add(NfListObject::Rule(rule)))
+        | NfObject::CmdObject(NfCmd::Insert(NfListObject::Rule(rule))) = x
+        {
           Some(rule.handle.unwrap())
         } else {
           None
         }
       })
       .collect();
 
+    debug_assert_eq!(nftables.objects.len(), handle.len());
+
     if let Some((next_hop, table_id)) = rt_info {
       let rtnl = self.rtnl.as_mut().expect("RtNetlink should be initialized");
       let real_table_id = rtnl.add(handle.clone(), spec, next_hop).await?;