From 5a85043fb184c598db3010314b005e5b4718787e Mon Sep 17 00:00:00 2001
From: arctic-alpaca <67190338+arctic-alpaca@users.noreply.github.com>
Date: Wed, 13 Nov 2024 20:07:41 +0100
Subject: [PATCH] replace hard-coded queue ID with queried queue ID

---
 test/integration-ebpf/src/redirect.rs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/test/integration-ebpf/src/redirect.rs b/test/integration-ebpf/src/redirect.rs
index 2b695945..215b43fa 100644
--- a/test/integration-ebpf/src/redirect.rs
+++ b/test/integration-ebpf/src/redirect.rs
@@ -31,7 +31,9 @@ pub fn redirect_sock(ctx: XdpContext) -> u32 {
     // Check whether incoming packet's queue ID matches the queue ID of the socket in XSKMAP at index `queue_id`.
     if SOCKS.get(queue_id) == Some(queue_id) {
         // Queue ID matches, redirect to AF_XDP socket.
-        SOCKS.redirect(0, 0).unwrap_or(xdp_action::XDP_ABORTED)
+        SOCKS
+            .redirect(queue_id, 0)
+            .unwrap_or(xdp_action::XDP_ABORTED)
     } else {
         // Queue ID did not match, pass packet to kernel network stack.
         xdp_action::XDP_PASS