/src/bind9/lib/isc/netmgr/http.c

Source
/*
 * 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.
 */

#include <ctype.h>
#include <inttypes.h>
#include <limits.h>
#include <nghttp2/nghttp2.h>
#include <signal.h>
#include <string.h>

#include <isc/async.h>
#include <isc/base64.h>
#include <isc/log.h>
#include <isc/netmgr.h>
#include <isc/sockaddr.h>
#include <isc/tls.h>
#include <isc/url.h>
#include <isc/util.h>

#include "netmgr-int.h"

#define AUTHEXTRA 7

#define MAX_DNS_MESSAGE_SIZE (UINT16_MAX)

#define DNS_MEDIA_TYPE "application/dns-message"

/*
 * See https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Cache-Control
 * for additional details. Basically it means "avoid caching by any
 * means."
 */
#define DEFAULT_CACHE_CONTROL "no-cache, no-store, must-revalidate"

/*
 * If server during request processing surpasses any of the limits
 * below, it will just reset the stream without returning any error
 * codes in a response.  Ideally, these parameters should be
 * configurable both globally and per every HTTP endpoint description
 * in the configuration file, but for now it should be enough.
 */

/*
 * 128K should be enough to encode 64K of data into base64url inside GET
 * request and have extra space for other headers
 */
#define MAX_ALLOWED_DATA_IN_HEADERS (MAX_DNS_MESSAGE_SIZE * 2)

#define MAX_ALLOWED_DATA_IN_POST \
  (MAX_DNS_MESSAGE_SIZE + MAX_DNS_MESSAGE_SIZE / 2)

#define HEADER_MATCH(header, name, namelen)   \
  (((namelen) == sizeof(header) - 1) && \
   (strncasecmp((header), (const char *)(name), (namelen)) == 0))

#define MIN_SUCCESSFUL_HTTP_STATUS (200)
#define MAX_SUCCESSFUL_HTTP_STATUS (299)

/* This definition sets the upper limit of pending write buffer to an
 * adequate enough value. That is done mostly to fight a limitation
 * for a max TLS record size in flamethrower (2K).  In a perfect world
 * this constant should not be required, if we ever move closer to
 * that state, the constant, and corresponding code, should be
 * removed. For now the limit seems adequate enough to fight
 * "tinygrams" problem. */
#define FLUSH_HTTP_WRITE_BUFFER_AFTER (1536)

/* This switch is here mostly to test the code interoperability with
 * buggy implementations */
#define ENABLE_HTTP_WRITE_BUFFERING 1

#define SUCCESSFUL_HTTP_STATUS(code)             \
  ((code) >= MIN_SUCCESSFUL_HTTP_STATUS && \
   (code) <= MAX_SUCCESSFUL_HTTP_STATUS)

#define INITIAL_DNS_MESSAGE_BUFFER_SIZE (512)

/*
 * The value should be small enough to not allow a server to open too
 * many streams at once. It should not be too small either because
 * the incoming data will be split into too many chunks with each of
 * them processed asynchronously.
 */
#define INCOMING_DATA_CHUNK_SIZE (256)

/*
 * Often processing a chunk does not change the number of streams. In
 * that case we can process more than once, but we still should have a
 * hard limit on that.
 */
#define INCOMING_DATA_MAX_CHUNKS_AT_ONCE (4)

/*
 * These constants define the grace period to help detect flooding clients.
 *
 * The first one defines how much data can be processed before opening
 * a first stream and received at least some useful (=DNS) data.
 *
 * The second one defines how much data from a client we read before
 * trying to drop a clients who sends not enough useful data.
 *
 * The third constant defines how many streams we agree to process
 * before checking if there was at least one DNS request received.
 */
#define INCOMING_DATA_INITIAL_STREAM_SIZE (1536)
#define INCOMING_DATA_GRACE_SIZE    (MAX_ALLOWED_DATA_IN_HEADERS)
#define MAX_STREAMS_BEFORE_FIRST_REQUEST  (50)

typedef struct isc_nm_http_response_status {
  size_t code;
  size_t content_length;
  bool content_type_valid;
} isc_nm_http_response_status_t;

typedef struct http_cstream {
  isc_nm_recv_cb_t read_cb;
  void *read_cbarg;
  isc_nm_cb_t connect_cb;
  void *connect_cbarg;

  bool sending;
  bool reading;

  char *uri;
  isc_url_parser_t up;

  char *authority;
  size_t authoritylen;
  char *path;

  isc_buffer_t *rbuf;

  size_t pathlen;
  int32_t stream_id;

  bool post; /* POST or GET */
  isc_buffer_t *postdata;
  char *GET_path;
  size_t GET_path_len;

  isc_nm_http_response_status_t response_status;
  isc_nmsocket_t *httpsock;
  ISC_LINK(struct http_cstream) link;
} http_cstream_t;

#define HTTP2_SESSION_MAGIC    ISC_MAGIC('H', '2', 'S', 'S')
#define VALID_HTTP2_SESSION(t) ISC_MAGIC_VALID(t, HTTP2_SESSION_MAGIC)

typedef ISC_LIST(isc__nm_uvreq_t) isc__nm_http_pending_callbacks_t;

struct isc_nm_http_session {
  unsigned int magic;
  isc_refcount_t references;
  isc_mem_t *mctx;

  size_t sending;
  bool reading;
  bool closed;
  bool closing;

  nghttp2_session *ngsession;
  bool client;

  ISC_LIST(http_cstream_t) cstreams;
  ISC_LIST(isc_nmsocket_h2_t) sstreams;
  size_t nsstreams;
  uint64_t total_opened_sstreams;

  isc_nmhandle_t *handle;
  isc_nmhandle_t *client_httphandle;
  isc_nmsocket_t *serversocket;

  isc_buffer_t *buf;

  isc_tlsctx_t *tlsctx;
  uint32_t max_concurrent_streams;

  isc__nm_http_pending_callbacks_t pending_write_callbacks;
  isc_buffer_t *pending_write_data;

  size_t data_in_flight;

  bool async_queued;

  /*
   * The statistical values below are for usage on server-side
   * only. They are meant to detect clients that are taking too many
   * resources from the server.
   */
  uint64_t received;  /* How many requests have been received. */
  uint64_t submitted; /* How many responses were submitted to send */
  uint64_t processed; /* How many responses were processed. */

  uint64_t processed_incoming_data;
  uint64_t processed_useful_data; /* DNS data */
};

typedef enum isc_http_error_responses {
  ISC_HTTP_ERROR_SUCCESS,          /* 200 */
  ISC_HTTP_ERROR_NOT_FOUND,        /* 404 */
  ISC_HTTP_ERROR_PAYLOAD_TOO_LARGE,      /* 413 */
  ISC_HTTP_ERROR_URI_TOO_LONG,         /* 414 */
  ISC_HTTP_ERROR_UNSUPPORTED_MEDIA_TYPE, /* 415 */
  ISC_HTTP_ERROR_BAD_REQUEST,        /* 400 */
  ISC_HTTP_ERROR_NOT_IMPLEMENTED,        /* 501 */
  ISC_HTTP_ERROR_GENERIC,          /* 500 Internal Server Error */
  ISC_HTTP_ERROR_MAX
} isc_http_error_responses_t;

typedef struct isc_http_send_req {
  isc_nm_http_session_t *session;
  isc_nmhandle_t *transphandle;
  isc_nmhandle_t *httphandle;
  isc_nm_cb_t cb;
  void *cbarg;
  isc_buffer_t *pending_write_data;
  isc__nm_http_pending_callbacks_t pending_write_callbacks;
  uint64_t submitted;
} isc_http_send_req_t;

#define HTTP_ENDPOINTS_MAGIC  ISC_MAGIC('H', 'T', 'E', 'P')
#define VALID_HTTP_ENDPOINTS(t) ISC_MAGIC_VALID(t, HTTP_ENDPOINTS_MAGIC)

#define HTTP_HANDLER_MAGIC    ISC_MAGIC('H', 'T', 'H', 'L')
#define VALID_HTTP_HANDLER(t) ISC_MAGIC_VALID(t, HTTP_HANDLER_MAGIC)

static void
http_send_outgoing(isc_nm_http_session_t *session, isc_nmhandle_t *httphandle,
       isc_nm_cb_t cb, void *cbarg);

static void
http_log_flooding_peer(isc_nm_http_session_t *session);

static bool
http_is_flooding_peer(isc_nm_http_session_t *session);

static ssize_t
http_process_input_data(isc_nm_http_session_t *session,
      isc_buffer_t *input_data);

static inline bool
http_too_many_active_streams(isc_nm_http_session_t *session);

static void
http_do_bio(isc_nm_http_session_t *session, isc_nmhandle_t *send_httphandle,
      isc_nm_cb_t send_cb, void *send_cbarg);

static void
http_do_bio_async(isc_nm_http_session_t *session);

static void
failed_httpstream_read_cb(isc_nmsocket_t *sock, isc_result_t result,
        isc_nm_http_session_t *session);

static void
client_call_failed_read_cb(isc_result_t result, isc_nm_http_session_t *session);

static void
server_call_failed_read_cb(isc_result_t result, isc_nm_http_session_t *session);

static void
failed_read_cb(isc_result_t result, isc_nm_http_session_t *session);

static isc_result_t
server_send_error_response(const isc_http_error_responses_t error,
         nghttp2_session *ngsession, isc_nmsocket_t *socket);

static isc_result_t
client_send(isc_nmhandle_t *handle, const isc_region_t *region);

static void
finish_http_session(isc_nm_http_session_t *session);

static void
http_transpost_tcp_nodelay(isc_nmhandle_t *transphandle);

static void
call_pending_callbacks(isc__nm_http_pending_callbacks_t pending_callbacks,
           isc_result_t result);

static void
server_call_cb(isc_nmsocket_t *socket, const isc_result_t result,
         isc_region_t *data);

static isc_nm_httphandler_t *
http_endpoints_find(const char *request_path,
        isc_nm_http_endpoints_t *restrict eps);

static void
http_init_listener_endpoints(isc_nmsocket_t *listener,
           isc_nm_http_endpoints_t *epset);

static void
http_cleanup_listener_endpoints(isc_nmsocket_t *listener);

static isc_nm_http_endpoints_t *
http_get_listener_endpoints(isc_nmsocket_t *listener, const isc_tid_t tid);

static void
http_initsocket(isc_nmsocket_t *sock);

static void *
http_malloc(size_t sz, isc_mem_t *mctx) {
  return isc_mem_allocate(mctx, sz);
}

static void *
http_calloc(size_t n, size_t sz, isc_mem_t *mctx) {
  return isc_mem_callocate(mctx, n, sz);
}

static void *
http_realloc(void *p, size_t newsz, isc_mem_t *mctx) {
  return isc_mem_reallocate(mctx, p, newsz);
}

static void
http_free(void *p, isc_mem_t *mctx) {
  if (p == NULL) { /* as standard free() behaves */
    return;
  }
  isc_mem_free(mctx, p);
}

static void
init_nghttp2_mem(isc_mem_t *mctx, nghttp2_mem *mem) {
  *mem = (nghttp2_mem){ .malloc = (nghttp2_malloc)http_malloc,
            .calloc = (nghttp2_calloc)http_calloc,
            .realloc = (nghttp2_realloc)http_realloc,
            .free = (nghttp2_free)http_free,
            .mem_user_data = mctx };
}

static void
new_session(isc_mem_t *mctx, isc_tlsctx_t *tctx,
      isc_nm_http_session_t **sessionp) {
  isc_nm_http_session_t *session = NULL;

  REQUIRE(sessionp != NULL && *sessionp == NULL);
  REQUIRE(mctx != NULL);

  session = isc_mem_get(mctx, sizeof(isc_nm_http_session_t));
  *session = (isc_nm_http_session_t){ .magic = HTTP2_SESSION_MAGIC,
              .tlsctx = tctx };
  isc_refcount_init(&session->references, 1);
  isc_mem_attach(mctx, &session->mctx);
  ISC_LIST_INIT(session->cstreams);
  ISC_LIST_INIT(session->sstreams);
  ISC_LIST_INIT(session->pending_write_callbacks);

  *sessionp = session;
}

void
isc__nm_httpsession_attach(isc_nm_http_session_t *source,
         isc_nm_http_session_t **targetp) {
  REQUIRE(VALID_HTTP2_SESSION(source));
  REQUIRE(targetp != NULL && *targetp == NULL);

  isc_refcount_increment(&source->references);

  *targetp = source;
}

void
isc__nm_httpsession_detach(isc_nm_http_session_t **sessionp) {
  isc_nm_http_session_t *session = NULL;

  REQUIRE(sessionp != NULL);

  session = *sessionp;
  *sessionp = NULL;

  REQUIRE(VALID_HTTP2_SESSION(session));

  if (isc_refcount_decrement(&session->references) > 1) {
    return;
  }

  finish_http_session(session);

  INSIST(ISC_LIST_EMPTY(session->sstreams));
  INSIST(ISC_LIST_EMPTY(session->cstreams));

  if (session->ngsession != NULL) {
    nghttp2_session_del(session->ngsession);
    session->ngsession = NULL;
  }

  if (session->buf != NULL) {
    isc_buffer_free(&session->buf);
  }

  /* We need an acquire memory barrier here */
  (void)isc_refcount_current(&session->references);

  session->magic = 0;
  isc_mem_putanddetach(&session->mctx, session,
           sizeof(isc_nm_http_session_t));
}

isc_nmhandle_t *
isc__nm_httpsession_handle(isc_nm_http_session_t *session) {
  REQUIRE(VALID_HTTP2_SESSION(session));

  return session->handle;
}

bool
isc__nm_httpsession_active(isc_nm_http_session_t *session) {
  REQUIRE(VALID_HTTP2_SESSION(session));

  return !session->closed && !session->closing && session->handle != NULL;
}

static http_cstream_t *
find_http_cstream(int32_t stream_id, isc_nm_http_session_t *session) {
  REQUIRE(VALID_HTTP2_SESSION(session));

  if (ISC_LIST_EMPTY(session->cstreams)) {
    return NULL;
  }

  ISC_LIST_FOREACH(session->cstreams, cstream, link) {
    if (cstream->stream_id == stream_id) {
      /* LRU-like behaviour */
      if (ISC_LIST_HEAD(session->cstreams) != cstream) {
        ISC_LIST_UNLINK(session->cstreams, cstream,
            link);
        ISC_LIST_PREPEND(session->cstreams, cstream,
             link);
      }

      return cstream;
    }
  }

  return NULL;
}

static isc_result_t
new_http_cstream(isc_nmsocket_t *sock, http_cstream_t **streamp) {
  isc_mem_t *mctx = sock->worker->mctx;
  const char *uri = NULL;
  bool post;
  http_cstream_t *stream = NULL;
  isc_result_t result;

  uri = sock->h2->session->handle->sock->h2->connect.uri;
  post = sock->h2->session->handle->sock->h2->connect.post;

  stream = isc_mem_get(mctx, sizeof(http_cstream_t));
  *stream = (http_cstream_t){ .stream_id = -1,
            .post = post,
            .uri = isc_mem_strdup(mctx, uri) };
  ISC_LINK_INIT(stream, link);

  result = isc_url_parse(stream->uri, strlen(stream->uri), 0,
             &stream->up);
  if (result != ISC_R_SUCCESS) {
    isc_mem_free(mctx, stream->uri);
    isc_mem_put(mctx, stream, sizeof(http_cstream_t));
    return result;
  }

  isc__nmsocket_attach(sock, &stream->httpsock);
  stream->authoritylen = stream->up.field_data[ISC_UF_HOST].len;
  stream->authority = isc_mem_get(mctx, stream->authoritylen + AUTHEXTRA);
  memmove(stream->authority, &uri[stream->up.field_data[ISC_UF_HOST].off],
    stream->up.field_data[ISC_UF_HOST].len);

  if (stream->up.field_set & (1 << ISC_UF_PORT)) {
    stream->authoritylen += (size_t)snprintf(
      stream->authority +
        stream->up.field_data[ISC_UF_HOST].len,
      AUTHEXTRA, ":%u", stream->up.port);
  }

  /* If we don't have path in URI, we use "/" as path. */
  stream->pathlen = 1;
  if (stream->up.field_set & (1 << ISC_UF_PATH)) {
    stream->pathlen = stream->up.field_data[ISC_UF_PATH].len;
  }
  if (stream->up.field_set & (1 << ISC_UF_QUERY)) {
    /* +1 for '?' character */
    stream->pathlen +=
      (size_t)(stream->up.field_data[ISC_UF_QUERY].len + 1);
  }

  stream->path = isc_mem_get(mctx, stream->pathlen);
  if (stream->up.field_set & (1 << ISC_UF_PATH)) {
    memmove(stream->path,
      &uri[stream->up.field_data[ISC_UF_PATH].off],
      stream->up.field_data[ISC_UF_PATH].len);
  } else {
    stream->path[0] = '/';
  }

  if (stream->up.field_set & (1 << ISC_UF_QUERY)) {
    stream->path[stream->pathlen -
           stream->up.field_data[ISC_UF_QUERY].len - 1] = '?';
    memmove(stream->path + stream->pathlen -
        stream->up.field_data[ISC_UF_QUERY].len,
      &uri[stream->up.field_data[ISC_UF_QUERY].off],
      stream->up.field_data[ISC_UF_QUERY].len);
  }

  isc_buffer_allocate(mctx, &stream->rbuf,
          INITIAL_DNS_MESSAGE_BUFFER_SIZE);

  ISC_LIST_PREPEND(sock->h2->session->cstreams, stream, link);
  *streamp = stream;

  return ISC_R_SUCCESS;
}

