1 /* $NetBSD: test-eof.c,v 1.1.1.3 2021/04/07 02:43:15 christos Exp $ */
2 /*
3 * Copyright (c) 2002-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 #include "../util-internal.h"
29 #include "event2/event-config.h"
30 #include <sys/cdefs.h>
31 __RCSID("$NetBSD: test-eof.c,v 1.1.1.3 2021/04/07 02:43:15 christos Exp $");
32
33 #ifdef _WIN32
34 #include <winsock2.h>
35 #else
36 #include <unistd.h>
37 #endif
38 #include <sys/types.h>
39 #include <sys/stat.h>
40 #ifdef EVENT__HAVE_SYS_TIME_H
41 #include <sys/time.h>
42 #endif
43 #ifdef EVENT__HAVE_SYS_SOCKET_H
44 #include <sys/socket.h>
45 #endif
46 #include <fcntl.h>
47 #include <stdlib.h>
48 #include <stdio.h>
49 #include <string.h>
50 #include <errno.h>
51
52 #include <event.h>
53 #include <evutil.h>
54
55 int test_okay = 1;
56 int called = 0;
57 struct timeval timeout = {60, 0};
58
59 static void
read_cb(evutil_socket_t fd,short event,void * arg)60 read_cb(evutil_socket_t fd, short event, void *arg)
61 {
62 char buf[256];
63 int len;
64
65 if (EV_TIMEOUT & event) {
66 printf("%s: Timeout!\n", __func__);
67 exit(1);
68 }
69
70 len = recv(fd, buf, sizeof(buf), 0);
71
72 printf("%s: read %d%s\n", __func__,
73 len, len ? "" : " - means EOF");
74
75 if (len) {
76 if (!called)
77 event_add(arg, &timeout);
78 } else if (called == 1)
79 test_okay = 0;
80
81 called++;
82 }
83
84 int
main(int argc,char ** argv)85 main(int argc, char **argv)
86 {
87 struct event ev;
88 const char *test = "test string";
89 evutil_socket_t pair[2];
90
91 #ifdef _WIN32
92 WORD wVersionRequested;
93 WSADATA wsaData;
94
95 wVersionRequested = MAKEWORD(2, 2);
96
97 (void) WSAStartup(wVersionRequested, &wsaData);
98 #endif
99
100 if (evutil_socketpair(AF_UNIX, SOCK_STREAM, 0, pair) == -1)
101 return (1);
102
103
104 if (send(pair[0], test, (int)strlen(test)+1, 0) < 0)
105 return (1);
106 shutdown(pair[0], EVUTIL_SHUT_WR);
107
108 /* Initialize the event library */
109 event_init();
110
111 /* Initialize one event */
112 event_set(&ev, pair[1], EV_READ | EV_TIMEOUT, read_cb, &ev);
113
114 event_add(&ev, &timeout);
115
116 event_dispatch();
117
118 return (test_okay);
119 }
120
121