1 #ifndef _LINUX_SCHED_ISOLATION_H
2 #define _LINUX_SCHED_ISOLATION_H
3 
4 #include <linux/cpumask.h>
5 #include <linux/init.h>
6 #include <linux/tick.h>
7 
8 enum hk_type {
9 	HK_TYPE_TIMER,
10 	HK_TYPE_RCU,
11 	HK_TYPE_MISC,
12 	HK_TYPE_SCHED,
13 	HK_TYPE_TICK,
14 	HK_TYPE_DOMAIN,
15 	HK_TYPE_WQ,
16 	HK_TYPE_MANAGED_IRQ,
17 	HK_TYPE_KTHREAD,
18 	HK_TYPE_MAX
19 };
20 
21 #ifdef CONFIG_CPU_ISOLATION
22 DECLARE_STATIC_KEY_FALSE(housekeeping_overridden);
23 extern int housekeeping_any_cpu(enum hk_type type);
24 extern const struct cpumask *housekeeping_cpumask(enum hk_type type);
25 extern bool housekeeping_enabled(enum hk_type type);
26 extern void housekeeping_affine(struct task_struct *t, enum hk_type type);
27 extern bool housekeeping_test_cpu(int cpu, enum hk_type type);
28 extern void __init housekeeping_init(void);
29 
30 #else
31 
housekeeping_any_cpu(enum hk_type type)32 static inline int housekeeping_any_cpu(enum hk_type type)
33 {
34 	return smp_processor_id();
35 }
36 
housekeeping_cpumask(enum hk_type type)37 static inline const struct cpumask *housekeeping_cpumask(enum hk_type type)
38 {
39 	return cpu_possible_mask;
40 }
41 
housekeeping_enabled(enum hk_type type)42 static inline bool housekeeping_enabled(enum hk_type type)
43 {
44 	return false;
45 }
46 
housekeeping_affine(struct task_struct * t,enum hk_type type)47 static inline void housekeeping_affine(struct task_struct *t,
48 				       enum hk_type type) { }
49 
housekeeping_test_cpu(int cpu,enum hk_type type)50 static inline bool housekeeping_test_cpu(int cpu, enum hk_type type)
51 {
52 	return true;
53 }
54 
housekeeping_init(void)55 static inline void housekeeping_init(void) { }
56 #endif /* CONFIG_CPU_ISOLATION */
57 
housekeeping_cpu(int cpu,enum hk_type type)58 static inline bool housekeeping_cpu(int cpu, enum hk_type type)
59 {
60 #ifdef CONFIG_CPU_ISOLATION
61 	if (static_branch_unlikely(&housekeeping_overridden))
62 		return housekeeping_test_cpu(cpu, type);
63 #endif
64 	return true;
65 }
66 
cpu_is_isolated(int cpu)67 static inline bool cpu_is_isolated(int cpu)
68 {
69 	return !housekeeping_test_cpu(cpu, HK_TYPE_DOMAIN) ||
70 		 !housekeeping_test_cpu(cpu, HK_TYPE_TICK);
71 }
72 
73 #endif /* _LINUX_SCHED_ISOLATION_H */
74