static void
put_http_cstream(isc_mem_t *mctx, http_cstream_t *stream) {
  isc_mem_put(mctx, stream->path, stream->pathlen);
  isc_mem_put(mctx, stream->authority,
        stream->up.field_data[ISC_UF_HOST].len + AUTHEXTRA);
  isc_mem_free(mctx, stream->uri);
  if (stream->GET_path != NULL) {
    isc_mem_free(mctx, stream->GET_path);
    stream->GET_path_len = 0;
  }

  if (stream->postdata != NULL) {
    INSIST(stream->post);
    isc_buffer_free(&stream->postdata);
  }

  if (stream == stream->httpsock->h2->connect.cstream) {
    stream->httpsock->h2->connect.cstream = NULL;
  }
  if (ISC_LINK_LINKED(stream, link)) {
    ISC_LIST_UNLINK(stream->httpsock->h2->session->cstreams, stream,
        link);
  }
  isc__nmsocket_detach(&stream->httpsock);

  isc_buffer_free(&stream->rbuf);
  isc_mem_put(mctx, stream, sizeof(http_cstream_t));
}

static void
finish_http_session(isc_nm_http_session_t *session) {
  if (session->closed) {
    return;
  }

  if (session->handle != NULL) {
    if (!session->closed) {
      session->closed = true;
      session->reading = false;
      isc_nm_read_stop(session->handle);
      isc__nmsocket_timer_stop(session->handle->sock);
      isc_nmhandle_close(session->handle);
    }

    /*
     * Free any unprocessed incoming data in order to not process
     * it during indirect calls to http_do_bio() that might happen
     * when calling the failed callbacks.
     */
    if (session->buf != NULL) {
      isc_buffer_free(&session->buf);
    }

    if (session->client) {
      client_call_failed_read_cb(ISC_R_UNEXPECTED, session);
    } else {
      server_call_failed_read_cb(ISC_R_UNEXPECTED, session);
    }

    call_pending_callbacks(session->pending_write_callbacks,
               ISC_R_UNEXPECTED);
    ISC_LIST_INIT(session->pending_write_callbacks);

    if (session->pending_write_data != NULL) {
      isc_buffer_free(&session->pending_write_data);
    }

    isc_nmhandle_detach(&session->handle);
  }

  if (session->client_httphandle != NULL) {
    isc_nmhandle_detach(&session->client_httphandle);
  }

  INSIST(ISC_LIST_EMPTY(session->cstreams));

  /* detach from server socket */
  if (session->serversocket != NULL) {
    isc__nmsocket_detach(&session->serversocket);
  }
  session->closed = true;
}

static int
on_client_data_chunk_recv_callback(int32_t stream_id, const uint8_t *data,
           size_t len, isc_nm_http_session_t *session) {
  http_cstream_t *cstream = find_http_cstream(stream_id, session);

  if (cstream != NULL) {
    size_t new_rbufsize = len;
    INSIST(cstream->rbuf != NULL);
    new_rbufsize += isc_buffer_usedlength(cstream->rbuf);
    if (new_rbufsize <= MAX_DNS_MESSAGE_SIZE &&
        new_rbufsize <= cstream->response_status.content_length)
    {
      isc_buffer_putmem(cstream->rbuf, data, len);
    } else {
      return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE;
    }
  } else {
    return NGHTTP2_ERR_CALLBACK_FAILURE;
  }

  return 0;
}

static int
on_server_data_chunk_recv_callback(int32_t stream_id, const uint8_t *data,
           size_t len, isc_nm_http_session_t *session) {
  isc_nmsocket_h2_t *h2 = ISC_LIST_HEAD(session->sstreams);
  isc_mem_t *mctx = h2->psock->worker->mctx;

  while (h2 != NULL) {
    if (stream_id == h2->stream_id) {
      if (isc_buffer_base(&h2->rbuf) == NULL) {
        isc_buffer_init(
          &h2->rbuf,
          isc_mem_allocate(mctx,
               h2->content_length),
          h2->content_length);
      }
      size_t new_bufsize = isc_buffer_usedlength(&h2->rbuf) +
               len;
      if (new_bufsize <= h2->content_length) {
        session->processed_useful_data += len;
        isc_buffer_putmem(&h2->rbuf, data, len);
        break;
      }

      return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE;
    }
    h2 = ISC_LIST_NEXT(h2, link);
  }
  if (h2 == NULL) {
    return NGHTTP2_ERR_CALLBACK_FAILURE;
  }

  return 0;
}

static int
on_data_chunk_recv_callback(nghttp2_session *ngsession, uint8_t flags,
          int32_t stream_id, const uint8_t *data, size_t len,
          void *user_data) {
  isc_nm_http_session_t *session = (isc_nm_http_session_t *)user_data;
  int rv;

  UNUSED(ngsession);
  UNUSED(flags);

  if (session->client) {
    rv = on_client_data_chunk_recv_callback(stream_id, data, len,
              session);
  } else {
    rv = on_server_data_chunk_recv_callback(stream_id, data, len,
              session);
  }

  return rv;
}

static void
call_unlink_cstream_readcb(http_cstream_t *cstream,
         isc_nm_http_session_t *session,
         isc_result_t result) {
  isc_region_t read_data;
  REQUIRE(VALID_HTTP2_SESSION(session));
  REQUIRE(cstream != NULL);
  ISC_LIST_UNLINK(session->cstreams, cstream, link);
  INSIST(VALID_NMHANDLE(session->client_httphandle));
  isc_buffer_usedregion(cstream->rbuf, &read_data);
  cstream->read_cb(session->client_httphandle, result, &read_data,
       cstream->read_cbarg);
  if (result == ISC_R_SUCCESS) {
    isc__nmsocket_timer_restart(session->handle->sock);
  }
  put_http_cstream(session->mctx, cstream);
}

static int
on_client_stream_close_callback(int32_t stream_id,
        isc_nm_http_session_t *session) {
  http_cstream_t *cstream = find_http_cstream(stream_id, session);

  if (cstream != NULL) {
    isc_result_t result =
      SUCCESSFUL_HTTP_STATUS(cstream->response_status.code)
        ? ISC_R_SUCCESS
        : ISC_R_FAILURE;
    call_unlink_cstream_readcb(cstream, session, result);
    if (ISC_LIST_EMPTY(session->cstreams)) {
      int rv = 0;
      rv = nghttp2_session_terminate_session(
        session->ngsession, NGHTTP2_NO_ERROR);
      if (rv != 0) {
        return rv;
      }
      /* Mark the session as closing one to finish it on a
       * subsequent call to http_do_bio() */
      session->closing = true;
    }
  } else {
    return NGHTTP2_ERR_CALLBACK_FAILURE;
  }

  return 0;
}

static int
on_server_stream_close_callback(int32_t stream_id,
        isc_nm_http_session_t *session) {
  isc_nmsocket_t *sock = nghttp2_session_get_stream_user_data(
    session->ngsession, stream_id);
  int rv = 0;

  ISC_LIST_UNLINK(session->sstreams, sock->h2, link);
  session->nsstreams--;
  if (sock->h2->request_received) {
    session->submitted++;
  }

  /*
   * By making a call to isc__nmsocket_prep_destroy(), we ensure that
   * the socket gets marked as inactive, allowing the HTTP/2 data
   * associated with it to be properly disposed of eventually.
   *
   * An HTTP/2 stream socket will normally be marked as inactive in
   * the normal course of operation. However, when browsers terminate
   * HTTP/2 streams prematurely (e.g. by sending RST_STREAM),
   * corresponding sockets can remain marked as active, retaining
   * references to the HTTP/2 data (most notably the session objects),
   * preventing them from being correctly freed and leading to BIND
   * hanging on shutdown.  Calling isc__nmsocket_prep_destroy()
   * ensures that this will not happen.
   */
  isc__nmsocket_prep_destroy(sock);
  isc__nmsocket_detach(&sock);
  return rv;
}

static int
on_stream_close_callback(nghttp2_session *ngsession, int32_t stream_id,
       uint32_t error_code, void *user_data) {
  isc_nm_http_session_t *session = (isc_nm_http_session_t *)user_data;
  int rv = 0;

  REQUIRE(VALID_HTTP2_SESSION(session));
  REQUIRE(session->ngsession == ngsession);

  UNUSED(error_code);

  if (session->client) {
    rv = on_client_stream_close_callback(stream_id, session);
  } else {
    rv = on_server_stream_close_callback(stream_id, session);
  }

  return rv;
}

static bool
client_handle_status_header(http_cstream_t *cstream, const uint8_t *value,
          const size_t valuelen) {
  char tmp[32] = { 0 };
  const size_t tmplen = sizeof(tmp) - 1;

  strncpy(tmp, (const char *)value, ISC_MIN(tmplen, valuelen));
  cstream->response_status.code = strtoul(tmp, NULL, 10);

  if (SUCCESSFUL_HTTP_STATUS(cstream->response_status.code)) {
    return true;
  }

  return false;
}

static bool
client_handle_content_length_header(http_cstream_t *cstream,
            const uint8_t *value,
            const size_t valuelen) {
  char tmp[32] = { 0 };
  const size_t tmplen = sizeof(tmp) - 1;

  strncpy(tmp, (const char *)value, ISC_MIN(tmplen, valuelen));
  cstream->response_status.content_length = strtoul(tmp, NULL, 10);

  if (cstream->response_status.content_length == 0 ||
      cstream->response_status.content_length > MAX_DNS_MESSAGE_SIZE)
  {
    return false;
  }

  return true;
}

static bool
client_handle_content_type_header(http_cstream_t *cstream, const uint8_t *value,
          const size_t valuelen) {
  const char type_dns_message[] = DNS_MEDIA_TYPE;

  if (HEADER_MATCH(type_dns_message, value, valuelen)) {
    cstream->response_status.content_type_valid = true;
    return true;
  }

  return false;
}

static int
client_on_header_callback(nghttp2_session *ngsession,
        const nghttp2_frame *frame, const uint8_t *name,
        size_t namelen, const uint8_t *value, size_t valuelen,
        uint8_t flags, void *user_data) {
  isc_nm_http_session_t *session = (isc_nm_http_session_t *)user_data;
  http_cstream_t *cstream = NULL;
  const char status[] = ":status";
  const char content_length[] = "Content-Length";
  const char content_type[] = "Content-Type";
  bool header_ok = true;

  REQUIRE(VALID_HTTP2_SESSION(session));
  REQUIRE(session->client);

  UNUSED(flags);
  UNUSED(ngsession);

  cstream = find_http_cstream(frame->hd.stream_id, session);
  if (cstream == NULL) {
    /*
     * This could happen in two cases:
     * - the server sent us bad data, or
     * - we closed the session prematurely before receiving all
     *   responses (i.e., because of a belated or partial response).
     */
    return NGHTTP2_ERR_CALLBACK_FAILURE;
  }

  INSIST(!ISC_LIST_EMPTY(session->cstreams));

  switch (frame->hd.type) {
  case NGHTTP2_HEADERS:
    if (frame->headers.cat != NGHTTP2_HCAT_RESPONSE) {
      break;
    }

    if (HEADER_MATCH(status, name, namelen)) {
      header_ok = client_handle_status_header(cstream, value,
                valuelen);
    } else if (HEADER_MATCH(content_length, name, namelen)) {
      header_ok = client_handle_content_length_header(
        cstream, value, valuelen);
    } else if (HEADER_MATCH(content_type, name, namelen)) {
      header_ok = client_handle_content_type_header(
        cstream, value, valuelen);
    }
    break;
  }

  if (!header_ok) {
    return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE;
  }

  return 0;
}

static void
initialize_nghttp2_client_session(isc_nm_http_session_t *session) {
  nghttp2_session_callbacks *callbacks = NULL;
  nghttp2_option *option = NULL;
  nghttp2_mem mem;

  init_nghttp2_mem(session->mctx, &mem);
  RUNTIME_CHECK(nghttp2_session_callbacks_new(&callbacks) == 0);
  RUNTIME_CHECK(nghttp2_option_new(&option) == 0);

#if NGHTTP2_VERSION_NUM >= (0x010c00)
  nghttp2_option_set_max_send_header_block_length(
    option, MAX_ALLOWED_DATA_IN_HEADERS);
#endif

  nghttp2_session_callbacks_set_on_data_chunk_recv_callback(
    callbacks, on_data_chunk_recv_callback);

  nghttp2_session_callbacks_set_on_stream_close_callback(
    callbacks, on_stream_close_callback);

  nghttp2_session_callbacks_set_on_header_callback(
    callbacks, client_on_header_callback);

  RUNTIME_CHECK(nghttp2_session_client_new3(&session->ngsession,
              callbacks, session, option,
              &mem) == 0);

  nghttp2_option_del(option);
  nghttp2_session_callbacks_del(callbacks);
}

static bool
send_client_connection_header(isc_nm_http_session_t *session) {
  nghttp2_settings_entry iv[] = { { NGHTTP2_SETTINGS_ENABLE_PUSH, 0 } };
  int rv;

  rv = nghttp2_submit_settings(session->ngsession, NGHTTP2_FLAG_NONE, iv,
             sizeof(iv) / sizeof(iv[0]));
  if (rv != 0) {
    return false;
  }

  return true;
}

#define MAKE_NV(NAME, VALUE, VALUELEN)                                 \
  { (uint8_t *)(uintptr_t)(NAME), (uint8_t *)(uintptr_t)(VALUE), \
    sizeof(NAME) - 1, VALUELEN, NGHTTP2_NV_FLAG_NONE }

#define MAKE_NV2(NAME, VALUE)                                          \
  { (uint8_t *)(uintptr_t)(NAME), (uint8_t *)(uintptr_t)(VALUE), \
    sizeof(NAME) - 1, sizeof(VALUE) - 1, NGHTTP2_NV_FLAG_NONE }

static ssize_t
client_read_callback(nghttp2_session *ngsession, int32_t stream_id,
         uint8_t *buf, size_t length, uint32_t *data_flags,
         nghttp2_data_source *source, void *user_data) {
  isc_nm_http_session_t *session = (isc_nm_http_session_t *)user_data;
  http_cstream_t *cstream = NULL;

  REQUIRE(session->client);
  REQUIRE(!ISC_LIST_EMPTY(session->cstreams));

  UNUSED(ngsession);
  UNUSED(source);

  cstream = find_http_cstream(stream_id, session);
  if (!cstream || cstream->stream_id != stream_id) {
    /* We haven't found the stream, so we are not reading */
    return NGHTTP2_ERR_CALLBACK_FAILURE;
  }

  if (cstream->post) {
    size_t len = isc_buffer_remaininglength(cstream->postdata);

    if (len > length) {
      len = length;
    }

    if (len > 0) {
      memmove(buf, isc_buffer_current(cstream->postdata),
        len);
      isc_buffer_forward(cstream->postdata, len);
    }

    if (isc_buffer_remaininglength(cstream->postdata) == 0) {
      *data_flags |= NGHTTP2_DATA_FLAG_EOF;
    }

    return len;
  } else {
    *data_flags |= NGHTTP2_DATA_FLAG_EOF;
    return 0;
  }

  return 0;
}

/*
 * Send HTTP request to the remote peer.
 */
static isc_result_t
client_submit_request(isc_nm_http_session_t *session, http_cstream_t *stream) {
  int32_t stream_id;
  char *uri = stream->uri;
  isc_url_parser_t *up = &stream->up;
  nghttp2_data_provider dp;

  if (stream->post) {
    char p[64];
    snprintf(p, sizeof(p), "%u",
       isc_buffer_usedlength(stream->postdata));
    nghttp2_nv hdrs[] = {
      MAKE_NV2(":method", "POST"),
      MAKE_NV(":scheme",
        &uri[up->field_data[ISC_UF_SCHEMA].off],
        up->field_data[ISC_UF_SCHEMA].len),
      MAKE_NV(":authority", stream->authority,
        stream->authoritylen),
      MAKE_NV(":path", stream->path, stream->pathlen),
      MAKE_NV2("content-type", DNS_MEDIA_TYPE),
      MAKE_NV2("accept", DNS_MEDIA_TYPE),
      MAKE_NV("content-length", p, strlen(p)),
      MAKE_NV2("cache-control", DEFAULT_CACHE_CONTROL)
    };

    dp = (nghttp2_data_provider){ .read_callback =
                  client_read_callback };
    stream_id = nghttp2_submit_request(
      session->ngsession, NULL, hdrs,
      sizeof(hdrs) / sizeof(hdrs[0]), &dp, stream);
  } else {
    INSIST(stream->GET_path != NULL);
    INSIST(stream->GET_path_len != 0);
    nghttp2_nv hdrs[] = {
      MAKE_NV2(":method", "GET"),
      MAKE_NV(":scheme",
        &uri[up->field_data[ISC_UF_SCHEMA].off],
        up->field_data[ISC_UF_SCHEMA].len),
      MAKE_NV(":authority", stream->authority,
        stream->authoritylen),
      MAKE_NV(":path", stream->GET_path,
        stream->GET_path_len),
      MAKE_NV2("accept", DNS_MEDIA_TYPE),
      MAKE_NV2("cache-control", DEFAULT_CACHE_CONTROL)
    };

    dp = (nghttp2_data_provider){ .read_callback =
                  client_read_callback };
    stream_id = nghttp2_submit_request(
      session->ngsession, NULL, hdrs,
      sizeof(hdrs) / sizeof(hdrs[0]), &dp, stream);
  }
  if (stream_id < 0) {
    return ISC_R_FAILURE;
  }

  stream->stream_id = stream_id;

  return ISC_R_SUCCESS;
}

