1 /*
2 * Copyright 2007-2012 Niels Provos and Nick Mathewson
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
7 * 1. Redistributions of source code must retain the above copyright
8 * notice, this list of conditions and the following disclaimer.
9 * 2. Redistributions in binary form must reproduce the above copyright
10 * notice, this list of conditions and the following disclaimer in the
11 * documentation and/or other materials provided with the distribution.
12 * 4. The name of the author may not be used to endorse or promote products
13 * derived from this software without specific prior written permission.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
16 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
17 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
18 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
19 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
20 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
21 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
22 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
23 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
24 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25 *
26 */
27
28 #include "event2/event-config.h"
29
30 #include <sys/types.h>
31 #include <sys/stat.h>
32 #ifdef EVENT__HAVE_SYS_TIME_H
33 #include <sys/time.h>
34 #endif
35 #ifdef _WIN32
36 #define WIN32_LEAN_AND_MEAN
37 #include <windows.h>
38 #else
39 #include <sys/socket.h>
40 #include <sys/resource.h>
41 #endif
42 #include <signal.h>
43 #include <fcntl.h>
44 #include <stdlib.h>
45 #include <stdio.h>
46 #include <string.h>
47 #ifdef EVENT__HAVE_UNISTD_H
48 #include <unistd.h>
49 #endif
50 #include <errno.h>
51 #include <getopt.h>
52 #include <event.h>
53 #include <evutil.h>
54
55 /*
56 * This benchmark tests how quickly we can propagate a write down a chain
57 * of socket pairs. We start by writing to the first socket pair and all
58 * events will fire subsequently until the last socket pair has been reached
59 * and the benchmark terminates.
60 */
61
62 static int fired;
63 static evutil_socket_t *pipes;
64 static struct event *events;
65
66 static void
read_cb(evutil_socket_t fd,short which,void * arg)67 read_cb(evutil_socket_t fd, short which, void *arg)
68 {
69 char ch;
70 evutil_socket_t sock = (evutil_socket_t)(ev_intptr_t)arg;
71
72 (void) recv(fd, &ch, sizeof(ch), 0);
73 if (sock >= 0) {
74 if (send(sock, "e", 1, 0) < 0)
75 perror("send");
76 }
77 fired++;
78 }
79
80 static struct timeval *
run_once(int num_pipes)81 run_once(int num_pipes)
82 {
83 int i;
84 evutil_socket_t *cp;
85 static struct timeval ts, te, tv_timeout;
86
87 events = (struct event *)calloc(num_pipes, sizeof(struct event));
88 pipes = (evutil_socket_t *)calloc(num_pipes * 2, sizeof(evutil_socket_t));
89
90 if (events == NULL || pipes == NULL) {
91 perror("malloc");
92 exit(1);
93 }
94
95 for (cp = pipes, i = 0; i < num_pipes; i++, cp += 2) {
96 if (evutil_socketpair(AF_UNIX, SOCK_STREAM, 0, cp) == -1) {
97 perror("socketpair");
98 exit(1);
99 }
100 }
101
102 /* measurements includes event setup */
103 evutil_gettimeofday(&ts, NULL);
104
105 /* provide a default timeout for events */
106 evutil_timerclear(&tv_timeout);
107 tv_timeout.tv_sec = 60;
108
109 for (cp = pipes, i = 0; i < num_pipes; i++, cp += 2) {
110 evutil_socket_t fd = i < num_pipes - 1 ? cp[3] : -1;
111 event_set(&events[i], cp[0], EV_READ, read_cb,
112 (void *)(ev_intptr_t)fd);
113 event_add(&events[i], &tv_timeout);
114 }
115
116 fired = 0;
117
118 /* kick everything off with a single write */
119 if (send(pipes[1], "e", 1, 0) < 0)
120 perror("send");
121
122 event_dispatch();
123
124 evutil_gettimeofday(&te, NULL);
125 evutil_timersub(&te, &ts, &te);
126
127 for (cp = pipes, i = 0; i < num_pipes; i++, cp += 2) {
128 event_del(&events[i]);
129 evutil_closesocket(cp[0]);
130 evutil_closesocket(cp[1]);
131 }
132
133 free(pipes);
134 free(events);
135
136 return (&te);
137 }
138
139 int
main(int argc,char ** argv)140 main(int argc, char **argv)
141 {
142 #ifdef HAVE_SETRLIMIT
143 struct rlimit rl;
144 #endif
145 int i, c;
146 struct timeval *tv;
147
148 int num_pipes = 100;
149 #ifdef _WIN32
150 WSADATA WSAData;
151 WSAStartup(0x101, &WSAData);
152 #endif
153
154 while ((c = getopt(argc, argv, "n:")) != -1) {
155 switch (c) {
156 case 'n':
157 num_pipes = atoi(optarg);
158 break;
159 default:
160 fprintf(stderr, "Illegal argument \"%c\"\n", c);
161 exit(1);
162 }
163 }
164
165 #ifdef HAVE_SETRLIMIT
166 rl.rlim_cur = rl.rlim_max = num_pipes * 2 + 50;
167 if (setrlimit(RLIMIT_NOFILE, &rl) == -1) {
168 perror("setrlimit");
169 exit(1);
170 }
171 #endif
172
173 event_init();
174
175 for (i = 0; i < 25; i++) {
176 tv = run_once(num_pipes);
177 if (tv == NULL)
178 exit(1);
179 fprintf(stdout, "%ld\n",
180 tv->tv_sec * 1000000L + tv->tv_usec);
181 }
182
183 #ifdef _WIN32
184 WSACleanup();
185 #endif
186
187 exit(0);
188 }
189