more fixes or something

src/example.rs

@@ -14,7 +14,7 @@ use std::{
 
 use ctor::ctor;
 use ipnet::{IpNet, Ipv4Net, Ipv6Net};
-use iptrie::{IpPrefix, RTrieSet};
+use iptrie::IpPrefix;
 use serde::{
     de::{Error, Visitor},
     Deserialize,
@@ -137,16 +137,17 @@ impl<T> Default for IpCache<T> {
 fn ignore<T>(_: &mut smallvec::SmallVec<[T; 4]>) {}
 
 impl<T> IpCache<T> {
-    fn extend_set_with_domain<J: IpPrefix + From<T>>(
+    fn extend_set_with_domain<J: Helper + From<T>>(
         &self,
-        ips: &mut RTrieSet<J>,
+        ips: &mut NftData<J>,
         domain_r: IpCacheKey,
     ) where
         T: Copy,
+        IpNet: From<J>,
     {
         self.get_maybe_update_rev(domain_r, |val| {
             if let Some(val) = val {
-                ips.extend(val.0.iter().copied().map(|x| J::from(x)));
+                ips.extend(val.0.iter().copied().map(From::from));
             }
             #[allow(unused_assignments)]
             let mut val = Some(ignore);
@@ -381,10 +382,10 @@ impl ExampleMod {
                 let rev_domain = IpCacheKey::from_split_rev_domain(rev_domain.into_iter());
                 self.caches
                     .0
-                    .extend_set_with_domain(r.0.ips_mut(), rev_domain.clone());
+                    .extend_set_with_domain(&mut r.0, rev_domain.clone());
                 self.caches
                     .1
-                    .extend_set_with_domain(r.1.ips_mut(), rev_domain.clone());
+                    .extend_set_with_domain(&mut r.1, rev_domain.clone());
             }
         }
     }
@@ -491,8 +492,9 @@ impl ExampleMod {
         split_domain: &[&[u8]],
         qnames: SmallVec<[usize; 5]>,
     ) -> Result<(), ()> {
-        println!("adding {ip4:?}/{ip6:?} for {split_domain:?} to {qnames:?}");
-        if !ip4.is_empty() || !ip6.is_empty() {
+        if ip4.is_empty() && ip6.is_empty() {
+            return Ok(());
+        }
         let mut first = true;
         let domain = match split_domain
             .iter()
@@ -527,11 +529,9 @@ impl ExampleMod {
                 .send((qnames, to_send))
                 .unwrap();
         }
-        }
         Ok(())
     }
     fn run_commands(&self, split_domain: &[&[u8]]) -> Option<ModuleExtState> {
-        println!("{split_domain:?} {:?}", self.nft_token);
         if let Some(split_domain) = self.nft_token.as_ref().and_then(|token| {
             split_domain
                 .split_last()

src/nftables.rs

@@ -332,8 +332,8 @@ fn iter_ip_trie<T: Helper>(trie: &RTrieSet<T>) -> impl '_ + Iterator<Item = T> {
 }
 
 pub(crate) struct NftData<T: Helper> {
+    all_ips: RTrieSet<T>,
     ips: RTrieSet<T>,
-    dirty: bool,
     set: Option<Set1>,
     name: String,
 }
@@ -343,7 +343,7 @@ impl<T: Helper> NftData<T> {
         Self {
             set: None,
             ips: RTrieSet::new(),
-            dirty: true,
+            all_ips: RTrieSet::new(),
             name: name.to_owned(),
         }
     }
@@ -361,17 +361,24 @@ where
     pub fn verify(&mut self) -> bool {
         if !self.name.is_empty() && self.set.is_none() {
             self.ips = RTrieSet::new();
+            self.all_ips = RTrieSet::new();
             false
         } else {
             true
         }
     }
+    fn dirty(&self) -> bool {
+        usize::from(self.ips.len()) > 1
+    }
     pub fn flush_changes(
         &mut self,
         socket: &mnl::Socket,
         flush_set: bool,
     ) -> Result<(), io::Error> {
-        if let Some(set) = self.set.as_mut().filter(|_| self.dirty) {
+        if !self.dirty() {
+            return Ok(());
+        }
+        if let Some(set) = self.set.as_mut() {
             if flush_set {
                 println!(
                     "initializing set {} with ~{} ips (e.g. {:?})",
@@ -380,7 +387,9 @@ where
                     iter_ip_trie(&self.ips).next(),
                 );
             }
-            set.add_cidrs(socket, flush_set, iter_ip_trie(&self.ips).map(IpNet::from))
+            let ret = set.add_cidrs(socket, flush_set, iter_ip_trie(&self.ips).map(IpNet::from));
+            self.ips = RTrieSet::new();
+            ret
         } else {
             Ok(())
         }
@@ -395,15 +404,12 @@ where
             !self.name.is_empty()
         } else {
             self.set.is_some()
-        }) && should_add(&self.ips, &ip)
+        }) && should_add(&self.all_ips, &ip)
         {
             self.ips.insert(ip);
-            self.dirty = true;
+            self.all_ips.insert(ip);
         }
     }
-    pub fn ips_mut(&mut self) -> &mut RTrieSet<T> {
-        &mut self.ips
-    }
     #[cfg(test)]
     pub fn ip_count(&self) -> usize {
         iter_ip_trie(&self.ips).count()