static inline size_t
http_in_flight_data_size(isc_nm_http_session_t *session) {
  size_t in_flight = 0;

  if (session->pending_write_data != NULL) {
    in_flight += isc_buffer_usedlength(session->pending_write_data);
  }

  in_flight += session->data_in_flight;

  return in_flight;
}

static ssize_t
http_process_input_data(isc_nm_http_session_t *session,
      isc_buffer_t *input_data) {
  ssize_t readlen = 0;
  ssize_t processed = 0;
  isc_region_t chunk = { 0 };
  size_t before, after;
  size_t i;

  REQUIRE(VALID_HTTP2_SESSION(session));
  REQUIRE(input_data != NULL);

  if (!isc__nm_httpsession_active(session)) {
    return 0;
  }

  /*
   * For clients that initiate request themselves just process
   * everything.
   */
  if (session->client) {
    isc_buffer_remainingregion(input_data, &chunk);
    if (chunk.length == 0) {
      return 0;
    }

    readlen = nghttp2_session_mem_recv(session->ngsession,
               chunk.base, chunk.length);

    if (readlen >= 0) {
      isc_buffer_forward(input_data, readlen);
      session->processed_incoming_data += readlen;
    }

    return readlen;
  }

  /*
   * If no streams are created during processing, we might process
   * more than one chunk at a time. Still we should not overdo that
   * to avoid processing too much data at once as such behaviour is
   * known for trashing the memory allocator at times.
   */
  for (before = after = session->nsstreams, i = 0;
       after <= before && i < INCOMING_DATA_MAX_CHUNKS_AT_ONCE;
       after = session->nsstreams, i++)
  {
    const uint64_t active_streams =
      (session->received - session->processed);

    /*
     * If there is too much outgoing data in flight - let's not
     * process any incoming data, as it could lead to piling up
     * too much send data in send buffers. With many clients
     * connected it can lead to excessive memory consumption on
     * the server instance.
     */
    const size_t in_flight = http_in_flight_data_size(session);
    if (in_flight >= ISC_NETMGR_TCP_SENDBUF_SIZE) {
      break;
    }

    /*
     * If we have reached the maximum number of streams used, we
     * might stop processing for now, as nghttp2 will happily
     * consume as much data as possible.
     */
    if (session->nsstreams >= session->max_concurrent_streams &&
        active_streams > 0)
    {
      break;
    }

    if (http_too_many_active_streams(session)) {
      break;
    }

    isc_buffer_remainingregion(input_data, &chunk);
    if (chunk.length == 0) {
      break;
    }

    chunk.length = ISC_MIN(chunk.length, INCOMING_DATA_CHUNK_SIZE);

    readlen = nghttp2_session_mem_recv(session->ngsession,
               chunk.base, chunk.length);

    if (readlen >= 0) {
      isc_buffer_forward(input_data, readlen);
      session->processed_incoming_data += readlen;
      processed += readlen;
    } else {
      isc_buffer_clear(input_data);
      return readlen;
    }
  }

  return processed;
}

static void
http_log_flooding_peer(isc_nm_http_session_t *session) {
  const int log_level = ISC_LOG_DEBUG(1);
  if (session->handle != NULL && isc_log_wouldlog(log_level)) {
    char client_sabuf[ISC_SOCKADDR_FORMATSIZE];
    char local_sabuf[ISC_SOCKADDR_FORMATSIZE];

    isc_sockaddr_format(&session->handle->sock->peer, client_sabuf,
            sizeof(client_sabuf));
    isc_sockaddr_format(&session->handle->sock->iface, local_sabuf,
            sizeof(local_sabuf));
    isc__nmsocket_log(session->handle->sock, log_level,
          "Dropping a flooding HTTP/2 peer "
          "%s (on %s) - processed: %" PRIu64
          " bytes, of them useful: %" PRIu64 "",
          client_sabuf, local_sabuf,
          session->processed_incoming_data,
          session->processed_useful_data);
  }
}

static bool
http_is_flooding_peer(isc_nm_http_session_t *session) {
  if (session->client) {
    return false;
  }

  /*
   * A flooding client can try to open a lot of streams before
   * submitting a request. Let's drop such clients.
   */
  if (session->received == 0 &&
      session->total_opened_sstreams > MAX_STREAMS_BEFORE_FIRST_REQUEST)
  {
    return true;
  }

  /*
   * We have processed enough data to open at least one stream and
   * get some useful data.
   */
  if (session->processed_incoming_data >
        INCOMING_DATA_INITIAL_STREAM_SIZE &&
      (session->total_opened_sstreams == 0 ||
       session->processed_useful_data == 0))
  {
    return true;
  }

  if (session->processed_incoming_data < INCOMING_DATA_GRACE_SIZE) {
    return false;
  }

  /*
   * The overhead of DoH per DNS message can be minimum 160-180
   * bytes. We should allow more for extra information that can be
   * included in headers, so let's use 256 bytes. Minimum DNS
   * message size is 12 bytes. So, (256+12)/12=22. Even that can be
   * too restricting for some edge cases, but should be good enough
   * for any practical purposes. Not to mention that HTTP/2 may
   * include legitimate data that is completely useless for DNS
   * purposes...
   *
   * Anyway, at that point we should have processed enough requests
   * for such clients (if any).
   */
  if (session->processed_useful_data == 0 ||
      (session->processed_incoming_data /
       session->processed_useful_data) > 22)
  {
    return true;
  }

  return false;
}

/*
 * Read callback from TLS socket.
 */
static void
http_readcb(isc_nmhandle_t *handle ISC_ATTR_UNUSED, isc_result_t result,
      isc_region_t *region, void *data) {
  isc_nm_http_session_t *session = (isc_nm_http_session_t *)data;
  isc_nm_http_session_t *tmpsess = NULL;
  ssize_t readlen;
  isc_buffer_t input;

  REQUIRE(VALID_HTTP2_SESSION(session));

  /*
   * Let's ensure that HTTP/2 session and its associated data will
   * not go "out of scope" too early.
   */
  isc__nm_httpsession_attach(session, &tmpsess);

  if (result != ISC_R_SUCCESS) {
    if (result != ISC_R_TIMEDOUT) {
      session->reading = false;
    }
    failed_read_cb(result, session);
    goto done;
  }

  isc_buffer_init(&input, region->base, region->length);
  isc_buffer_add(&input, region->length);

  readlen = http_process_input_data(session, &input);
  if (readlen < 0) {
    failed_read_cb(ISC_R_UNEXPECTED, session);
    goto done;
  } else if (http_is_flooding_peer(session)) {
    http_log_flooding_peer(session);
    failed_read_cb(ISC_R_RANGE, session);
    goto done;
  }

  if ((size_t)readlen < region->length) {
    size_t unread_size = region->length - readlen;
    if (session->buf == NULL) {
      isc_buffer_allocate(session->mctx, &session->buf,
              unread_size);
    }
    isc_buffer_putmem(session->buf, region->base + readlen,
          unread_size);
    if (session->handle != NULL) {
      INSIST(VALID_NMHANDLE(session->handle));
      isc_nm_read_stop(session->handle);
    }
    http_do_bio_async(session);
  } else {
    /* We might have something to receive or send, do IO */
    http_do_bio(session, NULL, NULL, NULL);
  }

done:
  isc__nm_httpsession_detach(&tmpsess);
}

static void
call_pending_callbacks(isc__nm_http_pending_callbacks_t pending_callbacks,
           isc_result_t result) {
  ISC_LIST_FOREACH(pending_callbacks, cbreq, link) {
    ISC_LIST_UNLINK(pending_callbacks, cbreq, link);
    isc__nm_sendcb(cbreq->handle->sock, cbreq, result, true);
  }
}

static void
http_writecb(isc_nmhandle_t *handle, isc_result_t result, void *arg) {
  isc_http_send_req_t *req = (isc_http_send_req_t *)arg;
  isc_nm_http_session_t *session = req->session;
  isc_nmhandle_t *transphandle = req->transphandle;

  REQUIRE(VALID_HTTP2_SESSION(session));
  REQUIRE(VALID_NMHANDLE(handle));

  if (isc__nm_httpsession_active(session)) {
    INSIST(session->handle == handle);
  }

  call_pending_callbacks(req->pending_write_callbacks, result);

  if (req->cb != NULL) {
    req->cb(req->httphandle, result, req->cbarg);
    isc_nmhandle_detach(&req->httphandle);
  }

  session->data_in_flight -=
    isc_buffer_usedlength(req->pending_write_data);
  isc_buffer_free(&req->pending_write_data);
  session->processed += req->submitted;
  isc_mem_put(session->mctx, req, sizeof(*req));

  session->sending--;

  if (result == ISC_R_SUCCESS) {
    http_do_bio(session, NULL, NULL, NULL);
  } else {
    finish_http_session(session);
  }
  isc_nmhandle_detach(&transphandle);

  isc__nm_httpsession_detach(&session);
}

static void
move_pending_send_callbacks(isc_nm_http_session_t *session,
          isc_http_send_req_t *send) {
  STATIC_ASSERT(
    sizeof(session->pending_write_callbacks) ==
      sizeof(send->pending_write_callbacks),
    "size of pending writes requests callbacks lists differs");
  memmove(&send->pending_write_callbacks,
    &session->pending_write_callbacks,
    sizeof(session->pending_write_callbacks));
  ISC_LIST_INIT(session->pending_write_callbacks);
}

static inline void
http_append_pending_send_request(isc_nm_http_session_t *session,
         isc_nmhandle_t *httphandle, isc_nm_cb_t cb,
         void *cbarg) {
  REQUIRE(VALID_HTTP2_SESSION(session));
  REQUIRE(VALID_NMHANDLE(httphandle));
  REQUIRE(cb != NULL);

  isc__nm_uvreq_t *newcb = isc__nm_uvreq_get(httphandle->sock);

  newcb->cb.send = cb;
  newcb->cbarg = cbarg;
  isc_nmhandle_attach(httphandle, &newcb->handle);
  ISC_LIST_APPEND(session->pending_write_callbacks, newcb, link);
}

static void
http_send_outgoing(isc_nm_http_session_t *session, isc_nmhandle_t *httphandle,
       isc_nm_cb_t cb, void *cbarg) {
  isc_http_send_req_t *send = NULL;
  size_t total = 0;
  isc_region_t send_data = { 0 };
  isc_nmhandle_t *transphandle = NULL;
#ifdef ENABLE_HTTP_WRITE_BUFFERING
  size_t max_total_write_size = 0;
#endif /* ENABLE_HTTP_WRITE_BUFFERING */

  if (!isc__nm_httpsession_active(session)) {
    if (cb != NULL) {
      isc__nm_uvreq_t *req =
        isc__nm_uvreq_get(httphandle->sock);

      req->cb.send = cb;
      req->cbarg = cbarg;
      isc_nmhandle_attach(httphandle, &req->handle);
      isc__nm_sendcb(httphandle->sock, req, ISC_R_CANCELED,
               true);
    }
    return;
  } else if (!nghttp2_session_want_write(session->ngsession) &&
       session->pending_write_data == NULL)
  {
    if (cb != NULL) {
      http_append_pending_send_request(session, httphandle,
               cb, cbarg);
    }
    return;
  }

  /*
   * We need to attach to the session->handle earlier because as an
   * indirect result of the nghttp2_session_mem_send() the session
   * might get closed and the handle detached. However, there is
   * still some outgoing data to handle and we need to call it
   * anyway if only to get the write callback passed here to get
   * called properly.
   */
  isc_nmhandle_attach(session->handle, &transphandle);

  while (nghttp2_session_want_write(session->ngsession)) {
    const uint8_t *data = NULL;
    const size_t pending =
      nghttp2_session_mem_send(session->ngsession, &data);
    const size_t new_total = total + pending;

    /*
     * Sometimes nghttp2_session_mem_send() does not return any
     * data to send even though nghttp2_session_want_write()
     * returns success.
     */
    if (pending == 0 || data == NULL) {
      break;
    }

    /* reallocate buffer if required */
    if (session->pending_write_data == NULL) {
      isc_buffer_allocate(session->mctx,
              &session->pending_write_data,
              INITIAL_DNS_MESSAGE_BUFFER_SIZE);
    }
    isc_buffer_putmem(session->pending_write_data, data, pending);
    total = new_total;
  }

#ifdef ENABLE_HTTP_WRITE_BUFFERING
  if (session->pending_write_data != NULL) {
    max_total_write_size =
      isc_buffer_usedlength(session->pending_write_data);
  }

  /*
   * Here we are trying to flush the pending writes buffer earlier
   * to avoid hitting unnecessary limitations on a TLS record size
   * within some tools (e.g. flamethrower).
   */
  if (cb != NULL) {
    /*
     * Case 0: The callback is specified, that means that a DNS
     * message is ready. Let's flush the buffer.
     */
    total = max_total_write_size;
  } else if (max_total_write_size >= FLUSH_HTTP_WRITE_BUFFER_AFTER) {
    /*
     * Case 1: We have equal or more than
     * FLUSH_HTTP_WRITE_BUFFER_AFTER bytes to send. Let's flush it.
     */
    total = max_total_write_size;
  } else if (session->sending > 0 && total > 0) {
    /*
     * Case 2: There is one or more write requests in flight and
     * we have some new data from nghttp2 to send.
     * Then let's return from the function: as soon as the
     * "in-flight" write callback gets called or we have reached
     * FLUSH_HTTP_WRITE_BUFFER_AFTER bytes in the write buffer, we
     * will flush the buffer. */
    INSIST(cb == NULL);
    goto nothing_to_send;
  } else if (session->sending == 0 && total == 0 &&
       session->pending_write_data != NULL)
  {
    /*
     * Case 3: There is no write in flight and we haven't got
     * anything new from nghttp2, but there is some data pending
     * in the write buffer. Let's flush the buffer.
     */
    isc_region_t region = { 0 };
    total = isc_buffer_usedlength(session->pending_write_data);
    INSIST(total > 0);
    isc_buffer_usedregion(session->pending_write_data, &region);
    INSIST(total == region.length);
  } else {
    /*
     * The other cases are uninteresting, fall-through ones.
     * In the following cases (4-6) we will just bail out:
     *
     * Case 4: There is nothing new to send, nor anything in the
     * write buffer.
     * Case 5: There is nothing new to send and there are write
     * request(s) in flight.
     * Case 6: There is nothing new to send nor are there any
     * write requests in flight.
     *
     * Case 7: There is some new data to send and there are no
     * write requests in flight: Let's send the data.
     */
    INSIST((total == 0 && session->pending_write_data == NULL) ||
           (total == 0 && session->sending > 0) ||
           (total == 0 && session->sending == 0) ||
           (total > 0 && session->sending == 0));
  }
#endif /* ENABLE_HTTP_WRITE_BUFFERING */

  if (total == 0) {
    /* No data returned */
    if (cb != NULL) {
      http_append_pending_send_request(session, httphandle,
               cb, cbarg);
    }
    goto nothing_to_send;
  }

  /*
   * If we have reached this point it means that we need to send some
   * data and flush the outgoing buffer. The code below does that.
   */
  send = isc_mem_get(session->mctx, sizeof(*send));

  *send = (isc_http_send_req_t){ .pending_write_data =
                 session->pending_write_data,
               .cb = cb,
               .cbarg = cbarg,
               .submitted = session->submitted };
  session->submitted = 0;
  session->pending_write_data = NULL;
  move_pending_send_callbacks(session, send);

  send->transphandle = transphandle;
  isc__nm_httpsession_attach(session, &send->session);

  if (cb != NULL) {
    INSIST(VALID_NMHANDLE(httphandle));
    isc_nmhandle_attach(httphandle, &send->httphandle);
  }

  session->sending++;
  isc_buffer_usedregion(send->pending_write_data, &send_data);
  session->data_in_flight += send_data.length;
  isc_nm_send(transphandle, &send_data, http_writecb, send);
  return;

nothing_to_send:
  isc_nmhandle_detach(&transphandle);
}

