testsuite/testsuite.h

   1 /*
   2  * Copyright (C) 2012-2013  ProFUSION embedded systems
   3  *
   4  * This program is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, see <http://www.gnu.org/licenses/>.
  16  */
  17
  18 #pragma once
  19
  20 #include <stdbool.h>
  21 #include <stdarg.h>
  22 #include <stdio.h>
  23
  24 #include <shared/macro.h>
  25
  26 struct test;
  27 typedef int (*testfunc)(const struct test *t);
  28
  29 enum test_config {
  30         /*
  31          * Where's the roots dir for this test. It will LD_PRELOAD path.so in
  32          * order to trap calls to functions using paths.
  33          */
  34         TC_ROOTFS = 0,
  35
  36         /*
  37          * What's the desired string to be returned by `uname -r`. It will
  38          * trap calls to uname(3P) by LD_PRELOAD'ing uname.so and then filling
  39          * in the information in u.release.
  40          */
  41         TC_UNAME_R,
  42
  43         /*
  44          * Fake calls to init_module(2), returning return-code and setting
  45          * errno to err-code. Set this variable with the following format:
  46          *
  47          *        modname:return-code:err-code
  48          *
  49          * When this variable is used, all calls to init_module() are trapped
  50          * and by default the return code is 0. In other words, they fake
  51          * "success" for all modules, except the ones in the list above, for
  52          * which the return codes are used.
  53          */
  54         TC_INIT_MODULE_RETCODES,
  55
  56         /*
  57          * Fake calls to delete_module(2), returning return-code and setting
  58          * errno to err-code. Set this variable with the following format:
  59          *
  60          *        modname:return-code:err-code
  61          *
  62          * When this variable is used, all calls to init_module() are trapped
  63          * and by default the return code is 0. In other words, they fake
  64          * "success" for all modules, except the ones in the list above, for
  65          * which the return codes are used.
  66          */
  67         TC_DELETE_MODULE_RETCODES,
  68
  69         _TC_LAST,
  70 };
  71
  72 #define S_TC_ROOTFS "TESTSUITE_ROOTFS"
  73 #define S_TC_UNAME_R "TESTSUITE_UNAME_R"
  74 #define S_TC_INIT_MODULE_RETCODES "TESTSUITE_INIT_MODULE_RETCODES"
  75 #define S_TC_DELETE_MODULE_RETCODES "TESTSUITE_DELETE_MODULE_RETCODES"
  76
  77 struct keyval {
  78         const char *key;
  79         const char *val;
  80 };
  81
  82 struct test {
  83         const char *name;
  84         const char *description;
  85         struct {
  86                 /* File with correct stdout */
  87                 const char *out;
  88                 /* File with correct stderr */
  89                 const char *err;
  90
  91                 /*
  92                  * Vector with pair of files
  93                  * key = correct file
  94                  * val = file to check
  95                  */
  96                 const struct keyval *files;
  97         } output;
  98         /* comma-separated list of loaded modules at the end of the test */
  99         const char *modules_loaded;
 100         testfunc func;
 101         const char *config[_TC_LAST];
 102         const char *path;
 103         const struct keyval *env_vars;
 104         bool need_spawn;
 105         bool expected_fail;
 106         bool print_outputs;
 107 } __attribute__((aligned(8)));
 108
 109
 110 int test_init(const struct test *start, const struct test *stop,
 111               int argc, char *const argv[]);
 112 const struct test *test_find(const struct test *start, const struct test *stop,
 113                              const char *name);
 114 int test_spawn_prog(const char *prog, const char *const args[]);
 115 int test_run(const struct test *t);
 116
 117 #define TS_EXPORT __attribute__ ((visibility("default")))
 118
 119 #define _LOG(prefix, fmt, ...) printf("TESTSUITE: " prefix fmt, ## __VA_ARGS__)
 120 #define LOG(fmt, ...) _LOG("", fmt, ## __VA_ARGS__)
 121 #define WARN(fmt, ...) _LOG("WARN: ", fmt, ## __VA_ARGS__)
 122 #define ERR(fmt, ...) _LOG("ERR: ", fmt, ## __VA_ARGS__)
 123
 124 #define assert_return(expr, r)                                          \
 125         do {                                                            \
 126                 if ((!(expr))) {                                        \
 127                         ERR("Failed assertion: " #expr " %s:%d %s\n",   \
 128                             __FILE__, __LINE__, __PRETTY_FUNCTION__);   \
 129                         return (r);                                     \
 130                 }                                                       \
 131         } while (false)
 132
 133
 134 /* Test definitions */
 135 #define DEFINE_TEST(_name, ...) \
 136         static const struct test s##_name##UNIQ \
 137         __attribute__((used, section("kmod_tests"), aligned(8))) = { \
 138                 .name = #_name, \
 139                 .func = _name, \
 140                 ## __VA_ARGS__ \
 141         };
 142
 143 #define TESTSUITE_MAIN() \
 144         extern struct test __start_kmod_tests[] __attribute__((weak, visibility("hidden")));    \
 145         extern struct test __stop_kmod_tests[] __attribute__((weak, visibility("hidden")));     \
 146         int main(int argc, char *argv[])                                                        \
 147         {                                                                                       \
 148                 const struct test *t;                                                           \
 149                 int arg;                                                                        \
 150                                                                                                 \
 151                 arg = test_init(__start_kmod_tests, __stop_kmod_tests, argc, argv);             \
 152                 if (arg == 0)                                                                   \
 153                         return 0;                                                               \
 154                                                                                                 \
 155                 if (arg < argc) {                                                               \
 156                         t = test_find(__start_kmod_tests, __stop_kmod_tests, argv[arg]);        \
 157                         if (t == NULL) {                                                        \
 158                                 fprintf(stderr, "could not find test %s\n", argv[arg]);         \
 159                                 exit(EXIT_FAILURE);                                             \
 160                         }                                                                       \
 161                                                                                                 \
 162                         return test_run(t);                                                     \
 163                 }                                                                               \
 164                                                                                                 \
 165                 for (t = __start_kmod_tests; t < __stop_kmod_tests; t++) {                      \
 166                         if (test_run(t) != 0)                                                   \
 167                                 exit(EXIT_FAILURE);                                             \
 168                 }                                                                               \
 169                                                                                                 \
 170                 exit(EXIT_SUCCESS);                                                             \
 171         }                                                                                       \
 172
 173 #ifdef noreturn
 174 # define __noreturn noreturn
 175 #elif __STDC_VERSION__ >= 201112L
 176 # define __noreturn _Noreturn
 177 #else
 178 # define __noreturn __attribute__((noreturn))
 179 #endif