• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /* SPDX-License-Identifier: GPL-2.0 */
2 #include <stdio.h>
3 #include <unistd.h>
4 #include <errno.h>
5 #include <string.h>
6 #include <assert.h>
7 #include <stdlib.h>
8 #include <stdarg.h>
9 #include <time.h>
10 #include <signal.h>
11 
12 #include <linux/types.h>
13 typedef __u16 __sum16;
14 #include <arpa/inet.h>
15 #include <linux/if_ether.h>
16 #include <linux/if_packet.h>
17 #include <linux/ip.h>
18 #include <linux/ipv6.h>
19 #include <netinet/tcp.h>
20 #include <linux/filter.h>
21 #include <linux/perf_event.h>
22 #include <linux/socket.h>
23 #include <linux/unistd.h>
24 
25 #include <sys/ioctl.h>
26 #include <sys/wait.h>
27 #include <sys/types.h>
28 #include <sys/time.h>
29 #include <fcntl.h>
30 #include <pthread.h>
31 #include <linux/bpf.h>
32 #include <linux/err.h>
33 #include <bpf/bpf.h>
34 #include <bpf/libbpf.h>
35 
36 #include "test_iptunnel_common.h"
37 #include "bpf_util.h"
38 #include <bpf/bpf_endian.h>
39 #include "trace_helpers.h"
40 #include "testing_helpers.h"
41 #include "flow_dissector_load.h"
42 
43 enum verbosity {
44 	VERBOSE_NONE,
45 	VERBOSE_NORMAL,
46 	VERBOSE_VERY,
47 	VERBOSE_SUPER,
48 };
49 
50 struct str_set {
51 	const char **strs;
52 	int cnt;
53 };
54 
55 struct test_selector {
56 	struct str_set whitelist;
57 	struct str_set blacklist;
58 	bool *num_set;
59 	int num_set_len;
60 };
61 
62 struct test_env {
63 	struct test_selector test_selector;
64 	struct test_selector subtest_selector;
65 	bool verifier_stats;
66 	enum verbosity verbosity;
67 
68 	bool jit_enabled;
69 	bool get_test_cnt;
70 	bool list_test_names;
71 
72 	struct prog_test_def *test;
73 	FILE *stdout;
74 	FILE *stderr;
75 	char *log_buf;
76 	size_t log_cnt;
77 	int nr_cpus;
78 
79 	int succ_cnt; /* successful tests */
80 	int sub_succ_cnt; /* successful sub-tests */
81 	int fail_cnt; /* total failed tests + sub-tests */
82 	int skip_cnt; /* skipped tests */
83 
84 	int saved_netns_fd;
85 };
86 
87 extern struct test_env env;
88 
89 extern void test__force_log();
90 extern bool test__start_subtest(const char *name);
91 extern void test__skip(void);
92 extern void test__fail(void);
93 extern int test__join_cgroup(const char *path);
94 
95 #define PRINT_FAIL(format...)                                                  \
96 	({                                                                     \
97 		test__fail();                                                  \
98 		fprintf(stdout, "%s:FAIL:%d ", __func__, __LINE__);            \
99 		fprintf(stdout, ##format);                                     \
100 	})
101 
102 #define _CHECK(condition, tag, duration, format...) ({			\
103 	int __ret = !!(condition);					\
104 	int __save_errno = errno;					\
105 	if (__ret) {							\
106 		test__fail();						\
107 		fprintf(stdout, "%s:FAIL:%s ", __func__, tag);		\
108 		fprintf(stdout, ##format);				\
109 	} else {							\
110 		fprintf(stdout, "%s:PASS:%s %d nsec\n",			\
111 		       __func__, tag, duration);			\
112 	}								\
113 	errno = __save_errno;						\
114 	__ret;								\
115 })
116 
117 #define CHECK_FAIL(condition) ({					\
118 	int __ret = !!(condition);					\
119 	int __save_errno = errno;					\
120 	if (__ret) {							\
121 		test__fail();						\
122 		fprintf(stdout, "%s:FAIL:%d\n", __func__, __LINE__);	\
123 	}								\
124 	errno = __save_errno;						\
125 	__ret;								\
126 })
127 
128 #define CHECK(condition, tag, format...) \
129 	_CHECK(condition, tag, duration, format)
130 #define CHECK_ATTR(condition, tag, format...) \
131 	_CHECK(condition, tag, tattr.duration, format)
132 
133 #define ASSERT_EQ(actual, expected, name) ({				\
134 	static int duration = 0;					\
135 	typeof(actual) ___act = (actual);				\
136 	typeof(expected) ___exp = (expected);				\
137 	bool ___ok = ___act == ___exp;					\
138 	CHECK(!___ok, (name),						\
139 	      "unexpected %s: actual %lld != expected %lld\n",		\
140 	      (name), (long long)(___act), (long long)(___exp));	\
141 	___ok;								\
142 })
143 
144 #define ASSERT_STREQ(actual, expected, name) ({				\
145 	static int duration = 0;					\
146 	const char *___act = actual;					\
147 	const char *___exp = expected;					\
148 	bool ___ok = strcmp(___act, ___exp) == 0;			\
149 	CHECK(!___ok, (name),						\
150 	      "unexpected %s: actual '%s' != expected '%s'\n",		\
151 	      (name), ___act, ___exp);					\
152 	___ok;								\
153 })
154 
155 #define ASSERT_OK(res, name) ({						\
156 	static int duration = 0;					\
157 	long long ___res = (res);					\
158 	bool ___ok = ___res == 0;					\
159 	CHECK(!___ok, (name), "unexpected error: %lld\n", ___res);	\
160 	___ok;								\
161 })
162 
163 #define ASSERT_ERR(res, name) ({					\
164 	static int duration = 0;					\
165 	long long ___res = (res);					\
166 	bool ___ok = ___res < 0;					\
167 	CHECK(!___ok, (name), "unexpected success: %lld\n", ___res);	\
168 	___ok;								\
169 })
170 
171 #define ASSERT_NULL(ptr, name) ({					\
172 	static int duration = 0;					\
173 	const void *___res = (ptr);					\
174 	bool ___ok = !___res;						\
175 	CHECK(!___ok, (name), "unexpected pointer: %p\n", ___res);	\
176 	___ok;								\
177 })
178 
179 #define ASSERT_OK_PTR(ptr, name) ({					\
180 	static int duration = 0;					\
181 	const void *___res = (ptr);					\
182 	bool ___ok = !IS_ERR_OR_NULL(___res);				\
183 	CHECK(!___ok, (name),						\
184 	      "unexpected error: %ld\n", PTR_ERR(___res));		\
185 	___ok;								\
186 })
187 
188 #define ASSERT_ERR_PTR(ptr, name) ({					\
189 	static int duration = 0;					\
190 	const void *___res = (ptr);					\
191 	bool ___ok = IS_ERR(___res)					\
192 	CHECK(!___ok, (name), "unexpected pointer: %p\n", ___res);	\
193 	___ok;								\
194 })
195 
ptr_to_u64(const void * ptr)196 static inline __u64 ptr_to_u64(const void *ptr)
197 {
198 	return (__u64) (unsigned long) ptr;
199 }
200 
u64_to_ptr(__u64 ptr)201 static inline void *u64_to_ptr(__u64 ptr)
202 {
203 	return (void *) (unsigned long) ptr;
204 }
205 
206 int bpf_find_map(const char *test, struct bpf_object *obj, const char *name);
207 int compare_map_keys(int map1_fd, int map2_fd);
208 int compare_stack_ips(int smap_fd, int amap_fd, int stack_trace_len);
209 int extract_build_id(char *build_id, size_t size);
210 
211 #ifdef __x86_64__
212 #define SYS_NANOSLEEP_KPROBE_NAME "__x64_sys_nanosleep"
213 #elif defined(__s390x__)
214 #define SYS_NANOSLEEP_KPROBE_NAME "__s390x_sys_nanosleep"
215 #else
216 #define SYS_NANOSLEEP_KPROBE_NAME "sys_nanosleep"
217 #endif
218