[2/2] selftests/bpf: Add a test for the type checking of iter args

Commit Message

Tao Lyu Nov. 7, 2024, 9:47 p.m. UTC

This is a selftest for the type checking of iter arguments.
When passing a PTR_TO_MAP_VALUE to bpf_iter_num_* kfuncs,
the program should be rejected.

Signed-off-by: Tao Lyu <[e-mail]>
---
 tools/testing/selftests/bpf/progs/iters.c | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

Patch

diff --git a/tools/testing/selftests/bpf/progs/iters.c b/tools/testing/selftests/bpf/progs/iters.c
index ef70b88bccb2..613037ececb9 100644
--- a/tools/testing/selftests/bpf/progs/iters.c
+++ b/tools/testing/selftests/bpf/progs/iters.c
@@ -1486,4 +1486,24 @@  int iter_subprog_check_stacksafe(const void *ctx)
 	return 0;
 }
 
+SEC("raw_tp")
+__failure __msg("arg#0 expected pointer to the iterator")
+int iter_check_arg_type(const void *ctx)
+{
+	struct bpf_iter_num it;
+
+	int *map_val = NULL;
+	int key = 0;
+
+	map_val = bpf_map_lookup_elem(&arr_map, &key);
+	if (!map_val)
+		return 0;
+
+	bpf_iter_num_new(&it, 0, 3);
+	while (bpf_iter_num_next((struct bpf_iter_num *)map_val))
+	bpf_iter_num_destroy(&it);
+
+	return 0;
+}
+
 char _license[] SEC("license") = "GPL";