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 
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