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