1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (c) 2021 Facebook */
3 #include <test_progs.h>
4 #include "test_legacy_printk.skel.h"
6 static int execute_one_variant(bool legacy)
8 struct test_legacy_printk *skel;
9 int err, zero = 0, my_pid = getpid(), res, map_fd;
11 skel = test_legacy_printk__open();
12 if (!ASSERT_OK_PTR(skel, "skel_open"))
15 bpf_program__set_autoload(skel->progs.handle_legacy, legacy);
16 bpf_program__set_autoload(skel->progs.handle_modern, !legacy);
18 err = test_legacy_printk__load(skel);
19 /* no ASSERT_OK, we expect one of two variants can fail here */
24 map_fd = bpf_map__fd(skel->maps.my_pid_map);
25 err = bpf_map_update_elem(map_fd, &zero, &my_pid, BPF_ANY);
26 if (!ASSERT_OK(err, "my_pid_map_update"))
28 err = bpf_map_lookup_elem(map_fd, &zero, &res);
30 skel->bss->my_pid_var = my_pid;
33 err = test_legacy_printk__attach(skel);
34 if (!ASSERT_OK(err, "skel_attach"))
37 usleep(1); /* trigger */
40 map_fd = bpf_map__fd(skel->maps.res_map);
41 err = bpf_map_lookup_elem(map_fd, &zero, &res);
42 if (!ASSERT_OK(err, "res_map_lookup"))
45 res = skel->bss->res_var;
48 if (!ASSERT_GT(res, 0, "res")) {
54 test_legacy_printk__destroy(skel);
58 void test_legacy_printk(void)
60 /* legacy variant should work everywhere */
61 ASSERT_OK(execute_one_variant(true /* legacy */), "legacy_case");
63 /* execute modern variant, can fail the load on old kernels */
64 execute_one_variant(false);