From 3335e0471bc89c8d12dc414c538c97025df1341b Mon Sep 17 00:00:00 2001
From: Pragyansh Chaturvedi <pragyanshchaturvedi18@gmail.com>
Date: Mon, 24 Nov 2025 23:51:08 +0530
Subject: [PATCH] Return None instead of KeyError for failure of BpfMap::lookup
 for missing key

---
 src/core/bpf_map.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/core/bpf_map.cpp b/src/core/bpf_map.cpp
index 213233f..794ad3a 100644
--- a/src/core/bpf_map.cpp
+++ b/src/core/bpf_map.cpp
@@ -42,7 +42,7 @@ py::object BpfMap::lookup(const py::object &key) const {
                                        value_span.data(), value_size_, BPF_ANY);
   if (ret < 0) {
     if (ret == -ENOENT)
-      throw py::key_error("Key not found in map '" + map_name_ + "'");
+      return py::none();
     throw BpfException("Failed to lookup key in map '" + map_name_ +
                        "': " + std::strerror(-ret));
   }