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()			\
13static unsigned int total_tests __initdata;	\
14static 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
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)			\
35static int __init __module##_init(void)			\
36{							\
37	pr_info("loaded.\n");				\
38	selftest();					\
39	return kstm_report(total_tests, failed_tests);	\
40}							\
41static void __exit __module##_exit(void)		\
42{							\
43	pr_info("unloaded.\n");				\
44}							\
45module_init(__module##_init);				\
46module_exit(__module##_exit)
47
48#endif	/* __KSELFTEST_MODULE_H */
49