Home | History | Annotate | Line # | Download | only in test 
      1 /*	$NetBSD: test-weof.c,v 1.1.1.3 2021/04/07 02:43:15 christos Exp $	*/
      2 /*
      3  * Copyright (c) 2002-2007 Niels Provos <provos (at) 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-weof.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 <signal.h>
     51 #include <errno.h>
     52 
     53 #include "event2/event.h"
     54 #include "event2/event_struct.h"
     55 #include "event2/event_compat.h"
     56 #include "event2/util.h"
     57 
     58 evutil_socket_t pair[2];
     59 int test_okay = 1;
     60 int called = 0;
     61 
     62 static void
     63 write_cb(evutil_socket_t fd, short event, void *arg)
     64 {
     65 	const char *test = "test string";
     66 	int len;
     67 
     68 	len = send(fd, test, (int)strlen(test) + 1, 0);
     69 
     70 	printf("%s: write %d%s\n", __func__,
     71 	    len, len ? "" : " - means EOF");
     72 
     73 	if (len > 0) {
     74 		if (!called)
     75 			event_add(arg, NULL);
     76 		evutil_closesocket(pair[0]);
     77 	} else if (called == 1)
     78 		test_okay = 0;
     79 
     80 	called++;
     81 }
     82 
     83 int
     84 main(int argc, char **argv)
     85 {
     86 	struct event ev;
     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 #ifndef _WIN32
     98 	if (signal(SIGPIPE, SIG_IGN) == SIG_ERR)
     99 		return (1);
    100 #endif
    101 
    102 	if (evutil_socketpair(AF_UNIX, SOCK_STREAM, 0, pair) == -1)
    103 		return (1);
    104 
    105 	/* Initialize the event library */
    106 	event_init();
    107 
    108 	/* Initialize one event */
    109 	event_set(&ev, pair[1], EV_WRITE, write_cb, &ev);
    110 
    111 	event_add(&ev, NULL);
    112 
    113 	event_dispatch();
    114 
    115 	return (test_okay);
    116 }
    117 
    118