1b915ebe6SJoe Stringer // SPDX-License-Identifier: GPL-2.0
2b915ebe6SJoe Stringer #include <test_progs.h>
3488a23b8SStanislav Fomichev #include <network_helpers.h>
4b915ebe6SJoe Stringer
test_global_data_number(struct bpf_object * obj,__u32 duration)5b915ebe6SJoe Stringer static void test_global_data_number(struct bpf_object *obj, __u32 duration)
6b915ebe6SJoe Stringer {
7b915ebe6SJoe Stringer int i, err, map_fd;
89028bbccSAndrii Nakryiko __u64 num;
9b915ebe6SJoe Stringer
10b915ebe6SJoe Stringer map_fd = bpf_find_map(__func__, obj, "result_number");
11d38835b7SStanislav Fomichev if (CHECK_FAIL(map_fd < 0))
12b915ebe6SJoe Stringer return;
13b915ebe6SJoe Stringer
14b915ebe6SJoe Stringer struct {
15b915ebe6SJoe Stringer char *name;
16b915ebe6SJoe Stringer uint32_t key;
179028bbccSAndrii Nakryiko __u64 num;
18b915ebe6SJoe Stringer } tests[] = {
19b915ebe6SJoe Stringer { "relocate .bss reference", 0, 0 },
20b915ebe6SJoe Stringer { "relocate .data reference", 1, 42 },
21b915ebe6SJoe Stringer { "relocate .rodata reference", 2, 24 },
22b915ebe6SJoe Stringer { "relocate .bss reference", 3, 0 },
23b915ebe6SJoe Stringer { "relocate .data reference", 4, 0xffeeff },
24b915ebe6SJoe Stringer { "relocate .rodata reference", 5, 0xabab },
25b915ebe6SJoe Stringer { "relocate .bss reference", 6, 1234 },
26b915ebe6SJoe Stringer { "relocate .bss reference", 7, 0 },
27b915ebe6SJoe Stringer { "relocate .rodata reference", 8, 0xab },
28b915ebe6SJoe Stringer { "relocate .rodata reference", 9, 0x1111111111111111 },
29b915ebe6SJoe Stringer { "relocate .rodata reference", 10, ~0 },
30b915ebe6SJoe Stringer };
31b915ebe6SJoe Stringer
32f98d6dd1SGuo Zhengkui for (i = 0; i < ARRAY_SIZE(tests); i++) {
33b915ebe6SJoe Stringer err = bpf_map_lookup_elem(map_fd, &tests[i].key, &num);
34b915ebe6SJoe Stringer CHECK(err || num != tests[i].num, tests[i].name,
359028bbccSAndrii Nakryiko "err %d result %llx expected %llx\n",
36b915ebe6SJoe Stringer err, num, tests[i].num);
37b915ebe6SJoe Stringer }
38b915ebe6SJoe Stringer }
39b915ebe6SJoe Stringer
test_global_data_string(struct bpf_object * obj,__u32 duration)40b915ebe6SJoe Stringer static void test_global_data_string(struct bpf_object *obj, __u32 duration)
41b915ebe6SJoe Stringer {
42b915ebe6SJoe Stringer int i, err, map_fd;
43b915ebe6SJoe Stringer char str[32];
44b915ebe6SJoe Stringer
45b915ebe6SJoe Stringer map_fd = bpf_find_map(__func__, obj, "result_string");
46d38835b7SStanislav Fomichev if (CHECK_FAIL(map_fd < 0))
47b915ebe6SJoe Stringer return;
48b915ebe6SJoe Stringer
49b915ebe6SJoe Stringer struct {
50b915ebe6SJoe Stringer char *name;
51b915ebe6SJoe Stringer uint32_t key;
52b915ebe6SJoe Stringer char str[32];
53b915ebe6SJoe Stringer } tests[] = {
54b915ebe6SJoe Stringer { "relocate .rodata reference", 0, "abcdefghijklmnopqrstuvwxyz" },
55b915ebe6SJoe Stringer { "relocate .data reference", 1, "abcdefghijklmnopqrstuvwxyz" },
56b915ebe6SJoe Stringer { "relocate .bss reference", 2, "" },
57b915ebe6SJoe Stringer { "relocate .data reference", 3, "abcdexghijklmnopqrstuvwxyz" },
58b915ebe6SJoe Stringer { "relocate .bss reference", 4, "\0\0hello" },
59b915ebe6SJoe Stringer };
60b915ebe6SJoe Stringer
61f98d6dd1SGuo Zhengkui for (i = 0; i < ARRAY_SIZE(tests); i++) {
62b915ebe6SJoe Stringer err = bpf_map_lookup_elem(map_fd, &tests[i].key, str);
63b915ebe6SJoe Stringer CHECK(err || memcmp(str, tests[i].str, sizeof(str)),
64b915ebe6SJoe Stringer tests[i].name, "err %d result \'%s\' expected \'%s\'\n",
65b915ebe6SJoe Stringer err, str, tests[i].str);
66b915ebe6SJoe Stringer }
67b915ebe6SJoe Stringer }
68b915ebe6SJoe Stringer
69b915ebe6SJoe Stringer struct foo {
70b915ebe6SJoe Stringer __u8 a;
71b915ebe6SJoe Stringer __u32 b;
72b915ebe6SJoe Stringer __u64 c;
73b915ebe6SJoe Stringer };
74b915ebe6SJoe Stringer
test_global_data_struct(struct bpf_object * obj,__u32 duration)75b915ebe6SJoe Stringer static void test_global_data_struct(struct bpf_object *obj, __u32 duration)
76b915ebe6SJoe Stringer {
77b915ebe6SJoe Stringer int i, err, map_fd;
78b915ebe6SJoe Stringer struct foo val;
79b915ebe6SJoe Stringer
80b915ebe6SJoe Stringer map_fd = bpf_find_map(__func__, obj, "result_struct");
81d38835b7SStanislav Fomichev if (CHECK_FAIL(map_fd < 0))
82b915ebe6SJoe Stringer return;
83b915ebe6SJoe Stringer
84b915ebe6SJoe Stringer struct {
85b915ebe6SJoe Stringer char *name;
86b915ebe6SJoe Stringer uint32_t key;
87b915ebe6SJoe Stringer struct foo val;
88b915ebe6SJoe Stringer } tests[] = {
89b915ebe6SJoe Stringer { "relocate .rodata reference", 0, { 42, 0xfefeefef, 0x1111111111111111ULL, } },
90b915ebe6SJoe Stringer { "relocate .bss reference", 1, { } },
91b915ebe6SJoe Stringer { "relocate .rodata reference", 2, { } },
92b915ebe6SJoe Stringer { "relocate .data reference", 3, { 41, 0xeeeeefef, 0x2111111111111111ULL, } },
93b915ebe6SJoe Stringer };
94b915ebe6SJoe Stringer
95f98d6dd1SGuo Zhengkui for (i = 0; i < ARRAY_SIZE(tests); i++) {
96b915ebe6SJoe Stringer err = bpf_map_lookup_elem(map_fd, &tests[i].key, &val);
97b915ebe6SJoe Stringer CHECK(err || memcmp(&val, &tests[i].val, sizeof(val)),
98b915ebe6SJoe Stringer tests[i].name, "err %d result { %u, %u, %llu } expected { %u, %u, %llu }\n",
99b915ebe6SJoe Stringer err, val.a, val.b, val.c, tests[i].val.a, tests[i].val.b, tests[i].val.c);
100b915ebe6SJoe Stringer }
101b915ebe6SJoe Stringer }
102b915ebe6SJoe Stringer
test_global_data_rdonly(struct bpf_object * obj,__u32 duration)103b915ebe6SJoe Stringer static void test_global_data_rdonly(struct bpf_object *obj, __u32 duration)
104b915ebe6SJoe Stringer {
105b915ebe6SJoe Stringer int err = -ENOMEM, map_fd, zero = 0;
1064f2511e1SAndrii Nakryiko struct bpf_map *map, *map2;
107b915ebe6SJoe Stringer __u8 *buff;
108b915ebe6SJoe Stringer
109b915ebe6SJoe Stringer map = bpf_object__find_map_by_name(obj, "test_glo.rodata");
1104f2511e1SAndrii Nakryiko if (!ASSERT_OK_PTR(map, "map"))
1114f2511e1SAndrii Nakryiko return;
1124f2511e1SAndrii Nakryiko if (!ASSERT_TRUE(bpf_map__is_internal(map), "is_internal"))
1134f2511e1SAndrii Nakryiko return;
1144f2511e1SAndrii Nakryiko
1154f2511e1SAndrii Nakryiko /* ensure we can lookup internal maps by their ELF names */
1164f2511e1SAndrii Nakryiko map2 = bpf_object__find_map_by_name(obj, ".rodata");
1174f2511e1SAndrii Nakryiko if (!ASSERT_EQ(map, map2, "same_maps"))
118b915ebe6SJoe Stringer return;
119b915ebe6SJoe Stringer
120b915ebe6SJoe Stringer map_fd = bpf_map__fd(map);
121d38835b7SStanislav Fomichev if (CHECK_FAIL(map_fd < 0))
122b915ebe6SJoe Stringer return;
123b915ebe6SJoe Stringer
1248d6fabf1SChristy Lee buff = malloc(bpf_map__value_size(map));
125b915ebe6SJoe Stringer if (buff)
126b915ebe6SJoe Stringer err = bpf_map_update_elem(map_fd, &zero, buff, 0);
127b915ebe6SJoe Stringer free(buff);
128b915ebe6SJoe Stringer CHECK(!err || errno != EPERM, "test .rodata read-only map",
129b915ebe6SJoe Stringer "err %d errno %d\n", err, errno);
130b915ebe6SJoe Stringer }
131b915ebe6SJoe Stringer
test_global_data(void)132b915ebe6SJoe Stringer void test_global_data(void)
133b915ebe6SJoe Stringer {
134*afef88e6SDaniel Müller const char *file = "./test_global_data.bpf.o";
135b915ebe6SJoe Stringer struct bpf_object *obj;
136b915ebe6SJoe Stringer int err, prog_fd;
13704fcb5f9SDelyan Kratunov LIBBPF_OPTS(bpf_test_run_opts, topts,
13804fcb5f9SDelyan Kratunov .data_in = &pkt_v4,
13904fcb5f9SDelyan Kratunov .data_size_in = sizeof(pkt_v4),
14004fcb5f9SDelyan Kratunov .repeat = 1,
14104fcb5f9SDelyan Kratunov );
142b915ebe6SJoe Stringer
143cbdb1461SAndrii Nakryiko err = bpf_prog_test_load(file, BPF_PROG_TYPE_SCHED_CLS, &obj, &prog_fd);
14404fcb5f9SDelyan Kratunov if (!ASSERT_OK(err, "load program"))
145b915ebe6SJoe Stringer return;
146b915ebe6SJoe Stringer
14704fcb5f9SDelyan Kratunov err = bpf_prog_test_run_opts(prog_fd, &topts);
14804fcb5f9SDelyan Kratunov ASSERT_OK(err, "pass global data run err");
14904fcb5f9SDelyan Kratunov ASSERT_OK(topts.retval, "pass global data run retval");
150b915ebe6SJoe Stringer
15104fcb5f9SDelyan Kratunov test_global_data_number(obj, topts.duration);
15204fcb5f9SDelyan Kratunov test_global_data_string(obj, topts.duration);
15304fcb5f9SDelyan Kratunov test_global_data_struct(obj, topts.duration);
15404fcb5f9SDelyan Kratunov test_global_data_rdonly(obj, topts.duration);
155b915ebe6SJoe Stringer
156b915ebe6SJoe Stringer bpf_object__close(obj);
157b915ebe6SJoe Stringer }
158