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