static inline bool
http_too_many_active_streams(isc_nm_http_session_t *session) {
  const uint64_t active_streams = session->received - session->processed;
  /*
   * The motivation behind capping the maximum active streams number
   * to a third of maximum streams is to allow the value to scale
   * with the max number of streams.
   *
   * We do not want to have too many active streams at once as every
   * stream is processed as a separate virtual connection by the
   * higher level code. If a client sends a bulk of requests without
   * waiting for the previous ones to complete we might want to
   * throttle it as it might be not a friend knocking at the
   * door. We already have some job to do for it.
   */
  const uint64_t max_active_streams =
    ISC_MAX(ISC_NETMGR_MAX_STREAM_CLIENTS_PER_CONN,
      (session->max_concurrent_streams * 6) / 10); /* 60% */

  if (session->client) {
    return false;
  }

  /*
   * Do not process incoming data if there are too many active DNS
   * clients (streams) per connection.
   */
  if (active_streams >= max_active_streams) {
    return true;
  }

  return false;
}

static void
http_do_bio(isc_nm_http_session_t *session, isc_nmhandle_t *send_httphandle,
      isc_nm_cb_t send_cb, void *send_cbarg) {
  isc__nm_uvreq_t *req = NULL;
  size_t remaining = 0;
  REQUIRE(VALID_HTTP2_SESSION(session));

  if (session->closed) {
    goto cancel;
  } else if (session->closing) {
    /*
     * There might be leftover callbacks waiting to be received
     */
    if (session->sending == 0) {
      finish_http_session(session);
    }
    goto cancel;
  } else if (nghttp2_session_want_read(session->ngsession) == 0 &&
       nghttp2_session_want_write(session->ngsession) == 0 &&
       session->pending_write_data == NULL)
  {
    session->closing = true;
    if (session->handle != NULL) {
      isc_nm_read_stop(session->handle);
    }
    if (session->sending == 0) {
      finish_http_session(session);
    }
    goto cancel;
  }

  else if (session->buf != NULL)
  {
    remaining = isc_buffer_remaininglength(session->buf);
  }

  if (nghttp2_session_want_read(session->ngsession) != 0) {
    if (!session->reading) {
      /* We have not yet started reading from this handle */
      isc__nmsocket_timer_start(session->handle->sock);
      isc_nm_read(session->handle, http_readcb, session);
      session->reading = true;
    } else if (session->buf != NULL && remaining > 0) {
      /* Leftover data in the buffer, use it */
      size_t remaining_after = 0;
      ssize_t readlen = 0;
      isc_nm_http_session_t *tmpsess = NULL;

      /*
       * Let's ensure that HTTP/2 session and its associated
       * data will not go "out of scope" too early.
       */
      isc__nm_httpsession_attach(session, &tmpsess);

      readlen = http_process_input_data(session,
                session->buf);

      remaining_after =
        isc_buffer_remaininglength(session->buf);

      if (readlen < 0) {
        failed_read_cb(ISC_R_UNEXPECTED, session);
      } else if (http_is_flooding_peer(session)) {
        http_log_flooding_peer(session);
        failed_read_cb(ISC_R_RANGE, session);
      } else if ((size_t)readlen == remaining) {
        isc_buffer_clear(session->buf);
        isc_buffer_compact(session->buf);
        http_do_bio(session, send_httphandle, send_cb,
              send_cbarg);
        isc__nm_httpsession_detach(&tmpsess);
        return;
      } else if (remaining_after > 0 &&
           remaining_after < remaining)
      {
        /*
         * We have processed a part of the data, now
         * let's delay processing of whatever is left
         * here. We want it to be an async operation so
         * that we will:
         *
         * a) let other things run;
         * b) have finer grained control over how much
         * data is processed at once, because nghttp2
         * would happily consume as much data we pass to
         * it and that could overwhelm the server.
         */
        http_do_bio_async(session);
      }
      isc__nm_httpsession_detach(&tmpsess);
    } else if (session->handle != NULL) {
      INSIST(VALID_NMHANDLE(session->handle));
      /*
       * Resume reading, it's idempotent, wait for more
       */
      isc__nmsocket_timer_start(session->handle->sock);
      isc_nm_read(session->handle, http_readcb, session);
    }
  } else if (session->handle != NULL) {
    INSIST(VALID_NMHANDLE(session->handle));
    /* We don't want more data, stop reading for now */
    isc_nm_read_stop(session->handle);
  }

  /* we might have some data to send after processing */
  http_send_outgoing(session, send_httphandle, send_cb, send_cbarg);

  return;
cancel:
  if (send_cb == NULL) {
    return;
  }
  req = isc__nm_uvreq_get(send_httphandle->sock);

  req->cb.send = send_cb;
  req->cbarg = send_cbarg;
  isc_nmhandle_attach(send_httphandle, &req->handle);
  isc__nm_sendcb(send_httphandle->sock, req, ISC_R_CANCELED, true);
}

static void
http_do_bio_async_cb(void *arg) {
  isc_nm_http_session_t *session = arg;

  REQUIRE(VALID_HTTP2_SESSION(session));

  session->async_queued = false;

  if (session->handle != NULL &&
      !isc__nmsocket_closing(session->handle->sock))
  {
    http_do_bio(session, NULL, NULL, NULL);
  }

  isc__nm_httpsession_detach(&session);
}

static void
http_do_bio_async(isc_nm_http_session_t *session) {
  isc_nm_http_session_t *tmpsess = NULL;

  REQUIRE(VALID_HTTP2_SESSION(session));

  if (session->handle == NULL ||
      isc__nmsocket_closing(session->handle->sock) ||
      session->async_queued)
  {
    return;
  }
  session->async_queued = true;
  isc__nm_httpsession_attach(session, &tmpsess);
  isc_async_run(session->handle->sock->worker->loop, http_do_bio_async_cb,
          tmpsess);
}

static isc_result_t
get_http_cstream(isc_nmsocket_t *sock, http_cstream_t **streamp) {
  http_cstream_t *cstream = sock->h2->connect.cstream;
  isc_result_t result;

  REQUIRE(streamp != NULL && *streamp == NULL);

  sock->h2->connect.cstream = NULL;

  if (cstream == NULL) {
    result = new_http_cstream(sock, &cstream);
    if (result != ISC_R_SUCCESS) {
      INSIST(cstream == NULL);
      return result;
    }
  }

  *streamp = cstream;
  return ISC_R_SUCCESS;
}

static void
http_call_connect_cb(isc_nmsocket_t *sock, isc_nm_http_session_t *session,
         isc_result_t result) {
  isc_nmhandle_t *httphandle = isc__nmhandle_get(sock, &sock->peer,
                   &sock->iface);
  void *cbarg;
  isc_nm_cb_t connect_cb;

  REQUIRE(sock->connect_cb != NULL);

  cbarg = sock->connect_cbarg;
  connect_cb = sock->connect_cb;
  isc__nmsocket_clearcb(sock);
  if (result == ISC_R_SUCCESS) {
    if (session != NULL) {
      session->client_httphandle = httphandle;
    }
    connect_cb(httphandle, result, cbarg);
  } else {
    connect_cb(httphandle, result, cbarg);
    isc_nmhandle_detach(&httphandle);
  }
}

static void
transport_connect_cb(isc_nmhandle_t *handle, isc_result_t result, void *cbarg) {
  isc_nmsocket_t *http_sock = (isc_nmsocket_t *)cbarg;
  isc_nmsocket_t *transp_sock = NULL;
  isc_nm_http_session_t *session = NULL;
  http_cstream_t *cstream = NULL;
  isc_mem_t *mctx = NULL;

  REQUIRE(VALID_NMSOCK(http_sock));
  REQUIRE(VALID_NMHANDLE(handle));

  transp_sock = handle->sock;

  REQUIRE(VALID_NMSOCK(transp_sock));

  mctx = transp_sock->worker->mctx;

  INSIST(http_sock->h2->connect.uri != NULL);

  http_sock->h2->connect.tls_peer_verify_string =
    isc_nm_verify_tls_peer_result_string(handle);
  if (result != ISC_R_SUCCESS) {
    goto error;
  }

  http_initsocket(transp_sock);
  new_session(mctx, http_sock->h2->connect.tlsctx, &session);
  session->client = true;
  transp_sock->h2->session = session;
  http_sock->h2->connect.tlsctx = NULL;
  /* otherwise we will get some garbage output in DIG */
  http_sock->iface = isc_nmhandle_localaddr(handle);
  http_sock->peer = isc_nmhandle_peeraddr(handle);

  transp_sock->h2->connect.post = http_sock->h2->connect.post;
  transp_sock->h2->connect.uri = http_sock->h2->connect.uri;
  http_sock->h2->connect.uri = NULL;
  isc__nm_httpsession_attach(session, &http_sock->h2->session);

  if (session->tlsctx != NULL) {
    const unsigned char *alpn = NULL;
    unsigned int alpnlen = 0;

    INSIST(transp_sock->type == isc_nm_tlssocket ||
           transp_sock->type == isc_nm_proxystreamsocket);

    isc__nmhandle_get_selected_alpn(handle, &alpn, &alpnlen);
    if (alpn == NULL || alpnlen != NGHTTP2_PROTO_VERSION_ID_LEN ||
        memcmp(NGHTTP2_PROTO_VERSION_ID, alpn,
         NGHTTP2_PROTO_VERSION_ID_LEN) != 0)
    {
      /*
       * HTTP/2 negotiation error.
       * Any sensible DoH client
       * will fail if HTTP/2 cannot
       * be negotiated via ALPN.
       */
      result = ISC_R_HTTP2ALPNERROR;
      goto error;
    }
  }

  isc_nmhandle_attach(handle, &session->handle);

  initialize_nghttp2_client_session(session);
  if (!send_client_connection_header(session)) {
    goto error;
  }

  result = get_http_cstream(http_sock, &cstream);
  http_sock->h2->connect.cstream = cstream;
  if (result != ISC_R_SUCCESS) {
    goto error;
  }

  http_transpost_tcp_nodelay(handle);
  isc__nmhandle_set_manual_timer(session->handle, true);

  http_call_connect_cb(http_sock, session, result);

  http_do_bio(session, NULL, NULL, NULL);
  isc__nmsocket_detach(&http_sock);
  return;

error:
  http_call_connect_cb(http_sock, session, result);

  if (http_sock->h2->connect.uri != NULL) {
    isc_mem_free(http_sock->worker->mctx,
           http_sock->h2->connect.uri);
  }

  isc__nmsocket_prep_destroy(http_sock);
  isc__nmsocket_detach(&http_sock);
}

void
isc_nm_httpconnect(isc_sockaddr_t *local, isc_sockaddr_t *peer, const char *uri,
       bool post, isc_nm_cb_t cb, void *cbarg, isc_tlsctx_t *tlsctx,
       const char *sni_hostname,
       isc_tlsctx_client_session_cache_t *client_sess_cache,
       unsigned int timeout, isc_nm_proxy_type_t proxy_type,
       isc_nm_proxyheader_info_t *proxy_info) {
  isc_sockaddr_t local_interface;
  isc_nmsocket_t *sock = NULL;
  isc__networker_t *worker = isc__networker_current();

  REQUIRE(cb != NULL);
  REQUIRE(peer != NULL);
  REQUIRE(uri != NULL);
  REQUIRE(*uri != '\0');

  if (isc__nm_closing(worker)) {
    cb(NULL, ISC_R_SHUTTINGDOWN, cbarg);
    return;
  }

  if (local == NULL) {
    isc_sockaddr_anyofpf(&local_interface, peer->type.sa.sa_family);
    local = &local_interface;
  }

  sock = isc_mempool_get(worker->nmsocket_pool);
  isc__nmsocket_init(sock, worker, isc_nm_httpsocket, local, NULL);
  http_initsocket(sock);

  sock->connect_timeout = timeout;
  sock->connect_cb = cb;
  sock->connect_cbarg = cbarg;
  sock->client = true;

  if (isc__nm_closing(worker)) {
    isc__nm_uvreq_t *req = isc__nm_uvreq_get(sock);

    req->cb.connect = cb;
    req->cbarg = cbarg;
    req->peer = *peer;
    req->local = *local;
    req->handle = isc__nmhandle_get(sock, &req->peer, &sock->iface);

    isc__nmsocket_clearcb(sock);
    isc__nm_connectcb(sock, req, ISC_R_SHUTTINGDOWN, true);
    isc__nmsocket_prep_destroy(sock);
    isc__nmsocket_detach(&sock);
    return;
  }

  *sock->h2 = (isc_nmsocket_h2_t){ .connect.uri = isc_mem_strdup(
             sock->worker->mctx, uri),
           .connect.post = post,
           .connect.tlsctx = tlsctx };
  ISC_LINK_INIT(sock->h2, link);

  /*
   * We need to prevent the interface object data from going out of
   * scope too early.
   */
  if (local == &local_interface) {
    sock->h2->connect.local_interface = local_interface;
    sock->iface = sock->h2->connect.local_interface;
  }

  switch (proxy_type) {
  case ISC_NM_PROXY_NONE:
    if (tlsctx != NULL) {
      isc_nm_tlsconnect(local, peer, transport_connect_cb,
            sock, tlsctx, sni_hostname,
            client_sess_cache, timeout, false,
            NULL);
    } else {
      isc_nm_tcpconnect(local, peer, transport_connect_cb,
            sock, timeout);
    }
    break;
  case ISC_NM_PROXY_PLAIN:
    if (tlsctx != NULL) {
      isc_nm_tlsconnect(local, peer, transport_connect_cb,
            sock, tlsctx, sni_hostname,
            client_sess_cache, timeout, true,
            proxy_info);
    } else {
      isc_nm_proxystreamconnect(
        local, peer, transport_connect_cb, sock,
        timeout, NULL, NULL, NULL, proxy_info);
    }
    break;
  case ISC_NM_PROXY_ENCRYPTED:
    INSIST(tlsctx != NULL);
    isc_nm_proxystreamconnect(local, peer, transport_connect_cb,
            sock, timeout, tlsctx, sni_hostname,
            client_sess_cache, proxy_info);
    break;
  default:
    UNREACHABLE();
  }
}

static isc_result_t
client_send(isc_nmhandle_t *handle, const isc_region_t *region) {
  isc_result_t result = ISC_R_SUCCESS;
  isc_nmsocket_t *sock = handle->sock;
  isc_mem_t *mctx = sock->worker->mctx;
  isc_nm_http_session_t *session = sock->h2->session;
  http_cstream_t *cstream = sock->h2->connect.cstream;

  REQUIRE(VALID_HTTP2_SESSION(handle->sock->h2->session));
  REQUIRE(session->client);
  REQUIRE(region != NULL);
  REQUIRE(region->base != NULL);
  REQUIRE(region->length <= MAX_DNS_MESSAGE_SIZE);

  if (session->closed) {
    return ISC_R_CANCELED;
  }

  INSIST(cstream != NULL);

  if (cstream->post) {
    /* POST */
    isc_buffer_allocate(mctx, &cstream->postdata, region->length);
    isc_buffer_putmem(cstream->postdata, region->base,
          region->length);
  } else {
    /* GET */
    size_t path_size = 0;
    char *base64url_data = NULL;
    size_t base64url_data_len = 0;
    isc_buffer_t *buf = NULL;
    isc_region_t data = *region;
    isc_region_t base64_region;
    size_t base64_len = ((4 * data.length / 3) + 3) & ~3;

    isc_buffer_allocate(mctx, &buf, base64_len);

    result = isc_base64_totext(&data, -1, "", buf);
    if (result != ISC_R_SUCCESS) {
      isc_buffer_free(&buf);
      goto error;
    }

    isc_buffer_usedregion(buf, &base64_region);
    INSIST(base64_region.length == base64_len);

    base64url_data = isc__nm_base64_to_base64url(
      mctx, (const char *)base64_region.base,
      base64_region.length, &base64url_data_len);
    isc_buffer_free(&buf);
    if (base64url_data == NULL) {
      goto error;
    }

    /* len("?dns=") + len(path) + len(base64url) + len("\0") */
    path_size = cstream->pathlen + base64url_data_len + 5 + 1;
    cstream->GET_path = isc_mem_allocate(mctx, path_size);
    cstream->GET_path_len = (size_t)snprintf(
      cstream->GET_path, path_size, "%.*s?dns=%s",
      (int)cstream->pathlen, cstream->path, base64url_data);

    INSIST(cstream->GET_path_len == (path_size - 1));
    isc_mem_free(mctx, base64url_data);
  }

  cstream->sending = true;

  sock->h2->connect.cstream = NULL;
  result = client_submit_request(session, cstream);
  if (result != ISC_R_SUCCESS) {
    put_http_cstream(session->mctx, cstream);
    goto error;
  }

error:
  return result;
}

