test-eof.c revision 285612
1303231Sdim/* 2303231Sdim * Copyright (c) 2002-2007 Niels Provos <provos@citi.umich.edu> 3353358Sdim * Copyright (c) 2007-2012 Niels Provos and Nick Mathewson 4353358Sdim * 5353358Sdim * Redistribution and use in source and binary forms, with or without 6303231Sdim * modification, are permitted provided that the following conditions 7303231Sdim * are met: 8303231Sdim * 1. Redistributions of source code must retain the above copyright 9303231Sdim * notice, this list of conditions and the following disclaimer. 10303231Sdim * 2. Redistributions in binary form must reproduce the above copyright 11303231Sdim * notice, this list of conditions and the following disclaimer in the 12303231Sdim * documentation and/or other materials provided with the distribution. 13303231Sdim * 3. The name of the author may not be used to endorse or promote products 14303231Sdim * derived from this software without specific prior written permission. 15303231Sdim * 16303231Sdim * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 17303231Sdim * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 18303231Sdim * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 19303231Sdim * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 20303231Sdim * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 21303231Sdim * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 22303231Sdim * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 23303231Sdim * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 24303231Sdim * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 25303231Sdim * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26303231Sdim */ 27303231Sdim#include "event2/event-config.h" 28303231Sdim 29303231Sdim#ifdef _WIN32 30303231Sdim#include <winsock2.h> 31303231Sdim#else 32#include <unistd.h> 33#endif 34#include <sys/types.h> 35#include <sys/stat.h> 36#ifdef EVENT__HAVE_SYS_TIME_H 37#include <sys/time.h> 38#endif 39#ifdef EVENT__HAVE_SYS_SOCKET_H 40#include <sys/socket.h> 41#endif 42#include <fcntl.h> 43#include <stdlib.h> 44#include <stdio.h> 45#include <string.h> 46#include <errno.h> 47 48#include <event.h> 49#include <evutil.h> 50 51#ifdef EVENT____func__ 52#define __func__ EVENT____func__ 53#endif 54 55int test_okay = 1; 56int called = 0; 57struct timeval timeout = {60, 0}; 58 59static void 60read_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#ifndef SHUT_WR 85#define SHUT_WR 1 86#endif 87 88int 89main(int argc, char **argv) 90{ 91 struct event ev; 92 const char *test = "test string"; 93 evutil_socket_t pair[2]; 94 95#ifdef _WIN32 96 WORD wVersionRequested; 97 WSADATA wsaData; 98 99 wVersionRequested = MAKEWORD(2, 2); 100 101 (void) WSAStartup(wVersionRequested, &wsaData); 102#endif 103 104 if (evutil_socketpair(AF_UNIX, SOCK_STREAM, 0, pair) == -1) 105 return (1); 106 107 108 if (send(pair[0], test, (int)strlen(test)+1, 0) < 0) 109 return (1); 110 shutdown(pair[0], SHUT_WR); 111 112 /* Initalize the event library */ 113 event_init(); 114 115 /* Initalize one event */ 116 event_set(&ev, pair[1], EV_READ | EV_TIMEOUT, read_cb, &ev); 117 118 event_add(&ev, &timeout); 119 120 event_dispatch(); 121 122 return (test_okay); 123} 124 125