1 /* SPDX-License-Identifier: GPL-2.0 */
2 /*
3 * Copyright (c) 2024 Meta Platforms, Inc. and affiliates.
4 */
5 #include <bpf/bpf.h>
6 #include <scx/common.h>
7 #include <sys/wait.h>
8 #include <unistd.h>
9 #include "maybe_null.bpf.skel.h"
10 #include "maybe_null_fail_dsp.bpf.skel.h"
11 #include "maybe_null_fail_yld.bpf.skel.h"
12 #include "scx_test.h"
13
14 static enum scx_test_status run(void *ctx)
15 {
16 struct maybe_null *skel;
17 struct maybe_null_fail_dsp *fail_dsp;
18 struct maybe_null_fail_yld *fail_yld;
19
20 skel = maybe_null__open_and_load();
21 if (!skel) {
22 SCX_ERR("Failed to open and load maybe_null skel");
23 return SCX_TEST_FAIL;
24 }
25 maybe_null__destroy(skel);
26
27 fail_dsp = maybe_null_fail_dsp__open_and_load();
28 if (fail_dsp) {
29 maybe_null_fail_dsp__destroy(fail_dsp);
30 SCX_ERR("Should failed to open and load maybe_null_fail_dsp skel");
31 return SCX_TEST_FAIL;
32 }
33
34 fail_yld = maybe_null_fail_yld__open_and_load();
35 if (fail_yld) {
36 maybe_null_fail_yld__destroy(fail_yld);
37 SCX_ERR("Should failed to open and load maybe_null_fail_yld skel");
38 return SCX_TEST_FAIL;
39 }
40
41 return SCX_TEST_PASS;
42 }
43
44 struct scx_test maybe_null = {
45 .name = "maybe_null",
46 .description = "Verify if PTR_MAYBE_NULL work for .dispatch",
47 .run = run,
48 };
49 REGISTER_SCX_TEST(&maybe_null)
50
Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.