| 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 */ |
| 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); |
| 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 | unsigned long get_torture_init_jiffies(void); |
| 108 | bool torture_cleanup_begin(void); |
| 109 | void torture_cleanup_end(void); |
| 110 | bool torture_must_stop(void); |
| 111 | bool torture_must_stop_irq(void); |
| 112 | void torture_kthread_stopping(char *title); |
| 113 | int _torture_create_kthread(int (*fn)(void *arg), void *arg, char *s, char *m, |
| 114 | char *f, struct task_struct **tp, void (*cbf)(struct task_struct *tp)); |
| 115 | void _torture_stop_kthread(char *m, struct task_struct **tp); |
| 116 | |
| 117 | #define torture_create_kthread(n, arg, tp) \ |
| 118 | _torture_create_kthread(n, (arg), #n, "Creating " #n " task", \ |
| 119 | "Failed to create " #n, &(tp), NULL) |
| 120 | #define torture_create_kthread_cb(n, arg, tp, cbf) \ |
| 121 | _torture_create_kthread(n, (arg), #n, "Creating " #n " task", \ |
| 122 | "Failed to create " #n, &(tp), cbf) |
| 123 | #define torture_stop_kthread(n, tp) \ |
| 124 | _torture_stop_kthread("Stopping " #n " task", &(tp)) |
| 125 | |
| 126 | /* Scheduler-related definitions. */ |
| 127 | #ifdef CONFIG_PREEMPTION |
| 128 | #define torture_preempt_schedule() __preempt_schedule() |
| 129 | #else |
| 130 | #define torture_preempt_schedule() do { } while (0) |
| 131 | #endif |
| 132 | |
| 133 | #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) |
| 134 | long torture_sched_setaffinity(pid_t pid, const struct cpumask *in_mask, bool dowarn); |
| 135 | #endif |
| 136 | |
| 137 | #endif /* __LINUX_TORTURE_H */ |