From e0db9861bcf7194c29888c28184785f969199c38 Mon Sep 17 00:00:00 2001
From: Mathias Hall-Andersen <mathias@hall-andersen.dk>
Date: Sat, 14 Dec 2019 13:37:51 +0100
Subject: Added profiler feature

---
 src/wireguard/router/outbound.rs | 34 +++++++++++++++++++---------------
 1 file changed, 19 insertions(+), 15 deletions(-)

(limited to 'src/wireguard/router/outbound.rs')

diff --git a/src/wireguard/router/outbound.rs b/src/wireguard/router/outbound.rs
index 9ecffd8..a555ecb 100644
--- a/src/wireguard/router/outbound.rs
+++ b/src/wireguard/router/outbound.rs
@@ -1,3 +1,4 @@
+use super::constants::MAX_INORDER_CONSUME;
 use super::device::Device;
 use super::messages::{TransportHeader, TYPE_TRANSPORT};
 use super::peer::Peer;
@@ -88,20 +89,23 @@ pub fn sequential<E: Endpoint, C: Callbacks, T: tun::Writer, B: udp::Writer<E>>(
     device: Device<E, C, T, B>,
 ) {
     device.run_outbound.run(|peer| {
-        peer.outbound.handle(|body| {
-            log::trace!("worker, sequential section, obtained job");
-
-            // send to peer
-            let xmit = peer.send(&body.msg[..]).is_ok();
-
-            // trigger callback
-            C::send(
-                &peer.opaque,
-                body.msg.len(),
-                xmit,
-                &body.keypair,
-                body.counter,
-            );
-        });
+        peer.outbound.handle(
+            |body| {
+                log::trace!("worker, sequential section, obtained job");
+
+                // send to peer
+                let xmit = peer.send(&body.msg[..]).is_ok();
+
+                // trigger callback
+                C::send(
+                    &peer.opaque,
+                    body.msg.len(),
+                    xmit,
+                    &body.keypair,
+                    body.counter,
+                );
+            },
+            MAX_INORDER_CONSUME,
+        )
     });
 }
-- 
cgit v1.2.3-59-g8ed1b