Home | History | Annotate | Download | only in test
      1 /*
      2  * Copyright (c) 2002-2007 Niels Provos <provos (at) citi.umich.edu>
      3  * Copyright (c) 2007-2012 Niels Provos and Nick Mathewson
      4  *
      5  * Redistribution and use in source and binary forms, with or without
      6  * modification, are permitted provided that the following conditions
      7  * are met:
      8  * 1. Redistributions of source code must retain the above copyright
      9  *    notice, this list of conditions and the following disclaimer.
     10  * 2. Redistributions in binary form must reproduce the above copyright
     11  *    notice, this list of conditions and the following disclaimer in the
     12  *    documentation and/or other materials provided with the distribution.
     13  * 3. The name of the author may not be used to endorse or promote products
     14  *    derived from this software without specific prior written permission.
     15  *
     16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
     17  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
     18  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
     19  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
     20  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
     21  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     22  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     23  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
     25  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     26  */
     27 #include "../util-internal.h"
     28 #include "event2/event-config.h"
     29 
     30 #ifdef _WIN32
     31 #include <winsock2.h>
     32 #else
     33 #include <unistd.h>
     34 #endif
     35 #include <sys/types.h>
     36 #include <sys/stat.h>
     37 #ifdef EVENT__HAVE_SYS_TIME_H
     38 #include <sys/time.h>
     39 #endif
     40 #ifdef EVENT__HAVE_SYS_SOCKET_H
     41 #include <sys/socket.h>
     42 #endif
     43 #include <fcntl.h>
     44 #include <stdlib.h>
     45 #include <stdio.h>
     46 #include <string.h>
     47 #include <errno.h>
     48 
     49 #include <event.h>
     50 #include <evutil.h>
     51 
     52 int test_okay = 1;
     53 int called = 0;
     54 struct timeval timeout = {60, 0};
     55 
     56 static void
     57 read_cb(evutil_socket_t fd, short event, void *arg)
     58 {
     59 	char buf[256];
     60 	int len;
     61 
     62 	if (EV_TIMEOUT & event) {
     63 		printf("%s: Timeout!\n", __func__);
     64 		exit(1);
     65 	}
     66 
     67 	len = recv(fd, buf, sizeof(buf), 0);
     68 
     69 	printf("%s: read %d%s\n", __func__,
     70 	    len, len ? "" : " - means EOF");
     71 
     72 	if (len) {
     73 		if (!called)
     74 			event_add(arg, &timeout);
     75 	} else if (called == 1)
     76 		test_okay = 0;
     77 
     78 	called++;
     79 }
     80 
     81 int
     82 main(int argc, char **argv)
     83 {
     84 	struct event ev;
     85 	const char *test = "test string";
     86 	evutil_socket_t pair[2];
     87 
     88 #ifdef _WIN32
     89 	WORD wVersionRequested;
     90 	WSADATA wsaData;
     91 
     92 	wVersionRequested = MAKEWORD(2, 2);
     93 
     94 	(void) WSAStartup(wVersionRequested, &wsaData);
     95 #endif
     96 
     97 	if (evutil_socketpair(AF_UNIX, SOCK_STREAM, 0, pair) == -1)
     98 		return (1);
     99 
    100 
    101 	if (send(pair[0], test, (int)strlen(test)+1, 0) < 0)
    102 		return (1);
    103 	shutdown(pair[0], EVUTIL_SHUT_WR);
    104 
    105 	/* Initalize the event library */
    106 	event_init();
    107 
    108 	/* Initalize one event */
    109 	event_set(&ev, pair[1], EV_READ | EV_TIMEOUT, read_cb, &ev);
    110 
    111 	event_add(&ev, &timeout);
    112 
    113 	event_dispatch();
    114 
    115 	return (test_okay);
    116 }
    117 
    118