isc_result_t
isc__nm_http_request(isc_nmhandle_t *handle, isc_region_t *region,
         isc_nm_recv_cb_t cb, void *cbarg) {
  isc_result_t result = ISC_R_SUCCESS;
  isc_nmsocket_t *sock = NULL;
  http_cstream_t *cstream = NULL;

  REQUIRE(VALID_NMHANDLE(handle));
  REQUIRE(VALID_NMSOCK(handle->sock));
  REQUIRE(handle->sock->tid == isc_tid());
  REQUIRE(handle->sock->client);

  REQUIRE(cb != NULL);

  sock = handle->sock;

  isc__nm_http_read(handle, cb, cbarg);
  if (!isc__nm_httpsession_active(handle->sock->h2->session)) {
    /* the callback was called by isc__nm_http_read() */
    return ISC_R_CANCELED;
  }
  result = client_send(handle, region);
  if (result != ISC_R_SUCCESS) {
    goto error;
  }

  return ISC_R_SUCCESS;

error:
  /*
   * client_send() detaches and frees the stream on a submit failure
   * (it nullifies sock->h2->connect.cstream before submitting, then
   * frees it on the failure branch), so the reloaded pointer can be
   * NULL here.  The caller still gets the error result and reports the
   * failure itself.
   */
  cstream = sock->h2->connect.cstream;
  if (cstream != NULL && cstream->read_cb != NULL) {
    cstream->read_cb(handle, result, NULL, cstream->read_cbarg);
  }
  return result;
}

static int
server_on_begin_headers_callback(nghttp2_session *ngsession,
         const nghttp2_frame *frame, void *user_data) {
  isc_nm_http_session_t *session = (isc_nm_http_session_t *)user_data;
  isc_nmsocket_t *socket = NULL;
  isc__networker_t *worker = NULL;
  isc_sockaddr_t local;

  if (frame->hd.type != NGHTTP2_HEADERS ||
      frame->headers.cat != NGHTTP2_HCAT_REQUEST)
  {
    return 0;
  } else if (frame->hd.length > MAX_ALLOWED_DATA_IN_HEADERS) {
    return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE;
  }

  if (session->nsstreams >= session->max_concurrent_streams) {
    return NGHTTP2_ERR_CALLBACK_FAILURE;
  }

  INSIST(session->handle->sock->tid == isc_tid());

  worker = session->handle->sock->worker;
  socket = isc_mempool_get(worker->nmsocket_pool);
  local = isc_nmhandle_localaddr(session->handle);
  isc__nmsocket_init(socket, worker, isc_nm_httpsocket, &local, NULL);
  http_initsocket(socket);
  socket->peer = isc_nmhandle_peeraddr(session->handle);
  *socket->h2 = (isc_nmsocket_h2_t){
    .psock = socket,
    .stream_id = frame->hd.stream_id,
    .headers_error_code = ISC_HTTP_ERROR_SUCCESS,
    .request_type = ISC_HTTP_REQ_UNSUPPORTED,
    .request_scheme = ISC_HTTP_SCHEME_UNSUPPORTED,
    .link = ISC_LINK_INITIALIZER,
  };
  isc_buffer_initnull(&socket->h2->rbuf);
  isc_buffer_initnull(&socket->h2->wbuf);
  isc_nm_http_endpoints_attach(
    http_get_listener_endpoints(session->serversocket, socket->tid),
    &socket->h2->peer_endpoints);
  session->nsstreams++;
  isc__nm_httpsession_attach(session, &socket->h2->session);
  ISC_LIST_APPEND(session->sstreams, socket->h2, link);
  session->total_opened_sstreams++;

  nghttp2_session_set_stream_user_data(ngsession, frame->hd.stream_id,
               socket);
  return 0;
}

static isc_http_error_responses_t
server_handle_path_header(isc_nmsocket_t *socket, const uint8_t *value,
        const size_t valuelen) {
  isc_nm_httphandler_t *handler = NULL;
  const uint8_t *qstr = NULL;
  size_t vlen = valuelen;

  qstr = memchr(value, '?', valuelen);
  if (qstr != NULL) {
    vlen = qstr - value;
  }

  if (socket->h2->request_path != NULL) {
    isc_mem_free(socket->worker->mctx, socket->h2->request_path);
  }
  socket->h2->request_path = isc_mem_allocate(socket->worker->mctx,
                vlen + 1);
  strlcpy(socket->h2->request_path, (const char *)value, vlen + 1);

  if (!isc_nm_http_path_isvalid(socket->h2->request_path)) {
    isc_mem_free(socket->worker->mctx, socket->h2->request_path);
    return ISC_HTTP_ERROR_BAD_REQUEST;
  }

  handler = http_endpoints_find(socket->h2->request_path,
              socket->h2->peer_endpoints);
  if (handler != NULL) {
    socket->h2->cb = handler->cb;
    socket->h2->cbarg = handler->cbarg;
  } else {
    isc_mem_free(socket->worker->mctx, socket->h2->request_path);
    return ISC_HTTP_ERROR_NOT_FOUND;
  }

  if (qstr != NULL) {
    const char *dns_value = NULL;
    size_t dns_value_len = 0;

    if (isc__nm_parse_httpquery((const char *)qstr, &dns_value,
              &dns_value_len))
    {
      const size_t decoded_size = dns_value_len / 4 * 3;
      if (decoded_size <= MAX_DNS_MESSAGE_SIZE) {
        if (socket->h2->query_data != NULL) {
          isc_mem_free(socket->worker->mctx,
                 socket->h2->query_data);
        }
        socket->h2->query_data =
          isc__nm_base64url_to_base64(
            socket->worker->mctx, dns_value,
            dns_value_len,
            &socket->h2->query_data_len);
        socket->h2->session->processed_useful_data +=
          dns_value_len;
      } else {
        socket->h2->query_too_large = true;
        return ISC_HTTP_ERROR_PAYLOAD_TOO_LARGE;
      }
    } else {
      return ISC_HTTP_ERROR_BAD_REQUEST;
    }
  }
  return ISC_HTTP_ERROR_SUCCESS;
}

static isc_http_error_responses_t
server_handle_method_header(isc_nmsocket_t *socket, const uint8_t *value,
          const size_t valuelen) {
  const char get[] = "GET";
  const char post[] = "POST";

  if (HEADER_MATCH(get, value, valuelen)) {
    socket->h2->request_type = ISC_HTTP_REQ_GET;
  } else if (HEADER_MATCH(post, value, valuelen)) {
    socket->h2->request_type = ISC_HTTP_REQ_POST;
  } else {
    return ISC_HTTP_ERROR_NOT_IMPLEMENTED;
  }
  return ISC_HTTP_ERROR_SUCCESS;
}

static isc_http_error_responses_t
server_handle_scheme_header(isc_nmsocket_t *socket, const uint8_t *value,
          const size_t valuelen) {
  const char http[] = "http";
  const char http_secure[] = "https";

  if (HEADER_MATCH(http_secure, value, valuelen)) {
    socket->h2->request_scheme = ISC_HTTP_SCHEME_HTTP_SECURE;
  } else if (HEADER_MATCH(http, value, valuelen)) {
    socket->h2->request_scheme = ISC_HTTP_SCHEME_HTTP;
  } else {
    return ISC_HTTP_ERROR_BAD_REQUEST;
  }
  return ISC_HTTP_ERROR_SUCCESS;
}

static isc_http_error_responses_t
server_handle_content_length_header(isc_nmsocket_t *socket,
            const uint8_t *value,
            const size_t valuelen) {
  char tmp[32] = { 0 };
  const size_t tmplen = sizeof(tmp) - 1;

  strncpy(tmp, (const char *)value,
    valuelen > tmplen ? tmplen : valuelen);
  socket->h2->content_length = strtoul(tmp, NULL, 10);
  if (socket->h2->content_length > MAX_DNS_MESSAGE_SIZE) {
    return ISC_HTTP_ERROR_PAYLOAD_TOO_LARGE;
  } else if (socket->h2->content_length == 0) {
    return ISC_HTTP_ERROR_BAD_REQUEST;
  }
  return ISC_HTTP_ERROR_SUCCESS;
}

static isc_http_error_responses_t
server_handle_content_type_header(isc_nmsocket_t *socket, const uint8_t *value,
          const size_t valuelen) {
  const char type_dns_message[] = DNS_MEDIA_TYPE;
  isc_http_error_responses_t resp = ISC_HTTP_ERROR_SUCCESS;

  UNUSED(socket);

  if (!HEADER_MATCH(type_dns_message, value, valuelen)) {
    resp = ISC_HTTP_ERROR_UNSUPPORTED_MEDIA_TYPE;
  }
  return resp;
}

static isc_http_error_responses_t
server_handle_header(isc_nmsocket_t *socket, const uint8_t *name,
         size_t namelen, const uint8_t *value,
         const size_t valuelen) {
  isc_http_error_responses_t code = ISC_HTTP_ERROR_SUCCESS;
  bool was_error;
  const char path[] = ":path";
  const char method[] = ":method";
  const char scheme[] = ":scheme";
  const char content_length[] = "Content-Length";
  const char content_type[] = "Content-Type";

  was_error = socket->h2->headers_error_code != ISC_HTTP_ERROR_SUCCESS;
  /*
   * process Content-Length even when there was an error,
   * to drop the connection earlier if required.
   */
  if (HEADER_MATCH(content_length, name, namelen)) {
    code = server_handle_content_length_header(socket, value,
                 valuelen);
  } else if (!was_error && HEADER_MATCH(path, name, namelen)) {
    code = server_handle_path_header(socket, value, valuelen);
  } else if (!was_error && HEADER_MATCH(method, name, namelen)) {
    code = server_handle_method_header(socket, value, valuelen);
  } else if (!was_error && HEADER_MATCH(scheme, name, namelen)) {
    code = server_handle_scheme_header(socket, value, valuelen);
  } else if (!was_error && HEADER_MATCH(content_type, name, namelen)) {
    code = server_handle_content_type_header(socket, value,
               valuelen);
  }

  return code;
}

static int
server_on_header_callback(nghttp2_session *session, const nghttp2_frame *frame,
        const uint8_t *name, size_t namelen,
        const uint8_t *value, size_t valuelen, uint8_t flags,
        void *user_data) {
  isc_nmsocket_t *socket = NULL;
  isc_http_error_responses_t code = ISC_HTTP_ERROR_SUCCESS;

  UNUSED(flags);
  UNUSED(user_data);

  socket = nghttp2_session_get_stream_user_data(session,
                  frame->hd.stream_id);
  if (socket == NULL) {
    return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE;
  }

  socket->h2->headers_data_processed += (namelen + valuelen);

  switch (frame->hd.type) {
  case NGHTTP2_HEADERS:
    if (frame->headers.cat != NGHTTP2_HCAT_REQUEST) {
      break;
    }
    code = server_handle_header(socket, name, namelen, value,
              valuelen);
    break;
  }

  INSIST(socket != NULL);

  if (socket->h2->headers_data_processed > MAX_ALLOWED_DATA_IN_HEADERS) {
    return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE;
  } else if (socket->h2->content_length > MAX_ALLOWED_DATA_IN_POST) {
    return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE;
  }

  if (code == ISC_HTTP_ERROR_SUCCESS) {
    return 0;
  } else {
    socket->h2->headers_error_code = code;
  }

  return 0;
}

static ssize_t
server_read_callback(nghttp2_session *ngsession, int32_t stream_id,
         uint8_t *buf, size_t length, uint32_t *data_flags,
         nghttp2_data_source *source, void *user_data) {
  isc_nm_http_session_t *session = (isc_nm_http_session_t *)user_data;
  isc_nmsocket_t *socket = (isc_nmsocket_t *)source->ptr;
  size_t buflen;

  REQUIRE(socket->h2->stream_id == stream_id);

  UNUSED(ngsession);
  UNUSED(session);

  buflen = isc_buffer_remaininglength(&socket->h2->wbuf);
  if (buflen > length) {
    buflen = length;
  }

  if (buflen > 0) {
    (void)memmove(buf, isc_buffer_current(&socket->h2->wbuf),
            buflen);
    isc_buffer_forward(&socket->h2->wbuf, buflen);
  }

  if (isc_buffer_remaininglength(&socket->h2->wbuf) == 0) {
    *data_flags |= NGHTTP2_DATA_FLAG_EOF;
  }

  return buflen;
}

static isc_result_t
server_send_response(nghttp2_session *ngsession, int32_t stream_id,
         const nghttp2_nv *nva, size_t nvlen,
         isc_nmsocket_t *socket) {
  nghttp2_data_provider data_prd;
  int rv;

  if (socket->h2->response_submitted) {
    /* NGHTTP2 will gladly accept new response (write request)
     * from us even though we cannot send more than one over the
     * same HTTP/2 stream. Thus, we need to handle this case
     * manually. We will return failure code so that it will be
     * passed to the write callback. */
    return ISC_R_FAILURE;
  }

  data_prd.source.ptr = socket;
  data_prd.read_callback = server_read_callback;

  rv = nghttp2_submit_response(ngsession, stream_id, nva, nvlen,
             &data_prd);
  if (rv != 0) {
    return ISC_R_FAILURE;
  }

  socket->h2->response_submitted = true;
  return ISC_R_SUCCESS;
}

