xref: /linux/tools/testing/selftests/sched_ext/maximal.bpf.c (revision 643e2e259c2b25a2af0ae4c23c6e16586d9fd19c)
1 /* SPDX-License-Identifier: GPL-2.0 */
2 /*
3  * A scheduler with every callback defined.
4  *
5  * This scheduler defines every callback.
6  *
7  * Copyright (c) 2024 Meta Platforms, Inc. and affiliates.
8  * Copyright (c) 2024 David Vernet <dvernet@meta.com>
9  */
10 
11 #include <scx/common.bpf.h>
12 
13 char _license[] SEC("license") = "GPL";
14 
15 #define DSQ_ID 0
16 
17 s32 BPF_STRUCT_OPS(maximal_select_cpu, struct task_struct *p, s32 prev_cpu,
18 		   u64 wake_flags)
19 {
20 	return prev_cpu;
21 }
22 
23 void BPF_STRUCT_OPS(maximal_enqueue, struct task_struct *p, u64 enq_flags)
24 {
25 	scx_bpf_dsq_insert(p, DSQ_ID, SCX_SLICE_DFL, enq_flags);
26 }
27 
28 void BPF_STRUCT_OPS(maximal_dequeue, struct task_struct *p, u64 deq_flags)
29 {}
30 
31 void BPF_STRUCT_OPS(maximal_dispatch, s32 cpu, struct task_struct *prev)
32 {
33 	scx_bpf_dsq_move_to_local(DSQ_ID);
34 }
35 
36 void BPF_STRUCT_OPS(maximal_runnable, struct task_struct *p, u64 enq_flags)
37 {}
38 
39 void BPF_STRUCT_OPS(maximal_running, struct task_struct *p)
40 {}
41 
42 void BPF_STRUCT_OPS(maximal_stopping, struct task_struct *p, bool runnable)
43 {}
44 
45 void BPF_STRUCT_OPS(maximal_quiescent, struct task_struct *p, u64 deq_flags)
46 {}
47 
48 bool BPF_STRUCT_OPS(maximal_yield, struct task_struct *from,
49 		    struct task_struct *to)
50 {
51 	return false;
52 }
53 
54 bool BPF_STRUCT_OPS(maximal_core_sched_before, struct task_struct *a,
55 		    struct task_struct *b)
56 {
57 	return false;
58 }
59 
60 void BPF_STRUCT_OPS(maximal_set_weight, struct task_struct *p, u32 weight)
61 {}
62 
63 void BPF_STRUCT_OPS(maximal_set_cpumask, struct task_struct *p,
64 		    const struct cpumask *cpumask)
65 {}
66 
67 void BPF_STRUCT_OPS(maximal_update_idle, s32 cpu, bool idle)
68 {}
69 
70 void BPF_STRUCT_OPS(maximal_cpu_acquire, s32 cpu,
71 		    struct scx_cpu_acquire_args *args)
72 {}
73 
74 void BPF_STRUCT_OPS(maximal_cpu_release, s32 cpu,
75 		    struct scx_cpu_release_args *args)
76 {}
77 
78 void BPF_STRUCT_OPS(maximal_cpu_online, s32 cpu)
79 {}
80 
81 void BPF_STRUCT_OPS(maximal_cpu_offline, s32 cpu)
82 {}
83 
84 s32 BPF_STRUCT_OPS(maximal_init_task, struct task_struct *p,
85 		   struct scx_init_task_args *args)
86 {
87 	return 0;
88 }
89 
90 void BPF_STRUCT_OPS(maximal_enable, struct task_struct *p)
91 {}
92 
93 void BPF_STRUCT_OPS(maximal_exit_task, struct task_struct *p,
94 		    struct scx_exit_task_args *args)
95 {}
96 
97 void BPF_STRUCT_OPS(maximal_disable, struct task_struct *p)
98 {}
99 
100 s32 BPF_STRUCT_OPS(maximal_cgroup_init, struct cgroup *cgrp,
101 		   struct scx_cgroup_init_args *args)
102 {
103 	return 0;
104 }
105 
106 void BPF_STRUCT_OPS(maximal_cgroup_exit, struct cgroup *cgrp)
107 {}
108 
109 s32 BPF_STRUCT_OPS(maximal_cgroup_prep_move, struct task_struct *p,
110 		   struct cgroup *from, struct cgroup *to)
111 {
112 	return 0;
113 }
114 
115 void BPF_STRUCT_OPS(maximal_cgroup_move, struct task_struct *p,
116 		    struct cgroup *from, struct cgroup *to)
117 {}
118 
119 void BPF_STRUCT_OPS(maximal_cgroup_cancel_move, struct task_struct *p,
120 	       struct cgroup *from, struct cgroup *to)
121 {}
122 
123 void BPF_STRUCT_OPS(maximal_cgroup_set_weight, struct cgroup *cgrp, u32 weight)
124 {}
125 
126 s32 BPF_STRUCT_OPS_SLEEPABLE(maximal_init)
127 {
128 	return scx_bpf_create_dsq(DSQ_ID, -1);
129 }
130 
131 void BPF_STRUCT_OPS(maximal_exit, struct scx_exit_info *info)
132 {}
133 
134 SEC(".struct_ops.link")
135 struct sched_ext_ops maximal_ops = {
136 	.select_cpu		= (void *) maximal_select_cpu,
137 	.enqueue		= (void *) maximal_enqueue,
138 	.dequeue		= (void *) maximal_dequeue,
139 	.dispatch		= (void *) maximal_dispatch,
140 	.runnable		= (void *) maximal_runnable,
141 	.running		= (void *) maximal_running,
142 	.stopping		= (void *) maximal_stopping,
143 	.quiescent		= (void *) maximal_quiescent,
144 	.yield			= (void *) maximal_yield,
145 	.core_sched_before	= (void *) maximal_core_sched_before,
146 	.set_weight		= (void *) maximal_set_weight,
147 	.set_cpumask		= (void *) maximal_set_cpumask,
148 	.update_idle		= (void *) maximal_update_idle,
149 	.cpu_acquire		= (void *) maximal_cpu_acquire,
150 	.cpu_release		= (void *) maximal_cpu_release,
151 	.cpu_online		= (void *) maximal_cpu_online,
152 	.cpu_offline		= (void *) maximal_cpu_offline,
153 	.init_task		= (void *) maximal_init_task,
154 	.enable			= (void *) maximal_enable,
155 	.exit_task		= (void *) maximal_exit_task,
156 	.disable		= (void *) maximal_disable,
157 	.cgroup_init		= (void *) maximal_cgroup_init,
158 	.cgroup_exit		= (void *) maximal_cgroup_exit,
159 	.cgroup_prep_move	= (void *) maximal_cgroup_prep_move,
160 	.cgroup_move		= (void *) maximal_cgroup_move,
161 	.cgroup_cancel_move	= (void *) maximal_cgroup_cancel_move,
162 	.cgroup_set_weight	= (void *) maximal_cgroup_set_weight,
163 	.init			= (void *) maximal_init,
164 	.exit			= (void *) maximal_exit,
165 	.name			= "maximal",
166 };
167