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