1 /* $NetBSD: regress_main.c,v 1.2 2013/04/11 16:56:42 christos Exp $ */ 2 /* 3 * Copyright (c) 2003-2007 Niels Provos <provos@citi.umich.edu> 4 * Copyright (c) 2007-2012 Niels Provos and Nick Mathewson 5 * 6 * Redistribution and use in source and binary forms, with or without 7 * modification, are permitted provided that the following conditions 8 * are met: 9 * 1. Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * 2. Redistributions in binary form must reproduce the above copyright 12 * notice, this list of conditions and the following disclaimer in the 13 * documentation and/or other materials provided with the distribution. 14 * 3. The name of the author may not be used to endorse or promote products 15 * derived from this software without specific prior written permission. 16 * 17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 18 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 19 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 20 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 21 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 22 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 23 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 24 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27 */ 28 29 #ifdef WIN32 30 #include <winsock2.h> 31 #include <windows.h> 32 #include <io.h> 33 #include <fcntl.h> 34 #endif 35 36 #if defined(__APPLE__) && defined(__ENVIRONMENT_MAC_OS_X_VERSION_MIN_REQUIRED__) 37 #if (__ENVIRONMENT_MAC_OS_X_VERSION_MIN_REQUIRED__ >= 1060 && \ 38 __ENVIRONMENT_MAC_OS_X_VERSION_MIN_REQUIRED__ < 1070) 39 #define FORK_BREAKS_GCOV 40 #include <vproc.h> 41 #endif 42 #endif 43 44 #include "event2/event-config.h" 45 #include <sys/cdefs.h> 46 __RCSID("$NetBSD: regress_main.c,v 1.2 2013/04/11 16:56:42 christos Exp $"); 47 48 #ifdef _EVENT___func__ 49 #define __func__ _EVENT___func__ 50 #endif 51 52 #if 0 53 #include <sys/types.h> 54 #include <sys/stat.h> 55 #ifdef _EVENT_HAVE_SYS_TIME_H 56 #include <sys/time.h> 57 #endif 58 #include <sys/queue.h> 59 #include <signal.h> 60 #include <errno.h> 61 #endif 62 63 #include <sys/types.h> 64 #ifdef _EVENT_HAVE_SYS_STAT_H 65 #include <sys/stat.h> 66 #endif 67 68 #ifndef WIN32 69 #include <sys/socket.h> 70 #include <sys/wait.h> 71 #include <signal.h> 72 #include <unistd.h> 73 #include <netdb.h> 74 #endif 75 76 #include <stdlib.h> 77 #include <stdio.h> 78 #include <string.h> 79 #include <assert.h> 80 81 #include "event2/util.h" 82 #include "event2/event.h" 83 #include "event2/event_compat.h" 84 #include "event2/dns.h" 85 #include "event2/dns_compat.h" 86 #include "event2/thread.h" 87 88 #include "event2/event-config.h" 89 #include <sys/cdefs.h> 90 __RCSID("$NetBSD: regress_main.c,v 1.2 2013/04/11 16:56:42 christos Exp $"); 91 #include "regress.h" 92 #include "tinytest.h" 93 #include "tinytest_macros.h" 94 #include "../iocp-internal.h" 95 #include "../event-internal.h" 96 97 long 98 timeval_msec_diff(const struct timeval *start, const struct timeval *end) 99 { 100 long ms = end->tv_sec - start->tv_sec; 101 ms *= 1000; 102 ms += ((end->tv_usec - start->tv_usec)+500) / 1000; 103 return ms; 104 105 } 106 107 /* ============================================================ */ 108 /* Code to wrap up old legacy test cases that used setup() and cleanup(). 109 * 110 * Not all of the tests designated "legacy" are ones that used setup() and 111 * cleanup(), of course. A test is legacy it it uses setup()/cleanup(), OR 112 * if it wants to find its event base/socketpair in global variables (ugh), 113 * OR if it wants to communicate success/failure through test_ok. 114 */ 115 116 /* This is set to true if we're inside a legacy test wrapper. It lets the 117 setup() and cleanup() functions in regress.c know they're not needed. 118 */ 119 int in_legacy_test_wrapper = 0; 120 121 static void dnslogcb(int w, const char *m) 122 { 123 TT_BLATHER(("%s", m)); 124 } 125 126 /* creates a temporary file with the data in it */ 127 int 128 regress_make_tmpfile(const void *data, size_t datalen) 129 { 130 #ifndef WIN32 131 char tmpfilename[32]; 132 int fd; 133 strcpy(tmpfilename, "/tmp/eventtmp.XXXXXX"); 134 #ifdef _EVENT_HAVE_UMASK 135 umask(0077); 136 #endif 137 fd = mkstemp(tmpfilename); 138 if (fd == -1) 139 return (-1); 140 if (write(fd, data, datalen) != (int)datalen) { 141 close(fd); 142 return (-1); 143 } 144 lseek(fd, 0, SEEK_SET); 145 /* remove it from the file system */ 146 unlink(tmpfilename); 147 return (fd); 148 #else 149 /* XXXX actually delete the file later */ 150 char tmpfilepath[MAX_PATH]; 151 char tmpfilename[MAX_PATH]; 152 DWORD r, written; 153 int tries = 16; 154 HANDLE h; 155 r = GetTempPathA(MAX_PATH, tmpfilepath); 156 if (r > MAX_PATH || r == 0) 157 return (-1); 158 for (; tries > 0; --tries) { 159 r = GetTempFileNameA(tmpfilepath, "LIBEVENT", 0, tmpfilename); 160 if (r == 0) 161 return (-1); 162 h = CreateFileA(tmpfilename, GENERIC_READ|GENERIC_WRITE, 163 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL); 164 if (h != INVALID_HANDLE_VALUE) 165 break; 166 } 167 if (tries == 0) 168 return (-1); 169 written = 0; 170 WriteFile(h, data, (DWORD)datalen, &written, NULL); 171 /* Closing the fd returned by this function will indeed close h. */ 172 return _open_osfhandle((intptr_t)h,_O_RDONLY); 173 #endif 174 } 175 176 #ifndef _WIN32 177 pid_t 178 regress_fork(void) 179 { 180 pid_t pid = fork(); 181 #ifdef FORK_BREAKS_GCOV 182 vproc_transaction_begin(0); 183 #endif 184 return pid; 185 } 186 #endif 187 188 static void 189 ignore_log_cb(int s, const char *msg) 190 { 191 } 192 193 static void * 194 basic_test_setup(const struct testcase_t *testcase) 195 { 196 struct event_base *base = NULL; 197 evutil_socket_t spair[2] = { -1, -1 }; 198 struct basic_test_data *data = NULL; 199 200 #ifndef WIN32 201 if (testcase->flags & TT_ENABLE_IOCP_FLAG) 202 return (void*)TT_SKIP; 203 #endif 204 205 if (testcase->flags & TT_NEED_THREADS) { 206 if (!(testcase->flags & TT_FORK)) 207 return NULL; 208 #if defined(EVTHREAD_USE_PTHREADS_IMPLEMENTED) 209 if (evthread_use_pthreads()) 210 exit(1); 211 #elif defined(EVTHREAD_USE_WINDOWS_THREADS_IMPLEMENTED) 212 if (evthread_use_windows_threads()) 213 exit(1); 214 #else 215 return (void*)TT_SKIP; 216 #endif 217 } 218 219 if (testcase->flags & TT_NEED_SOCKETPAIR) { 220 if (evutil_socketpair(AF_UNIX, SOCK_STREAM, 0, spair) == -1) { 221 fprintf(stderr, "%s: socketpair\n", __func__); 222 exit(1); 223 } 224 225 if (evutil_make_socket_nonblocking(spair[0]) == -1) { 226 fprintf(stderr, "fcntl(O_NONBLOCK)"); 227 exit(1); 228 } 229 230 if (evutil_make_socket_nonblocking(spair[1]) == -1) { 231 fprintf(stderr, "fcntl(O_NONBLOCK)"); 232 exit(1); 233 } 234 } 235 if (testcase->flags & TT_NEED_BASE) { 236 if (testcase->flags & TT_LEGACY) 237 base = event_init(); 238 else 239 base = event_base_new(); 240 if (!base) 241 exit(1); 242 } 243 if (testcase->flags & TT_ENABLE_IOCP_FLAG) { 244 if (event_base_start_iocp(base, 0)<0) { 245 event_base_free(base); 246 return (void*)TT_SKIP; 247 } 248 } 249 250 if (testcase->flags & TT_NEED_DNS) { 251 evdns_set_log_fn(dnslogcb); 252 if (evdns_init()) 253 return NULL; /* fast failure */ /*XXX asserts. */ 254 } 255 256 if (testcase->flags & TT_NO_LOGS) 257 event_set_log_callback(ignore_log_cb); 258 259 data = calloc(1, sizeof(*data)); 260 if (!data) 261 exit(1); 262 data->base = base; 263 data->pair[0] = spair[0]; 264 data->pair[1] = spair[1]; 265 data->setup_data = testcase->setup_data; 266 return data; 267 } 268 269 static int 270 basic_test_cleanup(const struct testcase_t *testcase, void *ptr) 271 { 272 struct basic_test_data *data = ptr; 273 274 if (testcase->flags & TT_NO_LOGS) 275 event_set_log_callback(NULL); 276 277 if (testcase->flags & TT_NEED_SOCKETPAIR) { 278 if (data->pair[0] != -1) 279 evutil_closesocket(data->pair[0]); 280 if (data->pair[1] != -1) 281 evutil_closesocket(data->pair[1]); 282 } 283 284 if (testcase->flags & TT_NEED_DNS) { 285 evdns_shutdown(0); 286 } 287 288 if (testcase->flags & TT_NEED_BASE) { 289 if (data->base) { 290 event_base_assert_ok(data->base); 291 event_base_free(data->base); 292 } 293 } 294 295 free(data); 296 297 return 1; 298 } 299 300 const struct testcase_setup_t basic_setup = { 301 basic_test_setup, basic_test_cleanup 302 }; 303 304 /* The "data" for a legacy test is just a pointer to the void fn(void) 305 function implementing the test case. We need to set up some globals, 306 though, since that's where legacy tests expect to find a socketpair 307 (sometimes) and a global event_base (sometimes). 308 */ 309 static void * 310 legacy_test_setup(const struct testcase_t *testcase) 311 { 312 struct basic_test_data *data = basic_test_setup(testcase); 313 if (data == (void*)TT_SKIP || data == NULL) 314 return data; 315 global_base = data->base; 316 pair[0] = data->pair[0]; 317 pair[1] = data->pair[1]; 318 data->legacy_test_fn = testcase->setup_data; 319 return data; 320 } 321 322 /* This function is the implementation of every legacy test case. It 323 sets test_ok to 0, invokes the test function, and tells tinytest that 324 the test failed if the test didn't set test_ok to 1. 325 */ 326 void 327 run_legacy_test_fn(void *ptr) 328 { 329 struct basic_test_data *data = ptr; 330 test_ok = called = 0; 331 332 in_legacy_test_wrapper = 1; 333 data->legacy_test_fn(); /* This part actually calls the test */ 334 in_legacy_test_wrapper = 0; 335 336 if (!test_ok) 337 tt_abort_msg("Legacy unit test failed"); 338 339 end: 340 test_ok = 0; 341 } 342 343 /* This function doesn't have to clean up ptr (which is just a pointer 344 to the test function), but it may need to close the socketpair or 345 free the event_base. 346 */ 347 static int 348 legacy_test_cleanup(const struct testcase_t *testcase, void *ptr) 349 { 350 int r = basic_test_cleanup(testcase, ptr); 351 pair[0] = pair[1] = -1; 352 global_base = NULL; 353 return r; 354 } 355 356 const struct testcase_setup_t legacy_setup = { 357 legacy_test_setup, legacy_test_cleanup 358 }; 359 360 /* ============================================================ */ 361 362 #if (!defined(_EVENT_HAVE_PTHREADS) && !defined(WIN32)) || defined(_EVENT_DISABLE_THREAD_SUPPORT) 363 struct testcase_t thread_testcases[] = { 364 { "basic", NULL, TT_SKIP, NULL, NULL }, 365 END_OF_TESTCASES 366 }; 367 #endif 368 369 struct testgroup_t testgroups[] = { 370 { "main/", main_testcases }, 371 { "heap/", minheap_testcases }, 372 { "et/", edgetriggered_testcases }, 373 { "evbuffer/", evbuffer_testcases }, 374 { "signal/", signal_testcases }, 375 { "util/", util_testcases }, 376 { "bufferevent/", bufferevent_testcases }, 377 { "http/", http_testcases }, 378 { "dns/", dns_testcases }, 379 { "evtag/", evtag_testcases }, 380 { "rpc/", rpc_testcases }, 381 { "thread/", thread_testcases }, 382 { "listener/", listener_testcases }, 383 #ifdef WIN32 384 { "iocp/", iocp_testcases }, 385 { "iocp/bufferevent/", bufferevent_iocp_testcases }, 386 { "iocp/listener/", listener_iocp_testcases }, 387 #endif 388 #ifdef LIBEVENT_CRYPTO 389 #ifdef _EVENT_HAVE_OPENSSL 390 { "ssl/", ssl_testcases }, 391 #endif 392 #endif 393 END_OF_GROUPS 394 }; 395 396 int 397 main(int argc, const char **argv) 398 { 399 #ifdef WIN32 400 WORD wVersionRequested; 401 WSADATA wsaData; 402 403 wVersionRequested = MAKEWORD(2, 2); 404 405 (void) WSAStartup(wVersionRequested, &wsaData); 406 #endif 407 408 #ifndef WIN32 409 if (signal(SIGPIPE, SIG_IGN) == SIG_ERR) 410 return 1; 411 #endif 412 413 #ifdef WIN32 414 tinytest_skip(testgroups, "http/connection_retry"); 415 #endif 416 417 #ifndef _EVENT_DISABLE_THREAD_SUPPORT 418 if (!getenv("EVENT_NO_DEBUG_LOCKS")) 419 evthread_enable_lock_debuging(); 420 #endif 421 422 if (tinytest_main(argc,argv,testgroups)) 423 return 1; 424 425 return 0; 426 } 427 428