This source file includes following definitions.
- timeval_msec_diff
- dnslogcb
- regress_make_tmpfile
- regress_fork
- ignore_log_cb
- basic_test_setup
- basic_test_cleanup
- legacy_test_setup
- run_legacy_test_fn
- legacy_test_cleanup
- main
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 #ifdef WIN32
29 #include <winsock2.h>
30 #include <windows.h>
31 #include <io.h>
32 #include <fcntl.h>
33 #endif
34
35 #if defined(__APPLE__) && defined(__ENVIRONMENT_MAC_OS_X_VERSION_MIN_REQUIRED__)
36 #if (__ENVIRONMENT_MAC_OS_X_VERSION_MIN_REQUIRED__ >= 1060 && \
37 __ENVIRONMENT_MAC_OS_X_VERSION_MIN_REQUIRED__ < 1070)
38 #define FORK_BREAKS_GCOV
39 #include <vproc.h>
40 #endif
41 #endif
42
43 #include "event2/event-config.h"
44
45 #ifdef _EVENT___func__
46 #define __func__ _EVENT___func__
47 #endif
48
49 #if 0
50 #include <sys/types.h>
51 #include <sys/stat.h>
52 #ifdef _EVENT_HAVE_SYS_TIME_H
53 #include <sys/time.h>
54 #endif
55 #include <sys/queue.h>
56 #include <signal.h>
57 #include <errno.h>
58 #endif
59
60 #include <sys/types.h>
61 #ifdef _EVENT_HAVE_SYS_STAT_H
62 #include <sys/stat.h>
63 #endif
64
65 #ifndef WIN32
66 #include <sys/socket.h>
67 #include <sys/wait.h>
68 #include <signal.h>
69 #include <unistd.h>
70 #include <netdb.h>
71 #endif
72
73 #include <stdlib.h>
74 #include <stdio.h>
75 #include <string.h>
76 #include <assert.h>
77
78 #include "event2/util.h"
79 #include "event2/event.h"
80 #include "event2/event_compat.h"
81 #include "event2/dns.h"
82 #include "event2/dns_compat.h"
83 #include "event2/thread.h"
84
85 #include "event2/event-config.h"
86 #include "regress.h"
87 #include "tinytest.h"
88 #include "tinytest_macros.h"
89 #include "../iocp-internal.h"
90 #include "../event-internal.h"
91
92 long
93 timeval_msec_diff(const struct timeval *start, const struct timeval *end)
94 {
95 long ms = end->tv_sec - start->tv_sec;
96 ms *= 1000;
97 ms += ((end->tv_usec - start->tv_usec)+500) / 1000;
98 return ms;
99
100 }
101
102
103
104
105
106
107
108
109
110
111
112
113
114 int in_legacy_test_wrapper = 0;
115
116 static void dnslogcb(int w, const char *m)
117 {
118 TT_BLATHER(("%s", m));
119 }
120
121
122 int
123 regress_make_tmpfile(const void *data, size_t datalen)
124 {
125 #ifndef WIN32
126 char tmpfilename[32];
127 int fd;
128 strcpy(tmpfilename, "/tmp/eventtmp.XXXXXX");
129 #ifdef _EVENT_HAVE_UMASK
130 umask(0077);
131 #endif
132 fd = mkstemp(tmpfilename);
133 if (fd == -1)
134 return (-1);
135 if (write(fd, data, datalen) != (int)datalen) {
136 close(fd);
137 return (-1);
138 }
139 lseek(fd, 0, SEEK_SET);
140
141 unlink(tmpfilename);
142 return (fd);
143 #else
144
145 char tmpfilepath[MAX_PATH];
146 char tmpfilename[MAX_PATH];
147 DWORD r, written;
148 int tries = 16;
149 HANDLE h;
150 r = GetTempPathA(MAX_PATH, tmpfilepath);
151 if (r > MAX_PATH || r == 0)
152 return (-1);
153 for (; tries > 0; --tries) {
154 r = GetTempFileNameA(tmpfilepath, "LIBEVENT", 0, tmpfilename);
155 if (r == 0)
156 return (-1);
157 h = CreateFileA(tmpfilename, GENERIC_READ|GENERIC_WRITE,
158 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
159 if (h != INVALID_HANDLE_VALUE)
160 break;
161 }
162 if (tries == 0)
163 return (-1);
164 written = 0;
165 WriteFile(h, data, (DWORD)datalen, &written, NULL);
166
167 return _open_osfhandle((intptr_t)h,_O_RDONLY);
168 #endif
169 }
170
171 #ifndef _WIN32
172 pid_t
173 regress_fork(void)
174 {
175 pid_t pid = fork();
176 #ifdef FORK_BREAKS_GCOV
177 vproc_transaction_begin(0);
178 #endif
179 return pid;
180 }
181 #endif
182
183 static void
184 ignore_log_cb(int s, const char *msg)
185 {
186 }
187
188 static void *
189 basic_test_setup(const struct testcase_t *testcase)
190 {
191 struct event_base *base = NULL;
192 evutil_socket_t spair[2] = { -1, -1 };
193 struct basic_test_data *data = NULL;
194
195 #ifndef WIN32
196 if (testcase->flags & TT_ENABLE_IOCP_FLAG)
197 return (void*)TT_SKIP;
198 #endif
199
200 if (testcase->flags & TT_NEED_THREADS) {
201 if (!(testcase->flags & TT_FORK))
202 return NULL;
203 #if defined(EVTHREAD_USE_PTHREADS_IMPLEMENTED)
204 if (evthread_use_pthreads())
205 exit(1);
206 #elif defined(EVTHREAD_USE_WINDOWS_THREADS_IMPLEMENTED)
207 if (evthread_use_windows_threads())
208 exit(1);
209 #else
210 return (void*)TT_SKIP;
211 #endif
212 }
213
214 if (testcase->flags & TT_NEED_SOCKETPAIR) {
215 if (evutil_socketpair(AF_UNIX, SOCK_STREAM, 0, spair) == -1) {
216 fprintf(stderr, "%s: socketpair\n", __func__);
217 exit(1);
218 }
219
220 if (evutil_make_socket_nonblocking(spair[0]) == -1) {
221 fprintf(stderr, "fcntl(O_NONBLOCK)");
222 exit(1);
223 }
224
225 if (evutil_make_socket_nonblocking(spair[1]) == -1) {
226 fprintf(stderr, "fcntl(O_NONBLOCK)");
227 exit(1);
228 }
229 }
230 if (testcase->flags & TT_NEED_BASE) {
231 if (testcase->flags & TT_LEGACY)
232 base = event_init();
233 else
234 base = event_base_new();
235 if (!base)
236 exit(1);
237 }
238 if (testcase->flags & TT_ENABLE_IOCP_FLAG) {
239 if (event_base_start_iocp(base, 0)<0) {
240 event_base_free(base);
241 return (void*)TT_SKIP;
242 }
243 }
244
245 if (testcase->flags & TT_NEED_DNS) {
246 evdns_set_log_fn(dnslogcb);
247 if (evdns_init())
248 return NULL;
249 }
250
251 if (testcase->flags & TT_NO_LOGS)
252 event_set_log_callback(ignore_log_cb);
253
254 data = calloc(1, sizeof(*data));
255 if (!data)
256 exit(1);
257 data->base = base;
258 data->pair[0] = spair[0];
259 data->pair[1] = spair[1];
260 data->setup_data = testcase->setup_data;
261 return data;
262 }
263
264 static int
265 basic_test_cleanup(const struct testcase_t *testcase, void *ptr)
266 {
267 struct basic_test_data *data = ptr;
268
269 if (testcase->flags & TT_NO_LOGS)
270 event_set_log_callback(NULL);
271
272 if (testcase->flags & TT_NEED_SOCKETPAIR) {
273 if (data->pair[0] != -1)
274 evutil_closesocket(data->pair[0]);
275 if (data->pair[1] != -1)
276 evutil_closesocket(data->pair[1]);
277 }
278
279 if (testcase->flags & TT_NEED_DNS) {
280 evdns_shutdown(0);
281 }
282
283 if (testcase->flags & TT_NEED_BASE) {
284 if (data->base) {
285 event_base_assert_ok(data->base);
286 event_base_free(data->base);
287 }
288 }
289
290 free(data);
291
292 return 1;
293 }
294
295 const struct testcase_setup_t basic_setup = {
296 basic_test_setup, basic_test_cleanup
297 };
298
299
300
301
302
303
304 static void *
305 legacy_test_setup(const struct testcase_t *testcase)
306 {
307 struct basic_test_data *data = basic_test_setup(testcase);
308 if (data == (void*)TT_SKIP || data == NULL)
309 return data;
310 global_base = data->base;
311 pair[0] = data->pair[0];
312 pair[1] = data->pair[1];
313 data->legacy_test_fn = testcase->setup_data;
314 return data;
315 }
316
317
318
319
320
321 void
322 run_legacy_test_fn(void *ptr)
323 {
324 struct basic_test_data *data = ptr;
325 test_ok = called = 0;
326
327 in_legacy_test_wrapper = 1;
328 data->legacy_test_fn();
329 in_legacy_test_wrapper = 0;
330
331 if (!test_ok)
332 tt_abort_msg("Legacy unit test failed");
333
334 end:
335 test_ok = 0;
336 }
337
338
339
340
341
342 static int
343 legacy_test_cleanup(const struct testcase_t *testcase, void *ptr)
344 {
345 int r = basic_test_cleanup(testcase, ptr);
346 pair[0] = pair[1] = -1;
347 global_base = NULL;
348 return r;
349 }
350
351 const struct testcase_setup_t legacy_setup = {
352 legacy_test_setup, legacy_test_cleanup
353 };
354
355
356
357 #if (!defined(_EVENT_HAVE_PTHREADS) && !defined(WIN32)) || defined(_EVENT_DISABLE_THREAD_SUPPORT)
358 struct testcase_t thread_testcases[] = {
359 { "basic", NULL, TT_SKIP, NULL, NULL },
360 END_OF_TESTCASES
361 };
362 #endif
363
364 struct testgroup_t testgroups[] = {
365 { "main/", main_testcases },
366 { "heap/", minheap_testcases },
367 { "et/", edgetriggered_testcases },
368 { "evbuffer/", evbuffer_testcases },
369 { "signal/", signal_testcases },
370 { "util/", util_testcases },
371 { "bufferevent/", bufferevent_testcases },
372 { "http/", http_testcases },
373 { "dns/", dns_testcases },
374 { "evtag/", evtag_testcases },
375 { "rpc/", rpc_testcases },
376 { "thread/", thread_testcases },
377 { "listener/", listener_testcases },
378 #ifdef WIN32
379 { "iocp/", iocp_testcases },
380 { "iocp/bufferevent/", bufferevent_iocp_testcases },
381 { "iocp/listener/", listener_iocp_testcases },
382 #endif
383 #ifdef _EVENT_HAVE_OPENSSL
384 { "ssl/", ssl_testcases },
385 #endif
386 END_OF_GROUPS
387 };
388
389 int
390 main(int argc, const char **argv)
391 {
392 #ifdef WIN32
393 WORD wVersionRequested;
394 WSADATA wsaData;
395
396 wVersionRequested = MAKEWORD(2, 2);
397
398 (void) WSAStartup(wVersionRequested, &wsaData);
399 #endif
400
401 #ifndef WIN32
402 if (signal(SIGPIPE, SIG_IGN) == SIG_ERR)
403 return 1;
404 #endif
405
406 #ifdef WIN32
407 tinytest_skip(testgroups, "http/connection_retry");
408 #endif
409
410 #ifndef _EVENT_DISABLE_THREAD_SUPPORT
411 if (!getenv("EVENT_NO_DEBUG_LOCKS"))
412 evthread_enable_lock_debuging();
413 #endif
414
415 if (tinytest_main(argc,argv,testgroups))
416 return 1;
417
418 return 0;
419 }
420