xref: /linux/include/linux/torture.h (revision a1ff5a7d78a036d6c2178ee5acd6ba4946243800)
1 /* SPDX-License-Identifier: GPL-2.0+ */
2 /*
3  * Common functions for in-kernel torture tests.
4  *
5  * Copyright IBM Corporation, 2014
6  *
7  * Author: Paul E. McKenney <paulmck@linux.ibm.com>
8  */
9 
10 #ifndef __LINUX_TORTURE_H
11 #define __LINUX_TORTURE_H
12 
13 #include <linux/types.h>
14 #include <linux/cache.h>
15 #include <linux/spinlock.h>
16 #include <linux/threads.h>
17 #include <linux/cpumask_types.h>
18 #include <linux/seqlock.h>
19 #include <linux/lockdep.h>
20 #include <linux/completion.h>
21 #include <linux/debugobjects.h>
22 #include <linux/bug.h>
23 #include <linux/compiler.h>
24 #include <linux/hrtimer.h>
25 
26 /* Definitions for a non-string torture-test module parameter. */
27 #define torture_param(type, name, init, msg) \
28 	static type name = init; \
29 	module_param(name, type, 0444); \
30 	MODULE_PARM_DESC(name, msg);
31 
32 #define TORTURE_FLAG "-torture:"
33 #define TOROUT_STRING(s) \
34 	pr_alert("%s" TORTURE_FLAG " %s\n", torture_type, s)
35 #define VERBOSE_TOROUT_STRING(s) \
36 do {										\
37 	if (verbose) {								\
38 		verbose_torout_sleep();						\
39 		pr_alert("%s" TORTURE_FLAG " %s\n", torture_type, s);		\
40 	}									\
41 } while (0)
42 #define TOROUT_ERRSTRING(s) \
43 	pr_alert("%s" TORTURE_FLAG "!!! %s\n", torture_type, s)
44 void verbose_torout_sleep(void);
45 
46 #define torture_init_error(firsterr)						\
47 ({										\
48 	int ___firsterr = (firsterr);						\
49 										\
50 	WARN_ONCE(!IS_MODULE(CONFIG_RCU_TORTURE_TEST) && ___firsterr < 0, "Torture-test initialization failed with error code %d\n", ___firsterr); \
51 	___firsterr < 0;								\
52 })
53 
54 /* Definitions for online/offline exerciser. */
55 #ifdef CONFIG_HOTPLUG_CPU
56 int torture_num_online_cpus(void);
57 #else /* #ifdef CONFIG_HOTPLUG_CPU */
torture_num_online_cpus(void)58 static inline int torture_num_online_cpus(void) { return 1; }
59 #endif /* #else #ifdef CONFIG_HOTPLUG_CPU */
60 typedef void torture_ofl_func(void);
61 bool torture_offline(int cpu, long *n_onl_attempts, long *n_onl_successes,
62 		     unsigned long *sum_offl, int *min_onl, int *max_onl);
63 bool torture_online(int cpu, long *n_onl_attempts, long *n_onl_successes,
64 		    unsigned long *sum_onl, int *min_onl, int *max_onl);
65 int torture_onoff_init(long ooholdoff, long oointerval, torture_ofl_func *f);
66 void torture_onoff_stats(void);
67 bool torture_onoff_failures(void);
68 
69 /* Low-rider random number generator. */
70 struct torture_random_state {
71 	unsigned long trs_state;
72 	long trs_count;
73 };
74 #define DEFINE_TORTURE_RANDOM(name) struct torture_random_state name = { 0, 0 }
75 #define DEFINE_TORTURE_RANDOM_PERCPU(name) \
76 	DEFINE_PER_CPU(struct torture_random_state, name)
77 unsigned long torture_random(struct torture_random_state *trsp);
torture_random_init(struct torture_random_state * trsp)78 static inline void torture_random_init(struct torture_random_state *trsp)
79 {
80 	trsp->trs_state = 0;
81 	trsp->trs_count = 0;
82 }
83 
84 /* Definitions for high-resolution-timer sleeps. */
85 int torture_hrtimeout_ns(ktime_t baset_ns, u32 fuzzt_ns, const enum hrtimer_mode mode,
86 			 struct torture_random_state *trsp);
87 int torture_hrtimeout_us(u32 baset_us, u32 fuzzt_ns, struct torture_random_state *trsp);
88 int torture_hrtimeout_ms(u32 baset_ms, u32 fuzzt_us, struct torture_random_state *trsp);
89 int torture_hrtimeout_jiffies(u32 baset_j, struct torture_random_state *trsp);
90 int torture_hrtimeout_s(u32 baset_s, u32 fuzzt_ms, struct torture_random_state *trsp);
91 
92 /* Task shuffler, which causes CPUs to occasionally go idle. */
93 void torture_shuffle_task_register(struct task_struct *tp);
94 int torture_shuffle_init(long shuffint);
95 
96 /* Test auto-shutdown handling. */
97 void torture_shutdown_absorb(const char *title);
98 int torture_shutdown_init(int ssecs, void (*cleanup)(void));
99 
100 /* Task stuttering, which forces load/no-load transitions. */
101 bool stutter_wait(const char *title);
102 int torture_stutter_init(int s, int sgap);
103 
104 /* Initialization and cleanup. */
105 bool torture_init_begin(char *ttype, int v);
106 void torture_init_end(void);
107 bool torture_cleanup_begin(void);
108 void torture_cleanup_end(void);
109 bool torture_must_stop(void);
110 bool torture_must_stop_irq(void);
111 void torture_kthread_stopping(char *title);
112 int _torture_create_kthread(int (*fn)(void *arg), void *arg, char *s, char *m,
113 			     char *f, struct task_struct **tp, void (*cbf)(struct task_struct *tp));
114 void _torture_stop_kthread(char *m, struct task_struct **tp);
115 
116 #define torture_create_kthread(n, arg, tp) \
117 	_torture_create_kthread(n, (arg), #n, "Creating " #n " task", \
118 				"Failed to create " #n, &(tp), NULL)
119 #define torture_create_kthread_cb(n, arg, tp, cbf) \
120 	_torture_create_kthread(n, (arg), #n, "Creating " #n " task", \
121 				"Failed to create " #n, &(tp), cbf)
122 #define torture_stop_kthread(n, tp) \
123 	_torture_stop_kthread("Stopping " #n " task", &(tp))
124 
125 /* Scheduler-related definitions. */
126 #ifdef CONFIG_PREEMPTION
127 #define torture_preempt_schedule() __preempt_schedule()
128 #else
129 #define torture_preempt_schedule()	do { } while (0)
130 #endif
131 
132 #if IS_ENABLED(CONFIG_RCU_TORTURE_TEST) || IS_MODULE(CONFIG_RCU_TORTURE_TEST) || IS_ENABLED(CONFIG_LOCK_TORTURE_TEST) || IS_MODULE(CONFIG_LOCK_TORTURE_TEST)
133 long torture_sched_setaffinity(pid_t pid, const struct cpumask *in_mask);
134 #endif
135 
136 #endif /* __LINUX_TORTURE_H */
137