fix: guard against division by zero in libbpf-rs

EHfive · Apr 5, 2024 · 3301262 · 3301262
1 parent 9ae58b6
commit 3301262
Showing 1 changed file with 16 additions and 12 deletions.
diff --git a/src/instance.rs b/src/instance.rs
@@ -919,12 +919,14 @@ fn remove_binding_and_ct_entires(skel: &EinatSkel, external_addr: IpAddr) -> Res
         }
     }
 
-    map_binding.delete_batch(
-        &to_delete_binding_keys,
-        (to_delete_binding_keys.len() / core::mem::size_of::<MapBindingKey>()) as _,
-        MapFlags::ANY,
-        MapFlags::ANY,
-    )?;
+    if !to_delete_binding_keys.is_empty() {
+        map_binding.delete_batch(
+            &to_delete_binding_keys,
+            (to_delete_binding_keys.len() / core::mem::size_of::<MapBindingKey>()) as _,
+            MapFlags::ANY,
+            MapFlags::ANY,
+        )?;
+    }
 
     let mut to_delete_ct_keys = Vec::new();
     for ct_key_raw in map_ct.keys() {
@@ -934,12 +936,14 @@ fn remove_binding_and_ct_entires(skel: &EinatSkel, external_addr: IpAddr) -> Res
         }
     }
 
-    map_ct.delete_batch(
-        &to_delete_ct_keys,
-        (to_delete_ct_keys.len() / core::mem::size_of::<MapCtKey>()) as _,
-        MapFlags::ANY,
-        MapFlags::ANY,
-    )?;
+    if !to_delete_ct_keys.is_empty() {
+        map_ct.delete_batch(
+            &to_delete_ct_keys,
+            (to_delete_ct_keys.len() / core::mem::size_of::<MapCtKey>()) as _,
+            MapFlags::ANY,
+            MapFlags::ANY,
+        )?;
+    }
 
     Ok(())
 }