1 /* SPDX-License-Identifier: GPL-2.0 */
2 /*
3 * Copyright (c) 2023 Meta Platforms, Inc. and affiliates.
4 * Copyright (c) 2023 David Vernet <dvernet@meta.com>
5 * Copyright (c) 2023 Tejun Heo <tj@kernel.org>
6 */
7 #include <bpf/bpf.h>
8 #include <scx/common.h>
9 #include <sys/wait.h>
10 #include <unistd.h>
11 #include "enq_select_cpu_fails.bpf.skel.h"
12 #include "scx_test.h"
13
14 static enum scx_test_status setup(void **ctx)
15 {
16 struct enq_select_cpu_fails *skel;
17
18 skel = enq_select_cpu_fails__open_and_load();
19 if (!skel) {
20 SCX_ERR("Failed to open and load skel");
21 return SCX_TEST_FAIL;
22 }
23 *ctx = skel;
24
25 return SCX_TEST_PASS;
26 }
27
28 static enum scx_test_status run(void *ctx)
29 {
30 struct enq_select_cpu_fails *skel = ctx;
31 struct bpf_link *link;
32
33 link = bpf_map__attach_struct_ops(skel->maps.enq_select_cpu_fails_ops);
34 if (!link) {
35 SCX_ERR("Failed to attach scheduler");
36 return SCX_TEST_FAIL;
37 }
38
39 sleep(1);
40
41 bpf_link__destroy(link);
42
43 return SCX_TEST_PASS;
44 }
45
46 static void cleanup(void *ctx)
47 {
48 struct enq_select_cpu_fails *skel = ctx;
49
50 enq_select_cpu_fails__destroy(skel);
51 }
52
53 struct scx_test enq_select_cpu_fails = {
54 .name = "enq_select_cpu_fails",
55 .description = "Verify we fail to call scx_bpf_select_cpu_dfl() "
56 "from ops.enqueue()",
57 .setup = setup,
58 .run = run,
59 .cleanup = cleanup,
60 };
61 REGISTER_SCX_TEST(&enq_select_cpu_fails)
62
Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.