1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (c) 2023 Meta Platforms, Inc. and affiliates. */
5 #include <sys/syscall.h>
8 #include "test_map_ops.skel.h"
9 #include "test_progs.h"
11 static void map_update(void)
13 (void)syscall(__NR_getpid);
16 static void map_delete(void)
18 (void)syscall(__NR_getppid);
21 static void map_push(void)
23 (void)syscall(__NR_getuid);
26 static void map_pop(void)
28 (void)syscall(__NR_geteuid);
31 static void map_peek(void)
33 (void)syscall(__NR_getgid);
36 static void map_for_each_pass(void)
38 (void)syscall(__NR_gettid);
41 static void map_for_each_fail(void)
43 (void)syscall(__NR_getpgid);
46 static int setup(struct test_map_ops **skel)
53 *skel = test_map_ops__open();
54 if (!ASSERT_OK_PTR(*skel, "test_map_ops__open"))
57 (*skel)->rodata->pid = getpid();
59 err = test_map_ops__load(*skel);
60 if (!ASSERT_OK(err, "test_map_ops__load"))
63 err = test_map_ops__attach(*skel);
64 if (!ASSERT_OK(err, "test_map_ops__attach"))
70 static void teardown(struct test_map_ops **skel)
73 test_map_ops__destroy(*skel);
76 static void map_ops_update_delete_subtest(void)
78 struct test_map_ops *skel;
84 ASSERT_OK(skel->bss->err, "map_update_initial");
87 ASSERT_LT(skel->bss->err, 0, "map_update_existing");
88 ASSERT_EQ(skel->bss->err, -EEXIST, "map_update_existing");
91 ASSERT_OK(skel->bss->err, "map_delete_existing");
94 ASSERT_LT(skel->bss->err, 0, "map_delete_non_existing");
95 ASSERT_EQ(skel->bss->err, -ENOENT, "map_delete_non_existing");
101 static void map_ops_push_peek_pop_subtest(void)
103 struct test_map_ops *skel;
109 ASSERT_OK(skel->bss->err, "map_push_initial");
112 ASSERT_LT(skel->bss->err, 0, "map_push_when_full");
113 ASSERT_EQ(skel->bss->err, -E2BIG, "map_push_when_full");
116 ASSERT_OK(skel->bss->err, "map_peek");
119 ASSERT_OK(skel->bss->err, "map_pop");
122 ASSERT_LT(skel->bss->err, 0, "map_peek_when_empty");
123 ASSERT_EQ(skel->bss->err, -ENOENT, "map_peek_when_empty");
126 ASSERT_LT(skel->bss->err, 0, "map_pop_when_empty");
127 ASSERT_EQ(skel->bss->err, -ENOENT, "map_pop_when_empty");
133 static void map_ops_for_each_subtest(void)
135 struct test_map_ops *skel;
141 /* expect to iterate over 1 element */
142 ASSERT_EQ(skel->bss->err, 1, "map_for_each_no_flags");
145 ASSERT_LT(skel->bss->err, 0, "map_for_each_with_flags");
146 ASSERT_EQ(skel->bss->err, -EINVAL, "map_for_each_with_flags");
152 void test_map_ops(void)
154 if (test__start_subtest("map_ops_update_delete"))
155 map_ops_update_delete_subtest();
157 if (test__start_subtest("map_ops_push_peek_pop"))
158 map_ops_push_peek_pop_subtest();
160 if (test__start_subtest("map_ops_for_each"))
161 map_ops_for_each_subtest();