xref: /minix3/external/bsd/libevent/dist/test/bench_http.c (revision e985b929927b5932e3b68f4b50587d458900107a)
1*e985b929SDavid van Moolenbroek /*	$NetBSD: bench_http.c,v 1.1.1.1 2013/04/11 16:43:33 christos Exp $	*/
2*e985b929SDavid van Moolenbroek /*
3*e985b929SDavid van Moolenbroek  * Copyright 2008-2012 Niels Provos and Nick Mathewson
4*e985b929SDavid van Moolenbroek  *
5*e985b929SDavid van Moolenbroek  * Redistribution and use in source and binary forms, with or without
6*e985b929SDavid van Moolenbroek  * modification, are permitted provided that the following conditions
7*e985b929SDavid van Moolenbroek  * are met:
8*e985b929SDavid van Moolenbroek  * 1. Redistributions of source code must retain the above copyright
9*e985b929SDavid van Moolenbroek  *    notice, this list of conditions and the following disclaimer.
10*e985b929SDavid van Moolenbroek  * 2. Redistributions in binary form must reproduce the above copyright
11*e985b929SDavid van Moolenbroek  *    notice, this list of conditions and the following disclaimer in the
12*e985b929SDavid van Moolenbroek  *    documentation and/or other materials provided with the distribution.
13*e985b929SDavid van Moolenbroek  * 4. The name of the author may not be used to endorse or promote products
14*e985b929SDavid van Moolenbroek  *    derived from this software without specific prior written permission.
15*e985b929SDavid van Moolenbroek  *
16*e985b929SDavid van Moolenbroek  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
17*e985b929SDavid van Moolenbroek  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
18*e985b929SDavid van Moolenbroek  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
19*e985b929SDavid van Moolenbroek  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
20*e985b929SDavid van Moolenbroek  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
21*e985b929SDavid van Moolenbroek  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22*e985b929SDavid van Moolenbroek  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23*e985b929SDavid van Moolenbroek  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24*e985b929SDavid van Moolenbroek  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
25*e985b929SDavid van Moolenbroek  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26*e985b929SDavid van Moolenbroek  *
27*e985b929SDavid van Moolenbroek  */
28*e985b929SDavid van Moolenbroek 
29*e985b929SDavid van Moolenbroek #include <sys/types.h>
30*e985b929SDavid van Moolenbroek #include <sys/stat.h>
31*e985b929SDavid van Moolenbroek #ifdef WIN32
32*e985b929SDavid van Moolenbroek #include <winsock2.h>
33*e985b929SDavid van Moolenbroek #else
34*e985b929SDavid van Moolenbroek #include <sys/socket.h>
35*e985b929SDavid van Moolenbroek #include <sys/resource.h>
36*e985b929SDavid van Moolenbroek #include <sys/time.h>
37*e985b929SDavid van Moolenbroek #include <unistd.h>
38*e985b929SDavid van Moolenbroek #endif
39*e985b929SDavid van Moolenbroek #include <fcntl.h>
40*e985b929SDavid van Moolenbroek #include <signal.h>
41*e985b929SDavid van Moolenbroek #include <stdlib.h>
42*e985b929SDavid van Moolenbroek #include <stdio.h>
43*e985b929SDavid van Moolenbroek #include <string.h>
44*e985b929SDavid van Moolenbroek #include <errno.h>
45*e985b929SDavid van Moolenbroek 
46*e985b929SDavid van Moolenbroek #include "event2/event.h"
47*e985b929SDavid van Moolenbroek #include "event2/buffer.h"
48*e985b929SDavid van Moolenbroek #include "event2/util.h"
49*e985b929SDavid van Moolenbroek #include "event2/http.h"
50*e985b929SDavid van Moolenbroek #include "event2/thread.h"
51*e985b929SDavid van Moolenbroek 
52*e985b929SDavid van Moolenbroek static void http_basic_cb(struct evhttp_request *req, void *arg);
53*e985b929SDavid van Moolenbroek 
54*e985b929SDavid van Moolenbroek static char *content;
55*e985b929SDavid van Moolenbroek static size_t content_len = 0;
56*e985b929SDavid van Moolenbroek 
57*e985b929SDavid van Moolenbroek static void
http_basic_cb(struct evhttp_request * req,void * arg)58*e985b929SDavid van Moolenbroek http_basic_cb(struct evhttp_request *req, void *arg)
59*e985b929SDavid van Moolenbroek {
60*e985b929SDavid van Moolenbroek 	struct evbuffer *evb = evbuffer_new();
61*e985b929SDavid van Moolenbroek 
62*e985b929SDavid van Moolenbroek 	evbuffer_add(evb, content, content_len);
63*e985b929SDavid van Moolenbroek 
64*e985b929SDavid van Moolenbroek 	/* allow sending of an empty reply */
65*e985b929SDavid van Moolenbroek 	evhttp_send_reply(req, HTTP_OK, "Everything is fine", evb);
66*e985b929SDavid van Moolenbroek 
67*e985b929SDavid van Moolenbroek 	evbuffer_free(evb);
68*e985b929SDavid van Moolenbroek }
69*e985b929SDavid van Moolenbroek 
70*e985b929SDavid van Moolenbroek #if LIBEVENT_VERSION_NUMBER >= 0x02000200
71*e985b929SDavid van Moolenbroek static void
http_ref_cb(struct evhttp_request * req,void * arg)72*e985b929SDavid van Moolenbroek http_ref_cb(struct evhttp_request *req, void *arg)
73*e985b929SDavid van Moolenbroek {
74*e985b929SDavid van Moolenbroek 	struct evbuffer *evb = evbuffer_new();
75*e985b929SDavid van Moolenbroek 
76*e985b929SDavid van Moolenbroek 	evbuffer_add_reference(evb, content, content_len, NULL, NULL);
77*e985b929SDavid van Moolenbroek 
78*e985b929SDavid van Moolenbroek 	/* allow sending of an empty reply */
79*e985b929SDavid van Moolenbroek 	evhttp_send_reply(req, HTTP_OK, "Everything is fine", evb);
80*e985b929SDavid van Moolenbroek 
81*e985b929SDavid van Moolenbroek 	evbuffer_free(evb);
82*e985b929SDavid van Moolenbroek }
83*e985b929SDavid van Moolenbroek #endif
84*e985b929SDavid van Moolenbroek 
85*e985b929SDavid van Moolenbroek int
main(int argc,char ** argv)86*e985b929SDavid van Moolenbroek main(int argc, char **argv)
87*e985b929SDavid van Moolenbroek {
88*e985b929SDavid van Moolenbroek 	struct event_config *cfg = event_config_new();
89*e985b929SDavid van Moolenbroek 	struct event_base *base;
90*e985b929SDavid van Moolenbroek 	struct evhttp *http;
91*e985b929SDavid van Moolenbroek 	int i;
92*e985b929SDavid van Moolenbroek 	int c;
93*e985b929SDavid van Moolenbroek 	int use_iocp = 0;
94*e985b929SDavid van Moolenbroek 	unsigned short port = 8080;
95*e985b929SDavid van Moolenbroek 	char *endptr = NULL;
96*e985b929SDavid van Moolenbroek 
97*e985b929SDavid van Moolenbroek #ifdef WIN32
98*e985b929SDavid van Moolenbroek 	WSADATA WSAData;
99*e985b929SDavid van Moolenbroek 	WSAStartup(0x101, &WSAData);
100*e985b929SDavid van Moolenbroek #else
101*e985b929SDavid van Moolenbroek 	if (signal(SIGPIPE, SIG_IGN) == SIG_ERR)
102*e985b929SDavid van Moolenbroek 		return (1);
103*e985b929SDavid van Moolenbroek #endif
104*e985b929SDavid van Moolenbroek 
105*e985b929SDavid van Moolenbroek 	for (i = 1; i < argc; ++i) {
106*e985b929SDavid van Moolenbroek 		if (*argv[i] != '-')
107*e985b929SDavid van Moolenbroek 			continue;
108*e985b929SDavid van Moolenbroek 
109*e985b929SDavid van Moolenbroek 		c = argv[i][1];
110*e985b929SDavid van Moolenbroek 
111*e985b929SDavid van Moolenbroek 		if ((c == 'p' || c == 'l') && i + 1 >= argc) {
112*e985b929SDavid van Moolenbroek 			fprintf(stderr, "-%c requires argument.\n", c);
113*e985b929SDavid van Moolenbroek 			exit(1);
114*e985b929SDavid van Moolenbroek 		}
115*e985b929SDavid van Moolenbroek 
116*e985b929SDavid van Moolenbroek 		switch (c) {
117*e985b929SDavid van Moolenbroek 		case 'p':
118*e985b929SDavid van Moolenbroek 			if (i+1 >= argc || !argv[i+1]) {
119*e985b929SDavid van Moolenbroek 				fprintf(stderr, "Missing port\n");
120*e985b929SDavid van Moolenbroek 				exit(1);
121*e985b929SDavid van Moolenbroek 			}
122*e985b929SDavid van Moolenbroek 			port = (int)strtol(argv[i+1], &endptr, 10);
123*e985b929SDavid van Moolenbroek 			if (*endptr != '\0') {
124*e985b929SDavid van Moolenbroek 				fprintf(stderr, "Bad port\n");
125*e985b929SDavid van Moolenbroek 				exit(1);
126*e985b929SDavid van Moolenbroek 			}
127*e985b929SDavid van Moolenbroek 			break;
128*e985b929SDavid van Moolenbroek 		case 'l':
129*e985b929SDavid van Moolenbroek 			if (i+1 >= argc || !argv[i+1]) {
130*e985b929SDavid van Moolenbroek 				fprintf(stderr, "Missing content length\n");
131*e985b929SDavid van Moolenbroek 				exit(1);
132*e985b929SDavid van Moolenbroek 			}
133*e985b929SDavid van Moolenbroek 			content_len = (size_t)strtol(argv[i+1], &endptr, 10);
134*e985b929SDavid van Moolenbroek 			if (*endptr != '\0' || content_len == 0) {
135*e985b929SDavid van Moolenbroek 				fprintf(stderr, "Bad content length\n");
136*e985b929SDavid van Moolenbroek 				exit(1);
137*e985b929SDavid van Moolenbroek 			}
138*e985b929SDavid van Moolenbroek 			break;
139*e985b929SDavid van Moolenbroek #ifdef WIN32
140*e985b929SDavid van Moolenbroek 		case 'i':
141*e985b929SDavid van Moolenbroek 			use_iocp = 1;
142*e985b929SDavid van Moolenbroek 			evthread_use_windows_threads();
143*e985b929SDavid van Moolenbroek 			event_config_set_flag(cfg,EVENT_BASE_FLAG_STARTUP_IOCP);
144*e985b929SDavid van Moolenbroek 			break;
145*e985b929SDavid van Moolenbroek #endif
146*e985b929SDavid van Moolenbroek 		default:
147*e985b929SDavid van Moolenbroek 			fprintf(stderr, "Illegal argument \"%c\"\n", c);
148*e985b929SDavid van Moolenbroek 			exit(1);
149*e985b929SDavid van Moolenbroek 		}
150*e985b929SDavid van Moolenbroek 	}
151*e985b929SDavid van Moolenbroek 
152*e985b929SDavid van Moolenbroek 	base = event_base_new_with_config(cfg);
153*e985b929SDavid van Moolenbroek 	if (!base) {
154*e985b929SDavid van Moolenbroek 		fprintf(stderr, "creating event_base failed. Exiting.\n");
155*e985b929SDavid van Moolenbroek 		return 1;
156*e985b929SDavid van Moolenbroek 	}
157*e985b929SDavid van Moolenbroek 
158*e985b929SDavid van Moolenbroek 	http = evhttp_new(base);
159*e985b929SDavid van Moolenbroek 
160*e985b929SDavid van Moolenbroek 	content = malloc(content_len);
161*e985b929SDavid van Moolenbroek 	if (content == NULL) {
162*e985b929SDavid van Moolenbroek 		fprintf(stderr, "Cannot allocate content\n");
163*e985b929SDavid van Moolenbroek 		exit(1);
164*e985b929SDavid van Moolenbroek 	} else {
165*e985b929SDavid van Moolenbroek 		int i = 0;
166*e985b929SDavid van Moolenbroek 		for (i = 0; i < (int)content_len; ++i)
167*e985b929SDavid van Moolenbroek 			content[i] = (i & 255);
168*e985b929SDavid van Moolenbroek 	}
169*e985b929SDavid van Moolenbroek 
170*e985b929SDavid van Moolenbroek 	evhttp_set_cb(http, "/ind", http_basic_cb, NULL);
171*e985b929SDavid van Moolenbroek 	fprintf(stderr, "/ind - basic content (memory copy)\n");
172*e985b929SDavid van Moolenbroek 
173*e985b929SDavid van Moolenbroek #ifdef _EVENT2_EVENT_H_
174*e985b929SDavid van Moolenbroek 	evhttp_set_cb(http, "/ref", http_ref_cb, NULL);
175*e985b929SDavid van Moolenbroek 	fprintf(stderr, "/ref - basic content (reference)\n");
176*e985b929SDavid van Moolenbroek #endif
177*e985b929SDavid van Moolenbroek 
178*e985b929SDavid van Moolenbroek 	fprintf(stderr, "Serving %d bytes on port %d using %s\n",
179*e985b929SDavid van Moolenbroek 	    (int)content_len, port,
180*e985b929SDavid van Moolenbroek 	    use_iocp? "IOCP" : event_base_get_method(base));
181*e985b929SDavid van Moolenbroek 
182*e985b929SDavid van Moolenbroek 	evhttp_bind_socket(http, "0.0.0.0", port);
183*e985b929SDavid van Moolenbroek 
184*e985b929SDavid van Moolenbroek 	if (use_iocp) {
185*e985b929SDavid van Moolenbroek 		struct timeval tv={99999999,0};
186*e985b929SDavid van Moolenbroek 		event_base_loopexit(base, &tv);
187*e985b929SDavid van Moolenbroek 	}
188*e985b929SDavid van Moolenbroek 	event_base_dispatch(base);
189*e985b929SDavid van Moolenbroek 
190*e985b929SDavid van Moolenbroek 	/* NOTREACHED */
191*e985b929SDavid van Moolenbroek 	return (0);
192*e985b929SDavid van Moolenbroek }
193