1 /* SPDX-License-Identifier: GPL-2.0+ */
2 #ifndef __KSELFTEST_MODULE_H
3 #define __KSELFTEST_MODULE_H
4
5 #include <linux/module.h>
6 #include <linux/panic.h>
7
8 /*
9 * Test framework for writing test modules to be loaded by kselftest.
10 * See Documentation/dev-tools/kselftest.rst for an example test module.
11 */
12
13 #define KSTM_MODULE_GLOBALS() \
14 static unsigned int total_tests __initdata; \
15 static unsigned int failed_tests __initdata; \
16 static unsigned int skipped_tests __initdata
17
18 #define KSTM_CHECK_ZERO(x) do { \
19 total_tests++; \
20 if (x) { \
21 pr_warn("TC failed at %s:%d\n", __func__, __LINE__); \
22 failed_tests++; \
23 } \
24 } while (0)
25
kstm_report(unsigned int total_tests,unsigned int failed_tests,unsigned int skipped_tests)26 static inline int kstm_report(unsigned int total_tests, unsigned int failed_tests,
27 unsigned int skipped_tests)
28 {
29 if (failed_tests == 0) {
30 if (skipped_tests) {
31 pr_info("skipped %u tests\n", skipped_tests);
32 pr_info("remaining %u tests passed\n", total_tests);
33 } else
34 pr_info("all %u tests passed\n", total_tests);
35 } else
36 pr_warn("failed %u out of %u tests\n", failed_tests, total_tests);
37
38 return failed_tests ? -EINVAL : 0;
39 }
40
41 #define KSTM_MODULE_LOADERS(__module) \
42 static int __init __module##_init(void) \
43 { \
44 pr_info("loaded.\n"); \
45 add_taint(TAINT_TEST, LOCKDEP_STILL_OK); \
46 selftest(); \
47 return kstm_report(total_tests, failed_tests, skipped_tests); \
48 } \
49 static void __exit __module##_exit(void) \
50 { \
51 pr_info("unloaded.\n"); \
52 } \
53 module_init(__module##_init); \
54 module_exit(__module##_exit)
55
56 MODULE_INFO(test, "Y");
57
58 #endif /* __KSELFTEST_MODULE_H */
59