Fix BloomFilter test

parttimenerd · Feb 16, 2025 · fb5f3fd · fb5f3fd
1 parent 2314cb0
commit fb5f3fd
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/bpf/src/test/java/me/bechberger/ebpf/bpf/BloomFilterMapTest.java b/bpf/src/test/java/me/bechberger/ebpf/bpf/BloomFilterMapTest.java
@@ -26,24 +26,24 @@ public static abstract class Program extends BPFProgram {
                  int kprobe__do_sys_openat2 (struct pt_regs *ctx)
             {
               int placed_value = 12;
-              int placed_value_in_filter = bpf_map_lookup_elem(&filter, &placed_value);
-              if (placed_value_in_filter == 0) {
+              bool placed_value_in_filter = !bpf_map_peek_elem(&filter, &placed_value);
+              if (placed_value_in_filter) {
                 bpf_printk("Placed OK");
               } else {
                 bpf_printk("Placed NOT OK");
               }
 
               int value = 11;
-              int maybe_in_filter = bpf_map_lookup_elem(&filter, &value);
-              if (maybe_in_filter != 0) {
+              bool maybe_in_filter = !bpf_map_peek_elem(&filter, &value);
+              if (!maybe_in_filter) {
                 bpf_printk("Before OK");
               } else {
                 bpf_printk("Before NOT OK");
               }
 
               bpf_map_push_elem(&filter, &value, BPF_ANY);
-              maybe_in_filter = bpf_map_lookup_elem(&filter, &value);
-              if (maybe_in_filter == 0) {
+              maybe_in_filter = !bpf_map_peek_elem(&filter, &value);
+              if (maybe_in_filter) {
                 bpf_printk("After OK");
               } else {
                 bpf_printk("After NOT OK");
@@ -54,7 +54,7 @@ int kprobe__do_sys_openat2 (struct pt_regs *ctx)
     }
 
     @Test
-    public void testBasicArrayMap() throws InterruptedException {
+    public void testBasicBloomFilter() throws InterruptedException {
         try (var program = BPFProgram.load(BloomFilterMapTest.Program.class)) {
             var filter = program.filter;
             filter.put(12);
@@ -63,23 +63,23 @@ public void testBasicArrayMap() throws InterruptedException {
 
             while (true) {
                 var msg = program.readTraceFields().msg();
-                if (msg != null) {
+                if (msg != null && msg.contains("Placed")) {
                     assertEquals("Placed OK", msg);
                     break;
                 }
             }
 
             while (true) {
                 var msg = program.readTraceFields().msg();
-                if (msg != null) {
+                if (msg != null && msg.contains("Before")) {
                     assertEquals("Before OK", msg);
                     break;
                 }
             }
 
             while (true) {
                 var msg = program.readTraceFields().msg();
-                if (msg != null) {
+                if (msg != null && msg.contains("After")) {
                     assertEquals("After OK", msg);
                     break;
                 }