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