#define MAKE_ERROR_REPLY(tag, code, desc) \
  { tag, MAKE_NV2(":status", #code), desc }

/*
 * Here we use roughly the same error codes that Unbound uses.
 * (https://blog.nlnetlabs.nl/dns-over-https-in-unbound/)
 */

static struct http_error_responses {
  const isc_http_error_responses_t type;
  const nghttp2_nv header;
  const char *desc;
} error_responses[] = {
  MAKE_ERROR_REPLY(ISC_HTTP_ERROR_BAD_REQUEST, 400, "Bad Request"),
  MAKE_ERROR_REPLY(ISC_HTTP_ERROR_NOT_FOUND, 404, "Not Found"),
  MAKE_ERROR_REPLY(ISC_HTTP_ERROR_PAYLOAD_TOO_LARGE, 413,
       "Payload Too Large"),
  MAKE_ERROR_REPLY(ISC_HTTP_ERROR_URI_TOO_LONG, 414, "URI Too Long"),
  MAKE_ERROR_REPLY(ISC_HTTP_ERROR_UNSUPPORTED_MEDIA_TYPE, 415,
       "Unsupported Media Type"),
  MAKE_ERROR_REPLY(ISC_HTTP_ERROR_GENERIC, 500, "Internal Server Error"),
  MAKE_ERROR_REPLY(ISC_HTTP_ERROR_NOT_IMPLEMENTED, 501, "Not Implemented")
};

static void
log_server_error_response(const isc_nmsocket_t *socket,
        const struct http_error_responses *response) {
  const int log_level = ISC_LOG_DEBUG(1);
  char client_sabuf[ISC_SOCKADDR_FORMATSIZE];
  char local_sabuf[ISC_SOCKADDR_FORMATSIZE];

  if (!isc_log_wouldlog(log_level)) {
    return;
  }

  isc_sockaddr_format(&socket->peer, client_sabuf, sizeof(client_sabuf));
  isc_sockaddr_format(&socket->iface, local_sabuf, sizeof(local_sabuf));
  isc__nmsocket_log(socket, log_level,
        "HTTP/2 request from %s (on %s) failed: %s %s",
        client_sabuf, local_sabuf, response->header.value,
        response->desc);
}

static isc_result_t
server_send_error_response(const isc_http_error_responses_t error,
         nghttp2_session *ngsession, isc_nmsocket_t *socket) {
  void *base;

  REQUIRE(error != ISC_HTTP_ERROR_SUCCESS);

  base = isc_buffer_base(&socket->h2->rbuf);
  if (base != NULL) {
    isc_mem_free(socket->h2->session->mctx, base);
    isc_buffer_initnull(&socket->h2->rbuf);
  }

  /* We do not want the error response to be cached anywhere. */
  socket->h2->min_ttl = 0;

  for (size_t i = 0;
       i < sizeof(error_responses) / sizeof(error_responses[0]); i++)
  {
    if (error_responses[i].type == error) {
      log_server_error_response(socket, &error_responses[i]);
      return server_send_response(
        ngsession, socket->h2->stream_id,
        &error_responses[i].header, 1, socket);
    }
  }

  return server_send_error_response(ISC_HTTP_ERROR_GENERIC, ngsession,
            socket);
}

static void
server_call_cb(isc_nmsocket_t *socket, const isc_result_t result,
         isc_region_t *data) {
  isc_nmhandle_t *handle = NULL;

  REQUIRE(VALID_NMSOCK(socket));

  /*
   * In some cases the callback could not have been set (e.g. when
   * the stream was closed prematurely (before processing its HTTP
   * path).
   */
  if (socket->h2->cb == NULL) {
    return;
  }

  handle = isc__nmhandle_get(socket, NULL, NULL);
  if (result != ISC_R_SUCCESS) {
    data = NULL;
  } else if (socket->h2->session->handle != NULL) {
    isc__nmsocket_timer_restart(socket->h2->session->handle->sock);
  }
  if (result == ISC_R_SUCCESS) {
    socket->h2->request_received = true;
    socket->h2->session->received++;
  }
  socket->h2->cb(handle, result, data, socket->h2->cbarg);
  isc_nmhandle_detach(&handle);
}

void
isc__nm_http_bad_request(isc_nmhandle_t *handle) {
  isc_nmsocket_t *sock = NULL;

  REQUIRE(VALID_NMHANDLE(handle));
  REQUIRE(VALID_NMSOCK(handle->sock));
  sock = handle->sock;
  REQUIRE(sock->type == isc_nm_httpsocket);
  REQUIRE(!sock->client);
  REQUIRE(VALID_HTTP2_SESSION(sock->h2->session));

  if (sock->h2->response_submitted ||
      !isc__nm_httpsession_active(sock->h2->session))
  {
    return;
  }

  (void)server_send_error_response(ISC_HTTP_ERROR_BAD_REQUEST,
           sock->h2->session->ngsession, sock);
}

static int
server_on_request_recv(nghttp2_session *ngsession, isc_nmsocket_t *socket) {
  isc_result_t result;
  isc_http_error_responses_t code = ISC_HTTP_ERROR_SUCCESS;
  isc_region_t data;
  uint8_t tmp_buf[MAX_DNS_MESSAGE_SIZE];

  code = socket->h2->headers_error_code;
  if (code != ISC_HTTP_ERROR_SUCCESS) {
    goto error;
  }

  if (socket->h2->request_path == NULL || socket->h2->cb == NULL) {
    code = ISC_HTTP_ERROR_NOT_FOUND;
  } else if (socket->h2->request_type == ISC_HTTP_REQ_POST &&
       socket->h2->content_length == 0)
  {
    code = ISC_HTTP_ERROR_BAD_REQUEST;
  } else if (socket->h2->request_type == ISC_HTTP_REQ_POST &&
       isc_buffer_usedlength(&socket->h2->rbuf) >
         socket->h2->content_length)
  {
    code = ISC_HTTP_ERROR_PAYLOAD_TOO_LARGE;
  } else if (socket->h2->request_type == ISC_HTTP_REQ_POST &&
       isc_buffer_usedlength(&socket->h2->rbuf) !=
         socket->h2->content_length)
  {
    code = ISC_HTTP_ERROR_BAD_REQUEST;
  } else if (socket->h2->request_type == ISC_HTTP_REQ_POST &&
       socket->h2->query_data != NULL)
  {
    /* The spec does not mention which value the query string for
     * POST should have. For GET we use its value to decode a DNS
     * message from it, for POST the message is transferred in the
     * body of the request. Taking it into account, it is much safer
     * to treat POST
     * requests with query strings as malformed ones. */
    code = ISC_HTTP_ERROR_BAD_REQUEST;
  } else if (socket->h2->request_type == ISC_HTTP_REQ_GET &&
       socket->h2->content_length > 0)
  {
    code = ISC_HTTP_ERROR_BAD_REQUEST;
  } else if (socket->h2->request_type == ISC_HTTP_REQ_GET &&
       socket->h2->query_data == NULL)
  {
    /* A GET request without any query data - there is nothing to
     * decode. */
    INSIST(socket->h2->query_data_len == 0);
    code = ISC_HTTP_ERROR_BAD_REQUEST;
  }

  if (code != ISC_HTTP_ERROR_SUCCESS) {
    goto error;
  }

  if (socket->h2->request_type == ISC_HTTP_REQ_GET) {
    isc_buffer_t decoded_buf;
    isc_buffer_init(&decoded_buf, tmp_buf, sizeof(tmp_buf));
    if (isc_base64_decodestring(socket->h2->query_data,
              &decoded_buf) != ISC_R_SUCCESS)
    {
      code = ISC_HTTP_ERROR_BAD_REQUEST;
      goto error;
    }
    isc_buffer_usedregion(&decoded_buf, &data);
  } else if (socket->h2->request_type == ISC_HTTP_REQ_POST) {
    INSIST(socket->h2->content_length > 0);
    isc_buffer_usedregion(&socket->h2->rbuf, &data);
  } else {
    UNREACHABLE();
  }

  server_call_cb(socket, ISC_R_SUCCESS, &data);

  return 0;

error:
  result = server_send_error_response(code, ngsession, socket);
  if (result != ISC_R_SUCCESS) {
    return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE;
  }
  return 0;
}

static void
http_send_cb(void *arg);

void
isc__nm_http_send(isc_nmhandle_t *handle, const isc_region_t *region,
      isc_nm_cb_t cb, void *cbarg) {
  isc_nmsocket_t *sock = NULL;
  isc__nm_uvreq_t *uvreq = NULL;

  REQUIRE(VALID_NMHANDLE(handle));

  sock = handle->sock;

  REQUIRE(VALID_NMSOCK(sock));
  REQUIRE(sock->tid == isc_tid());

  uvreq = isc__nm_uvreq_get(sock);
  isc_nmhandle_attach(handle, &uvreq->handle);
  uvreq->cb.send = cb;
  uvreq->cbarg = cbarg;

  uvreq->uvbuf.base = (char *)region->base;
  uvreq->uvbuf.len = region->length;

  isc_job_run(sock->worker->loop, &uvreq->job, http_send_cb, uvreq);
}

static void
failed_send_cb(isc_nmsocket_t *sock, isc__nm_uvreq_t *req,
         isc_result_t eresult) {
  REQUIRE(VALID_NMSOCK(sock));
  REQUIRE(VALID_UVREQ(req));

  if (req->cb.send != NULL) {
    isc__nm_sendcb(sock, req, eresult, true);
  } else {
    isc__nm_uvreq_put(&req);
  }
}

static void
client_httpsend(isc_nmhandle_t *handle, isc_nmsocket_t *sock,
    isc__nm_uvreq_t *req) {
  isc_result_t result = ISC_R_SUCCESS;
  isc_nm_cb_t cb = req->cb.send;
  void *cbarg = req->cbarg;

  result = client_send(
    handle, &(isc_region_t){ .base = (uint8_t *)req->uvbuf.base,
           .length = req->uvbuf.len });
  if (result != ISC_R_SUCCESS) {
    failed_send_cb(sock, req, result);
    return;
  }

  http_do_bio(sock->h2->session, handle, cb, cbarg);
  isc__nm_uvreq_put(&req);
}

static void
server_httpsend(isc_nmhandle_t *handle, isc_nmsocket_t *sock,
    isc__nm_uvreq_t *req) {
  size_t content_len_buf_len, cache_control_buf_len;
  isc_result_t result = ISC_R_SUCCESS;
  isc_nm_cb_t cb = req->cb.send;
  void *cbarg = req->cbarg;
  if (isc__nmsocket_closing(sock) ||
      !isc__nm_httpsession_active(handle->httpsession))
  {
    failed_send_cb(sock, req, ISC_R_CANCELED);
    return;
  }

  INSIST(handle->sock->tid == isc_tid());
  INSIST(VALID_NMHANDLE(handle->httpsession->handle));
  INSIST(VALID_NMSOCK(handle->httpsession->handle->sock));

  isc_buffer_init(&sock->h2->wbuf, req->uvbuf.base, req->uvbuf.len);
  isc_buffer_add(&sock->h2->wbuf, req->uvbuf.len);

  content_len_buf_len = snprintf(sock->h2->clenbuf,
               sizeof(sock->h2->clenbuf), "%lu",
               (unsigned long)req->uvbuf.len);
  if (sock->h2->min_ttl == 0) {
    cache_control_buf_len =
      snprintf(sock->h2->cache_control_buf,
         sizeof(sock->h2->cache_control_buf), "%s",
         DEFAULT_CACHE_CONTROL);
  } else {
    cache_control_buf_len =
      snprintf(sock->h2->cache_control_buf,
         sizeof(sock->h2->cache_control_buf),
         "max-age=%" PRIu32, sock->h2->min_ttl);
  }
  const nghttp2_nv hdrs[] = { MAKE_NV2(":status", "200"),
            MAKE_NV2("Content-Type", DNS_MEDIA_TYPE),
            MAKE_NV("Content-Length", sock->h2->clenbuf,
              content_len_buf_len),
            MAKE_NV("Cache-Control",
              sock->h2->cache_control_buf,
              cache_control_buf_len) };

  result = server_send_response(handle->httpsession->ngsession,
              sock->h2->stream_id, hdrs,
              sizeof(hdrs) / sizeof(nghttp2_nv), sock);

  if (result == ISC_R_SUCCESS) {
    http_do_bio(handle->httpsession, handle, cb, cbarg);
  } else {
    cb(handle, result, cbarg);
  }

  isc_buffer_initnull(&sock->h2->wbuf);
  isc__nm_uvreq_put(&req);
}

static void
http_send_cb(void *arg) {
  isc__nm_uvreq_t *req = arg;

  REQUIRE(VALID_UVREQ(req));

  isc_nmsocket_t *sock = req->sock;

  REQUIRE(VALID_NMSOCK(sock));
  REQUIRE(VALID_HTTP2_SESSION(sock->h2->session));

  isc_nmhandle_t *handle = req->handle;

  REQUIRE(VALID_NMHANDLE(handle));

  isc_nm_http_session_t *session = sock->h2->session;
  if (session != NULL && session->client) {
    client_httpsend(handle, sock, req);
  } else {
    server_httpsend(handle, sock, req);
  }
}

void
isc__nm_http_read(isc_nmhandle_t *handle, isc_nm_recv_cb_t cb, void *cbarg) {
  isc_result_t result;
  http_cstream_t *cstream = NULL;
  isc_nm_http_session_t *session = NULL;

  REQUIRE(VALID_NMHANDLE(handle));

  session = handle->sock->h2->session;
  if (!isc__nm_httpsession_active(session)) {
    cb(handle, ISC_R_CANCELED, NULL, cbarg);
    return;
  }

  result = get_http_cstream(handle->sock, &cstream);
  if (result != ISC_R_SUCCESS) {
    return;
  }

  handle->sock->h2->connect.cstream = cstream;
  cstream->read_cb = cb;
  cstream->read_cbarg = cbarg;
  cstream->reading = true;

  if (cstream->sending) {
    result = client_submit_request(session, cstream);
    if (result != ISC_R_SUCCESS) {
      put_http_cstream(session->mctx, cstream);
      return;
    }

    http_do_bio(session, NULL, NULL, NULL);
  }
}

static int
server_on_frame_recv_callback(nghttp2_session *ngsession,
            const nghttp2_frame *frame, void *user_data) {
  isc_nmsocket_t *socket = NULL;

  UNUSED(user_data);

  switch (frame->hd.type) {
  case NGHTTP2_DATA:
  case NGHTTP2_HEADERS:
    /* Check that the client request has finished */
    if (frame->hd.flags & NGHTTP2_FLAG_END_STREAM) {
      socket = nghttp2_session_get_stream_user_data(
        ngsession, frame->hd.stream_id);

      /*
       * For DATA and HEADERS frame,
       * this callback may be called
       * after
       * on_stream_close_callback.
       * Check that the stream is
       * still alive.
       */
      if (socket == NULL) {
        return 0;
      }

      return server_on_request_recv(ngsession, socket);
    }
    break;
  default:
    break;
  }
  return 0;
}

static void
initialize_nghttp2_server_session(isc_nm_http_session_t *session) {
  nghttp2_session_callbacks *callbacks = NULL;
  nghttp2_mem mem;

  init_nghttp2_mem(session->mctx, &mem);

  RUNTIME_CHECK(nghttp2_session_callbacks_new(&callbacks) == 0);

  nghttp2_session_callbacks_set_on_data_chunk_recv_callback(
    callbacks, on_data_chunk_recv_callback);

  nghttp2_session_callbacks_set_on_stream_close_callback(
    callbacks, on_stream_close_callback);

  nghttp2_session_callbacks_set_on_header_callback(
    callbacks, server_on_header_callback);

  nghttp2_session_callbacks_set_on_begin_headers_callback(
    callbacks, server_on_begin_headers_callback);

  nghttp2_session_callbacks_set_on_frame_recv_callback(
    callbacks, server_on_frame_recv_callback);

  RUNTIME_CHECK(nghttp2_session_server_new3(&session->ngsession,
              callbacks, session, NULL,
              &mem) == 0);

  nghttp2_session_callbacks_del(callbacks);
}

static int
server_send_connection_header(isc_nm_http_session_t *session) {
  nghttp2_settings_entry iv[1] = {
    { NGHTTP2_SETTINGS_MAX_CONCURRENT_STREAMS,
      session->max_concurrent_streams }
  };
  int rv;

  rv = nghttp2_submit_settings(session->ngsession, NGHTTP2_FLAG_NONE, iv,
             1);
  if (rv != 0) {
    return -1;
  }
  return 0;
}

/*
 * It is advisable to disable Nagle's algorithm for HTTP/2
 * connections because multiple HTTP/2 streams could be multiplexed
 * over one transport connection. Thus, delays when delivering small
 * packets could bring down performance for the whole session.
 * HTTP/2 is meant to be used this way.
 */
static void
http_transpost_tcp_nodelay(isc_nmhandle_t *transphandle) {
  (void)isc_nmhandle_set_tcp_nodelay(transphandle, true);
}

static isc_result_t
httplisten_acceptcb(isc_nmhandle_t *handle, isc_result_t result, void *cbarg) {
  isc_nmsocket_t *httpserver = (isc_nmsocket_t *)cbarg;
  isc_nm_http_session_t *session = NULL;

  REQUIRE(VALID_NMHANDLE(handle));
  REQUIRE(VALID_NMSOCK(handle->sock));

  if (isc__nm_closing(handle->sock->worker)) {
    return ISC_R_SHUTTINGDOWN;
  } else if (result != ISC_R_SUCCESS) {
    return result;
  }

  REQUIRE(VALID_NMSOCK(httpserver));
  REQUIRE(httpserver->type == isc_nm_httplistener);

  http_initsocket(handle->sock);

  http_transpost_tcp_nodelay(handle);

  new_session(handle->sock->worker->mctx, NULL, &session);
  session->max_concurrent_streams =
    atomic_load_relaxed(&httpserver->h2->max_concurrent_streams);
  initialize_nghttp2_server_session(session);
  handle->sock->h2->session = session;

  isc_nmhandle_attach(handle, &session->handle);
  isc__nmsocket_attach(httpserver, &session->serversocket);
  server_send_connection_header(session);

  isc__nmhandle_set_manual_timer(session->handle, true);

  /* TODO H2 */
  http_do_bio(session, NULL, NULL, NULL);
  return ISC_R_SUCCESS;
}

isc_result_t
isc_nm_listenhttp(uint32_t workers, isc_sockaddr_t *iface, int backlog,
      isc_quota_t *quota, isc_tlsctx_t *ctx,
      isc_nm_http_endpoints_t *eps, uint32_t max_concurrent_streams,
      isc_nm_proxy_type_t proxy_type, isc_nmsocket_t **sockp) {
  isc_nmsocket_t *sock = NULL;
  isc_result_t result = ISC_R_FAILURE;
  isc__networker_t *worker = isc__networker_current();

  REQUIRE(!ISC_LIST_EMPTY(eps->handlers));
  REQUIRE(atomic_load(&eps->in_use) == false);
  REQUIRE(isc_tid() == 0);

  sock = isc_mempool_get(worker->nmsocket_pool);
  isc__nmsocket_init(sock, worker, isc_nm_httplistener, iface, NULL);
  http_initsocket(sock);
  atomic_init(&sock->h2->max_concurrent_streams,
        NGHTTP2_INITIAL_MAX_CONCURRENT_STREAMS);

  isc_nmsocket_set_max_streams(sock, max_concurrent_streams);

  atomic_store(&eps->in_use, true);
  http_init_listener_endpoints(sock, eps);

  switch (proxy_type) {
  case ISC_NM_PROXY_NONE:
    if (ctx != NULL) {
      result = isc_nm_listentls(
        workers, iface, httplisten_acceptcb, sock,
        backlog, quota, ctx, false, &sock->outer);
    } else {
      result = isc_nm_listentcp(workers, iface,
              httplisten_acceptcb, sock,
              backlog, quota, &sock->outer);
    }
    break;
  case ISC_NM_PROXY_PLAIN:
    if (ctx != NULL) {
      result = isc_nm_listentls(
        workers, iface, httplisten_acceptcb, sock,
        backlog, quota, ctx, true, &sock->outer);
    } else {
      result = isc_nm_listenproxystream(
        workers, iface, httplisten_acceptcb, sock,
        backlog, quota, NULL, &sock->outer);
    }
    break;
  case ISC_NM_PROXY_ENCRYPTED:
    INSIST(ctx != NULL);
    result = isc_nm_listenproxystream(
      workers, iface, httplisten_acceptcb, sock, backlog,
      quota, ctx, &sock->outer);
    break;
  default:
    UNREACHABLE();
  }

  if (result != ISC_R_SUCCESS) {
    sock->closed = true;
    isc__nmsocket_detach(&sock);
    return result;
  }

  sock->nchildren = sock->outer->nchildren;
  sock->fd = (uv_os_sock_t)-1;

  *sockp = sock;
  return ISC_R_SUCCESS;
}

isc_nm_http_endpoints_t *
isc_nm_http_endpoints_new(isc_mem_t *mctx) {
  isc_nm_http_endpoints_t *restrict eps;
  REQUIRE(mctx != NULL);

  eps = isc_mem_get(mctx, sizeof(*eps));
  *eps = (isc_nm_http_endpoints_t){ .mctx = NULL };

  isc_mem_attach(mctx, &eps->mctx);
  ISC_LIST_INIT(eps->handlers);
  isc_refcount_init(&eps->references, 1);
  atomic_init(&eps->in_use, false);
  eps->magic = HTTP_ENDPOINTS_MAGIC;

  return eps;
}

void
isc_nm_http_endpoints_detach(isc_nm_http_endpoints_t **restrict epsp) {
  isc_nm_http_endpoints_t *restrict eps;
  isc_mem_t *mctx;

  REQUIRE(epsp != NULL);
  eps = *epsp;
  REQUIRE(VALID_HTTP_ENDPOINTS(eps));

  if (isc_refcount_decrement(&eps->references) > 1) {
    *epsp = NULL;
    return;
  }

  mctx = eps->mctx;

  /* Delete all handlers */
  ISC_LIST_FOREACH(eps->handlers, handler, link) {
    ISC_LIST_DEQUEUE(eps->handlers, handler, link);
    isc_mem_free(mctx, handler->path);
    handler->magic = 0;
    isc_mem_put(mctx, handler, sizeof(*handler));
  }

  eps->magic = 0;

  isc_mem_putanddetach(&mctx, eps, sizeof(*eps));
  *epsp = NULL;
}

void
isc_nm_http_endpoints_attach(isc_nm_http_endpoints_t *source,
           isc_nm_http_endpoints_t **targetp) {
  REQUIRE(VALID_HTTP_ENDPOINTS(source));
  REQUIRE(targetp != NULL && *targetp == NULL);

  isc_refcount_increment(&source->references);

  *targetp = source;
}

static isc_nm_httphandler_t *
http_endpoints_find(const char *request_path,
        isc_nm_http_endpoints_t *restrict eps) {
  REQUIRE(VALID_HTTP_ENDPOINTS(eps));

  if (request_path == NULL || *request_path == '\0') {
    return NULL;
  }

  ISC_LIST_FOREACH(eps->handlers, handler, link) {
    if (!strcmp(request_path, handler->path)) {
      INSIST(VALID_HTTP_HANDLER(handler));
      INSIST(handler->cb != NULL);
      return handler;
    }
  }

  return NULL;
}

isc_result_t
isc_nm_http_endpoints_add(isc_nm_http_endpoints_t *restrict eps,
        const char *uri, const isc_nm_recv_cb_t cb,
        void *cbarg) {
  isc_mem_t *mctx;
  isc_nm_httphandler_t *restrict handler = NULL;

  REQUIRE(VALID_HTTP_ENDPOINTS(eps));
  REQUIRE(isc_nm_http_path_isvalid(uri));
  REQUIRE(cb != NULL);
  REQUIRE(atomic_load(&eps->in_use) == false);

  mctx = eps->mctx;

  if (http_endpoints_find(uri, eps) == NULL) {
    handler = isc_mem_get(mctx, sizeof(*handler));
    *handler = (isc_nm_httphandler_t){
      .cb = cb,
      .cbarg = cbarg,
      .path = isc_mem_strdup(mctx, uri),
      .link = ISC_LINK_INITIALIZER,
      .magic = HTTP_HANDLER_MAGIC
    };

    ISC_LIST_APPEND(eps->handlers, handler, link);
  }

  return ISC_R_SUCCESS;
}

void
isc__nm_http_stoplistening(isc_nmsocket_t *sock) {
  REQUIRE(VALID_NMSOCK(sock));
  REQUIRE(sock->type == isc_nm_httplistener);
  REQUIRE(isc_tid() == sock->tid);

  isc__nmsocket_stop(sock);
}

static void
http_close_direct(isc_nmsocket_t *sock) {
  isc_nm_http_session_t *session = NULL;

  REQUIRE(VALID_NMSOCK(sock));

  sock->closed = true;
  sock->active = false;
  session = sock->h2->session;

  if (session != NULL && session->sending == 0 && !session->reading) {
    /*
     * The socket is going to be closed too early without been
     * used even once (might happen in a case of low level
     * error).
     */
    finish_http_session(session);
  } else if (session != NULL && session->handle) {
    http_do_bio(session, NULL, NULL, NULL);
  }
}

static void
http_close_cb(void *arg) {
  isc_nmsocket_t *sock = arg;
  REQUIRE(VALID_NMSOCK(sock));

  http_close_direct(sock);
  isc__nmsocket_detach(&sock);
}

void
isc__nm_http_close(isc_nmsocket_t *sock) {
  bool destroy = false;
  REQUIRE(VALID_NMSOCK(sock));
  REQUIRE(sock->type == isc_nm_httpsocket);
  REQUIRE(!isc__nmsocket_active(sock));
  REQUIRE(!sock->closing);

  sock->closing = true;

  if (sock->h2->session != NULL && sock->h2->session->closed &&
      sock->tid == isc_tid())
  {
    isc__nm_httpsession_detach(&sock->h2->session);
    destroy = true;
  } else if (sock->h2->session == NULL && sock->tid == isc_tid()) {
    destroy = true;
  }

  if (destroy) {
    http_close_direct(sock);
    isc__nmsocket_prep_destroy(sock);
    return;
  }

  isc__nmsocket_attach(sock, &(isc_nmsocket_t *){ NULL });
  isc_async_run(sock->worker->loop, http_close_cb, sock);
}

static void
failed_httpstream_read_cb(isc_nmsocket_t *sock, isc_result_t result,
        isc_nm_http_session_t *session) {
  isc_region_t data;
  REQUIRE(VALID_NMSOCK(sock));
  INSIST(sock->type == isc_nm_httpsocket);

  if (sock->h2->request_path == NULL) {
    return;
  }

  (void)nghttp2_submit_rst_stream(
    session->ngsession, NGHTTP2_FLAG_END_STREAM,
    sock->h2->stream_id, NGHTTP2_REFUSED_STREAM);
  isc_buffer_usedregion(&sock->h2->rbuf, &data);
  server_call_cb(sock, result, &data);
}

static void
client_call_failed_read_cb(isc_result_t result,
         isc_nm_http_session_t *session) {
  REQUIRE(VALID_HTTP2_SESSION(session));
  REQUIRE(result != ISC_R_SUCCESS);

  ISC_LIST_FOREACH(session->cstreams, cstream, link) {
    /*
     * read_cb could be NULL if cstream was allocated and added
     * to the tracking list, but was not properly initialized due
     * to a low-level error. It is safe to get rid of the object
     * in such a case.
     */
    if (cstream->read_cb != NULL) {
      isc_region_t read_data;
      isc_buffer_usedregion(cstream->rbuf, &read_data);
      cstream->read_cb(session->client_httphandle, result,
           &read_data, cstream->read_cbarg);
    }

    if (result != ISC_R_TIMEDOUT || cstream->read_cb == NULL ||
        !(session->handle != NULL &&
          isc__nmsocket_timer_running(session->handle->sock)))
    {
      ISC_LIST_DEQUEUE(session->cstreams, cstream, link);
      put_http_cstream(session->mctx, cstream);
    }
  }
}

static void
server_call_failed_read_cb(isc_result_t result,
         isc_nm_http_session_t *session) {
  REQUIRE(VALID_HTTP2_SESSION(session));
  REQUIRE(result != ISC_R_SUCCESS);

  ISC_LIST_FOREACH(session->sstreams, h2data, link) {
    failed_httpstream_read_cb(h2data->psock, result, session);
  }

  ISC_LIST_FOREACH(session->sstreams, h2data, link) {
    ISC_LIST_DEQUEUE(session->sstreams, h2data, link);

    /* Cleanup socket in place */
    h2data->psock->active = false;
    h2data->psock->closed = true;
    isc__nmsocket_detach(&h2data->psock);
  }
}

static void
failed_read_cb(isc_result_t result, isc_nm_http_session_t *session) {
  if (session->client) {
    client_call_failed_read_cb(result, session);
    /*
     * If result was ISC_R_TIMEDOUT and the timer was reset,
     * then we still have active streams and should not close
     * the session.
     */
    if (ISC_LIST_EMPTY(session->cstreams)) {
      finish_http_session(session);
    }
  } else {
    server_call_failed_read_cb(result, session);
    /*
     * All streams are now destroyed; close the session.
     */
    finish_http_session(session);
  }
}

void
isc__nm_http_set_maxage(isc_nmhandle_t *handle, const uint32_t ttl) {
  isc_nm_http_session_t *session;
  isc_nmsocket_t *sock;

  REQUIRE(VALID_NMHANDLE(handle));
  REQUIRE(VALID_NMSOCK(handle->sock));

  sock = handle->sock;
  session = sock->h2->session;

  INSIST(VALID_HTTP2_SESSION(session));
  INSIST(!session->client);

  sock->h2->min_ttl = ttl;
}

bool
isc__nm_http_has_encryption(const isc_nmhandle_t *handle) {
  isc_nm_http_session_t *session;
  isc_nmsocket_t *sock;

  REQUIRE(VALID_NMHANDLE(handle));
  REQUIRE(VALID_NMSOCK(handle->sock));

  sock = handle->sock;
  session = sock->h2->session;

  INSIST(VALID_HTTP2_SESSION(session));

  if (session->handle == NULL) {
    return false;
  }

  return isc_nm_has_encryption(session->handle);
}

const char *
isc__nm_http_verify_tls_peer_result_string(const isc_nmhandle_t *handle) {
  isc_nmsocket_t *sock = NULL;
  isc_nm_http_session_t *session;

  REQUIRE(VALID_NMHANDLE(handle));
  REQUIRE(VALID_NMSOCK(handle->sock));
  REQUIRE(handle->sock->type == isc_nm_httpsocket);

  sock = handle->sock;
  session = sock->h2->session;

  /*
   * In the case of a low-level error the session->handle is not
   * attached nor session object is created.
   */
  if (session == NULL && sock->h2->connect.tls_peer_verify_string != NULL)
  {
    return sock->h2->connect.tls_peer_verify_string;
  }

  if (session == NULL) {
    return NULL;
  }

  INSIST(VALID_HTTP2_SESSION(session));

  if (session->handle == NULL) {
    return NULL;
  }

  return isc_nm_verify_tls_peer_result_string(session->handle);
}

void
isc__nm_http_set_tlsctx(isc_nmsocket_t *listener, isc_tlsctx_t *tlsctx) {
  REQUIRE(VALID_NMSOCK(listener));
  REQUIRE(listener->type == isc_nm_httplistener);

  isc_nmsocket_set_tlsctx(listener->outer, tlsctx);
}

void
isc__nm_http_set_max_streams(isc_nmsocket_t *listener,
           const uint32_t max_concurrent_streams) {
  uint32_t max_streams = NGHTTP2_INITIAL_MAX_CONCURRENT_STREAMS;

  REQUIRE(VALID_NMSOCK(listener));
  REQUIRE(listener->type == isc_nm_httplistener);

  if (max_concurrent_streams > 0 &&
      max_concurrent_streams < NGHTTP2_INITIAL_MAX_CONCURRENT_STREAMS)
  {
    max_streams = max_concurrent_streams;
  }

  atomic_store_relaxed(&listener->h2->max_concurrent_streams,
           max_streams);
}

typedef struct http_endpoints_data {
  isc_nmsocket_t *listener;
  isc_nm_http_endpoints_t *endpoints;
} http_endpoints_data_t;

static void
http_set_endpoints_cb(void *arg) {
  http_endpoints_data_t *data = arg;
  const isc_tid_t tid = isc_tid();
  isc_nmsocket_t *listener = data->listener;
  isc_nm_http_endpoints_t *endpoints = data->endpoints;
  isc__networker_t *worker = isc__networker_current();

  isc_mem_put(worker->loop->mctx, data, sizeof(*data));

  isc_nm_http_endpoints_detach(&listener->h2->listener_endpoints[tid]);
  isc_nm_http_endpoints_attach(endpoints,
             &listener->h2->listener_endpoints[tid]);

  isc_nm_http_endpoints_detach(&endpoints);
  isc__nmsocket_detach(&listener);
}

void
isc_nm_http_set_endpoints(isc_nmsocket_t *listener,
        isc_nm_http_endpoints_t *eps) {
  REQUIRE(VALID_NMSOCK(listener));
  REQUIRE(listener->type == isc_nm_httplistener);
  REQUIRE(VALID_HTTP_ENDPOINTS(eps));

  atomic_store(&eps->in_use, true);

  for (size_t i = 0; i < isc_loopmgr_nloops(); i++) {
    isc__networker_t *worker = isc__networker_get(i);
    http_endpoints_data_t *data = isc_mem_cget(worker->loop->mctx,
                 1, sizeof(*data));

    isc__nmsocket_attach(listener, &data->listener);
    isc_nm_http_endpoints_attach(eps, &data->endpoints);

    isc_async_run(worker->loop, http_set_endpoints_cb, data);
  }
}

static void
http_init_listener_endpoints(isc_nmsocket_t *listener,
           isc_nm_http_endpoints_t *epset) {
  size_t nworkers;

  REQUIRE(VALID_NMSOCK(listener));
  REQUIRE(listener->worker != NULL);
  REQUIRE(VALID_HTTP_ENDPOINTS(epset));

  nworkers = (size_t)isc_loopmgr_nloops();
  INSIST(nworkers > 0);

  listener->h2->listener_endpoints =
    isc_mem_cget(listener->worker->mctx, nworkers,
           sizeof(isc_nm_http_endpoints_t *));
  listener->h2->n_listener_endpoints = nworkers;
  for (size_t i = 0; i < nworkers; i++) {
    listener->h2->listener_endpoints[i] = NULL;
    isc_nm_http_endpoints_attach(
      epset, &listener->h2->listener_endpoints[i]);
  }
}

static void
http_cleanup_listener_endpoints(isc_nmsocket_t *listener) {
  REQUIRE(listener->worker != NULL);

  if (listener->h2->listener_endpoints == NULL) {
    return;
  }

  for (size_t i = 0; i < listener->h2->n_listener_endpoints; i++) {
    isc_nm_http_endpoints_detach(
      &listener->h2->listener_endpoints[i]);
  }
  isc_mem_cput(listener->worker->mctx, listener->h2->listener_endpoints,
         listener->h2->n_listener_endpoints,
         sizeof(isc_nm_http_endpoints_t *));
  listener->h2->n_listener_endpoints = 0;
}

static isc_nm_http_endpoints_t *
http_get_listener_endpoints(isc_nmsocket_t *listener, const isc_tid_t tid) {
  isc_nm_http_endpoints_t *eps;
  REQUIRE(VALID_NMSOCK(listener));
  REQUIRE(tid >= 0);
  REQUIRE((size_t)tid < listener->h2->n_listener_endpoints);

  eps = listener->h2->listener_endpoints[tid];
  INSIST(eps != NULL);
  return eps;
}

static const bool base64url_validation_table[256] = {
  false, false, false, false, false, false, false, false, false, false,
  false, false, false, false, false, false, false, false, false, false,
  false, false, false, false, false, false, false, false, false, false,
  false, false, false, false, false, false, false, false, false, false,
  false, false, false, false, false, true,  false, false, true,  true,
  true,  true,  true,  true,  true,  true,  true,  true,  false, false,
  false, false, false, false, false, true,  true,  true,  true,  true,
  true,  true,  true,  true,  true,  true,  true,  true,  true,  true,
  true,  true,  true,  true,  true,  true,  true,  true,  true,  true,
  true,  false, false, false, false, true,  false, true,  true,  true,
  true,  true,  true,  true,  true,  true,  true,  true,  true,  true,
  true,  true,  true,  true,  true,  true,  true,  true,  true,  true,
  true,  true,  true,  false, false, false, false, false, false, false,
  false, false, false, false, false, false, false, false, false, false,
  false, false, false, false, false, false, false, false, false, false,
  false, false, false, false, false, false, false, false, false, false,
  false, false, false, false, false, false, false, false, false, false,
  false, false, false, false, false, false, false, false, false, false,
  false, false, false, false, false, false, false, false, false, false,
  false, false, false, false, false, false, false, false, false, false,
  false, false, false, false, false, false, false, false, false, false,
  false, false, false, false, false, false, false, false, false, false,
  false, false, false, false, false, false, false, false, false, false,
  false, false, false, false, false, false, false, false, false, false,
  false, false, false, false, false, false, false, false, false, false,
  false, false, false, false, false, false
};

char *
isc__nm_base64url_to_base64(isc_mem_t *mem, const char *base64url,
          const size_t base64url_len, size_t *res_len) {
  char *res = NULL;
  size_t i, k, len;

  if (mem == NULL || base64url == NULL || base64url_len == 0) {
    return NULL;
  }

  len = base64url_len % 4 ? base64url_len + (4 - base64url_len % 4)
        : base64url_len;
  res = isc_mem_allocate(mem, len + 1); /* '\0' */

  for (i = 0; i < base64url_len; i++) {
    switch (base64url[i]) {
    case '-':
      res[i] = '+';
      break;
    case '_':
      res[i] = '/';
      break;
    default:
      if (base64url_validation_table[(size_t)base64url[i]]) {
        res[i] = base64url[i];
      } else {
        isc_mem_free(mem, res);
        return NULL;
      }
      break;
    }
  }

  if (base64url_len % 4 != 0) {
    for (k = 0; k < (4 - base64url_len % 4); k++, i++) {
      res[i] = '=';
    }
  }

  INSIST(i == len);

  SET_IF_NOT_NULL(res_len, len);

  res[len] = '\0';

  return res;
}

char *
isc__nm_base64_to_base64url(isc_mem_t *mem, const char *base64,
          const size_t base64_len, size_t *res_len) {
  char *res = NULL;
  size_t i;

  if (mem == NULL || base64 == NULL || base64_len == 0) {
    return NULL;
  }

  res = isc_mem_allocate(mem, base64_len + 1); /* '\0' */

  for (i = 0; i < base64_len; i++) {
    switch (base64[i]) {
    case '+':
      res[i] = '-';
      break;
    case '/':
      res[i] = '_';
      break;
    case '=':
      goto end;
      break;
    default:
      /*
       * All other characters from
       * the alphabet are the same
       * for both base64 and
       * base64url, so we can reuse
       * the validation table for
       * the rest of the characters.
       */
      if (base64[i] != '-' && base64[i] != '_' &&
          base64url_validation_table[(size_t)base64[i]])
      {
        res[i] = base64[i];
      } else {
        isc_mem_free(mem, res);
        return NULL;
      }
      break;
    }
  }
end:
  SET_IF_NOT_NULL(res_len, i);

  res[i] = '\0';

  return res;
}

static void
http_initsocket(isc_nmsocket_t *sock) {
  REQUIRE(sock != NULL);

  sock->h2 = isc_mem_get(sock->worker->mctx, sizeof(*sock->h2));
  *sock->h2 = (isc_nmsocket_h2_t){
    .request_type = ISC_HTTP_REQ_UNSUPPORTED,
    .request_scheme = ISC_HTTP_SCHEME_UNSUPPORTED,
  };
}

void
isc__nm_http_cleanup_data(isc_nmsocket_t *sock) {
  switch (sock->type) {
  case isc_nm_httplistener:
  case isc_nm_httpsocket:
    if (sock->type == isc_nm_httplistener &&
        sock->h2->listener_endpoints != NULL)
    {
      /* Delete all handlers */
      http_cleanup_listener_endpoints(sock);
    }

    if (sock->type == isc_nm_httpsocket &&
        sock->h2->peer_endpoints != NULL)
    {
      isc_nm_http_endpoints_detach(&sock->h2->peer_endpoints);
    }

    if (sock->h2->request_path != NULL) {
      isc_mem_free(sock->worker->mctx,
             sock->h2->request_path);
    }

    if (sock->h2->query_data != NULL) {
      isc_mem_free(sock->worker->mctx, sock->h2->query_data);
    }

    INSIST(sock->h2->connect.cstream == NULL);

    if (isc_buffer_base(&sock->h2->rbuf) != NULL) {
      void *base = isc_buffer_base(&sock->h2->rbuf);
      isc_mem_free(sock->worker->mctx, base);
      isc_buffer_initnull(&sock->h2->rbuf);
    }
    FALLTHROUGH;
  case isc_nm_proxystreamlistener:
  case isc_nm_proxystreamsocket:
  case isc_nm_tcpsocket:
  case isc_nm_tlssocket:
    if (sock->h2 != NULL) {
      if (sock->h2->session != NULL) {
        if (sock->h2->connect.uri != NULL) {
          isc_mem_free(sock->worker->mctx,
                 sock->h2->connect.uri);
        }
        isc__nm_httpsession_detach(&sock->h2->session);
      }

      isc_mem_put(sock->worker->mctx, sock->h2,
            sizeof(*sock->h2));
    };
    break;
  default:
    break;
  }
}

void
isc__nm_http_cleartimeout(isc_nmhandle_t *handle) {
  isc_nmsocket_t *sock = NULL;

  REQUIRE(VALID_NMHANDLE(handle));
  REQUIRE(VALID_NMSOCK(handle->sock));
  REQUIRE(handle->sock->type == isc_nm_httpsocket);

  sock = handle->sock;
  if (sock->h2->session != NULL && sock->h2->session->handle != NULL) {
    INSIST(VALID_HTTP2_SESSION(sock->h2->session));
    INSIST(VALID_NMHANDLE(sock->h2->session->handle));
    isc_nmhandle_cleartimeout(sock->h2->session->handle);
  }
}

void
isc__nm_http_settimeout(isc_nmhandle_t *handle, uint32_t timeout) {
  isc_nmsocket_t *sock = NULL;

  REQUIRE(VALID_NMHANDLE(handle));
  REQUIRE(VALID_NMSOCK(handle->sock));
  REQUIRE(handle->sock->type == isc_nm_httpsocket);

  sock = handle->sock;
  if (sock->h2->session != NULL && sock->h2->session->handle != NULL) {
    INSIST(VALID_HTTP2_SESSION(sock->h2->session));
    INSIST(VALID_NMHANDLE(sock->h2->session->handle));
    isc_nmhandle_settimeout(sock->h2->session->handle, timeout);
  }
}

void
isc__nmhandle_http_keepalive(isc_nmhandle_t *handle, bool value) {
  isc_nmsocket_t *sock = NULL;

  REQUIRE(VALID_NMHANDLE(handle));
  REQUIRE(VALID_NMSOCK(handle->sock));
  REQUIRE(handle->sock->type == isc_nm_httpsocket);

  sock = handle->sock;
  if (sock->h2->session != NULL && sock->h2->session->handle) {
    INSIST(VALID_HTTP2_SESSION(sock->h2->session));
    INSIST(VALID_NMHANDLE(sock->h2->session->handle));

    isc_nmhandle_keepalive(sock->h2->session->handle, value);
  }
}

void
isc_nm_http_makeuri(const bool https, const isc_sockaddr_t *sa,
        const char *hostname, const uint16_t http_port,
        const char *abs_path, char *outbuf,
        const size_t outbuf_len) {
  char saddr[INET6_ADDRSTRLEN] = { 0 };
  int family;
  bool ipv6_addr = false;
  struct sockaddr_in6 sa6;
  uint16_t host_port = http_port;
  const char *host = NULL;

  REQUIRE(outbuf != NULL);
  REQUIRE(outbuf_len != 0);
  REQUIRE(isc_nm_http_path_isvalid(abs_path));

  /* If hostname is specified, use that. */
  if (hostname != NULL && hostname[0] != '\0') {
    /*
     * The host name could be an IPv6 address. If so,
     * wrap it between [ and ].
     */
    if (inet_pton(AF_INET6, hostname, &sa6) == 1 &&
        hostname[0] != '[')
    {
      ipv6_addr = true;
    }
    host = hostname;
  } else {
    /*
     * A hostname was not specified; build one from
     * the given IP address.
     */
    INSIST(sa != NULL);
    family = ((const struct sockaddr *)&sa->type.sa)->sa_family;
    host_port = ntohs(family == AF_INET ? sa->type.sin.sin_port
                : sa->type.sin6.sin6_port);
    ipv6_addr = family == AF_INET6;
    (void)inet_ntop(
      family,
      family == AF_INET
        ? (const struct sockaddr *)&sa->type.sin.sin_addr
        : (const struct sockaddr *)&sa->type.sin6
            .sin6_addr,
      saddr, sizeof(saddr));
    host = saddr;
  }

  /*
   * If the port number was not specified, the default
   * depends on whether we're using encryption or not.
   */
  if (host_port == 0) {
    host_port = https ? 443 : 80;
  }

  (void)snprintf(outbuf, outbuf_len, "%s://%s%s%s:%u%s",
           https ? "https" : "http", ipv6_addr ? "[" : "", host,
           ipv6_addr ? "]" : "", host_port, abs_path);
}

/*
 * DoH GET Query String Scanner-less Recursive Descent Parser/Verifier
 *
 * It is based on the following grammar (using WSN/EBNF):
 *
 * S                = query-string.
 * query-string     = ['?'] { key-value-pair } EOF.
 * key-value-pair   = key '=' value [ '&' ].
 * key              = ('_' | alpha) { '_' | alnum}.
 * value            = value-char {value-char}.
 * value-char       = unreserved-char | percent-charcode.
 * unreserved-char  = alnum |'_' | '.' | '-' | '~'. (* RFC3986, Section 2.3 *)
 * percent-charcode = '%' hexdigit hexdigit.
 * ...
 *
 * Should be good enough.
 */
typedef struct isc_httpparser_state {
  const char *str;

  const char *last_key;
  size_t last_key_len;

  const char *last_value;
  size_t last_value_len;

  bool query_found;
  const char *query;
  size_t query_len;
} isc_httpparser_state_t;

#define MATCH(ch)      (st->str[0] == (ch))
#define MATCH_ALPHA()  isalpha((unsigned char)(st->str[0]))
#define MATCH_DIGIT()  isdigit((unsigned char)(st->str[0]))
#define MATCH_ALNUM()  isalnum((unsigned char)(st->str[0]))
#define MATCH_XDIGIT() isxdigit((unsigned char)(st->str[0]))
#define ADVANCE()      st->str++
#define GETP()         (st->str)

static bool
rule_query_string(isc_httpparser_state_t *st);

bool
isc__nm_parse_httpquery(const char *query_string, const char **start,
      size_t *len) {
  isc_httpparser_state_t state;

  REQUIRE(start != NULL);
  REQUIRE(len != NULL);

  if (query_string == NULL || query_string[0] == '\0') {
    return false;
  }

  state = (isc_httpparser_state_t){ .str = query_string };
  if (!rule_query_string(&state)) {
    return false;
  }

  if (!state.query_found) {
    return false;
  }

  *start = state.query;
  *len = state.query_len;

  return true;
}

static bool
rule_key_value_pair(isc_httpparser_state_t *st);

static bool
rule_key(isc_httpparser_state_t *st);

static bool
rule_value(isc_httpparser_state_t *st);

static bool
rule_value_char(isc_httpparser_state_t *st);

static bool
rule_percent_charcode(isc_httpparser_state_t *st);

static bool
rule_unreserved_char(isc_httpparser_state_t *st);

static bool
rule_query_string(isc_httpparser_state_t *st) {
  if (MATCH('?')) {
    ADVANCE();
  }

  while (rule_key_value_pair(st)) {
    /* skip */;
  }

  if (!MATCH('\0')) {
    return false;
  }

  ADVANCE();
  return true;
}

static bool
rule_key_value_pair(isc_httpparser_state_t *st) {
  if (!rule_key(st)) {
    return false;
  }

  if (MATCH('=')) {
    ADVANCE();
  } else {
    return false;
  }

  if (rule_value(st)) {
    const char dns[] = "dns";
    if (st->last_key_len == sizeof(dns) - 1 &&
        memcmp(st->last_key, dns, sizeof(dns) - 1) == 0)
    {
      st->query_found = true;
      st->query = st->last_value;
      st->query_len = st->last_value_len;
    }
  } else {
    return false;
  }

  if (MATCH('&')) {
    ADVANCE();
  }

  return true;
}

static bool
rule_key(isc_httpparser_state_t *st) {
  if (MATCH('_') || MATCH_ALPHA()) {
    st->last_key = GETP();
    ADVANCE();
  } else {
    return false;
  }

  while (MATCH('_') || MATCH_ALNUM()) {
    ADVANCE();
  }

  st->last_key_len = GETP() - st->last_key;
  return true;
}

static bool
rule_value(isc_httpparser_state_t *st) {
  const char *s = GETP();
  if (!rule_value_char(st)) {
    return false;
  }

  st->last_value = s;
  while (rule_value_char(st)) {
    /* skip */;
  }
  st->last_value_len = GETP() - st->last_value;
  return true;
}

static bool
rule_value_char(isc_httpparser_state_t *st) {
  if (rule_unreserved_char(st)) {
    return true;
  }

  return rule_percent_charcode(st);
}

static bool
rule_unreserved_char(isc_httpparser_state_t *st) {
  if (MATCH_ALNUM() || MATCH('_') || MATCH('.') || MATCH('-') ||
      MATCH('~'))
  {
    ADVANCE();
    return true;
  }
  return false;
}

static bool
rule_percent_charcode(isc_httpparser_state_t *st) {
  if (MATCH('%')) {
    ADVANCE();
  } else {
    return false;
  }

  if (!MATCH_XDIGIT()) {
    return false;
  }
  ADVANCE();

  if (!MATCH_XDIGIT()) {
    return false;
  }
  ADVANCE();

  return true;
}

/*
 * DoH URL Location Verifier. Based on the following grammar (EBNF/WSN
 * notation):
 *
 * S             = path_absolute.
 * path_absolute = '/' [ segments ] '\0'.
 * segments      = segment_nz { slash_segment }.
 * slash_segment = '/' segment.
 * segment       = { pchar }.
 * segment_nz    = pchar { pchar }.
 * pchar         = unreserved | pct_encoded | sub_delims | ':' | '@'.
 * unreserved    = ALPHA | DIGIT | '-' | '.' | '_' | '~'.
 * pct_encoded   = '%' XDIGIT XDIGIT.
 * sub_delims    = '!' | '$' | '&' | '\'' | '(' | ')' | '*' | '+' |
 *                 ',' | ';' | '='.
 *
 * The grammar is extracted from RFC 3986. It is slightly modified to
 * aid in parser creation, but the end result is the same
 * (path_absolute is defined slightly differently - split into
 * multiple productions).
 *
 * https://datatracker.ietf.org/doc/html/rfc3986#appendix-A
 */

typedef struct isc_http_location_parser_state {
  const char *str;
} isc_http_location_parser_state_t;

static bool
rule_loc_path_absolute(isc_http_location_parser_state_t *);

static bool
rule_loc_segments(isc_http_location_parser_state_t *);

static bool
rule_loc_slash_segment(isc_http_location_parser_state_t *);

static bool
rule_loc_segment(isc_http_location_parser_state_t *);

static bool
rule_loc_segment_nz(isc_http_location_parser_state_t *);

static bool
rule_loc_pchar(isc_http_location_parser_state_t *);

static bool
rule_loc_unreserved(isc_http_location_parser_state_t *);

static bool
rule_loc_pct_encoded(isc_http_location_parser_state_t *);

static bool
rule_loc_sub_delims(isc_http_location_parser_state_t *);

static bool
rule_loc_path_absolute(isc_http_location_parser_state_t *st) {
  if (MATCH('/')) {
    ADVANCE();
  } else {
    return false;
  }

  (void)rule_loc_segments(st);

  if (MATCH('\0')) {
    ADVANCE();
  } else {
    return false;
  }

  return true;
}

static bool
rule_loc_segments(isc_http_location_parser_state_t *st) {
  if (!rule_loc_segment_nz(st)) {
    return false;
  }

  while (rule_loc_slash_segment(st)) {
    /* zero or more */;
  }

  return true;
}

static bool
rule_loc_slash_segment(isc_http_location_parser_state_t *st) {
  if (MATCH('/')) {
    ADVANCE();
  } else {
    return false;
  }

  return rule_loc_segment(st);
}

static bool
rule_loc_segment(isc_http_location_parser_state_t *st) {
  while (rule_loc_pchar(st)) {
    /* zero or more */;
  }

  return true;
}

static bool
rule_loc_segment_nz(isc_http_location_parser_state_t *st) {
  if (!rule_loc_pchar(st)) {
    return false;
  }

  while (rule_loc_pchar(st)) {
    /* zero or more */;
  }

  return true;
}

static bool
rule_loc_pchar(isc_http_location_parser_state_t *st) {
  if (rule_loc_unreserved(st)) {
    return true;
  } else if (rule_loc_pct_encoded(st)) {
    return true;
  } else if (rule_loc_sub_delims(st)) {
    return true;
  } else if (MATCH(':') || MATCH('@')) {
    ADVANCE();
    return true;
  }

  return false;
}

static bool
rule_loc_unreserved(isc_http_location_parser_state_t *st) {
  if (MATCH_ALPHA() | MATCH_DIGIT() | MATCH('-') | MATCH('.') |
      MATCH('_') | MATCH('~'))
  {
    ADVANCE();
    return true;
  }
  return false;
}

static bool
rule_loc_pct_encoded(isc_http_location_parser_state_t *st) {
  if (!MATCH('%')) {
    return false;
  }
  ADVANCE();

  if (!MATCH_XDIGIT()) {
    return false;
  }
  ADVANCE();

  if (!MATCH_XDIGIT()) {
    return false;
  }
  ADVANCE();

  return true;
}

static bool
rule_loc_sub_delims(isc_http_location_parser_state_t *st) {
  if (MATCH('!') | MATCH('$') | MATCH('&') | MATCH('\'') | MATCH('(') |
      MATCH(')') | MATCH('*') | MATCH('+') | MATCH(',') | MATCH(';') |
      MATCH('='))
  {
    ADVANCE();
    return true;
  }

  return false;
}

bool
isc_nm_http_path_isvalid(const char *path) {
  isc_http_location_parser_state_t state = { 0 };

  REQUIRE(path != NULL);

  state.str = path;

  return rule_loc_path_absolute(&state);
}

Coverage Report

Created: 2026-10-17 06:53