Files
bind9/lib/isc/httpd.c
Ondřej Surý 9274876dec Serialize the HTTP/1.1 statschannel requests
The statschannel truncated test still terminates abruptly sometimes and
it doesn't return the answer for the first query.  This might happen
when the second process_request() discovers there's not enough space
before the sending is complete and the connection is terminated before
the client gets the data.

Change the isc_http, so it pauses the reading when it receives the data
and resumes it only after the sending has completed or there's
incomplete request waiting for more data.

This makes the request processing slightly less efficient, but also less
taxing for the server, because previously all requests that has been
received via single TCP read would be processed in the loop and the
sends would be queued after the read callback has processed a full
buffer.

(cherry picked from commit 13959781cb)
2022-10-20 17:23:36 +02:00

1107 lines
27 KiB
C

/*
* Copyright (C) Internet Systems Consortium, Inc. ("ISC")
*
* SPDX-License-Identifier: MPL-2.0
*
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, you can obtain one at https://mozilla.org/MPL/2.0/.
*
* See the COPYRIGHT file distributed with this work for additional
* information regarding copyright ownership.
*/
/*! \file */
#include <ctype.h>
#include <inttypes.h>
#include <stdbool.h>
#include <string.h>
#include <isc/buffer.h>
#include <isc/httpd.h>
#include <isc/mem.h>
#include <isc/netmgr.h>
#include <isc/print.h>
#include <isc/refcount.h>
#include <isc/sockaddr.h>
#include <isc/string.h>
#include <isc/time.h>
#include <isc/url.h>
#include <isc/util.h>
#include "netmgr/netmgr-int.h"
#include "picohttpparser.h"
#ifdef HAVE_ZLIB
#include <zlib.h>
#endif /* ifdef HAVE_ZLIB */
#define CHECK(m) \
do { \
result = (m); \
if (result != ISC_R_SUCCESS) { \
goto cleanup; \
} \
} while (0)
/*
* Size the recv buffer to hold at maximum two full buffers from isc_nm_read(),
* so we don't have to handle the truncation.
*/
#define HTTP_RECVLEN ISC_NETMGR_TCP_RECVBUF_SIZE * 2
#define HTTP_SENDLEN ISC_NETMGR_TCP_RECVBUF_SIZE
#define HTTP_HEADERS_NUM 10
#define HTTP_MAX_REQUEST_LEN 4096
#define HTTPD_CLOSE 0x0001 /* Got a Connection: close header */
#define HTTPD_FOUNDHOST 0x0002 /* Got a Host: header */
#define HTTPD_KEEPALIVE 0x0004 /* Got a Connection: Keep-Alive */
#define HTTPD_ACCEPT_DEFLATE 0x0008
#define HTTPD_MAGIC ISC_MAGIC('H', 't', 'p', 'd')
#define VALID_HTTPD(m) ISC_MAGIC_VALID(m, HTTPD_MAGIC)
#define HTTPDMGR_MAGIC ISC_MAGIC('H', 'p', 'd', 'm')
#define VALID_HTTPDMGR(m) ISC_MAGIC_VALID(m, HTTPDMGR_MAGIC)
/*%
* HTTP methods.
*/
typedef enum { METHOD_UNKNOWN = 0, METHOD_GET = 1, METHOD_POST = 2 } method_t;
/*%
* HTTP urls. These are the URLs we manage, and the function to call to
* provide the data for it. We pass in the base url (so the same function
* can handle multiple requests), and a structure to fill in to return a
* result to the client. We also pass in a pointer to be filled in for
* the data cleanup function.
*/
struct isc_httpdurl {
char *url;
isc_httpdaction_t *action;
void *action_arg;
bool isstatic;
isc_time_t loadtime;
ISC_LINK(isc_httpdurl_t) link;
};
/*% http client */
struct isc_httpd {
unsigned int magic; /* HTTPD_MAGIC */
isc_httpdmgr_t *mgr; /*%< our parent */
ISC_LINK(isc_httpd_t) link;
isc_nmhandle_t *handle; /* Permanent pointer to handle */
isc_nmhandle_t *readhandle; /* Waiting for a read callback */
int flags;
/*%
* Received data state.
*/
char recvbuf[HTTP_RECVLEN]; /*%< receive buffer */
size_t recvlen; /*%< length recv'd */
size_t consume; /*%< length of last command */
method_t method;
int minor_version;
const char *path;
isc_url_parser_t up;
isc_time_t if_modified_since;
};
struct isc_httpdmgr {
unsigned int magic; /* HTTPDMGR_MAGIC */
isc_refcount_t references;
isc_mem_t *mctx;
isc_nmsocket_t *sock;
isc_httpdclientok_t *client_ok; /*%< client validator */
isc_httpdondestroy_t *ondestroy; /*%< cleanup callback */
void *cb_arg; /*%< argument for the above */
unsigned int flags;
ISC_LIST(isc_httpd_t) running; /*%< running clients */
isc_mutex_t lock;
ISC_LIST(isc_httpdurl_t) urls; /*%< urls we manage */
isc_httpdaction_t *render_404;
isc_httpdaction_t *render_500;
};
typedef struct isc_httpd_sendreq {
isc_mem_t *mctx;
isc_httpd_t *httpd;
isc_nmhandle_t *handle;
/*%
* Transmit data state.
*
* This is the data buffer we will transmit.
*
* This free function pointer is filled in by the rendering function
* we call. The free function is called after the data is transmitted
* to the client.
*
* We currently use three buffers total:
*
* sendbuffer - gets filled as we gather the data
*
* bodybuffer - for the client to fill in (which it manages, it provides
* the space for it, etc) -- we will pass that buffer structure back to
* the caller, who is responsible for managing the space it may have
* allocated as backing store for it. we only allocate the buffer
* itself, not the backing store.
*
* compbuffer - managed by us, that contains the compressed HTTP data,
* if compression is used.
*/
isc_buffer_t *sendbuffer;
isc_buffer_t *compbuffer;
isc_buffer_t bodybuffer;
const char *mimetype;
unsigned int retcode;
const char *retmsg;
isc_httpdfree_t *freecb;
void *freecb_arg;
} isc_httpd_sendreq_t;
static isc_result_t
httpd_newconn(isc_nmhandle_t *, isc_result_t, void *);
static void
httpd_request(isc_nmhandle_t *, isc_result_t, isc_region_t *, void *);
static void
httpd_senddone(isc_nmhandle_t *, isc_result_t, void *);
static void
httpd_reset(void *);
static void
httpd_put(void *);
static void
httpd_addheader(isc_httpd_sendreq_t *, const char *, const char *);
static void
httpd_addheaderuint(isc_httpd_sendreq_t *, const char *, int);
static void
httpd_endheaders(isc_httpd_sendreq_t *);
static void
httpd_response(isc_httpd_t *, isc_httpd_sendreq_t *);
static isc_result_t
process_request(isc_httpd_t *, size_t);
static isc_httpdaction_t render_404;
static isc_httpdaction_t render_500;
#if ENABLE_AFL
static void (*finishhook)(void) = NULL;
#endif /* ENABLE_AFL */
static void
destroy_httpdmgr(isc_httpdmgr_t *);
static void
httpdmgr_attach(isc_httpdmgr_t *, isc_httpdmgr_t **);
static void
httpdmgr_detach(isc_httpdmgr_t **);
isc_result_t
isc_httpdmgr_create(isc_nm_t *nm, isc_mem_t *mctx, isc_sockaddr_t *addr,
isc_httpdclientok_t *client_ok,
isc_httpdondestroy_t *ondestroy, void *cb_arg,
isc_httpdmgr_t **httpdmgrp) {
isc_result_t result;
isc_httpdmgr_t *httpdmgr = NULL;
REQUIRE(nm != NULL);
REQUIRE(mctx != NULL);
REQUIRE(httpdmgrp != NULL && *httpdmgrp == NULL);
httpdmgr = isc_mem_get(mctx, sizeof(isc_httpdmgr_t));
*httpdmgr = (isc_httpdmgr_t){ .client_ok = client_ok,
.ondestroy = ondestroy,
.cb_arg = cb_arg,
.render_404 = render_404,
.render_500 = render_500 };
isc_mutex_init(&httpdmgr->lock);
isc_mem_attach(mctx, &httpdmgr->mctx);
ISC_LIST_INIT(httpdmgr->running);
ISC_LIST_INIT(httpdmgr->urls);
isc_refcount_init(&httpdmgr->references, 1);
CHECK(isc_nm_listentcp(nm, addr, httpd_newconn, httpdmgr,
sizeof(isc_httpd_t), 5, NULL, &httpdmgr->sock));
httpdmgr->magic = HTTPDMGR_MAGIC;
*httpdmgrp = httpdmgr;
return (ISC_R_SUCCESS);
cleanup:
httpdmgr->magic = 0;
isc_refcount_decrementz(&httpdmgr->references);
isc_refcount_destroy(&httpdmgr->references);
isc_mem_detach(&httpdmgr->mctx);
isc_mutex_destroy(&httpdmgr->lock);
isc_mem_put(mctx, httpdmgr, sizeof(isc_httpdmgr_t));
return (result);
}
static void
httpdmgr_attach(isc_httpdmgr_t *source, isc_httpdmgr_t **targetp) {
REQUIRE(VALID_HTTPDMGR(source));
REQUIRE(targetp != NULL && *targetp == NULL);
isc_refcount_increment(&source->references);
*targetp = source;
}
static void
httpdmgr_detach(isc_httpdmgr_t **httpdmgrp) {
isc_httpdmgr_t *httpdmgr = NULL;
REQUIRE(httpdmgrp != NULL);
REQUIRE(VALID_HTTPDMGR(*httpdmgrp));
httpdmgr = *httpdmgrp;
*httpdmgrp = NULL;
if (isc_refcount_decrement(&httpdmgr->references) == 1) {
destroy_httpdmgr(httpdmgr);
}
}
static void
destroy_httpdmgr(isc_httpdmgr_t *httpdmgr) {
isc_httpdurl_t *url;
isc_refcount_destroy(&httpdmgr->references);
LOCK(&httpdmgr->lock);
REQUIRE((httpdmgr->flags & ISC_HTTPDMGR_SHUTTINGDOWN) != 0);
REQUIRE(ISC_LIST_EMPTY(httpdmgr->running));
httpdmgr->magic = 0;
if (httpdmgr->sock != NULL) {
isc_nmsocket_close(&httpdmgr->sock);
}
/*
* Clear out the list of all actions we know about. Just free the
* memory.
*/
url = ISC_LIST_HEAD(httpdmgr->urls);
while (url != NULL) {
isc_mem_free(httpdmgr->mctx, url->url);
ISC_LIST_UNLINK(httpdmgr->urls, url, link);
isc_mem_put(httpdmgr->mctx, url, sizeof(isc_httpdurl_t));
url = ISC_LIST_HEAD(httpdmgr->urls);
}
UNLOCK(&httpdmgr->lock);
isc_mutex_destroy(&httpdmgr->lock);
if (httpdmgr->ondestroy != NULL) {
(httpdmgr->ondestroy)(httpdmgr->cb_arg);
}
isc_mem_putanddetach(&httpdmgr->mctx, httpdmgr, sizeof(isc_httpdmgr_t));
}
static bool
name_match(const struct phr_header *header, const char *match) {
size_t match_len = strlen(match);
if (match_len != header->name_len) {
return (false);
}
return (strncasecmp(header->name, match, match_len) == 0);
}
static bool
value_match(const struct phr_header *header, const char *match) {
size_t match_len = strlen(match);
size_t limit;
if (match_len > header->value_len) {
return (false);
}
limit = header->value_len - match_len + 1;
for (size_t i = 0; i < limit; i++) {
if (isspace(header->value[i])) {
while (i < limit && isspace(header->value[i])) {
i++;
}
continue;
}
if (strncasecmp(&header->value[i], match, match_len) == 0) {
i += match_len;
/*
* Sanity check; f.e. for 'deflate' match only
* 'deflate[,;]', but not 'deflateyou'
*/
if (i == header->value_len || header->value[i] == ',' ||
header->value[i] == ';')
{
return (true);
}
}
while (i < limit && header->value[i] != ',') {
i++;
}
}
return (false);
}
static isc_result_t
process_request(isc_httpd_t *httpd, size_t last_len) {
int pret;
const char *method = NULL;
size_t method_len = 0;
const char *path;
size_t path_len = 0;
struct phr_header headers[HTTP_HEADERS_NUM];
size_t num_headers;
isc_result_t result;
num_headers = ARRAY_SIZE(headers);
pret = phr_parse_request(httpd->recvbuf, httpd->recvlen, &method,
&method_len, &path, &path_len,
&httpd->minor_version, headers, &num_headers,
last_len);
if (pret == -1) {
/* Parse Error */
return (ISC_R_UNEXPECTED);
} else if (pret == -2) {
/* Need more data */
return (ISC_R_NOMORE);
}
INSIST(pret > 0);
if (pret > HTTP_MAX_REQUEST_LEN) {
return (ISC_R_RANGE);
}
httpd->consume = pret;
/*
* Determine if this is a POST or GET method. Any other values will
* cause an error to be returned.
*/
if (strncmp(method, "GET ", method_len) == 0) {
httpd->method = METHOD_GET;
} else if (strncmp(method, "POST ", method_len) == 0) {
httpd->method = METHOD_POST;
} else {
return (ISC_R_RANGE);
}
/*
* Parse the URL
*/
result = isc_url_parse(path, path_len, 0, &httpd->up);
if (result != ISC_R_SUCCESS) {
return (result);
}
httpd->path = path;
ssize_t content_len = 0;
bool keep_alive = false;
isc_time_set(&httpd->if_modified_since, 0, 0);
for (size_t i = 0; i < num_headers; i++) {
struct phr_header *header = &headers[i];
if (name_match(header, "Content-Length")) {
char *endptr;
content_len = (size_t)strtoul(header->value, &endptr,
10);
/* Consistency check, if we consumed all numbers */
if ((header->value + header->value_len) != endptr) {
return (ISC_R_RANGE);
}
} else if (name_match(header, "Connection")) {
if (value_match(header, "close")) {
httpd->flags |= HTTPD_CLOSE;
} else if (value_match(header, "keep-alive")) {
keep_alive = true;
}
} else if (name_match(header, "Host")) {
httpd->flags |= HTTPD_FOUNDHOST;
} else if (name_match(header, "Accept-Encoding")) {
if (value_match(header, "deflate")) {
httpd->flags |= HTTPD_ACCEPT_DEFLATE;
}
} else if (name_match(header, "If-Modified-Since")) {
char timestamp[ISC_FORMATHTTPTIMESTAMP_SIZE + 1];
memmove(timestamp, header->value, header->value_len);
timestamp[header->value_len] = 0;
/* Ignore the value if it can't be parsed */
(void)isc_time_parsehttptimestamp(
timestamp, &httpd->if_modified_since);
}
}
/*
* The Content-Length is optional in an HTTP request.
* For a GET the length must be zero.
*/
if (httpd->method == METHOD_GET && content_len != 0) {
return (ISC_R_BADNUMBER);
}
if (content_len == (ssize_t)ULONG_MAX) {
/* Invalid number in the header value. */
return (ISC_R_BADNUMBER);
}
if (httpd->consume + content_len > httpd->recvlen) {
/* The request data isn't complete yet. */
return (ISC_R_NOMORE);
}
/* Consume the request's data, which we do not use. */
httpd->consume += content_len;
switch (httpd->minor_version) {
case 0:
if (keep_alive == true) {
httpd->flags |= HTTPD_KEEPALIVE;
} else {
httpd->flags |= HTTPD_CLOSE;
}
break;
case 1:
if ((httpd->flags & HTTPD_FOUNDHOST) == 0) {
return (ISC_R_RANGE);
}
break;
default:
return (ISC_R_UNEXPECTED);
}
/*
* Looks like a a valid request, so now we know we won't have
* to process this buffer again. We can NULL-terminate the
* URL for the caller's benefit, and set recvlen to 0 so
* the next read will overwrite this one instead of appending
* to the buffer.
*/
return (ISC_R_SUCCESS);
}
static void
httpd_reset(void *arg) {
isc_httpd_t *httpd = (isc_httpd_t *)arg;
isc_httpdmgr_t *httpdmgr = NULL;
REQUIRE(VALID_HTTPD(httpd));
httpdmgr = httpd->mgr;
REQUIRE(VALID_HTTPDMGR(httpdmgr));
LOCK(&httpdmgr->lock);
ISC_LIST_UNLINK(httpdmgr->running, httpd, link);
UNLOCK(&httpdmgr->lock);
httpd->recvbuf[0] = 0;
httpd->recvlen = 0;
httpd->consume = 0;
httpd->method = METHOD_UNKNOWN;
httpd->flags = 0;
httpd->minor_version = -1;
httpd->path = NULL;
httpd->up = (isc_url_parser_t){ 0 };
isc_time_set(&httpd->if_modified_since, 0, 0);
}
static void
isc__httpd_sendreq_free(isc_httpd_sendreq_t *req) {
/* Clean up buffers */
isc_buffer_free(&req->sendbuffer);
isc_mem_putanddetach(&req->mctx, req, sizeof(*req));
}
static isc_httpd_sendreq_t *
isc__httpd_sendreq_new(isc_httpd_t *httpd) {
isc_httpdmgr_t *httpdmgr = httpd->mgr;
isc_httpd_sendreq_t *req;
REQUIRE(VALID_HTTPDMGR(httpdmgr));
req = isc_mem_get(httpdmgr->mctx, sizeof(*req));
*req = (isc_httpd_sendreq_t){ 0 };
isc_mem_attach(httpdmgr->mctx, &req->mctx);
/*
* Initialize the buffer for our headers.
*/
isc_buffer_allocate(req->mctx, &req->sendbuffer, HTTP_SENDLEN);
isc_buffer_clear(req->sendbuffer);
isc_buffer_setautorealloc(req->sendbuffer, true);
isc_buffer_initnull(&req->bodybuffer);
return (req);
}
static void
httpd_put(void *arg) {
isc_httpd_t *httpd = (isc_httpd_t *)arg;
isc_httpdmgr_t *mgr = NULL;
REQUIRE(VALID_HTTPD(httpd));
mgr = httpd->mgr;
REQUIRE(VALID_HTTPDMGR(mgr));
httpd->magic = 0;
httpd->mgr = NULL;
isc_mem_put(mgr->mctx, httpd, sizeof(*httpd));
httpdmgr_detach(&mgr);
#if ENABLE_AFL
if (finishhook != NULL) {
finishhook();
}
#endif /* ENABLE_AFL */
}
static void
new_httpd(isc_httpdmgr_t *httpdmgr, isc_nmhandle_t *handle) {
isc_httpd_t *httpd = NULL;
REQUIRE(VALID_HTTPDMGR(httpdmgr));
httpd = isc_nmhandle_getdata(handle);
if (httpd == NULL) {
httpd = isc_mem_get(httpdmgr->mctx, sizeof(*httpd));
*httpd = (isc_httpd_t){ .handle = NULL };
httpdmgr_attach(httpdmgr, &httpd->mgr);
}
if (httpd->handle == NULL) {
isc_nmhandle_setdata(handle, httpd, httpd_reset, httpd_put);
httpd->handle = handle;
} else {
INSIST(httpd->handle == handle);
}
ISC_LINK_INIT(httpd, link);
httpd->magic = HTTPD_MAGIC;
LOCK(&httpdmgr->lock);
ISC_LIST_APPEND(httpdmgr->running, httpd, link);
UNLOCK(&httpdmgr->lock);
isc_nmhandle_attach(httpd->handle, &httpd->readhandle);
isc_nm_read(handle, httpd_request, httpdmgr);
}
static isc_result_t
httpd_newconn(isc_nmhandle_t *handle, isc_result_t result, void *arg) {
isc_httpdmgr_t *httpdmgr = (isc_httpdmgr_t *)arg;
isc_sockaddr_t peeraddr;
REQUIRE(VALID_HTTPDMGR(httpdmgr));
if ((httpdmgr->flags & ISC_HTTPDMGR_SHUTTINGDOWN) != 0) {
return (ISC_R_CANCELED);
} else if (result == ISC_R_CANCELED) {
isc_httpdmgr_shutdown(&httpdmgr);
return (result);
} else if (result != ISC_R_SUCCESS) {
return (result);
}
peeraddr = isc_nmhandle_peeraddr(handle);
if (httpdmgr->client_ok != NULL &&
!(httpdmgr->client_ok)(&peeraddr, httpdmgr->cb_arg))
{
return (ISC_R_FAILURE);
}
new_httpd(httpdmgr, handle);
return (ISC_R_SUCCESS);
}
static isc_result_t
render_404(const isc_httpd_t *httpd, const isc_httpdurl_t *urlinfo, void *arg,
unsigned int *retcode, const char **retmsg, const char **mimetype,
isc_buffer_t *b, isc_httpdfree_t **freecb, void **freecb_args) {
static char msg[] = "No such URL.\r\n";
UNUSED(httpd);
UNUSED(urlinfo);
UNUSED(arg);
*retcode = 404;
*retmsg = "No such URL";
*mimetype = "text/plain";
isc_buffer_reinit(b, msg, strlen(msg));
isc_buffer_add(b, strlen(msg));
*freecb = NULL;
*freecb_args = NULL;
return (ISC_R_SUCCESS);
}
static isc_result_t
render_500(const isc_httpd_t *httpd, const isc_httpdurl_t *urlinfo, void *arg,
unsigned int *retcode, const char **retmsg, const char **mimetype,
isc_buffer_t *b, isc_httpdfree_t **freecb, void **freecb_args) {
static char msg[] = "Internal server failure.\r\n";
UNUSED(httpd);
UNUSED(urlinfo);
UNUSED(arg);
*retcode = 500;
*retmsg = "Internal server failure";
*mimetype = "text/plain";
isc_buffer_reinit(b, msg, strlen(msg));
isc_buffer_add(b, strlen(msg));
*freecb = NULL;
*freecb_args = NULL;
return (ISC_R_SUCCESS);
}
#ifdef HAVE_ZLIB
/*%<
* Tries to compress httpd->bodybuffer to httpd->compbuffer, extending it
* if necessary.
*
* Requires:
*\li httpd a valid isc_httpd_t object
*
* Returns:
*\li #ISC_R_SUCCESS -- all is well.
*\li #ISC_R_NOMEMORY -- not enough memory to compress data
*\li #ISC_R_FAILURE -- error during compression or compressed
* data would be larger than input data
*/
static isc_result_t
httpd_compress(isc_httpd_sendreq_t *req) {
z_stream zstr;
int ret, inputlen;
/*
* We're setting output buffer size to input size so it fails if the
* compressed data size would be bigger than the input size.
*/
inputlen = isc_buffer_usedlength(&req->bodybuffer);
if (inputlen == 0) {
return (ISC_R_FAILURE);
}
isc_buffer_allocate(req->mctx, &req->compbuffer, inputlen);
isc_buffer_clear(req->compbuffer);
zstr = (z_stream){
.total_in = inputlen,
.avail_out = inputlen,
.avail_in = inputlen,
.next_in = isc_buffer_base(&req->bodybuffer),
.next_out = isc_buffer_base(req->compbuffer),
};
ret = deflateInit(&zstr, Z_DEFAULT_COMPRESSION);
if (ret == Z_OK) {
ret = deflate(&zstr, Z_FINISH);
}
deflateEnd(&zstr);
if (ret == Z_STREAM_END) {
isc_buffer_add(req->compbuffer, zstr.total_out);
return (ISC_R_SUCCESS);
} else {
isc_buffer_free(&req->compbuffer);
return (ISC_R_FAILURE);
}
}
#endif /* ifdef HAVE_ZLIB */
static void
prepare_response(isc_httpdmgr_t *mgr, isc_httpd_t *httpd,
isc_httpd_sendreq_t **reqp) {
isc_httpd_sendreq_t *req = NULL;
isc_time_t now;
char datebuf[ISC_FORMATHTTPTIMESTAMP_SIZE];
const char *path = "/";
size_t path_len = 1;
bool is_compressed = false;
isc_httpdurl_t *url = NULL;
isc_result_t result;
REQUIRE(VALID_HTTPD(httpd));
REQUIRE(reqp != NULL && *reqp == NULL);
isc_time_now(&now);
isc_time_formathttptimestamp(&now, datebuf, sizeof(datebuf));
if (httpd->up.field_set & (1 << ISC_UF_PATH)) {
path = &httpd->path[httpd->up.field_data[ISC_UF_PATH].off];
path_len = httpd->up.field_data[ISC_UF_PATH].len;
}
LOCK(&mgr->lock);
url = ISC_LIST_HEAD(mgr->urls);
while (url != NULL) {
if (strncmp(path, url->url, path_len) == 0) {
break;
}
url = ISC_LIST_NEXT(url, link);
}
UNLOCK(&mgr->lock);
req = isc__httpd_sendreq_new(httpd);
if (url == NULL) {
result = mgr->render_404(httpd, NULL, NULL, &req->retcode,
&req->retmsg, &req->mimetype,
&req->bodybuffer, &req->freecb,
&req->freecb_arg);
} else {
result = url->action(httpd, url, url->action_arg, &req->retcode,
&req->retmsg, &req->mimetype,
&req->bodybuffer, &req->freecb,
&req->freecb_arg);
}
if (result != ISC_R_SUCCESS) {
result = mgr->render_500(httpd, url, NULL, &req->retcode,
&req->retmsg, &req->mimetype,
&req->bodybuffer, &req->freecb,
&req->freecb_arg);
RUNTIME_CHECK(result == ISC_R_SUCCESS);
}
#ifdef HAVE_ZLIB
if ((httpd->flags & HTTPD_ACCEPT_DEFLATE) != 0) {
result = httpd_compress(req);
if (result == ISC_R_SUCCESS) {
is_compressed = true;
}
}
#endif /* ifdef HAVE_ZLIB */
httpd_response(httpd, req);
if ((httpd->flags & HTTPD_KEEPALIVE) != 0) {
httpd_addheader(req, "Connection", "Keep-Alive");
}
httpd_addheader(req, "Content-Type", req->mimetype);
httpd_addheader(req, "Date", datebuf);
httpd_addheader(req, "Expires", datebuf);
if (url != NULL && url->isstatic) {
char loadbuf[ISC_FORMATHTTPTIMESTAMP_SIZE];
isc_time_formathttptimestamp(&url->loadtime, loadbuf,
sizeof(loadbuf));
httpd_addheader(req, "Last-Modified", loadbuf);
httpd_addheader(req, "Cache-Control: public", NULL);
} else {
httpd_addheader(req, "Last-Modified", datebuf);
httpd_addheader(req, "Pragma: no-cache", NULL);
httpd_addheader(req, "Cache-Control: no-cache", NULL);
}
httpd_addheader(req, "Server: libisc", NULL);
if (is_compressed) {
httpd_addheader(req, "Content-Encoding", "deflate");
httpd_addheaderuint(req, "Content-Length",
isc_buffer_usedlength(req->compbuffer));
} else {
httpd_addheaderuint(req, "Content-Length",
isc_buffer_usedlength(&req->bodybuffer));
}
httpd_endheaders(req); /* done */
/*
* Append either the compressed or the non-compressed response body to
* the response headers and store the result in httpd->sendbuffer.
*/
if (is_compressed) {
isc_buffer_putmem(req->sendbuffer,
isc_buffer_base(req->compbuffer),
isc_buffer_usedlength(req->compbuffer));
isc_buffer_free(&req->compbuffer);
} else {
isc_buffer_putmem(req->sendbuffer,
isc_buffer_base(&req->bodybuffer),
isc_buffer_usedlength(&req->bodybuffer));
}
/* Free the bodybuffer */
if (req->freecb != NULL && isc_buffer_length(&req->bodybuffer) > 0) {
req->freecb(&req->bodybuffer, req->freecb_arg);
}
/* Consume the request from the recv buffer. */
INSIST(httpd->consume != 0);
INSIST(httpd->consume <= httpd->recvlen);
if (httpd->consume < httpd->recvlen) {
memmove(httpd->recvbuf, httpd->recvbuf + httpd->consume,
httpd->recvlen - httpd->consume);
}
httpd->recvlen -= httpd->consume;
httpd->consume = 0;
/*
* We don't need to attach to httpd here because it gets only cleaned
* when the last handle has been detached
*/
req->httpd = httpd;
*reqp = req;
}
static void
httpd_request(isc_nmhandle_t *handle, isc_result_t eresult,
isc_region_t *region, void *arg) {
isc_result_t result;
isc_httpdmgr_t *mgr = arg;
isc_httpd_t *httpd = NULL;
isc_httpd_sendreq_t *req = NULL;
isc_region_t r;
size_t last_len = 0;
httpd = isc_nmhandle_getdata(handle);
REQUIRE(VALID_HTTPD(httpd));
REQUIRE(httpd->handle == handle);
REQUIRE(httpd->readhandle == handle);
isc_nm_pauseread(httpd->readhandle);
if (eresult != ISC_R_SUCCESS) {
goto close_readhandle;
}
/*
* If we are being called from httpd_senddone(), the last HTTP request
* was processed successfully, reset the last_len to 0, even if there's
* data in the httpd->recvbuf.
*/
last_len = (region == NULL) ? 0 : httpd->recvlen;
/* Store the received data into the recvbuf */
if (region != NULL) {
if (httpd->recvlen + region->length > sizeof(httpd->recvbuf)) {
goto close_readhandle;
}
memmove(httpd->recvbuf + httpd->recvlen, region->base,
region->length);
httpd->recvlen += region->length;
}
result = process_request(httpd, last_len);
if (result == ISC_R_NOMORE) {
if (httpd->recvlen > HTTP_MAX_REQUEST_LEN) {
goto close_readhandle;
}
/* Wait for more data, the readhandle is still attached */
isc_nm_resumeread(httpd->readhandle);
return;
}
if (result != ISC_R_SUCCESS) {
goto close_readhandle;
}
prepare_response(mgr, httpd, &req);
/*
* Determine total response size.
*/
isc_buffer_usedregion(req->sendbuffer, &r);
isc_nmhandle_attach(httpd->handle, &req->handle);
isc_nm_send(httpd->handle, &r, httpd_senddone, req);
isc_nmhandle_detach(&httpd->readhandle);
return;
close_readhandle:
isc_nmhandle_detach(&httpd->readhandle);
}
void
isc_httpdmgr_shutdown(isc_httpdmgr_t **httpdmgrp) {
isc_httpdmgr_t *httpdmgr;
isc_httpd_t *httpd;
REQUIRE(httpdmgrp != NULL);
REQUIRE(VALID_HTTPDMGR(*httpdmgrp));
httpdmgr = *httpdmgrp;
*httpdmgrp = NULL;
isc_nm_stoplistening(httpdmgr->sock);
LOCK(&httpdmgr->lock);
httpdmgr->flags |= ISC_HTTPDMGR_SHUTTINGDOWN;
httpd = ISC_LIST_HEAD(httpdmgr->running);
while (httpd != NULL) {
isc_nm_pauseread(httpd->readhandle);
isc_nmhandle_detach(&httpd->readhandle);
httpd = ISC_LIST_NEXT(httpd, link);
}
UNLOCK(&httpdmgr->lock);
isc_nmsocket_close(&httpdmgr->sock);
httpdmgr_detach(&httpdmgr);
}
static void
httpd_response(isc_httpd_t *httpd, isc_httpd_sendreq_t *req) {
isc_result_t result;
result = isc_buffer_printf(req->sendbuffer, "HTTP/1.%u %03u %s\r\n",
httpd->minor_version, req->retcode,
req->retmsg);
RUNTIME_CHECK(result == ISC_R_SUCCESS);
}
static void
httpd_addheader(isc_httpd_sendreq_t *req, const char *name, const char *val) {
isc_result_t result;
if (val != NULL) {
result = isc_buffer_printf(req->sendbuffer, "%s: %s\r\n", name,
val);
} else {
result = isc_buffer_printf(req->sendbuffer, "%s\r\n", name);
}
RUNTIME_CHECK(result == ISC_R_SUCCESS);
}
static void
httpd_endheaders(isc_httpd_sendreq_t *req) {
isc_result_t result;
result = isc_buffer_printf(req->sendbuffer, "\r\n");
RUNTIME_CHECK(result == ISC_R_SUCCESS);
}
static void
httpd_addheaderuint(isc_httpd_sendreq_t *req, const char *name, int val) {
isc_result_t result;
result = isc_buffer_printf(req->sendbuffer, "%s: %d\r\n", name, val);
RUNTIME_CHECK(result == ISC_R_SUCCESS);
}
static void
httpd_senddone(isc_nmhandle_t *handle, isc_result_t eresult, void *arg) {
isc_httpd_sendreq_t *req = (isc_httpd_sendreq_t *)arg;
isc_httpd_t *httpd = req->httpd;
REQUIRE(VALID_HTTPD(httpd));
if (eresult == ISC_R_SUCCESS && (httpd->flags & HTTPD_CLOSE) == 0) {
/*
* Calling httpd_request() with region NULL restarts
* reading.
*/
isc_nmhandle_attach(handle, &httpd->readhandle);
httpd_request(handle, ISC_R_SUCCESS, NULL, httpd->mgr);
}
isc_nmhandle_detach(&handle);
isc__httpd_sendreq_free(req);
}
isc_result_t
isc_httpdmgr_addurl(isc_httpdmgr_t *httpdmgr, const char *url, bool isstatic,
isc_httpdaction_t *func, void *arg) {
isc_httpdurl_t *item;
REQUIRE(VALID_HTTPDMGR(httpdmgr));
if (url == NULL) {
httpdmgr->render_404 = func;
return (ISC_R_SUCCESS);
}
item = isc_mem_get(httpdmgr->mctx, sizeof(isc_httpdurl_t));
item->url = isc_mem_strdup(httpdmgr->mctx, url);
item->action = func;
item->action_arg = arg;
item->isstatic = isstatic;
isc_time_now(&item->loadtime);
ISC_LINK_INIT(item, link);
LOCK(&httpdmgr->lock);
ISC_LIST_APPEND(httpdmgr->urls, item, link);
UNLOCK(&httpdmgr->lock);
return (ISC_R_SUCCESS);
}
void
isc_httpd_setfinishhook(void (*fn)(void)) {
#if ENABLE_AFL
finishhook = fn;
#else /* ENABLE_AFL */
UNUSED(fn);
#endif /* ENABLE_AFL */
}
bool
isc_httpdurl_isstatic(const isc_httpdurl_t *url) {
return (url->isstatic);
}
const isc_time_t *
isc_httpdurl_loadtime(const isc_httpdurl_t *url) {
return (&url->loadtime);
}
const isc_time_t *
isc_httpd_if_modified_since(const isc_httpd_t *httpd) {
return ((const isc_time_t *)&httpd->if_modified_since);
}