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