diff options
Diffstat (limited to 'src/platform')
| -rw-r--r-- | src/platform/linux.rs | 4 | ||||
| -rw-r--r-- | src/platform/windows.rs | 7 |
2 files changed, 9 insertions, 2 deletions
diff --git a/src/platform/linux.rs b/src/platform/linux.rs index f19adc3..0faef31 100644 --- a/src/platform/linux.rs +++ b/src/platform/linux.rs @@ -213,6 +213,7 @@ pub fn run() -> Result<()> { }; use nfq::Queue; use crate::handle_packet; + use super::PACKET_SIZE_CAP; let mut q = Queue::open()?; q.bind(queue_num())?; @@ -227,6 +228,8 @@ pub fn run() -> Result<()> { splash!("{MESSAGE_AT_RUN}"); + let mut buf = Vec::<u8>::with_capacity(PACKET_SIZE_CAP); + while crate::RUNNING.load(Ordering::SeqCst) { { let fd = q.as_fd(); @@ -246,6 +249,7 @@ pub fn run() -> Result<()> { while let Ok(mut msg) = q.recv() { let verdict = handle_packet!( &msg.get_payload(), + &mut buf, handled => nfq::Verdict::Drop, rejected => nfq::Verdict::Accept, ); diff --git a/src/platform/windows.rs b/src/platform/windows.rs index 35ce956..f3ee2c0 100644 --- a/src/platform/windows.rs +++ b/src/platform/windows.rs @@ -62,16 +62,19 @@ pub fn send_to_raw(pkt: &[u8]) -> Result<()> { pub fn run() -> Result<()> { use crate::{handle_packet, RUNNING, MESSAGE_AT_RUN}; + use super::PACKET_SIZE_CAP; - let mut buf = vec![0u8; 65536]; + let mut windivert_buf = vec![0u8; 65536]; + let mut buf = Vec::<u8>::with_capacity(PACKET_SIZE_CAP); splash!("{MESSAGE_AT_RUN}"); while RUNNING.load(Ordering::SeqCst) { - let pkt = WINDIVERT_HANDLE.recv(Some(&mut buf))?; + let pkt = WINDIVERT_HANDLE.recv(Some(&mut windivert_buf))?; handle_packet!( &pkt.data, + &mut buf, handled => {}, rejected => { WINDIVERT_HANDLE.send(&pkt)?; } ); |