2 * arch/x86/kernel/nmi-selftest.c
4 * Testsuite for NMI: IPIs
6 * Started by Don Zickus:
7 * (using lib/locking-selftest.c as a guide)
9 * Copyright (C) 2011 Red Hat, Inc., Don Zickus <dzickus@redhat.com>
12 #include <linux/smp.h>
13 #include <linux/cpumask.h>
14 #include <linux/delay.h>
15 #include <linux/init.h>
24 static int __initdata nmi_fail;
26 /* check to see if NMI IPIs work on this machine */
27 static DECLARE_BITMAP(nmi_ipi_mask, NR_CPUS) __initdata;
29 static int __initdata testcase_total;
30 static int __initdata testcase_successes;
31 static int __initdata expected_testcase_failures;
32 static int __initdata unexpected_testcase_failures;
33 static int __initdata unexpected_testcase_unknowns;
35 static int __init nmi_unk_cb(unsigned int val, struct pt_regs *regs)
37 unexpected_testcase_unknowns++;
41 static void __init init_nmi_testsuite(void)
43 /* trap all the unknown NMIs we may generate */
44 register_nmi_handler(NMI_UNKNOWN, nmi_unk_cb, 0, "nmi_selftest_unk");
47 static void __init cleanup_nmi_testsuite(void)
49 unregister_nmi_handler(NMI_UNKNOWN, "nmi_selftest_unk");
52 static int __init test_nmi_ipi_callback(unsigned int val, struct pt_regs *regs)
54 int cpu = raw_smp_processor_id();
56 if (cpumask_test_and_clear_cpu(cpu, to_cpumask(nmi_ipi_mask)))
62 static void __init test_nmi_ipi(struct cpumask *mask)
64 unsigned long timeout;
66 if (register_nmi_handler(NMI_LOCAL, test_nmi_ipi_callback,
67 NMI_FLAG_FIRST, "nmi_selftest")) {
72 /* sync above data before sending NMI */
75 apic->send_IPI_mask(mask, NMI_VECTOR);
77 /* Don't wait longer than a second */
78 timeout = USEC_PER_SEC;
79 while (!cpumask_empty(mask) && timeout--)
82 /* What happens if we timeout, do we still unregister?? */
83 unregister_nmi_handler(NMI_LOCAL, "nmi_selftest");
90 static void __init remote_ipi(void)
92 cpumask_copy(to_cpumask(nmi_ipi_mask), cpu_online_mask);
93 cpumask_clear_cpu(smp_processor_id(), to_cpumask(nmi_ipi_mask));
94 if (!cpumask_empty(to_cpumask(nmi_ipi_mask)))
95 test_nmi_ipi(to_cpumask(nmi_ipi_mask));
98 static void __init local_ipi(void)
100 cpumask_clear(to_cpumask(nmi_ipi_mask));
101 cpumask_set_cpu(smp_processor_id(), to_cpumask(nmi_ipi_mask));
102 test_nmi_ipi(to_cpumask(nmi_ipi_mask));
105 static void __init reset_nmi(void)
110 static void __init dotest(void (*testcase_fn)(void), int expected)
114 * Filter out expected failures:
116 if (nmi_fail != expected) {
117 unexpected_testcase_failures++;
119 if (nmi_fail == FAILURE)
121 else if (nmi_fail == TIMEOUT)
127 testcase_successes++;
135 static inline void __init print_testname(const char *testname)
137 printk("%12s:", testname);
140 void __init nmi_selftest(void)
142 init_nmi_testsuite();
147 printk("----------------\n");
148 printk("| NMI testsuite:\n");
149 printk("--------------------\n");
151 print_testname("remote IPI");
152 dotest(remote_ipi, SUCCESS);
154 print_testname("local IPI");
155 dotest(local_ipi, SUCCESS);
158 cleanup_nmi_testsuite();
160 if (unexpected_testcase_failures) {
161 printk("--------------------\n");
162 printk("BUG: %3d unexpected failures (out of %3d) - debugging disabled! |\n",
163 unexpected_testcase_failures, testcase_total);
164 printk("-----------------------------------------------------------------\n");
165 } else if (expected_testcase_failures && testcase_successes) {
166 printk("--------------------\n");
167 printk("%3d out of %3d testcases failed, as expected. |\n",
168 expected_testcase_failures, testcase_total);
169 printk("----------------------------------------------------\n");
170 } else if (expected_testcase_failures && !testcase_successes) {
171 printk("--------------------\n");
172 printk("All %3d testcases failed, as expected. |\n",
173 expected_testcase_failures);
174 printk("----------------------------------------\n");
176 printk("--------------------\n");
177 printk("Good, all %3d testcases passed! |\n",
179 printk("---------------------------------\n");