Skip to content

Commit fb5f3fd

Browse files
Mr-Pineparttimenerd
authored andcommitted
Fix BloomFilter test
1 parent 2314cb0 commit fb5f3fd

1 file changed

Lines changed: 10 additions & 10 deletions

File tree

bpf/src/test/java/me/bechberger/ebpf/bpf/BloomFilterMapTest.java

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)