xref: /linux-6.15/kernel/sched/clock.c (revision 5d2a4e91)
1 /*
2  * sched_clock() for unstable CPU clocks
3  *
4  *  Copyright (C) 2008 Red Hat, Inc., Peter Zijlstra
5  *
6  *  Updates and enhancements:
7  *    Copyright (C) 2008 Red Hat, Inc. Steven Rostedt <[email protected]>
8  *
9  * Based on code by:
10  *   Ingo Molnar <[email protected]>
11  *   Guillaume Chazarain <[email protected]>
12  *
13  *
14  * What this file implements:
15  *
16  * cpu_clock(i) provides a fast (execution time) high resolution
17  * clock with bounded drift between CPUs. The value of cpu_clock(i)
18  * is monotonic for constant i. The timestamp returned is in nanoseconds.
19  *
20  * ######################### BIG FAT WARNING ##########################
21  * # when comparing cpu_clock(i) to cpu_clock(j) for i != j, time can #
22  * # go backwards !!                                                  #
23  * ####################################################################
24  *
25  * There is no strict promise about the base, although it tends to start
26  * at 0 on boot (but people really shouldn't rely on that).
27  *
28  * cpu_clock(i)       -- can be used from any context, including NMI.
29  * local_clock()      -- is cpu_clock() on the current CPU.
30  *
31  * sched_clock_cpu(i)
32  *
33  * How it is implemented:
34  *
35  * The implementation either uses sched_clock() when
36  * !CONFIG_HAVE_UNSTABLE_SCHED_CLOCK, which means in that case the
37  * sched_clock() is assumed to provide these properties (mostly it means
38  * the architecture provides a globally synchronized highres time source).
39  *
40  * Otherwise it tries to create a semi stable clock from a mixture of other
41  * clocks, including:
42  *
43  *  - GTOD (clock monotomic)
44  *  - sched_clock()
45  *  - explicit idle events
46  *
47  * We use GTOD as base and use sched_clock() deltas to improve resolution. The
48  * deltas are filtered to provide monotonicity and keeping it within an
49  * expected window.
50  *
51  * Furthermore, explicit sleep and wakeup hooks allow us to account for time
52  * that is otherwise invisible (TSC gets stopped).
53  *
54  */
55 #include "sched.h"
56 #include <linux/sched_clock.h>
57 
58 /*
59  * Scheduler clock - returns current time in nanosec units.
60  * This is default implementation.
61  * Architectures and sub-architectures can override this.
62  */
63 unsigned long long __weak sched_clock(void)
64 {
65 	return (unsigned long long)(jiffies - INITIAL_JIFFIES)
66 					* (NSEC_PER_SEC / HZ);
67 }
68 EXPORT_SYMBOL_GPL(sched_clock);
69 
70 __read_mostly int sched_clock_running;
71 
72 #ifdef CONFIG_HAVE_UNSTABLE_SCHED_CLOCK
73 /*
74  * We must start with !__sched_clock_stable because the unstable -> stable
75  * transition is accurate, while the stable -> unstable transition is not.
76  *
77  * Similarly we start with __sched_clock_stable_early, thereby assuming we
78  * will become stable, such that there's only a single 1 -> 0 transition.
79  */
80 static DEFINE_STATIC_KEY_FALSE(__sched_clock_stable);
81 static int __sched_clock_stable_early = 1;
82 
83 /*
84  * We want: ktime_get_ns() + __gtod_offset == sched_clock() + __sched_clock_offset
85  */
86 __read_mostly u64 __sched_clock_offset;
87 static __read_mostly u64 __gtod_offset;
88 
89 struct sched_clock_data {
90 	u64			tick_raw;
91 	u64			tick_gtod;
92 	u64			clock;
93 };
94 
95 static DEFINE_PER_CPU_SHARED_ALIGNED(struct sched_clock_data, sched_clock_data);
96 
97 static inline struct sched_clock_data *this_scd(void)
98 {
99 	return this_cpu_ptr(&sched_clock_data);
100 }
101 
102 static inline struct sched_clock_data *cpu_sdc(int cpu)
103 {
104 	return &per_cpu(sched_clock_data, cpu);
105 }
106 
107 int sched_clock_stable(void)
108 {
109 	return static_branch_likely(&__sched_clock_stable);
110 }
111 
112 static void __scd_stamp(struct sched_clock_data *scd)
113 {
114 	scd->tick_gtod = ktime_get_ns();
115 	scd->tick_raw = sched_clock();
116 }
117 
118 static void __set_sched_clock_stable(void)
119 {
120 	struct sched_clock_data *scd;
121 
122 	/*
123 	 * Since we're still unstable and the tick is already running, we have
124 	 * to disable IRQs in order to get a consistent scd->tick* reading.
125 	 */
126 	local_irq_disable();
127 	scd = this_scd();
128 	/*
129 	 * Attempt to make the (initial) unstable->stable transition continuous.
130 	 */
131 	__sched_clock_offset = (scd->tick_gtod + __gtod_offset) - (scd->tick_raw);
132 	local_irq_enable();
133 
134 	printk(KERN_INFO "sched_clock: Marking stable (%lld, %lld)->(%lld, %lld)\n",
135 			scd->tick_gtod, __gtod_offset,
136 			scd->tick_raw,  __sched_clock_offset);
137 
138 	static_branch_enable(&__sched_clock_stable);
139 	tick_dep_clear(TICK_DEP_BIT_CLOCK_UNSTABLE);
140 }
141 
142 /*
143  * If we ever get here, we're screwed, because we found out -- typically after
144  * the fact -- that TSC wasn't good. This means all our clocksources (including
145  * ktime) could have reported wrong values.
146  *
147  * What we do here is an attempt to fix up and continue sort of where we left
148  * off in a coherent manner.
149  *
150  * The only way to fully avoid random clock jumps is to boot with:
151  * "tsc=unstable".
152  */
153 static void __sched_clock_work(struct work_struct *work)
154 {
155 	struct sched_clock_data *scd;
156 	int cpu;
157 
158 	/* take a current timestamp and set 'now' */
159 	preempt_disable();
160 	scd = this_scd();
161 	__scd_stamp(scd);
162 	scd->clock = scd->tick_gtod + __gtod_offset;
163 	preempt_enable();
164 
165 	/* clone to all CPUs */
166 	for_each_possible_cpu(cpu)
167 		per_cpu(sched_clock_data, cpu) = *scd;
168 
169 	printk(KERN_WARNING "TSC found unstable after boot, most likely due to broken BIOS. Use 'tsc=unstable'.\n");
170 	printk(KERN_INFO "sched_clock: Marking unstable (%lld, %lld)<-(%lld, %lld)\n",
171 			scd->tick_gtod, __gtod_offset,
172 			scd->tick_raw,  __sched_clock_offset);
173 
174 	static_branch_disable(&__sched_clock_stable);
175 }
176 
177 static DECLARE_WORK(sched_clock_work, __sched_clock_work);
178 
179 static void __clear_sched_clock_stable(void)
180 {
181 	if (!sched_clock_stable())
182 		return;
183 
184 	tick_dep_set(TICK_DEP_BIT_CLOCK_UNSTABLE);
185 	schedule_work(&sched_clock_work);
186 }
187 
188 void clear_sched_clock_stable(void)
189 {
190 	__sched_clock_stable_early = 0;
191 
192 	smp_mb(); /* matches sched_clock_init_late() */
193 
194 	if (sched_clock_running == 2)
195 		__clear_sched_clock_stable();
196 }
197 
198 static void __sched_clock_gtod_offset(void)
199 {
200 	__gtod_offset = (sched_clock() + __sched_clock_offset) - ktime_get_ns();
201 }
202 
203 void __init sched_clock_init(void)
204 {
205 	sched_clock_running = 1;
206 }
207 /*
208  * We run this as late_initcall() such that it runs after all built-in drivers,
209  * notably: acpi_processor and intel_idle, which can mark the TSC as unstable.
210  */
211 static int __init sched_clock_init_late(void)
212 {
213 	sched_clock_running = 2;
214 	/*
215 	 * Ensure that it is impossible to not do a static_key update.
216 	 *
217 	 * Either {set,clear}_sched_clock_stable() must see sched_clock_running
218 	 * and do the update, or we must see their __sched_clock_stable_early
219 	 * and do the update, or both.
220 	 */
221 	smp_mb(); /* matches {set,clear}_sched_clock_stable() */
222 
223 	if (__sched_clock_stable_early)
224 		__set_sched_clock_stable();
225 
226 	return 0;
227 }
228 late_initcall(sched_clock_init_late);
229 
230 /*
231  * min, max except they take wrapping into account
232  */
233 
234 static inline u64 wrap_min(u64 x, u64 y)
235 {
236 	return (s64)(x - y) < 0 ? x : y;
237 }
238 
239 static inline u64 wrap_max(u64 x, u64 y)
240 {
241 	return (s64)(x - y) > 0 ? x : y;
242 }
243 
244 /*
245  * update the percpu scd from the raw @now value
246  *
247  *  - filter out backward motion
248  *  - use the GTOD tick value to create a window to filter crazy TSC values
249  */
250 static u64 sched_clock_local(struct sched_clock_data *scd)
251 {
252 	u64 now, clock, old_clock, min_clock, max_clock, gtod;
253 	s64 delta;
254 
255 again:
256 	now = sched_clock();
257 	delta = now - scd->tick_raw;
258 	if (unlikely(delta < 0))
259 		delta = 0;
260 
261 	old_clock = scd->clock;
262 
263 	/*
264 	 * scd->clock = clamp(scd->tick_gtod + delta,
265 	 *		      max(scd->tick_gtod, scd->clock),
266 	 *		      scd->tick_gtod + TICK_NSEC);
267 	 */
268 
269 	gtod = scd->tick_gtod + __gtod_offset;
270 	clock = gtod + delta;
271 	min_clock = wrap_max(gtod, old_clock);
272 	max_clock = wrap_max(old_clock, gtod + TICK_NSEC);
273 
274 	clock = wrap_max(clock, min_clock);
275 	clock = wrap_min(clock, max_clock);
276 
277 	if (cmpxchg64(&scd->clock, old_clock, clock) != old_clock)
278 		goto again;
279 
280 	return clock;
281 }
282 
283 static u64 sched_clock_remote(struct sched_clock_data *scd)
284 {
285 	struct sched_clock_data *my_scd = this_scd();
286 	u64 this_clock, remote_clock;
287 	u64 *ptr, old_val, val;
288 
289 #if BITS_PER_LONG != 64
290 again:
291 	/*
292 	 * Careful here: The local and the remote clock values need to
293 	 * be read out atomic as we need to compare the values and
294 	 * then update either the local or the remote side. So the
295 	 * cmpxchg64 below only protects one readout.
296 	 *
297 	 * We must reread via sched_clock_local() in the retry case on
298 	 * 32-bit kernels as an NMI could use sched_clock_local() via the
299 	 * tracer and hit between the readout of
300 	 * the low 32-bit and the high 32-bit portion.
301 	 */
302 	this_clock = sched_clock_local(my_scd);
303 	/*
304 	 * We must enforce atomic readout on 32-bit, otherwise the
305 	 * update on the remote CPU can hit inbetween the readout of
306 	 * the low 32-bit and the high 32-bit portion.
307 	 */
308 	remote_clock = cmpxchg64(&scd->clock, 0, 0);
309 #else
310 	/*
311 	 * On 64-bit kernels the read of [my]scd->clock is atomic versus the
312 	 * update, so we can avoid the above 32-bit dance.
313 	 */
314 	sched_clock_local(my_scd);
315 again:
316 	this_clock = my_scd->clock;
317 	remote_clock = scd->clock;
318 #endif
319 
320 	/*
321 	 * Use the opportunity that we have both locks
322 	 * taken to couple the two clocks: we take the
323 	 * larger time as the latest time for both
324 	 * runqueues. (this creates monotonic movement)
325 	 */
326 	if (likely((s64)(remote_clock - this_clock) < 0)) {
327 		ptr = &scd->clock;
328 		old_val = remote_clock;
329 		val = this_clock;
330 	} else {
331 		/*
332 		 * Should be rare, but possible:
333 		 */
334 		ptr = &my_scd->clock;
335 		old_val = this_clock;
336 		val = remote_clock;
337 	}
338 
339 	if (cmpxchg64(ptr, old_val, val) != old_val)
340 		goto again;
341 
342 	return val;
343 }
344 
345 /*
346  * Similar to cpu_clock(), but requires local IRQs to be disabled.
347  *
348  * See cpu_clock().
349  */
350 u64 sched_clock_cpu(int cpu)
351 {
352 	struct sched_clock_data *scd;
353 	u64 clock;
354 
355 	if (sched_clock_stable())
356 		return sched_clock() + __sched_clock_offset;
357 
358 	if (unlikely(!sched_clock_running))
359 		return 0ull;
360 
361 	preempt_disable_notrace();
362 	scd = cpu_sdc(cpu);
363 
364 	if (cpu != smp_processor_id())
365 		clock = sched_clock_remote(scd);
366 	else
367 		clock = sched_clock_local(scd);
368 	preempt_enable_notrace();
369 
370 	return clock;
371 }
372 EXPORT_SYMBOL_GPL(sched_clock_cpu);
373 
374 void sched_clock_tick(void)
375 {
376 	struct sched_clock_data *scd;
377 
378 	if (sched_clock_stable())
379 		return;
380 
381 	if (unlikely(!sched_clock_running))
382 		return;
383 
384 	lockdep_assert_irqs_disabled();
385 
386 	scd = this_scd();
387 	__scd_stamp(scd);
388 	sched_clock_local(scd);
389 }
390 
391 void sched_clock_tick_stable(void)
392 {
393 	if (!sched_clock_stable())
394 		return;
395 
396 	/*
397 	 * Called under watchdog_lock.
398 	 *
399 	 * The watchdog just found this TSC to (still) be stable, so now is a
400 	 * good moment to update our __gtod_offset. Because once we find the
401 	 * TSC to be unstable, any computation will be computing crap.
402 	 */
403 	local_irq_disable();
404 	__sched_clock_gtod_offset();
405 	local_irq_enable();
406 }
407 
408 /*
409  * We are going deep-idle (irqs are disabled):
410  */
411 void sched_clock_idle_sleep_event(void)
412 {
413 	sched_clock_cpu(smp_processor_id());
414 }
415 EXPORT_SYMBOL_GPL(sched_clock_idle_sleep_event);
416 
417 /*
418  * We just idled; resync with ktime.
419  */
420 void sched_clock_idle_wakeup_event(void)
421 {
422 	unsigned long flags;
423 
424 	if (sched_clock_stable())
425 		return;
426 
427 	if (unlikely(timekeeping_suspended))
428 		return;
429 
430 	local_irq_save(flags);
431 	sched_clock_tick();
432 	local_irq_restore(flags);
433 }
434 EXPORT_SYMBOL_GPL(sched_clock_idle_wakeup_event);
435 
436 #else /* CONFIG_HAVE_UNSTABLE_SCHED_CLOCK */
437 
438 void __init sched_clock_init(void)
439 {
440 	sched_clock_running = 1;
441 	generic_sched_clock_init();
442 }
443 
444 u64 sched_clock_cpu(int cpu)
445 {
446 	if (unlikely(!sched_clock_running))
447 		return 0;
448 
449 	return sched_clock();
450 }
451 
452 #endif /* CONFIG_HAVE_UNSTABLE_SCHED_CLOCK */
453 
454 /*
455  * Running clock - returns the time that has elapsed while a guest has been
456  * running.
457  * On a guest this value should be local_clock minus the time the guest was
458  * suspended by the hypervisor (for any reason).
459  * On bare metal this function should return the same as local_clock.
460  * Architectures and sub-architectures can override this.
461  */
462 u64 __weak running_clock(void)
463 {
464 	return local_clock();
465 }
466