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