httpd.c revision 1.2.2.3 1 1.2.2.3 pgoyette /* $NetBSD: httpd.c,v 1.2.2.3 2019/01/18 08:49:57 pgoyette Exp $ */
2 1.2.2.2 pgoyette
3 1.2.2.2 pgoyette /*
4 1.2.2.2 pgoyette * Copyright (C) Internet Systems Consortium, Inc. ("ISC")
5 1.2.2.2 pgoyette *
6 1.2.2.2 pgoyette * This Source Code Form is subject to the terms of the Mozilla Public
7 1.2.2.2 pgoyette * License, v. 2.0. If a copy of the MPL was not distributed with this
8 1.2.2.2 pgoyette * file, You can obtain one at http://mozilla.org/MPL/2.0/.
9 1.2.2.2 pgoyette *
10 1.2.2.2 pgoyette * See the COPYRIGHT file distributed with this work for additional
11 1.2.2.2 pgoyette * information regarding copyright ownership.
12 1.2.2.2 pgoyette */
13 1.2.2.2 pgoyette
14 1.2.2.2 pgoyette
15 1.2.2.2 pgoyette /*! \file */
16 1.2.2.2 pgoyette
17 1.2.2.2 pgoyette #include <config.h>
18 1.2.2.2 pgoyette
19 1.2.2.3 pgoyette #include <inttypes.h>
20 1.2.2.3 pgoyette #include <stdbool.h>
21 1.2.2.3 pgoyette #include <string.h>
22 1.2.2.3 pgoyette
23 1.2.2.2 pgoyette #include <isc/buffer.h>
24 1.2.2.2 pgoyette #include <isc/httpd.h>
25 1.2.2.2 pgoyette #include <isc/mem.h>
26 1.2.2.2 pgoyette #include <isc/print.h>
27 1.2.2.2 pgoyette #include <isc/socket.h>
28 1.2.2.2 pgoyette #include <isc/string.h>
29 1.2.2.2 pgoyette #include <isc/task.h>
30 1.2.2.2 pgoyette #include <isc/time.h>
31 1.2.2.2 pgoyette #include <isc/util.h>
32 1.2.2.2 pgoyette
33 1.2.2.2 pgoyette #ifdef HAVE_ZLIB
34 1.2.2.2 pgoyette #include <zlib.h>
35 1.2.2.2 pgoyette #endif
36 1.2.2.2 pgoyette
37 1.2.2.2 pgoyette /*%
38 1.2.2.2 pgoyette * TODO:
39 1.2.2.2 pgoyette *
40 1.2.2.2 pgoyette * o Put in better checks to make certain things are passed in correctly.
41 1.2.2.2 pgoyette * This includes a magic number for externally-visible structures,
42 1.2.2.2 pgoyette * checking for NULL-ness before dereferencing, etc.
43 1.2.2.2 pgoyette * o Make the URL processing external functions which will fill-in a buffer
44 1.2.2.2 pgoyette * structure we provide, or return an error and we will render a generic
45 1.2.2.2 pgoyette * page and close the client.
46 1.2.2.2 pgoyette */
47 1.2.2.2 pgoyette
48 1.2.2.2 pgoyette #define MSHUTTINGDOWN(cm) ((cm->flags & ISC_HTTPDMGR_FLAGSHUTTINGDOWN) != 0)
49 1.2.2.2 pgoyette #define MSETSHUTTINGDOWN(cm) (cm->flags |= ISC_HTTPDMGR_FLAGSHUTTINGDOWN)
50 1.2.2.2 pgoyette
51 1.2.2.2 pgoyette #ifdef DEBUG_HTTPD
52 1.2.2.2 pgoyette #define ENTER(x) do { fprintf(stderr, "ENTER %s\n", (x)); } while (/*CONSTCOND*/0)
53 1.2.2.2 pgoyette #define EXIT(x) do { fprintf(stderr, "EXIT %s\n", (x)); } while (/*CONSTCOND*/0)
54 1.2.2.2 pgoyette #define NOTICE(x) do { fprintf(stderr, "NOTICE %s\n", (x)); } while (/*CONSTCOND*/0)
55 1.2.2.2 pgoyette #else
56 1.2.2.2 pgoyette #define ENTER(x) do { } while(/*CONSTCOND*/0)
57 1.2.2.2 pgoyette #define EXIT(x) do { } while(/*CONSTCOND*/0)
58 1.2.2.2 pgoyette #define NOTICE(x) do { } while(/*CONSTCOND*/0)
59 1.2.2.2 pgoyette #endif
60 1.2.2.2 pgoyette
61 1.2.2.2 pgoyette #define HTTP_RECVLEN 1024
62 1.2.2.2 pgoyette #define HTTP_SENDGROW 1024
63 1.2.2.2 pgoyette #define HTTP_SEND_MAXLEN 10240
64 1.2.2.2 pgoyette
65 1.2.2.2 pgoyette #define HTTPD_CLOSE 0x0001 /* Got a Connection: close header */
66 1.2.2.2 pgoyette #define HTTPD_FOUNDHOST 0x0002 /* Got a Host: header */
67 1.2.2.2 pgoyette #define HTTPD_KEEPALIVE 0x0004 /* Got a Connection: Keep-Alive */
68 1.2.2.2 pgoyette #define HTTPD_ACCEPT_DEFLATE 0x0008
69 1.2.2.2 pgoyette
70 1.2.2.2 pgoyette /*% http client */
71 1.2.2.2 pgoyette struct isc_httpd {
72 1.2.2.2 pgoyette isc_httpdmgr_t *mgr; /*%< our parent */
73 1.2.2.2 pgoyette ISC_LINK(isc_httpd_t) link;
74 1.2.2.2 pgoyette unsigned int state;
75 1.2.2.2 pgoyette isc_socket_t *sock;
76 1.2.2.2 pgoyette
77 1.2.2.2 pgoyette /*%
78 1.2.2.2 pgoyette * Received data state.
79 1.2.2.2 pgoyette */
80 1.2.2.2 pgoyette char recvbuf[HTTP_RECVLEN]; /*%< receive buffer */
81 1.2.2.3 pgoyette uint32_t recvlen; /*%< length recv'd */
82 1.2.2.2 pgoyette char *headers; /*%< set in process_request() */
83 1.2.2.2 pgoyette unsigned int method;
84 1.2.2.2 pgoyette char *url;
85 1.2.2.2 pgoyette char *querystring;
86 1.2.2.2 pgoyette char *protocol;
87 1.2.2.2 pgoyette
88 1.2.2.2 pgoyette /*
89 1.2.2.2 pgoyette * Flags on the httpd client.
90 1.2.2.2 pgoyette */
91 1.2.2.2 pgoyette int flags;
92 1.2.2.2 pgoyette
93 1.2.2.2 pgoyette /*%
94 1.2.2.2 pgoyette * Transmit data state.
95 1.2.2.2 pgoyette *
96 1.2.2.2 pgoyette * This is the data buffer we will transmit.
97 1.2.2.2 pgoyette *
98 1.2.2.2 pgoyette * This free function pointer is filled in by the rendering function
99 1.2.2.2 pgoyette * we call. The free function is called after the data is transmitted
100 1.2.2.2 pgoyette * to the client.
101 1.2.2.2 pgoyette *
102 1.2.2.2 pgoyette * The bufflist is the list of buffers we are currently transmitting.
103 1.2.2.2 pgoyette * The headerbuffer is where we render our headers to. If we run out of
104 1.2.2.2 pgoyette * space when rendering a header, we will change the size of our
105 1.2.2.2 pgoyette * buffer. We will not free it until we are finished, and will
106 1.2.2.2 pgoyette * allocate an additional HTTP_SENDGROW bytes per header space grow.
107 1.2.2.2 pgoyette *
108 1.2.2.2 pgoyette * We currently use three buffers total, one for the headers (which
109 1.2.2.2 pgoyette * we manage), another for the client to fill in (which it manages,
110 1.2.2.2 pgoyette * it provides the space for it, etc) -- we will pass that buffer
111 1.2.2.2 pgoyette * structure back to the caller, who is responsible for managing the
112 1.2.2.2 pgoyette * space it may have allocated as backing store for it. This second
113 1.2.2.2 pgoyette * buffer is bodybuffer, and we only allocate the buffer itself, not
114 1.2.2.2 pgoyette * the backing store.
115 1.2.2.2 pgoyette * The third buffer is compbuffer, managed by us, that contains the
116 1.2.2.2 pgoyette * compressed HTTP data, if compression is used.
117 1.2.2.2 pgoyette *
118 1.2.2.2 pgoyette */
119 1.2.2.2 pgoyette isc_buffer_t headerbuffer;
120 1.2.2.2 pgoyette isc_buffer_t compbuffer;
121 1.2.2.3 pgoyette isc_buffer_t *sendbuffer;
122 1.2.2.2 pgoyette
123 1.2.2.2 pgoyette const char *mimetype;
124 1.2.2.2 pgoyette unsigned int retcode;
125 1.2.2.2 pgoyette const char *retmsg;
126 1.2.2.2 pgoyette isc_buffer_t bodybuffer;
127 1.2.2.2 pgoyette isc_httpdfree_t *freecb;
128 1.2.2.2 pgoyette void *freecb_arg;
129 1.2.2.2 pgoyette };
130 1.2.2.2 pgoyette
131 1.2.2.2 pgoyette /*% lightweight socket manager for httpd output */
132 1.2.2.2 pgoyette struct isc_httpdmgr {
133 1.2.2.2 pgoyette isc_mem_t *mctx;
134 1.2.2.2 pgoyette isc_socket_t *sock; /*%< listening socket */
135 1.2.2.2 pgoyette isc_task_t *task; /*%< owning task */
136 1.2.2.2 pgoyette isc_timermgr_t *timermgr;
137 1.2.2.2 pgoyette
138 1.2.2.2 pgoyette isc_httpdclientok_t *client_ok; /*%< client validator */
139 1.2.2.2 pgoyette isc_httpdondestroy_t *ondestroy; /*%< cleanup callback */
140 1.2.2.2 pgoyette void *cb_arg; /*%< argument for the above */
141 1.2.2.2 pgoyette
142 1.2.2.2 pgoyette unsigned int flags;
143 1.2.2.2 pgoyette ISC_LIST(isc_httpd_t) running; /*%< running clients */
144 1.2.2.2 pgoyette
145 1.2.2.2 pgoyette isc_mutex_t lock;
146 1.2.2.2 pgoyette
147 1.2.2.2 pgoyette ISC_LIST(isc_httpdurl_t) urls; /*%< urls we manage */
148 1.2.2.2 pgoyette isc_httpdaction_t *render_404;
149 1.2.2.2 pgoyette isc_httpdaction_t *render_500;
150 1.2.2.2 pgoyette };
151 1.2.2.2 pgoyette
152 1.2.2.2 pgoyette /*%
153 1.2.2.2 pgoyette * HTTP methods.
154 1.2.2.2 pgoyette */
155 1.2.2.2 pgoyette #define ISC_HTTPD_METHODUNKNOWN 0
156 1.2.2.2 pgoyette #define ISC_HTTPD_METHODGET 1
157 1.2.2.2 pgoyette #define ISC_HTTPD_METHODPOST 2
158 1.2.2.2 pgoyette
159 1.2.2.2 pgoyette /*%
160 1.2.2.2 pgoyette * Client states.
161 1.2.2.2 pgoyette *
162 1.2.2.2 pgoyette * _IDLE The client is not doing anything at all. This state should
163 1.2.2.2 pgoyette * only occur just after creation, and just before being
164 1.2.2.2 pgoyette * destroyed.
165 1.2.2.2 pgoyette *
166 1.2.2.2 pgoyette * _RECV The client is waiting for data after issuing a socket recv().
167 1.2.2.2 pgoyette *
168 1.2.2.2 pgoyette * _RECVDONE Data has been received, and is being processed.
169 1.2.2.2 pgoyette *
170 1.2.2.2 pgoyette * _SEND All data for a response has completed, and a reply was
171 1.2.2.2 pgoyette * sent via a socket send() call.
172 1.2.2.2 pgoyette *
173 1.2.2.2 pgoyette * _SENDDONE Send is completed.
174 1.2.2.2 pgoyette *
175 1.2.2.2 pgoyette * Badly formatted state table:
176 1.2.2.2 pgoyette *
177 1.2.2.2 pgoyette * IDLE -> RECV when client has a recv() queued.
178 1.2.2.2 pgoyette *
179 1.2.2.2 pgoyette * RECV -> RECVDONE when recvdone event received.
180 1.2.2.2 pgoyette *
181 1.2.2.2 pgoyette * RECVDONE -> SEND if the data for a reply is at hand.
182 1.2.2.2 pgoyette *
183 1.2.2.2 pgoyette * SEND -> RECV when a senddone event was received.
184 1.2.2.2 pgoyette *
185 1.2.2.2 pgoyette * At any time -> RECV on error. If RECV fails, the client will
186 1.2.2.2 pgoyette * self-destroy, closing the socket and freeing memory.
187 1.2.2.2 pgoyette */
188 1.2.2.2 pgoyette #define ISC_HTTPD_STATEIDLE 0
189 1.2.2.2 pgoyette #define ISC_HTTPD_STATERECV 1
190 1.2.2.2 pgoyette #define ISC_HTTPD_STATERECVDONE 2
191 1.2.2.2 pgoyette #define ISC_HTTPD_STATESEND 3
192 1.2.2.2 pgoyette #define ISC_HTTPD_STATESENDDONE 4
193 1.2.2.2 pgoyette
194 1.2.2.2 pgoyette #define ISC_HTTPD_ISRECV(c) ((c)->state == ISC_HTTPD_STATERECV)
195 1.2.2.2 pgoyette #define ISC_HTTPD_ISRECVDONE(c) ((c)->state == ISC_HTTPD_STATERECVDONE)
196 1.2.2.2 pgoyette #define ISC_HTTPD_ISSEND(c) ((c)->state == ISC_HTTPD_STATESEND)
197 1.2.2.2 pgoyette #define ISC_HTTPD_ISSENDDONE(c) ((c)->state == ISC_HTTPD_STATESENDDONE)
198 1.2.2.2 pgoyette
199 1.2.2.2 pgoyette /*%
200 1.2.2.2 pgoyette * Overall magic test that means we're not idle.
201 1.2.2.2 pgoyette */
202 1.2.2.2 pgoyette #define ISC_HTTPD_SETRECV(c) ((c)->state = ISC_HTTPD_STATERECV)
203 1.2.2.2 pgoyette #define ISC_HTTPD_SETRECVDONE(c) ((c)->state = ISC_HTTPD_STATERECVDONE)
204 1.2.2.2 pgoyette #define ISC_HTTPD_SETSEND(c) ((c)->state = ISC_HTTPD_STATESEND)
205 1.2.2.2 pgoyette #define ISC_HTTPD_SETSENDDONE(c) ((c)->state = ISC_HTTPD_STATESENDDONE)
206 1.2.2.2 pgoyette
207 1.2.2.2 pgoyette static void isc_httpd_accept(isc_task_t *, isc_event_t *);
208 1.2.2.2 pgoyette static void isc_httpd_recvdone(isc_task_t *, isc_event_t *);
209 1.2.2.2 pgoyette static void isc_httpd_senddone(isc_task_t *, isc_event_t *);
210 1.2.2.2 pgoyette static void destroy_client(isc_httpd_t **);
211 1.2.2.2 pgoyette static isc_result_t process_request(isc_httpd_t *, int);
212 1.2.2.2 pgoyette static void httpdmgr_destroy(isc_httpdmgr_t *);
213 1.2.2.2 pgoyette static isc_result_t grow_headerspace(isc_httpd_t *);
214 1.2.2.2 pgoyette static void reset_client(isc_httpd_t *httpd);
215 1.2.2.2 pgoyette
216 1.2.2.2 pgoyette static isc_httpdaction_t render_404;
217 1.2.2.2 pgoyette static isc_httpdaction_t render_500;
218 1.2.2.2 pgoyette
219 1.2.2.2 pgoyette static void (*finishhook)(void) = NULL;
220 1.2.2.2 pgoyette
221 1.2.2.2 pgoyette static void
222 1.2.2.2 pgoyette destroy_client(isc_httpd_t **httpdp) {
223 1.2.2.2 pgoyette isc_httpd_t *httpd = *httpdp;
224 1.2.2.2 pgoyette isc_httpdmgr_t *httpdmgr = httpd->mgr;
225 1.2.2.2 pgoyette isc_region_t r;
226 1.2.2.2 pgoyette
227 1.2.2.2 pgoyette *httpdp = NULL;
228 1.2.2.2 pgoyette
229 1.2.2.2 pgoyette LOCK(&httpdmgr->lock);
230 1.2.2.2 pgoyette
231 1.2.2.2 pgoyette isc_socket_detach(&httpd->sock);
232 1.2.2.2 pgoyette ISC_LIST_UNLINK(httpdmgr->running, httpd, link);
233 1.2.2.2 pgoyette
234 1.2.2.2 pgoyette isc_buffer_region(&httpd->headerbuffer, &r);
235 1.2.2.2 pgoyette if (r.length > 0) {
236 1.2.2.2 pgoyette isc_mem_put(httpdmgr->mctx, r.base, r.length);
237 1.2.2.2 pgoyette }
238 1.2.2.2 pgoyette
239 1.2.2.2 pgoyette isc_buffer_region(&httpd->compbuffer, &r);
240 1.2.2.2 pgoyette if (r.length > 0) {
241 1.2.2.2 pgoyette isc_mem_put(httpdmgr->mctx, r.base, r.length);
242 1.2.2.2 pgoyette }
243 1.2.2.2 pgoyette
244 1.2.2.2 pgoyette isc_mem_put(httpdmgr->mctx, httpd, sizeof(isc_httpd_t));
245 1.2.2.2 pgoyette
246 1.2.2.2 pgoyette UNLOCK(&httpdmgr->lock);
247 1.2.2.2 pgoyette
248 1.2.2.2 pgoyette if (finishhook != NULL)
249 1.2.2.2 pgoyette finishhook();
250 1.2.2.2 pgoyette
251 1.2.2.2 pgoyette httpdmgr_destroy(httpdmgr);
252 1.2.2.2 pgoyette }
253 1.2.2.2 pgoyette
254 1.2.2.2 pgoyette isc_result_t
255 1.2.2.2 pgoyette isc_httpdmgr_create(isc_mem_t *mctx, isc_socket_t *sock, isc_task_t *task,
256 1.2.2.2 pgoyette isc_httpdclientok_t *client_ok,
257 1.2.2.2 pgoyette isc_httpdondestroy_t *ondestroy, void *cb_arg,
258 1.2.2.2 pgoyette isc_timermgr_t *tmgr, isc_httpdmgr_t **httpdmgrp)
259 1.2.2.2 pgoyette {
260 1.2.2.2 pgoyette isc_result_t result;
261 1.2.2.2 pgoyette isc_httpdmgr_t *httpdmgr;
262 1.2.2.2 pgoyette
263 1.2.2.2 pgoyette REQUIRE(mctx != NULL);
264 1.2.2.2 pgoyette REQUIRE(sock != NULL);
265 1.2.2.2 pgoyette REQUIRE(task != NULL);
266 1.2.2.2 pgoyette REQUIRE(tmgr != NULL);
267 1.2.2.2 pgoyette REQUIRE(httpdmgrp != NULL && *httpdmgrp == NULL);
268 1.2.2.2 pgoyette
269 1.2.2.2 pgoyette httpdmgr = isc_mem_get(mctx, sizeof(isc_httpdmgr_t));
270 1.2.2.2 pgoyette if (httpdmgr == NULL)
271 1.2.2.2 pgoyette return (ISC_R_NOMEMORY);
272 1.2.2.2 pgoyette
273 1.2.2.3 pgoyette isc_mutex_init(&httpdmgr->lock);
274 1.2.2.2 pgoyette httpdmgr->mctx = NULL;
275 1.2.2.2 pgoyette isc_mem_attach(mctx, &httpdmgr->mctx);
276 1.2.2.2 pgoyette httpdmgr->sock = NULL;
277 1.2.2.2 pgoyette isc_socket_attach(sock, &httpdmgr->sock);
278 1.2.2.2 pgoyette httpdmgr->task = NULL;
279 1.2.2.2 pgoyette isc_task_attach(task, &httpdmgr->task);
280 1.2.2.2 pgoyette httpdmgr->timermgr = tmgr; /* XXXMLG no attach function? */
281 1.2.2.2 pgoyette httpdmgr->client_ok = client_ok;
282 1.2.2.2 pgoyette httpdmgr->ondestroy = ondestroy;
283 1.2.2.2 pgoyette httpdmgr->cb_arg = cb_arg;
284 1.2.2.2 pgoyette httpdmgr->flags = 0;
285 1.2.2.2 pgoyette
286 1.2.2.2 pgoyette ISC_LIST_INIT(httpdmgr->running);
287 1.2.2.2 pgoyette ISC_LIST_INIT(httpdmgr->urls);
288 1.2.2.2 pgoyette
289 1.2.2.2 pgoyette /* XXXMLG ignore errors on isc_socket_listen() */
290 1.2.2.2 pgoyette result = isc_socket_listen(sock, SOMAXCONN);
291 1.2.2.2 pgoyette if (result != ISC_R_SUCCESS) {
292 1.2.2.2 pgoyette UNEXPECTED_ERROR(__FILE__, __LINE__,
293 1.2.2.2 pgoyette "isc_socket_listen() failed: %s",
294 1.2.2.2 pgoyette isc_result_totext(result));
295 1.2.2.2 pgoyette goto cleanup;
296 1.2.2.2 pgoyette }
297 1.2.2.2 pgoyette
298 1.2.2.2 pgoyette (void)isc_socket_filter(sock, "httpready");
299 1.2.2.2 pgoyette
300 1.2.2.2 pgoyette result = isc_socket_accept(sock, task, isc_httpd_accept, httpdmgr);
301 1.2.2.2 pgoyette if (result != ISC_R_SUCCESS)
302 1.2.2.2 pgoyette goto cleanup;
303 1.2.2.2 pgoyette
304 1.2.2.2 pgoyette httpdmgr->render_404 = render_404;
305 1.2.2.2 pgoyette httpdmgr->render_500 = render_500;
306 1.2.2.2 pgoyette
307 1.2.2.2 pgoyette *httpdmgrp = httpdmgr;
308 1.2.2.2 pgoyette return (ISC_R_SUCCESS);
309 1.2.2.2 pgoyette
310 1.2.2.2 pgoyette cleanup:
311 1.2.2.2 pgoyette isc_task_detach(&httpdmgr->task);
312 1.2.2.2 pgoyette isc_socket_detach(&httpdmgr->sock);
313 1.2.2.2 pgoyette isc_mem_detach(&httpdmgr->mctx);
314 1.2.2.3 pgoyette isc_mutex_destroy(&httpdmgr->lock);
315 1.2.2.2 pgoyette isc_mem_put(mctx, httpdmgr, sizeof(isc_httpdmgr_t));
316 1.2.2.2 pgoyette return (result);
317 1.2.2.2 pgoyette }
318 1.2.2.2 pgoyette
319 1.2.2.2 pgoyette static void
320 1.2.2.2 pgoyette httpdmgr_destroy(isc_httpdmgr_t *httpdmgr) {
321 1.2.2.2 pgoyette isc_mem_t *mctx;
322 1.2.2.2 pgoyette isc_httpdurl_t *url;
323 1.2.2.2 pgoyette
324 1.2.2.2 pgoyette ENTER("httpdmgr_destroy");
325 1.2.2.2 pgoyette
326 1.2.2.2 pgoyette LOCK(&httpdmgr->lock);
327 1.2.2.2 pgoyette
328 1.2.2.2 pgoyette if (!MSHUTTINGDOWN(httpdmgr)) {
329 1.2.2.2 pgoyette NOTICE("httpdmgr_destroy not shutting down yet");
330 1.2.2.2 pgoyette UNLOCK(&httpdmgr->lock);
331 1.2.2.2 pgoyette return;
332 1.2.2.2 pgoyette }
333 1.2.2.2 pgoyette
334 1.2.2.2 pgoyette /*
335 1.2.2.2 pgoyette * If all clients are not shut down, don't do anything yet.
336 1.2.2.2 pgoyette */
337 1.2.2.2 pgoyette if (!ISC_LIST_EMPTY(httpdmgr->running)) {
338 1.2.2.2 pgoyette NOTICE("httpdmgr_destroy clients still active");
339 1.2.2.2 pgoyette UNLOCK(&httpdmgr->lock);
340 1.2.2.2 pgoyette return;
341 1.2.2.2 pgoyette }
342 1.2.2.2 pgoyette
343 1.2.2.2 pgoyette NOTICE("httpdmgr_destroy detaching socket, task, and timermgr");
344 1.2.2.2 pgoyette
345 1.2.2.2 pgoyette isc_socket_detach(&httpdmgr->sock);
346 1.2.2.2 pgoyette isc_task_detach(&httpdmgr->task);
347 1.2.2.2 pgoyette httpdmgr->timermgr = NULL;
348 1.2.2.2 pgoyette
349 1.2.2.2 pgoyette /*
350 1.2.2.2 pgoyette * Clear out the list of all actions we know about. Just free the
351 1.2.2.2 pgoyette * memory.
352 1.2.2.2 pgoyette */
353 1.2.2.2 pgoyette url = ISC_LIST_HEAD(httpdmgr->urls);
354 1.2.2.2 pgoyette while (url != NULL) {
355 1.2.2.2 pgoyette isc_mem_free(httpdmgr->mctx, url->url);
356 1.2.2.2 pgoyette ISC_LIST_UNLINK(httpdmgr->urls, url, link);
357 1.2.2.2 pgoyette isc_mem_put(httpdmgr->mctx, url, sizeof(isc_httpdurl_t));
358 1.2.2.2 pgoyette url = ISC_LIST_HEAD(httpdmgr->urls);
359 1.2.2.2 pgoyette }
360 1.2.2.2 pgoyette
361 1.2.2.2 pgoyette UNLOCK(&httpdmgr->lock);
362 1.2.2.3 pgoyette isc_mutex_destroy(&httpdmgr->lock);
363 1.2.2.2 pgoyette
364 1.2.2.2 pgoyette if (httpdmgr->ondestroy != NULL)
365 1.2.2.2 pgoyette (httpdmgr->ondestroy)(httpdmgr->cb_arg);
366 1.2.2.2 pgoyette
367 1.2.2.2 pgoyette mctx = httpdmgr->mctx;
368 1.2.2.2 pgoyette isc_mem_putanddetach(&mctx, httpdmgr, sizeof(isc_httpdmgr_t));
369 1.2.2.2 pgoyette
370 1.2.2.2 pgoyette EXIT("httpdmgr_destroy");
371 1.2.2.2 pgoyette }
372 1.2.2.2 pgoyette
373 1.2.2.2 pgoyette #define LENGTHOK(s) (httpd->recvbuf - (s) < (int)httpd->recvlen)
374 1.2.2.2 pgoyette #define BUFLENOK(s) (httpd->recvbuf - (s) < HTTP_RECVLEN)
375 1.2.2.2 pgoyette
376 1.2.2.2 pgoyette /*
377 1.2.2.2 pgoyette * Look for the given header in headers.
378 1.2.2.2 pgoyette * If value is specified look for it terminated with a character in eov.
379 1.2.2.2 pgoyette */
380 1.2.2.3 pgoyette static bool
381 1.2.2.2 pgoyette have_header(isc_httpd_t *httpd, const char *header, const char *value,
382 1.2.2.2 pgoyette const char *eov)
383 1.2.2.2 pgoyette {
384 1.2.2.2 pgoyette char *cr, *nl, *h;
385 1.2.2.2 pgoyette size_t hlen, vlen = 0;
386 1.2.2.2 pgoyette
387 1.2.2.2 pgoyette h = httpd->headers;
388 1.2.2.2 pgoyette hlen = strlen(header);
389 1.2.2.2 pgoyette if (value != NULL) {
390 1.2.2.2 pgoyette INSIST(eov != NULL);
391 1.2.2.2 pgoyette vlen = strlen(value);
392 1.2.2.2 pgoyette }
393 1.2.2.2 pgoyette
394 1.2.2.2 pgoyette for (;;) {
395 1.2.2.2 pgoyette if (strncasecmp(h, header, hlen) != 0) {
396 1.2.2.2 pgoyette /*
397 1.2.2.2 pgoyette * Skip to next line;
398 1.2.2.2 pgoyette */
399 1.2.2.2 pgoyette cr = strchr(h, '\r');
400 1.2.2.2 pgoyette if (cr != NULL && cr[1] == '\n')
401 1.2.2.2 pgoyette cr++;
402 1.2.2.2 pgoyette nl = strchr(h, '\n');
403 1.2.2.2 pgoyette
404 1.2.2.2 pgoyette /* last header? */
405 1.2.2.2 pgoyette h = cr;
406 1.2.2.2 pgoyette if (h == NULL || (nl != NULL && nl < h))
407 1.2.2.2 pgoyette h = nl;
408 1.2.2.2 pgoyette if (h == NULL)
409 1.2.2.3 pgoyette return (false);
410 1.2.2.2 pgoyette h++;
411 1.2.2.2 pgoyette continue;
412 1.2.2.2 pgoyette }
413 1.2.2.2 pgoyette
414 1.2.2.2 pgoyette if (value == NULL)
415 1.2.2.3 pgoyette return (true);
416 1.2.2.2 pgoyette
417 1.2.2.2 pgoyette /*
418 1.2.2.2 pgoyette * Skip optional leading white space.
419 1.2.2.2 pgoyette */
420 1.2.2.2 pgoyette h += hlen;
421 1.2.2.2 pgoyette while (*h == ' ' || *h == '\t')
422 1.2.2.2 pgoyette h++;
423 1.2.2.2 pgoyette /*
424 1.2.2.2 pgoyette * Terminate token search on NULL or EOL.
425 1.2.2.2 pgoyette */
426 1.2.2.2 pgoyette while (*h != 0 && *h != '\r' && *h != '\n') {
427 1.2.2.2 pgoyette if (strncasecmp(h, value, vlen) == 0)
428 1.2.2.2 pgoyette if (strchr(eov, h[vlen]) != NULL)
429 1.2.2.3 pgoyette return (true);
430 1.2.2.2 pgoyette /*
431 1.2.2.2 pgoyette * Skip to next token.
432 1.2.2.2 pgoyette */
433 1.2.2.2 pgoyette h += strcspn(h, eov);
434 1.2.2.2 pgoyette if (h[0] == '\r' && h[1] == '\n')
435 1.2.2.2 pgoyette h++;
436 1.2.2.2 pgoyette if (h[0] != 0)
437 1.2.2.2 pgoyette h++;
438 1.2.2.2 pgoyette }
439 1.2.2.3 pgoyette return (false);
440 1.2.2.2 pgoyette }
441 1.2.2.2 pgoyette }
442 1.2.2.2 pgoyette
443 1.2.2.2 pgoyette static isc_result_t
444 1.2.2.2 pgoyette process_request(isc_httpd_t *httpd, int length) {
445 1.2.2.2 pgoyette char *s;
446 1.2.2.2 pgoyette char *p;
447 1.2.2.2 pgoyette int delim;
448 1.2.2.2 pgoyette
449 1.2.2.2 pgoyette ENTER("request");
450 1.2.2.2 pgoyette
451 1.2.2.2 pgoyette httpd->recvlen += length;
452 1.2.2.2 pgoyette
453 1.2.2.2 pgoyette httpd->recvbuf[httpd->recvlen] = 0;
454 1.2.2.2 pgoyette httpd->headers = NULL;
455 1.2.2.2 pgoyette
456 1.2.2.2 pgoyette /*
457 1.2.2.2 pgoyette * If we don't find a blank line in our buffer, return that we need
458 1.2.2.2 pgoyette * more data.
459 1.2.2.2 pgoyette */
460 1.2.2.2 pgoyette s = strstr(httpd->recvbuf, "\r\n\r\n");
461 1.2.2.2 pgoyette delim = 2;
462 1.2.2.2 pgoyette if (s == NULL) {
463 1.2.2.2 pgoyette s = strstr(httpd->recvbuf, "\n\n");
464 1.2.2.2 pgoyette delim = 1;
465 1.2.2.2 pgoyette }
466 1.2.2.2 pgoyette if (s == NULL)
467 1.2.2.2 pgoyette return (ISC_R_NOTFOUND);
468 1.2.2.2 pgoyette
469 1.2.2.2 pgoyette /*
470 1.2.2.2 pgoyette * NUL terminate request at the blank line.
471 1.2.2.2 pgoyette */
472 1.2.2.2 pgoyette s[delim] = 0;
473 1.2.2.2 pgoyette
474 1.2.2.2 pgoyette /*
475 1.2.2.2 pgoyette * Determine if this is a POST or GET method. Any other values will
476 1.2.2.2 pgoyette * cause an error to be returned.
477 1.2.2.2 pgoyette */
478 1.2.2.2 pgoyette if (strncmp(httpd->recvbuf, "GET ", 4) == 0) {
479 1.2.2.2 pgoyette httpd->method = ISC_HTTPD_METHODGET;
480 1.2.2.2 pgoyette p = httpd->recvbuf + 4;
481 1.2.2.2 pgoyette } else if (strncmp(httpd->recvbuf, "POST ", 5) == 0) {
482 1.2.2.2 pgoyette httpd->method = ISC_HTTPD_METHODPOST;
483 1.2.2.2 pgoyette p = httpd->recvbuf + 5;
484 1.2.2.2 pgoyette } else {
485 1.2.2.2 pgoyette return (ISC_R_RANGE);
486 1.2.2.2 pgoyette }
487 1.2.2.2 pgoyette
488 1.2.2.2 pgoyette /*
489 1.2.2.2 pgoyette * From now on, p is the start of our buffer.
490 1.2.2.2 pgoyette */
491 1.2.2.2 pgoyette
492 1.2.2.2 pgoyette /*
493 1.2.2.2 pgoyette * Extract the URL.
494 1.2.2.2 pgoyette */
495 1.2.2.2 pgoyette s = p;
496 1.2.2.2 pgoyette while (LENGTHOK(s) && BUFLENOK(s) &&
497 1.2.2.2 pgoyette (*s != '\n' && *s != '\r' && *s != '\0' && *s != ' '))
498 1.2.2.2 pgoyette s++;
499 1.2.2.2 pgoyette if (!LENGTHOK(s))
500 1.2.2.2 pgoyette return (ISC_R_NOTFOUND);
501 1.2.2.2 pgoyette if (!BUFLENOK(s))
502 1.2.2.2 pgoyette return (ISC_R_NOMEMORY);
503 1.2.2.2 pgoyette *s = 0;
504 1.2.2.2 pgoyette
505 1.2.2.2 pgoyette /*
506 1.2.2.2 pgoyette * Make the URL relative.
507 1.2.2.2 pgoyette */
508 1.2.2.2 pgoyette if ((strncmp(p, "http:/", 6) == 0)
509 1.2.2.2 pgoyette || (strncmp(p, "https:/", 7) == 0)) {
510 1.2.2.2 pgoyette /* Skip first / */
511 1.2.2.2 pgoyette while (*p != '/' && *p != 0)
512 1.2.2.2 pgoyette p++;
513 1.2.2.2 pgoyette if (*p == 0)
514 1.2.2.2 pgoyette return (ISC_R_RANGE);
515 1.2.2.2 pgoyette p++;
516 1.2.2.2 pgoyette /* Skip second / */
517 1.2.2.2 pgoyette while (*p != '/' && *p != 0)
518 1.2.2.2 pgoyette p++;
519 1.2.2.2 pgoyette if (*p == 0)
520 1.2.2.2 pgoyette return (ISC_R_RANGE);
521 1.2.2.2 pgoyette p++;
522 1.2.2.2 pgoyette /* Find third / */
523 1.2.2.2 pgoyette while (*p != '/' && *p != 0)
524 1.2.2.2 pgoyette p++;
525 1.2.2.2 pgoyette if (*p == 0) {
526 1.2.2.2 pgoyette p--;
527 1.2.2.2 pgoyette *p = '/';
528 1.2.2.2 pgoyette }
529 1.2.2.2 pgoyette }
530 1.2.2.2 pgoyette
531 1.2.2.2 pgoyette httpd->url = p;
532 1.2.2.2 pgoyette p = s + 1;
533 1.2.2.2 pgoyette s = p;
534 1.2.2.2 pgoyette
535 1.2.2.2 pgoyette /*
536 1.2.2.2 pgoyette * Now, see if there is a ? mark in the URL. If so, this is
537 1.2.2.2 pgoyette * part of the query string, and we will split it from the URL.
538 1.2.2.2 pgoyette */
539 1.2.2.2 pgoyette httpd->querystring = strchr(httpd->url, '?');
540 1.2.2.2 pgoyette if (httpd->querystring != NULL) {
541 1.2.2.2 pgoyette *(httpd->querystring) = 0;
542 1.2.2.2 pgoyette httpd->querystring++;
543 1.2.2.2 pgoyette }
544 1.2.2.2 pgoyette
545 1.2.2.2 pgoyette /*
546 1.2.2.2 pgoyette * Extract the HTTP/1.X protocol. We will bounce on anything but
547 1.2.2.2 pgoyette * HTTP/1.0 or HTTP/1.1 for now.
548 1.2.2.2 pgoyette */
549 1.2.2.2 pgoyette while (LENGTHOK(s) && BUFLENOK(s) &&
550 1.2.2.2 pgoyette (*s != '\n' && *s != '\r' && *s != '\0'))
551 1.2.2.2 pgoyette s++;
552 1.2.2.2 pgoyette if (!LENGTHOK(s))
553 1.2.2.2 pgoyette return (ISC_R_NOTFOUND);
554 1.2.2.2 pgoyette if (!BUFLENOK(s))
555 1.2.2.2 pgoyette return (ISC_R_NOMEMORY);
556 1.2.2.2 pgoyette /*
557 1.2.2.2 pgoyette * Check that we have the expected eol delimiter.
558 1.2.2.2 pgoyette */
559 1.2.2.2 pgoyette if (strncmp(s, delim == 1 ? "\n" : "\r\n", delim) != 0)
560 1.2.2.2 pgoyette return (ISC_R_RANGE);
561 1.2.2.2 pgoyette *s = 0;
562 1.2.2.2 pgoyette if ((strncmp(p, "HTTP/1.0", 8) != 0)
563 1.2.2.2 pgoyette && (strncmp(p, "HTTP/1.1", 8) != 0))
564 1.2.2.2 pgoyette return (ISC_R_RANGE);
565 1.2.2.2 pgoyette httpd->protocol = p;
566 1.2.2.2 pgoyette p = s + delim; /* skip past eol */
567 1.2.2.2 pgoyette s = p;
568 1.2.2.2 pgoyette
569 1.2.2.2 pgoyette httpd->headers = s;
570 1.2.2.2 pgoyette
571 1.2.2.2 pgoyette if (have_header(httpd, "Connection:", "close", ", \t\r\n"))
572 1.2.2.2 pgoyette httpd->flags |= HTTPD_CLOSE;
573 1.2.2.2 pgoyette
574 1.2.2.2 pgoyette if (have_header(httpd, "Host:", NULL, NULL))
575 1.2.2.2 pgoyette httpd->flags |= HTTPD_FOUNDHOST;
576 1.2.2.2 pgoyette
577 1.2.2.2 pgoyette if (strncmp(httpd->protocol, "HTTP/1.0", 8) == 0) {
578 1.2.2.2 pgoyette if (have_header(httpd, "Connection:", "Keep-Alive",
579 1.2.2.2 pgoyette ", \t\r\n"))
580 1.2.2.2 pgoyette httpd->flags |= HTTPD_KEEPALIVE;
581 1.2.2.2 pgoyette else
582 1.2.2.2 pgoyette httpd->flags |= HTTPD_CLOSE;
583 1.2.2.2 pgoyette }
584 1.2.2.2 pgoyette
585 1.2.2.2 pgoyette /*
586 1.2.2.2 pgoyette * Check for Accept-Encoding:
587 1.2.2.2 pgoyette */
588 1.2.2.2 pgoyette #ifdef HAVE_ZLIB
589 1.2.2.2 pgoyette if (have_header(httpd, "Accept-Encoding:", "deflate", ";, \t\r\n"))
590 1.2.2.2 pgoyette httpd->flags |= HTTPD_ACCEPT_DEFLATE;
591 1.2.2.2 pgoyette #endif
592 1.2.2.2 pgoyette
593 1.2.2.2 pgoyette /*
594 1.2.2.2 pgoyette * Standards compliance hooks here.
595 1.2.2.2 pgoyette */
596 1.2.2.2 pgoyette if (strcmp(httpd->protocol, "HTTP/1.1") == 0
597 1.2.2.2 pgoyette && ((httpd->flags & HTTPD_FOUNDHOST) == 0))
598 1.2.2.2 pgoyette return (ISC_R_RANGE);
599 1.2.2.2 pgoyette
600 1.2.2.2 pgoyette EXIT("request");
601 1.2.2.2 pgoyette
602 1.2.2.2 pgoyette return (ISC_R_SUCCESS);
603 1.2.2.2 pgoyette }
604 1.2.2.2 pgoyette
605 1.2.2.2 pgoyette static void
606 1.2.2.2 pgoyette isc_httpd_accept(isc_task_t *task, isc_event_t *ev) {
607 1.2.2.2 pgoyette isc_result_t result;
608 1.2.2.2 pgoyette isc_httpdmgr_t *httpdmgr = ev->ev_arg;
609 1.2.2.2 pgoyette isc_httpd_t *httpd;
610 1.2.2.2 pgoyette isc_region_t r;
611 1.2.2.2 pgoyette isc_socket_newconnev_t *nev = (isc_socket_newconnev_t *)ev;
612 1.2.2.2 pgoyette isc_sockaddr_t peeraddr;
613 1.2.2.2 pgoyette char *headerdata;
614 1.2.2.2 pgoyette
615 1.2.2.2 pgoyette ENTER("accept");
616 1.2.2.2 pgoyette
617 1.2.2.2 pgoyette LOCK(&httpdmgr->lock);
618 1.2.2.2 pgoyette if (MSHUTTINGDOWN(httpdmgr)) {
619 1.2.2.2 pgoyette NOTICE("accept shutting down, goto out");
620 1.2.2.2 pgoyette goto out;
621 1.2.2.2 pgoyette }
622 1.2.2.2 pgoyette
623 1.2.2.2 pgoyette if (nev->result == ISC_R_CANCELED) {
624 1.2.2.2 pgoyette NOTICE("accept canceled, goto out");
625 1.2.2.2 pgoyette goto out;
626 1.2.2.2 pgoyette }
627 1.2.2.2 pgoyette
628 1.2.2.2 pgoyette if (nev->result != ISC_R_SUCCESS) {
629 1.2.2.2 pgoyette /* XXXMLG log failure */
630 1.2.2.2 pgoyette NOTICE("accept returned failure, goto requeue");
631 1.2.2.2 pgoyette goto requeue;
632 1.2.2.2 pgoyette }
633 1.2.2.2 pgoyette
634 1.2.2.2 pgoyette (void)isc_socket_getpeername(nev->newsocket, &peeraddr);
635 1.2.2.2 pgoyette if (httpdmgr->client_ok != NULL &&
636 1.2.2.2 pgoyette !(httpdmgr->client_ok)(&peeraddr, httpdmgr->cb_arg)) {
637 1.2.2.2 pgoyette isc_socket_detach(&nev->newsocket);
638 1.2.2.2 pgoyette goto requeue;
639 1.2.2.2 pgoyette }
640 1.2.2.2 pgoyette
641 1.2.2.2 pgoyette httpd = isc_mem_get(httpdmgr->mctx, sizeof(isc_httpd_t));
642 1.2.2.2 pgoyette if (httpd == NULL) {
643 1.2.2.2 pgoyette /* XXXMLG log failure */
644 1.2.2.2 pgoyette NOTICE("accept failed to allocate memory, goto requeue");
645 1.2.2.2 pgoyette isc_socket_detach(&nev->newsocket);
646 1.2.2.2 pgoyette goto requeue;
647 1.2.2.2 pgoyette }
648 1.2.2.2 pgoyette
649 1.2.2.2 pgoyette httpd->mgr = httpdmgr;
650 1.2.2.2 pgoyette ISC_LINK_INIT(httpd, link);
651 1.2.2.2 pgoyette ISC_LIST_APPEND(httpdmgr->running, httpd, link);
652 1.2.2.2 pgoyette ISC_HTTPD_SETRECV(httpd);
653 1.2.2.2 pgoyette httpd->sock = nev->newsocket;
654 1.2.2.2 pgoyette isc_socket_setname(httpd->sock, "httpd", NULL);
655 1.2.2.2 pgoyette httpd->flags = 0;
656 1.2.2.2 pgoyette
657 1.2.2.2 pgoyette /*
658 1.2.2.2 pgoyette * Initialize the buffer for our headers.
659 1.2.2.2 pgoyette */
660 1.2.2.2 pgoyette headerdata = isc_mem_get(httpdmgr->mctx, HTTP_SENDGROW);
661 1.2.2.2 pgoyette if (headerdata == NULL) {
662 1.2.2.2 pgoyette isc_mem_put(httpdmgr->mctx, httpd, sizeof(isc_httpd_t));
663 1.2.2.2 pgoyette isc_socket_detach(&nev->newsocket);
664 1.2.2.2 pgoyette goto requeue;
665 1.2.2.2 pgoyette }
666 1.2.2.2 pgoyette isc_buffer_init(&httpd->headerbuffer, headerdata, HTTP_SENDGROW);
667 1.2.2.2 pgoyette
668 1.2.2.2 pgoyette isc_buffer_initnull(&httpd->compbuffer);
669 1.2.2.2 pgoyette isc_buffer_initnull(&httpd->bodybuffer);
670 1.2.2.3 pgoyette httpd->sendbuffer = NULL;
671 1.2.2.2 pgoyette reset_client(httpd);
672 1.2.2.2 pgoyette
673 1.2.2.2 pgoyette r.base = (unsigned char *)httpd->recvbuf;
674 1.2.2.2 pgoyette r.length = HTTP_RECVLEN - 1;
675 1.2.2.2 pgoyette result = isc_socket_recv(httpd->sock, &r, 1, task, isc_httpd_recvdone,
676 1.2.2.2 pgoyette httpd);
677 1.2.2.2 pgoyette /* FIXME!!! */
678 1.2.2.2 pgoyette POST(result);
679 1.2.2.2 pgoyette NOTICE("accept queued recv on socket");
680 1.2.2.2 pgoyette
681 1.2.2.2 pgoyette requeue:
682 1.2.2.2 pgoyette result = isc_socket_accept(httpdmgr->sock, task, isc_httpd_accept,
683 1.2.2.2 pgoyette httpdmgr);
684 1.2.2.2 pgoyette if (result != ISC_R_SUCCESS) {
685 1.2.2.2 pgoyette /* XXXMLG what to do? Log failure... */
686 1.2.2.2 pgoyette NOTICE("accept could not reaccept due to failure");
687 1.2.2.2 pgoyette }
688 1.2.2.2 pgoyette
689 1.2.2.2 pgoyette out:
690 1.2.2.2 pgoyette UNLOCK(&httpdmgr->lock);
691 1.2.2.2 pgoyette
692 1.2.2.2 pgoyette httpdmgr_destroy(httpdmgr);
693 1.2.2.2 pgoyette
694 1.2.2.2 pgoyette isc_event_free(&ev);
695 1.2.2.2 pgoyette
696 1.2.2.2 pgoyette EXIT("accept");
697 1.2.2.2 pgoyette }
698 1.2.2.2 pgoyette
699 1.2.2.2 pgoyette static isc_result_t
700 1.2.2.2 pgoyette render_404(const char *url, isc_httpdurl_t *urlinfo,
701 1.2.2.2 pgoyette const char *querystring, const char *headers, void *arg,
702 1.2.2.2 pgoyette unsigned int *retcode, const char **retmsg,
703 1.2.2.2 pgoyette const char **mimetype, isc_buffer_t *b,
704 1.2.2.2 pgoyette isc_httpdfree_t **freecb, void **freecb_args)
705 1.2.2.2 pgoyette {
706 1.2.2.2 pgoyette static char msg[] = "No such URL.\r\n";
707 1.2.2.2 pgoyette
708 1.2.2.2 pgoyette UNUSED(url);
709 1.2.2.2 pgoyette UNUSED(urlinfo);
710 1.2.2.2 pgoyette UNUSED(querystring);
711 1.2.2.2 pgoyette UNUSED(headers);
712 1.2.2.2 pgoyette UNUSED(arg);
713 1.2.2.2 pgoyette
714 1.2.2.2 pgoyette *retcode = 404;
715 1.2.2.2 pgoyette *retmsg = "No such URL";
716 1.2.2.2 pgoyette *mimetype = "text/plain";
717 1.2.2.2 pgoyette isc_buffer_reinit(b, msg, strlen(msg));
718 1.2.2.2 pgoyette isc_buffer_add(b, strlen(msg));
719 1.2.2.2 pgoyette *freecb = NULL;
720 1.2.2.2 pgoyette *freecb_args = NULL;
721 1.2.2.2 pgoyette
722 1.2.2.2 pgoyette return (ISC_R_SUCCESS);
723 1.2.2.2 pgoyette }
724 1.2.2.2 pgoyette
725 1.2.2.2 pgoyette static isc_result_t
726 1.2.2.2 pgoyette render_500(const char *url, isc_httpdurl_t *urlinfo,
727 1.2.2.2 pgoyette const char *querystring, const char *headers, void *arg,
728 1.2.2.2 pgoyette unsigned int *retcode, const char **retmsg,
729 1.2.2.2 pgoyette const char **mimetype, isc_buffer_t *b,
730 1.2.2.2 pgoyette isc_httpdfree_t **freecb, void **freecb_args)
731 1.2.2.2 pgoyette {
732 1.2.2.2 pgoyette static char msg[] = "Internal server failure.\r\n";
733 1.2.2.2 pgoyette
734 1.2.2.2 pgoyette UNUSED(url);
735 1.2.2.2 pgoyette UNUSED(urlinfo);
736 1.2.2.2 pgoyette UNUSED(querystring);
737 1.2.2.2 pgoyette UNUSED(headers);
738 1.2.2.2 pgoyette UNUSED(arg);
739 1.2.2.2 pgoyette
740 1.2.2.2 pgoyette *retcode = 500;
741 1.2.2.2 pgoyette *retmsg = "Internal server failure";
742 1.2.2.2 pgoyette *mimetype = "text/plain";
743 1.2.2.2 pgoyette isc_buffer_reinit(b, msg, strlen(msg));
744 1.2.2.2 pgoyette isc_buffer_add(b, strlen(msg));
745 1.2.2.2 pgoyette *freecb = NULL;
746 1.2.2.2 pgoyette *freecb_args = NULL;
747 1.2.2.2 pgoyette
748 1.2.2.2 pgoyette return (ISC_R_SUCCESS);
749 1.2.2.2 pgoyette }
750 1.2.2.2 pgoyette
751 1.2.2.2 pgoyette #ifdef HAVE_ZLIB
752 1.2.2.2 pgoyette /*%<
753 1.2.2.2 pgoyette * Reallocates compbuffer to size, does nothing if compbuffer is already
754 1.2.2.2 pgoyette * larger than size.
755 1.2.2.2 pgoyette *
756 1.2.2.2 pgoyette * Requires:
757 1.2.2.2 pgoyette *\li httpd a valid isc_httpd_t object
758 1.2.2.2 pgoyette *
759 1.2.2.2 pgoyette * Returns:
760 1.2.2.2 pgoyette *\li #ISC_R_SUCCESS -- all is well.
761 1.2.2.2 pgoyette *\li #ISC_R_NOMEMORY -- not enough memory to extend buffer
762 1.2.2.2 pgoyette */
763 1.2.2.2 pgoyette static isc_result_t
764 1.2.2.2 pgoyette alloc_compspace(isc_httpd_t *httpd, unsigned int size) {
765 1.2.2.2 pgoyette char *newspace;
766 1.2.2.2 pgoyette isc_region_t r;
767 1.2.2.2 pgoyette
768 1.2.2.2 pgoyette isc_buffer_region(&httpd->compbuffer, &r);
769 1.2.2.2 pgoyette if (size < r.length) {
770 1.2.2.2 pgoyette return (ISC_R_SUCCESS);
771 1.2.2.2 pgoyette }
772 1.2.2.2 pgoyette
773 1.2.2.2 pgoyette newspace = isc_mem_get(httpd->mgr->mctx, size);
774 1.2.2.2 pgoyette if (newspace == NULL)
775 1.2.2.2 pgoyette return (ISC_R_NOMEMORY);
776 1.2.2.2 pgoyette isc_buffer_reinit(&httpd->compbuffer, newspace, size);
777 1.2.2.2 pgoyette
778 1.2.2.2 pgoyette if (r.base != NULL) {
779 1.2.2.2 pgoyette isc_mem_put(httpd->mgr->mctx, r.base, r.length);
780 1.2.2.2 pgoyette }
781 1.2.2.2 pgoyette
782 1.2.2.2 pgoyette return (ISC_R_SUCCESS);
783 1.2.2.2 pgoyette }
784 1.2.2.2 pgoyette
785 1.2.2.2 pgoyette /*%<
786 1.2.2.2 pgoyette * Tries to compress httpd->bodybuffer to httpd->compbuffer, extending it
787 1.2.2.2 pgoyette * if necessary.
788 1.2.2.2 pgoyette *
789 1.2.2.2 pgoyette * Requires:
790 1.2.2.2 pgoyette *\li httpd a valid isc_httpd_t object
791 1.2.2.2 pgoyette *
792 1.2.2.2 pgoyette * Returns:
793 1.2.2.2 pgoyette *\li #ISC_R_SUCCESS -- all is well.
794 1.2.2.2 pgoyette *\li #ISC_R_NOMEMORY -- not enough memory to compress data
795 1.2.2.2 pgoyette *\li #ISC_R_FAILURE -- error during compression or compressed
796 1.2.2.2 pgoyette * data would be larger than input data
797 1.2.2.2 pgoyette */
798 1.2.2.2 pgoyette static isc_result_t
799 1.2.2.2 pgoyette isc_httpd_compress(isc_httpd_t *httpd) {
800 1.2.2.2 pgoyette z_stream zstr;
801 1.2.2.2 pgoyette isc_region_t r;
802 1.2.2.2 pgoyette isc_result_t result;
803 1.2.2.2 pgoyette int ret;
804 1.2.2.2 pgoyette int inputlen;
805 1.2.2.2 pgoyette
806 1.2.2.2 pgoyette inputlen = isc_buffer_usedlength(&httpd->bodybuffer);
807 1.2.2.2 pgoyette result = alloc_compspace(httpd, inputlen);
808 1.2.2.2 pgoyette if (result != ISC_R_SUCCESS) {
809 1.2.2.2 pgoyette return result;
810 1.2.2.2 pgoyette }
811 1.2.2.2 pgoyette isc_buffer_region(&httpd->compbuffer, &r);
812 1.2.2.2 pgoyette
813 1.2.2.2 pgoyette /*
814 1.2.2.2 pgoyette * We're setting output buffer size to input size so it fails if the
815 1.2.2.2 pgoyette * compressed data size would be bigger than the input size.
816 1.2.2.2 pgoyette */
817 1.2.2.2 pgoyette memset(&zstr, 0, sizeof(zstr));
818 1.2.2.2 pgoyette zstr.total_in = zstr.avail_in =
819 1.2.2.2 pgoyette zstr.total_out = zstr.avail_out = inputlen;
820 1.2.2.2 pgoyette
821 1.2.2.2 pgoyette zstr.next_in = isc_buffer_base(&httpd->bodybuffer);
822 1.2.2.2 pgoyette zstr.next_out = r.base;
823 1.2.2.2 pgoyette
824 1.2.2.2 pgoyette ret = deflateInit(&zstr, Z_DEFAULT_COMPRESSION);
825 1.2.2.2 pgoyette if (ret == Z_OK) {
826 1.2.2.2 pgoyette ret = deflate(&zstr, Z_FINISH);
827 1.2.2.2 pgoyette }
828 1.2.2.2 pgoyette deflateEnd(&zstr);
829 1.2.2.2 pgoyette if (ret == Z_STREAM_END) {
830 1.2.2.2 pgoyette isc_buffer_add(&httpd->compbuffer, inputlen - zstr.avail_out);
831 1.2.2.2 pgoyette return (ISC_R_SUCCESS);
832 1.2.2.2 pgoyette } else {
833 1.2.2.2 pgoyette return (ISC_R_FAILURE);
834 1.2.2.2 pgoyette }
835 1.2.2.2 pgoyette }
836 1.2.2.2 pgoyette #endif
837 1.2.2.2 pgoyette
838 1.2.2.2 pgoyette static void
839 1.2.2.2 pgoyette isc_httpd_recvdone(isc_task_t *task, isc_event_t *ev) {
840 1.2.2.2 pgoyette isc_result_t result;
841 1.2.2.2 pgoyette isc_httpd_t *httpd = ev->ev_arg;
842 1.2.2.2 pgoyette isc_socketevent_t *sev = (isc_socketevent_t *)ev;
843 1.2.2.3 pgoyette isc_buffer_t *databuffer;
844 1.2.2.2 pgoyette isc_httpdurl_t *url;
845 1.2.2.2 pgoyette isc_time_t now;
846 1.2.2.3 pgoyette isc_region_t r;
847 1.2.2.3 pgoyette bool is_compressed = false;
848 1.2.2.2 pgoyette char datebuf[ISC_FORMATHTTPTIMESTAMP_SIZE];
849 1.2.2.2 pgoyette
850 1.2.2.2 pgoyette ENTER("recv");
851 1.2.2.2 pgoyette
852 1.2.2.2 pgoyette INSIST(ISC_HTTPD_ISRECV(httpd));
853 1.2.2.2 pgoyette
854 1.2.2.2 pgoyette if (sev->result != ISC_R_SUCCESS) {
855 1.2.2.2 pgoyette NOTICE("recv destroying client");
856 1.2.2.2 pgoyette destroy_client(&httpd);
857 1.2.2.2 pgoyette goto out;
858 1.2.2.2 pgoyette }
859 1.2.2.2 pgoyette
860 1.2.2.2 pgoyette result = process_request(httpd, sev->n);
861 1.2.2.2 pgoyette if (result == ISC_R_NOTFOUND) {
862 1.2.2.2 pgoyette if (httpd->recvlen >= HTTP_RECVLEN - 1) {
863 1.2.2.2 pgoyette destroy_client(&httpd);
864 1.2.2.2 pgoyette goto out;
865 1.2.2.2 pgoyette }
866 1.2.2.2 pgoyette r.base = (unsigned char *)httpd->recvbuf + httpd->recvlen;
867 1.2.2.2 pgoyette r.length = HTTP_RECVLEN - httpd->recvlen - 1;
868 1.2.2.2 pgoyette /* check return code? */
869 1.2.2.2 pgoyette (void)isc_socket_recv(httpd->sock, &r, 1, task,
870 1.2.2.2 pgoyette isc_httpd_recvdone, httpd);
871 1.2.2.2 pgoyette goto out;
872 1.2.2.2 pgoyette } else if (result != ISC_R_SUCCESS) {
873 1.2.2.2 pgoyette destroy_client(&httpd);
874 1.2.2.2 pgoyette goto out;
875 1.2.2.2 pgoyette }
876 1.2.2.2 pgoyette
877 1.2.2.2 pgoyette ISC_HTTPD_SETSEND(httpd);
878 1.2.2.2 pgoyette
879 1.2.2.2 pgoyette /*
880 1.2.2.2 pgoyette * XXXMLG Call function here. Provide an add-header function
881 1.2.2.2 pgoyette * which will append the common headers to a response we generate.
882 1.2.2.2 pgoyette */
883 1.2.2.2 pgoyette isc_buffer_initnull(&httpd->bodybuffer);
884 1.2.2.2 pgoyette isc_time_now(&now);
885 1.2.2.2 pgoyette isc_time_formathttptimestamp(&now, datebuf, sizeof(datebuf));
886 1.2.2.2 pgoyette url = ISC_LIST_HEAD(httpd->mgr->urls);
887 1.2.2.2 pgoyette while (url != NULL) {
888 1.2.2.2 pgoyette if (strcmp(httpd->url, url->url) == 0)
889 1.2.2.2 pgoyette break;
890 1.2.2.2 pgoyette url = ISC_LIST_NEXT(url, link);
891 1.2.2.2 pgoyette }
892 1.2.2.2 pgoyette if (url == NULL)
893 1.2.2.2 pgoyette result = httpd->mgr->render_404(httpd->url, NULL,
894 1.2.2.2 pgoyette httpd->querystring,
895 1.2.2.2 pgoyette NULL, NULL,
896 1.2.2.2 pgoyette &httpd->retcode,
897 1.2.2.2 pgoyette &httpd->retmsg,
898 1.2.2.2 pgoyette &httpd->mimetype,
899 1.2.2.2 pgoyette &httpd->bodybuffer,
900 1.2.2.2 pgoyette &httpd->freecb,
901 1.2.2.2 pgoyette &httpd->freecb_arg);
902 1.2.2.2 pgoyette else
903 1.2.2.2 pgoyette result = url->action(httpd->url, url,
904 1.2.2.2 pgoyette httpd->querystring,
905 1.2.2.2 pgoyette httpd->headers,
906 1.2.2.2 pgoyette url->action_arg,
907 1.2.2.2 pgoyette &httpd->retcode, &httpd->retmsg,
908 1.2.2.2 pgoyette &httpd->mimetype, &httpd->bodybuffer,
909 1.2.2.2 pgoyette &httpd->freecb, &httpd->freecb_arg);
910 1.2.2.2 pgoyette if (result != ISC_R_SUCCESS) {
911 1.2.2.2 pgoyette result = httpd->mgr->render_500(httpd->url, url,
912 1.2.2.2 pgoyette httpd->querystring,
913 1.2.2.2 pgoyette NULL, NULL,
914 1.2.2.2 pgoyette &httpd->retcode,
915 1.2.2.2 pgoyette &httpd->retmsg,
916 1.2.2.2 pgoyette &httpd->mimetype,
917 1.2.2.2 pgoyette &httpd->bodybuffer,
918 1.2.2.2 pgoyette &httpd->freecb,
919 1.2.2.2 pgoyette &httpd->freecb_arg);
920 1.2.2.2 pgoyette RUNTIME_CHECK(result == ISC_R_SUCCESS);
921 1.2.2.2 pgoyette }
922 1.2.2.2 pgoyette
923 1.2.2.2 pgoyette #ifdef HAVE_ZLIB
924 1.2.2.3 pgoyette if ((httpd->flags & HTTPD_ACCEPT_DEFLATE) != 0) {
925 1.2.2.2 pgoyette result = isc_httpd_compress(httpd);
926 1.2.2.2 pgoyette if (result == ISC_R_SUCCESS) {
927 1.2.2.3 pgoyette is_compressed = true;
928 1.2.2.2 pgoyette }
929 1.2.2.2 pgoyette }
930 1.2.2.2 pgoyette #endif
931 1.2.2.2 pgoyette
932 1.2.2.2 pgoyette isc_httpd_response(httpd);
933 1.2.2.2 pgoyette if ((httpd->flags & HTTPD_KEEPALIVE) != 0)
934 1.2.2.2 pgoyette isc_httpd_addheader(httpd, "Connection", "Keep-Alive");
935 1.2.2.2 pgoyette isc_httpd_addheader(httpd, "Content-Type", httpd->mimetype);
936 1.2.2.2 pgoyette isc_httpd_addheader(httpd, "Date", datebuf);
937 1.2.2.2 pgoyette isc_httpd_addheader(httpd, "Expires", datebuf);
938 1.2.2.2 pgoyette
939 1.2.2.2 pgoyette if (url != NULL && url->isstatic) {
940 1.2.2.2 pgoyette char loadbuf[ISC_FORMATHTTPTIMESTAMP_SIZE];
941 1.2.2.2 pgoyette isc_time_formathttptimestamp(&url->loadtime,
942 1.2.2.2 pgoyette loadbuf, sizeof(loadbuf));
943 1.2.2.2 pgoyette isc_httpd_addheader(httpd, "Last-Modified", loadbuf);
944 1.2.2.2 pgoyette isc_httpd_addheader(httpd, "Cache-Control: public", NULL);
945 1.2.2.2 pgoyette } else {
946 1.2.2.2 pgoyette isc_httpd_addheader(httpd, "Last-Modified", datebuf);
947 1.2.2.2 pgoyette isc_httpd_addheader(httpd, "Pragma: no-cache", NULL);
948 1.2.2.2 pgoyette isc_httpd_addheader(httpd, "Cache-Control: no-cache", NULL);
949 1.2.2.2 pgoyette }
950 1.2.2.2 pgoyette
951 1.2.2.2 pgoyette isc_httpd_addheader(httpd, "Server: libisc", NULL);
952 1.2.2.2 pgoyette
953 1.2.2.3 pgoyette if (is_compressed == true) {
954 1.2.2.2 pgoyette isc_httpd_addheader(httpd, "Content-Encoding", "deflate");
955 1.2.2.2 pgoyette isc_httpd_addheaderuint(httpd, "Content-Length",
956 1.2.2.2 pgoyette isc_buffer_usedlength(&httpd->compbuffer));
957 1.2.2.2 pgoyette } else {
958 1.2.2.2 pgoyette isc_httpd_addheaderuint(httpd, "Content-Length",
959 1.2.2.2 pgoyette isc_buffer_usedlength(&httpd->bodybuffer));
960 1.2.2.2 pgoyette }
961 1.2.2.2 pgoyette
962 1.2.2.2 pgoyette isc_httpd_endheaders(httpd); /* done */
963 1.2.2.2 pgoyette
964 1.2.2.2 pgoyette /*
965 1.2.2.3 pgoyette * Append either the compressed or the non-compressed response body to
966 1.2.2.3 pgoyette * the response headers and store the result in httpd->sendbuffer.
967 1.2.2.2 pgoyette */
968 1.2.2.3 pgoyette isc_buffer_dup(httpd->mgr->mctx,
969 1.2.2.3 pgoyette &httpd->sendbuffer, &httpd->headerbuffer);
970 1.2.2.3 pgoyette isc_buffer_setautorealloc(httpd->sendbuffer, true);
971 1.2.2.3 pgoyette databuffer = (is_compressed ? &httpd->compbuffer : &httpd->bodybuffer);
972 1.2.2.3 pgoyette isc_buffer_usedregion(databuffer, &r);
973 1.2.2.3 pgoyette result = isc_buffer_copyregion(httpd->sendbuffer, &r);
974 1.2.2.3 pgoyette RUNTIME_CHECK(result == ISC_R_SUCCESS);
975 1.2.2.3 pgoyette /*
976 1.2.2.3 pgoyette * Determine total response size.
977 1.2.2.3 pgoyette */
978 1.2.2.3 pgoyette isc_buffer_usedregion(httpd->sendbuffer, &r);
979 1.2.2.2 pgoyette
980 1.2.2.2 pgoyette /* check return code? */
981 1.2.2.3 pgoyette (void)isc_socket_send(httpd->sock, &r, task,
982 1.2.2.2 pgoyette isc_httpd_senddone, httpd);
983 1.2.2.2 pgoyette
984 1.2.2.2 pgoyette out:
985 1.2.2.2 pgoyette isc_event_free(&ev);
986 1.2.2.2 pgoyette EXIT("recv");
987 1.2.2.2 pgoyette }
988 1.2.2.2 pgoyette
989 1.2.2.2 pgoyette void
990 1.2.2.2 pgoyette isc_httpdmgr_shutdown(isc_httpdmgr_t **httpdmgrp) {
991 1.2.2.2 pgoyette isc_httpdmgr_t *httpdmgr;
992 1.2.2.2 pgoyette isc_httpd_t *httpd;
993 1.2.2.2 pgoyette httpdmgr = *httpdmgrp;
994 1.2.2.2 pgoyette *httpdmgrp = NULL;
995 1.2.2.2 pgoyette
996 1.2.2.2 pgoyette ENTER("isc_httpdmgr_shutdown");
997 1.2.2.2 pgoyette
998 1.2.2.2 pgoyette LOCK(&httpdmgr->lock);
999 1.2.2.2 pgoyette
1000 1.2.2.2 pgoyette MSETSHUTTINGDOWN(httpdmgr);
1001 1.2.2.2 pgoyette
1002 1.2.2.2 pgoyette isc_socket_cancel(httpdmgr->sock, httpdmgr->task, ISC_SOCKCANCEL_ALL);
1003 1.2.2.2 pgoyette
1004 1.2.2.2 pgoyette httpd = ISC_LIST_HEAD(httpdmgr->running);
1005 1.2.2.2 pgoyette while (httpd != NULL) {
1006 1.2.2.2 pgoyette isc_socket_cancel(httpd->sock, httpdmgr->task,
1007 1.2.2.2 pgoyette ISC_SOCKCANCEL_ALL);
1008 1.2.2.2 pgoyette httpd = ISC_LIST_NEXT(httpd, link);
1009 1.2.2.2 pgoyette }
1010 1.2.2.2 pgoyette
1011 1.2.2.2 pgoyette UNLOCK(&httpdmgr->lock);
1012 1.2.2.2 pgoyette
1013 1.2.2.2 pgoyette EXIT("isc_httpdmgr_shutdown");
1014 1.2.2.2 pgoyette }
1015 1.2.2.2 pgoyette
1016 1.2.2.2 pgoyette static isc_result_t
1017 1.2.2.2 pgoyette grow_headerspace(isc_httpd_t *httpd) {
1018 1.2.2.2 pgoyette char *newspace;
1019 1.2.2.2 pgoyette unsigned int newlen;
1020 1.2.2.2 pgoyette isc_region_t r;
1021 1.2.2.2 pgoyette
1022 1.2.2.2 pgoyette isc_buffer_region(&httpd->headerbuffer, &r);
1023 1.2.2.2 pgoyette newlen = r.length + HTTP_SENDGROW;
1024 1.2.2.2 pgoyette if (newlen > HTTP_SEND_MAXLEN)
1025 1.2.2.2 pgoyette return (ISC_R_NOSPACE);
1026 1.2.2.2 pgoyette
1027 1.2.2.2 pgoyette newspace = isc_mem_get(httpd->mgr->mctx, newlen);
1028 1.2.2.2 pgoyette if (newspace == NULL)
1029 1.2.2.2 pgoyette return (ISC_R_NOMEMORY);
1030 1.2.2.2 pgoyette
1031 1.2.2.2 pgoyette isc_buffer_reinit(&httpd->headerbuffer, newspace, newlen);
1032 1.2.2.2 pgoyette
1033 1.2.2.2 pgoyette isc_mem_put(httpd->mgr->mctx, r.base, r.length);
1034 1.2.2.2 pgoyette
1035 1.2.2.2 pgoyette return (ISC_R_SUCCESS);
1036 1.2.2.2 pgoyette }
1037 1.2.2.2 pgoyette
1038 1.2.2.2 pgoyette isc_result_t
1039 1.2.2.2 pgoyette isc_httpd_response(isc_httpd_t *httpd) {
1040 1.2.2.2 pgoyette isc_result_t result;
1041 1.2.2.2 pgoyette unsigned int needlen;
1042 1.2.2.2 pgoyette
1043 1.2.2.2 pgoyette needlen = strlen(httpd->protocol) + 1; /* protocol + space */
1044 1.2.2.2 pgoyette needlen += 3 + 1; /* room for response code, always 3 bytes */
1045 1.2.2.2 pgoyette needlen += strlen(httpd->retmsg) + 2; /* return msg + CRLF */
1046 1.2.2.2 pgoyette
1047 1.2.2.2 pgoyette while (isc_buffer_availablelength(&httpd->headerbuffer) < needlen) {
1048 1.2.2.2 pgoyette result = grow_headerspace(httpd);
1049 1.2.2.2 pgoyette if (result != ISC_R_SUCCESS)
1050 1.2.2.2 pgoyette return (result);
1051 1.2.2.2 pgoyette }
1052 1.2.2.2 pgoyette
1053 1.2.2.2 pgoyette return (isc_buffer_printf(&httpd->headerbuffer, "%s %03u %s\r\n",
1054 1.2.2.2 pgoyette httpd->protocol, httpd->retcode,
1055 1.2.2.2 pgoyette httpd->retmsg));
1056 1.2.2.2 pgoyette }
1057 1.2.2.2 pgoyette
1058 1.2.2.2 pgoyette isc_result_t
1059 1.2.2.2 pgoyette isc_httpd_addheader(isc_httpd_t *httpd, const char *name,
1060 1.2.2.2 pgoyette const char *val)
1061 1.2.2.2 pgoyette {
1062 1.2.2.2 pgoyette isc_result_t result;
1063 1.2.2.2 pgoyette unsigned int needlen;
1064 1.2.2.2 pgoyette
1065 1.2.2.2 pgoyette needlen = strlen(name); /* name itself */
1066 1.2.2.2 pgoyette if (val != NULL)
1067 1.2.2.2 pgoyette needlen += 2 + strlen(val); /* :<space> and val */
1068 1.2.2.2 pgoyette needlen += 2; /* CRLF */
1069 1.2.2.2 pgoyette
1070 1.2.2.2 pgoyette while (isc_buffer_availablelength(&httpd->headerbuffer) < needlen) {
1071 1.2.2.2 pgoyette result = grow_headerspace(httpd);
1072 1.2.2.2 pgoyette if (result != ISC_R_SUCCESS)
1073 1.2.2.2 pgoyette return (result);
1074 1.2.2.2 pgoyette }
1075 1.2.2.2 pgoyette
1076 1.2.2.2 pgoyette if (val != NULL) {
1077 1.2.2.2 pgoyette return (isc_buffer_printf(&httpd->headerbuffer, "%s: %s\r\n",
1078 1.2.2.2 pgoyette name, val));
1079 1.2.2.2 pgoyette } else {
1080 1.2.2.2 pgoyette return (isc_buffer_printf(&httpd->headerbuffer, "%s\r\n",
1081 1.2.2.2 pgoyette name));
1082 1.2.2.2 pgoyette }
1083 1.2.2.2 pgoyette }
1084 1.2.2.2 pgoyette
1085 1.2.2.2 pgoyette isc_result_t
1086 1.2.2.2 pgoyette isc_httpd_endheaders(isc_httpd_t *httpd) {
1087 1.2.2.2 pgoyette isc_result_t result;
1088 1.2.2.2 pgoyette
1089 1.2.2.2 pgoyette while (isc_buffer_availablelength(&httpd->headerbuffer) < 2) {
1090 1.2.2.2 pgoyette result = grow_headerspace(httpd);
1091 1.2.2.2 pgoyette if (result != ISC_R_SUCCESS)
1092 1.2.2.2 pgoyette return (result);
1093 1.2.2.2 pgoyette }
1094 1.2.2.2 pgoyette
1095 1.2.2.2 pgoyette return (isc_buffer_printf(&httpd->headerbuffer, "\r\n"));
1096 1.2.2.2 pgoyette }
1097 1.2.2.2 pgoyette
1098 1.2.2.2 pgoyette isc_result_t
1099 1.2.2.2 pgoyette isc_httpd_addheaderuint(isc_httpd_t *httpd, const char *name, int val) {
1100 1.2.2.2 pgoyette isc_result_t result;
1101 1.2.2.2 pgoyette unsigned int needlen;
1102 1.2.2.2 pgoyette char buf[sizeof "18446744073709551616"];
1103 1.2.2.2 pgoyette
1104 1.2.2.2 pgoyette snprintf(buf, sizeof(buf), "%d", val);
1105 1.2.2.2 pgoyette
1106 1.2.2.2 pgoyette needlen = strlen(name); /* name itself */
1107 1.2.2.2 pgoyette needlen += 2 + strlen(buf); /* :<space> and val */
1108 1.2.2.2 pgoyette needlen += 2; /* CRLF */
1109 1.2.2.2 pgoyette
1110 1.2.2.2 pgoyette while (isc_buffer_availablelength(&httpd->headerbuffer) < needlen) {
1111 1.2.2.2 pgoyette result = grow_headerspace(httpd);
1112 1.2.2.2 pgoyette if (result != ISC_R_SUCCESS)
1113 1.2.2.2 pgoyette return (result);
1114 1.2.2.2 pgoyette }
1115 1.2.2.2 pgoyette
1116 1.2.2.2 pgoyette return (isc_buffer_printf(&httpd->headerbuffer, "%s: %s\r\n", name,
1117 1.2.2.2 pgoyette buf));
1118 1.2.2.2 pgoyette }
1119 1.2.2.2 pgoyette
1120 1.2.2.2 pgoyette static void
1121 1.2.2.2 pgoyette isc_httpd_senddone(isc_task_t *task, isc_event_t *ev) {
1122 1.2.2.2 pgoyette isc_httpd_t *httpd = ev->ev_arg;
1123 1.2.2.2 pgoyette isc_region_t r;
1124 1.2.2.2 pgoyette isc_socketevent_t *sev = (isc_socketevent_t *)ev;
1125 1.2.2.2 pgoyette
1126 1.2.2.2 pgoyette ENTER("senddone");
1127 1.2.2.2 pgoyette INSIST(ISC_HTTPD_ISSEND(httpd));
1128 1.2.2.2 pgoyette
1129 1.2.2.3 pgoyette isc_buffer_free(&httpd->sendbuffer);
1130 1.2.2.2 pgoyette
1131 1.2.2.2 pgoyette /*
1132 1.2.2.2 pgoyette * We will always want to clean up our receive buffer, even if we
1133 1.2.2.2 pgoyette * got an error on send or we are shutting down.
1134 1.2.2.2 pgoyette *
1135 1.2.2.2 pgoyette * We will pass in the buffer only if there is data in it. If
1136 1.2.2.2 pgoyette * there is no data, we will pass in a NULL.
1137 1.2.2.2 pgoyette */
1138 1.2.2.2 pgoyette if (httpd->freecb != NULL) {
1139 1.2.2.2 pgoyette isc_buffer_t *b = NULL;
1140 1.2.2.2 pgoyette if (isc_buffer_length(&httpd->bodybuffer) > 0) {
1141 1.2.2.2 pgoyette b = &httpd->bodybuffer;
1142 1.2.2.2 pgoyette httpd->freecb(b, httpd->freecb_arg);
1143 1.2.2.2 pgoyette }
1144 1.2.2.2 pgoyette NOTICE("senddone free callback performed");
1145 1.2.2.2 pgoyette }
1146 1.2.2.2 pgoyette
1147 1.2.2.2 pgoyette if (sev->result != ISC_R_SUCCESS) {
1148 1.2.2.2 pgoyette destroy_client(&httpd);
1149 1.2.2.2 pgoyette goto out;
1150 1.2.2.2 pgoyette }
1151 1.2.2.2 pgoyette
1152 1.2.2.2 pgoyette if ((httpd->flags & HTTPD_CLOSE) != 0) {
1153 1.2.2.2 pgoyette destroy_client(&httpd);
1154 1.2.2.2 pgoyette goto out;
1155 1.2.2.2 pgoyette }
1156 1.2.2.2 pgoyette
1157 1.2.2.2 pgoyette ISC_HTTPD_SETRECV(httpd);
1158 1.2.2.2 pgoyette
1159 1.2.2.2 pgoyette NOTICE("senddone restarting recv on socket");
1160 1.2.2.2 pgoyette
1161 1.2.2.2 pgoyette reset_client(httpd);
1162 1.2.2.2 pgoyette
1163 1.2.2.2 pgoyette r.base = (unsigned char *)httpd->recvbuf;
1164 1.2.2.2 pgoyette r.length = HTTP_RECVLEN - 1;
1165 1.2.2.2 pgoyette /* check return code? */
1166 1.2.2.2 pgoyette (void)isc_socket_recv(httpd->sock, &r, 1, task,
1167 1.2.2.2 pgoyette isc_httpd_recvdone, httpd);
1168 1.2.2.2 pgoyette
1169 1.2.2.2 pgoyette out:
1170 1.2.2.2 pgoyette isc_event_free(&ev);
1171 1.2.2.2 pgoyette EXIT("senddone");
1172 1.2.2.2 pgoyette }
1173 1.2.2.2 pgoyette
1174 1.2.2.2 pgoyette static void
1175 1.2.2.2 pgoyette reset_client(isc_httpd_t *httpd) {
1176 1.2.2.2 pgoyette /*
1177 1.2.2.2 pgoyette * Catch errors here. We MUST be in RECV mode, and we MUST NOT have
1178 1.2.2.2 pgoyette * any outstanding buffers. If we have buffers, we have a leak.
1179 1.2.2.2 pgoyette */
1180 1.2.2.2 pgoyette INSIST(ISC_HTTPD_ISRECV(httpd));
1181 1.2.2.2 pgoyette INSIST(!ISC_LINK_LINKED(&httpd->headerbuffer, link));
1182 1.2.2.2 pgoyette INSIST(!ISC_LINK_LINKED(&httpd->bodybuffer, link));
1183 1.2.2.2 pgoyette
1184 1.2.2.2 pgoyette httpd->recvbuf[0] = 0;
1185 1.2.2.2 pgoyette httpd->recvlen = 0;
1186 1.2.2.2 pgoyette httpd->headers = NULL;
1187 1.2.2.2 pgoyette httpd->method = ISC_HTTPD_METHODUNKNOWN;
1188 1.2.2.2 pgoyette httpd->url = NULL;
1189 1.2.2.2 pgoyette httpd->querystring = NULL;
1190 1.2.2.2 pgoyette httpd->protocol = NULL;
1191 1.2.2.2 pgoyette httpd->flags = 0;
1192 1.2.2.2 pgoyette
1193 1.2.2.2 pgoyette isc_buffer_clear(&httpd->headerbuffer);
1194 1.2.2.2 pgoyette isc_buffer_clear(&httpd->compbuffer);
1195 1.2.2.2 pgoyette isc_buffer_invalidate(&httpd->bodybuffer);
1196 1.2.2.2 pgoyette }
1197 1.2.2.2 pgoyette
1198 1.2.2.2 pgoyette isc_result_t
1199 1.2.2.2 pgoyette isc_httpdmgr_addurl(isc_httpdmgr_t *httpdmgr, const char *url,
1200 1.2.2.2 pgoyette isc_httpdaction_t *func, void *arg)
1201 1.2.2.2 pgoyette {
1202 1.2.2.3 pgoyette return (isc_httpdmgr_addurl2(httpdmgr, url, false, func, arg));
1203 1.2.2.2 pgoyette }
1204 1.2.2.2 pgoyette
1205 1.2.2.2 pgoyette isc_result_t
1206 1.2.2.2 pgoyette isc_httpdmgr_addurl2(isc_httpdmgr_t *httpdmgr, const char *url,
1207 1.2.2.3 pgoyette bool isstatic,
1208 1.2.2.2 pgoyette isc_httpdaction_t *func, void *arg)
1209 1.2.2.2 pgoyette {
1210 1.2.2.2 pgoyette isc_httpdurl_t *item;
1211 1.2.2.2 pgoyette
1212 1.2.2.2 pgoyette if (url == NULL) {
1213 1.2.2.2 pgoyette httpdmgr->render_404 = func;
1214 1.2.2.2 pgoyette return (ISC_R_SUCCESS);
1215 1.2.2.2 pgoyette }
1216 1.2.2.2 pgoyette
1217 1.2.2.2 pgoyette item = isc_mem_get(httpdmgr->mctx, sizeof(isc_httpdurl_t));
1218 1.2.2.2 pgoyette if (item == NULL)
1219 1.2.2.2 pgoyette return (ISC_R_NOMEMORY);
1220 1.2.2.2 pgoyette
1221 1.2.2.2 pgoyette item->url = isc_mem_strdup(httpdmgr->mctx, url);
1222 1.2.2.2 pgoyette if (item->url == NULL) {
1223 1.2.2.2 pgoyette isc_mem_put(httpdmgr->mctx, item, sizeof(isc_httpdurl_t));
1224 1.2.2.2 pgoyette return (ISC_R_NOMEMORY);
1225 1.2.2.2 pgoyette }
1226 1.2.2.2 pgoyette
1227 1.2.2.2 pgoyette item->action = func;
1228 1.2.2.2 pgoyette item->action_arg = arg;
1229 1.2.2.2 pgoyette item->isstatic = isstatic;
1230 1.2.2.2 pgoyette isc_time_now(&item->loadtime);
1231 1.2.2.2 pgoyette
1232 1.2.2.2 pgoyette ISC_LINK_INIT(item, link);
1233 1.2.2.2 pgoyette ISC_LIST_APPEND(httpdmgr->urls, item, link);
1234 1.2.2.2 pgoyette
1235 1.2.2.2 pgoyette return (ISC_R_SUCCESS);
1236 1.2.2.2 pgoyette }
1237 1.2.2.2 pgoyette
1238 1.2.2.2 pgoyette void
1239 1.2.2.2 pgoyette isc_httpd_setfinishhook(void (*fn)(void))
1240 1.2.2.2 pgoyette {
1241 1.2.2.2 pgoyette finishhook = fn;
1242 1.2.2.2 pgoyette }
1243