mirror of
https://gitlab.isc.org/isc-projects/bind9
synced 2025-08-28 21:17:54 +00:00
Rewrite the isc_httpd to be more robust. 1. Replace the hand-crafted HTTP request parser with picohttpparser for parsing the whole HTTP/1.0 and HTTP/1.1 requests. Limit the number of allowed headers to 10 (arbitrary number). 2. Replace the hand-crafted URL parser with isc_url_parse for parsing the URL from the HTTP request. 3. Increase the receive buffer to match the isc_netmgr buffers, so we can at least receive two full isc_nm_read()s. This makes the truncation processing much simpler. 4. Process the received buffer from single isc_nm_read() in a single loop and schedule the sends to be independent of each other. The first two changes makes the code simpler and rely on already existing libraries that we already had (isc_url based on nodejs) or are used elsewhere (picohttpparser). The second two changes remove the artificial "truncation" limit on parsing multiple request. Now only a request that has too many headers (currently 10) or is too big (so, the receive buffer fills up without reaching end of the request) will end the connection. We can be benevolent here with the limites, because the statschannel channel is by definition private and access must be allowed only to administrators of the server. There are no timers, no rate-limiting, no upper limit on the number of requests that can be served, etc.
1098 lines
26 KiB
C
1098 lines
26 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 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, ISC_NM_LISTEN_ONE, addr, httpd_newconn,
|
|
httpdmgr, 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);
|
|
|
|
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
|
|
httpd_request(isc_nmhandle_t *handle, isc_result_t eresult,
|
|
isc_region_t *region, void *arg) {
|
|
isc_result_t result;
|
|
isc_httpd_t *httpd = NULL;
|
|
isc_httpdmgr_t *mgr = (isc_httpdmgr_t *)arg;
|
|
isc_httpdurl_t *url = NULL;
|
|
isc_time_t now;
|
|
isc_region_t r;
|
|
bool is_compressed = false;
|
|
char datebuf[ISC_FORMATHTTPTIMESTAMP_SIZE];
|
|
size_t limit = 0;
|
|
size_t last_len;
|
|
|
|
httpd = isc_nmhandle_getdata(handle);
|
|
|
|
REQUIRE(VALID_HTTPD(httpd));
|
|
|
|
REQUIRE(httpd->handle == handle);
|
|
|
|
if (eresult != ISC_R_SUCCESS) {
|
|
goto cleanup_readhandle;
|
|
}
|
|
|
|
REQUIRE(region != NULL);
|
|
|
|
last_len = httpd->recvlen;
|
|
|
|
if (httpd->recvlen + region->length > sizeof(httpd->recvbuf)) {
|
|
goto cleanup_readhandle;
|
|
}
|
|
|
|
/* Store the received data into the recvbuf */
|
|
REQUIRE(httpd->recvlen + region->length < sizeof(httpd->recvbuf));
|
|
memmove(httpd->recvbuf + httpd->recvlen, region->base, region->length);
|
|
httpd->recvlen += region->length;
|
|
|
|
again:
|
|
result = process_request(httpd, last_len);
|
|
|
|
if (result == ISC_R_NOMORE) {
|
|
limit = sizeof(httpd->recvbuf) - httpd->recvlen;
|
|
if (region->length > limit) {
|
|
/*
|
|
* We need more data, but we don't have space to store
|
|
* it
|
|
*/
|
|
goto cleanup_readhandle;
|
|
}
|
|
|
|
memmove(httpd->recvbuf + httpd->recvlen, region->base,
|
|
region->length);
|
|
|
|
/* Just wait for more data */
|
|
return;
|
|
}
|
|
|
|
if (result != ISC_R_SUCCESS) {
|
|
goto cleanup_readhandle;
|
|
}
|
|
|
|
isc_time_now(&now);
|
|
isc_time_formathttptimestamp(&now, datebuf, sizeof(datebuf));
|
|
|
|
const char *path = "/";
|
|
size_t path_len = 1;
|
|
|
|
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);
|
|
|
|
isc_httpd_sendreq_t *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;
|
|
last_len = 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;
|
|
|
|
/*
|
|
* 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);
|
|
|
|
if ((httpd->flags & HTTPD_CLOSE) != 0) {
|
|
goto cleanup_readhandle;
|
|
}
|
|
|
|
/*
|
|
* The request was successfully completed;
|
|
*/
|
|
if (httpd->recvlen > 0) {
|
|
goto again;
|
|
}
|
|
|
|
return;
|
|
|
|
cleanup_readhandle:
|
|
isc_nm_read_stop(httpd->readhandle);
|
|
isc_nmhandle_detach(&httpd->readhandle);
|
|
}
|
|
|
|
void
|
|
isc_httpdmgr_shutdown(isc_httpdmgr_t **httpdmgrp) {
|
|
isc_httpdmgr_t *httpdmgr = NULL;
|
|
isc_httpd_t *httpd = NULL;
|
|
|
|
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_read_stop(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 result, void *arg) {
|
|
isc_httpd_sendreq_t *req = (isc_httpd_sendreq_t *)arg;
|
|
isc_httpd_t *httpd = req->httpd;
|
|
|
|
REQUIRE(VALID_HTTPD(httpd));
|
|
|
|
if (result != ISC_R_SUCCESS && httpd->readhandle != NULL) {
|
|
isc_nm_read_stop(httpd->readhandle);
|
|
isc_nmhandle_close(httpd->readhandle);
|
|
isc_nmhandle_detach(&httpd->readhandle);
|
|
}
|
|
|
|
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);
|
|
}
|