1 // SPDX-License-Identifier: GPL-2.0
2 /*
3 * stop-task scheduling class.
4 *
5 * The stop task is the highest priority task in the system, it preempts
6 * everything and will be preempted by nothing.
7 *
8 * See kernel/stop_machine.c
9 */
10
11 #ifdef CONFIG_SMP
12 static int
select_task_rq_stop(struct task_struct * p,int cpu,int flags)13 select_task_rq_stop(struct task_struct *p, int cpu, int flags)
14 {
15 return task_cpu(p); /* stop tasks as never migrate */
16 }
17
18 static int
balance_stop(struct rq * rq,struct task_struct * prev,struct rq_flags * rf)19 balance_stop(struct rq *rq, struct task_struct *prev, struct rq_flags *rf)
20 {
21 return sched_stop_runnable(rq);
22 }
23 #endif /* CONFIG_SMP */
24
25 static void
wakeup_preempt_stop(struct rq * rq,struct task_struct * p,int flags)26 wakeup_preempt_stop(struct rq *rq, struct task_struct *p, int flags)
27 {
28 /* we're never preempted */
29 }
30
set_next_task_stop(struct rq * rq,struct task_struct * stop,bool first)31 static void set_next_task_stop(struct rq *rq, struct task_struct *stop, bool first)
32 {
33 stop->se.exec_start = rq_clock_task(rq);
34 }
35
pick_task_stop(struct rq * rq)36 static struct task_struct *pick_task_stop(struct rq *rq)
37 {
38 if (!sched_stop_runnable(rq))
39 return NULL;
40
41 return rq->stop;
42 }
43
pick_next_task_stop(struct rq * rq)44 static struct task_struct *pick_next_task_stop(struct rq *rq)
45 {
46 struct task_struct *p = pick_task_stop(rq);
47
48 if (p)
49 set_next_task_stop(rq, p, true);
50
51 return p;
52 }
53
54 static void
enqueue_task_stop(struct rq * rq,struct task_struct * p,int flags)55 enqueue_task_stop(struct rq *rq, struct task_struct *p, int flags)
56 {
57 add_nr_running(rq, 1);
58 }
59
60 static void
dequeue_task_stop(struct rq * rq,struct task_struct * p,int flags)61 dequeue_task_stop(struct rq *rq, struct task_struct *p, int flags)
62 {
63 sub_nr_running(rq, 1);
64 }
65
yield_task_stop(struct rq * rq)66 static void yield_task_stop(struct rq *rq)
67 {
68 BUG(); /* the stop task should never yield, its pointless. */
69 }
70
put_prev_task_stop(struct rq * rq,struct task_struct * prev)71 static void put_prev_task_stop(struct rq *rq, struct task_struct *prev)
72 {
73 update_curr_common(rq);
74 }
75
76 /*
77 * scheduler tick hitting a task of our scheduling class.
78 *
79 * NOTE: This function can be called remotely by the tick offload that
80 * goes along full dynticks. Therefore no local assumption can be made
81 * and everything must be accessed through the @rq and @curr passed in
82 * parameters.
83 */
task_tick_stop(struct rq * rq,struct task_struct * curr,int queued)84 static void task_tick_stop(struct rq *rq, struct task_struct *curr, int queued)
85 {
86 }
87
switched_to_stop(struct rq * rq,struct task_struct * p)88 static void switched_to_stop(struct rq *rq, struct task_struct *p)
89 {
90 BUG(); /* its impossible to change to this class */
91 }
92
93 static void
prio_changed_stop(struct rq * rq,struct task_struct * p,int oldprio)94 prio_changed_stop(struct rq *rq, struct task_struct *p, int oldprio)
95 {
96 BUG(); /* how!?, what priority? */
97 }
98
update_curr_stop(struct rq * rq)99 static void update_curr_stop(struct rq *rq)
100 {
101 }
102
103 /*
104 * Simple, special scheduling class for the per-CPU stop tasks:
105 */
106 DEFINE_SCHED_CLASS(stop) = {
107
108 .enqueue_task = enqueue_task_stop,
109 .dequeue_task = dequeue_task_stop,
110 .yield_task = yield_task_stop,
111
112 .wakeup_preempt = wakeup_preempt_stop,
113
114 .pick_next_task = pick_next_task_stop,
115 .put_prev_task = put_prev_task_stop,
116 .set_next_task = set_next_task_stop,
117
118 #ifdef CONFIG_SMP
119 .balance = balance_stop,
120 .pick_task = pick_task_stop,
121 .select_task_rq = select_task_rq_stop,
122 .set_cpus_allowed = set_cpus_allowed_common,
123 #endif
124
125 .task_tick = task_tick_stop,
126
127 .prio_changed = prio_changed_stop,
128 .switched_to = switched_to_stop,
129 .update_curr = update_curr_stop,
130 };
131