tools/testing/selftests/kselftest_module.h

   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
  24 static 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 */