Hotfix: send_ex improper use of iterator

Rubensei · Jan 7, 2023 · df70e96 · df70e96
1 parent b8cfd2a
commit df70e96
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/windivert/Cargo.toml b/windivert/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "windivert"
-version = "0.5.3"
+version = "0.5.4"
 description = "Wrapper library around windivert-sys"
 authors = ["Ruben Serrano Izquierdo <rserranoizq@gmail.com>"]
 repository = "https://github.com/Rubensei/windivert-rust.git"

diff --git a/windivert/src/divert/blocking.rs b/windivert/src/divert/blocking.rs
@@ -108,16 +108,15 @@ impl<L: layer::WinDivertLayerTrait> WinDivert<L> {
         Ok(injected_length)
     }
 
-    fn internal_send_ex<'data, 'packets, P>(&self, mut packets: P) -> Result<u32, WinDivertError>
+    fn internal_send_ex<'data, 'packets, P>(&self, packets: P) -> Result<u32, WinDivertError>
     where
         P: ExactSizeIterator<Item = &'packets WinDivertPacket<'data, L>>,
         'data: 'packets,
         L: 'packets,
     {
         let packet_count = packets.len();
         let mut injected_length = 0;
-        let capacity = packets.by_ref().map(|p| p.data.len()).sum();
-        let mut packet_buffer: Vec<u8> = Vec::with_capacity(capacity);
+        let mut packet_buffer: Vec<u8> = Vec::new();
         let mut address_buffer: Vec<WINDIVERT_ADDRESS> = Vec::with_capacity(packet_count);
         packets.for_each(|packet: &'packets WinDivertPacket<'data, L>| {
             packet_buffer.extend(&packet.data[..]);