@@ -26,24 +26,24 @@ public static abstract class Program extends BPFProgram {
2626 int kprobe__do_sys_openat2 (struct pt_regs *ctx)
2727 {
2828 int placed_value = 12;
29- int placed_value_in_filter = bpf_map_lookup_elem (&filter, &placed_value);
30- if (placed_value_in_filter == 0 ) {
29+ bool placed_value_in_filter = !bpf_map_peek_elem (&filter, &placed_value);
30+ if (placed_value_in_filter) {
3131 bpf_printk("Placed OK");
3232 } else {
3333 bpf_printk("Placed NOT OK");
3434 }
3535
3636 int value = 11;
37- int maybe_in_filter = bpf_map_lookup_elem (&filter, &value);
38- if (maybe_in_filter != 0 ) {
37+ bool maybe_in_filter = !bpf_map_peek_elem (&filter, &value);
38+ if (!maybe_in_filter ) {
3939 bpf_printk("Before OK");
4040 } else {
4141 bpf_printk("Before NOT OK");
4242 }
4343
4444 bpf_map_push_elem(&filter, &value, BPF_ANY);
45- maybe_in_filter = bpf_map_lookup_elem (&filter, &value);
46- if (maybe_in_filter == 0 ) {
45+ maybe_in_filter = !bpf_map_peek_elem (&filter, &value);
46+ if (maybe_in_filter) {
4747 bpf_printk("After OK");
4848 } else {
4949 bpf_printk("After NOT OK");
@@ -54,7 +54,7 @@ int kprobe__do_sys_openat2 (struct pt_regs *ctx)
5454 }
5555
5656 @ Test
57- public void testBasicArrayMap () throws InterruptedException {
57+ public void testBasicBloomFilter () throws InterruptedException {
5858 try (var program = BPFProgram .load (BloomFilterMapTest .Program .class )) {
5959 var filter = program .filter ;
6060 filter .put (12 );
@@ -63,23 +63,23 @@ public void testBasicArrayMap() throws InterruptedException {
6363
6464 while (true ) {
6565 var msg = program .readTraceFields ().msg ();
66- if (msg != null ) {
66+ if (msg != null && msg . contains ( "Placed" ) ) {
6767 assertEquals ("Placed OK" , msg );
6868 break ;
6969 }
7070 }
7171
7272 while (true ) {
7373 var msg = program .readTraceFields ().msg ();
74- if (msg != null ) {
74+ if (msg != null && msg . contains ( "Before" ) ) {
7575 assertEquals ("Before OK" , msg );
7676 break ;
7777 }
7878 }
7979
8080 while (true ) {
8181 var msg = program .readTraceFields ().msg ();
82- if (msg != null ) {
82+ if (msg != null && msg . contains ( "After" ) ) {
8383 assertEquals ("After OK" , msg );
8484 break ;
8585 }
0 commit comments