2014-03-30 12:09:21 +02:00
|
|
|
Tutorial: HTTP/2 server
|
2013-12-26 15:37:42 +01:00
|
|
|
=========================
|
|
|
|
|
|
|
|
In this tutorial, we are going to write single-threaded, event-based
|
2014-03-30 12:09:21 +02:00
|
|
|
HTTP/2 web server, which supports HTTPS only. It can handle
|
2014-08-01 10:09:17 +02:00
|
|
|
concurrent multiple requests, but only the GET method is supported. The
|
2013-12-28 14:09:17 +01:00
|
|
|
complete source code, `libevent-server.c`_, is attached at the end of
|
|
|
|
this page. It also resides in examples directory in the archive or
|
2013-12-26 15:37:42 +01:00
|
|
|
repository.
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
This simple server takes 3 arguments, a port number to listen to, a path to
|
|
|
|
your SSL/TLS private key file and a path to your certificate file. Its
|
|
|
|
synopsis is like this::
|
2013-12-26 15:37:42 +01:00
|
|
|
|
|
|
|
$ libevent-server PORT /path/to/server.key /path/to/server.crt
|
|
|
|
|
|
|
|
We use libevent in this tutorial to handle networking I/O. Please
|
2013-12-28 14:09:17 +01:00
|
|
|
note that nghttp2 itself does not depend on libevent.
|
2013-12-26 15:37:42 +01:00
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
First we create a setup routine for libevent and OpenSSL in the functions
|
|
|
|
``main()`` and ``run()``. One thing in there you should look at, is the setup
|
|
|
|
of the NPN callback. The NPN callback is used for the server to advertise
|
|
|
|
which application protocols the server supports to a client. In this example
|
|
|
|
program, when creating ``SSL_CTX`` object, we store the application protocol
|
|
|
|
name in the wire format of NPN in a statically allocated buffer. This is safe
|
|
|
|
because we only create one ``SSL_CTX`` object in the program's entire life
|
|
|
|
time::
|
2013-12-26 15:37:42 +01:00
|
|
|
|
|
|
|
static unsigned char next_proto_list[256];
|
|
|
|
static size_t next_proto_list_len;
|
|
|
|
|
|
|
|
static int next_proto_cb(SSL *s, const unsigned char **data, unsigned int *len,
|
|
|
|
void *arg)
|
|
|
|
{
|
|
|
|
*data = next_proto_list;
|
2014-06-14 15:47:56 +02:00
|
|
|
*len = (unsigned int)next_proto_list_len;
|
2013-12-26 15:37:42 +01:00
|
|
|
return SSL_TLSEXT_ERR_OK;
|
|
|
|
}
|
|
|
|
|
|
|
|
static SSL_CTX* create_ssl_ctx(const char *key_file, const char *cert_file)
|
|
|
|
{
|
|
|
|
SSL_CTX *ssl_ctx;
|
|
|
|
ssl_ctx = SSL_CTX_new(SSLv23_server_method());
|
|
|
|
|
|
|
|
...
|
|
|
|
|
|
|
|
next_proto_list[0] = NGHTTP2_PROTO_VERSION_ID_LEN;
|
|
|
|
memcpy(&next_proto_list[1], NGHTTP2_PROTO_VERSION_ID,
|
|
|
|
NGHTTP2_PROTO_VERSION_ID_LEN);
|
|
|
|
next_proto_list_len = 1 + NGHTTP2_PROTO_VERSION_ID_LEN;
|
|
|
|
|
|
|
|
SSL_CTX_set_next_protos_advertised_cb(ssl_ctx, next_proto_cb, NULL);
|
|
|
|
return ssl_ctx;
|
|
|
|
}
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
The wire format of NPN is a sequence of length prefixed string. Exactly one
|
|
|
|
byte is used to specify the length of each protocol identifier. In this
|
|
|
|
tutorial, we advertise the specific HTTP/2 protocol version the current
|
|
|
|
nghttp2 library supports. The nghttp2 library exports its identifier in
|
|
|
|
:macro:`NGHTTP2_PROTO_VERSION_ID`. The ``next_proto_cb()`` function is the
|
|
|
|
server-side NPN callback. In the OpenSSL implementation, we just assign the
|
|
|
|
pointer to the NPN buffers we filled in earlier. The NPN callback function is
|
|
|
|
set to the ``SSL_CTX`` object using
|
2013-12-26 15:37:42 +01:00
|
|
|
``SSL_CTX_set_next_protos_advertised_cb()``.
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
We use the ``app_content`` structure to store application-wide data::
|
2013-12-26 15:37:42 +01:00
|
|
|
|
|
|
|
struct app_context {
|
|
|
|
SSL_CTX *ssl_ctx;
|
|
|
|
struct event_base *evbase;
|
|
|
|
};
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
We use the ``http2_session_data`` structure to store session-level
|
|
|
|
(which corresponds to one HTTP/2 connection) data::
|
2013-12-26 15:37:42 +01:00
|
|
|
|
|
|
|
typedef struct http2_session_data {
|
|
|
|
struct http2_stream_data root;
|
|
|
|
struct bufferevent *bev;
|
|
|
|
app_context *app_ctx;
|
|
|
|
nghttp2_session *session;
|
|
|
|
char *client_addr;
|
|
|
|
size_t handshake_leftlen;
|
|
|
|
} http2_session_data;
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
We use the ``http2_stream_data`` structure to store stream-level data::
|
2013-12-26 15:37:42 +01:00
|
|
|
|
|
|
|
typedef struct http2_stream_data {
|
|
|
|
struct http2_stream_data *prev, *next;
|
|
|
|
char *request_path;
|
|
|
|
int32_t stream_id;
|
|
|
|
int fd;
|
|
|
|
} http2_stream_data;
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
A single HTTP/2 session can have multiple streams. We manage these multiple
|
|
|
|
streams with a doubly linked list. The first element of this list is pointed
|
|
|
|
to by the ``root->next`` in ``http2_session_data``. Initially, ``root->next``
|
|
|
|
is ``NULL``. The ``handshake_leftlen`` member of ``http2_session_data`` is
|
|
|
|
used to track the number of bytes remaining when receiving the first client
|
|
|
|
connection preface (:macro:`NGHTTP2_CLIENT_CONNECTION_PREFACE`), which is a 24
|
|
|
|
bytes long magic string from the client. We use libevent's bufferevent
|
|
|
|
structure to perform network I/O. Note that the bufferevent object is kept in
|
|
|
|
``http2_session_data`` and not in ``http2_stream_data``. This is because
|
|
|
|
``http2_stream_data`` is just a logical stream multiplexed over the single
|
|
|
|
connection managed by bufferevent in ``http2_session_data``.
|
|
|
|
|
|
|
|
We first create a listener object to accept incoming connections. We use
|
|
|
|
libevent's ``struct evconnlistener`` for this purpose::
|
2013-12-26 15:37:42 +01:00
|
|
|
|
|
|
|
static void start_listen(struct event_base *evbase, const char *service,
|
|
|
|
app_context *app_ctx)
|
|
|
|
{
|
|
|
|
int rv;
|
|
|
|
struct addrinfo hints;
|
|
|
|
struct addrinfo *res, *rp;
|
|
|
|
|
|
|
|
memset(&hints, 0, sizeof(hints));
|
|
|
|
hints.ai_family = AF_UNSPEC;
|
|
|
|
hints.ai_socktype = SOCK_STREAM;
|
|
|
|
hints.ai_flags = AI_PASSIVE;
|
|
|
|
#ifdef AI_ADDRCONFIG
|
|
|
|
hints.ai_flags |= AI_ADDRCONFIG;
|
|
|
|
#endif // AI_ADDRCONFIG
|
|
|
|
|
|
|
|
rv = getaddrinfo(NULL, service, &hints, &res);
|
|
|
|
if(rv != 0) {
|
|
|
|
errx(1, NULL);
|
|
|
|
}
|
|
|
|
for(rp = res; rp; rp = rp->ai_next) {
|
|
|
|
struct evconnlistener *listener;
|
|
|
|
listener = evconnlistener_new_bind(evbase, acceptcb, app_ctx,
|
|
|
|
LEV_OPT_CLOSE_ON_FREE |
|
|
|
|
LEV_OPT_REUSEABLE, -1,
|
|
|
|
rp->ai_addr, rp->ai_addrlen);
|
|
|
|
if(listener) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
errx(1, "Could not start listener");
|
|
|
|
}
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
We specify the ``acceptcb`` callback which is called when a new connection is
|
|
|
|
accepted::
|
2013-12-26 15:37:42 +01:00
|
|
|
|
|
|
|
static void acceptcb(struct evconnlistener *listener, int fd,
|
|
|
|
struct sockaddr *addr, int addrlen, void *arg)
|
|
|
|
{
|
|
|
|
app_context *app_ctx = (app_context*)arg;
|
|
|
|
http2_session_data *session_data;
|
|
|
|
|
|
|
|
session_data = create_http2_session_data(app_ctx, fd, addr, addrlen);
|
|
|
|
bufferevent_setcb(session_data->bev, handshake_readcb, NULL, eventcb,
|
|
|
|
session_data);
|
|
|
|
}
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
Here we create the ``http2_session_data`` object. The bufferevent for this
|
|
|
|
connection is also initialized at this time. We specify two callbacks for the
|
|
|
|
bufferevent: ``handshake_readcb`` and ``eventcb``.
|
2013-12-26 15:37:42 +01:00
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
The ``eventcb()`` callback is invoked by the libevent event loop when an event
|
2013-12-28 14:09:17 +01:00
|
|
|
(e.g., connection has been established, timeout, etc) happens on the
|
|
|
|
underlying network socket::
|
2013-12-26 15:37:42 +01:00
|
|
|
|
|
|
|
static void eventcb(struct bufferevent *bev, short events, void *ptr)
|
|
|
|
{
|
|
|
|
http2_session_data *session_data = (http2_session_data*)ptr;
|
|
|
|
if(events & BEV_EVENT_CONNECTED) {
|
|
|
|
fprintf(stderr, "%s connected\n", session_data->client_addr);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
if(events & BEV_EVENT_EOF) {
|
2014-01-17 17:17:09 +01:00
|
|
|
fprintf(stderr, "%s EOF\n", session_data->client_addr);
|
2013-12-26 15:37:42 +01:00
|
|
|
} else if(events & BEV_EVENT_ERROR) {
|
|
|
|
fprintf(stderr, "%s network error\n", session_data->client_addr);
|
|
|
|
} else if(events & BEV_EVENT_TIMEOUT) {
|
|
|
|
fprintf(stderr, "%s timeout\n", session_data->client_addr);
|
|
|
|
}
|
|
|
|
delete_http2_session_data(session_data);
|
|
|
|
}
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
For the ``BEV_EVENT_EOF``, ``BEV_EVENT_ERROR`` and ``BEV_EVENT_TIMEOUT``
|
|
|
|
events, we just simply tear down the connection. The
|
|
|
|
``delete_http2_session_data()`` function destroys the ``http2_session_data``
|
|
|
|
object and thus also its bufferevent member. As a result, the underlying
|
|
|
|
connection is closed. The ``BEV_EVENT_CONNECTED`` event is invoked when
|
|
|
|
SSL/TLS handshake is finished successfully.
|
2013-12-26 15:37:42 +01:00
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
``handshake_readcb()`` is a callback function to handle a 24 bytes magic byte
|
|
|
|
string coming from a client, since the nghttp2 library does not handle it::
|
2013-12-26 15:37:42 +01:00
|
|
|
|
|
|
|
static void handshake_readcb(struct bufferevent *bev, void *ptr)
|
|
|
|
{
|
|
|
|
http2_session_data *session_data = (http2_session_data*)ptr;
|
|
|
|
uint8_t data[24];
|
|
|
|
struct evbuffer *input = bufferevent_get_input(session_data->bev);
|
|
|
|
int readlen = evbuffer_remove(input, data, session_data->handshake_leftlen);
|
2014-03-30 14:08:04 +02:00
|
|
|
const char *conhead = NGHTTP2_CLIENT_CONNECTION_PREFACE;
|
2013-12-26 15:37:42 +01:00
|
|
|
|
2014-03-30 14:08:04 +02:00
|
|
|
if(memcmp(conhead + NGHTTP2_CLIENT_CONNECTION_PREFACE_LEN
|
2013-12-26 15:37:42 +01:00
|
|
|
- session_data->handshake_leftlen, data, readlen) != 0) {
|
|
|
|
delete_http2_session_data(session_data);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
session_data->handshake_leftlen -= readlen;
|
|
|
|
if(session_data->handshake_leftlen == 0) {
|
|
|
|
bufferevent_setcb(session_data->bev, readcb, writecb, eventcb, ptr);
|
|
|
|
/* Process pending data in buffer since they are not notified
|
|
|
|
further */
|
|
|
|
initialize_nghttp2_session(session_data);
|
|
|
|
if(send_server_connection_header(session_data) != 0) {
|
|
|
|
delete_http2_session_data(session_data);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
if(session_recv(session_data) != 0) {
|
|
|
|
delete_http2_session_data(session_data);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-12-28 14:09:17 +01:00
|
|
|
We check that the received byte string matches
|
2014-08-01 10:09:17 +02:00
|
|
|
:macro:`NGHTTP2_CLIENT_CONNECTION_PREFACE`. When they match, the connection
|
|
|
|
state is ready to start the HTTP/2 communication. First we change the callback
|
|
|
|
functions for the bufferevent object. We use the same ``eventcb`` callback as
|
|
|
|
before, but we specify new ``readcb`` and ``writecb`` functions to handle the
|
|
|
|
HTTP/2 communication. These two functions are described later.
|
|
|
|
|
|
|
|
We initialize a nghttp2 session object which is done in
|
2013-12-26 15:37:42 +01:00
|
|
|
``initialize_nghttp2_session()``::
|
|
|
|
|
|
|
|
static void initialize_nghttp2_session(http2_session_data *session_data)
|
|
|
|
{
|
|
|
|
nghttp2_session_callbacks callbacks = {0};
|
|
|
|
|
|
|
|
callbacks.send_callback = send_callback;
|
|
|
|
callbacks.on_frame_recv_callback = on_frame_recv_callback;
|
|
|
|
callbacks.on_stream_close_callback = on_stream_close_callback;
|
2014-01-17 16:06:24 +01:00
|
|
|
callbacks.on_header_callback = on_header_callback;
|
2014-01-29 13:49:50 +01:00
|
|
|
callbacks.on_begin_headers_callback = on_begin_headers_callback;
|
2013-12-26 15:37:42 +01:00
|
|
|
nghttp2_session_server_new(&session_data->session, &callbacks, session_data);
|
|
|
|
}
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
Since we are creating a server, the nghttp2 session object is created using
|
|
|
|
`nghttp2_session_server_new()` function. We registers five callbacks for
|
2013-12-26 15:37:42 +01:00
|
|
|
nghttp2 session object. We'll talk about these callbacks later.
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
After initialization of the nghttp2 session object, we are going to send
|
|
|
|
a server connection header in ``send_server_connection_header()``::
|
2013-12-26 15:37:42 +01:00
|
|
|
|
|
|
|
static int send_server_connection_header(http2_session_data *session_data)
|
|
|
|
{
|
|
|
|
nghttp2_settings_entry iv[1] = {
|
|
|
|
{ NGHTTP2_SETTINGS_MAX_CONCURRENT_STREAMS, 100 }
|
|
|
|
};
|
|
|
|
int rv;
|
|
|
|
|
|
|
|
rv = nghttp2_submit_settings(session_data->session, NGHTTP2_FLAG_NONE,
|
|
|
|
iv, ARRLEN(iv));
|
|
|
|
if(rv != 0) {
|
|
|
|
warnx("Fatal error: %s", nghttp2_strerror(rv));
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
The server connection header is a SETTINGS frame. We specify
|
|
|
|
SETTINGS_MAX_CONCURRENT_STREAMS to 100 in the SETTINGS frame. To queue
|
2013-12-26 15:37:42 +01:00
|
|
|
the SETTINGS frame for the transmission, we use
|
|
|
|
`nghttp2_submit_settings()`. Note that `nghttp2_submit_settings()`
|
2014-08-01 10:09:17 +02:00
|
|
|
function only queues the frame and it does not actually send it. All
|
|
|
|
functions in the ``nghttp2_submit_*()`` family have this property. To
|
|
|
|
actually send the frame, `nghttp2_session_send()` should be used, as
|
|
|
|
described later.
|
|
|
|
|
|
|
|
Since bufferevent may buffer more than the first 24 bytes from the client, we
|
|
|
|
have to process them here since libevent won't invoke callback functions for
|
|
|
|
this pending data. To process the received data, we call the
|
2013-12-26 15:37:42 +01:00
|
|
|
``session_recv()`` function::
|
|
|
|
|
|
|
|
static int session_recv(http2_session_data *session_data)
|
|
|
|
{
|
2014-06-14 15:47:56 +02:00
|
|
|
ssize_t readlen;
|
2013-12-26 15:37:42 +01:00
|
|
|
struct evbuffer *input = bufferevent_get_input(session_data->bev);
|
|
|
|
size_t datalen = evbuffer_get_length(input);
|
|
|
|
unsigned char *data = evbuffer_pullup(input, -1);
|
2014-06-14 15:47:56 +02:00
|
|
|
|
|
|
|
readlen = nghttp2_session_mem_recv(session_data->session, data, datalen);
|
|
|
|
if(readlen < 0) {
|
|
|
|
warnx("Fatal error: %s", nghttp2_strerror((int)readlen));
|
2013-12-26 15:37:42 +01:00
|
|
|
return -1;
|
|
|
|
}
|
2014-06-18 06:14:07 +02:00
|
|
|
if(evbuffer_drain(input, readlen) != 0) {
|
|
|
|
warnx("Fatal error: evbuffer_drain failed");
|
|
|
|
return -1;
|
|
|
|
}
|
2013-12-26 15:37:42 +01:00
|
|
|
if(session_send(session_data) != 0) {
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
In this function, we feed all unprocessed but already received data to the
|
|
|
|
nghttp2 session object using the `nghttp2_session_mem_recv()` function. The
|
|
|
|
`nghttp2_session_mem_recv()` function processes the data and may invoke the
|
|
|
|
nghttp2 callbacks and also queue outgoing frames. Since there may be pending
|
|
|
|
outgoing frames, we call ``session_send()`` function to send off those
|
|
|
|
frames. The ``session_send()`` function is defined as follows::
|
2013-12-26 15:37:42 +01:00
|
|
|
|
|
|
|
static int session_send(http2_session_data *session_data)
|
|
|
|
{
|
|
|
|
int rv;
|
|
|
|
rv = nghttp2_session_send(session_data->session);
|
|
|
|
if(rv != 0) {
|
|
|
|
warnx("Fatal error: %s", nghttp2_strerror(rv));
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
The `nghttp2_session_send()` function serializes the frame into wire
|
2014-08-01 10:09:17 +02:00
|
|
|
format and calls :member:`nghttp2_session_callbacks.send_callback` with
|
|
|
|
it. We set the ``send_callback()`` function to
|
2013-12-26 15:37:42 +01:00
|
|
|
:member:`nghttp2_session_callbacks.send_callback` in
|
|
|
|
``initialize_nghttp2_session()`` function described earlier. It is
|
|
|
|
defined as follows::
|
|
|
|
|
|
|
|
static ssize_t send_callback(nghttp2_session *session,
|
|
|
|
const uint8_t *data, size_t length,
|
|
|
|
int flags, void *user_data)
|
|
|
|
{
|
|
|
|
http2_session_data *session_data = (http2_session_data*)user_data;
|
|
|
|
struct bufferevent *bev = session_data->bev;
|
|
|
|
/* Avoid excessive buffering in server side. */
|
|
|
|
if(evbuffer_get_length(bufferevent_get_output(session_data->bev)) >=
|
|
|
|
OUTPUT_WOULDBLOCK_THRESHOLD) {
|
|
|
|
return NGHTTP2_ERR_WOULDBLOCK;
|
|
|
|
}
|
|
|
|
bufferevent_write(bev, data, length);
|
|
|
|
return length;
|
|
|
|
}
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
Since we use bufferevent to abstract network I/O, we just write the data to
|
|
|
|
the bufferevent object. Note that `nghttp2_session_send()` continues to write
|
|
|
|
all frames queued so far. If we were writing the data to a non-blocking socket
|
|
|
|
directly using ``write()`` system call in the
|
|
|
|
:member:`nghttp2_session_callbacks.send_callback`, we would surely get
|
|
|
|
``EAGAIN`` or ``EWOULDBLOCK`` back since the socket has limited send
|
|
|
|
buffer. If that happens, we can return :macro:`NGHTTP2_ERR_WOULDBLOCK` to
|
|
|
|
signal the nghttp2 library to stop sending further data. But when writing to
|
|
|
|
the bufferevent, we have to regulate the amount data to get buffered ourselves
|
|
|
|
to avoid using huge amounts of memory. To achieve this, we check the size of
|
|
|
|
the output buffer and if it reaches more than or equal to
|
2013-12-26 15:37:42 +01:00
|
|
|
``OUTPUT_WOULDBLOCK_THRESHOLD`` bytes, we stop writing data and return
|
|
|
|
:macro:`NGHTTP2_ERR_WOULDBLOCK` to tell the library to stop calling
|
|
|
|
send_callback.
|
|
|
|
|
|
|
|
The next bufferevent callback is ``readcb()``, which is invoked when
|
|
|
|
data is available to read in the bufferevent input buffer::
|
|
|
|
|
|
|
|
static void readcb(struct bufferevent *bev, void *ptr)
|
|
|
|
{
|
|
|
|
http2_session_data *session_data = (http2_session_data*)ptr;
|
|
|
|
if(session_recv(session_data) != 0) {
|
|
|
|
delete_http2_session_data(session_data);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
In this function, we just call ``session_recv()`` to process incoming
|
|
|
|
data.
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
The third bufferevent callback is ``writecb()``, which is invoked when all
|
|
|
|
data in the bufferevent output buffer has been sent::
|
2013-12-26 15:37:42 +01:00
|
|
|
|
|
|
|
static void writecb(struct bufferevent *bev, void *ptr)
|
|
|
|
{
|
|
|
|
http2_session_data *session_data = (http2_session_data*)ptr;
|
|
|
|
if(evbuffer_get_length(bufferevent_get_output(bev)) > 0) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
if(nghttp2_session_want_read(session_data->session) == 0 &&
|
|
|
|
nghttp2_session_want_write(session_data->session) == 0) {
|
|
|
|
delete_http2_session_data(session_data);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
if(session_send(session_data) != 0) {
|
|
|
|
delete_http2_session_data(session_data);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
First we check whether we should drop the connection or not. The nghttp2
|
|
|
|
session object keeps track of reception and transmission of GOAWAY frames and
|
|
|
|
other error conditions as well. Using this information, the nghttp2 session
|
|
|
|
object will tell whether the connection should be dropped or not. More
|
|
|
|
specifically, if both `nghttp2_session_want_read()` and
|
|
|
|
`nghttp2_session_want_write()` return 0, we have no business left in the
|
|
|
|
connection. But since we are using bufferevent and its deferred callback
|
|
|
|
option, the bufferevent output buffer may contain pending data when the
|
|
|
|
``writecb()`` is called. To handle this, we check whether the output buffer is
|
|
|
|
empty or not. If all these conditions are met, we drop connection.
|
|
|
|
|
|
|
|
Otherwise, we call ``session_send()`` to process the pending output
|
|
|
|
data. Remember that in ``send_callback()``, we must not write all data to
|
|
|
|
bufferevent to avoid excessive buffering. We continue processing pending data
|
|
|
|
when the output buffer becomes empty.
|
|
|
|
|
|
|
|
We have already described the nghttp2 callback ``send_callback()``. Let's
|
|
|
|
learn about the remaining nghttp2 callbacks we setup in
|
2013-12-26 15:37:42 +01:00
|
|
|
``initialize_nghttp2_setup()`` function.
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
The ``on_begin_headers_callback()`` function is invoked when the reception of
|
|
|
|
a header block in HEADERS or PUSH_PROMISE frame is started::
|
2013-12-26 15:37:42 +01:00
|
|
|
|
2014-01-29 13:49:50 +01:00
|
|
|
static int on_begin_headers_callback(nghttp2_session *session,
|
|
|
|
const nghttp2_frame *frame,
|
|
|
|
void *user_data)
|
2013-12-26 15:37:42 +01:00
|
|
|
{
|
|
|
|
http2_session_data *session_data = (http2_session_data*)user_data;
|
|
|
|
http2_stream_data *stream_data;
|
2014-01-29 13:49:50 +01:00
|
|
|
|
|
|
|
if(frame->hd.type != NGHTTP2_HEADERS ||
|
|
|
|
frame->headers.cat != NGHTTP2_HCAT_REQUEST) {
|
|
|
|
return 0;
|
2013-12-26 15:37:42 +01:00
|
|
|
}
|
2014-01-29 13:49:50 +01:00
|
|
|
stream_data = create_http2_stream_data(session_data, frame->hd.stream_id);
|
|
|
|
nghttp2_session_set_stream_user_data(session, frame->hd.stream_id,
|
|
|
|
stream_data);
|
2013-12-26 15:37:42 +01:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
We are only interested in the HEADERS frame in this function. Since the
|
|
|
|
HEADERS frame has several roles in the HTTP/2 protocol, we check that it is a
|
|
|
|
request HEADERS, which opens new stream. If the frame is a request HEADERS, we
|
|
|
|
create a ``http2_stream_data`` object to store the stream related data. We
|
|
|
|
associate the created ``http2_stream_data`` object with the stream in the
|
|
|
|
nghttp2 session object using `nghttp2_set_stream_user_data()` to get the
|
|
|
|
object without searching through the doubly linked list.
|
|
|
|
|
|
|
|
In this example server, we want to serve files relative to the current working
|
|
|
|
directory in which the program was invoked. Each header name/value pair is
|
|
|
|
emitted via ``on_header_callback`` function, which is called after
|
2014-01-29 13:49:50 +01:00
|
|
|
``on_begin_headers_callback()``::
|
2014-01-17 16:06:24 +01:00
|
|
|
|
|
|
|
static int on_header_callback(nghttp2_session *session,
|
|
|
|
const nghttp2_frame *frame,
|
|
|
|
const uint8_t *name, size_t namelen,
|
|
|
|
const uint8_t *value, size_t valuelen,
|
|
|
|
void *user_data)
|
|
|
|
{
|
|
|
|
http2_stream_data *stream_data;
|
|
|
|
const char PATH[] = ":path";
|
|
|
|
switch(frame->hd.type) {
|
|
|
|
case NGHTTP2_HEADERS:
|
|
|
|
if(frame->headers.cat != NGHTTP2_HCAT_REQUEST) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
stream_data = nghttp2_session_get_stream_user_data(session,
|
|
|
|
frame->hd.stream_id);
|
2014-02-20 13:30:05 +01:00
|
|
|
if(!stream_data || stream_data->request_path) {
|
2014-01-17 16:06:24 +01:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
if(namelen == sizeof(PATH) - 1 && memcmp(PATH, name, namelen) == 0) {
|
|
|
|
size_t j;
|
|
|
|
for(j = 0; j < valuelen && value[j] != '?'; ++j);
|
|
|
|
stream_data->request_path = percent_decode(value, j);
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
We search for the ``:path`` header field among the request headers and store
|
|
|
|
the requested path in the ``http2_stream_data`` object. In this example
|
|
|
|
program, we ignore ``:method`` header field and always treat the request as a
|
|
|
|
GET request.
|
2013-12-26 15:37:42 +01:00
|
|
|
|
2014-01-29 13:49:50 +01:00
|
|
|
The ``on_frame_recv_callback()`` function is invoked when a frame is
|
|
|
|
fully received::
|
2013-12-26 15:37:42 +01:00
|
|
|
|
2014-01-29 13:49:50 +01:00
|
|
|
static int on_frame_recv_callback(nghttp2_session *session,
|
|
|
|
const nghttp2_frame *frame, void *user_data)
|
2013-12-26 15:37:42 +01:00
|
|
|
{
|
|
|
|
http2_session_data *session_data = (http2_session_data*)user_data;
|
|
|
|
http2_stream_data *stream_data;
|
2014-01-29 13:49:50 +01:00
|
|
|
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) {
|
|
|
|
stream_data = nghttp2_session_get_stream_user_data(session,
|
|
|
|
frame->hd.stream_id);
|
|
|
|
/* For DATA and HEADERS frame, this callback may be called after
|
|
|
|
on_stream_close_callback. Check that stream still alive. */
|
|
|
|
if(!stream_data) {
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
return on_request_recv(session, session_data, stream_data);
|
2013-12-26 15:37:42 +01:00
|
|
|
}
|
2014-01-29 13:49:50 +01:00
|
|
|
break;
|
|
|
|
default:
|
|
|
|
break;
|
2013-12-26 15:37:42 +01:00
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
First we retrieve the ``http2_stream_data`` object associated with the stream
|
|
|
|
in ``on_begin_headers_callback()``. It is done using
|
|
|
|
`nghttp2_session_get_stream_user_data()`. If the requested path cannot be
|
|
|
|
served for some reason (e.g., file is not found), we send a 404 response,
|
|
|
|
which is done in ``error_reply()``. Otherwise, we open the requested file and
|
|
|
|
send its content. We send the header field ``:status`` as a single response
|
|
|
|
header.
|
2013-12-26 15:37:42 +01:00
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
Sending the content of the file is done in ``send_response()`` function::
|
2013-12-26 15:37:42 +01:00
|
|
|
|
|
|
|
static int send_response(nghttp2_session *session, int32_t stream_id,
|
|
|
|
nghttp2_nv *nva, size_t nvlen, int fd)
|
|
|
|
{
|
|
|
|
int rv;
|
|
|
|
nghttp2_data_provider data_prd;
|
|
|
|
data_prd.source.fd = fd;
|
|
|
|
data_prd.read_callback = file_read_callback;
|
|
|
|
|
|
|
|
rv = nghttp2_submit_response(session, stream_id, nva, nvlen, &data_prd);
|
|
|
|
if(rv != 0) {
|
|
|
|
warnx("Fatal error: %s", nghttp2_strerror(rv));
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
The nghttp2 library uses the :type:`nghttp2_data_provider` structure to
|
2013-12-26 15:37:42 +01:00
|
|
|
send entity body to the remote peer. The ``source`` member of this
|
|
|
|
structure is a union and it can be either void pointer or int which is
|
|
|
|
intended to be used as file descriptor. In this example server, we use
|
2014-08-01 10:09:17 +02:00
|
|
|
the file descriptor. We also set the ``file_read_callback()`` callback
|
|
|
|
function to read the contents of the file::
|
2013-12-26 15:37:42 +01:00
|
|
|
|
|
|
|
static ssize_t file_read_callback
|
|
|
|
(nghttp2_session *session, int32_t stream_id,
|
2014-04-05 11:04:21 +02:00
|
|
|
uint8_t *buf, size_t length, uint32_t *data_flags,
|
2013-12-26 15:37:42 +01:00
|
|
|
nghttp2_data_source *source, void *user_data)
|
|
|
|
{
|
|
|
|
int fd = source->fd;
|
|
|
|
ssize_t r;
|
|
|
|
while((r = read(fd, buf, length)) == -1 && errno == EINTR);
|
|
|
|
if(r == -1) {
|
|
|
|
return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE;
|
|
|
|
}
|
|
|
|
if(r == 0) {
|
2014-04-05 11:04:21 +02:00
|
|
|
*data_flags |= NGHTTP2_DATA_FLAG_EOF;
|
2013-12-26 15:37:42 +01:00
|
|
|
}
|
|
|
|
return r;
|
|
|
|
}
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
If an error happens while reading the file, we return
|
2014-04-05 11:04:21 +02:00
|
|
|
:macro:`NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE`. This tells the
|
2014-08-01 10:09:17 +02:00
|
|
|
library to send RST_STREAM to the stream. When all data has been read, set
|
|
|
|
the :macro:`NGHTTP2_DATA_FLAG_EOF` flag to ``*data_flags`` to tell the
|
|
|
|
nghttp2 library that we have finished reading the file.
|
2013-12-26 15:37:42 +01:00
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
The `nghttp2_submit_response()` function is used to send the response to the
|
|
|
|
remote peer.
|
2013-12-26 15:37:42 +01:00
|
|
|
|
|
|
|
The ``on_stream_close_callback()`` function is invoked when the stream
|
|
|
|
is about to close::
|
|
|
|
|
|
|
|
static int on_stream_close_callback(nghttp2_session *session,
|
|
|
|
int32_t stream_id,
|
|
|
|
nghttp2_error_code error_code,
|
|
|
|
void *user_data)
|
|
|
|
{
|
|
|
|
http2_session_data *session_data = (http2_session_data*)user_data;
|
|
|
|
http2_stream_data *stream_data;
|
|
|
|
|
|
|
|
stream_data = nghttp2_session_get_stream_user_data(session, stream_id);
|
2014-02-20 13:30:05 +01:00
|
|
|
if(!stream_data) {
|
|
|
|
return 0;
|
|
|
|
}
|
2013-12-26 15:37:42 +01:00
|
|
|
remove_stream(session_data, stream_data);
|
|
|
|
delete_http2_stream_data(stream_data);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2014-08-01 10:09:17 +02:00
|
|
|
We destroy the ``http2_stream_data`` object in this function since the stream
|
|
|
|
is about to close and we no longer use that object.
|