1b94fa9f9SRoberto Sassu // SPDX-License-Identifier: GPL-2.0
2b94fa9f9SRoberto Sassu
3b94fa9f9SRoberto Sassu /*
4b94fa9f9SRoberto Sassu * Copyright (c) 2022 Facebook
5b94fa9f9SRoberto Sassu * Copyright (C) 2022 Huawei Technologies Duesseldorf GmbH
6b94fa9f9SRoberto Sassu *
7b94fa9f9SRoberto Sassu * Author: Roberto Sassu <roberto.sassu@huawei.com>
8b94fa9f9SRoberto Sassu */
9b94fa9f9SRoberto Sassu
10b94fa9f9SRoberto Sassu #include <test_progs.h>
11b94fa9f9SRoberto Sassu #include "test_kfunc_dynptr_param.skel.h"
12b94fa9f9SRoberto Sassu
13b94fa9f9SRoberto Sassu static struct {
14b94fa9f9SRoberto Sassu const char *prog_name;
15b94fa9f9SRoberto Sassu int expected_runtime_err;
16b94fa9f9SRoberto Sassu } kfunc_dynptr_tests[] = {
17*8032cad1SJoanne Koong {"dynptr_data_null", -EBADMSG},
18b94fa9f9SRoberto Sassu };
19b94fa9f9SRoberto Sassu
20b94fa9f9SRoberto Sassu static bool kfunc_not_supported;
21b94fa9f9SRoberto Sassu
libbpf_print_cb(enum libbpf_print_level level,const char * fmt,va_list args)22b94fa9f9SRoberto Sassu static int libbpf_print_cb(enum libbpf_print_level level, const char *fmt,
23b94fa9f9SRoberto Sassu va_list args)
24b94fa9f9SRoberto Sassu {
25b94fa9f9SRoberto Sassu if (strcmp(fmt, "libbpf: extern (func ksym) '%s': not found in kernel or module BTFs\n"))
26b94fa9f9SRoberto Sassu return 0;
27b94fa9f9SRoberto Sassu
28b94fa9f9SRoberto Sassu if (strcmp(va_arg(args, char *), "bpf_verify_pkcs7_signature"))
29b94fa9f9SRoberto Sassu return 0;
30b94fa9f9SRoberto Sassu
31b94fa9f9SRoberto Sassu kfunc_not_supported = true;
32b94fa9f9SRoberto Sassu return 0;
33b94fa9f9SRoberto Sassu }
34b94fa9f9SRoberto Sassu
has_pkcs7_kfunc_support(void)35*8032cad1SJoanne Koong static bool has_pkcs7_kfunc_support(void)
36b94fa9f9SRoberto Sassu {
37b94fa9f9SRoberto Sassu struct test_kfunc_dynptr_param *skel;
38b94fa9f9SRoberto Sassu libbpf_print_fn_t old_print_cb;
39b94fa9f9SRoberto Sassu int err;
40b94fa9f9SRoberto Sassu
41*8032cad1SJoanne Koong skel = test_kfunc_dynptr_param__open();
42*8032cad1SJoanne Koong if (!ASSERT_OK_PTR(skel, "test_kfunc_dynptr_param__open"))
43*8032cad1SJoanne Koong return false;
44b94fa9f9SRoberto Sassu
45b94fa9f9SRoberto Sassu kfunc_not_supported = false;
46b94fa9f9SRoberto Sassu
47b94fa9f9SRoberto Sassu old_print_cb = libbpf_set_print(libbpf_print_cb);
48b94fa9f9SRoberto Sassu err = test_kfunc_dynptr_param__load(skel);
49b94fa9f9SRoberto Sassu libbpf_set_print(old_print_cb);
50b94fa9f9SRoberto Sassu
51b94fa9f9SRoberto Sassu if (err < 0 && kfunc_not_supported) {
52b94fa9f9SRoberto Sassu fprintf(stderr,
53b94fa9f9SRoberto Sassu "%s:SKIP:bpf_verify_pkcs7_signature() kfunc not supported\n",
54b94fa9f9SRoberto Sassu __func__);
55b94fa9f9SRoberto Sassu test_kfunc_dynptr_param__destroy(skel);
56*8032cad1SJoanne Koong return false;
57*8032cad1SJoanne Koong }
58*8032cad1SJoanne Koong
59*8032cad1SJoanne Koong test_kfunc_dynptr_param__destroy(skel);
60*8032cad1SJoanne Koong
61*8032cad1SJoanne Koong return true;
62b94fa9f9SRoberto Sassu }
63b94fa9f9SRoberto Sassu
verify_success(const char * prog_name,int expected_runtime_err)64b94fa9f9SRoberto Sassu static void verify_success(const char *prog_name, int expected_runtime_err)
65b94fa9f9SRoberto Sassu {
66b94fa9f9SRoberto Sassu struct test_kfunc_dynptr_param *skel;
67b94fa9f9SRoberto Sassu struct bpf_program *prog;
68b94fa9f9SRoberto Sassu struct bpf_link *link;
69b94fa9f9SRoberto Sassu __u32 next_id;
70b94fa9f9SRoberto Sassu int err;
71b94fa9f9SRoberto Sassu
72b94fa9f9SRoberto Sassu skel = test_kfunc_dynptr_param__open();
73b94fa9f9SRoberto Sassu if (!ASSERT_OK_PTR(skel, "test_kfunc_dynptr_param__open"))
74b94fa9f9SRoberto Sassu return;
75b94fa9f9SRoberto Sassu
76b94fa9f9SRoberto Sassu skel->bss->pid = getpid();
77b94fa9f9SRoberto Sassu
78b94fa9f9SRoberto Sassu err = test_kfunc_dynptr_param__load(skel);
79b94fa9f9SRoberto Sassu
80b94fa9f9SRoberto Sassu if (!ASSERT_OK(err, "test_kfunc_dynptr_param__load"))
81b94fa9f9SRoberto Sassu goto cleanup;
82b94fa9f9SRoberto Sassu
83b94fa9f9SRoberto Sassu prog = bpf_object__find_program_by_name(skel->obj, prog_name);
84b94fa9f9SRoberto Sassu if (!ASSERT_OK_PTR(prog, "bpf_object__find_program_by_name"))
85b94fa9f9SRoberto Sassu goto cleanup;
86b94fa9f9SRoberto Sassu
87b94fa9f9SRoberto Sassu link = bpf_program__attach(prog);
88b94fa9f9SRoberto Sassu if (!ASSERT_OK_PTR(link, "bpf_program__attach"))
89b94fa9f9SRoberto Sassu goto cleanup;
90b94fa9f9SRoberto Sassu
91b94fa9f9SRoberto Sassu err = bpf_prog_get_next_id(0, &next_id);
92b94fa9f9SRoberto Sassu
93b94fa9f9SRoberto Sassu bpf_link__destroy(link);
94b94fa9f9SRoberto Sassu
95b94fa9f9SRoberto Sassu if (!ASSERT_OK(err, "bpf_prog_get_next_id"))
96b94fa9f9SRoberto Sassu goto cleanup;
97b94fa9f9SRoberto Sassu
98b94fa9f9SRoberto Sassu ASSERT_EQ(skel->bss->err, expected_runtime_err, "err");
99b94fa9f9SRoberto Sassu
100b94fa9f9SRoberto Sassu cleanup:
101b94fa9f9SRoberto Sassu test_kfunc_dynptr_param__destroy(skel);
102b94fa9f9SRoberto Sassu }
103b94fa9f9SRoberto Sassu
test_kfunc_dynptr_param(void)104b94fa9f9SRoberto Sassu void test_kfunc_dynptr_param(void)
105b94fa9f9SRoberto Sassu {
106b94fa9f9SRoberto Sassu int i;
107b94fa9f9SRoberto Sassu
108*8032cad1SJoanne Koong if (!has_pkcs7_kfunc_support())
109*8032cad1SJoanne Koong return;
110*8032cad1SJoanne Koong
111b94fa9f9SRoberto Sassu for (i = 0; i < ARRAY_SIZE(kfunc_dynptr_tests); i++) {
112b94fa9f9SRoberto Sassu if (!test__start_subtest(kfunc_dynptr_tests[i].prog_name))
113b94fa9f9SRoberto Sassu continue;
114b94fa9f9SRoberto Sassu
115b94fa9f9SRoberto Sassu verify_success(kfunc_dynptr_tests[i].prog_name,
116b94fa9f9SRoberto Sassu kfunc_dynptr_tests[i].expected_runtime_err);
117b94fa9f9SRoberto Sassu }
118*8032cad1SJoanne Koong RUN_TESTS(test_kfunc_dynptr_param);
119b94fa9f9SRoberto Sassu }
120