| @@ -0,0 +1,202 @@ |
| | 1 | // SPDX-License-Identifier: GPL-2.0+ |
| | 2 | /* |
| | 3 | * Unit test for the clocksource watchdog. |
| | 4 | * |
| | 5 | * Copyright (C) 2021 Facebook, Inc. |
| | 6 | * |
| | 7 | * Author: Paul E. McKenney <paulmck@kernel.org> |
| | 8 | */ |
| | 9 | #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt |
| | 10 | |
| | 11 | #include <linux/device.h> |
| | 12 | #include <linux/clocksource.h> |
| | 13 | #include <linux/init.h> |
| | 14 | #include <linux/module.h> |
| | 15 | #include <linux/sched.h> /* for spin_unlock_irq() using preempt_count() m68k */ |
| | 16 | #include <linux/tick.h> |
| | 17 | #include <linux/kthread.h> |
| | 18 | #include <linux/delay.h> |
| | 19 | #include <linux/prandom.h> |
| | 20 | #include <linux/cpu.h> |
| | 21 | |
| | 22 | MODULE_LICENSE("GPL"); |
| | 23 | MODULE_AUTHOR("Paul E. McKenney <paulmck@kernel.org>"); |
| | 24 | |
| | 25 | static int holdoff = IS_BUILTIN(CONFIG_TEST_CLOCKSOURCE_WATCHDOG) ? 10 : 0; |
| | 26 | module_param(holdoff, int, 0444); |
| | 27 | MODULE_PARM_DESC(holdoff, "Time to wait to start test (s)."); |
| | 28 | |
| | 29 | /* Watchdog kthread's task_struct pointer for debug purposes. */ |
| | 30 | static struct task_struct *wdtest_task; |
| | 31 | |
| | 32 | static u64 wdtest_jiffies_read(struct clocksource *cs) |
| | 33 | { |
| | 34 | return (u64)jiffies; |
| | 35 | } |
| | 36 | |
| | 37 | /* Assume HZ > 100. */ |
| | 38 | #define JIFFIES_SHIFT 8 |
| | 39 | |
| | 40 | static struct clocksource clocksource_wdtest_jiffies = { |
| | 41 | .name = "wdtest-jiffies", |
| | 42 | .rating = 1, /* lowest valid rating*/ |
| | 43 | .uncertainty_margin = TICK_NSEC, |
| | 44 | .read = wdtest_jiffies_read, |
| | 45 | .mask = CLOCKSOURCE_MASK(32), |
| | 46 | .flags = CLOCK_SOURCE_MUST_VERIFY, |
| | 47 | .mult = TICK_NSEC << JIFFIES_SHIFT, /* details above */ |
| | 48 | .shift = JIFFIES_SHIFT, |
| | 49 | .max_cycles = 10, |
| | 50 | }; |
| | 51 | |
| | 52 | static int wdtest_ktime_read_ndelays; |
| | 53 | static bool wdtest_ktime_read_fuzz; |
| | 54 | |
| | 55 | static u64 wdtest_ktime_read(struct clocksource *cs) |
| | 56 | { |
| | 57 | int wkrn = READ_ONCE(wdtest_ktime_read_ndelays); |
| | 58 | static int sign = 1; |
| | 59 | u64 ret; |
| | 60 | |
| | 61 | if (wkrn) { |
| | 62 | udelay(cs->uncertainty_margin / 250); |
| | 63 | WRITE_ONCE(wdtest_ktime_read_ndelays, wkrn - 1); |
| | 64 | } |
| | 65 | ret = ktime_get_real_fast_ns(); |
| | 66 | if (READ_ONCE(wdtest_ktime_read_fuzz)) { |
| | 67 | sign = -sign; |
| | 68 | ret = ret + sign * 100 * NSEC_PER_MSEC; |
| | 69 | } |
| | 70 | return ret; |
| | 71 | } |
| | 72 | |
| | 73 | static void wdtest_ktime_cs_mark_unstable(struct clocksource *cs) |
| | 74 | { |
| | 75 | pr_info("--- Marking %s unstable due to clocksource watchdog.\n", cs->name); |
| | 76 | } |
| | 77 | |
| | 78 | #define KTIME_FLAGS (CLOCK_SOURCE_IS_CONTINUOUS | \ |
| | 79 | CLOCK_SOURCE_VALID_FOR_HRES | \ |
| | 80 | CLOCK_SOURCE_MUST_VERIFY | \ |
| | 81 | CLOCK_SOURCE_VERIFY_PERCPU) |
| | 82 | |
| | 83 | static struct clocksource clocksource_wdtest_ktime = { |
| | 84 | .name = "wdtest-ktime", |
| | 85 | .rating = 300, |
| | 86 | .read = wdtest_ktime_read, |
| | 87 | .mask = CLOCKSOURCE_MASK(64), |
| | 88 | .flags = KTIME_FLAGS, |
| | 89 | .mark_unstable = wdtest_ktime_cs_mark_unstable, |
| | 90 | .list = LIST_HEAD_INIT(clocksource_wdtest_ktime.list), |
| | 91 | }; |
| | 92 | |
| | 93 | /* Reset the clocksource if needed. */ |
| | 94 | static void wdtest_ktime_clocksource_reset(void) |
| | 95 | { |
| | 96 | if (clocksource_wdtest_ktime.flags & CLOCK_SOURCE_UNSTABLE) { |
| | 97 | clocksource_unregister(&clocksource_wdtest_ktime); |
| | 98 | clocksource_wdtest_ktime.flags = KTIME_FLAGS; |
| | 99 | schedule_timeout_uninterruptible(HZ / 10); |
| | 100 | clocksource_register_khz(&clocksource_wdtest_ktime, 1000 * 1000); |
| | 101 | } |
| | 102 | } |
| | 103 | |
| | 104 | /* Run the specified series of watchdog tests. */ |
| | 105 | static int wdtest_func(void *arg) |
| | 106 | { |
| | 107 | unsigned long j1, j2; |
| | 108 | char *s; |
| | 109 | int i; |
| | 110 | |
| | 111 | schedule_timeout_uninterruptible(holdoff * HZ); |
| | 112 | |
| | 113 | /* |
| | 114 | * Verify that jiffies-like clocksources get the manually |
| | 115 | * specified uncertainty margin. |
| | 116 | */ |
| | 117 | pr_info("--- Verify jiffies-like uncertainty margin.\n"); |
| | 118 | __clocksource_register(&clocksource_wdtest_jiffies); |
| | 119 | WARN_ON_ONCE(clocksource_wdtest_jiffies.uncertainty_margin != TICK_NSEC); |
| | 120 | |
| | 121 | j1 = clocksource_wdtest_jiffies.read(&clocksource_wdtest_jiffies); |
| | 122 | schedule_timeout_uninterruptible(HZ); |
| | 123 | j2 = clocksource_wdtest_jiffies.read(&clocksource_wdtest_jiffies); |
| | 124 | WARN_ON_ONCE(j1 == j2); |
| | 125 | |
| | 126 | clocksource_unregister(&clocksource_wdtest_jiffies); |
| | 127 | |
| | 128 | /* |
| | 129 | * Verify that tsc-like clocksources are assigned a reasonable |
| | 130 | * uncertainty margin. |
| | 131 | */ |
| | 132 | pr_info("--- Verify tsc-like uncertainty margin.\n"); |
| | 133 | clocksource_register_khz(&clocksource_wdtest_ktime, 1000 * 1000); |
| | 134 | WARN_ON_ONCE(clocksource_wdtest_ktime.uncertainty_margin < NSEC_PER_USEC); |
| | 135 | |
| | 136 | j1 = clocksource_wdtest_ktime.read(&clocksource_wdtest_ktime); |
| | 137 | udelay(1); |
| | 138 | j2 = clocksource_wdtest_ktime.read(&clocksource_wdtest_ktime); |
| | 139 | pr_info("--- tsc-like times: %lu - %lu = %lu.\n", j2, j1, j2 - j1); |
| | 140 | WARN_ON_ONCE(time_before(j2, j1 + NSEC_PER_USEC)); |
| | 141 | |
| | 142 | /* Verify tsc-like stability with various numbers of errors injected. */ |
| | 143 | for (i = 0; i <= max_cswd_read_retries + 1; i++) { |
| | 144 | if (i <= 1 && i < max_cswd_read_retries) |
| | 145 | s = ""; |
| | 146 | else if (i <= max_cswd_read_retries) |
| | 147 | s = ", expect message"; |
| | 148 | else |
| | 149 | s = ", expect clock skew"; |
| | 150 | pr_info("--- Watchdog with %dx error injection, %lu retries%s.\n", i, max_cswd_read_retries, s); |
| | 151 | WRITE_ONCE(wdtest_ktime_read_ndelays, i); |
| | 152 | schedule_timeout_uninterruptible(2 * HZ); |
| | 153 | WARN_ON_ONCE(READ_ONCE(wdtest_ktime_read_ndelays)); |
| | 154 | WARN_ON_ONCE((i <= max_cswd_read_retries) != |
| | 155 | !(clocksource_wdtest_ktime.flags & CLOCK_SOURCE_UNSTABLE)); |
| | 156 | wdtest_ktime_clocksource_reset(); |
| | 157 | } |
| | 158 | |
| | 159 | /* Verify tsc-like stability with clock-value-fuzz error injection. */ |
| | 160 | pr_info("--- Watchdog clock-value-fuzz error injection, expect clock skew and per-CPU mismatches.\n"); |
| | 161 | WRITE_ONCE(wdtest_ktime_read_fuzz, true); |
| | 162 | schedule_timeout_uninterruptible(2 * HZ); |
| | 163 | WARN_ON_ONCE(!(clocksource_wdtest_ktime.flags & CLOCK_SOURCE_UNSTABLE)); |
| | 164 | clocksource_verify_percpu(&clocksource_wdtest_ktime); |
| | 165 | WRITE_ONCE(wdtest_ktime_read_fuzz, false); |
| | 166 | |
| | 167 | clocksource_unregister(&clocksource_wdtest_ktime); |
| | 168 | |
| | 169 | pr_info("--- Done with test.\n"); |
| | 170 | return 0; |
| | 171 | } |
| | 172 | |
| | 173 | static void wdtest_print_module_parms(void) |
| | 174 | { |
| | 175 | pr_alert("--- holdoff=%d\n", holdoff); |
| | 176 | } |
| | 177 | |
| | 178 | /* Cleanup function. */ |
| | 179 | static void clocksource_wdtest_cleanup(void) |
| | 180 | { |
| | 181 | } |
| | 182 | |
| | 183 | static int __init clocksource_wdtest_init(void) |
| | 184 | { |
| | 185 | int ret = 0; |
| | 186 | |
| | 187 | wdtest_print_module_parms(); |
| | 188 | |
| | 189 | /* Create watchdog-test task. */ |
| | 190 | wdtest_task = kthread_run(wdtest_func, NULL, "wdtest"); |
| | 191 | if (IS_ERR(wdtest_task)) { |
| | 192 | ret = PTR_ERR(wdtest_task); |
| | 193 | pr_warn("%s: Failed to create wdtest kthread.\n", __func__); |
| | 194 | wdtest_task = NULL; |
| | 195 | return ret; |
| | 196 | } |
| | 197 | |
| | 198 | return 0; |
| | 199 | } |
| | 200 | |
| | 201 | module_init(clocksource_wdtest_init); |
| | 202 | module_exit(clocksource_wdtest_cleanup); |