/* * nghttp2 - HTTP/2 C Library * * Copyright (c) 2013 Tatsuhiro Tsujikawa * * Permission is hereby granted, free of charge, to any person obtaining * a copy of this software and associated documentation files (the * "Software"), to deal in the Software without restriction, including * without limitation the rights to use, copy, modify, merge, publish, * distribute, sublicense, and/or sell copies of the Software, and to * permit persons to whom the Software is furnished to do so, subject to * the following conditions: * * The above copyright notice and this permission notice shall be * included in all copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */ #include "nghttp2_session_test.h" #include #include #include #include "nghttp2_session.h" #include "nghttp2_stream.h" #include "nghttp2_net.h" #include "nghttp2_helper.h" #include "nghttp2_test_helper.h" #include "nghttp2_priority_spec.h" #define OB_CTRL(ITEM) nghttp2_outbound_item_get_ctrl_frame(ITEM) #define OB_CTRL_TYPE(ITEM) nghttp2_outbound_item_get_ctrl_frame_type(ITEM) #define OB_DATA(ITEM) nghttp2_outbound_item_get_data_frame(ITEM) typedef struct { uint8_t buf[65535]; size_t length; } accumulator; typedef struct { uint8_t data[8192]; uint8_t *datamark; uint8_t *datalimit; size_t feedseq[8192]; size_t seqidx; } scripted_data_feed; typedef struct { accumulator *acc; scripted_data_feed *df; int frame_recv_cb_called, invalid_frame_recv_cb_called; uint8_t recv_frame_type; int frame_send_cb_called; uint8_t sent_frame_type; int frame_not_send_cb_called; uint8_t not_sent_frame_type; int not_sent_error; int stream_close_cb_called; uint32_t stream_close_error_code; size_t data_source_length; int32_t stream_id; size_t block_count; int data_chunk_recv_cb_called; const nghttp2_frame *frame; size_t fixed_sendlen; int header_cb_called; int begin_headers_cb_called; nghttp2_nv nv; size_t data_chunk_len; size_t padlen; int begin_frame_cb_called; } my_user_data; static const nghttp2_nv reqnv[] = { MAKE_NV(":method", "GET"), MAKE_NV(":path", "/"), MAKE_NV(":scheme", "https"), MAKE_NV(":authority", "localhost"), }; static const nghttp2_nv resnv[] = { MAKE_NV(":status", "200"), }; static const nghttp2_nv trailernv[] = { // from http://tools.ietf.org/html/rfc6249#section-7 MAKE_NV("digest", "SHA-256=" "MWVkMWQxYTRiMzk5MDQ0MzI3NGU5NDEyZTk5OWY1ZGFmNzgyZTJlODYz" "YjRjYzFhOTlmNTQwYzI2M2QwM2U2MQ=="), }; static void scripted_data_feed_init2(scripted_data_feed *df, nghttp2_bufs *bufs) { nghttp2_buf_chain *ci; nghttp2_buf *buf; uint8_t *ptr; size_t len; memset(df, 0, sizeof(scripted_data_feed)); ptr = df->data; len = 0; for (ci = bufs->head; ci; ci = ci->next) { buf = &ci->buf; ptr = nghttp2_cpymem(ptr, buf->pos, nghttp2_buf_len(buf)); len += nghttp2_buf_len(buf); } df->datamark = df->data; df->datalimit = df->data + len; df->feedseq[0] = len; } static ssize_t null_send_callback(nghttp2_session *session _U_, const uint8_t *data _U_, size_t len, int flags _U_, void *user_data _U_) { return (ssize_t)len; } static ssize_t fail_send_callback(nghttp2_session *session _U_, const uint8_t *data _U_, size_t len _U_, int flags _U_, void *user_data _U_) { return NGHTTP2_ERR_CALLBACK_FAILURE; } static ssize_t fixed_bytes_send_callback(nghttp2_session *session _U_, const uint8_t *data _U_, size_t len, int flags _U_, void *user_data) { size_t fixed_sendlen = ((my_user_data *)user_data)->fixed_sendlen; return (ssize_t)(fixed_sendlen < len ? fixed_sendlen : len); } static ssize_t scripted_recv_callback(nghttp2_session *session _U_, uint8_t *data, size_t len, int flags _U_, void *user_data) { scripted_data_feed *df = ((my_user_data *)user_data)->df; size_t wlen = df->feedseq[df->seqidx] > len ? len : df->feedseq[df->seqidx]; memcpy(data, df->datamark, wlen); df->datamark += wlen; df->feedseq[df->seqidx] -= wlen; if (df->feedseq[df->seqidx] == 0) { ++df->seqidx; } return (ssize_t)wlen; } static ssize_t eof_recv_callback(nghttp2_session *session _U_, uint8_t *data _U_, size_t len _U_, int flags _U_, void *user_data _U_) { return NGHTTP2_ERR_EOF; } static ssize_t accumulator_send_callback(nghttp2_session *session _U_, const uint8_t *buf, size_t len, int flags _U_, void *user_data) { accumulator *acc = ((my_user_data *)user_data)->acc; assert(acc->length + len < sizeof(acc->buf)); memcpy(acc->buf + acc->length, buf, len); acc->length += len; return (ssize_t)len; } static int on_begin_frame_callback(nghttp2_session *session _U_, const nghttp2_frame_hd *hd _U_, void *user_data) { my_user_data *ud = (my_user_data *)user_data; ++ud->begin_frame_cb_called; return 0; } static int on_frame_recv_callback(nghttp2_session *session _U_, const nghttp2_frame *frame, void *user_data) { my_user_data *ud = (my_user_data *)user_data; ++ud->frame_recv_cb_called; ud->recv_frame_type = frame->hd.type; return 0; } static int on_invalid_frame_recv_callback(nghttp2_session *session _U_, const nghttp2_frame *frame _U_, int lib_error_code _U_, void *user_data) { my_user_data *ud = (my_user_data *)user_data; ++ud->invalid_frame_recv_cb_called; return 0; } static int on_frame_send_callback(nghttp2_session *session _U_, const nghttp2_frame *frame, void *user_data) { my_user_data *ud = (my_user_data *)user_data; ++ud->frame_send_cb_called; ud->sent_frame_type = frame->hd.type; return 0; } static int on_frame_not_send_callback(nghttp2_session *session _U_, const nghttp2_frame *frame, int lib_error, void *user_data) { my_user_data *ud = (my_user_data *)user_data; ++ud->frame_not_send_cb_called; ud->not_sent_frame_type = frame->hd.type; ud->not_sent_error = lib_error; return 0; } static int on_data_chunk_recv_callback(nghttp2_session *session _U_, uint8_t flags _U_, int32_t stream_id _U_, const uint8_t *data _U_, size_t len, void *user_data) { my_user_data *ud = (my_user_data *)user_data; ++ud->data_chunk_recv_cb_called; ud->data_chunk_len = len; return 0; } static int pause_on_data_chunk_recv_callback(nghttp2_session *session _U_, uint8_t flags _U_, int32_t stream_id _U_, const uint8_t *data _U_, size_t len _U_, void *user_data) { my_user_data *ud = (my_user_data *)user_data; ++ud->data_chunk_recv_cb_called; return NGHTTP2_ERR_PAUSE; } static ssize_t select_padding_callback(nghttp2_session *session _U_, const nghttp2_frame *frame, size_t max_payloadlen, void *user_data) { my_user_data *ud = (my_user_data *)user_data; return (ssize_t)nghttp2_min(max_payloadlen, frame->hd.length + ud->padlen); } static ssize_t too_large_data_source_length_callback( nghttp2_session *session _U_, uint8_t frame_type _U_, int32_t stream_id _U_, int32_t session_remote_window_size _U_, int32_t stream_remote_window_size _U_, uint32_t remote_max_frame_size _U_, void *user_data _U_) { return NGHTTP2_MAX_FRAME_SIZE_MAX + 1; } static ssize_t smallest_length_data_source_length_callback( nghttp2_session *session _U_, uint8_t frame_type _U_, int32_t stream_id _U_, int32_t session_remote_window_size _U_, int32_t stream_remote_window_size _U_, uint32_t remote_max_frame_size _U_, void *user_data _U_) { return 1; } static ssize_t fixed_length_data_source_read_callback( nghttp2_session *session _U_, int32_t stream_id _U_, uint8_t *buf _U_, size_t len, uint32_t *data_flags, nghttp2_data_source *source _U_, void *user_data) { my_user_data *ud = (my_user_data *)user_data; size_t wlen; if (len < ud->data_source_length) { wlen = len; } else { wlen = ud->data_source_length; } ud->data_source_length -= wlen; if (ud->data_source_length == 0) { *data_flags |= NGHTTP2_DATA_FLAG_EOF; } return (ssize_t)wlen; } static ssize_t temporal_failure_data_source_read_callback( nghttp2_session *session _U_, int32_t stream_id _U_, uint8_t *buf _U_, size_t len _U_, uint32_t *data_flags _U_, nghttp2_data_source *source _U_, void *user_data _U_) { return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE; } static ssize_t fail_data_source_read_callback(nghttp2_session *session _U_, int32_t stream_id _U_, uint8_t *buf _U_, size_t len _U_, uint32_t *data_flags _U_, nghttp2_data_source *source _U_, void *user_data _U_) { return NGHTTP2_ERR_CALLBACK_FAILURE; } static ssize_t no_end_stream_data_source_read_callback( nghttp2_session *session _U_, int32_t stream_id _U_, uint8_t *buf _U_, size_t len _U_, uint32_t *data_flags, nghttp2_data_source *source _U_, void *user_data _U_) { *data_flags |= NGHTTP2_DATA_FLAG_EOF | NGHTTP2_DATA_FLAG_NO_END_STREAM; return 0; } static ssize_t no_copy_data_source_read_callback( nghttp2_session *session _U_, int32_t stream_id _U_, uint8_t *buf _U_, size_t len, uint32_t *data_flags, nghttp2_data_source *source _U_, void *user_data) { my_user_data *ud = (my_user_data *)user_data; size_t wlen; if (len < ud->data_source_length) { wlen = len; } else { wlen = ud->data_source_length; } ud->data_source_length -= wlen; *data_flags |= NGHTTP2_DATA_FLAG_NO_COPY; if (ud->data_source_length == 0) { *data_flags |= NGHTTP2_DATA_FLAG_EOF; } return (ssize_t)wlen; } static int send_data_callback(nghttp2_session *session _U_, nghttp2_frame *frame, const uint8_t *framehd, size_t length, nghttp2_data_source *source _U_, void *user_data) { accumulator *acc = ((my_user_data *)user_data)->acc; memcpy(acc->buf + acc->length, framehd, NGHTTP2_FRAME_HDLEN); acc->length += NGHTTP2_FRAME_HDLEN; if (frame->data.padlen) { *(acc->buf + acc->length++) = (uint8_t)(frame->data.padlen - 1); } acc->length += length; if (frame->data.padlen) { acc->length += frame->data.padlen - 1; } return 0; } /* static void no_stream_user_data_stream_close_callback */ /* (nghttp2_session *session, */ /* int32_t stream_id, */ /* nghttp2_error_code error_code, */ /* void *user_data) */ /* { */ /* my_user_data* my_data = (my_user_data*)user_data; */ /* ++my_data->stream_close_cb_called; */ /* } */ static ssize_t block_count_send_callback(nghttp2_session *session _U_, const uint8_t *data _U_, size_t len, int flags _U_, void *user_data) { my_user_data *ud = (my_user_data *)user_data; if (ud->block_count == 0) { return NGHTTP2_ERR_WOULDBLOCK; } --ud->block_count; return (ssize_t)len; } static int on_header_callback(nghttp2_session *session _U_, const nghttp2_frame *frame, const uint8_t *name, size_t namelen, const uint8_t *value, size_t valuelen, uint8_t flags _U_, void *user_data) { my_user_data *ud = (my_user_data *)user_data; ++ud->header_cb_called; ud->nv.name = (uint8_t *)name; ud->nv.namelen = namelen; ud->nv.value = (uint8_t *)value; ud->nv.valuelen = valuelen; ud->frame = frame; return 0; } static int pause_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) { on_header_callback(session, frame, name, namelen, value, valuelen, flags, user_data); return NGHTTP2_ERR_PAUSE; } static int temporal_failure_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) { on_header_callback(session, frame, name, namelen, value, valuelen, flags, user_data); return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE; } static int on_begin_headers_callback(nghttp2_session *session _U_, const nghttp2_frame *frame _U_, void *user_data) { my_user_data *ud = (my_user_data *)user_data; ++ud->begin_headers_cb_called; return 0; } static int temporal_failure_on_begin_headers_callback( nghttp2_session *session, const nghttp2_frame *frame, void *user_data) { on_begin_headers_callback(session, frame, user_data); return NGHTTP2_ERR_TEMPORAL_CALLBACK_FAILURE; } static ssize_t defer_data_source_read_callback(nghttp2_session *session _U_, int32_t stream_id _U_, uint8_t *buf _U_, size_t len _U_, uint32_t *data_flags _U_, nghttp2_data_source *source _U_, void *user_data _U_) { return NGHTTP2_ERR_DEFERRED; } static int on_stream_close_callback(nghttp2_session *session _U_, int32_t stream_id _U_, nghttp2_error_code error_code _U_, void *user_data) { my_user_data *my_data = (my_user_data *)user_data; ++my_data->stream_close_cb_called; my_data->stream_close_error_code = error_code; return 0; } static nghttp2_settings_entry *dup_iv(const nghttp2_settings_entry *iv, size_t niv) { return nghttp2_frame_iv_copy(iv, niv, nghttp2_mem_default()); } static nghttp2_priority_spec pri_spec_default = {0, NGHTTP2_DEFAULT_WEIGHT, 0}; void test_nghttp2_session_recv(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; scripted_data_feed df; my_user_data user_data; nghttp2_bufs bufs; size_t framelen; nghttp2_frame frame; size_t i; nghttp2_outbound_item *item; nghttp2_nv *nva; size_t nvlen; nghttp2_hd_deflater deflater; int rv; nghttp2_mem *mem; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; callbacks.recv_callback = scripted_recv_callback; callbacks.on_frame_recv_callback = on_frame_recv_callback; callbacks.on_begin_frame_callback = on_begin_frame_callback; user_data.df = &df; nghttp2_session_server_new(&session, &callbacks, &user_data); nghttp2_hd_deflate_init(&deflater, mem); nvlen = ARRLEN(reqnv); nghttp2_nv_array_copy(&nva, reqnv, nvlen, mem); nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS, 1, NGHTTP2_HCAT_HEADERS, NULL, nva, nvlen); rv = nghttp2_frame_pack_headers(&bufs, &frame.headers, &deflater); CU_ASSERT(0 == rv); scripted_data_feed_init2(&df, &bufs); framelen = nghttp2_bufs_len(&bufs); /* Send 1 byte per each read */ for (i = 0; i < framelen; ++i) { df.feedseq[i] = 1; } nghttp2_frame_headers_free(&frame.headers, mem); user_data.frame_recv_cb_called = 0; user_data.begin_frame_cb_called = 0; while (df.seqidx < framelen) { CU_ASSERT(0 == nghttp2_session_recv(session)); } CU_ASSERT(1 == user_data.frame_recv_cb_called); CU_ASSERT(1 == user_data.begin_frame_cb_called); nghttp2_bufs_reset(&bufs); /* Receive PRIORITY */ nghttp2_frame_priority_init(&frame.priority, 5, &pri_spec_default); rv = nghttp2_frame_pack_priority(&bufs, &frame.priority); CU_ASSERT(0 == rv); nghttp2_frame_priority_free(&frame.priority); scripted_data_feed_init2(&df, &bufs); user_data.frame_recv_cb_called = 0; user_data.begin_frame_cb_called = 0; CU_ASSERT(0 == nghttp2_session_recv(session)); CU_ASSERT(1 == user_data.frame_recv_cb_called); CU_ASSERT(1 == user_data.begin_frame_cb_called); nghttp2_bufs_reset(&bufs); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); /* Some tests for frame too large */ nghttp2_session_server_new(&session, &callbacks, &user_data); /* Receive PING with too large payload */ nghttp2_frame_ping_init(&frame.ping, NGHTTP2_FLAG_NONE, NULL); rv = nghttp2_frame_pack_ping(&bufs, &frame.ping); CU_ASSERT(0 == rv); /* Add extra 16 bytes */ nghttp2_bufs_seek_last_present(&bufs); assert(nghttp2_buf_len(&bufs.cur->buf) >= 16); bufs.cur->buf.last += 16; nghttp2_put_uint32be( bufs.cur->buf.pos, (uint32_t)(((frame.hd.length + 16) << 8) + bufs.cur->buf.pos[3])); nghttp2_frame_ping_free(&frame.ping); scripted_data_feed_init2(&df, &bufs); user_data.frame_recv_cb_called = 0; user_data.begin_frame_cb_called = 0; CU_ASSERT(0 == nghttp2_session_recv(session)); CU_ASSERT(0 == user_data.frame_recv_cb_called); CU_ASSERT(0 == user_data.begin_frame_cb_called); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_GOAWAY == item->frame.hd.type); CU_ASSERT(NGHTTP2_FRAME_SIZE_ERROR == item->frame.goaway.error_code); CU_ASSERT(0 == nghttp2_session_send(session)); nghttp2_bufs_free(&bufs); nghttp2_session_del(session); } void test_nghttp2_session_recv_invalid_stream_id(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; scripted_data_feed df; my_user_data user_data; nghttp2_bufs bufs; nghttp2_frame frame; nghttp2_hd_deflater deflater; int rv; nghttp2_mem *mem; nghttp2_nv *nva; size_t nvlen; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.recv_callback = scripted_recv_callback; callbacks.on_invalid_frame_recv_callback = on_invalid_frame_recv_callback; user_data.df = &df; user_data.invalid_frame_recv_cb_called = 0; nghttp2_session_server_new(&session, &callbacks, &user_data); nghttp2_hd_deflate_init(&deflater, mem); nvlen = ARRLEN(reqnv); nghttp2_nv_array_copy(&nva, reqnv, nvlen, mem); nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS, 2, NGHTTP2_HCAT_HEADERS, NULL, nva, nvlen); rv = nghttp2_frame_pack_headers(&bufs, &frame.headers, &deflater); CU_ASSERT(0 == rv); CU_ASSERT(nghttp2_bufs_len(&bufs) > 0); scripted_data_feed_init2(&df, &bufs); nghttp2_frame_headers_free(&frame.headers, mem); CU_ASSERT(0 == nghttp2_session_recv(session)); CU_ASSERT(1 == user_data.invalid_frame_recv_cb_called); nghttp2_bufs_free(&bufs); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); } void test_nghttp2_session_recv_invalid_frame(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; scripted_data_feed df; my_user_data user_data; nghttp2_bufs bufs; nghttp2_frame frame; nghttp2_nv *nva; size_t nvlen; nghttp2_hd_deflater deflater; int rv; nghttp2_mem *mem; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.recv_callback = scripted_recv_callback; callbacks.send_callback = null_send_callback; callbacks.on_frame_send_callback = on_frame_send_callback; user_data.df = &df; user_data.frame_send_cb_called = 0; nghttp2_session_server_new(&session, &callbacks, &user_data); nghttp2_hd_deflate_init(&deflater, mem); nvlen = ARRLEN(reqnv); nghttp2_nv_array_copy(&nva, reqnv, nvlen, mem); nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS, 1, NGHTTP2_HCAT_HEADERS, NULL, nva, nvlen); rv = nghttp2_frame_pack_headers(&bufs, &frame.headers, &deflater); CU_ASSERT(0 == rv); CU_ASSERT(nghttp2_bufs_len(&bufs) > 0); scripted_data_feed_init2(&df, &bufs); CU_ASSERT(0 == nghttp2_session_recv(session)); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(0 == user_data.frame_send_cb_called); /* Receive exactly same bytes of HEADERS is treated as error, because it has * pseudo headers and without END_STREAM flag set */ scripted_data_feed_init2(&df, &bufs); CU_ASSERT(0 == nghttp2_session_recv(session)); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(1 == user_data.frame_send_cb_called); CU_ASSERT(NGHTTP2_RST_STREAM == user_data.sent_frame_type); nghttp2_bufs_free(&bufs); nghttp2_frame_headers_free(&frame.headers, mem); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); } void test_nghttp2_session_recv_eof(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; callbacks.recv_callback = eof_recv_callback; nghttp2_session_client_new(&session, &callbacks, NULL); CU_ASSERT(NGHTTP2_ERR_EOF == nghttp2_session_recv(session)); nghttp2_session_del(session); } void test_nghttp2_session_recv_data(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data ud; uint8_t data[8092]; ssize_t rv; nghttp2_outbound_item *item; nghttp2_stream *stream; nghttp2_frame_hd hd; int i; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; callbacks.on_data_chunk_recv_callback = on_data_chunk_recv_callback; callbacks.on_frame_recv_callback = on_frame_recv_callback; callbacks.on_frame_send_callback = on_frame_send_callback; nghttp2_session_client_new(&session, &callbacks, &ud); /* Create DATA frame with length 4KiB */ memset(data, 0, sizeof(data)); hd.length = 4096; hd.type = NGHTTP2_DATA; hd.flags = NGHTTP2_FLAG_NONE; hd.stream_id = 1; nghttp2_frame_pack_frame_hd(data, &hd); /* stream 1 is not opened, so it must be responded with connection error. This is not mandated by the spec */ ud.data_chunk_recv_cb_called = 0; ud.frame_recv_cb_called = 0; rv = nghttp2_session_mem_recv(session, data, NGHTTP2_FRAME_HDLEN + 4096); CU_ASSERT(NGHTTP2_FRAME_HDLEN + 4096 == rv); CU_ASSERT(0 == ud.data_chunk_recv_cb_called); CU_ASSERT(0 == ud.frame_recv_cb_called); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_GOAWAY == item->frame.hd.type); nghttp2_session_del(session); nghttp2_session_client_new(&session, &callbacks, &ud); /* Create stream 1 with CLOSING state. DATA is ignored. */ stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_CLOSING, NULL); /* Set initial window size 16383 to check stream flow control, isolating it from the conneciton flow control */ stream->local_window_size = 16383; ud.data_chunk_recv_cb_called = 0; ud.frame_recv_cb_called = 0; rv = nghttp2_session_mem_recv(session, data, NGHTTP2_FRAME_HDLEN + 4096); CU_ASSERT(NGHTTP2_FRAME_HDLEN + 4096 == rv); CU_ASSERT(0 == ud.data_chunk_recv_cb_called); CU_ASSERT(0 == ud.frame_recv_cb_called); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NULL == item); /* This is normal case. DATA is acceptable. */ stream->state = NGHTTP2_STREAM_OPENED; ud.data_chunk_recv_cb_called = 0; ud.frame_recv_cb_called = 0; rv = nghttp2_session_mem_recv(session, data, NGHTTP2_FRAME_HDLEN + 4096); CU_ASSERT(NGHTTP2_FRAME_HDLEN + 4096 == rv); CU_ASSERT(1 == ud.data_chunk_recv_cb_called); CU_ASSERT(1 == ud.frame_recv_cb_called); CU_ASSERT(NULL == nghttp2_session_get_next_ob_item(session)); ud.data_chunk_recv_cb_called = 0; ud.frame_recv_cb_called = 0; rv = nghttp2_session_mem_recv(session, data, NGHTTP2_FRAME_HDLEN + 4096); CU_ASSERT(NGHTTP2_FRAME_HDLEN + 4096 == rv); /* Now we got data more than initial-window-size / 2, WINDOW_UPDATE must be queued */ CU_ASSERT(1 == ud.data_chunk_recv_cb_called); CU_ASSERT(1 == ud.frame_recv_cb_called); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_WINDOW_UPDATE == item->frame.hd.type); CU_ASSERT(1 == item->frame.window_update.hd.stream_id); CU_ASSERT(0 == nghttp2_session_send(session)); /* Set initial window size to 1MiB, so that we can check connection flow control individually */ stream->local_window_size = 1 << 20; /* Connection flow control takes into account DATA which is received in the error condition. We have received 4096 * 4 bytes of DATA. Additional 4 DATA frames, connection flow control will kick in. */ for (i = 0; i < 5; ++i) { rv = nghttp2_session_mem_recv(session, data, NGHTTP2_FRAME_HDLEN + 4096); CU_ASSERT(NGHTTP2_FRAME_HDLEN + 4096 == rv); } item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_WINDOW_UPDATE == item->frame.hd.type); CU_ASSERT(0 == item->frame.window_update.hd.stream_id); CU_ASSERT(0 == nghttp2_session_send(session)); /* Reception of DATA with stream ID = 0 causes connection error */ hd.length = 4096; hd.type = NGHTTP2_DATA; hd.flags = NGHTTP2_FLAG_NONE; hd.stream_id = 0; nghttp2_frame_pack_frame_hd(data, &hd); ud.data_chunk_recv_cb_called = 0; ud.frame_recv_cb_called = 0; rv = nghttp2_session_mem_recv(session, data, NGHTTP2_FRAME_HDLEN + 4096); CU_ASSERT(NGHTTP2_FRAME_HDLEN + 4096 == rv); CU_ASSERT(0 == ud.data_chunk_recv_cb_called); CU_ASSERT(0 == ud.frame_recv_cb_called); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_GOAWAY == item->frame.hd.type); CU_ASSERT(NGHTTP2_PROTOCOL_ERROR == item->frame.goaway.error_code); nghttp2_session_del(session); /* Check window_update_queued flag in both session and stream */ nghttp2_session_server_new(&session, &callbacks, &ud); hd.length = 4096; hd.type = NGHTTP2_DATA; hd.flags = NGHTTP2_FLAG_NONE; hd.stream_id = 1; nghttp2_frame_pack_frame_hd(data, &hd); stream = open_stream(session, 1); /* Send 32767 bytes of DATA. In our current flow control algorithm, it triggers first WINDOW_UPDATE of window_size_increment 32767. */ for (i = 0; i < 7; ++i) { rv = nghttp2_session_mem_recv(session, data, NGHTTP2_FRAME_HDLEN + 4096); CU_ASSERT(NGHTTP2_FRAME_HDLEN + 4096 == rv); } hd.length = 4095; nghttp2_frame_pack_frame_hd(data, &hd); rv = nghttp2_session_mem_recv(session, data, NGHTTP2_FRAME_HDLEN + 4095); CU_ASSERT(NGHTTP2_FRAME_HDLEN + 4095 == rv); /* Now 2 WINDOW_UPDATEs for session and stream should be queued. */ CU_ASSERT(0 == stream->recv_window_size); CU_ASSERT(0 == session->recv_window_size); CU_ASSERT(1 == stream->window_update_queued); CU_ASSERT(1 == session->window_update_queued); /* Then send 32768 bytes of DATA. Since we have not sent queued WINDOW_UDPATE frame, recv_window_size should not be decreased */ hd.length = 4096; nghttp2_frame_pack_frame_hd(data, &hd); for (i = 0; i < 8; ++i) { rv = nghttp2_session_mem_recv(session, data, NGHTTP2_FRAME_HDLEN + 4096); CU_ASSERT(NGHTTP2_FRAME_HDLEN + 4096 == rv); } /* WINDOW_UPDATE is blocked for session and stream, so recv_window_size must not be decreased. */ CU_ASSERT(32768 == stream->recv_window_size); CU_ASSERT(32768 == session->recv_window_size); CU_ASSERT(1 == stream->window_update_queued); CU_ASSERT(1 == session->window_update_queued); ud.frame_send_cb_called = 0; /* This sends queued WINDOW_UPDATES. And then check recv_window_size, and queue WINDOW_UPDATEs for both session and stream, and send them at once. */ CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(4 == ud.frame_send_cb_called); CU_ASSERT(0 == stream->recv_window_size); CU_ASSERT(0 == session->recv_window_size); CU_ASSERT(0 == stream->window_update_queued); CU_ASSERT(0 == session->window_update_queued); nghttp2_session_del(session); } void test_nghttp2_session_recv_data_no_auto_flow_control(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data ud; nghttp2_option *option; nghttp2_frame_hd hd; size_t padlen; uint8_t data[8192]; ssize_t rv; size_t sendlen; nghttp2_stream *stream; size_t i; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; callbacks.on_frame_send_callback = on_frame_send_callback; nghttp2_option_new(&option); nghttp2_option_set_no_auto_window_update(option, 1); nghttp2_session_server_new2(&session, &callbacks, &ud, option); /* Create DATA frame with length 4KiB + 11 bytes padding*/ padlen = 11; memset(data, 0, sizeof(data)); hd.length = 4096 + 1 + padlen; hd.type = NGHTTP2_DATA; hd.flags = NGHTTP2_FLAG_PADDED; hd.stream_id = 1; nghttp2_frame_pack_frame_hd(data, &hd); data[NGHTTP2_FRAME_HDLEN] = (uint8_t)padlen; /* First create stream 1, then close it. Check that data is consumed for connection in this situation */ open_stream(session, 1); /* Receive first 100 bytes */ sendlen = 100; rv = nghttp2_session_mem_recv(session, data, sendlen); CU_ASSERT((ssize_t)sendlen == rv); /* We consumed pad length field (1 byte) */ CU_ASSERT(1 == session->consumed_size); /* close stream here */ nghttp2_submit_rst_stream(session, NGHTTP2_FLAG_NONE, 1, NGHTTP2_NO_ERROR); nghttp2_session_send(session); /* stream 1 has been closed, and we disabled auto flow-control, so data must be immediately consumed for connection. */ rv = nghttp2_session_mem_recv(session, data + sendlen, NGHTTP2_FRAME_HDLEN + hd.length - sendlen); CU_ASSERT((ssize_t)(NGHTTP2_FRAME_HDLEN + hd.length - sendlen) == rv); /* We already consumed pad length field (1 byte), so do +1 here */ CU_ASSERT((int32_t)(NGHTTP2_FRAME_HDLEN + hd.length - sendlen + 1) == session->consumed_size); nghttp2_session_del(session); /* Reuse DATA created previously. */ nghttp2_session_server_new2(&session, &callbacks, &ud, option); /* Now we are expecting final response header, which means receiving DATA for that stream is illegal. */ stream = open_stream(session, 1); stream->http_flags |= NGHTTP2_HTTP_FLAG_EXPECT_FINAL_RESPONSE; rv = nghttp2_session_mem_recv(session, data, NGHTTP2_FRAME_HDLEN + hd.length); CU_ASSERT((ssize_t)(NGHTTP2_FRAME_HDLEN + hd.length) == rv); /* Whole payload must be consumed now because HTTP messaging rule was not honored. */ CU_ASSERT((int32_t)hd.length == session->consumed_size); nghttp2_session_del(session); /* Check window_update_queued flag in both session and stream */ nghttp2_session_server_new2(&session, &callbacks, &ud, option); stream = open_stream(session, 1); hd.length = 4096; hd.type = NGHTTP2_DATA; hd.flags = NGHTTP2_FLAG_NONE; hd.stream_id = 1; nghttp2_frame_pack_frame_hd(data, &hd); /* Receive up to 65535 bytes of DATA */ for (i = 0; i < 15; ++i) { rv = nghttp2_session_mem_recv(session, data, NGHTTP2_FRAME_HDLEN + 4096); CU_ASSERT(NGHTTP2_FRAME_HDLEN + 4096 == rv); } hd.length = 4095; nghttp2_frame_pack_frame_hd(data, &hd); rv = nghttp2_session_mem_recv(session, data, NGHTTP2_FRAME_HDLEN + 4095); CU_ASSERT(NGHTTP2_FRAME_HDLEN + 4095 == rv); CU_ASSERT(65535 == session->recv_window_size); CU_ASSERT(65535 == stream->recv_window_size); /* The first call of nghttp2_session_consume_connection() will queue WINDOW_UPDATE. Next call does not. */ nghttp2_session_consume_connection(session, 32767); nghttp2_session_consume_connection(session, 32768); CU_ASSERT(32768 == session->recv_window_size); CU_ASSERT(65535 == stream->recv_window_size); CU_ASSERT(1 == session->window_update_queued); CU_ASSERT(0 == stream->window_update_queued); ud.frame_send_cb_called = 0; /* This will send WINDOW_UPDATE, and check whether we should send WINDOW_UPDATE, and queue and send it at once. */ CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(0 == session->recv_window_size); CU_ASSERT(65535 == stream->recv_window_size); CU_ASSERT(0 == session->window_update_queued); CU_ASSERT(0 == stream->window_update_queued); CU_ASSERT(2 == ud.frame_send_cb_called); /* Do the same for stream */ nghttp2_session_consume_stream(session, 1, 32767); nghttp2_session_consume_stream(session, 1, 32768); CU_ASSERT(0 == session->recv_window_size); CU_ASSERT(32768 == stream->recv_window_size); CU_ASSERT(0 == session->window_update_queued); CU_ASSERT(1 == stream->window_update_queued); ud.frame_send_cb_called = 0; CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(0 == session->recv_window_size); CU_ASSERT(0 == stream->recv_window_size); CU_ASSERT(0 == session->window_update_queued); CU_ASSERT(0 == stream->window_update_queued); CU_ASSERT(2 == ud.frame_send_cb_called); nghttp2_session_del(session); nghttp2_option_del(option); } void test_nghttp2_session_recv_continuation(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_nv *nva; size_t nvlen; nghttp2_frame frame; nghttp2_bufs bufs; nghttp2_buf *buf; ssize_t rv; my_user_data ud; nghttp2_hd_deflater deflater; uint8_t data[1024]; size_t datalen; nghttp2_frame_hd cont_hd; nghttp2_priority_spec pri_spec; nghttp2_mem *mem; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.on_header_callback = on_header_callback; callbacks.on_begin_headers_callback = on_begin_headers_callback; callbacks.on_begin_frame_callback = on_begin_frame_callback; nghttp2_session_server_new(&session, &callbacks, &ud); nghttp2_hd_deflate_init(&deflater, mem); /* Make 1 HEADERS and insert CONTINUATION header */ nvlen = ARRLEN(reqnv); nghttp2_nv_array_copy(&nva, reqnv, nvlen, mem); nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_NONE, 1, NGHTTP2_HCAT_HEADERS, NULL, nva, nvlen); rv = nghttp2_frame_pack_headers(&bufs, &frame.headers, &deflater); CU_ASSERT(0 == rv); CU_ASSERT(nghttp2_bufs_len(&bufs) > 0); /* make sure that all data is in the first buf */ buf = &bufs.head->buf; assert(nghttp2_bufs_len(&bufs) == nghttp2_buf_len(buf)); nghttp2_frame_headers_free(&frame.headers, mem); /* HEADERS's payload is 1 byte */ memcpy(data, buf->pos, NGHTTP2_FRAME_HDLEN + 1); datalen = NGHTTP2_FRAME_HDLEN + 1; buf->pos += NGHTTP2_FRAME_HDLEN + 1; nghttp2_put_uint32be(data, (uint32_t)((1 << 8) + data[3])); /* First CONTINUATION, 2 bytes */ nghttp2_frame_hd_init(&cont_hd, 2, NGHTTP2_CONTINUATION, NGHTTP2_FLAG_NONE, 1); nghttp2_frame_pack_frame_hd(data + datalen, &cont_hd); datalen += NGHTTP2_FRAME_HDLEN; memcpy(data + datalen, buf->pos, cont_hd.length); datalen += cont_hd.length; buf->pos += cont_hd.length; /* Second CONTINUATION, rest of the bytes */ nghttp2_frame_hd_init(&cont_hd, nghttp2_buf_len(buf), NGHTTP2_CONTINUATION, NGHTTP2_FLAG_END_HEADERS, 1); nghttp2_frame_pack_frame_hd(data + datalen, &cont_hd); datalen += NGHTTP2_FRAME_HDLEN; memcpy(data + datalen, buf->pos, cont_hd.length); datalen += cont_hd.length; buf->pos += cont_hd.length; CU_ASSERT(0 == nghttp2_buf_len(buf)); ud.header_cb_called = 0; ud.begin_frame_cb_called = 0; rv = nghttp2_session_mem_recv(session, data, datalen); CU_ASSERT((ssize_t)datalen == rv); CU_ASSERT(4 == ud.header_cb_called); CU_ASSERT(3 == ud.begin_frame_cb_called); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); /* HEADERS with padding followed by CONTINUATION */ nghttp2_session_server_new(&session, &callbacks, &ud); nghttp2_hd_deflate_init(&deflater, mem); nvlen = ARRLEN(reqnv); nghttp2_nv_array_copy(&nva, reqnv, nvlen, mem); nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_NONE, 1, NGHTTP2_HCAT_HEADERS, NULL, nva, nvlen); nghttp2_bufs_reset(&bufs); rv = nghttp2_frame_pack_headers(&bufs, &frame.headers, &deflater); CU_ASSERT(0 == rv); nghttp2_frame_headers_free(&frame.headers, mem); /* make sure that all data is in the first buf */ buf = &bufs.head->buf; assert(nghttp2_bufs_len(&bufs) == nghttp2_buf_len(buf)); /* HEADERS payload is 3 byte (1 for padding field, 1 for padding) */ memcpy(data, buf->pos, NGHTTP2_FRAME_HDLEN); nghttp2_put_uint32be(data, (uint32_t)((3 << 8) + data[3])); data[4] |= NGHTTP2_FLAG_PADDED; /* padding field */ data[NGHTTP2_FRAME_HDLEN] = 1; data[NGHTTP2_FRAME_HDLEN + 1] = buf->pos[NGHTTP2_FRAME_HDLEN]; /* padding */ data[NGHTTP2_FRAME_HDLEN + 2] = 0; datalen = NGHTTP2_FRAME_HDLEN + 3; buf->pos += NGHTTP2_FRAME_HDLEN + 1; /* CONTINUATION, rest of the bytes */ nghttp2_frame_hd_init(&cont_hd, nghttp2_buf_len(buf), NGHTTP2_CONTINUATION, NGHTTP2_FLAG_END_HEADERS, 1); nghttp2_frame_pack_frame_hd(data + datalen, &cont_hd); datalen += NGHTTP2_FRAME_HDLEN; memcpy(data + datalen, buf->pos, cont_hd.length); datalen += cont_hd.length; buf->pos += cont_hd.length; CU_ASSERT(0 == nghttp2_buf_len(buf)); ud.header_cb_called = 0; ud.begin_frame_cb_called = 0; rv = nghttp2_session_mem_recv(session, data, datalen); CU_ASSERT((ssize_t)datalen == rv); CU_ASSERT(4 == ud.header_cb_called); CU_ASSERT(2 == ud.begin_frame_cb_called); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); /* Expecting CONTINUATION, but get the other frame */ nghttp2_session_server_new(&session, &callbacks, &ud); nghttp2_hd_deflate_init(&deflater, mem); /* HEADERS without END_HEADERS flag */ nvlen = ARRLEN(reqnv); nghttp2_nv_array_copy(&nva, reqnv, nvlen, mem); nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_NONE, 1, NGHTTP2_HCAT_HEADERS, NULL, nva, nvlen); nghttp2_bufs_reset(&bufs); rv = nghttp2_frame_pack_headers(&bufs, &frame.headers, &deflater); CU_ASSERT(0 == rv); CU_ASSERT(nghttp2_bufs_len(&bufs) > 0); nghttp2_frame_headers_free(&frame.headers, mem); /* make sure that all data is in the first buf */ buf = &bufs.head->buf; assert(nghttp2_bufs_len(&bufs) == nghttp2_buf_len(buf)); memcpy(data, buf->pos, nghttp2_buf_len(buf)); datalen = nghttp2_buf_len(buf); /* Followed by PRIORITY */ nghttp2_priority_spec_default_init(&pri_spec); nghttp2_frame_priority_init(&frame.priority, 1, &pri_spec); nghttp2_bufs_reset(&bufs); rv = nghttp2_frame_pack_priority(&bufs, &frame.priority); CU_ASSERT(0 == rv); CU_ASSERT(nghttp2_bufs_len(&bufs) > 0); memcpy(data + datalen, buf->pos, nghttp2_buf_len(buf)); datalen += nghttp2_buf_len(buf); ud.begin_headers_cb_called = 0; rv = nghttp2_session_mem_recv(session, data, datalen); CU_ASSERT((ssize_t)datalen == rv); CU_ASSERT(1 == ud.begin_headers_cb_called); CU_ASSERT(NGHTTP2_GOAWAY == nghttp2_session_get_next_ob_item(session)->frame.hd.type); nghttp2_bufs_free(&bufs); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); } void test_nghttp2_session_recv_headers_with_priority(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_nv *nva; size_t nvlen; nghttp2_frame frame; nghttp2_bufs bufs; nghttp2_buf *buf; ssize_t rv; my_user_data ud; nghttp2_hd_deflater deflater; nghttp2_outbound_item *item; nghttp2_priority_spec pri_spec; nghttp2_stream *stream; nghttp2_mem *mem; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.on_frame_recv_callback = on_frame_recv_callback; nghttp2_session_server_new(&session, &callbacks, &ud); nghttp2_hd_deflate_init(&deflater, mem); open_stream(session, 1); /* With NGHTTP2_FLAG_PRIORITY without exclusive flag set */ nvlen = ARRLEN(reqnv); nghttp2_nv_array_copy(&nva, reqnv, nvlen, mem); nghttp2_priority_spec_init(&pri_spec, 1, 99, 0); nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS | NGHTTP2_FLAG_PRIORITY, 3, NGHTTP2_HCAT_HEADERS, &pri_spec, nva, nvlen); rv = nghttp2_frame_pack_headers(&bufs, &frame.headers, &deflater); CU_ASSERT(0 == rv); CU_ASSERT(nghttp2_bufs_len(&bufs) > 0); nghttp2_frame_headers_free(&frame.headers, mem); buf = &bufs.head->buf; assert(nghttp2_bufs_len(&bufs) == nghttp2_buf_len(buf)); ud.frame_recv_cb_called = 0; rv = nghttp2_session_mem_recv(session, buf->pos, nghttp2_buf_len(buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(buf) == rv); CU_ASSERT(1 == ud.frame_recv_cb_called); stream = nghttp2_session_get_stream(session, 3); CU_ASSERT(99 == stream->weight); CU_ASSERT(1 == stream->dep_prev->stream_id); nghttp2_bufs_reset(&bufs); /* With NGHTTP2_FLAG_PRIORITY, but cut last 1 byte to make it invalid. */ nvlen = ARRLEN(reqnv); nghttp2_nv_array_copy(&nva, reqnv, nvlen, mem); nghttp2_priority_spec_init(&pri_spec, 0, 99, 0); nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS | NGHTTP2_FLAG_PRIORITY, 5, NGHTTP2_HCAT_HEADERS, &pri_spec, nva, nvlen); rv = nghttp2_frame_pack_headers(&bufs, &frame.headers, &deflater); CU_ASSERT(0 == rv); CU_ASSERT(nghttp2_bufs_len(&bufs) > NGHTTP2_FRAME_HDLEN + 5); nghttp2_frame_headers_free(&frame.headers, mem); buf = &bufs.head->buf; /* Make payload shorter than required length to store priority group */ nghttp2_put_uint32be(buf->pos, (uint32_t)((4 << 8) + buf->pos[3])); ud.frame_recv_cb_called = 0; rv = nghttp2_session_mem_recv(session, buf->pos, nghttp2_buf_len(buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(buf) == rv); CU_ASSERT(0 == ud.frame_recv_cb_called); stream = nghttp2_session_get_stream(session, 5); CU_ASSERT(NULL == stream); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NULL != item); CU_ASSERT(NGHTTP2_GOAWAY == item->frame.hd.type); CU_ASSERT(NGHTTP2_FRAME_SIZE_ERROR == item->frame.goaway.error_code); nghttp2_bufs_reset(&bufs); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); /* Check dep_stream_id == stream_id */ nghttp2_session_server_new(&session, &callbacks, &ud); nghttp2_hd_deflate_init(&deflater, mem); nvlen = ARRLEN(reqnv); nghttp2_nv_array_copy(&nva, reqnv, nvlen, mem); nghttp2_priority_spec_init(&pri_spec, 1, 0, 0); nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS | NGHTTP2_FLAG_PRIORITY, 1, NGHTTP2_HCAT_HEADERS, &pri_spec, nva, nvlen); rv = nghttp2_frame_pack_headers(&bufs, &frame.headers, &deflater); CU_ASSERT(0 == rv); CU_ASSERT(nghttp2_bufs_len(&bufs) > 0); nghttp2_frame_headers_free(&frame.headers, mem); buf = &bufs.head->buf; assert(nghttp2_bufs_len(&bufs) == nghttp2_buf_len(buf)); ud.frame_recv_cb_called = 0; rv = nghttp2_session_mem_recv(session, buf->pos, nghttp2_buf_len(buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(buf) == rv); CU_ASSERT(0 == ud.frame_recv_cb_called); stream = nghttp2_session_get_stream(session, 1); CU_ASSERT(NULL == stream); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NULL != item); CU_ASSERT(NGHTTP2_GOAWAY == item->frame.hd.type); CU_ASSERT(NGHTTP2_PROTOCOL_ERROR == item->frame.goaway.error_code); nghttp2_bufs_reset(&bufs); nghttp2_bufs_free(&bufs); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); } static int response_on_begin_frame_callback(nghttp2_session *session, const nghttp2_frame_hd *hd, void *user_data _U_) { int rv; if (hd->type != NGHTTP2_HEADERS) { return 0; } rv = nghttp2_submit_response(session, hd->stream_id, resnv, ARRLEN(resnv), NULL); CU_ASSERT(0 == rv); return 0; } void test_nghttp2_session_recv_headers_early_response(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_bufs bufs; nghttp2_buf *buf; nghttp2_hd_deflater deflater; nghttp2_mem *mem; nghttp2_nv *nva; size_t nvlen; nghttp2_frame frame; ssize_t rv; nghttp2_stream *stream; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; callbacks.on_begin_frame_callback = response_on_begin_frame_callback; nghttp2_session_server_new(&session, &callbacks, NULL); nghttp2_hd_deflate_init(&deflater, mem); nvlen = ARRLEN(reqnv); nghttp2_nv_array_copy(&nva, reqnv, nvlen, mem); nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS | NGHTTP2_FLAG_END_STREAM, 1, NGHTTP2_HCAT_REQUEST, NULL, nva, nvlen); rv = nghttp2_frame_pack_headers(&bufs, &frame.headers, &deflater); CU_ASSERT(0 == rv); nghttp2_frame_headers_free(&frame.headers, mem); buf = &bufs.head->buf; /* Only receive 9 bytes headers, and invoke on_begin_frame_callback */ rv = nghttp2_session_mem_recv(session, buf->pos, 9); CU_ASSERT(9 == rv); rv = nghttp2_session_send(session); CU_ASSERT(0 == rv); rv = nghttp2_session_mem_recv(session, buf->pos + 9, nghttp2_buf_len(buf) - 9); CU_ASSERT((ssize_t)nghttp2_buf_len(buf) - 9 == rv); stream = nghttp2_session_get_stream_raw(session, 1); CU_ASSERT(stream->flags & NGHTTP2_STREAM_FLAG_CLOSED); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); nghttp2_bufs_free(&bufs); } void test_nghttp2_session_server_recv_push_response(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_bufs bufs; nghttp2_buf *buf; ssize_t rv; my_user_data ud; nghttp2_mem *mem; nghttp2_frame frame; nghttp2_hd_deflater deflater; nghttp2_nv *nva; size_t nvlen; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.on_invalid_frame_recv_callback = on_invalid_frame_recv_callback; nghttp2_session_server_new(&session, &callbacks, &ud); nghttp2_hd_deflate_init(&deflater, mem); nghttp2_session_open_stream(session, 2, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_RESERVED, NULL); nvlen = ARRLEN(resnv); nghttp2_nv_array_copy(&nva, resnv, nvlen, mem); nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS, 2, NGHTTP2_HCAT_HEADERS, &pri_spec_default, nva, nvlen); rv = nghttp2_frame_pack_headers(&bufs, &frame.headers, &deflater); CU_ASSERT(0 == rv); CU_ASSERT(nghttp2_bufs_len(&bufs) > 0); nghttp2_frame_headers_free(&frame.headers, mem); buf = &bufs.head->buf; ud.invalid_frame_recv_cb_called = 0; rv = nghttp2_session_mem_recv(session, buf->pos, nghttp2_buf_len(buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(buf) == rv); CU_ASSERT(1 == ud.invalid_frame_recv_cb_called); nghttp2_bufs_free(&bufs); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); } void test_nghttp2_session_recv_premature_headers(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_bufs bufs; nghttp2_buf *buf; ssize_t rv; my_user_data ud; nghttp2_hd_deflater deflater; nghttp2_outbound_item *item; nghttp2_mem *mem; uint32_t payloadlen; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_server_new(&session, &callbacks, &ud); nghttp2_hd_deflate_init(&deflater, mem); pack_headers(&bufs, &deflater, 1, NGHTTP2_FLAG_END_HEADERS, reqnv, ARRLEN(reqnv), mem); buf = &bufs.head->buf; /* Intentionally feed payload cutting last 1 byte off */ payloadlen = nghttp2_get_uint32(buf->pos) >> 8; nghttp2_put_uint32be(buf->pos, ((payloadlen - 1) << 8) + buf->pos[3]); rv = nghttp2_session_mem_recv(session, buf->pos, nghttp2_buf_len(buf) - 1); CU_ASSERT((ssize_t)(nghttp2_buf_len(buf) - 1) == rv); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NULL != item); CU_ASSERT(NGHTTP2_RST_STREAM == item->frame.hd.type); CU_ASSERT(NGHTTP2_COMPRESSION_ERROR == item->frame.rst_stream.error_code); CU_ASSERT(1 == item->frame.hd.stream_id); CU_ASSERT(0 == nghttp2_session_send(session)); nghttp2_bufs_reset(&bufs); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); /* Test for PUSH_PROMISE */ nghttp2_session_client_new(&session, &callbacks, &ud); nghttp2_hd_deflate_init(&deflater, mem); nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); rv = pack_push_promise(&bufs, &deflater, 1, NGHTTP2_FLAG_END_HEADERS, 2, reqnv, ARRLEN(reqnv), mem); CU_ASSERT(0 == rv); buf = &bufs.head->buf; payloadlen = nghttp2_get_uint32(buf->pos) >> 8; /* Intentionally feed payload cutting last 1 byte off */ nghttp2_put_uint32be(buf->pos, ((payloadlen - 1) << 8) + buf->pos[3]); rv = nghttp2_session_mem_recv(session, buf->pos, nghttp2_buf_len(buf) - 1); CU_ASSERT((ssize_t)(nghttp2_buf_len(buf) - 1) == rv); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NULL != item); CU_ASSERT(NGHTTP2_RST_STREAM == item->frame.hd.type); CU_ASSERT(NGHTTP2_COMPRESSION_ERROR == item->frame.rst_stream.error_code); CU_ASSERT(2 == item->frame.hd.stream_id); CU_ASSERT(0 == nghttp2_session_send(session)); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); nghttp2_bufs_free(&bufs); } void test_nghttp2_session_recv_unknown_frame(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data ud; uint8_t data[16384]; size_t datalen; nghttp2_frame_hd hd; ssize_t rv; nghttp2_frame_hd_init(&hd, 16000, 99, NGHTTP2_FLAG_NONE, 0); nghttp2_frame_pack_frame_hd(data, &hd); datalen = NGHTTP2_FRAME_HDLEN + hd.length; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.on_frame_recv_callback = on_frame_recv_callback; nghttp2_session_server_new(&session, &callbacks, &ud); ud.frame_recv_cb_called = 0; /* Unknown frame must be ignored */ rv = nghttp2_session_mem_recv(session, data, datalen); CU_ASSERT(rv == (ssize_t)datalen); CU_ASSERT(0 == ud.frame_recv_cb_called); CU_ASSERT(NULL == nghttp2_session_get_next_ob_item(session)); nghttp2_session_del(session); } void test_nghttp2_session_recv_unexpected_continuation(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data ud; uint8_t data[16384]; size_t datalen; nghttp2_frame_hd hd; ssize_t rv; nghttp2_outbound_item *item; nghttp2_frame_hd_init(&hd, 16000, NGHTTP2_CONTINUATION, NGHTTP2_FLAG_END_HEADERS, 1); nghttp2_frame_pack_frame_hd(data, &hd); datalen = NGHTTP2_FRAME_HDLEN + hd.length; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.on_frame_recv_callback = on_frame_recv_callback; nghttp2_session_server_new(&session, &callbacks, &ud); open_stream(session, 1); ud.frame_recv_cb_called = 0; /* unexpected CONTINUATION must be treated as connection error */ rv = nghttp2_session_mem_recv(session, data, datalen); CU_ASSERT(rv == (ssize_t)datalen); CU_ASSERT(0 == ud.frame_recv_cb_called); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_GOAWAY == item->frame.hd.type); nghttp2_session_del(session); } void test_nghttp2_session_recv_settings_header_table_size(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_frame frame; nghttp2_bufs bufs; nghttp2_buf *buf; ssize_t rv; my_user_data ud; nghttp2_settings_entry iv[3]; nghttp2_nv nv = MAKE_NV(":authority", "example.org"); nghttp2_mem *mem; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.on_frame_recv_callback = on_frame_recv_callback; callbacks.send_callback = null_send_callback; nghttp2_session_client_new(&session, &callbacks, &ud); iv[0].settings_id = NGHTTP2_SETTINGS_HEADER_TABLE_SIZE; iv[0].value = 3000; iv[1].settings_id = NGHTTP2_SETTINGS_INITIAL_WINDOW_SIZE; iv[1].value = 16384; nghttp2_frame_settings_init(&frame.settings, NGHTTP2_FLAG_NONE, dup_iv(iv, 2), 2); rv = nghttp2_frame_pack_settings(&bufs, &frame.settings); CU_ASSERT(0 == rv); CU_ASSERT(nghttp2_bufs_len(&bufs) > 0); nghttp2_frame_settings_free(&frame.settings, mem); buf = &bufs.head->buf; assert(nghttp2_bufs_len(&bufs) == nghttp2_buf_len(buf)); ud.frame_recv_cb_called = 0; rv = nghttp2_session_mem_recv(session, buf->pos, nghttp2_buf_len(buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(buf) == rv); CU_ASSERT(1 == ud.frame_recv_cb_called); CU_ASSERT(3000 == session->remote_settings.header_table_size); CU_ASSERT(16384 == session->remote_settings.initial_window_size); nghttp2_bufs_reset(&bufs); /* 2 SETTINGS_HEADER_TABLE_SIZE */ iv[0].settings_id = NGHTTP2_SETTINGS_HEADER_TABLE_SIZE; iv[0].value = 3001; iv[1].settings_id = NGHTTP2_SETTINGS_INITIAL_WINDOW_SIZE; iv[1].value = 16383; iv[2].settings_id = NGHTTP2_SETTINGS_HEADER_TABLE_SIZE; iv[2].value = 3001; nghttp2_frame_settings_init(&frame.settings, NGHTTP2_FLAG_NONE, dup_iv(iv, 3), 3); rv = nghttp2_frame_pack_settings(&bufs, &frame.settings); CU_ASSERT(0 == rv); CU_ASSERT(nghttp2_bufs_len(&bufs) > 0); nghttp2_frame_settings_free(&frame.settings, mem); buf = &bufs.head->buf; assert(nghttp2_bufs_len(&bufs) == nghttp2_buf_len(buf)); ud.frame_recv_cb_called = 0; rv = nghttp2_session_mem_recv(session, buf->pos, nghttp2_buf_len(buf)); CU_ASSERT((ssize_t)(nghttp2_buf_len(buf)) == rv); CU_ASSERT(1 == ud.frame_recv_cb_called); CU_ASSERT(3001 == session->remote_settings.header_table_size); CU_ASSERT(16383 == session->remote_settings.initial_window_size); nghttp2_bufs_reset(&bufs); /* 2 SETTINGS_HEADER_TABLE_SIZE; first entry clears dynamic header table. */ nghttp2_submit_request(session, NULL, &nv, 1, NULL, NULL); nghttp2_session_send(session); CU_ASSERT(0 < session->hd_deflater.ctx.hd_table.len); iv[0].settings_id = NGHTTP2_SETTINGS_HEADER_TABLE_SIZE; iv[0].value = 0; iv[1].settings_id = NGHTTP2_SETTINGS_INITIAL_WINDOW_SIZE; iv[1].value = 16382; iv[2].settings_id = NGHTTP2_SETTINGS_HEADER_TABLE_SIZE; iv[2].value = 4096; nghttp2_frame_settings_init(&frame.settings, NGHTTP2_FLAG_NONE, dup_iv(iv, 3), 3); rv = nghttp2_frame_pack_settings(&bufs, &frame.settings); CU_ASSERT(0 == rv); CU_ASSERT(nghttp2_bufs_len(&bufs) > 0); nghttp2_frame_settings_free(&frame.settings, mem); buf = &bufs.head->buf; assert(nghttp2_bufs_len(&bufs) == nghttp2_buf_len(buf)); ud.frame_recv_cb_called = 0; rv = nghttp2_session_mem_recv(session, buf->pos, nghttp2_buf_len(buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(buf) == rv); CU_ASSERT(1 == ud.frame_recv_cb_called); CU_ASSERT(4096 == session->remote_settings.header_table_size); CU_ASSERT(16382 == session->remote_settings.initial_window_size); CU_ASSERT(0 == session->hd_deflater.ctx.hd_table.len); nghttp2_bufs_reset(&bufs); /* 2 SETTINGS_HEADER_TABLE_SIZE; second entry clears dynamic header table. */ nghttp2_submit_request(session, NULL, &nv, 1, NULL, NULL); nghttp2_session_send(session); CU_ASSERT(0 < session->hd_deflater.ctx.hd_table.len); iv[0].settings_id = NGHTTP2_SETTINGS_HEADER_TABLE_SIZE; iv[0].value = 3000; iv[1].settings_id = NGHTTP2_SETTINGS_INITIAL_WINDOW_SIZE; iv[1].value = 16381; iv[2].settings_id = NGHTTP2_SETTINGS_HEADER_TABLE_SIZE; iv[2].value = 0; nghttp2_frame_settings_init(&frame.settings, NGHTTP2_FLAG_NONE, dup_iv(iv, 3), 3); rv = nghttp2_frame_pack_settings(&bufs, &frame.settings); CU_ASSERT(0 == rv); CU_ASSERT(nghttp2_bufs_len(&bufs) > 0); nghttp2_frame_settings_free(&frame.settings, mem); buf = &bufs.head->buf; assert(nghttp2_bufs_len(&bufs) == nghttp2_buf_len(buf)); ud.frame_recv_cb_called = 0; rv = nghttp2_session_mem_recv(session, buf->pos, nghttp2_buf_len(buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(buf) == rv); CU_ASSERT(1 == ud.frame_recv_cb_called); CU_ASSERT(0 == session->remote_settings.header_table_size); CU_ASSERT(16381 == session->remote_settings.initial_window_size); CU_ASSERT(0 == session->hd_deflater.ctx.hd_table.len); nghttp2_bufs_reset(&bufs); nghttp2_bufs_free(&bufs); nghttp2_session_del(session); } void test_nghttp2_session_recv_too_large_frame_length(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; uint8_t buf[NGHTTP2_FRAME_HDLEN]; nghttp2_outbound_item *item; nghttp2_frame_hd hd; /* Initial max frame size is NGHTTP2_MAX_FRAME_SIZE_MIN */ nghttp2_frame_hd_init(&hd, NGHTTP2_MAX_FRAME_SIZE_MIN + 1, NGHTTP2_HEADERS, NGHTTP2_FLAG_NONE, 1); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); nghttp2_session_server_new(&session, &callbacks, NULL); nghttp2_frame_pack_frame_hd(buf, &hd); CU_ASSERT(sizeof(buf) == nghttp2_session_mem_recv(session, buf, sizeof(buf))); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(item != NULL); CU_ASSERT(NGHTTP2_GOAWAY == item->frame.hd.type); nghttp2_session_del(session); } void test_nghttp2_session_continue(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data user_data; const nghttp2_nv nv1[] = {MAKE_NV(":method", "GET"), MAKE_NV(":path", "/")}; const nghttp2_nv nv2[] = {MAKE_NV("user-agent", "nghttp2/1.0.0"), MAKE_NV("alpha", "bravo")}; nghttp2_bufs bufs; nghttp2_buf *buf; size_t framelen1, framelen2; ssize_t rv; uint8_t buffer[4096]; nghttp2_buf databuf; nghttp2_frame frame; nghttp2_nv *nva; size_t nvlen; const nghttp2_frame *recv_frame; nghttp2_frame_hd data_hd; nghttp2_hd_deflater deflater; nghttp2_mem *mem; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); nghttp2_buf_wrap_init(&databuf, buffer, sizeof(buffer)); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; callbacks.on_frame_recv_callback = on_frame_recv_callback; callbacks.on_data_chunk_recv_callback = pause_on_data_chunk_recv_callback; callbacks.on_header_callback = pause_on_header_callback; callbacks.on_begin_headers_callback = on_begin_headers_callback; nghttp2_session_server_new(&session, &callbacks, &user_data); /* disable strict HTTP layering checks */ session->opt_flags |= NGHTTP2_OPTMASK_NO_HTTP_MESSAGING; nghttp2_hd_deflate_init(&deflater, mem); /* Make 2 HEADERS frames */ nvlen = ARRLEN(nv1); nghttp2_nv_array_copy(&nva, nv1, nvlen, mem); nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS, 1, NGHTTP2_HCAT_HEADERS, NULL, nva, nvlen); rv = nghttp2_frame_pack_headers(&bufs, &frame.headers, &deflater); CU_ASSERT(0 == rv); CU_ASSERT(nghttp2_bufs_len(&bufs) > 0); nghttp2_frame_headers_free(&frame.headers, mem); buf = &bufs.head->buf; assert(nghttp2_bufs_len(&bufs) == nghttp2_buf_len(buf)); framelen1 = nghttp2_buf_len(buf); databuf.last = nghttp2_cpymem(databuf.last, buf->pos, nghttp2_buf_len(buf)); nvlen = ARRLEN(nv2); nghttp2_nv_array_copy(&nva, nv2, nvlen, mem); nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS, 3, NGHTTP2_HCAT_HEADERS, NULL, nva, nvlen); nghttp2_bufs_reset(&bufs); rv = nghttp2_frame_pack_headers(&bufs, &frame.headers, &deflater); CU_ASSERT(0 == rv); CU_ASSERT(nghttp2_bufs_len(&bufs) > 0); nghttp2_frame_headers_free(&frame.headers, mem); assert(nghttp2_bufs_len(&bufs) == nghttp2_buf_len(buf)); framelen2 = nghttp2_buf_len(buf); databuf.last = nghttp2_cpymem(databuf.last, buf->pos, nghttp2_buf_len(buf)); /* Receive 1st HEADERS and pause */ user_data.begin_headers_cb_called = 0; user_data.header_cb_called = 0; rv = nghttp2_session_mem_recv(session, databuf.pos, nghttp2_buf_len(&databuf)); CU_ASSERT(rv >= 0); databuf.pos += rv; recv_frame = user_data.frame; CU_ASSERT(NGHTTP2_HEADERS == recv_frame->hd.type); CU_ASSERT(framelen1 - NGHTTP2_FRAME_HDLEN == recv_frame->hd.length); CU_ASSERT(1 == user_data.begin_headers_cb_called); CU_ASSERT(1 == user_data.header_cb_called); CU_ASSERT(nghttp2_nv_equal(&nv1[0], &user_data.nv)); /* get 2nd header field */ user_data.begin_headers_cb_called = 0; user_data.header_cb_called = 0; rv = nghttp2_session_mem_recv(session, databuf.pos, nghttp2_buf_len(&databuf)); CU_ASSERT(rv >= 0); databuf.pos += rv; CU_ASSERT(0 == user_data.begin_headers_cb_called); CU_ASSERT(1 == user_data.header_cb_called); CU_ASSERT(nghttp2_nv_equal(&nv1[1], &user_data.nv)); /* will call end_headers_callback and receive 2nd HEADERS and pause */ user_data.begin_headers_cb_called = 0; user_data.header_cb_called = 0; rv = nghttp2_session_mem_recv(session, databuf.pos, nghttp2_buf_len(&databuf)); CU_ASSERT(rv >= 0); databuf.pos += rv; recv_frame = user_data.frame; CU_ASSERT(NGHTTP2_HEADERS == recv_frame->hd.type); CU_ASSERT(framelen2 - NGHTTP2_FRAME_HDLEN == recv_frame->hd.length); CU_ASSERT(1 == user_data.begin_headers_cb_called); CU_ASSERT(1 == user_data.header_cb_called); CU_ASSERT(nghttp2_nv_equal(&nv2[0], &user_data.nv)); /* get 2nd header field */ user_data.begin_headers_cb_called = 0; user_data.header_cb_called = 0; rv = nghttp2_session_mem_recv(session, databuf.pos, nghttp2_buf_len(&databuf)); CU_ASSERT(rv >= 0); databuf.pos += rv; CU_ASSERT(0 == user_data.begin_headers_cb_called); CU_ASSERT(1 == user_data.header_cb_called); CU_ASSERT(nghttp2_nv_equal(&nv2[1], &user_data.nv)); /* No input data, frame_recv_callback is called */ user_data.begin_headers_cb_called = 0; user_data.header_cb_called = 0; user_data.frame_recv_cb_called = 0; rv = nghttp2_session_mem_recv(session, databuf.pos, nghttp2_buf_len(&databuf)); CU_ASSERT(rv >= 0); databuf.pos += rv; CU_ASSERT(0 == user_data.begin_headers_cb_called); CU_ASSERT(0 == user_data.header_cb_called); CU_ASSERT(1 == user_data.frame_recv_cb_called); /* Receive DATA */ nghttp2_frame_hd_init(&data_hd, 16, NGHTTP2_DATA, NGHTTP2_FLAG_NONE, 1); nghttp2_buf_reset(&databuf); nghttp2_frame_pack_frame_hd(databuf.pos, &data_hd); /* Intentionally specify larger buffer size to see pause is kicked in. */ databuf.last = databuf.end; user_data.frame_recv_cb_called = 0; rv = nghttp2_session_mem_recv(session, databuf.pos, nghttp2_buf_len(&databuf)); CU_ASSERT(16 + NGHTTP2_FRAME_HDLEN == rv); CU_ASSERT(0 == user_data.frame_recv_cb_called); /* Next nghttp2_session_mem_recv invokes on_frame_recv_callback and pause again in on_data_chunk_recv_callback since we pass same DATA frame. */ user_data.frame_recv_cb_called = 0; rv = nghttp2_session_mem_recv(session, databuf.pos, nghttp2_buf_len(&databuf)); CU_ASSERT(16 + NGHTTP2_FRAME_HDLEN == rv); CU_ASSERT(1 == user_data.frame_recv_cb_called); /* And finally call on_frame_recv_callback with 0 size input */ user_data.frame_recv_cb_called = 0; rv = nghttp2_session_mem_recv(session, NULL, 0); CU_ASSERT(0 == rv); CU_ASSERT(1 == user_data.frame_recv_cb_called); nghttp2_bufs_free(&bufs); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); } void test_nghttp2_session_add_frame(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; accumulator acc; my_user_data user_data; nghttp2_outbound_item *item; nghttp2_frame *frame; nghttp2_nv *nva; size_t nvlen; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = accumulator_send_callback; acc.length = 0; user_data.acc = &acc; CU_ASSERT(0 == nghttp2_session_client_new(&session, &callbacks, &user_data)); item = mem->malloc(sizeof(nghttp2_outbound_item), NULL); nghttp2_outbound_item_init(item); frame = &item->frame; nvlen = ARRLEN(reqnv); nghttp2_nv_array_copy(&nva, reqnv, nvlen, mem); nghttp2_frame_headers_init( &frame->headers, NGHTTP2_FLAG_END_HEADERS | NGHTTP2_FLAG_PRIORITY, (int32_t)session->next_stream_id, NGHTTP2_HCAT_REQUEST, NULL, nva, nvlen); session->next_stream_id += 2; CU_ASSERT(0 == nghttp2_session_add_item(session, item)); CU_ASSERT(NULL != nghttp2_outbound_queue_top(&session->ob_syn)); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(NGHTTP2_HEADERS == acc.buf[3]); CU_ASSERT((NGHTTP2_FLAG_END_HEADERS | NGHTTP2_FLAG_PRIORITY) == acc.buf[4]); /* check stream id */ CU_ASSERT(1 == nghttp2_get_uint32(&acc.buf[5])); nghttp2_session_del(session); } void test_nghttp2_session_on_request_headers_received(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data user_data; nghttp2_frame frame; nghttp2_stream *stream; int32_t stream_id = 1; nghttp2_nv malformed_nva[] = {MAKE_NV(":path", "\x01")}; nghttp2_nv *nva; size_t nvlen; nghttp2_priority_spec pri_spec; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.on_begin_headers_callback = on_begin_headers_callback; callbacks.on_invalid_frame_recv_callback = on_invalid_frame_recv_callback; nghttp2_session_server_new(&session, &callbacks, &user_data); nghttp2_priority_spec_init(&pri_spec, 0, 255, 0); nghttp2_frame_headers_init( &frame.headers, NGHTTP2_FLAG_END_HEADERS | NGHTTP2_FLAG_PRIORITY, stream_id, NGHTTP2_HCAT_REQUEST, &pri_spec, NULL, 0); user_data.begin_headers_cb_called = 0; user_data.invalid_frame_recv_cb_called = 0; CU_ASSERT(0 == nghttp2_session_on_request_headers_received(session, &frame)); CU_ASSERT(1 == user_data.begin_headers_cb_called); stream = nghttp2_session_get_stream(session, stream_id); CU_ASSERT(NGHTTP2_STREAM_OPENING == stream->state); CU_ASSERT(255 == stream->weight); nghttp2_frame_headers_free(&frame.headers, mem); /* More than un-ACKed max concurrent streams leads REFUSED_STREAM */ session->pending_local_max_concurrent_stream = 1; nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS | NGHTTP2_FLAG_PRIORITY, 3, NGHTTP2_HCAT_HEADERS, NULL, NULL, 0); user_data.invalid_frame_recv_cb_called = 0; CU_ASSERT(NGHTTP2_ERR_IGN_HEADER_BLOCK == nghttp2_session_on_request_headers_received(session, &frame)); CU_ASSERT(1 == user_data.invalid_frame_recv_cb_called); CU_ASSERT(0 == (session->goaway_flags & NGHTTP2_GOAWAY_TERM_ON_SEND)); nghttp2_frame_headers_free(&frame.headers, mem); session->local_settings.max_concurrent_streams = NGHTTP2_INITIAL_MAX_CONCURRENT_STREAMS; /* Stream ID less than or equal to the previouly received request HEADERS is just ignored due to race condition */ nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS | NGHTTP2_FLAG_PRIORITY, 3, NGHTTP2_HCAT_HEADERS, NULL, NULL, 0); user_data.invalid_frame_recv_cb_called = 0; CU_ASSERT(NGHTTP2_ERR_IGN_HEADER_BLOCK == nghttp2_session_on_request_headers_received(session, &frame)); CU_ASSERT(0 == user_data.invalid_frame_recv_cb_called); CU_ASSERT(0 == (session->goaway_flags & NGHTTP2_GOAWAY_TERM_ON_SEND)); nghttp2_frame_headers_free(&frame.headers, mem); /* Stream ID is our side and it is idle stream ID, then treat it as connection error */ nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS | NGHTTP2_FLAG_PRIORITY, 2, NGHTTP2_HCAT_HEADERS, NULL, NULL, 0); user_data.invalid_frame_recv_cb_called = 0; CU_ASSERT(NGHTTP2_ERR_IGN_HEADER_BLOCK == nghttp2_session_on_request_headers_received(session, &frame)); CU_ASSERT(1 == user_data.invalid_frame_recv_cb_called); CU_ASSERT(session->goaway_flags & NGHTTP2_GOAWAY_TERM_ON_SEND); nghttp2_frame_headers_free(&frame.headers, mem); nghttp2_session_del(session); /* Check malformed headers. The library accept it. */ nghttp2_session_server_new(&session, &callbacks, &user_data); nvlen = ARRLEN(malformed_nva); nghttp2_nv_array_copy(&nva, malformed_nva, nvlen, mem); nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS | NGHTTP2_FLAG_PRIORITY, 1, NGHTTP2_HCAT_HEADERS, NULL, nva, nvlen); user_data.begin_headers_cb_called = 0; user_data.invalid_frame_recv_cb_called = 0; CU_ASSERT(0 == nghttp2_session_on_request_headers_received(session, &frame)); CU_ASSERT(1 == user_data.begin_headers_cb_called); CU_ASSERT(0 == user_data.invalid_frame_recv_cb_called); nghttp2_frame_headers_free(&frame.headers, mem); nghttp2_session_del(session); /* Check client side */ nghttp2_session_client_new(&session, &callbacks, &user_data); /* Receiving peer's idle stream ID is subject to connection error */ nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS, 2, NGHTTP2_HCAT_REQUEST, NULL, NULL, 0); user_data.invalid_frame_recv_cb_called = 0; CU_ASSERT(NGHTTP2_ERR_IGN_HEADER_BLOCK == nghttp2_session_on_request_headers_received(session, &frame)); CU_ASSERT(1 == user_data.invalid_frame_recv_cb_called); CU_ASSERT(session->goaway_flags & NGHTTP2_GOAWAY_TERM_ON_SEND); nghttp2_frame_headers_free(&frame.headers, mem); nghttp2_session_del(session); nghttp2_session_client_new(&session, &callbacks, &user_data); /* Receiving our's idle stream ID is subject to connection error */ nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS, 1, NGHTTP2_HCAT_REQUEST, NULL, NULL, 0); user_data.invalid_frame_recv_cb_called = 0; CU_ASSERT(NGHTTP2_ERR_IGN_HEADER_BLOCK == nghttp2_session_on_request_headers_received(session, &frame)); CU_ASSERT(1 == user_data.invalid_frame_recv_cb_called); CU_ASSERT(session->goaway_flags & NGHTTP2_GOAWAY_TERM_ON_SEND); nghttp2_frame_headers_free(&frame.headers, mem); nghttp2_session_del(session); nghttp2_session_client_new(&session, &callbacks, &user_data); session->next_stream_id = 5; /* Stream ID which is not idle and not in stream map is just ignored */ nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS, 3, NGHTTP2_HCAT_REQUEST, NULL, NULL, 0); user_data.invalid_frame_recv_cb_called = 0; CU_ASSERT(NGHTTP2_ERR_IGN_HEADER_BLOCK == nghttp2_session_on_request_headers_received(session, &frame)); CU_ASSERT(0 == user_data.invalid_frame_recv_cb_called); CU_ASSERT(0 == (session->goaway_flags & NGHTTP2_GOAWAY_TERM_ON_SEND)); nghttp2_frame_headers_free(&frame.headers, mem); nghttp2_session_del(session); nghttp2_session_server_new(&session, &callbacks, &user_data); /* Stream ID which is equal to local_last_stream_id is ok. */ session->local_last_stream_id = 3; nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS, 3, NGHTTP2_HCAT_REQUEST, NULL, NULL, 0); CU_ASSERT(0 == nghttp2_session_on_request_headers_received(session, &frame)); nghttp2_frame_headers_free(&frame.headers, mem); /* If GOAWAY has been sent, new stream is ignored */ nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS, 5, NGHTTP2_HCAT_REQUEST, NULL, NULL, 0); session->goaway_flags |= NGHTTP2_GOAWAY_SENT; user_data.invalid_frame_recv_cb_called = 0; CU_ASSERT(NGHTTP2_ERR_IGN_HEADER_BLOCK == nghttp2_session_on_request_headers_received(session, &frame)); CU_ASSERT(0 == user_data.invalid_frame_recv_cb_called); CU_ASSERT(0 == (session->goaway_flags & NGHTTP2_GOAWAY_TERM_ON_SEND)); nghttp2_frame_headers_free(&frame.headers, mem); nghttp2_session_del(session); } void test_nghttp2_session_on_response_headers_received(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data user_data; nghttp2_frame frame; nghttp2_stream *stream; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.on_begin_headers_callback = on_begin_headers_callback; callbacks.on_invalid_frame_recv_callback = on_invalid_frame_recv_callback; nghttp2_session_client_new(&session, &callbacks, &user_data); stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS, 1, NGHTTP2_HCAT_HEADERS, NULL, NULL, 0); user_data.begin_headers_cb_called = 0; user_data.invalid_frame_recv_cb_called = 0; CU_ASSERT(0 == nghttp2_session_on_response_headers_received(session, &frame, stream)); CU_ASSERT(1 == user_data.begin_headers_cb_called); CU_ASSERT(NGHTTP2_STREAM_OPENED == stream->state); nghttp2_frame_headers_free(&frame.headers, mem); nghttp2_session_del(session); } void test_nghttp2_session_on_headers_received(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data user_data; nghttp2_frame frame; nghttp2_stream *stream; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.on_begin_headers_callback = on_begin_headers_callback; callbacks.on_invalid_frame_recv_callback = on_invalid_frame_recv_callback; nghttp2_session_client_new(&session, &callbacks, &user_data); stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENED, NULL); nghttp2_stream_shutdown(stream, NGHTTP2_SHUT_WR); nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS, 1, NGHTTP2_HCAT_HEADERS, NULL, NULL, 0); user_data.begin_headers_cb_called = 0; user_data.invalid_frame_recv_cb_called = 0; CU_ASSERT(0 == nghttp2_session_on_headers_received(session, &frame, stream)); CU_ASSERT(1 == user_data.begin_headers_cb_called); CU_ASSERT(NGHTTP2_STREAM_OPENED == stream->state); /* stream closed */ frame.hd.flags |= NGHTTP2_FLAG_END_STREAM; CU_ASSERT(0 == nghttp2_session_on_headers_received(session, &frame, stream)); CU_ASSERT(2 == user_data.begin_headers_cb_called); /* Check to see when NGHTTP2_STREAM_CLOSING, incoming HEADERS is discarded. */ stream = nghttp2_session_open_stream(session, 3, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_CLOSING, NULL); frame.hd.stream_id = 3; frame.hd.flags = NGHTTP2_FLAG_END_HEADERS; CU_ASSERT(NGHTTP2_ERR_IGN_HEADER_BLOCK == nghttp2_session_on_headers_received(session, &frame, stream)); /* See no counters are updated */ CU_ASSERT(2 == user_data.begin_headers_cb_called); CU_ASSERT(0 == user_data.invalid_frame_recv_cb_called); /* Server initiated stream */ stream = nghttp2_session_open_stream(session, 2, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); /* half closed (remote) */ frame.hd.flags = NGHTTP2_FLAG_END_HEADERS | NGHTTP2_FLAG_END_STREAM; frame.hd.stream_id = 2; CU_ASSERT(0 == nghttp2_session_on_headers_received(session, &frame, stream)); CU_ASSERT(3 == user_data.begin_headers_cb_called); CU_ASSERT(NGHTTP2_STREAM_OPENING == stream->state); nghttp2_stream_shutdown(stream, NGHTTP2_SHUT_RD); /* Further reception of HEADERS is subject to stream error */ CU_ASSERT(NGHTTP2_ERR_IGN_HEADER_BLOCK == nghttp2_session_on_headers_received(session, &frame, stream)); CU_ASSERT(1 == user_data.invalid_frame_recv_cb_called); nghttp2_frame_headers_free(&frame.headers, mem); nghttp2_session_del(session); } void test_nghttp2_session_on_push_response_headers_received(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data user_data; nghttp2_frame frame; nghttp2_stream *stream; nghttp2_outbound_item *item; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; callbacks.on_begin_headers_callback = on_begin_headers_callback; callbacks.on_invalid_frame_recv_callback = on_invalid_frame_recv_callback; nghttp2_session_client_new(&session, &callbacks, &user_data); stream = nghttp2_session_open_stream(session, 2, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_RESERVED, NULL); nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS, 2, NGHTTP2_HCAT_HEADERS, NULL, NULL, 0); /* nghttp2_session_on_push_response_headers_received assumes stream's state is NGHTTP2_STREAM_RESERVED and session->server is 0. */ user_data.begin_headers_cb_called = 0; user_data.invalid_frame_recv_cb_called = 0; CU_ASSERT(1 == session->num_incoming_reserved_streams); CU_ASSERT(0 == nghttp2_session_on_push_response_headers_received( session, &frame, stream)); CU_ASSERT(1 == user_data.begin_headers_cb_called); CU_ASSERT(0 == session->num_incoming_reserved_streams); CU_ASSERT(NGHTTP2_STREAM_OPENED == stream->state); CU_ASSERT(1 == session->num_incoming_streams); CU_ASSERT(0 == (stream->flags & NGHTTP2_STREAM_FLAG_PUSH)); /* If un-ACKed max concurrent streams limit is exceeded, RST_STREAMed */ session->pending_local_max_concurrent_stream = 1; stream = nghttp2_session_open_stream(session, 4, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_RESERVED, NULL); frame.hd.stream_id = 4; CU_ASSERT(NGHTTP2_ERR_IGN_HEADER_BLOCK == nghttp2_session_on_push_response_headers_received(session, &frame, stream)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_RST_STREAM == item->frame.hd.type); CU_ASSERT(NGHTTP2_REFUSED_STREAM == item->frame.rst_stream.error_code); CU_ASSERT(1 == session->num_incoming_streams); CU_ASSERT(1 == session->num_incoming_reserved_streams); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(1 == session->num_incoming_streams); /* If ACKed max concurrent streams limit is exceeded, GOAWAY is issued */ session->local_settings.max_concurrent_streams = 1; stream = nghttp2_session_open_stream(session, 6, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_RESERVED, NULL); frame.hd.stream_id = 6; CU_ASSERT(NGHTTP2_ERR_IGN_HEADER_BLOCK == nghttp2_session_on_push_response_headers_received(session, &frame, stream)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_GOAWAY == item->frame.hd.type); CU_ASSERT(NGHTTP2_PROTOCOL_ERROR == item->frame.goaway.error_code); CU_ASSERT(1 == session->num_incoming_streams); CU_ASSERT(1 == session->num_incoming_reserved_streams); nghttp2_frame_headers_free(&frame.headers, mem); nghttp2_session_del(session); } void test_nghttp2_session_on_priority_received(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data user_data; nghttp2_frame frame; nghttp2_stream *stream, *dep_stream; nghttp2_priority_spec pri_spec; nghttp2_outbound_item *item; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.on_frame_recv_callback = on_frame_recv_callback; callbacks.on_invalid_frame_recv_callback = on_invalid_frame_recv_callback; nghttp2_session_server_new(&session, &callbacks, &user_data); stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); nghttp2_priority_spec_init(&pri_spec, 0, 2, 0); nghttp2_frame_priority_init(&frame.priority, 1, &pri_spec); /* depend on stream 0 */ CU_ASSERT(0 == nghttp2_session_on_priority_received(session, &frame)); CU_ASSERT(2 == stream->weight); stream = nghttp2_session_open_stream(session, 2, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); dep_stream = nghttp2_session_open_stream(session, 3, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); frame.hd.stream_id = 2; /* using dependency stream */ nghttp2_priority_spec_init(&frame.priority.pri_spec, 3, 1, 0); CU_ASSERT(0 == nghttp2_session_on_priority_received(session, &frame)); CU_ASSERT(dep_stream == stream->dep_prev); /* PRIORITY against idle stream */ frame.hd.stream_id = 100; CU_ASSERT(0 == nghttp2_session_on_priority_received(session, &frame)); stream = nghttp2_session_get_stream_raw(session, frame.hd.stream_id); CU_ASSERT(NGHTTP2_STREAM_IDLE == stream->state); CU_ASSERT(dep_stream == stream->dep_prev); nghttp2_frame_priority_free(&frame.priority); nghttp2_session_del(session); /* Check dep_stream_id == stream_id case */ nghttp2_session_server_new(&session, &callbacks, &user_data); nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENED, NULL); nghttp2_priority_spec_init(&pri_spec, 1, 0, 0); nghttp2_frame_priority_init(&frame.priority, 1, &pri_spec); CU_ASSERT(0 == nghttp2_session_on_priority_received(session, &frame)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_GOAWAY == item->frame.hd.type); nghttp2_frame_priority_free(&frame.priority); nghttp2_session_del(session); /* Check again dep_stream_id == stream_id, and stream_id is idle */ nghttp2_session_server_new(&session, &callbacks, &user_data); nghttp2_priority_spec_init(&pri_spec, 1, 16, 0); nghttp2_frame_priority_init(&frame.priority, 1, &pri_spec); CU_ASSERT(0 == nghttp2_session_on_priority_received(session, &frame)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_GOAWAY == item->frame.hd.type); nghttp2_frame_priority_free(&frame.priority); nghttp2_session_del(session); } void test_nghttp2_session_on_rst_stream_received(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data user_data; nghttp2_frame frame; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); nghttp2_session_server_new(&session, &callbacks, &user_data); nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); nghttp2_frame_rst_stream_init(&frame.rst_stream, 1, NGHTTP2_PROTOCOL_ERROR); CU_ASSERT(0 == nghttp2_session_on_rst_stream_received(session, &frame)); CU_ASSERT(NULL == nghttp2_session_get_stream(session, 1)); nghttp2_frame_rst_stream_free(&frame.rst_stream); nghttp2_session_del(session); } void test_nghttp2_session_on_settings_received(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data user_data; nghttp2_stream *stream1, *stream2; nghttp2_frame frame; const size_t niv = 5; nghttp2_settings_entry iv[255]; nghttp2_outbound_item *item; nghttp2_nv nv = MAKE_NV(":authority", "example.org"); nghttp2_mem *mem; mem = nghttp2_mem_default(); iv[0].settings_id = NGHTTP2_SETTINGS_MAX_CONCURRENT_STREAMS; iv[0].value = 50; iv[1].settings_id = NGHTTP2_SETTINGS_MAX_CONCURRENT_STREAMS; iv[1].value = 1000000009; iv[2].settings_id = NGHTTP2_SETTINGS_INITIAL_WINDOW_SIZE; iv[2].value = 64 * 1024; iv[3].settings_id = NGHTTP2_SETTINGS_HEADER_TABLE_SIZE; iv[3].value = 1024; iv[4].settings_id = NGHTTP2_SETTINGS_ENABLE_PUSH; iv[4].value = 0; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_client_new(&session, &callbacks, &user_data); session->remote_settings.initial_window_size = 16 * 1024; stream1 = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); stream2 = nghttp2_session_open_stream(session, 2, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); /* Set window size for each streams and will see how settings updates these values */ stream1->remote_window_size = 16 * 1024; stream2->remote_window_size = -48 * 1024; nghttp2_frame_settings_init(&frame.settings, NGHTTP2_FLAG_NONE, dup_iv(iv, niv), niv); CU_ASSERT(0 == nghttp2_session_on_settings_received(session, &frame, 0)); CU_ASSERT(1000000009 == session->remote_settings.max_concurrent_streams); CU_ASSERT(64 * 1024 == session->remote_settings.initial_window_size); CU_ASSERT(1024 == session->remote_settings.header_table_size); CU_ASSERT(0 == session->remote_settings.enable_push); CU_ASSERT(64 * 1024 == stream1->remote_window_size); CU_ASSERT(0 == stream2->remote_window_size); frame.settings.iv[2].value = 16 * 1024; CU_ASSERT(0 == nghttp2_session_on_settings_received(session, &frame, 0)); CU_ASSERT(16 * 1024 == stream1->remote_window_size); CU_ASSERT(-48 * 1024 == stream2->remote_window_size); CU_ASSERT(16 * 1024 == nghttp2_session_get_stream_remote_window_size( session, stream1->stream_id)); CU_ASSERT(0 == nghttp2_session_get_stream_remote_window_size( session, stream2->stream_id)); nghttp2_frame_settings_free(&frame.settings, mem); nghttp2_session_del(session); /* Check ACK with niv > 0 */ nghttp2_session_server_new(&session, &callbacks, NULL); nghttp2_frame_settings_init(&frame.settings, NGHTTP2_FLAG_ACK, dup_iv(iv, 1), 1); CU_ASSERT(0 == nghttp2_session_on_settings_received(session, &frame, 0)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(item != NULL); CU_ASSERT(NGHTTP2_GOAWAY == item->frame.hd.type); nghttp2_frame_settings_free(&frame.settings, mem); nghttp2_session_del(session); /* Check ACK against no inflight SETTINGS */ nghttp2_session_server_new(&session, &callbacks, NULL); nghttp2_frame_settings_init(&frame.settings, NGHTTP2_FLAG_ACK, NULL, 0); CU_ASSERT(0 == nghttp2_session_on_settings_received(session, &frame, 0)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(item != NULL); CU_ASSERT(NGHTTP2_GOAWAY == item->frame.hd.type); nghttp2_frame_settings_free(&frame.settings, mem); nghttp2_session_del(session); /* Check that 2 SETTINGS_HEADER_TABLE_SIZE 0 and 4096 are included and header table size is once cleared to 0. */ nghttp2_session_client_new(&session, &callbacks, NULL); nghttp2_submit_request(session, NULL, &nv, 1, NULL, NULL); nghttp2_session_send(session); CU_ASSERT(session->hd_deflater.ctx.hd_table.len > 0); iv[0].settings_id = NGHTTP2_SETTINGS_HEADER_TABLE_SIZE; iv[0].value = 0; iv[1].settings_id = NGHTTP2_SETTINGS_HEADER_TABLE_SIZE; iv[1].value = 2048; nghttp2_frame_settings_init(&frame.settings, NGHTTP2_FLAG_NONE, dup_iv(iv, 2), 2); CU_ASSERT(0 == nghttp2_session_on_settings_received(session, &frame, 0)); CU_ASSERT(0 == session->hd_deflater.ctx.hd_table.len); CU_ASSERT(2048 == session->hd_deflater.ctx.hd_table_bufsize_max); CU_ASSERT(2048 == session->remote_settings.header_table_size); nghttp2_frame_settings_free(&frame.settings, mem); nghttp2_session_del(session); /* Check too large SETTINGS_MAX_FRAME_SIZE */ nghttp2_session_server_new(&session, &callbacks, NULL); iv[0].settings_id = NGHTTP2_SETTINGS_MAX_FRAME_SIZE; iv[0].value = NGHTTP2_MAX_FRAME_SIZE_MAX + 1; nghttp2_frame_settings_init(&frame.settings, NGHTTP2_FLAG_NONE, dup_iv(iv, 1), 1); CU_ASSERT(0 == nghttp2_session_on_settings_received(session, &frame, 0)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(item != NULL); CU_ASSERT(NGHTTP2_GOAWAY == item->frame.hd.type); nghttp2_frame_settings_free(&frame.settings, mem); nghttp2_session_del(session); } void test_nghttp2_session_on_push_promise_received(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data user_data; nghttp2_frame frame; nghttp2_stream *stream, *promised_stream; nghttp2_outbound_item *item; nghttp2_nv malformed_nva[] = {MAKE_NV(":path", "\x01")}; nghttp2_nv *nva; size_t nvlen; nghttp2_mem *mem; nghttp2_settings_entry iv = {NGHTTP2_SETTINGS_ENABLE_PUSH, 0}; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; callbacks.on_begin_headers_callback = on_begin_headers_callback; callbacks.on_invalid_frame_recv_callback = on_invalid_frame_recv_callback; nghttp2_session_client_new(&session, &callbacks, &user_data); stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); nghttp2_frame_push_promise_init(&frame.push_promise, NGHTTP2_FLAG_END_HEADERS, 1, 2, NULL, 0); user_data.begin_headers_cb_called = 0; user_data.invalid_frame_recv_cb_called = 0; CU_ASSERT(0 == nghttp2_session_on_push_promise_received(session, &frame)); CU_ASSERT(1 == user_data.begin_headers_cb_called); CU_ASSERT(1 == session->num_incoming_reserved_streams); promised_stream = nghttp2_session_get_stream(session, 2); CU_ASSERT(NGHTTP2_STREAM_RESERVED == promised_stream->state); CU_ASSERT(2 == session->last_recv_stream_id); /* Attempt to PUSH_PROMISE against half close (remote) */ nghttp2_stream_shutdown(stream, NGHTTP2_SHUT_RD); frame.push_promise.promised_stream_id = 4; user_data.begin_headers_cb_called = 0; user_data.invalid_frame_recv_cb_called = 0; CU_ASSERT(NGHTTP2_ERR_IGN_HEADER_BLOCK == nghttp2_session_on_push_promise_received(session, &frame)); CU_ASSERT(0 == user_data.begin_headers_cb_called); CU_ASSERT(1 == user_data.invalid_frame_recv_cb_called); CU_ASSERT(1 == session->num_incoming_reserved_streams); CU_ASSERT(NULL == nghttp2_session_get_stream(session, 4)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_RST_STREAM == item->frame.hd.type); CU_ASSERT(4 == item->frame.hd.stream_id); CU_ASSERT(NGHTTP2_PROTOCOL_ERROR == item->frame.rst_stream.error_code); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(4 == session->last_recv_stream_id); /* Attempt to PUSH_PROMISE against stream in closing state */ stream->shut_flags = NGHTTP2_SHUT_NONE; stream->state = NGHTTP2_STREAM_CLOSING; frame.push_promise.promised_stream_id = 6; user_data.begin_headers_cb_called = 0; user_data.invalid_frame_recv_cb_called = 0; CU_ASSERT(NGHTTP2_ERR_IGN_HEADER_BLOCK == nghttp2_session_on_push_promise_received(session, &frame)); CU_ASSERT(0 == user_data.begin_headers_cb_called); CU_ASSERT(1 == session->num_incoming_reserved_streams); CU_ASSERT(NULL == nghttp2_session_get_stream(session, 6)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_RST_STREAM == item->frame.hd.type); CU_ASSERT(6 == item->frame.hd.stream_id); CU_ASSERT(NGHTTP2_CANCEL == item->frame.rst_stream.error_code); CU_ASSERT(0 == nghttp2_session_send(session)); /* Attempt to PUSH_PROMISE against non-existent stream */ frame.hd.stream_id = 3; frame.push_promise.promised_stream_id = 8; user_data.begin_headers_cb_called = 0; user_data.invalid_frame_recv_cb_called = 0; CU_ASSERT(NGHTTP2_ERR_IGN_HEADER_BLOCK == nghttp2_session_on_push_promise_received(session, &frame)); CU_ASSERT(0 == user_data.begin_headers_cb_called); CU_ASSERT(1 == session->num_incoming_reserved_streams); CU_ASSERT(NULL == nghttp2_session_get_stream(session, 8)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_GOAWAY == item->frame.hd.type); CU_ASSERT(0 == item->frame.hd.stream_id); CU_ASSERT(NGHTTP2_PROTOCOL_ERROR == item->frame.goaway.error_code); CU_ASSERT(0 == nghttp2_session_send(session)); nghttp2_session_del(session); nghttp2_session_client_new(&session, &callbacks, &user_data); stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); /* Same ID twice */ frame.hd.stream_id = 1; frame.push_promise.promised_stream_id = 2; user_data.begin_headers_cb_called = 0; user_data.invalid_frame_recv_cb_called = 0; CU_ASSERT(0 == nghttp2_session_on_push_promise_received(session, &frame)); CU_ASSERT(1 == user_data.begin_headers_cb_called); CU_ASSERT(1 == session->num_incoming_reserved_streams); CU_ASSERT(NULL != nghttp2_session_get_stream(session, 2)); user_data.begin_headers_cb_called = 0; user_data.invalid_frame_recv_cb_called = 0; CU_ASSERT(NGHTTP2_ERR_IGN_HEADER_BLOCK == nghttp2_session_on_push_promise_received(session, &frame)); CU_ASSERT(0 == user_data.begin_headers_cb_called); CU_ASSERT(1 == session->num_incoming_reserved_streams); CU_ASSERT(NULL == nghttp2_session_get_stream(session, 8)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_GOAWAY == item->frame.hd.type); CU_ASSERT(NGHTTP2_PROTOCOL_ERROR == item->frame.goaway.error_code); CU_ASSERT(0 == nghttp2_session_send(session)); /* After GOAWAY, PUSH_PROMISE will be discarded */ frame.push_promise.promised_stream_id = 10; user_data.begin_headers_cb_called = 0; user_data.invalid_frame_recv_cb_called = 0; CU_ASSERT(NGHTTP2_ERR_IGN_HEADER_BLOCK == nghttp2_session_on_push_promise_received(session, &frame)); CU_ASSERT(0 == user_data.begin_headers_cb_called); CU_ASSERT(1 == session->num_incoming_reserved_streams); CU_ASSERT(NULL == nghttp2_session_get_stream(session, 10)); CU_ASSERT(NULL == nghttp2_session_get_next_ob_item(session)); nghttp2_frame_push_promise_free(&frame.push_promise, mem); nghttp2_session_del(session); nghttp2_session_client_new(&session, &callbacks, &user_data); nghttp2_session_open_stream(session, 2, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_RESERVED, NULL); /* Attempt to PUSH_PROMISE against reserved (remote) stream */ nghttp2_frame_push_promise_init(&frame.push_promise, NGHTTP2_FLAG_END_HEADERS, 2, 4, NULL, 0); user_data.begin_headers_cb_called = 0; user_data.invalid_frame_recv_cb_called = 0; CU_ASSERT(NGHTTP2_ERR_IGN_HEADER_BLOCK == nghttp2_session_on_push_promise_received(session, &frame)); CU_ASSERT(0 == user_data.begin_headers_cb_called); CU_ASSERT(1 == user_data.invalid_frame_recv_cb_called); CU_ASSERT(1 == session->num_incoming_reserved_streams); nghttp2_frame_push_promise_free(&frame.push_promise, mem); nghttp2_session_del(session); /* Disable PUSH */ nghttp2_session_client_new(&session, &callbacks, &user_data); nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); session->local_settings.enable_push = 0; nghttp2_frame_push_promise_init(&frame.push_promise, NGHTTP2_FLAG_END_HEADERS, 1, 2, NULL, 0); user_data.begin_headers_cb_called = 0; user_data.invalid_frame_recv_cb_called = 0; CU_ASSERT(NGHTTP2_ERR_IGN_HEADER_BLOCK == nghttp2_session_on_push_promise_received(session, &frame)); CU_ASSERT(0 == user_data.begin_headers_cb_called); CU_ASSERT(1 == user_data.invalid_frame_recv_cb_called); CU_ASSERT(0 == session->num_incoming_reserved_streams); nghttp2_frame_push_promise_free(&frame.push_promise, mem); nghttp2_session_del(session); /* Check malformed headers. We accept malformed headers */ nghttp2_session_client_new(&session, &callbacks, &user_data); nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); nvlen = ARRLEN(malformed_nva); nghttp2_nv_array_copy(&nva, malformed_nva, nvlen, mem); nghttp2_frame_push_promise_init(&frame.push_promise, NGHTTP2_FLAG_END_HEADERS, 1, 2, nva, nvlen); user_data.begin_headers_cb_called = 0; user_data.invalid_frame_recv_cb_called = 0; CU_ASSERT(0 == nghttp2_session_on_push_promise_received(session, &frame)); CU_ASSERT(1 == user_data.begin_headers_cb_called); CU_ASSERT(0 == user_data.invalid_frame_recv_cb_called); nghttp2_frame_push_promise_free(&frame.push_promise, mem); nghttp2_session_del(session); /* If local_settings.enable_push = 0 is pending, but not acked from peer, incoming PUSH_PROMISE is rejected */ nghttp2_session_client_new(&session, &callbacks, &user_data); nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); /* Submit settings with ENABLE_PUSH = 0 (thus disabling push) */ nghttp2_submit_settings(session, NGHTTP2_FLAG_NONE, &iv, 1); nghttp2_frame_push_promise_init(&frame.push_promise, NGHTTP2_FLAG_END_HEADERS, 1, 2, NULL, 0); CU_ASSERT(NGHTTP2_ERR_IGN_HEADER_BLOCK == nghttp2_session_on_push_promise_received(session, &frame)); CU_ASSERT(0 == session->num_incoming_reserved_streams); nghttp2_frame_push_promise_free(&frame.push_promise, mem); nghttp2_session_del(session); /* Check max_incoming_reserved_streams */ nghttp2_session_client_new(&session, &callbacks, &user_data); session->max_incoming_reserved_streams = 1; nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); nghttp2_session_open_stream(session, 2, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_RESERVED, NULL); CU_ASSERT(1 == session->num_incoming_reserved_streams); nghttp2_frame_push_promise_init(&frame.push_promise, NGHTTP2_FLAG_END_HEADERS, 1, 4, NULL, 0); CU_ASSERT(NGHTTP2_ERR_IGN_HEADER_BLOCK == nghttp2_session_on_push_promise_received(session, &frame)); CU_ASSERT(1 == session->num_incoming_reserved_streams); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_RST_STREAM == item->frame.hd.type); CU_ASSERT(NGHTTP2_CANCEL == item->frame.rst_stream.error_code); nghttp2_frame_push_promise_free(&frame.push_promise, mem); nghttp2_session_del(session); } void test_nghttp2_session_on_ping_received(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data user_data; nghttp2_frame frame; nghttp2_outbound_item *top; const uint8_t opaque_data[] = "01234567"; user_data.frame_recv_cb_called = 0; user_data.invalid_frame_recv_cb_called = 0; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.on_frame_recv_callback = on_frame_recv_callback; callbacks.on_invalid_frame_recv_callback = on_invalid_frame_recv_callback; nghttp2_session_client_new(&session, &callbacks, &user_data); nghttp2_frame_ping_init(&frame.ping, NGHTTP2_FLAG_ACK, opaque_data); CU_ASSERT(0 == nghttp2_session_on_ping_received(session, &frame)); CU_ASSERT(1 == user_data.frame_recv_cb_called); /* Since this ping frame has ACK flag set, no further action is performed. */ CU_ASSERT(NULL == nghttp2_outbound_queue_top(&session->ob_urgent)); /* Clear the flag, and receive it again */ frame.hd.flags = NGHTTP2_FLAG_NONE; CU_ASSERT(0 == nghttp2_session_on_ping_received(session, &frame)); CU_ASSERT(2 == user_data.frame_recv_cb_called); top = nghttp2_outbound_queue_top(&session->ob_urgent); CU_ASSERT(NGHTTP2_PING == top->frame.hd.type); CU_ASSERT(NGHTTP2_FLAG_ACK == top->frame.hd.flags); CU_ASSERT(memcmp(opaque_data, top->frame.ping.opaque_data, 8) == 0); nghttp2_frame_ping_free(&frame.ping); nghttp2_session_del(session); } void test_nghttp2_session_on_goaway_received(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data user_data; nghttp2_frame frame; int i; nghttp2_mem *mem; mem = nghttp2_mem_default(); user_data.frame_recv_cb_called = 0; user_data.invalid_frame_recv_cb_called = 0; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.on_frame_recv_callback = on_frame_recv_callback; callbacks.on_invalid_frame_recv_callback = on_invalid_frame_recv_callback; callbacks.on_stream_close_callback = on_stream_close_callback; nghttp2_session_client_new(&session, &callbacks, &user_data); for (i = 1; i <= 7; ++i) { open_stream(session, i); } nghttp2_frame_goaway_init(&frame.goaway, 3, NGHTTP2_PROTOCOL_ERROR, NULL, 0); user_data.stream_close_cb_called = 0; CU_ASSERT(0 == nghttp2_session_on_goaway_received(session, &frame)); CU_ASSERT(1 == user_data.frame_recv_cb_called); CU_ASSERT(3 == session->remote_last_stream_id); /* on_stream_close should be callsed for 2 times (stream 5 and 7) */ CU_ASSERT(2 == user_data.stream_close_cb_called); CU_ASSERT(NULL != nghttp2_session_get_stream(session, 1)); CU_ASSERT(NULL != nghttp2_session_get_stream(session, 2)); CU_ASSERT(NULL != nghttp2_session_get_stream(session, 3)); CU_ASSERT(NULL != nghttp2_session_get_stream(session, 4)); CU_ASSERT(NULL == nghttp2_session_get_stream(session, 5)); CU_ASSERT(NULL != nghttp2_session_get_stream(session, 6)); CU_ASSERT(NULL == nghttp2_session_get_stream(session, 7)); nghttp2_frame_goaway_free(&frame.goaway, mem); nghttp2_session_del(session); } void test_nghttp2_session_on_window_update_received(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data user_data; nghttp2_frame frame; nghttp2_stream *stream; nghttp2_outbound_item *data_item; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.on_frame_recv_callback = on_frame_recv_callback; callbacks.on_invalid_frame_recv_callback = on_invalid_frame_recv_callback; user_data.frame_recv_cb_called = 0; user_data.invalid_frame_recv_cb_called = 0; nghttp2_session_client_new(&session, &callbacks, &user_data); stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENED, NULL); data_item = create_data_ob_item(mem); CU_ASSERT(0 == nghttp2_stream_attach_item(stream, data_item)); nghttp2_frame_window_update_init(&frame.window_update, NGHTTP2_FLAG_NONE, 1, 16 * 1024); CU_ASSERT(0 == nghttp2_session_on_window_update_received(session, &frame)); CU_ASSERT(1 == user_data.frame_recv_cb_called); CU_ASSERT(NGHTTP2_INITIAL_WINDOW_SIZE + 16 * 1024 == stream->remote_window_size); CU_ASSERT(0 == nghttp2_stream_defer_item( stream, NGHTTP2_STREAM_FLAG_DEFERRED_FLOW_CONTROL)); CU_ASSERT(0 == nghttp2_session_on_window_update_received(session, &frame)); CU_ASSERT(2 == user_data.frame_recv_cb_called); CU_ASSERT(NGHTTP2_INITIAL_WINDOW_SIZE + 16 * 1024 * 2 == stream->remote_window_size); CU_ASSERT(0 == (stream->flags & NGHTTP2_STREAM_FLAG_DEFERRED_ALL)); nghttp2_frame_window_update_free(&frame.window_update); /* Receiving WINDOW_UPDATE on reserved (remote) stream is a connection error */ nghttp2_session_open_stream(session, 2, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_RESERVED, NULL); nghttp2_frame_window_update_init(&frame.window_update, NGHTTP2_FLAG_NONE, 2, 4096); CU_ASSERT(!(session->goaway_flags & NGHTTP2_GOAWAY_TERM_ON_SEND)); CU_ASSERT(0 == nghttp2_session_on_window_update_received(session, &frame)); CU_ASSERT(session->goaway_flags & NGHTTP2_GOAWAY_TERM_ON_SEND); nghttp2_frame_window_update_free(&frame.window_update); nghttp2_session_del(session); /* Receiving WINDOW_UPDATE on reserved (local) stream is allowed */ nghttp2_session_server_new(&session, &callbacks, &user_data); stream = nghttp2_session_open_stream(session, 2, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_RESERVED, NULL); nghttp2_frame_window_update_init(&frame.window_update, NGHTTP2_FLAG_NONE, 2, 4096); CU_ASSERT(0 == nghttp2_session_on_window_update_received(session, &frame)); CU_ASSERT(!(session->goaway_flags & NGHTTP2_GOAWAY_TERM_ON_SEND)); CU_ASSERT(NGHTTP2_INITIAL_WINDOW_SIZE + 4096 == stream->remote_window_size); nghttp2_frame_window_update_free(&frame.window_update); nghttp2_session_del(session); } void test_nghttp2_session_on_data_received(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data user_data; nghttp2_outbound_item *top; nghttp2_stream *stream; nghttp2_frame frame; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); nghttp2_session_client_new(&session, &callbacks, &user_data); stream = nghttp2_session_open_stream(session, 2, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); nghttp2_frame_hd_init(&frame.hd, 4096, NGHTTP2_DATA, NGHTTP2_FLAG_NONE, 2); CU_ASSERT(0 == nghttp2_session_on_data_received(session, &frame)); CU_ASSERT(0 == stream->shut_flags); frame.hd.flags = NGHTTP2_FLAG_END_STREAM; CU_ASSERT(0 == nghttp2_session_on_data_received(session, &frame)); CU_ASSERT(NGHTTP2_SHUT_RD == stream->shut_flags); /* If NGHTTP2_STREAM_CLOSING state, DATA frame is discarded. */ nghttp2_session_open_stream(session, 4, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_CLOSING, NULL); frame.hd.flags = NGHTTP2_FLAG_NONE; frame.hd.stream_id = 4; CU_ASSERT(0 == nghttp2_session_on_data_received(session, &frame)); CU_ASSERT(NULL == nghttp2_outbound_queue_top(&session->ob_reg)); /* Check INVALID_STREAM case: DATA frame with stream ID which does not exist. */ frame.hd.stream_id = 6; CU_ASSERT(0 == nghttp2_session_on_data_received(session, &frame)); top = nghttp2_outbound_queue_top(&session->ob_reg); /* DATA against nonexistent stream is just ignored for now */ CU_ASSERT(top == NULL); /* CU_ASSERT(NGHTTP2_RST_STREAM == top->frame.hd.type); */ /* CU_ASSERT(NGHTTP2_PROTOCOL_ERROR == top->frame.rst_stream.error_code); */ nghttp2_session_del(session); } void test_nghttp2_session_send_headers_start_stream(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_outbound_item *item; nghttp2_frame *frame; nghttp2_stream *stream; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_client_new(&session, &callbacks, NULL); item = mem->malloc(sizeof(nghttp2_outbound_item), NULL); nghttp2_outbound_item_init(item); frame = &item->frame; nghttp2_frame_headers_init(&frame->headers, NGHTTP2_FLAG_END_HEADERS, (int32_t)session->next_stream_id, NGHTTP2_HCAT_REQUEST, NULL, NULL, 0); session->next_stream_id += 2; nghttp2_session_add_item(session, item); CU_ASSERT(0 == nghttp2_session_send(session)); stream = nghttp2_session_get_stream(session, 1); CU_ASSERT(NGHTTP2_STREAM_OPENING == stream->state); nghttp2_session_del(session); } void test_nghttp2_session_send_headers_reply(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_outbound_item *item; nghttp2_frame *frame; nghttp2_stream *stream; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; CU_ASSERT(0 == nghttp2_session_client_new(&session, &callbacks, NULL)); nghttp2_session_open_stream(session, 2, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); item = mem->malloc(sizeof(nghttp2_outbound_item), NULL); nghttp2_outbound_item_init(item); frame = &item->frame; nghttp2_frame_headers_init(&frame->headers, NGHTTP2_FLAG_END_HEADERS, 2, NGHTTP2_HCAT_HEADERS, NULL, NULL, 0); nghttp2_session_add_item(session, item); CU_ASSERT(0 == nghttp2_session_send(session)); stream = nghttp2_session_get_stream(session, 2); CU_ASSERT(NGHTTP2_STREAM_OPENED == stream->state); nghttp2_session_del(session); } void test_nghttp2_session_send_headers_frame_size_error(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_outbound_item *item; nghttp2_frame *frame; nghttp2_nv *nva; size_t nvlen; size_t vallen = NGHTTP2_HD_MAX_NV; nghttp2_nv nv[28]; size_t nnv = ARRLEN(nv); size_t i; my_user_data ud; nghttp2_mem *mem; mem = nghttp2_mem_default(); for (i = 0; i < nnv; ++i) { nv[i].name = (uint8_t *)"header"; nv[i].namelen = strlen((const char *)nv[i].name); nv[i].value = mem->malloc(vallen + 1, NULL); memset(nv[i].value, '0' + (int)i, vallen); nv[i].value[vallen] = '\0'; nv[i].valuelen = vallen; nv[i].flags = NGHTTP2_NV_FLAG_NONE; } memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; callbacks.on_frame_not_send_callback = on_frame_not_send_callback; nghttp2_session_client_new(&session, &callbacks, &ud); nvlen = nnv; nghttp2_nv_array_copy(&nva, nv, nvlen, mem); item = mem->malloc(sizeof(nghttp2_outbound_item), NULL); nghttp2_outbound_item_init(item); frame = &item->frame; nghttp2_frame_headers_init(&frame->headers, NGHTTP2_FLAG_END_HEADERS, (int32_t)session->next_stream_id, NGHTTP2_HCAT_REQUEST, NULL, nva, nvlen); session->next_stream_id += 2; nghttp2_session_add_item(session, item); ud.frame_not_send_cb_called = 0; CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(1 == ud.frame_not_send_cb_called); CU_ASSERT(NGHTTP2_HEADERS == ud.not_sent_frame_type); CU_ASSERT(NGHTTP2_ERR_FRAME_SIZE_ERROR == ud.not_sent_error); for (i = 0; i < nnv; ++i) { mem->free(nv[i].value, NULL); } nghttp2_session_del(session); } void test_nghttp2_session_send_headers_push_reply(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_outbound_item *item; nghttp2_frame *frame; nghttp2_stream *stream; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; CU_ASSERT(0 == nghttp2_session_server_new(&session, &callbacks, NULL)); nghttp2_session_open_stream(session, 2, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_RESERVED, NULL); item = mem->malloc(sizeof(nghttp2_outbound_item), NULL); nghttp2_outbound_item_init(item); frame = &item->frame; nghttp2_frame_headers_init(&frame->headers, NGHTTP2_FLAG_END_HEADERS, 2, NGHTTP2_HCAT_HEADERS, NULL, NULL, 0); nghttp2_session_add_item(session, item); CU_ASSERT(0 == session->num_outgoing_streams); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(1 == session->num_outgoing_streams); stream = nghttp2_session_get_stream(session, 2); CU_ASSERT(NGHTTP2_STREAM_OPENED == stream->state); CU_ASSERT(0 == (stream->flags & NGHTTP2_STREAM_FLAG_PUSH)); nghttp2_session_del(session); } void test_nghttp2_session_send_rst_stream(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data user_data; nghttp2_outbound_item *item; nghttp2_frame *frame; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_client_new(&session, &callbacks, &user_data); nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); item = mem->malloc(sizeof(nghttp2_outbound_item), NULL); nghttp2_outbound_item_init(item); frame = &item->frame; nghttp2_frame_rst_stream_init(&frame->rst_stream, 1, NGHTTP2_PROTOCOL_ERROR); nghttp2_session_add_item(session, item); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(NULL == nghttp2_session_get_stream(session, 1)); nghttp2_session_del(session); } void test_nghttp2_session_send_push_promise(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_outbound_item *item; nghttp2_frame *frame; nghttp2_stream *stream; nghttp2_settings_entry iv; my_user_data ud; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; callbacks.on_frame_not_send_callback = on_frame_not_send_callback; nghttp2_session_server_new(&session, &callbacks, &ud); nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); item = mem->malloc(sizeof(nghttp2_outbound_item), NULL); nghttp2_outbound_item_init(item); frame = &item->frame; nghttp2_frame_push_promise_init(&frame->push_promise, NGHTTP2_FLAG_END_HEADERS, 1, (int32_t)session->next_stream_id, NULL, 0); session->next_stream_id += 2; nghttp2_session_add_item(session, item); CU_ASSERT(0 == nghttp2_session_send(session)); stream = nghttp2_session_get_stream(session, 2); CU_ASSERT(NGHTTP2_STREAM_RESERVED == stream->state); /* Received ENABLE_PUSH = 0 */ iv.settings_id = NGHTTP2_SETTINGS_ENABLE_PUSH; iv.value = 0; frame = mem->malloc(sizeof(nghttp2_frame), NULL); nghttp2_frame_settings_init(&frame->settings, NGHTTP2_FLAG_NONE, dup_iv(&iv, 1), 1); nghttp2_session_on_settings_received(session, frame, 1); nghttp2_frame_settings_free(&frame->settings, mem); mem->free(frame, NULL); item = mem->malloc(sizeof(nghttp2_outbound_item), NULL); nghttp2_outbound_item_init(item); frame = &item->frame; nghttp2_frame_push_promise_init(&frame->push_promise, NGHTTP2_FLAG_END_HEADERS, 1, -1, NULL, 0); nghttp2_session_add_item(session, item); ud.frame_not_send_cb_called = 0; CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(1 == ud.frame_not_send_cb_called); CU_ASSERT(NGHTTP2_PUSH_PROMISE == ud.not_sent_frame_type); CU_ASSERT(NGHTTP2_ERR_PUSH_DISABLED == ud.not_sent_error); nghttp2_session_del(session); /* PUSH_PROMISE from client is error */ nghttp2_session_client_new(&session, &callbacks, &ud); nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); item = mem->malloc(sizeof(nghttp2_outbound_item), NULL); nghttp2_outbound_item_init(item); frame = &item->frame; nghttp2_frame_push_promise_init(&frame->push_promise, NGHTTP2_FLAG_END_HEADERS, 1, -1, NULL, 0); nghttp2_session_add_item(session, item); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(NULL == nghttp2_session_get_stream(session, 3)); nghttp2_session_del(session); } void test_nghttp2_session_is_my_stream_id(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); nghttp2_session_server_new(&session, &callbacks, NULL); CU_ASSERT(0 == nghttp2_session_is_my_stream_id(session, 0)); CU_ASSERT(0 == nghttp2_session_is_my_stream_id(session, 1)); CU_ASSERT(1 == nghttp2_session_is_my_stream_id(session, 2)); nghttp2_session_del(session); nghttp2_session_client_new(&session, &callbacks, NULL); CU_ASSERT(0 == nghttp2_session_is_my_stream_id(session, 0)); CU_ASSERT(1 == nghttp2_session_is_my_stream_id(session, 1)); CU_ASSERT(0 == nghttp2_session_is_my_stream_id(session, 2)); nghttp2_session_del(session); } void test_nghttp2_session_upgrade2(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; uint8_t settings_payload[128]; size_t settings_payloadlen; nghttp2_settings_entry iv[16]; nghttp2_stream *stream; nghttp2_outbound_item *item; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; iv[0].settings_id = NGHTTP2_SETTINGS_MAX_CONCURRENT_STREAMS; iv[0].value = 1; iv[1].settings_id = NGHTTP2_SETTINGS_INITIAL_WINDOW_SIZE; iv[1].value = 4095; settings_payloadlen = (size_t)nghttp2_pack_settings_payload( settings_payload, sizeof(settings_payload), iv, 2); /* Check client side */ nghttp2_session_client_new(&session, &callbacks, NULL); CU_ASSERT(0 == nghttp2_session_upgrade2(session, settings_payload, settings_payloadlen, 0, &callbacks)); stream = nghttp2_session_get_stream(session, 1); CU_ASSERT(stream != NULL); CU_ASSERT(&callbacks == stream->stream_user_data); CU_ASSERT(NGHTTP2_SHUT_WR == stream->shut_flags); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_SETTINGS == item->frame.hd.type); CU_ASSERT(2 == item->frame.settings.niv); CU_ASSERT(NGHTTP2_SETTINGS_MAX_CONCURRENT_STREAMS == item->frame.settings.iv[0].settings_id); CU_ASSERT(1 == item->frame.settings.iv[0].value); CU_ASSERT(NGHTTP2_SETTINGS_INITIAL_WINDOW_SIZE == item->frame.settings.iv[1].settings_id); CU_ASSERT(4095 == item->frame.settings.iv[1].value); /* Call nghttp2_session_upgrade2() again is error */ CU_ASSERT(NGHTTP2_ERR_PROTO == nghttp2_session_upgrade2(session, settings_payload, settings_payloadlen, 0, &callbacks)); nghttp2_session_del(session); /* Check server side */ nghttp2_session_server_new(&session, &callbacks, NULL); CU_ASSERT(0 == nghttp2_session_upgrade2(session, settings_payload, settings_payloadlen, 0, &callbacks)); stream = nghttp2_session_get_stream(session, 1); CU_ASSERT(stream != NULL); CU_ASSERT(NULL == stream->stream_user_data); CU_ASSERT(NGHTTP2_SHUT_RD == stream->shut_flags); CU_ASSERT(NULL == nghttp2_session_get_next_ob_item(session)); CU_ASSERT(1 == session->remote_settings.max_concurrent_streams); CU_ASSERT(4095 == session->remote_settings.initial_window_size); /* Call nghttp2_session_upgrade2() again is error */ CU_ASSERT(NGHTTP2_ERR_PROTO == nghttp2_session_upgrade2(session, settings_payload, settings_payloadlen, 0, &callbacks)); nghttp2_session_del(session); /* Empty SETTINGS is OK */ settings_payloadlen = (size_t)nghttp2_pack_settings_payload( settings_payload, sizeof(settings_payload), NULL, 0); nghttp2_session_client_new(&session, &callbacks, NULL); CU_ASSERT(0 == nghttp2_session_upgrade2(session, settings_payload, settings_payloadlen, 0, NULL)); nghttp2_session_del(session); } void test_nghttp2_session_reprioritize_stream(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data ud; nghttp2_stream *stream; nghttp2_stream *dep_stream; nghttp2_priority_spec pri_spec; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = block_count_send_callback; nghttp2_session_server_new(&session, &callbacks, &ud); stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); nghttp2_priority_spec_init(&pri_spec, 0, 10, 0); nghttp2_session_reprioritize_stream(session, stream, &pri_spec); CU_ASSERT(10 == stream->weight); CU_ASSERT(&session->root == stream->dep_prev); /* If depenency to idle stream which is not in depdenency tree yet */ nghttp2_priority_spec_init(&pri_spec, 3, 99, 0); nghttp2_session_reprioritize_stream(session, stream, &pri_spec); CU_ASSERT(99 == stream->weight); CU_ASSERT(3 == stream->dep_prev->stream_id); dep_stream = nghttp2_session_get_stream_raw(session, 3); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == dep_stream->weight); dep_stream = open_stream(session, 3); /* Change weight */ pri_spec.weight = 128; nghttp2_session_reprioritize_stream(session, stream, &pri_spec); CU_ASSERT(128 == stream->weight); CU_ASSERT(dep_stream == stream->dep_prev); /* Change weight again to test short-path case */ pri_spec.weight = 100; nghttp2_session_reprioritize_stream(session, stream, &pri_spec); CU_ASSERT(100 == stream->weight); CU_ASSERT(dep_stream == stream->dep_prev); CU_ASSERT(100 == dep_stream->sum_dep_weight); /* Test circular dependency; stream 1 is first removed and becomes root. Then stream 3 depends on it. */ nghttp2_priority_spec_init(&pri_spec, 1, 1, 0); nghttp2_session_reprioritize_stream(session, dep_stream, &pri_spec); CU_ASSERT(1 == dep_stream->weight); CU_ASSERT(stream == dep_stream->dep_prev); /* Making priority to closed stream will result in default priority */ session->last_recv_stream_id = 9; nghttp2_priority_spec_init(&pri_spec, 5, 5, 0); nghttp2_session_reprioritize_stream(session, stream, &pri_spec); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == stream->weight); nghttp2_session_del(session); } void test_nghttp2_session_reprioritize_stream_with_idle_stream_dep(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_stream *stream; nghttp2_priority_spec pri_spec; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = block_count_send_callback; nghttp2_session_server_new(&session, &callbacks, NULL); stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); session->pending_local_max_concurrent_stream = 1; nghttp2_priority_spec_init(&pri_spec, 101, 10, 0); nghttp2_session_reprioritize_stream(session, stream, &pri_spec); /* idle stream is not counteed to max concurrent streams */ CU_ASSERT(10 == stream->weight); CU_ASSERT(101 == stream->dep_prev->stream_id); stream = nghttp2_session_get_stream_raw(session, 101); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == stream->weight); nghttp2_session_del(session); } void test_nghttp2_submit_data(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_data_provider data_prd; my_user_data ud; nghttp2_frame *frame; nghttp2_frame_hd hd; nghttp2_active_outbound_item *aob; nghttp2_bufs *framebufs; nghttp2_buf *buf; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = block_count_send_callback; data_prd.read_callback = fixed_length_data_source_read_callback; ud.data_source_length = NGHTTP2_DATA_PAYLOADLEN * 2; CU_ASSERT(0 == nghttp2_session_client_new(&session, &callbacks, &ud)); aob = &session->aob; framebufs = &aob->framebufs; nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); CU_ASSERT( 0 == nghttp2_submit_data(session, NGHTTP2_FLAG_END_STREAM, 1, &data_prd)); ud.block_count = 0; CU_ASSERT(0 == nghttp2_session_send(session)); frame = &aob->item->frame; buf = &framebufs->head->buf; nghttp2_frame_unpack_frame_hd(&hd, buf->pos); CU_ASSERT(NGHTTP2_FLAG_NONE == hd.flags); CU_ASSERT(NGHTTP2_FLAG_NONE == frame->hd.flags); /* aux_data.data.flags has these flags */ CU_ASSERT(NGHTTP2_FLAG_END_STREAM == aob->item->aux_data.data.flags); nghttp2_session_del(session); } void test_nghttp2_submit_data_read_length_too_large(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_data_provider data_prd; my_user_data ud; nghttp2_frame *frame; nghttp2_frame_hd hd; nghttp2_active_outbound_item *aob; nghttp2_bufs *framebufs; nghttp2_buf *buf; size_t payloadlen; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = block_count_send_callback; callbacks.read_length_callback = too_large_data_source_length_callback; data_prd.read_callback = fixed_length_data_source_read_callback; ud.data_source_length = NGHTTP2_DATA_PAYLOADLEN * 2; CU_ASSERT(0 == nghttp2_session_client_new(&session, &callbacks, &ud)); aob = &session->aob; framebufs = &aob->framebufs; nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); CU_ASSERT( 0 == nghttp2_submit_data(session, NGHTTP2_FLAG_END_STREAM, 1, &data_prd)); ud.block_count = 0; CU_ASSERT(0 == nghttp2_session_send(session)); frame = &aob->item->frame; buf = &framebufs->head->buf; nghttp2_frame_unpack_frame_hd(&hd, buf->pos); CU_ASSERT(NGHTTP2_FLAG_NONE == hd.flags); CU_ASSERT(NGHTTP2_FLAG_NONE == frame->hd.flags); CU_ASSERT(16384 == hd.length) /* aux_data.data.flags has these flags */ CU_ASSERT(NGHTTP2_FLAG_END_STREAM == aob->item->aux_data.data.flags); nghttp2_session_del(session); /* Check that buffers are expanded */ CU_ASSERT(0 == nghttp2_session_client_new(&session, &callbacks, &ud)); ud.data_source_length = NGHTTP2_MAX_FRAME_SIZE_MAX; session->remote_settings.max_frame_size = NGHTTP2_MAX_FRAME_SIZE_MAX; nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); CU_ASSERT( 0 == nghttp2_submit_data(session, NGHTTP2_FLAG_END_STREAM, 1, &data_prd)); ud.block_count = 0; CU_ASSERT(0 == nghttp2_session_send(session)); aob = &session->aob; frame = &aob->item->frame; framebufs = &aob->framebufs; buf = &framebufs->head->buf; nghttp2_frame_unpack_frame_hd(&hd, buf->pos); payloadlen = nghttp2_min(NGHTTP2_INITIAL_CONNECTION_WINDOW_SIZE, NGHTTP2_INITIAL_WINDOW_SIZE); CU_ASSERT(NGHTTP2_FRAME_HDLEN + 1 + payloadlen == (size_t)nghttp2_buf_cap(buf)); CU_ASSERT(NGHTTP2_FLAG_NONE == hd.flags); CU_ASSERT(NGHTTP2_FLAG_NONE == frame->hd.flags); CU_ASSERT(payloadlen == hd.length); /* aux_data.data.flags has these flags */ CU_ASSERT(NGHTTP2_FLAG_END_STREAM == aob->item->aux_data.data.flags); nghttp2_session_del(session); } void test_nghttp2_submit_data_read_length_smallest(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_data_provider data_prd; my_user_data ud; nghttp2_frame *frame; nghttp2_frame_hd hd; nghttp2_active_outbound_item *aob; nghttp2_bufs *framebufs; nghttp2_buf *buf; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = block_count_send_callback; callbacks.read_length_callback = smallest_length_data_source_length_callback; data_prd.read_callback = fixed_length_data_source_read_callback; ud.data_source_length = NGHTTP2_DATA_PAYLOADLEN * 2; CU_ASSERT(0 == nghttp2_session_client_new(&session, &callbacks, &ud)); aob = &session->aob; framebufs = &aob->framebufs; nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); CU_ASSERT( 0 == nghttp2_submit_data(session, NGHTTP2_FLAG_END_STREAM, 1, &data_prd)); ud.block_count = 0; CU_ASSERT(0 == nghttp2_session_send(session)); frame = &aob->item->frame; buf = &framebufs->head->buf; nghttp2_frame_unpack_frame_hd(&hd, buf->pos); CU_ASSERT(NGHTTP2_FLAG_NONE == hd.flags); CU_ASSERT(NGHTTP2_FLAG_NONE == frame->hd.flags); CU_ASSERT(1 == hd.length) /* aux_data.data.flags has these flags */ CU_ASSERT(NGHTTP2_FLAG_END_STREAM == aob->item->aux_data.data.flags); nghttp2_session_del(session); } static ssize_t submit_data_twice_data_source_read_callback( nghttp2_session *session _U_, int32_t stream_id _U_, uint8_t *buf _U_, size_t len, uint32_t *data_flags, nghttp2_data_source *source _U_, void *user_data _U_) { *data_flags |= NGHTTP2_DATA_FLAG_EOF; return (ssize_t)nghttp2_min(len, 16); } static int submit_data_twice_on_frame_send_callback(nghttp2_session *session, const nghttp2_frame *frame, void *user_data _U_) { static int called = 0; int rv; nghttp2_data_provider data_prd; if (called == 0) { called = 1; data_prd.read_callback = submit_data_twice_data_source_read_callback; rv = nghttp2_submit_data(session, NGHTTP2_FLAG_END_STREAM, frame->hd.stream_id, &data_prd); CU_ASSERT(0 == rv); } return 0; } void test_nghttp2_submit_data_twice(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_data_provider data_prd; my_user_data ud; accumulator acc; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = accumulator_send_callback; callbacks.on_frame_send_callback = submit_data_twice_on_frame_send_callback; data_prd.read_callback = submit_data_twice_data_source_read_callback; acc.length = 0; ud.acc = &acc; CU_ASSERT(0 == nghttp2_session_client_new(&session, &callbacks, &ud)); nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); CU_ASSERT(0 == nghttp2_submit_data(session, NGHTTP2_FLAG_NONE, 1, &data_prd)); CU_ASSERT(0 == nghttp2_session_send(session)); /* We should have sent 2 DATA frame with 16 bytes payload each */ CU_ASSERT(NGHTTP2_FRAME_HDLEN * 2 + 16 * 2 == acc.length); nghttp2_session_del(session); } void test_nghttp2_submit_request_with_data(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_data_provider data_prd; my_user_data ud; nghttp2_outbound_item *item; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; data_prd.read_callback = fixed_length_data_source_read_callback; ud.data_source_length = 64 * 1024 - 1; CU_ASSERT(0 == nghttp2_session_client_new(&session, &callbacks, &ud)); CU_ASSERT(1 == nghttp2_submit_request(session, NULL, reqnv, ARRLEN(reqnv), &data_prd, NULL)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(ARRLEN(reqnv) == item->frame.headers.nvlen); assert_nv_equal(reqnv, item->frame.headers.nva, item->frame.headers.nvlen, mem); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(0 == ud.data_source_length); nghttp2_session_del(session); } void test_nghttp2_submit_request_without_data(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; accumulator acc; nghttp2_data_provider data_prd = {{-1}, NULL}; nghttp2_outbound_item *item; my_user_data ud; nghttp2_frame frame; nghttp2_hd_inflater inflater; nva_out out; nghttp2_bufs bufs; nghttp2_mem *mem; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); nva_out_init(&out); acc.length = 0; ud.acc = &acc; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = accumulator_send_callback; CU_ASSERT(0 == nghttp2_session_client_new(&session, &callbacks, &ud)); nghttp2_hd_inflate_init(&inflater, mem); CU_ASSERT(1 == nghttp2_submit_request(session, NULL, reqnv, ARRLEN(reqnv), &data_prd, NULL)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(ARRLEN(reqnv) == item->frame.headers.nvlen); assert_nv_equal(reqnv, item->frame.headers.nva, item->frame.headers.nvlen, mem); CU_ASSERT(item->frame.hd.flags & NGHTTP2_FLAG_END_STREAM); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(0 == unpack_frame(&frame, acc.buf, acc.length)); nghttp2_bufs_add(&bufs, acc.buf, acc.length); inflate_hd(&inflater, &out, &bufs, NGHTTP2_FRAME_HDLEN, mem); CU_ASSERT(ARRLEN(reqnv) == out.nvlen); assert_nv_equal(reqnv, out.nva, out.nvlen, mem); nghttp2_frame_headers_free(&frame.headers, mem); nva_out_reset(&out, mem); nghttp2_bufs_free(&bufs); nghttp2_hd_inflate_free(&inflater); nghttp2_session_del(session); } void test_nghttp2_submit_response_with_data(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_data_provider data_prd; my_user_data ud; nghttp2_outbound_item *item; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; data_prd.read_callback = fixed_length_data_source_read_callback; ud.data_source_length = 64 * 1024 - 1; CU_ASSERT(0 == nghttp2_session_server_new(&session, &callbacks, &ud)); nghttp2_session_open_stream(session, 1, NGHTTP2_FLAG_END_STREAM, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); CU_ASSERT(0 == nghttp2_submit_response(session, 1, resnv, ARRLEN(resnv), &data_prd)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(ARRLEN(resnv) == item->frame.headers.nvlen); assert_nv_equal(resnv, item->frame.headers.nva, item->frame.headers.nvlen, mem); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(0 == ud.data_source_length); nghttp2_session_del(session); } void test_nghttp2_submit_response_without_data(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; accumulator acc; nghttp2_data_provider data_prd = {{-1}, NULL}; nghttp2_outbound_item *item; my_user_data ud; nghttp2_frame frame; nghttp2_hd_inflater inflater; nva_out out; nghttp2_bufs bufs; nghttp2_mem *mem; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); nva_out_init(&out); acc.length = 0; ud.acc = &acc; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = accumulator_send_callback; CU_ASSERT(0 == nghttp2_session_server_new(&session, &callbacks, &ud)); nghttp2_hd_inflate_init(&inflater, mem); nghttp2_session_open_stream(session, 1, NGHTTP2_FLAG_END_STREAM, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); CU_ASSERT(0 == nghttp2_submit_response(session, 1, resnv, ARRLEN(resnv), &data_prd)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(ARRLEN(resnv) == item->frame.headers.nvlen); assert_nv_equal(resnv, item->frame.headers.nva, item->frame.headers.nvlen, mem); CU_ASSERT(item->frame.hd.flags & NGHTTP2_FLAG_END_STREAM); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(0 == unpack_frame(&frame, acc.buf, acc.length)); nghttp2_bufs_add(&bufs, acc.buf, acc.length); inflate_hd(&inflater, &out, &bufs, NGHTTP2_FRAME_HDLEN, mem); CU_ASSERT(ARRLEN(resnv) == out.nvlen); assert_nv_equal(resnv, out.nva, out.nvlen, mem); nva_out_reset(&out, mem); nghttp2_bufs_free(&bufs); nghttp2_frame_headers_free(&frame.headers, mem); nghttp2_hd_inflate_free(&inflater); nghttp2_session_del(session); } void test_nghttp2_submit_response_push_response(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data ud; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; callbacks.on_frame_not_send_callback = on_frame_not_send_callback; nghttp2_session_server_new(&session, &callbacks, &ud); nghttp2_session_open_stream(session, 2, NGHTTP2_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_RESERVED, NULL); session->goaway_flags |= NGHTTP2_GOAWAY_RECV; CU_ASSERT(0 == nghttp2_submit_response(session, 2, resnv, ARRLEN(resnv), NULL)); ud.frame_not_send_cb_called = 0; CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(1 == ud.frame_not_send_cb_called); nghttp2_session_del(session); } void test_nghttp2_submit_trailer(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; accumulator acc; nghttp2_data_provider data_prd; nghttp2_outbound_item *item; my_user_data ud; nghttp2_frame frame; nghttp2_hd_inflater inflater; nva_out out; nghttp2_bufs bufs; nghttp2_mem *mem; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); data_prd.read_callback = no_end_stream_data_source_read_callback; nva_out_init(&out); acc.length = 0; ud.acc = &acc; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; CU_ASSERT(0 == nghttp2_session_server_new(&session, &callbacks, &ud)); nghttp2_hd_inflate_init(&inflater, mem); nghttp2_session_open_stream(session, 1, NGHTTP2_FLAG_END_STREAM, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); CU_ASSERT(0 == nghttp2_submit_response(session, 1, resnv, ARRLEN(resnv), &data_prd)); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(0 == nghttp2_submit_trailer(session, 1, trailernv, ARRLEN(trailernv))); session->callbacks.send_callback = accumulator_send_callback; item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_HEADERS == item->frame.hd.type); CU_ASSERT(NGHTTP2_HCAT_HEADERS == item->frame.headers.cat); CU_ASSERT(item->frame.hd.flags & NGHTTP2_FLAG_END_STREAM); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(0 == unpack_frame(&frame, acc.buf, acc.length)); nghttp2_bufs_add(&bufs, acc.buf, acc.length); inflate_hd(&inflater, &out, &bufs, NGHTTP2_FRAME_HDLEN, mem); CU_ASSERT(ARRLEN(trailernv) == out.nvlen); assert_nv_equal(trailernv, out.nva, out.nvlen, mem); nva_out_reset(&out, mem); nghttp2_bufs_free(&bufs); nghttp2_frame_headers_free(&frame.headers, mem); nghttp2_hd_inflate_free(&inflater); nghttp2_session_del(session); } void test_nghttp2_submit_headers_start_stream(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_outbound_item *item; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); CU_ASSERT(0 == nghttp2_session_client_new(&session, &callbacks, NULL)); CU_ASSERT(1 == nghttp2_submit_headers(session, NGHTTP2_FLAG_END_STREAM, -1, NULL, reqnv, ARRLEN(reqnv), NULL)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(ARRLEN(reqnv) == item->frame.headers.nvlen); assert_nv_equal(reqnv, item->frame.headers.nva, item->frame.headers.nvlen, mem); CU_ASSERT((NGHTTP2_FLAG_END_HEADERS | NGHTTP2_FLAG_END_STREAM) == item->frame.hd.flags); CU_ASSERT(0 == (item->frame.hd.flags & NGHTTP2_FLAG_PRIORITY)); nghttp2_session_del(session); } void test_nghttp2_submit_headers_reply(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data ud; nghttp2_outbound_item *item; nghttp2_stream *stream; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; callbacks.on_frame_send_callback = on_frame_send_callback; CU_ASSERT(0 == nghttp2_session_server_new(&session, &callbacks, &ud)); CU_ASSERT(0 == nghttp2_submit_headers(session, NGHTTP2_FLAG_END_STREAM, 1, NULL, resnv, ARRLEN(resnv), NULL)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(ARRLEN(resnv) == item->frame.headers.nvlen); assert_nv_equal(resnv, item->frame.headers.nva, item->frame.headers.nvlen, mem); CU_ASSERT((NGHTTP2_FLAG_END_STREAM | NGHTTP2_FLAG_END_HEADERS) == item->frame.hd.flags); ud.frame_send_cb_called = 0; ud.sent_frame_type = 0; /* The transimission will be canceled because the stream 1 is not open. */ CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(0 == ud.frame_send_cb_called); stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); CU_ASSERT(0 == nghttp2_submit_headers(session, NGHTTP2_FLAG_END_STREAM, 1, NULL, resnv, ARRLEN(resnv), NULL)); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(1 == ud.frame_send_cb_called); CU_ASSERT(NGHTTP2_HEADERS == ud.sent_frame_type); CU_ASSERT(stream->shut_flags & NGHTTP2_SHUT_WR); nghttp2_session_del(session); } void test_nghttp2_submit_headers_push_reply(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data ud; nghttp2_stream *stream; int foo; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; callbacks.on_frame_send_callback = on_frame_send_callback; CU_ASSERT(0 == nghttp2_session_server_new(&session, &callbacks, &ud)); stream = nghttp2_session_open_stream(session, 2, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_RESERVED, NULL); CU_ASSERT(0 == nghttp2_submit_headers(session, NGHTTP2_FLAG_NONE, 2, NULL, resnv, ARRLEN(resnv), &foo)); ud.frame_send_cb_called = 0; ud.sent_frame_type = 0; CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(1 == ud.frame_send_cb_called); CU_ASSERT(NGHTTP2_HEADERS == ud.sent_frame_type); CU_ASSERT(NGHTTP2_STREAM_OPENED == stream->state); CU_ASSERT(&foo == stream->stream_user_data); nghttp2_session_del(session); /* Sending HEADERS from client against stream in reserved state is error */ CU_ASSERT(0 == nghttp2_session_client_new(&session, &callbacks, &ud)); nghttp2_session_open_stream(session, 2, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_RESERVED, NULL); CU_ASSERT(0 == nghttp2_submit_headers(session, NGHTTP2_FLAG_NONE, 2, NULL, reqnv, ARRLEN(reqnv), NULL)); ud.frame_send_cb_called = 0; ud.sent_frame_type = 0; CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(0 == ud.frame_send_cb_called); nghttp2_session_del(session); } void test_nghttp2_submit_headers(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data ud; nghttp2_outbound_item *item; nghttp2_stream *stream; accumulator acc; nghttp2_frame frame; nghttp2_hd_inflater inflater; nva_out out; nghttp2_bufs bufs; nghttp2_mem *mem; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); nva_out_init(&out); acc.length = 0; ud.acc = &acc; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = accumulator_send_callback; callbacks.on_frame_send_callback = on_frame_send_callback; CU_ASSERT(0 == nghttp2_session_client_new(&session, &callbacks, &ud)); nghttp2_hd_inflate_init(&inflater, mem); CU_ASSERT(0 == nghttp2_submit_headers(session, NGHTTP2_FLAG_END_STREAM, 1, NULL, reqnv, ARRLEN(reqnv), NULL)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(ARRLEN(reqnv) == item->frame.headers.nvlen); assert_nv_equal(reqnv, item->frame.headers.nva, item->frame.headers.nvlen, mem); CU_ASSERT((NGHTTP2_FLAG_END_STREAM | NGHTTP2_FLAG_END_HEADERS) == item->frame.hd.flags); ud.frame_send_cb_called = 0; ud.sent_frame_type = 0; /* The transimission will be canceled because the stream 1 is not open. */ CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(0 == ud.frame_send_cb_called); stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); CU_ASSERT(0 == nghttp2_submit_headers(session, NGHTTP2_FLAG_END_STREAM, 1, NULL, reqnv, ARRLEN(reqnv), NULL)); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(1 == ud.frame_send_cb_called); CU_ASSERT(NGHTTP2_HEADERS == ud.sent_frame_type); CU_ASSERT(stream->shut_flags & NGHTTP2_SHUT_WR); CU_ASSERT(0 == unpack_frame(&frame, acc.buf, acc.length)); nghttp2_bufs_add(&bufs, acc.buf, acc.length); inflate_hd(&inflater, &out, &bufs, NGHTTP2_FRAME_HDLEN, mem); CU_ASSERT(ARRLEN(reqnv) == out.nvlen); assert_nv_equal(reqnv, out.nva, out.nvlen, mem); nva_out_reset(&out, mem); nghttp2_bufs_free(&bufs); nghttp2_frame_headers_free(&frame.headers, mem); nghttp2_hd_inflate_free(&inflater); nghttp2_session_del(session); } void test_nghttp2_submit_headers_continuation(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_nv nv[] = { MAKE_NV("h1", ""), MAKE_NV("h1", ""), MAKE_NV("h1", ""), MAKE_NV("h1", ""), MAKE_NV("h1", ""), MAKE_NV("h1", ""), MAKE_NV("h1", ""), }; nghttp2_outbound_item *item; uint8_t data[4096]; size_t i; my_user_data ud; memset(data, '0', sizeof(data)); for (i = 0; i < ARRLEN(nv); ++i) { nv[i].valuelen = sizeof(data); nv[i].value = data; } memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; callbacks.on_frame_send_callback = on_frame_send_callback; CU_ASSERT(0 == nghttp2_session_client_new(&session, &callbacks, &ud)); CU_ASSERT(1 == nghttp2_submit_headers(session, NGHTTP2_FLAG_END_STREAM, -1, NULL, nv, ARRLEN(nv), NULL)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_HEADERS == item->frame.hd.type); CU_ASSERT((NGHTTP2_FLAG_END_STREAM | NGHTTP2_FLAG_END_HEADERS) == item->frame.hd.flags); CU_ASSERT(0 == (item->frame.hd.flags & NGHTTP2_FLAG_PRIORITY)); ud.frame_send_cb_called = 0; CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(1 == ud.frame_send_cb_called); nghttp2_session_del(session); } void test_nghttp2_submit_priority(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_stream *stream; my_user_data ud; nghttp2_priority_spec pri_spec; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; callbacks.on_frame_send_callback = on_frame_send_callback; nghttp2_session_client_new(&session, &callbacks, &ud); stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); nghttp2_priority_spec_init(&pri_spec, 0, 3, 0); /* depends on stream 0 */ CU_ASSERT(0 == nghttp2_submit_priority(session, NGHTTP2_FLAG_NONE, 1, &pri_spec)); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(3 == stream->weight); /* submit against idle stream */ CU_ASSERT(0 == nghttp2_submit_priority(session, NGHTTP2_FLAG_NONE, 3, &pri_spec)); ud.frame_send_cb_called = 0; CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(1 == ud.frame_send_cb_called); nghttp2_session_del(session); } void test_nghttp2_submit_settings(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data ud; nghttp2_outbound_item *item; nghttp2_frame *frame; nghttp2_settings_entry iv[7]; nghttp2_frame ack_frame; const int32_t UNKNOWN_ID = 1000000007; nghttp2_mem *mem; mem = nghttp2_mem_default(); iv[0].settings_id = NGHTTP2_SETTINGS_MAX_CONCURRENT_STREAMS; iv[0].value = 5; iv[1].settings_id = NGHTTP2_SETTINGS_INITIAL_WINDOW_SIZE; iv[1].value = 16 * 1024; iv[2].settings_id = NGHTTP2_SETTINGS_MAX_CONCURRENT_STREAMS; iv[2].value = 50; iv[3].settings_id = NGHTTP2_SETTINGS_HEADER_TABLE_SIZE; iv[3].value = 111; iv[4].settings_id = UNKNOWN_ID; iv[4].value = 999; iv[5].settings_id = NGHTTP2_SETTINGS_HEADER_TABLE_SIZE; iv[5].value = 1023; iv[6].settings_id = NGHTTP2_SETTINGS_INITIAL_WINDOW_SIZE; iv[6].value = (uint32_t)NGHTTP2_MAX_WINDOW_SIZE + 1; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; callbacks.on_frame_send_callback = on_frame_send_callback; nghttp2_session_server_new(&session, &callbacks, &ud); CU_ASSERT(NGHTTP2_ERR_INVALID_ARGUMENT == nghttp2_submit_settings(session, NGHTTP2_FLAG_NONE, iv, 7)); /* Make sure that local settings are not changed */ CU_ASSERT(NGHTTP2_INITIAL_MAX_CONCURRENT_STREAMS == session->local_settings.max_concurrent_streams); CU_ASSERT(NGHTTP2_INITIAL_WINDOW_SIZE == session->local_settings.initial_window_size); /* Now sends without 6th one */ CU_ASSERT(0 == nghttp2_submit_settings(session, NGHTTP2_FLAG_NONE, iv, 6)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_SETTINGS == item->frame.hd.type); frame = &item->frame; CU_ASSERT(6 == frame->settings.niv); CU_ASSERT(5 == frame->settings.iv[0].value); CU_ASSERT(NGHTTP2_SETTINGS_MAX_CONCURRENT_STREAMS == frame->settings.iv[0].settings_id); CU_ASSERT(16 * 1024 == frame->settings.iv[1].value); CU_ASSERT(NGHTTP2_SETTINGS_INITIAL_WINDOW_SIZE == frame->settings.iv[1].settings_id); CU_ASSERT(UNKNOWN_ID == frame->settings.iv[4].settings_id); CU_ASSERT(999 == frame->settings.iv[4].value); ud.frame_send_cb_called = 0; CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(1 == ud.frame_send_cb_called); CU_ASSERT(50 == session->pending_local_max_concurrent_stream); nghttp2_frame_settings_init(&ack_frame.settings, NGHTTP2_FLAG_ACK, NULL, 0); CU_ASSERT(0 == nghttp2_session_on_settings_received(session, &ack_frame, 0)); nghttp2_frame_settings_free(&ack_frame.settings, mem); CU_ASSERT(16 * 1024 == session->local_settings.initial_window_size); CU_ASSERT(1023 == session->hd_inflater.ctx.hd_table_bufsize_max); CU_ASSERT(111 == session->hd_inflater.min_hd_table_bufsize_max); CU_ASSERT(50 == session->local_settings.max_concurrent_streams); /* We just keep the last seen value */ CU_ASSERT(50 == session->pending_local_max_concurrent_stream); nghttp2_session_del(session); } void test_nghttp2_submit_settings_update_local_window_size(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_outbound_item *item; nghttp2_settings_entry iv[4]; nghttp2_stream *stream; nghttp2_frame ack_frame; nghttp2_mem *mem; mem = nghttp2_mem_default(); nghttp2_frame_settings_init(&ack_frame.settings, NGHTTP2_FLAG_ACK, NULL, 0); iv[0].settings_id = NGHTTP2_SETTINGS_INITIAL_WINDOW_SIZE; iv[0].value = 16 * 1024; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_server_new(&session, &callbacks, NULL); stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENED, NULL); stream->local_window_size = NGHTTP2_INITIAL_WINDOW_SIZE + 100; stream->recv_window_size = 32768; nghttp2_session_open_stream(session, 3, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENED, NULL); CU_ASSERT(0 == nghttp2_submit_settings(session, NGHTTP2_FLAG_NONE, iv, 1)); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(0 == nghttp2_session_on_settings_received(session, &ack_frame, 0)); stream = nghttp2_session_get_stream(session, 1); CU_ASSERT(0 == stream->recv_window_size); CU_ASSERT(16 * 1024 + 100 == stream->local_window_size); stream = nghttp2_session_get_stream(session, 3); CU_ASSERT(16 * 1024 == stream->local_window_size); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_WINDOW_UPDATE == item->frame.hd.type); CU_ASSERT(32768 == item->frame.window_update.window_size_increment); nghttp2_session_del(session); /* Check overflow case */ iv[0].value = 128 * 1024; nghttp2_session_server_new(&session, &callbacks, NULL); stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENED, NULL); stream->local_window_size = NGHTTP2_MAX_WINDOW_SIZE; CU_ASSERT(0 == nghttp2_submit_settings(session, NGHTTP2_FLAG_NONE, iv, 1)); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(0 == nghttp2_session_on_settings_received(session, &ack_frame, 0)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_GOAWAY == item->frame.hd.type); CU_ASSERT(NGHTTP2_FLOW_CONTROL_ERROR == item->frame.goaway.error_code); nghttp2_session_del(session); nghttp2_frame_settings_free(&ack_frame.settings, mem); } void test_nghttp2_submit_settings_multiple_times(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_settings_entry iv[4]; nghttp2_frame frame; nghttp2_inflight_settings *inflight_settings; memset(&callbacks, 0, sizeof(callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_client_new(&session, &callbacks, NULL); /* first SETTINGS */ iv[0].settings_id = NGHTTP2_SETTINGS_MAX_CONCURRENT_STREAMS; iv[0].value = 100; iv[1].settings_id = NGHTTP2_SETTINGS_ENABLE_PUSH; iv[1].value = 0; CU_ASSERT(0 == nghttp2_submit_settings(session, NGHTTP2_FLAG_NONE, iv, 2)); inflight_settings = session->inflight_settings_head; CU_ASSERT(NULL != inflight_settings); CU_ASSERT(NGHTTP2_SETTINGS_MAX_CONCURRENT_STREAMS == inflight_settings->iv[0].settings_id); CU_ASSERT(100 == inflight_settings->iv[0].value); CU_ASSERT(2 == inflight_settings->niv); CU_ASSERT(NULL == inflight_settings->next); CU_ASSERT(100 == session->pending_local_max_concurrent_stream); CU_ASSERT(0 == session->pending_enable_push); /* second SETTINGS */ iv[0].settings_id = NGHTTP2_SETTINGS_MAX_CONCURRENT_STREAMS; iv[0].value = 99; CU_ASSERT(0 == nghttp2_submit_settings(session, NGHTTP2_FLAG_NONE, iv, 1)); inflight_settings = session->inflight_settings_head->next; CU_ASSERT(NULL != inflight_settings); CU_ASSERT(NGHTTP2_SETTINGS_MAX_CONCURRENT_STREAMS == inflight_settings->iv[0].settings_id); CU_ASSERT(99 == inflight_settings->iv[0].value); CU_ASSERT(1 == inflight_settings->niv); CU_ASSERT(NULL == inflight_settings->next); CU_ASSERT(99 == session->pending_local_max_concurrent_stream); CU_ASSERT(0 == session->pending_enable_push); nghttp2_frame_settings_init(&frame.settings, NGHTTP2_FLAG_ACK, NULL, 0); /* receive SETTINGS ACK */ CU_ASSERT(0 == nghttp2_session_on_settings_received(session, &frame, 0)); inflight_settings = session->inflight_settings_head; /* first inflight SETTINGS was removed */ CU_ASSERT(NULL != inflight_settings); CU_ASSERT(NGHTTP2_SETTINGS_MAX_CONCURRENT_STREAMS == inflight_settings->iv[0].settings_id); CU_ASSERT(99 == inflight_settings->iv[0].value); CU_ASSERT(1 == inflight_settings->niv); CU_ASSERT(NULL == inflight_settings->next); CU_ASSERT(100 == session->local_settings.max_concurrent_streams); /* receive SETTINGS ACK again */ CU_ASSERT(0 == nghttp2_session_on_settings_received(session, &frame, 0)); CU_ASSERT(NULL == session->inflight_settings_head); CU_ASSERT(99 == session->local_settings.max_concurrent_streams); nghttp2_session_del(session); } void test_nghttp2_submit_push_promise(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data ud; nghttp2_stream *stream; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; callbacks.on_frame_send_callback = on_frame_send_callback; callbacks.on_frame_not_send_callback = on_frame_not_send_callback; CU_ASSERT(0 == nghttp2_session_server_new(&session, &callbacks, &ud)); nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); CU_ASSERT(2 == nghttp2_submit_push_promise(session, NGHTTP2_FLAG_NONE, 1, reqnv, ARRLEN(reqnv), &ud)); stream = nghttp2_session_get_stream(session, 2); CU_ASSERT(NULL != stream); CU_ASSERT(NGHTTP2_STREAM_RESERVED == stream->state); CU_ASSERT(&ud == nghttp2_session_get_stream_user_data(session, 2)); ud.frame_send_cb_called = 0; ud.sent_frame_type = 0; CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(1 == ud.frame_send_cb_called); CU_ASSERT(NGHTTP2_PUSH_PROMISE == ud.sent_frame_type); stream = nghttp2_session_get_stream(session, 2); CU_ASSERT(NGHTTP2_STREAM_RESERVED == stream->state); CU_ASSERT(&ud == nghttp2_session_get_stream_user_data(session, 2)); /* submit PUSH_PROMISE while associated stream is not opened */ CU_ASSERT(NGHTTP2_ERR_STREAM_CLOSED == nghttp2_submit_push_promise(session, NGHTTP2_FLAG_NONE, 3, reqnv, ARRLEN(reqnv), &ud)); nghttp2_session_del(session); } void test_nghttp2_submit_window_update(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data ud; nghttp2_outbound_item *item; nghttp2_stream *stream; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_client_new(&session, &callbacks, &ud); stream = nghttp2_session_open_stream(session, 2, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENED, NULL); stream->recv_window_size = 4096; CU_ASSERT(0 == nghttp2_submit_window_update(session, NGHTTP2_FLAG_NONE, 2, 1024)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_WINDOW_UPDATE == item->frame.hd.type); CU_ASSERT(1024 == item->frame.window_update.window_size_increment); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(3072 == stream->recv_window_size); CU_ASSERT(0 == nghttp2_submit_window_update(session, NGHTTP2_FLAG_NONE, 2, 4096)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_WINDOW_UPDATE == item->frame.hd.type); CU_ASSERT(4096 == item->frame.window_update.window_size_increment); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(0 == stream->recv_window_size); CU_ASSERT(0 == nghttp2_submit_window_update(session, NGHTTP2_FLAG_NONE, 2, 4096)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_WINDOW_UPDATE == item->frame.hd.type); CU_ASSERT(4096 == item->frame.window_update.window_size_increment); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(0 == stream->recv_window_size); CU_ASSERT(0 == nghttp2_submit_window_update(session, NGHTTP2_FLAG_NONE, 2, 0)); /* It is ok if stream is closed or does not exist at the call time */ CU_ASSERT(0 == nghttp2_submit_window_update(session, NGHTTP2_FLAG_NONE, 4, 4096)); nghttp2_session_del(session); } void test_nghttp2_submit_window_update_local_window_size(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_outbound_item *item; nghttp2_stream *stream; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_client_new(&session, &callbacks, NULL); stream = nghttp2_session_open_stream(session, 2, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENED, NULL); stream->recv_window_size = 4096; CU_ASSERT(0 == nghttp2_submit_window_update(session, NGHTTP2_FLAG_NONE, 2, stream->recv_window_size + 1)); CU_ASSERT(NGHTTP2_INITIAL_WINDOW_SIZE + 1 == stream->local_window_size); CU_ASSERT(0 == stream->recv_window_size); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_WINDOW_UPDATE == item->frame.hd.type); CU_ASSERT(4097 == item->frame.window_update.window_size_increment); CU_ASSERT(0 == nghttp2_session_send(session)); /* Let's decrement local window size */ stream->recv_window_size = 4096; CU_ASSERT(0 == nghttp2_submit_window_update(session, NGHTTP2_FLAG_NONE, 2, -stream->local_window_size / 2)); CU_ASSERT(32768 == stream->local_window_size); CU_ASSERT(-28672 == stream->recv_window_size); CU_ASSERT(32768 == stream->recv_reduction); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(item == NULL); /* Increase local window size */ CU_ASSERT(0 == nghttp2_submit_window_update(session, NGHTTP2_FLAG_NONE, 2, 16384)); CU_ASSERT(49152 == stream->local_window_size); CU_ASSERT(-12288 == stream->recv_window_size); CU_ASSERT(16384 == stream->recv_reduction); CU_ASSERT(NULL == nghttp2_session_get_next_ob_item(session)); CU_ASSERT(NGHTTP2_ERR_FLOW_CONTROL == nghttp2_submit_window_update(session, NGHTTP2_FLAG_NONE, 2, NGHTTP2_MAX_WINDOW_SIZE)); CU_ASSERT(0 == nghttp2_session_send(session)); /* Check connection-level flow control */ session->recv_window_size = 4096; CU_ASSERT(0 == nghttp2_submit_window_update(session, NGHTTP2_FLAG_NONE, 0, session->recv_window_size + 1)); CU_ASSERT(NGHTTP2_INITIAL_CONNECTION_WINDOW_SIZE + 1 == session->local_window_size); CU_ASSERT(0 == session->recv_window_size); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_WINDOW_UPDATE == item->frame.hd.type); CU_ASSERT(4097 == item->frame.window_update.window_size_increment); CU_ASSERT(0 == nghttp2_session_send(session)); /* Go decrement part */ session->recv_window_size = 4096; CU_ASSERT(0 == nghttp2_submit_window_update(session, NGHTTP2_FLAG_NONE, 0, -session->local_window_size / 2)); CU_ASSERT(32768 == session->local_window_size); CU_ASSERT(-28672 == session->recv_window_size); CU_ASSERT(32768 == session->recv_reduction); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(item == NULL); /* Increase local window size */ CU_ASSERT(0 == nghttp2_submit_window_update(session, NGHTTP2_FLAG_NONE, 0, 16384)); CU_ASSERT(49152 == session->local_window_size); CU_ASSERT(-12288 == session->recv_window_size); CU_ASSERT(16384 == session->recv_reduction); CU_ASSERT(NULL == nghttp2_session_get_next_ob_item(session)); CU_ASSERT(NGHTTP2_ERR_FLOW_CONTROL == nghttp2_submit_window_update(session, NGHTTP2_FLAG_NONE, 0, NGHTTP2_MAX_WINDOW_SIZE)); nghttp2_session_del(session); } void test_nghttp2_submit_shutdown_notice(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data ud; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; callbacks.on_frame_send_callback = on_frame_send_callback; callbacks.on_frame_not_send_callback = on_frame_not_send_callback; nghttp2_session_server_new(&session, &callbacks, &ud); CU_ASSERT(0 == nghttp2_submit_shutdown_notice(session)); ud.frame_send_cb_called = 0; nghttp2_session_send(session); CU_ASSERT(1 == ud.frame_send_cb_called); CU_ASSERT(NGHTTP2_GOAWAY == ud.sent_frame_type); CU_ASSERT((1u << 31) - 1 == session->local_last_stream_id); /* After another GOAWAY, nghttp2_submit_shutdown_notice() is noop. */ CU_ASSERT(0 == nghttp2_session_terminate_session(session, NGHTTP2_NO_ERROR)); ud.frame_send_cb_called = 0; nghttp2_session_send(session); CU_ASSERT(1 == ud.frame_send_cb_called); CU_ASSERT(NGHTTP2_GOAWAY == ud.sent_frame_type); CU_ASSERT(0 == session->local_last_stream_id); CU_ASSERT(0 == nghttp2_submit_shutdown_notice(session)); ud.frame_send_cb_called = 0; ud.frame_not_send_cb_called = 0; nghttp2_session_send(session); CU_ASSERT(0 == ud.frame_send_cb_called); CU_ASSERT(0 == ud.frame_not_send_cb_called); nghttp2_session_del(session); /* Using nghttp2_submit_shutdown_notice() with client side session is error */ nghttp2_session_client_new(&session, &callbacks, NULL); CU_ASSERT(NGHTTP2_ERR_INVALID_STATE == nghttp2_submit_shutdown_notice(session)); nghttp2_session_del(session); } void test_nghttp2_submit_invalid_nv(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_nv empty_name_nv[] = {MAKE_NV("Version", "HTTP/1.1"), MAKE_NV("", "empty name")}; /* Now invalid header name/value pair in HTTP/1.1 is accepted in nghttp2 */ memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); CU_ASSERT(0 == nghttp2_session_server_new(&session, &callbacks, NULL)); /* nghttp2_submit_request */ CU_ASSERT(0 < nghttp2_submit_request(session, NULL, empty_name_nv, ARRLEN(empty_name_nv), NULL, NULL)); /* nghttp2_submit_response */ CU_ASSERT(0 == nghttp2_submit_response(session, 2, empty_name_nv, ARRLEN(empty_name_nv), NULL)); /* nghttp2_submit_headers */ CU_ASSERT(0 < nghttp2_submit_headers(session, NGHTTP2_FLAG_NONE, -1, NULL, empty_name_nv, ARRLEN(empty_name_nv), NULL)); /* nghttp2_submit_push_promise */ open_stream(session, 1); CU_ASSERT(0 < nghttp2_submit_push_promise(session, NGHTTP2_FLAG_NONE, 1, empty_name_nv, ARRLEN(empty_name_nv), NULL)); nghttp2_session_del(session); } void test_nghttp2_session_open_stream(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_stream *stream; nghttp2_priority_spec pri_spec; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); nghttp2_session_server_new(&session, &callbacks, NULL); nghttp2_priority_spec_init(&pri_spec, 0, 245, 0); stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec, NGHTTP2_STREAM_OPENED, NULL); CU_ASSERT(1 == session->num_incoming_streams); CU_ASSERT(0 == session->num_outgoing_streams); CU_ASSERT(NGHTTP2_STREAM_OPENED == stream->state); CU_ASSERT(245 == stream->weight); CU_ASSERT(&session->root == stream->dep_prev); CU_ASSERT(NGHTTP2_SHUT_NONE == stream->shut_flags); stream = nghttp2_session_open_stream(session, 2, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); CU_ASSERT(1 == session->num_incoming_streams); CU_ASSERT(1 == session->num_outgoing_streams); CU_ASSERT(&session->root == stream->dep_prev); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == stream->weight); CU_ASSERT(NGHTTP2_SHUT_NONE == stream->shut_flags); stream = nghttp2_session_open_stream(session, 4, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_RESERVED, NULL); CU_ASSERT(1 == session->num_incoming_streams); CU_ASSERT(1 == session->num_outgoing_streams); CU_ASSERT(&session->root == stream->dep_prev); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == stream->weight); CU_ASSERT(NGHTTP2_SHUT_RD == stream->shut_flags); nghttp2_priority_spec_init(&pri_spec, 1, 17, 1); stream = nghttp2_session_open_stream(session, 3, NGHTTP2_STREAM_FLAG_NONE, &pri_spec, NGHTTP2_STREAM_OPENED, NULL); CU_ASSERT(17 == stream->weight); CU_ASSERT(1 == stream->dep_prev->stream_id); /* Dependency to idle stream */ nghttp2_priority_spec_init(&pri_spec, 1000000007, 240, 1); stream = nghttp2_session_open_stream(session, 5, NGHTTP2_STREAM_FLAG_NONE, &pri_spec, NGHTTP2_STREAM_OPENED, NULL); CU_ASSERT(240 == stream->weight); CU_ASSERT(1000000007 == stream->dep_prev->stream_id); stream = nghttp2_session_get_stream_raw(session, 1000000007); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == stream->weight); CU_ASSERT(&session->root == stream->dep_prev); /* Dependency to closed stream which is not in dependency tree */ session->last_recv_stream_id = 7; nghttp2_priority_spec_init(&pri_spec, 7, 10, 0); stream = nghttp2_session_open_stream(session, 9, NGHTTP2_FLAG_NONE, &pri_spec, NGHTTP2_STREAM_OPENED, NULL); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == stream->weight); nghttp2_session_del(session); nghttp2_session_client_new(&session, &callbacks, NULL); stream = nghttp2_session_open_stream(session, 4, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_RESERVED, NULL); CU_ASSERT(0 == session->num_incoming_streams); CU_ASSERT(0 == session->num_outgoing_streams); CU_ASSERT(&session->root == stream->dep_prev); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == stream->weight); CU_ASSERT(NGHTTP2_SHUT_WR == stream->shut_flags); nghttp2_session_del(session); } void test_nghttp2_session_open_stream_with_idle_stream_dep(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_stream *stream; nghttp2_priority_spec pri_spec; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); nghttp2_session_server_new(&session, &callbacks, NULL); /* Dependency to idle stream */ nghttp2_priority_spec_init(&pri_spec, 101, 245, 0); stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec, NGHTTP2_STREAM_OPENED, NULL); CU_ASSERT(245 == stream->weight); CU_ASSERT(101 == stream->dep_prev->stream_id); stream = nghttp2_session_get_stream_raw(session, 101); CU_ASSERT(NGHTTP2_STREAM_IDLE == stream->state); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == stream->weight); nghttp2_priority_spec_init(&pri_spec, 211, 1, 0); /* stream 101 was already created as idle. */ stream = nghttp2_session_open_stream(session, 101, NGHTTP2_STREAM_FLAG_NONE, &pri_spec, NGHTTP2_STREAM_OPENED, NULL); CU_ASSERT(1 == stream->weight); CU_ASSERT(211 == stream->dep_prev->stream_id); stream = nghttp2_session_get_stream_raw(session, 211); CU_ASSERT(NGHTTP2_STREAM_IDLE == stream->state); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == stream->weight); nghttp2_session_del(session); } void test_nghttp2_session_get_next_ob_item(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_priority_spec pri_spec; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_server_new(&session, &callbacks, NULL); session->remote_settings.max_concurrent_streams = 2; CU_ASSERT(NULL == nghttp2_session_get_next_ob_item(session)); nghttp2_submit_ping(session, NGHTTP2_FLAG_NONE, NULL); CU_ASSERT(NGHTTP2_PING == nghttp2_session_get_next_ob_item(session)->frame.hd.type); nghttp2_submit_request(session, NULL, NULL, 0, NULL, NULL); CU_ASSERT(NGHTTP2_PING == nghttp2_session_get_next_ob_item(session)->frame.hd.type); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(NULL == nghttp2_session_get_next_ob_item(session)); /* Incoming stream does not affect the number of outgoing max concurrent streams. */ nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); nghttp2_priority_spec_init(&pri_spec, 0, NGHTTP2_MAX_WEIGHT, 0); nghttp2_submit_request(session, &pri_spec, NULL, 0, NULL, NULL); CU_ASSERT(NGHTTP2_HEADERS == nghttp2_session_get_next_ob_item(session)->frame.hd.type); CU_ASSERT(0 == nghttp2_session_send(session)); nghttp2_submit_request(session, &pri_spec, NULL, 0, NULL, NULL); CU_ASSERT(NULL == nghttp2_session_get_next_ob_item(session)); session->remote_settings.max_concurrent_streams = 3; CU_ASSERT(NGHTTP2_HEADERS == nghttp2_session_get_next_ob_item(session)->frame.hd.type); nghttp2_session_del(session); } void test_nghttp2_session_pop_next_ob_item(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_outbound_item *item; nghttp2_priority_spec pri_spec; nghttp2_stream *stream; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_server_new(&session, &callbacks, NULL); session->remote_settings.max_concurrent_streams = 1; CU_ASSERT(NULL == nghttp2_session_pop_next_ob_item(session)); nghttp2_submit_ping(session, NGHTTP2_FLAG_NONE, NULL); nghttp2_priority_spec_init(&pri_spec, 0, 254, 0); nghttp2_submit_request(session, &pri_spec, NULL, 0, NULL, NULL); item = nghttp2_session_pop_next_ob_item(session); CU_ASSERT(NGHTTP2_PING == item->frame.hd.type); nghttp2_outbound_item_free(item, mem); mem->free(item, NULL); item = nghttp2_session_pop_next_ob_item(session); CU_ASSERT(NGHTTP2_HEADERS == item->frame.hd.type); nghttp2_outbound_item_free(item, mem); mem->free(item, NULL); CU_ASSERT(NULL == nghttp2_session_pop_next_ob_item(session)); /* Incoming stream does not affect the number of outgoing max concurrent streams. */ nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); /* In-flight outgoing stream */ nghttp2_session_open_stream(session, 4, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); nghttp2_priority_spec_init(&pri_spec, 0, NGHTTP2_MAX_WEIGHT, 0); nghttp2_submit_request(session, &pri_spec, NULL, 0, NULL, NULL); nghttp2_submit_response(session, 1, NULL, 0, NULL); item = nghttp2_session_pop_next_ob_item(session); CU_ASSERT(NGHTTP2_HEADERS == item->frame.hd.type); CU_ASSERT(1 == item->frame.hd.stream_id); stream = nghttp2_session_get_stream(session, 1); nghttp2_stream_detach_item(stream); nghttp2_outbound_item_free(item, mem); mem->free(item, NULL); CU_ASSERT(NULL == nghttp2_session_pop_next_ob_item(session)); session->remote_settings.max_concurrent_streams = 2; item = nghttp2_session_pop_next_ob_item(session); CU_ASSERT(NGHTTP2_HEADERS == item->frame.hd.type); nghttp2_outbound_item_free(item, mem); mem->free(item, NULL); nghttp2_session_del(session); /* Check that push reply HEADERS are queued into ob_ss_pq */ nghttp2_session_server_new(&session, &callbacks, NULL); session->remote_settings.max_concurrent_streams = 0; nghttp2_session_open_stream(session, 2, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_RESERVED, NULL); CU_ASSERT(0 == nghttp2_submit_headers(session, NGHTTP2_FLAG_END_STREAM, 2, NULL, NULL, 0, NULL)); CU_ASSERT(NULL == nghttp2_session_pop_next_ob_item(session)); CU_ASSERT(1 == nghttp2_outbound_queue_size(&session->ob_syn)); nghttp2_session_del(session); } void test_nghttp2_session_reply_fail(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_data_provider data_prd; my_user_data ud; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = fail_send_callback; data_prd.read_callback = fixed_length_data_source_read_callback; ud.data_source_length = 4 * 1024; CU_ASSERT(0 == nghttp2_session_server_new(&session, &callbacks, &ud)); nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); CU_ASSERT(0 == nghttp2_submit_response(session, 1, NULL, 0, &data_prd)); CU_ASSERT(NGHTTP2_ERR_CALLBACK_FAILURE == nghttp2_session_send(session)); nghttp2_session_del(session); } void test_nghttp2_session_max_concurrent_streams(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_frame frame; nghttp2_outbound_item *item; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_server_new(&session, &callbacks, NULL); nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENED, NULL); /* Check un-ACKed SETTINGS_MAX_CONCURRENT_STREAMS */ nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS, 3, NGHTTP2_HCAT_HEADERS, NULL, NULL, 0); session->pending_local_max_concurrent_stream = 1; CU_ASSERT(NGHTTP2_ERR_IGN_HEADER_BLOCK == nghttp2_session_on_request_headers_received(session, &frame)); item = nghttp2_outbound_queue_top(&session->ob_reg); CU_ASSERT(NGHTTP2_RST_STREAM == item->frame.hd.type); CU_ASSERT(NGHTTP2_REFUSED_STREAM == item->frame.rst_stream.error_code); CU_ASSERT(0 == nghttp2_session_send(session)); /* Check ACKed SETTINGS_MAX_CONCURRENT_STREAMS */ session->local_settings.max_concurrent_streams = 1; frame.hd.stream_id = 5; CU_ASSERT(NGHTTP2_ERR_IGN_HEADER_BLOCK == nghttp2_session_on_request_headers_received(session, &frame)); item = nghttp2_outbound_queue_top(&session->ob_reg); CU_ASSERT(NGHTTP2_GOAWAY == item->frame.hd.type); CU_ASSERT(NGHTTP2_PROTOCOL_ERROR == item->frame.goaway.error_code); nghttp2_frame_headers_free(&frame.headers, mem); nghttp2_session_del(session); } /* * Check that on_stream_close_callback is called when server pushed * HEADERS have NGHTTP2_FLAG_END_STREAM. */ void test_nghttp2_session_stream_close_on_headers_push(void) { /* nghttp2_session *session; */ /* nghttp2_session_callbacks callbacks; */ /* const char *nv[] = { NULL }; */ /* my_user_data ud; */ /* nghttp2_frame frame; */ /* memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); */ /* callbacks.on_stream_close_callback = */ /* no_stream_user_data_stream_close_callback; */ /* ud.stream_close_cb_called = 0; */ /* nghttp2_session_client_new(&session, NGHTTP2_PROTO_SPDY2, &callbacks, &ud); */ /* nghttp2_session_open_stream(session, 1, NGHTTP2_CTRL_FLAG_NONE, 3, */ /* NGHTTP2_STREAM_OPENING, NULL); */ /* nghttp2_frame_syn_stream_init(&frame.syn_stream, NGHTTP2_PROTO_SPDY2, */ /* NGHTTP2_CTRL_FLAG_FIN | */ /* NGHTTP2_CTRL_FLAG_UNIDIRECTIONAL, */ /* 2, 1, 3, dup_nv(nv)); */ /* CU_ASSERT(0 == nghttp2_session_on_request_headers_received(session, * &frame)); */ /* nghttp2_frame_syn_stream_free(&frame.syn_stream); */ /* nghttp2_session_del(session); */ } void test_nghttp2_session_stop_data_with_rst_stream(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data ud; nghttp2_data_provider data_prd; nghttp2_frame frame; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.on_frame_send_callback = on_frame_send_callback; callbacks.send_callback = block_count_send_callback; data_prd.read_callback = fixed_length_data_source_read_callback; ud.frame_send_cb_called = 0; ud.data_source_length = NGHTTP2_DATA_PAYLOADLEN * 4; nghttp2_session_server_new(&session, &callbacks, &ud); nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); nghttp2_submit_response(session, 1, NULL, 0, &data_prd); ud.block_count = 2; /* Sends response HEADERS + DATA[0] */ CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(NGHTTP2_DATA == ud.sent_frame_type); /* data for DATA[1] is read from data_prd but it is not sent */ CU_ASSERT(ud.data_source_length == NGHTTP2_DATA_PAYLOADLEN * 2); nghttp2_frame_rst_stream_init(&frame.rst_stream, 1, NGHTTP2_CANCEL); CU_ASSERT(0 == nghttp2_session_on_rst_stream_received(session, &frame)); nghttp2_frame_rst_stream_free(&frame.rst_stream); /* Big enough number to send all DATA frames potentially. */ ud.block_count = 100; /* Nothing will be sent in the following call. */ CU_ASSERT(0 == nghttp2_session_send(session)); /* With RST_STREAM, stream is canceled and further DATA on that stream are not sent. */ CU_ASSERT(ud.data_source_length == NGHTTP2_DATA_PAYLOADLEN * 2); CU_ASSERT(NULL == nghttp2_session_get_stream(session, 1)); nghttp2_session_del(session); } void test_nghttp2_session_defer_data(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data ud; nghttp2_data_provider data_prd; nghttp2_outbound_item *item; nghttp2_stream *stream; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.on_frame_send_callback = on_frame_send_callback; callbacks.send_callback = block_count_send_callback; data_prd.read_callback = defer_data_source_read_callback; ud.frame_send_cb_called = 0; ud.data_source_length = NGHTTP2_DATA_PAYLOADLEN * 4; nghttp2_session_server_new(&session, &callbacks, &ud); stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); session->remote_window_size = 1 << 20; stream->remote_window_size = 1 << 20; nghttp2_submit_response(session, 1, NULL, 0, &data_prd); ud.block_count = 1; /* Sends HEADERS reply */ CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(NGHTTP2_HEADERS == ud.sent_frame_type); /* No data is read */ CU_ASSERT(ud.data_source_length == NGHTTP2_DATA_PAYLOADLEN * 4); ud.block_count = 1; nghttp2_submit_ping(session, NGHTTP2_FLAG_NONE, NULL); /* Sends PING */ CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(NGHTTP2_PING == ud.sent_frame_type); /* Resume deferred DATA */ CU_ASSERT(0 == nghttp2_session_resume_data(session, 1)); item = stream->item; item->aux_data.data.data_prd.read_callback = fixed_length_data_source_read_callback; ud.block_count = 1; /* Reads 2 DATA chunks */ CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(ud.data_source_length == NGHTTP2_DATA_PAYLOADLEN * 2); /* Deferred again */ item->aux_data.data.data_prd.read_callback = defer_data_source_read_callback; /* This is needed since 16KiB block is already read and waiting to be sent. No read_callback invocation. */ ud.block_count = 1; CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(ud.data_source_length == NGHTTP2_DATA_PAYLOADLEN * 2); /* Resume deferred DATA */ CU_ASSERT(0 == nghttp2_session_resume_data(session, 1)); item->aux_data.data.data_prd.read_callback = fixed_length_data_source_read_callback; ud.block_count = 1; /* Reads 2 16KiB blocks */ CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(ud.data_source_length == 0); nghttp2_session_del(session); } void test_nghttp2_session_flow_control(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data ud; nghttp2_data_provider data_prd; nghttp2_frame frame; nghttp2_stream *stream; int32_t new_initial_window_size; nghttp2_settings_entry iv[1]; nghttp2_frame settings_frame; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = fixed_bytes_send_callback; callbacks.on_frame_send_callback = on_frame_send_callback; data_prd.read_callback = fixed_length_data_source_read_callback; ud.frame_send_cb_called = 0; ud.data_source_length = 128 * 1024; /* Use smaller emission count so that we can check outbound flow control window calculation is correct. */ ud.fixed_sendlen = 2 * 1024; /* Initial window size to 64KiB - 1*/ nghttp2_session_client_new(&session, &callbacks, &ud); /* Change it to 64KiB for easy calculation */ session->remote_window_size = 64 * 1024; session->remote_settings.initial_window_size = 64 * 1024; nghttp2_submit_request(session, NULL, NULL, 0, &data_prd, NULL); /* Sends 64KiB - 1 data */ CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(64 * 1024 == ud.data_source_length); /* Back 32KiB in stream window */ nghttp2_frame_window_update_init(&frame.window_update, NGHTTP2_FLAG_NONE, 1, 32 * 1024); nghttp2_session_on_window_update_received(session, &frame); /* Send nothing because of connection-level window */ CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(64 * 1024 == ud.data_source_length); /* Back 32KiB in connection-level window */ frame.hd.stream_id = 0; nghttp2_session_on_window_update_received(session, &frame); /* Sends another 32KiB data */ CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(32 * 1024 == ud.data_source_length); stream = nghttp2_session_get_stream(session, 1); /* Change initial window size to 16KiB. The window_size becomes negative. */ new_initial_window_size = 16 * 1024; stream->remote_window_size = new_initial_window_size - ((int32_t)session->remote_settings.initial_window_size - stream->remote_window_size); session->remote_settings.initial_window_size = (uint32_t)new_initial_window_size; CU_ASSERT(-48 * 1024 == stream->remote_window_size); /* Back 48KiB to stream window */ frame.hd.stream_id = 1; frame.window_update.window_size_increment = 48 * 1024; nghttp2_session_on_window_update_received(session, &frame); /* Nothing is sent because window_size is 0 */ CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(32 * 1024 == ud.data_source_length); /* Back 16KiB in stream window */ frame.hd.stream_id = 1; frame.window_update.window_size_increment = 16 * 1024; nghttp2_session_on_window_update_received(session, &frame); /* Back 24KiB in connection-level window */ frame.hd.stream_id = 0; frame.window_update.window_size_increment = 24 * 1024; nghttp2_session_on_window_update_received(session, &frame); /* Sends another 16KiB data */ CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(16 * 1024 == ud.data_source_length); /* Increase initial window size to 32KiB */ iv[0].settings_id = NGHTTP2_SETTINGS_INITIAL_WINDOW_SIZE; iv[0].value = 32 * 1024; nghttp2_frame_settings_init(&settings_frame.settings, NGHTTP2_FLAG_NONE, dup_iv(iv, 1), 1); nghttp2_session_on_settings_received(session, &settings_frame, 1); nghttp2_frame_settings_free(&settings_frame.settings, mem); /* Sends another 8KiB data */ CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(8 * 1024 == ud.data_source_length); /* Back 8KiB in connection-level window */ frame.hd.stream_id = 0; frame.window_update.window_size_increment = 8 * 1024; nghttp2_session_on_window_update_received(session, &frame); /* Sends last 8KiB data */ CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(0 == ud.data_source_length); CU_ASSERT(nghttp2_session_get_stream(session, 1)->shut_flags & NGHTTP2_SHUT_WR); nghttp2_frame_window_update_free(&frame.window_update); nghttp2_session_del(session); } void test_nghttp2_session_flow_control_data_recv(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; uint8_t data[64 * 1024 + 16]; nghttp2_frame_hd hd; nghttp2_outbound_item *item; nghttp2_stream *stream; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; /* Initial window size to 64KiB - 1*/ nghttp2_session_client_new(&session, &callbacks, NULL); stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENED, NULL); session->next_stream_id = 3; nghttp2_stream_shutdown(stream, NGHTTP2_SHUT_WR); session->local_window_size = NGHTTP2_MAX_PAYLOADLEN; stream->local_window_size = NGHTTP2_MAX_PAYLOADLEN; /* Create DATA frame */ memset(data, 0, sizeof(data)); nghttp2_frame_hd_init(&hd, NGHTTP2_MAX_PAYLOADLEN, NGHTTP2_DATA, NGHTTP2_FLAG_END_STREAM, 1); nghttp2_frame_pack_frame_hd(data, &hd); CU_ASSERT(NGHTTP2_MAX_PAYLOADLEN + NGHTTP2_FRAME_HDLEN == nghttp2_session_mem_recv(session, data, NGHTTP2_MAX_PAYLOADLEN + NGHTTP2_FRAME_HDLEN)); item = nghttp2_session_get_next_ob_item(session); /* Since this is the last frame, stream-level WINDOW_UPDATE is not issued, but connection-level is. */ CU_ASSERT(NGHTTP2_WINDOW_UPDATE == item->frame.hd.type); CU_ASSERT(0 == item->frame.hd.stream_id); CU_ASSERT(NGHTTP2_MAX_PAYLOADLEN == item->frame.window_update.window_size_increment); CU_ASSERT(0 == nghttp2_session_send(session)); /* Receive DATA for closed stream. They are still subject to under connection-level flow control, since this situation arises when RST_STREAM is issued by the remote, but the local side keeps sending DATA frames. Without calculating connection-level window, the subsequent flow control gets confused. */ CU_ASSERT(NGHTTP2_MAX_PAYLOADLEN + NGHTTP2_FRAME_HDLEN == nghttp2_session_mem_recv(session, data, NGHTTP2_MAX_PAYLOADLEN + NGHTTP2_FRAME_HDLEN)); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_WINDOW_UPDATE == item->frame.hd.type); CU_ASSERT(0 == item->frame.hd.stream_id); CU_ASSERT(NGHTTP2_MAX_PAYLOADLEN == item->frame.window_update.window_size_increment); nghttp2_session_del(session); } void test_nghttp2_session_flow_control_data_with_padding_recv(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; uint8_t data[1024]; nghttp2_frame_hd hd; nghttp2_stream *stream; nghttp2_option *option; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; nghttp2_option_new(&option); /* Disable auto window update so that we can check padding is consumed automatically */ nghttp2_option_set_no_auto_window_update(option, 1); /* Initial window size to 64KiB - 1*/ nghttp2_session_client_new2(&session, &callbacks, NULL, option); nghttp2_option_del(option); stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENED, NULL); /* Create DATA frame */ memset(data, 0, sizeof(data)); nghttp2_frame_hd_init(&hd, 357, NGHTTP2_DATA, NGHTTP2_FLAG_PADDED, 1); nghttp2_frame_pack_frame_hd(data, &hd); /* Set Pad Length field, which itself is padding */ data[NGHTTP2_FRAME_HDLEN] = 255; CU_ASSERT( (ssize_t)(NGHTTP2_FRAME_HDLEN + hd.length) == nghttp2_session_mem_recv(session, data, NGHTTP2_FRAME_HDLEN + hd.length)); CU_ASSERT((int32_t)hd.length == session->recv_window_size); CU_ASSERT((int32_t)hd.length == stream->recv_window_size); CU_ASSERT(256 == session->consumed_size); CU_ASSERT(256 == stream->consumed_size); CU_ASSERT(357 == session->recv_window_size); CU_ASSERT(357 == stream->recv_window_size); /* Receive the same DATA frame, but in 2 parts: first 9 + 1 + 102 bytes which includes 1st padding byte, and remainder */ CU_ASSERT((ssize_t)(NGHTTP2_FRAME_HDLEN + 103) == nghttp2_session_mem_recv(session, data, NGHTTP2_FRAME_HDLEN + 103)); CU_ASSERT(258 == session->consumed_size); CU_ASSERT(258 == stream->consumed_size); CU_ASSERT(460 == session->recv_window_size); CU_ASSERT(460 == stream->recv_window_size); /* 357 - 103 = 254 bytes left */ CU_ASSERT(254 == nghttp2_session_mem_recv(session, data, 254)); CU_ASSERT(512 == session->consumed_size); CU_ASSERT(512 == stream->consumed_size); CU_ASSERT(714 == session->recv_window_size); CU_ASSERT(714 == stream->recv_window_size); /* Receive the same DATA frame, but in 2 parts: first 9 = 1 + 101 bytes which only includes data without padding, 2nd part is padding only */ CU_ASSERT((ssize_t)(NGHTTP2_FRAME_HDLEN + 102) == nghttp2_session_mem_recv(session, data, NGHTTP2_FRAME_HDLEN + 102)); CU_ASSERT(513 == session->consumed_size); CU_ASSERT(513 == stream->consumed_size); CU_ASSERT(816 == session->recv_window_size); CU_ASSERT(816 == stream->recv_window_size); /* 357 - 102 = 255 bytes left */ CU_ASSERT(255 == nghttp2_session_mem_recv(session, data, 255)); CU_ASSERT(768 == session->consumed_size); CU_ASSERT(768 == stream->consumed_size); CU_ASSERT(1071 == session->recv_window_size); CU_ASSERT(1071 == stream->recv_window_size); /* Receive the same DATA frame, but in 2 parts: first 9 = 1 + 50 bytes which includes byte up to middle of data, 2nd part is the remainder */ CU_ASSERT((ssize_t)(NGHTTP2_FRAME_HDLEN + 51) == nghttp2_session_mem_recv(session, data, NGHTTP2_FRAME_HDLEN + 51)); CU_ASSERT(769 == session->consumed_size); CU_ASSERT(769 == stream->consumed_size); CU_ASSERT(1122 == session->recv_window_size); CU_ASSERT(1122 == stream->recv_window_size); /* 357 - 51 = 306 bytes left */ CU_ASSERT(306 == nghttp2_session_mem_recv(session, data, 306)); CU_ASSERT(1024 == session->consumed_size); CU_ASSERT(1024 == stream->consumed_size); CU_ASSERT(1428 == session->recv_window_size); CU_ASSERT(1428 == stream->recv_window_size); nghttp2_session_del(session); } void test_nghttp2_session_data_read_temporal_failure(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data ud; nghttp2_data_provider data_prd; nghttp2_frame frame; nghttp2_stream *stream; size_t data_size = 128 * 1024; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; callbacks.on_frame_send_callback = on_frame_send_callback; data_prd.read_callback = fixed_length_data_source_read_callback; ud.data_source_length = data_size; /* Initial window size is 64KiB - 1 */ nghttp2_session_client_new(&session, &callbacks, &ud); nghttp2_submit_request(session, NULL, NULL, 0, &data_prd, NULL); /* Sends NGHTTP2_INITIAL_WINDOW_SIZE data, assuming, it is equal to or smaller than NGHTTP2_INITIAL_CONNECTION_WINDOW_SIZE */ CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(data_size - NGHTTP2_INITIAL_WINDOW_SIZE == ud.data_source_length); stream = nghttp2_session_get_stream(session, 1); CU_ASSERT(NGHTTP2_DATA == stream->item->frame.hd.type); stream->item->aux_data.data.data_prd.read_callback = temporal_failure_data_source_read_callback; /* Back NGHTTP2_INITIAL_WINDOW_SIZE to both connection-level and stream-wise window */ nghttp2_frame_window_update_init(&frame.window_update, NGHTTP2_FLAG_NONE, 1, NGHTTP2_INITIAL_WINDOW_SIZE); nghttp2_session_on_window_update_received(session, &frame); frame.hd.stream_id = 0; nghttp2_session_on_window_update_received(session, &frame); nghttp2_frame_window_update_free(&frame.window_update); /* Sending data will fail (soft fail) and treated as stream error */ ud.frame_send_cb_called = 0; CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(data_size - NGHTTP2_INITIAL_WINDOW_SIZE == ud.data_source_length); CU_ASSERT(1 == ud.frame_send_cb_called); CU_ASSERT(NGHTTP2_RST_STREAM == ud.sent_frame_type); data_prd.read_callback = fail_data_source_read_callback; nghttp2_submit_request(session, NULL, NULL, 0, &data_prd, NULL); /* Sending data will fail (hard fail) and session tear down */ CU_ASSERT(NGHTTP2_ERR_CALLBACK_FAILURE == nghttp2_session_send(session)); nghttp2_session_del(session); } void test_nghttp2_session_on_stream_close(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data user_data; nghttp2_stream *stream; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.on_stream_close_callback = on_stream_close_callback; user_data.stream_close_cb_called = 0; nghttp2_session_client_new(&session, &callbacks, &user_data); stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENED, &user_data); CU_ASSERT(stream != NULL); CU_ASSERT(nghttp2_session_close_stream(session, 1, NGHTTP2_NO_ERROR) == 0); CU_ASSERT(user_data.stream_close_cb_called == 1); nghttp2_session_del(session); } void test_nghttp2_session_on_ctrl_not_send(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data user_data; nghttp2_stream *stream; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.on_frame_not_send_callback = on_frame_not_send_callback; callbacks.send_callback = null_send_callback; user_data.frame_not_send_cb_called = 0; user_data.not_sent_frame_type = 0; user_data.not_sent_error = 0; nghttp2_session_server_new(&session, &callbacks, &user_data); stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, &user_data); /* Check response HEADERS */ /* Send bogus stream ID */ CU_ASSERT(0 == nghttp2_submit_headers(session, NGHTTP2_FLAG_END_STREAM, 3, NULL, NULL, 0, NULL)); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(1 == user_data.frame_not_send_cb_called); CU_ASSERT(NGHTTP2_HEADERS == user_data.not_sent_frame_type); CU_ASSERT(NGHTTP2_ERR_STREAM_CLOSED == user_data.not_sent_error); user_data.frame_not_send_cb_called = 0; /* Shutdown transmission */ stream->shut_flags |= NGHTTP2_SHUT_WR; CU_ASSERT(0 == nghttp2_submit_headers(session, NGHTTP2_FLAG_END_STREAM, 1, NULL, NULL, 0, NULL)); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(1 == user_data.frame_not_send_cb_called); CU_ASSERT(NGHTTP2_HEADERS == user_data.not_sent_frame_type); CU_ASSERT(NGHTTP2_ERR_STREAM_SHUT_WR == user_data.not_sent_error); stream->shut_flags = NGHTTP2_SHUT_NONE; user_data.frame_not_send_cb_called = 0; /* Queue RST_STREAM */ CU_ASSERT(0 == nghttp2_submit_headers(session, NGHTTP2_FLAG_END_STREAM, 1, NULL, NULL, 0, NULL)); CU_ASSERT(0 == nghttp2_submit_rst_stream(session, NGHTTP2_FLAG_NONE, 1, NGHTTP2_INTERNAL_ERROR)); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(1 == user_data.frame_not_send_cb_called); CU_ASSERT(NGHTTP2_HEADERS == user_data.not_sent_frame_type); CU_ASSERT(NGHTTP2_ERR_STREAM_CLOSING == user_data.not_sent_error); nghttp2_session_del(session); /* Check request HEADERS */ user_data.frame_not_send_cb_called = 0; CU_ASSERT(nghttp2_session_client_new(&session, &callbacks, &user_data) == 0); /* Maximum Stream ID is reached */ session->next_stream_id = (1u << 31) + 1; CU_ASSERT(NGHTTP2_ERR_STREAM_ID_NOT_AVAILABLE == nghttp2_submit_headers(session, NGHTTP2_FLAG_END_STREAM, -1, NULL, NULL, 0, NULL)); user_data.frame_not_send_cb_called = 0; /* GOAWAY received */ session->goaway_flags |= NGHTTP2_GOAWAY_RECV; session->next_stream_id = 9; CU_ASSERT(0 < nghttp2_submit_headers(session, NGHTTP2_FLAG_END_STREAM, -1, NULL, NULL, 0, NULL)); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(1 == user_data.frame_not_send_cb_called); CU_ASSERT(NGHTTP2_HEADERS == user_data.not_sent_frame_type); CU_ASSERT(NGHTTP2_ERR_START_STREAM_NOT_ALLOWED == user_data.not_sent_error); nghttp2_session_del(session); } void test_nghttp2_session_get_outbound_queue_size(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); CU_ASSERT(0 == nghttp2_session_client_new(&session, &callbacks, NULL)); CU_ASSERT(0 == nghttp2_session_get_outbound_queue_size(session)); CU_ASSERT(0 == nghttp2_submit_ping(session, NGHTTP2_FLAG_NONE, NULL)); CU_ASSERT(1 == nghttp2_session_get_outbound_queue_size(session)); CU_ASSERT(0 == nghttp2_submit_goaway(session, NGHTTP2_FLAG_NONE, 2, NGHTTP2_NO_ERROR, NULL, 0)); CU_ASSERT(2 == nghttp2_session_get_outbound_queue_size(session)); nghttp2_session_del(session); } void test_nghttp2_session_get_effective_local_window_size(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_stream *stream; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); CU_ASSERT(0 == nghttp2_session_client_new(&session, &callbacks, NULL)); stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENED, NULL); CU_ASSERT(NGHTTP2_INITIAL_CONNECTION_WINDOW_SIZE == nghttp2_session_get_effective_local_window_size(session)); CU_ASSERT(0 == nghttp2_session_get_effective_recv_data_length(session)); CU_ASSERT(NGHTTP2_INITIAL_WINDOW_SIZE == nghttp2_session_get_stream_effective_local_window_size(session, 1)); CU_ASSERT(0 == nghttp2_session_get_stream_effective_recv_data_length(session, 1)); /* Check connection flow control */ session->recv_window_size = 100; nghttp2_submit_window_update(session, NGHTTP2_FLAG_NONE, 0, 1100); CU_ASSERT(NGHTTP2_INITIAL_CONNECTION_WINDOW_SIZE + 1000 == nghttp2_session_get_effective_local_window_size(session)); CU_ASSERT(0 == nghttp2_session_get_effective_recv_data_length(session)); nghttp2_submit_window_update(session, NGHTTP2_FLAG_NONE, 0, -50); /* Now session->recv_window_size = -50 */ CU_ASSERT(NGHTTP2_INITIAL_CONNECTION_WINDOW_SIZE + 950 == nghttp2_session_get_effective_local_window_size(session)); CU_ASSERT(0 == nghttp2_session_get_effective_recv_data_length(session)); session->recv_window_size += 50; /* Now session->recv_window_size = 0 */ nghttp2_submit_window_update(session, NGHTTP2_FLAG_NONE, 0, 100); CU_ASSERT(NGHTTP2_INITIAL_CONNECTION_WINDOW_SIZE + 1050 == nghttp2_session_get_effective_local_window_size(session)); CU_ASSERT(50 == nghttp2_session_get_effective_recv_data_length(session)); /* Check stream flow control */ stream->recv_window_size = 100; nghttp2_submit_window_update(session, NGHTTP2_FLAG_NONE, 1, 1100); CU_ASSERT(NGHTTP2_INITIAL_WINDOW_SIZE + 1000 == nghttp2_session_get_stream_effective_local_window_size(session, 1)); CU_ASSERT(0 == nghttp2_session_get_stream_effective_recv_data_length(session, 1)); nghttp2_submit_window_update(session, NGHTTP2_FLAG_NONE, 1, -50); /* Now stream->recv_window_size = -50 */ CU_ASSERT(NGHTTP2_INITIAL_WINDOW_SIZE + 950 == nghttp2_session_get_stream_effective_local_window_size(session, 1)); CU_ASSERT(0 == nghttp2_session_get_stream_effective_recv_data_length(session, 1)); stream->recv_window_size += 50; /* Now stream->recv_window_size = 0 */ nghttp2_submit_window_update(session, NGHTTP2_FLAG_NONE, 1, 100); CU_ASSERT(NGHTTP2_INITIAL_WINDOW_SIZE + 1050 == nghttp2_session_get_stream_effective_local_window_size(session, 1)); CU_ASSERT(50 == nghttp2_session_get_stream_effective_recv_data_length(session, 1)); nghttp2_session_del(session); } void test_nghttp2_session_set_option(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_option *option; nghttp2_option_new(&option); nghttp2_option_set_no_auto_window_update(option, 1); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); nghttp2_session_client_new2(&session, &callbacks, NULL, option); CU_ASSERT(session->opt_flags & NGHTTP2_OPTMASK_NO_AUTO_WINDOW_UPDATE); nghttp2_session_del(session); nghttp2_option_set_peer_max_concurrent_streams(option, 100); nghttp2_session_client_new2(&session, &callbacks, NULL, option); CU_ASSERT(100 == session->remote_settings.max_concurrent_streams); nghttp2_session_del(session); nghttp2_option_set_max_reserved_remote_streams(option, 99); nghttp2_session_client_new2(&session, &callbacks, NULL, option); CU_ASSERT(99 == session->max_incoming_reserved_streams); nghttp2_session_del(session); nghttp2_option_del(option); } void test_nghttp2_session_data_backoff_by_high_pri_frame(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data ud; nghttp2_data_provider data_prd; nghttp2_stream *stream; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = block_count_send_callback; callbacks.on_frame_send_callback = on_frame_send_callback; data_prd.read_callback = fixed_length_data_source_read_callback; ud.frame_send_cb_called = 0; ud.data_source_length = NGHTTP2_DATA_PAYLOADLEN * 4; nghttp2_session_client_new(&session, &callbacks, &ud); nghttp2_submit_request(session, NULL, NULL, 0, &data_prd, NULL); session->remote_window_size = 1 << 20; ud.block_count = 2; /* Sends request HEADERS + DATA[0] */ CU_ASSERT(0 == nghttp2_session_send(session)); stream = nghttp2_session_get_stream(session, 1); stream->remote_window_size = 1 << 20; CU_ASSERT(NGHTTP2_DATA == ud.sent_frame_type); /* data for DATA[1] is read from data_prd but it is not sent */ CU_ASSERT(ud.data_source_length == NGHTTP2_DATA_PAYLOADLEN * 2); nghttp2_submit_ping(session, NGHTTP2_FLAG_NONE, NULL); ud.block_count = 2; /* Sends DATA[1] + PING, PING is interleaved in DATA sequence */ CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(NGHTTP2_PING == ud.sent_frame_type); /* data for DATA[2] is read from data_prd but it is not sent */ CU_ASSERT(ud.data_source_length == NGHTTP2_DATA_PAYLOADLEN); ud.block_count = 2; /* Sends DATA[2..3] */ CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(stream->shut_flags & NGHTTP2_SHUT_WR); nghttp2_session_del(session); } static void check_session_recv_data_with_padding(nghttp2_bufs *bufs, size_t datalen, nghttp2_mem *mem) { nghttp2_session *session; my_user_data ud; nghttp2_session_callbacks callbacks; uint8_t *in; size_t inlen; memset(&callbacks, 0, sizeof(callbacks)); callbacks.on_frame_recv_callback = on_frame_recv_callback; callbacks.on_data_chunk_recv_callback = on_data_chunk_recv_callback; nghttp2_session_server_new(&session, &callbacks, &ud); nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); inlen = (size_t)nghttp2_bufs_remove(bufs, &in); ud.frame_recv_cb_called = 0; ud.data_chunk_len = 0; CU_ASSERT((ssize_t)inlen == nghttp2_session_mem_recv(session, in, inlen)); CU_ASSERT(1 == ud.frame_recv_cb_called); CU_ASSERT(datalen == ud.data_chunk_len); mem->free(in, NULL); nghttp2_session_del(session); } void test_nghttp2_session_pack_data_with_padding(void) { nghttp2_session *session; my_user_data ud; nghttp2_session_callbacks callbacks; nghttp2_data_provider data_prd; nghttp2_frame *frame; size_t datalen = 55; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(callbacks)); callbacks.send_callback = block_count_send_callback; callbacks.on_frame_send_callback = on_frame_send_callback; callbacks.select_padding_callback = select_padding_callback; data_prd.read_callback = fixed_length_data_source_read_callback; nghttp2_session_client_new(&session, &callbacks, &ud); ud.padlen = 63; nghttp2_submit_request(session, NULL, NULL, 0, &data_prd, NULL); ud.block_count = 1; ud.data_source_length = datalen; /* Sends HEADERS */ CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(NGHTTP2_HEADERS == ud.sent_frame_type); frame = &session->aob.item->frame; CU_ASSERT(ud.padlen == frame->data.padlen); CU_ASSERT(frame->hd.flags & NGHTTP2_FLAG_PADDED); /* Check reception of this DATA frame */ check_session_recv_data_with_padding(&session->aob.framebufs, datalen, mem); nghttp2_session_del(session); } void test_nghttp2_session_pack_headers_with_padding(void) { nghttp2_session *session, *sv_session; accumulator acc; my_user_data ud; nghttp2_session_callbacks callbacks; memset(&callbacks, 0, sizeof(callbacks)); callbacks.send_callback = accumulator_send_callback; callbacks.on_frame_send_callback = on_frame_send_callback; callbacks.select_padding_callback = select_padding_callback; callbacks.on_frame_recv_callback = on_frame_recv_callback; acc.length = 0; ud.acc = &acc; nghttp2_session_client_new(&session, &callbacks, &ud); nghttp2_session_server_new(&sv_session, &callbacks, &ud); ud.padlen = 163; CU_ASSERT(1 == nghttp2_submit_request(session, NULL, reqnv, ARRLEN(reqnv), NULL, NULL)); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(acc.length < NGHTTP2_MAX_PAYLOADLEN); ud.frame_recv_cb_called = 0; CU_ASSERT((ssize_t)acc.length == nghttp2_session_mem_recv(sv_session, acc.buf, acc.length)); CU_ASSERT(1 == ud.frame_recv_cb_called); CU_ASSERT(NULL == nghttp2_session_get_next_ob_item(sv_session)); nghttp2_session_del(sv_session); nghttp2_session_del(session); } void test_nghttp2_pack_settings_payload(void) { nghttp2_settings_entry iv[2]; uint8_t buf[64]; ssize_t len; nghttp2_settings_entry *resiv; size_t resniv; nghttp2_mem *mem; mem = nghttp2_mem_default(); iv[0].settings_id = NGHTTP2_SETTINGS_HEADER_TABLE_SIZE; iv[0].value = 1023; iv[1].settings_id = NGHTTP2_SETTINGS_INITIAL_WINDOW_SIZE; iv[1].value = 4095; len = nghttp2_pack_settings_payload(buf, sizeof(buf), iv, 2); CU_ASSERT(2 * NGHTTP2_FRAME_SETTINGS_ENTRY_LENGTH == len); CU_ASSERT(0 == nghttp2_frame_unpack_settings_payload2(&resiv, &resniv, buf, (size_t)len, mem)); CU_ASSERT(2 == resniv); CU_ASSERT(NGHTTP2_SETTINGS_HEADER_TABLE_SIZE == resiv[0].settings_id); CU_ASSERT(1023 == resiv[0].value); CU_ASSERT(NGHTTP2_SETTINGS_INITIAL_WINDOW_SIZE == resiv[1].settings_id); CU_ASSERT(4095 == resiv[1].value); mem->free(resiv, NULL); len = nghttp2_pack_settings_payload(buf, 9 /* too small */, iv, 2); CU_ASSERT(NGHTTP2_ERR_INSUFF_BUFSIZE == len); } #define check_stream_dep_sib(STREAM, DEP_PREV, DEP_NEXT, SIB_PREV, SIB_NEXT) \ do { \ CU_ASSERT(DEP_PREV == STREAM->dep_prev); \ CU_ASSERT(DEP_NEXT == STREAM->dep_next); \ CU_ASSERT(SIB_PREV == STREAM->sib_prev); \ CU_ASSERT(SIB_NEXT == STREAM->sib_next); \ } while (0) /* nghttp2_stream_dep_add() and its families functions should be tested in nghttp2_stream_test.c, but it is easier to use nghttp2_session_open_stream(). Therefore, we test them here. */ void test_nghttp2_session_stream_dep_add(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_stream *a, *b, *c, *d, *e, *root; memset(&callbacks, 0, sizeof(callbacks)); nghttp2_session_server_new(&session, &callbacks, NULL); root = &session->root; a = open_stream(session, 1); c = open_stream_with_dep(session, 5, a); b = open_stream_with_dep(session, 3, a); d = open_stream_with_dep(session, 7, c); /* a * | * b--c * | * d */ CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT * 2 == a->sum_dep_weight); CU_ASSERT(0 == b->sum_dep_weight); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == c->sum_dep_weight); CU_ASSERT(0 == d->sum_dep_weight); check_stream_dep_sib(a, root, b, NULL, NULL); check_stream_dep_sib(b, a, NULL, NULL, c); check_stream_dep_sib(c, a, d, b, NULL); check_stream_dep_sib(d, c, NULL, NULL, NULL); CU_ASSERT(a == session->root.dep_next); e = open_stream_with_dep_excl(session, 9, a); /* a * | * e * | * b--c * | * d */ CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == a->sum_dep_weight); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT * 2 == e->sum_dep_weight); CU_ASSERT(0 == b->sum_dep_weight); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == c->sum_dep_weight); CU_ASSERT(0 == d->sum_dep_weight); check_stream_dep_sib(a, root, e, NULL, NULL); check_stream_dep_sib(e, a, b, NULL, NULL); check_stream_dep_sib(b, e, NULL, NULL, c); check_stream_dep_sib(c, e, d, b, NULL); check_stream_dep_sib(d, c, NULL, NULL, NULL); CU_ASSERT(a == session->root.dep_next); nghttp2_session_del(session); } void test_nghttp2_session_stream_dep_remove(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_stream *a, *b, *c, *d, *e, *f, *root; memset(&callbacks, 0, sizeof(callbacks)); /* Remove root */ nghttp2_session_server_new(&session, &callbacks, NULL); root = &session->root; a = open_stream(session, 1); b = open_stream_with_dep(session, 3, a); c = open_stream_with_dep(session, 5, a); d = open_stream_with_dep(session, 7, c); /* a * | * c--b * | * d */ nghttp2_stream_dep_remove(a); /* becomes: * c b * | * d */ CU_ASSERT(0 == a->sum_dep_weight); CU_ASSERT(0 == b->sum_dep_weight); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == c->sum_dep_weight); CU_ASSERT(0 == d->sum_dep_weight); check_stream_dep_sib(a, NULL, NULL, NULL, NULL); check_stream_dep_sib(b, root, NULL, c, NULL); check_stream_dep_sib(c, root, d, NULL, b); check_stream_dep_sib(d, c, NULL, NULL, NULL); CU_ASSERT(c == session->root.dep_next); nghttp2_session_del(session); /* Remove right most stream */ nghttp2_session_server_new(&session, &callbacks, NULL); root = &session->root; a = open_stream(session, 1); b = open_stream_with_dep(session, 3, a); c = open_stream_with_dep(session, 5, a); d = open_stream_with_dep(session, 7, c); /* a * | * c--b * | * d */ nghttp2_stream_dep_remove(b); /* becomes: * a * | * c * | * d */ CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == a->sum_dep_weight); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == c->sum_dep_weight); CU_ASSERT(0 == d->sum_dep_weight); CU_ASSERT(0 == b->sum_dep_weight); check_stream_dep_sib(a, root, c, NULL, NULL); check_stream_dep_sib(b, NULL, NULL, NULL, NULL); check_stream_dep_sib(c, a, d, NULL, NULL); check_stream_dep_sib(d, c, NULL, NULL, NULL); CU_ASSERT(a == session->root.dep_next); nghttp2_session_del(session); /* Remove left most stream */ nghttp2_session_server_new(&session, &callbacks, NULL); root = &session->root; a = open_stream(session, 1); b = open_stream_with_dep(session, 3, a); c = open_stream_with_dep(session, 5, a); d = open_stream_with_dep(session, 7, c); e = open_stream_with_dep(session, 9, c); /* a * | * c--b * | * e--d */ nghttp2_stream_dep_remove(c); /* becomes: * a * | * e--d--b */ CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT * 2 == a->sum_dep_weight); CU_ASSERT(0 == b->sum_dep_weight); CU_ASSERT(0 == d->sum_dep_weight); CU_ASSERT(0 == c->sum_dep_weight); CU_ASSERT(0 == e->sum_dep_weight); check_stream_dep_sib(a, root, e, NULL, NULL); check_stream_dep_sib(b, a, NULL, d, NULL); check_stream_dep_sib(c, NULL, NULL, NULL, NULL); check_stream_dep_sib(d, a, NULL, e, b); check_stream_dep_sib(e, a, NULL, NULL, d); nghttp2_session_del(session); /* Remove middle stream */ nghttp2_session_server_new(&session, &callbacks, NULL); root = &session->root; a = open_stream(session, 1); b = open_stream_with_dep(session, 3, a); c = open_stream_with_dep(session, 5, a); d = open_stream_with_dep(session, 7, a); e = open_stream_with_dep(session, 9, c); f = open_stream_with_dep(session, 11, c); /* a * | * d--c--b * | * f--e */ CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT * 3 == a->sum_dep_weight); CU_ASSERT(0 == b->sum_dep_weight); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT * 2 == c->sum_dep_weight); CU_ASSERT(0 == d->sum_dep_weight); CU_ASSERT(0 == e->sum_dep_weight); CU_ASSERT(0 == f->sum_dep_weight); nghttp2_stream_dep_remove(c); /* becomes: * a * | * d--f--e--b */ /* c's weight 16 is distributed evenly to e and f. Each weight of e and f becomes 8. */ CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT * 2 + 8 * 2 == a->sum_dep_weight); CU_ASSERT(0 == b->sum_dep_weight); CU_ASSERT(0 == c->sum_dep_weight); CU_ASSERT(0 == d->sum_dep_weight); CU_ASSERT(0 == e->sum_dep_weight); CU_ASSERT(0 == f->sum_dep_weight); check_stream_dep_sib(a, root, d, NULL, NULL); check_stream_dep_sib(b, a, NULL, e, NULL); check_stream_dep_sib(c, NULL, NULL, NULL, NULL); check_stream_dep_sib(e, a, NULL, f, b); check_stream_dep_sib(f, a, NULL, d, e); check_stream_dep_sib(d, a, NULL, NULL, f); nghttp2_session_del(session); } void test_nghttp2_session_stream_dep_add_subtree(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_stream *a, *b, *c, *d, *e, *f, *root; memset(&callbacks, 0, sizeof(callbacks)); /* dep_stream has dep_next */ nghttp2_session_server_new(&session, &callbacks, NULL); root = &session->root; a = open_stream(session, 1); b = open_stream_with_dep(session, 3, a); c = open_stream_with_dep(session, 5, a); d = open_stream_with_dep(session, 7, c); e = open_stream(session, 9); f = open_stream_with_dep(session, 11, e); /* a e * | | * c--b f * | * d */ nghttp2_stream_dep_remove_subtree(e); nghttp2_stream_dep_add_subtree(a, e); /* becomes * a * | * e--c--b * | | * f d */ CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT * 3 == a->sum_dep_weight); CU_ASSERT(0 == b->sum_dep_weight); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == c->sum_dep_weight); CU_ASSERT(0 == d->sum_dep_weight); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == e->sum_dep_weight); CU_ASSERT(0 == f->sum_dep_weight); check_stream_dep_sib(a, root, e, NULL, NULL); check_stream_dep_sib(b, a, NULL, c, NULL); check_stream_dep_sib(c, a, d, e, b); check_stream_dep_sib(d, c, NULL, NULL, NULL); check_stream_dep_sib(e, a, f, NULL, c); check_stream_dep_sib(f, e, NULL, NULL, NULL); nghttp2_session_del(session); /* dep_stream has dep_next and now we insert subtree */ nghttp2_session_server_new(&session, &callbacks, NULL); root = &session->root; a = open_stream(session, 1); b = open_stream_with_dep(session, 3, a); c = open_stream_with_dep(session, 5, a); d = open_stream_with_dep(session, 7, c); e = open_stream(session, 9); f = open_stream_with_dep(session, 11, e); /* a e * | | * c--b f * | * d */ nghttp2_stream_dep_remove_subtree(e); nghttp2_stream_dep_insert_subtree(a, e); /* becomes * a * | * e * | * f--c--b * | * d */ CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == a->sum_dep_weight); CU_ASSERT(0 == b->sum_dep_weight); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == c->sum_dep_weight); CU_ASSERT(0 == d->sum_dep_weight); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT * 3 == e->sum_dep_weight); CU_ASSERT(0 == f->sum_dep_weight); check_stream_dep_sib(a, root, e, NULL, NULL); check_stream_dep_sib(e, a, f, NULL, NULL); check_stream_dep_sib(f, e, NULL, NULL, c); check_stream_dep_sib(b, e, NULL, c, NULL); check_stream_dep_sib(c, e, d, f, b); check_stream_dep_sib(d, c, NULL, NULL, NULL); nghttp2_session_del(session); } void test_nghttp2_session_stream_dep_remove_subtree(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_stream *a, *b, *c, *d, *e, *root; memset(&callbacks, 0, sizeof(callbacks)); /* Remove left most stream */ nghttp2_session_server_new(&session, &callbacks, NULL); root = &session->root; a = open_stream(session, 1); b = open_stream_with_dep(session, 3, a); c = open_stream_with_dep(session, 5, a); d = open_stream_with_dep(session, 7, c); /* a * | * c--b * | * d */ nghttp2_stream_dep_remove_subtree(c); /* becomes * a c * | | * b d */ CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == a->sum_dep_weight); CU_ASSERT(0 == b->sum_dep_weight); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == c->sum_dep_weight); CU_ASSERT(0 == d->sum_dep_weight); check_stream_dep_sib(a, root, b, NULL, NULL); check_stream_dep_sib(b, a, NULL, NULL, NULL); check_stream_dep_sib(c, NULL, d, NULL, NULL); check_stream_dep_sib(d, c, NULL, NULL, NULL); nghttp2_session_del(session); /* Remove right most stream */ nghttp2_session_server_new(&session, &callbacks, NULL); root = &session->root; a = open_stream(session, 1); b = open_stream_with_dep(session, 3, a); c = open_stream_with_dep(session, 5, a); d = open_stream_with_dep(session, 7, c); /* a * | * c--b * | * d */ nghttp2_stream_dep_remove_subtree(b); /* becomes * a b * | * c * | * d */ CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == a->sum_dep_weight); CU_ASSERT(0 == b->sum_dep_weight); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == c->sum_dep_weight); CU_ASSERT(0 == d->sum_dep_weight); check_stream_dep_sib(a, root, c, NULL, NULL); check_stream_dep_sib(c, a, d, NULL, NULL); check_stream_dep_sib(d, c, NULL, NULL, NULL); check_stream_dep_sib(b, NULL, NULL, NULL, NULL); nghttp2_session_del(session); /* Remove middle stream */ nghttp2_session_server_new(&session, &callbacks, NULL); root = &session->root; a = open_stream(session, 1); e = open_stream_with_dep(session, 9, a); c = open_stream_with_dep(session, 5, a); b = open_stream_with_dep(session, 3, a); d = open_stream_with_dep(session, 7, c); /* a * | * b--c--e * | * d */ nghttp2_stream_dep_remove_subtree(c); /* becomes * a c * | | * b--e d */ CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT * 2 == a->sum_dep_weight); CU_ASSERT(0 == b->sum_dep_weight); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == c->sum_dep_weight); CU_ASSERT(0 == d->sum_dep_weight); CU_ASSERT(0 == e->sum_dep_weight); check_stream_dep_sib(a, root, b, NULL, NULL); check_stream_dep_sib(b, a, NULL, NULL, e); check_stream_dep_sib(e, a, NULL, b, NULL); check_stream_dep_sib(c, NULL, d, NULL, NULL); check_stream_dep_sib(d, c, NULL, NULL, NULL); nghttp2_session_del(session); } void test_nghttp2_session_stream_dep_all_your_stream_are_belong_to_us(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_stream *a, *b, *c, *d, *root; nghttp2_outbound_item *db, *dc; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(callbacks)); nghttp2_session_server_new(&session, &callbacks, NULL); root = &session->root; a = open_stream(session, 1); b = open_stream_with_dep(session, 3, a); c = open_stream(session, 5); /* a c * | * b */ nghttp2_stream_dep_remove_subtree(c); CU_ASSERT(0 == nghttp2_stream_dep_insert_subtree(&session->root, c)); /* * c * | * a * | * b */ CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == c->sum_dep_weight); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == a->sum_dep_weight); CU_ASSERT(0 == b->sum_dep_weight); CU_ASSERT(nghttp2_pq_empty(&a->obq)); CU_ASSERT(nghttp2_pq_empty(&b->obq)); CU_ASSERT(nghttp2_pq_empty(&c->obq)); check_stream_dep_sib(c, root, a, NULL, NULL); check_stream_dep_sib(a, c, b, NULL, NULL); check_stream_dep_sib(b, a, NULL, NULL, NULL); nghttp2_session_del(session); nghttp2_session_server_new(&session, &callbacks, NULL); root = &session->root; a = open_stream(session, 1); b = open_stream(session, 3); c = open_stream(session, 5); /* * a b c */ nghttp2_stream_dep_remove_subtree(c); CU_ASSERT(0 == nghttp2_stream_dep_insert_subtree(&session->root, c)); /* * c * | * b--a */ CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT * 2 == c->sum_dep_weight); CU_ASSERT(0 == b->sum_dep_weight); CU_ASSERT(0 == a->sum_dep_weight); CU_ASSERT(nghttp2_pq_empty(&a->obq)); CU_ASSERT(nghttp2_pq_empty(&b->obq)); CU_ASSERT(nghttp2_pq_empty(&c->obq)); check_stream_dep_sib(c, root, b, NULL, NULL); check_stream_dep_sib(b, c, NULL, NULL, a); check_stream_dep_sib(a, c, NULL, b, NULL); nghttp2_session_del(session); nghttp2_session_server_new(&session, &callbacks, NULL); root = &session->root; a = open_stream(session, 1); b = open_stream_with_dep(session, 3, a); c = open_stream(session, 5); d = open_stream_with_dep(session, 7, c); /* a c * | | * b d */ nghttp2_stream_dep_remove_subtree(c); CU_ASSERT(0 == nghttp2_stream_dep_insert_subtree(&session->root, c)); /* * c * | * d--a * | * b */ CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT * 2 == c->sum_dep_weight); CU_ASSERT(0 == d->sum_dep_weight); CU_ASSERT(NGHTTP2_DEFAULT_WEIGHT == a->sum_dep_weight); CU_ASSERT(0 == b->sum_dep_weight); CU_ASSERT(nghttp2_pq_empty(&a->obq)); CU_ASSERT(nghttp2_pq_empty(&b->obq)); CU_ASSERT(nghttp2_pq_empty(&c->obq)); CU_ASSERT(nghttp2_pq_empty(&d->obq)); check_stream_dep_sib(c, root, d, NULL, NULL); check_stream_dep_sib(d, c, NULL, NULL, a); check_stream_dep_sib(a, c, b, d, NULL); check_stream_dep_sib(b, a, NULL, NULL, NULL); nghttp2_session_del(session); nghttp2_session_server_new(&session, &callbacks, NULL); root = &session->root; a = open_stream(session, 1); b = open_stream_with_dep(session, 3, a); c = open_stream(session, 5); d = open_stream_with_dep(session, 7, c); /* a c * | | * b d */ db = create_data_ob_item(mem); nghttp2_stream_attach_item(b, db); nghttp2_stream_dep_remove_subtree(c); CU_ASSERT(0 == nghttp2_stream_dep_insert_subtree(&session->root, c)); /* * c * | * d--a * | * b */ CU_ASSERT(c->queued); CU_ASSERT(a->queued); CU_ASSERT(b->queued); CU_ASSERT(!d->queued); CU_ASSERT(1 == nghttp2_pq_size(&a->obq)); CU_ASSERT(1 == nghttp2_pq_size(&c->obq)); CU_ASSERT(nghttp2_pq_empty(&d->obq)); check_stream_dep_sib(c, root, d, NULL, NULL); check_stream_dep_sib(d, c, NULL, NULL, a); check_stream_dep_sib(a, c, b, d, NULL); check_stream_dep_sib(b, a, NULL, NULL, NULL); nghttp2_session_del(session); nghttp2_session_server_new(&session, &callbacks, NULL); root = &session->root; a = open_stream(session, 1); b = open_stream_with_dep(session, 3, a); c = open_stream(session, 5); d = open_stream_with_dep(session, 7, c); /* a c * | | * b d */ db = create_data_ob_item(mem); dc = create_data_ob_item(mem); nghttp2_stream_attach_item(b, db); nghttp2_stream_attach_item(c, dc); nghttp2_stream_dep_remove_subtree(c); CU_ASSERT(0 == nghttp2_stream_dep_insert_subtree(&session->root, c)); /* * c * | * d--a * | * b */ CU_ASSERT(c->queued); CU_ASSERT(a->queued); CU_ASSERT(b->queued); CU_ASSERT(!d->queued); check_stream_dep_sib(c, root, d, NULL, NULL); check_stream_dep_sib(d, c, NULL, NULL, a); check_stream_dep_sib(a, c, b, d, NULL); check_stream_dep_sib(b, a, NULL, NULL, NULL); nghttp2_session_del(session); } void test_nghttp2_session_stream_attach_item(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_stream *a, *b, *c, *d, *e; nghttp2_outbound_item *da, *db, *dc, *dd; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(callbacks)); nghttp2_session_server_new(&session, &callbacks, NULL); a = open_stream(session, 1); b = open_stream_with_dep(session, 3, a); c = open_stream_with_dep(session, 5, a); d = open_stream_with_dep(session, 7, c); /* a * | * c--b * | * d */ db = create_data_ob_item(mem); nghttp2_stream_attach_item(b, db); CU_ASSERT(a->queued); CU_ASSERT(b->queued); CU_ASSERT(!c->queued); CU_ASSERT(!d->queued); CU_ASSERT(1 == nghttp2_pq_size(&a->obq)); /* Attach item to c */ dc = create_data_ob_item(mem); nghttp2_stream_attach_item(c, dc); CU_ASSERT(a->queued); CU_ASSERT(b->queued); CU_ASSERT(c->queued); CU_ASSERT(!d->queued); CU_ASSERT(2 == nghttp2_pq_size(&a->obq)); /* Attach item to a */ da = create_data_ob_item(mem); nghttp2_stream_attach_item(a, da); CU_ASSERT(a->queued); CU_ASSERT(b->queued); CU_ASSERT(c->queued); CU_ASSERT(!d->queued); CU_ASSERT(2 == nghttp2_pq_size(&a->obq)); /* Detach item from a */ nghttp2_stream_detach_item(a); CU_ASSERT(a->queued); CU_ASSERT(b->queued); CU_ASSERT(c->queued); CU_ASSERT(!d->queued); CU_ASSERT(2 == nghttp2_pq_size(&a->obq)); /* Attach item to d */ dd = create_data_ob_item(mem); nghttp2_stream_attach_item(d, dd); CU_ASSERT(a->queued); CU_ASSERT(b->queued); CU_ASSERT(c->queued); CU_ASSERT(d->queued); CU_ASSERT(2 == nghttp2_pq_size(&a->obq)); CU_ASSERT(1 == nghttp2_pq_size(&c->obq)); /* Detach item from c */ nghttp2_stream_detach_item(c); CU_ASSERT(a->queued); CU_ASSERT(b->queued); CU_ASSERT(c->queued); CU_ASSERT(d->queued); CU_ASSERT(2 == nghttp2_pq_size(&a->obq)); CU_ASSERT(1 == nghttp2_pq_size(&c->obq)); /* Detach item from b */ nghttp2_stream_detach_item(b); CU_ASSERT(a->queued); CU_ASSERT(!b->queued); CU_ASSERT(c->queued); CU_ASSERT(d->queued); CU_ASSERT(1 == nghttp2_pq_size(&a->obq)); /* exercises insertion */ e = open_stream_with_dep_excl(session, 9, a); /* a * | * e * | * c--b * | * d */ CU_ASSERT(a->queued); CU_ASSERT(e->queued); CU_ASSERT(!b->queued); CU_ASSERT(c->queued); CU_ASSERT(d->queued); CU_ASSERT(1 == nghttp2_pq_size(&a->obq)); CU_ASSERT(1 == nghttp2_pq_size(&e->obq)); CU_ASSERT(nghttp2_pq_empty(&b->obq)); CU_ASSERT(1 == nghttp2_pq_size(&c->obq)); CU_ASSERT(nghttp2_pq_empty(&d->obq)); /* exercises deletion */ nghttp2_stream_dep_remove(e); /* a * | * c--b * | * d */ CU_ASSERT(a->queued); CU_ASSERT(!b->queued); CU_ASSERT(c->queued); CU_ASSERT(d->queued); CU_ASSERT(1 == nghttp2_pq_size(&a->obq)); CU_ASSERT(nghttp2_pq_empty(&b->obq)); CU_ASSERT(1 == nghttp2_pq_size(&c->obq)); CU_ASSERT(nghttp2_pq_empty(&d->obq)); /* e's weight 16 is distributed equally among c and b, both now have weight 8 each. */ CU_ASSERT(8 == b->weight); CU_ASSERT(8 == c->weight); /* da, db, dc have been detached */ nghttp2_outbound_item_free(da, mem); nghttp2_outbound_item_free(db, mem); nghttp2_outbound_item_free(dc, mem); free(da); free(db); free(dc); nghttp2_session_del(session); nghttp2_session_server_new(&session, &callbacks, NULL); a = open_stream(session, 1); b = open_stream_with_dep(session, 3, a); c = open_stream_with_dep(session, 5, a); d = open_stream_with_dep(session, 7, c); /* a * | * c--b * | * d */ da = create_data_ob_item(mem); db = create_data_ob_item(mem); dc = create_data_ob_item(mem); nghttp2_stream_attach_item(a, da); nghttp2_stream_attach_item(b, db); nghttp2_stream_attach_item(c, dc); CU_ASSERT(a->queued); CU_ASSERT(b->queued); CU_ASSERT(c->queued); CU_ASSERT(!d->queued); CU_ASSERT(2 == nghttp2_pq_size(&a->obq)); CU_ASSERT(nghttp2_pq_empty(&b->obq)); CU_ASSERT(nghttp2_pq_empty(&c->obq)); CU_ASSERT(nghttp2_pq_empty(&d->obq)); /* Detach item from a */ nghttp2_stream_detach_item(a); CU_ASSERT(a->queued); CU_ASSERT(b->queued); CU_ASSERT(c->queued); CU_ASSERT(!d->queued); CU_ASSERT(2 == nghttp2_pq_size(&a->obq)); CU_ASSERT(nghttp2_pq_empty(&b->obq)); CU_ASSERT(nghttp2_pq_empty(&c->obq)); CU_ASSERT(nghttp2_pq_empty(&d->obq)); /* da has been detached */ nghttp2_outbound_item_free(da, mem); free(da); nghttp2_session_del(session); } void test_nghttp2_session_stream_attach_item_subtree(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_stream *a, *b, *c, *d, *e, *f; nghttp2_outbound_item *da, *db, *dd, *de; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(callbacks)); nghttp2_session_server_new(&session, &callbacks, NULL); a = open_stream(session, 1); b = open_stream_with_dep(session, 3, a); c = open_stream_with_dep(session, 5, a); d = open_stream_with_dep(session, 7, c); e = open_stream_with_dep_weight(session, 9, 32, &session->root); f = open_stream_with_dep(session, 11, e); /* * a e * | | * c--b f * | * d */ de = create_data_ob_item(mem); nghttp2_stream_attach_item(e, de); db = create_data_ob_item(mem); nghttp2_stream_attach_item(b, db); CU_ASSERT(a->queued); CU_ASSERT(b->queued); CU_ASSERT(!c->queued); CU_ASSERT(!d->queued); CU_ASSERT(e->queued); CU_ASSERT(!f->queued); CU_ASSERT(1 == nghttp2_pq_size(&a->obq)); CU_ASSERT(nghttp2_pq_empty(&b->obq)); CU_ASSERT(nghttp2_pq_empty(&c->obq)); CU_ASSERT(nghttp2_pq_empty(&d->obq)); CU_ASSERT(nghttp2_pq_empty(&e->obq)); CU_ASSERT(nghttp2_pq_empty(&f->obq)); /* Insert subtree e under a */ nghttp2_stream_dep_remove_subtree(e); nghttp2_stream_dep_insert_subtree(a, e); /* * a * | * e * | * f--c--b * | * d */ CU_ASSERT(a->queued); CU_ASSERT(b->queued); CU_ASSERT(!c->queued); CU_ASSERT(!d->queued); CU_ASSERT(e->queued); CU_ASSERT(!f->queued); CU_ASSERT(1 == nghttp2_pq_size(&a->obq)); CU_ASSERT(nghttp2_pq_empty(&b->obq)); CU_ASSERT(nghttp2_pq_empty(&c->obq)); CU_ASSERT(nghttp2_pq_empty(&d->obq)); CU_ASSERT(1 == nghttp2_pq_size(&e->obq)); CU_ASSERT(nghttp2_pq_empty(&f->obq)); /* Remove subtree b */ nghttp2_stream_dep_remove_subtree(b); CU_ASSERT(0 == nghttp2_stream_dep_add_subtree(&session->root, b)); /* * a b * | * e * | * f--c * | * d */ CU_ASSERT(a->queued); CU_ASSERT(b->queued); CU_ASSERT(!c->queued); CU_ASSERT(!d->queued); CU_ASSERT(e->queued); CU_ASSERT(!f->queued); CU_ASSERT(1 == nghttp2_pq_size(&a->obq)); CU_ASSERT(nghttp2_pq_empty(&b->obq)); CU_ASSERT(nghttp2_pq_empty(&c->obq)); CU_ASSERT(nghttp2_pq_empty(&d->obq)); CU_ASSERT(nghttp2_pq_empty(&e->obq)); CU_ASSERT(nghttp2_pq_empty(&f->obq)); /* Remove subtree a, and add it to root again */ nghttp2_stream_dep_remove_subtree(a); CU_ASSERT(0 == nghttp2_stream_dep_add_subtree(&session->root, a)); CU_ASSERT(a->queued); CU_ASSERT(b->queued); CU_ASSERT(!c->queued); CU_ASSERT(!d->queued); CU_ASSERT(e->queued); CU_ASSERT(!f->queued); CU_ASSERT(1 == nghttp2_pq_size(&a->obq)); CU_ASSERT(nghttp2_pq_empty(&b->obq)); CU_ASSERT(nghttp2_pq_empty(&c->obq)); CU_ASSERT(nghttp2_pq_empty(&d->obq)); CU_ASSERT(nghttp2_pq_empty(&e->obq)); CU_ASSERT(nghttp2_pq_empty(&f->obq)); /* Remove subtree c */ nghttp2_stream_dep_remove_subtree(c); CU_ASSERT(0 == nghttp2_stream_dep_add_subtree(&session->root, c)); /* * a b c * | | * e d * | * f */ CU_ASSERT(a->queued); CU_ASSERT(b->queued); CU_ASSERT(!c->queued); CU_ASSERT(!d->queued); CU_ASSERT(e->queued); CU_ASSERT(!f->queued); CU_ASSERT(1 == nghttp2_pq_size(&a->obq)); CU_ASSERT(nghttp2_pq_empty(&b->obq)); CU_ASSERT(nghttp2_pq_empty(&c->obq)); CU_ASSERT(nghttp2_pq_empty(&d->obq)); CU_ASSERT(nghttp2_pq_empty(&e->obq)); CU_ASSERT(nghttp2_pq_empty(&f->obq)); dd = create_data_ob_item(mem); nghttp2_stream_attach_item(d, dd); /* Add subtree c to a */ nghttp2_stream_dep_remove_subtree(c); nghttp2_stream_dep_add_subtree(a, c); /* * a b * | * c--e * | | * d f */ CU_ASSERT(a->queued); CU_ASSERT(b->queued); CU_ASSERT(c->queued); CU_ASSERT(d->queued); CU_ASSERT(e->queued); CU_ASSERT(!f->queued); CU_ASSERT(2 == nghttp2_pq_size(&a->obq)); CU_ASSERT(nghttp2_pq_empty(&b->obq)); CU_ASSERT(1 == nghttp2_pq_size(&c->obq)); CU_ASSERT(nghttp2_pq_empty(&d->obq)); CU_ASSERT(nghttp2_pq_empty(&e->obq)); CU_ASSERT(nghttp2_pq_empty(&f->obq)); /* Insert b under a */ nghttp2_stream_dep_remove_subtree(b); nghttp2_stream_dep_insert_subtree(a, b); /* * a * | * b * | * c--e * | | * d f */ CU_ASSERT(a->queued); CU_ASSERT(b->queued); CU_ASSERT(c->queued); CU_ASSERT(d->queued); CU_ASSERT(e->queued); CU_ASSERT(!f->queued); CU_ASSERT(1 == nghttp2_pq_size(&a->obq)); CU_ASSERT(2 == nghttp2_pq_size(&b->obq)); CU_ASSERT(1 == nghttp2_pq_size(&c->obq)); CU_ASSERT(nghttp2_pq_empty(&d->obq)); CU_ASSERT(nghttp2_pq_empty(&e->obq)); CU_ASSERT(nghttp2_pq_empty(&f->obq)); /* Remove subtree b */ nghttp2_stream_dep_remove_subtree(b); CU_ASSERT(0 == nghttp2_stream_dep_add_subtree(&session->root, b)); /* * b a * | * e--c * | | * f d */ CU_ASSERT(!a->queued); CU_ASSERT(b->queued); CU_ASSERT(c->queued); CU_ASSERT(d->queued); CU_ASSERT(e->queued); CU_ASSERT(!f->queued); CU_ASSERT(nghttp2_pq_empty(&a->obq)); CU_ASSERT(2 == nghttp2_pq_size(&b->obq)); CU_ASSERT(1 == nghttp2_pq_size(&c->obq)); CU_ASSERT(nghttp2_pq_empty(&d->obq)); CU_ASSERT(nghttp2_pq_empty(&e->obq)); CU_ASSERT(nghttp2_pq_empty(&f->obq)); /* Remove subtree c, and detach item from b, and then re-add subtree c under b */ nghttp2_stream_dep_remove_subtree(c); nghttp2_stream_detach_item(b); nghttp2_stream_dep_add_subtree(b, c); /* * b a * | * e--c * | | * f d */ CU_ASSERT(!a->queued); CU_ASSERT(b->queued); CU_ASSERT(c->queued); CU_ASSERT(d->queued); CU_ASSERT(e->queued); CU_ASSERT(!f->queued); CU_ASSERT(nghttp2_pq_empty(&a->obq)); CU_ASSERT(2 == nghttp2_pq_size(&b->obq)); CU_ASSERT(1 == nghttp2_pq_size(&c->obq)); CU_ASSERT(nghttp2_pq_empty(&d->obq)); CU_ASSERT(nghttp2_pq_empty(&e->obq)); CU_ASSERT(nghttp2_pq_empty(&f->obq)); /* Attach data to a, and add subtree a under b */ da = create_data_ob_item(mem); nghttp2_stream_attach_item(a, da); nghttp2_stream_dep_remove_subtree(a); nghttp2_stream_dep_add_subtree(b, a); /* * b * | * a--e--c * | | * f d */ CU_ASSERT(a->queued); CU_ASSERT(b->queued); CU_ASSERT(c->queued); CU_ASSERT(d->queued); CU_ASSERT(e->queued); CU_ASSERT(!f->queued); CU_ASSERT(nghttp2_pq_empty(&a->obq)); CU_ASSERT(3 == nghttp2_pq_size(&b->obq)); CU_ASSERT(1 == nghttp2_pq_size(&c->obq)); CU_ASSERT(nghttp2_pq_empty(&d->obq)); CU_ASSERT(nghttp2_pq_empty(&e->obq)); CU_ASSERT(nghttp2_pq_empty(&f->obq)); /* Remove subtree c, and add under f */ nghttp2_stream_dep_remove_subtree(c); nghttp2_stream_dep_insert_subtree(f, c); /* * b * | * a--e * | * f * | * c * | * d */ CU_ASSERT(a->queued); CU_ASSERT(b->queued); CU_ASSERT(c->queued); CU_ASSERT(d->queued); CU_ASSERT(e->queued); CU_ASSERT(f->queued); CU_ASSERT(nghttp2_pq_empty(&a->obq)); CU_ASSERT(2 == nghttp2_pq_size(&b->obq)); CU_ASSERT(1 == nghttp2_pq_size(&c->obq)); CU_ASSERT(nghttp2_pq_empty(&d->obq)); CU_ASSERT(1 == nghttp2_pq_size(&e->obq)); CU_ASSERT(1 == nghttp2_pq_size(&f->obq)); /* db has been detached */ nghttp2_outbound_item_free(db, mem); free(db); nghttp2_session_del(session); } void test_nghttp2_session_stream_get_state(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_mem *mem; nghttp2_hd_deflater deflater; nghttp2_bufs bufs; nghttp2_buf *buf; nghttp2_stream *stream; ssize_t rv; nghttp2_data_provider data_prd; nghttp2_frame frame; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); memset(&data_prd, 0, sizeof(data_prd)); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_server_new(&session, &callbacks, NULL); nghttp2_hd_deflate_init(&deflater, mem); CU_ASSERT(NGHTTP2_STREAM_STATE_IDLE == nghttp2_stream_get_state(nghttp2_session_get_root_stream(session))); /* stream 1 HEADERS; without END_STREAM flag set */ pack_headers(&bufs, &deflater, 1, NGHTTP2_FLAG_END_HEADERS, reqnv, ARRLEN(reqnv), mem); buf = &bufs.head->buf; rv = nghttp2_session_mem_recv(session, buf->pos, nghttp2_buf_len(buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(buf) == rv); stream = nghttp2_session_find_stream(session, 1); CU_ASSERT(NULL != stream); CU_ASSERT(1 == stream->stream_id); CU_ASSERT(NGHTTP2_STREAM_STATE_OPEN == nghttp2_stream_get_state(stream)); nghttp2_bufs_reset(&bufs); /* stream 3 HEADERS; with END_STREAM flag set */ pack_headers(&bufs, &deflater, 3, NGHTTP2_FLAG_END_HEADERS | NGHTTP2_FLAG_END_STREAM, reqnv, ARRLEN(reqnv), mem); buf = &bufs.head->buf; rv = nghttp2_session_mem_recv(session, buf->pos, nghttp2_buf_len(buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(buf) == rv); stream = nghttp2_session_find_stream(session, 3); CU_ASSERT(NULL != stream); CU_ASSERT(3 == stream->stream_id); CU_ASSERT(NGHTTP2_STREAM_STATE_HALF_CLOSED_REMOTE == nghttp2_stream_get_state(stream)); nghttp2_bufs_reset(&bufs); /* Respond to stream 1 */ nghttp2_submit_response(session, 1, resnv, ARRLEN(resnv), NULL); rv = nghttp2_session_send(session); CU_ASSERT(0 == rv); stream = nghttp2_session_find_stream(session, 1); CU_ASSERT(NGHTTP2_STREAM_STATE_HALF_CLOSED_LOCAL == nghttp2_stream_get_state(stream)); /* Respond to stream 3 */ nghttp2_submit_response(session, 3, resnv, ARRLEN(resnv), NULL); rv = nghttp2_session_send(session); CU_ASSERT(0 == rv); stream = nghttp2_session_find_stream(session, 3); CU_ASSERT(NGHTTP2_STREAM_STATE_CLOSED == nghttp2_stream_get_state(stream)); /* stream 5 HEADERS; with END_STREAM flag set */ pack_headers(&bufs, &deflater, 5, NGHTTP2_FLAG_END_HEADERS | NGHTTP2_FLAG_END_STREAM, reqnv, ARRLEN(reqnv), mem); buf = &bufs.head->buf; rv = nghttp2_session_mem_recv(session, buf->pos, nghttp2_buf_len(buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(buf) == rv); nghttp2_bufs_reset(&bufs); /* Push stream 2 associated to stream 5 */ rv = nghttp2_submit_push_promise(session, NGHTTP2_FLAG_NONE, 5, reqnv, ARRLEN(reqnv), NULL); CU_ASSERT(2 == rv); rv = nghttp2_session_send(session); CU_ASSERT(0 == rv); stream = nghttp2_session_find_stream(session, 2); CU_ASSERT(NGHTTP2_STREAM_STATE_RESERVED_LOCAL == nghttp2_stream_get_state(stream)); /* Send resposne to push stream 2 with END_STREAM set */ nghttp2_submit_response(session, 2, resnv, ARRLEN(resnv), NULL); rv = nghttp2_session_send(session); CU_ASSERT(0 == rv); stream = nghttp2_session_find_stream(session, 2); /* At server, pushed stream object is not retained after closed */ CU_ASSERT(NULL == stream); /* Push stream 4 associated to stream 5 */ rv = nghttp2_submit_push_promise(session, NGHTTP2_FLAG_NONE, 5, reqnv, ARRLEN(reqnv), NULL); CU_ASSERT(4 == rv); rv = nghttp2_session_send(session); CU_ASSERT(0 == rv); stream = nghttp2_session_find_stream(session, 4); CU_ASSERT(NGHTTP2_STREAM_STATE_RESERVED_LOCAL == nghttp2_stream_get_state(stream)); /* Send response to push stream 4 without closing */ data_prd.read_callback = defer_data_source_read_callback; nghttp2_submit_response(session, 4, resnv, ARRLEN(resnv), &data_prd); rv = nghttp2_session_send(session); CU_ASSERT(0 == rv); stream = nghttp2_session_find_stream(session, 4); CU_ASSERT(NGHTTP2_STREAM_STATE_HALF_CLOSED_REMOTE == nghttp2_stream_get_state(stream)); /* Create idle stream by PRIORITY frame */ nghttp2_frame_priority_init(&frame.priority, 7, &pri_spec_default); rv = nghttp2_frame_pack_priority(&bufs, &frame.priority); CU_ASSERT(0 == rv); nghttp2_frame_priority_free(&frame.priority); buf = &bufs.head->buf; rv = nghttp2_session_mem_recv(session, buf->pos, nghttp2_buf_len(buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(buf) == rv); stream = nghttp2_session_find_stream(session, 7); CU_ASSERT(NGHTTP2_STREAM_STATE_IDLE == nghttp2_stream_get_state(stream)); nghttp2_bufs_reset(&bufs); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); /* Test for client side */ nghttp2_session_client_new(&session, &callbacks, NULL); nghttp2_hd_deflate_init(&deflater, mem); nghttp2_submit_request(session, NULL, reqnv, ARRLEN(reqnv), NULL, NULL); rv = nghttp2_session_send(session); CU_ASSERT(0 == rv); /* Receive PUSH_PROMISE 2 associated to stream 1 */ pack_push_promise(&bufs, &deflater, 1, NGHTTP2_FLAG_END_HEADERS, 2, reqnv, ARRLEN(reqnv), mem); buf = &bufs.head->buf; rv = nghttp2_session_mem_recv(session, buf->pos, nghttp2_buf_len(buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(buf) == rv); stream = nghttp2_session_find_stream(session, 2); CU_ASSERT(NGHTTP2_STREAM_STATE_RESERVED_REMOTE == nghttp2_stream_get_state(stream)); nghttp2_bufs_reset(&bufs); /* Receive push response for stream 2 without END_STREAM set */ pack_headers(&bufs, &deflater, 2, NGHTTP2_FLAG_END_HEADERS, resnv, ARRLEN(resnv), mem); buf = &bufs.head->buf; rv = nghttp2_session_mem_recv(session, buf->pos, nghttp2_buf_len(buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(buf) == rv); stream = nghttp2_session_find_stream(session, 2); CU_ASSERT(NGHTTP2_STREAM_STATE_HALF_CLOSED_LOCAL == nghttp2_stream_get_state(stream)); nghttp2_bufs_reset(&bufs); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); nghttp2_bufs_free(&bufs); } void test_nghttp2_session_stream_get_something(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_stream *a, *b, *c; memset(&callbacks, 0, sizeof(callbacks)); nghttp2_session_server_new(&session, &callbacks, NULL); a = open_stream(session, 1); CU_ASSERT(nghttp2_session_get_root_stream(session) == nghttp2_stream_get_parent(a)); CU_ASSERT(NULL == nghttp2_stream_get_previous_sibling(a)); CU_ASSERT(NULL == nghttp2_stream_get_next_sibling(a)); CU_ASSERT(NULL == nghttp2_stream_get_first_child(a)); b = open_stream_with_dep(session, 3, a); c = open_stream_with_dep_weight(session, 5, 11, a); CU_ASSERT(a == nghttp2_stream_get_parent(c)); CU_ASSERT(a == nghttp2_stream_get_parent(b)); CU_ASSERT(c == nghttp2_stream_get_first_child(a)); CU_ASSERT(b == nghttp2_stream_get_next_sibling(c)); CU_ASSERT(c == nghttp2_stream_get_previous_sibling(b)); CU_ASSERT(27 == nghttp2_stream_get_sum_dependency_weight(a)); CU_ASSERT(11 == nghttp2_stream_get_weight(c)); CU_ASSERT(5 == nghttp2_stream_get_stream_id(c)); CU_ASSERT(0 == nghttp2_stream_get_stream_id(&session->root)); nghttp2_session_del(session); } void test_nghttp2_session_find_stream(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_stream *stream; memset(&callbacks, 0, sizeof(callbacks)); nghttp2_session_server_new(&session, &callbacks, NULL); open_stream(session, 1); stream = nghttp2_session_find_stream(session, 1); CU_ASSERT(NULL != stream); CU_ASSERT(1 == stream->stream_id); stream = nghttp2_session_find_stream(session, 0); CU_ASSERT(&session->root == stream); CU_ASSERT(0 == stream->stream_id); stream = nghttp2_session_find_stream(session, 2); CU_ASSERT(NULL == stream); nghttp2_session_del(session); } void test_nghttp2_session_keep_closed_stream(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; const size_t max_concurrent_streams = 5; nghttp2_settings_entry iv = {NGHTTP2_SETTINGS_MAX_CONCURRENT_STREAMS, max_concurrent_streams}; size_t i; memset(&callbacks, 0, sizeof(callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_server_new(&session, &callbacks, NULL); nghttp2_submit_settings(session, NGHTTP2_FLAG_NONE, &iv, 1); for (i = 0; i < max_concurrent_streams; ++i) { open_stream(session, (int)i * 2 + 1); } CU_ASSERT(0 == session->num_closed_streams); nghttp2_session_close_stream(session, 1, NGHTTP2_NO_ERROR); CU_ASSERT(1 == session->num_closed_streams); CU_ASSERT(1 == session->closed_stream_tail->stream_id); CU_ASSERT(session->closed_stream_tail == session->closed_stream_head); nghttp2_session_close_stream(session, 5, NGHTTP2_NO_ERROR); CU_ASSERT(2 == session->num_closed_streams); CU_ASSERT(5 == session->closed_stream_tail->stream_id); CU_ASSERT(1 == session->closed_stream_head->stream_id); CU_ASSERT(session->closed_stream_head == session->closed_stream_tail->closed_prev); CU_ASSERT(NULL == session->closed_stream_tail->closed_next); CU_ASSERT(session->closed_stream_tail == session->closed_stream_head->closed_next); CU_ASSERT(NULL == session->closed_stream_head->closed_prev); open_stream(session, 11); CU_ASSERT(1 == session->num_closed_streams); CU_ASSERT(5 == session->closed_stream_tail->stream_id); CU_ASSERT(session->closed_stream_tail == session->closed_stream_head); CU_ASSERT(NULL == session->closed_stream_head->closed_prev); CU_ASSERT(NULL == session->closed_stream_head->closed_next); open_stream(session, 13); CU_ASSERT(0 == session->num_closed_streams); CU_ASSERT(NULL == session->closed_stream_tail); CU_ASSERT(NULL == session->closed_stream_head); nghttp2_session_close_stream(session, 3, NGHTTP2_NO_ERROR); CU_ASSERT(1 == session->num_closed_streams); CU_ASSERT(3 == session->closed_stream_head->stream_id); /* server initiated stream is not counted to max concurrent limit */ open_stream(session, 2); CU_ASSERT(1 == session->num_closed_streams); CU_ASSERT(3 == session->closed_stream_head->stream_id); nghttp2_session_close_stream(session, 2, NGHTTP2_NO_ERROR); CU_ASSERT(1 == session->num_closed_streams); CU_ASSERT(3 == session->closed_stream_head->stream_id); nghttp2_session_del(session); } void test_nghttp2_session_keep_idle_stream(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; const size_t max_concurrent_streams = 1; nghttp2_settings_entry iv = {NGHTTP2_SETTINGS_MAX_CONCURRENT_STREAMS, max_concurrent_streams}; int i; memset(&callbacks, 0, sizeof(callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_server_new(&session, &callbacks, NULL); nghttp2_submit_settings(session, NGHTTP2_FLAG_NONE, &iv, 1); /* We at least allow 2 idle streams even if max concurrent streams is very low. */ for (i = 0; i < 2; ++i) { nghttp2_session_open_stream(session, i * 2 + 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_IDLE, NULL); } CU_ASSERT(2 == session->num_idle_streams); CU_ASSERT(1 == session->idle_stream_head->stream_id); CU_ASSERT(3 == session->idle_stream_tail->stream_id); nghttp2_session_open_stream(session, 5, NGHTTP2_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_IDLE, NULL); CU_ASSERT(2 == session->num_idle_streams); CU_ASSERT(3 == session->idle_stream_head->stream_id); CU_ASSERT(5 == session->idle_stream_tail->stream_id); nghttp2_session_del(session); } void test_nghttp2_session_detach_idle_stream(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; int i; nghttp2_stream *stream; memset(&callbacks, 0, sizeof(callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_server_new(&session, &callbacks, NULL); for (i = 1; i <= 3; ++i) { nghttp2_session_open_stream(session, i, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_IDLE, NULL); } CU_ASSERT(3 == session->num_idle_streams); /* Detach middle stream */ stream = nghttp2_session_get_stream_raw(session, 2); CU_ASSERT(session->idle_stream_head == stream->closed_prev); CU_ASSERT(session->idle_stream_tail == stream->closed_next); CU_ASSERT(stream == session->idle_stream_head->closed_next); CU_ASSERT(stream == session->idle_stream_tail->closed_prev); nghttp2_session_detach_idle_stream(session, stream); CU_ASSERT(2 == session->num_idle_streams); CU_ASSERT(NULL == stream->closed_prev); CU_ASSERT(NULL == stream->closed_next); CU_ASSERT(session->idle_stream_head == session->idle_stream_tail->closed_prev); CU_ASSERT(session->idle_stream_tail == session->idle_stream_head->closed_next); /* Detach head stream */ stream = session->idle_stream_head; nghttp2_session_detach_idle_stream(session, stream); CU_ASSERT(1 == session->num_idle_streams); CU_ASSERT(session->idle_stream_head == session->idle_stream_tail); CU_ASSERT(NULL == session->idle_stream_head->closed_prev); CU_ASSERT(NULL == session->idle_stream_head->closed_next); /* Detach last stream */ stream = session->idle_stream_head; nghttp2_session_detach_idle_stream(session, stream); CU_ASSERT(0 == session->num_idle_streams); CU_ASSERT(NULL == session->idle_stream_head); CU_ASSERT(NULL == session->idle_stream_tail); for (i = 4; i <= 5; ++i) { nghttp2_session_open_stream(session, i, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_IDLE, NULL); } CU_ASSERT(2 == session->num_idle_streams); /* Detach tail stream */ stream = session->idle_stream_tail; nghttp2_session_detach_idle_stream(session, stream); CU_ASSERT(1 == session->num_idle_streams); CU_ASSERT(session->idle_stream_head == session->idle_stream_tail); CU_ASSERT(NULL == session->idle_stream_head->closed_prev); CU_ASSERT(NULL == session->idle_stream_head->closed_next); nghttp2_session_del(session); } void test_nghttp2_session_large_dep_tree(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; size_t i; nghttp2_stream *dep_stream = NULL; nghttp2_stream *stream; int32_t stream_id; memset(&callbacks, 0, sizeof(callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_server_new(&session, &callbacks, NULL); stream_id = 1; for (i = 0; i < 250; ++i, stream_id += 2) { dep_stream = open_stream_with_dep(session, stream_id, dep_stream); } stream_id = 1; for (i = 0; i < 250; ++i, stream_id += 2) { stream = nghttp2_session_get_stream(session, stream_id); CU_ASSERT(nghttp2_stream_dep_find_ancestor(stream, &session->root)); CU_ASSERT(nghttp2_stream_in_dep_tree(stream)); } nghttp2_session_del(session); } void test_nghttp2_session_graceful_shutdown(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data ud; memset(&callbacks, 0, sizeof(callbacks)); callbacks.send_callback = null_send_callback; callbacks.on_frame_send_callback = on_frame_send_callback; callbacks.on_stream_close_callback = on_stream_close_callback; nghttp2_session_server_new(&session, &callbacks, &ud); open_stream(session, 301); open_stream(session, 302); open_stream(session, 309); open_stream(session, 311); open_stream(session, 319); CU_ASSERT(0 == nghttp2_submit_shutdown_notice(session)); ud.frame_send_cb_called = 0; CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(1 == ud.frame_send_cb_called); CU_ASSERT((1u << 31) - 1 == session->local_last_stream_id); CU_ASSERT(0 == nghttp2_submit_goaway(session, NGHTTP2_FLAG_NONE, 311, NGHTTP2_NO_ERROR, NULL, 0)); ud.frame_send_cb_called = 0; ud.stream_close_cb_called = 0; CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(1 == ud.frame_send_cb_called); CU_ASSERT(311 == session->local_last_stream_id); CU_ASSERT(1 == ud.stream_close_cb_called); CU_ASSERT(0 == nghttp2_session_terminate_session2(session, 301, NGHTTP2_NO_ERROR)); ud.frame_send_cb_called = 0; ud.stream_close_cb_called = 0; CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(1 == ud.frame_send_cb_called); CU_ASSERT(301 == session->local_last_stream_id); CU_ASSERT(2 == ud.stream_close_cb_called); CU_ASSERT(NULL != nghttp2_session_get_stream(session, 301)); CU_ASSERT(NULL != nghttp2_session_get_stream(session, 302)); CU_ASSERT(NULL == nghttp2_session_get_stream(session, 309)); CU_ASSERT(NULL == nghttp2_session_get_stream(session, 311)); CU_ASSERT(NULL == nghttp2_session_get_stream(session, 319)); nghttp2_session_del(session); } void test_nghttp2_session_on_header_temporal_failure(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data ud; nghttp2_bufs bufs; nghttp2_buf *buf; nghttp2_hd_deflater deflater; nghttp2_nv nv[] = {MAKE_NV("alpha", "bravo"), MAKE_NV("charlie", "delta")}; nghttp2_nv *nva; size_t hdpos; ssize_t rv; nghttp2_frame frame; nghttp2_frame_hd hd; nghttp2_outbound_item *item; nghttp2_mem *mem; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(callbacks)); callbacks.on_header_callback = temporal_failure_on_header_callback; nghttp2_session_server_new(&session, &callbacks, &ud); frame_pack_bufs_init(&bufs); nghttp2_hd_deflate_init(&deflater, mem); nghttp2_nv_array_copy(&nva, reqnv, ARRLEN(reqnv), mem); nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_STREAM, 1, NGHTTP2_HCAT_REQUEST, NULL, nva, ARRLEN(reqnv)); nghttp2_frame_pack_headers(&bufs, &frame.headers, &deflater); nghttp2_frame_headers_free(&frame.headers, mem); /* We are going to create CONTINUATION. First serialize header block, and then frame header. */ hdpos = nghttp2_bufs_len(&bufs); buf = &bufs.head->buf; buf->last += NGHTTP2_FRAME_HDLEN; nghttp2_hd_deflate_hd_bufs(&deflater, &bufs, &nv[1], 1); nghttp2_frame_hd_init(&hd, nghttp2_bufs_len(&bufs) - hdpos - NGHTTP2_FRAME_HDLEN, NGHTTP2_CONTINUATION, NGHTTP2_FLAG_END_HEADERS, 1); nghttp2_frame_pack_frame_hd(&buf->pos[hdpos], &hd); ud.header_cb_called = 0; rv = nghttp2_session_mem_recv(session, buf->pos, nghttp2_bufs_len(&bufs)); CU_ASSERT((ssize_t)nghttp2_bufs_len(&bufs) == rv); CU_ASSERT(1 == ud.header_cb_called); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_RST_STREAM == item->frame.hd.type); CU_ASSERT(1 == item->frame.hd.stream_id); /* Make sure no header decompression error occurred */ CU_ASSERT(NGHTTP2_GOAWAY_NONE == session->goaway_flags); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); nghttp2_bufs_reset(&bufs); /* Check for PUSH_PROMISE */ nghttp2_hd_deflate_init(&deflater, mem); nghttp2_session_client_new(&session, &callbacks, &ud); nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); rv = pack_push_promise(&bufs, &deflater, 1, NGHTTP2_FLAG_END_HEADERS, 2, reqnv, ARRLEN(reqnv), mem); CU_ASSERT(0 == rv); ud.header_cb_called = 0; rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_bufs_len(&bufs)); CU_ASSERT((ssize_t)nghttp2_bufs_len(&bufs) == rv); CU_ASSERT(1 == ud.header_cb_called); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_RST_STREAM == item->frame.hd.type); CU_ASSERT(2 == item->frame.hd.stream_id); CU_ASSERT(NGHTTP2_INTERNAL_ERROR == item->frame.rst_stream.error_code); nghttp2_session_del(session); nghttp2_hd_deflate_free(&deflater); nghttp2_bufs_free(&bufs); } void test_nghttp2_session_recv_client_magic(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; ssize_t rv; nghttp2_frame ping_frame; uint8_t buf[16]; /* enable global nghttp2_enable_strict_preface here */ nghttp2_enable_strict_preface = 1; memset(&callbacks, 0, sizeof(callbacks)); /* Check success case */ nghttp2_session_server_new(&session, &callbacks, NULL); rv = nghttp2_session_mem_recv(session, (const uint8_t *)NGHTTP2_CLIENT_MAGIC, NGHTTP2_CLIENT_MAGIC_LEN); CU_ASSERT(rv == NGHTTP2_CLIENT_MAGIC_LEN); CU_ASSERT(NGHTTP2_IB_READ_FIRST_SETTINGS == session->iframe.state); /* Receiving PING is error because we want SETTINGS. */ nghttp2_frame_ping_init(&ping_frame.ping, NGHTTP2_FLAG_NONE, NULL); nghttp2_frame_pack_frame_hd(buf, &ping_frame.ping.hd); rv = nghttp2_session_mem_recv(session, buf, NGHTTP2_FRAME_HDLEN); CU_ASSERT(NGHTTP2_FRAME_HDLEN == rv); CU_ASSERT(NGHTTP2_IB_IGN_ALL == session->iframe.state); CU_ASSERT(0 == session->iframe.payloadleft); nghttp2_frame_ping_free(&ping_frame.ping); nghttp2_session_del(session); /* Check bad case */ nghttp2_session_server_new(&session, &callbacks, NULL); /* Feed magic with one byte less */ rv = nghttp2_session_mem_recv(session, (const uint8_t *)NGHTTP2_CLIENT_MAGIC, NGHTTP2_CLIENT_MAGIC_LEN - 1); CU_ASSERT(rv == NGHTTP2_CLIENT_MAGIC_LEN - 1); CU_ASSERT(NGHTTP2_IB_READ_CLIENT_MAGIC == session->iframe.state); CU_ASSERT(1 == session->iframe.payloadleft); rv = nghttp2_session_mem_recv(session, (const uint8_t *)"\0", 1); CU_ASSERT(NGHTTP2_ERR_BAD_CLIENT_MAGIC == rv); nghttp2_session_del(session); /* disable global nghttp2_enable_strict_preface here */ nghttp2_enable_strict_preface = 0; } void test_nghttp2_session_delete_data_item(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_stream *a; nghttp2_data_provider prd; memset(&callbacks, 0, sizeof(callbacks)); nghttp2_session_server_new(&session, &callbacks, NULL); a = open_stream(session, 1); open_stream_with_dep(session, 3, a); /* We don't care about these members, since we won't send data */ prd.source.ptr = NULL; prd.read_callback = fail_data_source_read_callback; /* This data item will be marked as TOP */ CU_ASSERT(0 == nghttp2_submit_data(session, NGHTTP2_FLAG_NONE, 1, &prd)); /* This data item will be marked as REST */ CU_ASSERT(0 == nghttp2_submit_data(session, NGHTTP2_FLAG_NONE, 3, &prd)); nghttp2_session_del(session); } void test_nghttp2_session_open_idle_stream(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_stream *stream; nghttp2_stream *opened_stream; nghttp2_priority_spec pri_spec; nghttp2_frame frame; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); nghttp2_session_server_new(&session, &callbacks, NULL); nghttp2_priority_spec_init(&pri_spec, 0, 3, 0); nghttp2_frame_priority_init(&frame.priority, 1, &pri_spec); CU_ASSERT(0 == nghttp2_session_on_priority_received(session, &frame)); stream = nghttp2_session_get_stream_raw(session, 1); CU_ASSERT(NGHTTP2_STREAM_IDLE == stream->state); CU_ASSERT(NULL == stream->closed_prev); CU_ASSERT(NULL == stream->closed_next); CU_ASSERT(1 == session->num_idle_streams); CU_ASSERT(session->idle_stream_head == stream); CU_ASSERT(session->idle_stream_tail == stream); opened_stream = nghttp2_session_open_stream( session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); CU_ASSERT(stream == opened_stream); CU_ASSERT(NGHTTP2_STREAM_OPENING == stream->state); CU_ASSERT(0 == session->num_idle_streams); CU_ASSERT(NULL == session->idle_stream_head); CU_ASSERT(NULL == session->idle_stream_tail); nghttp2_frame_priority_free(&frame.priority); nghttp2_session_del(session); } void test_nghttp2_session_cancel_reserved_remote(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_stream *stream; nghttp2_frame frame; nghttp2_nv *nva; size_t nvlen; nghttp2_hd_deflater deflater; nghttp2_mem *mem; nghttp2_bufs bufs; ssize_t rv; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_client_new(&session, &callbacks, NULL); nghttp2_hd_deflate_init(&deflater, mem); stream = nghttp2_session_open_stream(session, 2, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_RESERVED, NULL); session->last_recv_stream_id = 2; nghttp2_submit_rst_stream(session, NGHTTP2_FLAG_NONE, 2, NGHTTP2_CANCEL); CU_ASSERT(NGHTTP2_STREAM_CLOSING == stream->state); CU_ASSERT(0 == nghttp2_session_send(session)); nvlen = ARRLEN(resnv); nghttp2_nv_array_copy(&nva, resnv, nvlen, mem); nghttp2_frame_headers_init(&frame.headers, NGHTTP2_FLAG_END_HEADERS, 2, NGHTTP2_HCAT_PUSH_RESPONSE, NULL, nva, nvlen); rv = nghttp2_frame_pack_headers(&bufs, &frame.headers, &deflater); CU_ASSERT(0 == rv); rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); /* stream is not dangling, so assign NULL */ stream = NULL; /* No RST_STREAM or GOAWAY is generated since stream should be in NGHTTP2_STREAM_CLOSING and push response should be ignored. */ CU_ASSERT(0 == nghttp2_outbound_queue_size(&session->ob_reg)); /* Check that we can receive push response HEADERS while RST_STREAM is just queued. */ nghttp2_session_open_stream(session, 4, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_RESERVED, NULL); session->last_recv_stream_id = 4; nghttp2_submit_rst_stream(session, NGHTTP2_FLAG_NONE, 2, NGHTTP2_CANCEL); nghttp2_bufs_reset(&bufs); frame.hd.stream_id = 4; rv = nghttp2_frame_pack_headers(&bufs, &frame.headers, &deflater); CU_ASSERT(0 == rv); rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); CU_ASSERT(1 == nghttp2_outbound_queue_size(&session->ob_reg)); nghttp2_frame_headers_free(&frame.headers, mem); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); nghttp2_bufs_free(&bufs); } void test_nghttp2_session_reset_pending_headers(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_stream *stream; int32_t stream_id; my_user_data ud; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; callbacks.on_frame_send_callback = on_frame_send_callback; callbacks.on_frame_not_send_callback = on_frame_not_send_callback; callbacks.on_stream_close_callback = on_stream_close_callback; nghttp2_session_client_new(&session, &callbacks, &ud); stream_id = nghttp2_submit_request(session, NULL, NULL, 0, NULL, NULL); CU_ASSERT(stream_id >= 1); nghttp2_submit_rst_stream(session, NGHTTP2_FLAG_NONE, stream_id, NGHTTP2_CANCEL); session->remote_settings.max_concurrent_streams = 0; /* RST_STREAM cancels pending HEADERS and is not actually sent. */ ud.frame_send_cb_called = 0; CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(0 == ud.frame_send_cb_called); stream = nghttp2_session_get_stream(session, stream_id); CU_ASSERT(NULL == stream); /* See HEADERS is not sent. on_stream_close is called just like transmission failure. */ session->remote_settings.max_concurrent_streams = 1; ud.frame_not_send_cb_called = 0; ud.stream_close_error_code = 0; CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT(1 == ud.frame_not_send_cb_called); CU_ASSERT(NGHTTP2_HEADERS == ud.not_sent_frame_type); CU_ASSERT(NGHTTP2_CANCEL == ud.stream_close_error_code); stream = nghttp2_session_get_stream(session, stream_id); CU_ASSERT(NULL == stream); nghttp2_session_del(session); } void test_nghttp2_session_send_data_callback(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_data_provider data_prd; my_user_data ud; accumulator acc; nghttp2_frame_hd hd; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = accumulator_send_callback; callbacks.send_data_callback = send_data_callback; data_prd.read_callback = no_copy_data_source_read_callback; acc.length = 0; ud.acc = &acc; ud.data_source_length = NGHTTP2_DATA_PAYLOADLEN * 2; nghttp2_session_client_new(&session, &callbacks, &ud); nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); nghttp2_submit_data(session, NGHTTP2_FLAG_END_STREAM, 1, &data_prd); CU_ASSERT(0 == nghttp2_session_send(session)); CU_ASSERT((NGHTTP2_FRAME_HDLEN + NGHTTP2_DATA_PAYLOADLEN) * 2 == acc.length); nghttp2_frame_unpack_frame_hd(&hd, acc.buf); CU_ASSERT(16384 == hd.length); CU_ASSERT(NGHTTP2_DATA == hd.type); CU_ASSERT(NGHTTP2_FLAG_NONE == hd.flags); nghttp2_frame_unpack_frame_hd(&hd, acc.buf + NGHTTP2_FRAME_HDLEN + hd.length); CU_ASSERT(16384 == hd.length); CU_ASSERT(NGHTTP2_DATA == hd.type); CU_ASSERT(NGHTTP2_FLAG_END_STREAM == hd.flags); nghttp2_session_del(session); } void test_nghttp2_session_on_begin_headers_temporal_failure(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; my_user_data ud; nghttp2_bufs bufs; nghttp2_mem *mem; ssize_t rv; nghttp2_hd_deflater deflater; nghttp2_outbound_item *item; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); nghttp2_hd_deflate_init(&deflater, mem); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.on_begin_headers_callback = temporal_failure_on_begin_headers_callback; callbacks.on_header_callback = on_header_callback; callbacks.on_frame_recv_callback = on_frame_recv_callback; callbacks.send_callback = null_send_callback; nghttp2_session_server_new(&session, &callbacks, &ud); rv = pack_headers(&bufs, &deflater, 1, NGHTTP2_FLAG_END_HEADERS, reqnv, ARRLEN(reqnv), mem); CU_ASSERT(0 == rv); ud.header_cb_called = 0; ud.frame_recv_cb_called = 0; rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_bufs_len(&bufs)); CU_ASSERT((ssize_t)nghttp2_bufs_len(&bufs) == rv); CU_ASSERT(0 == ud.header_cb_called); CU_ASSERT(0 == ud.frame_recv_cb_called); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_RST_STREAM == item->frame.hd.type); CU_ASSERT(1 == item->frame.hd.stream_id); CU_ASSERT(NGHTTP2_INTERNAL_ERROR == item->frame.rst_stream.error_code); nghttp2_session_del(session); nghttp2_hd_deflate_free(&deflater); nghttp2_bufs_reset(&bufs); /* check for PUSH_PROMISE */ nghttp2_hd_deflate_init(&deflater, mem); nghttp2_session_client_new(&session, &callbacks, &ud); nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); rv = pack_push_promise(&bufs, &deflater, 1, NGHTTP2_FLAG_END_HEADERS, 2, reqnv, ARRLEN(reqnv), mem); CU_ASSERT(0 == rv); ud.header_cb_called = 0; ud.frame_recv_cb_called = 0; rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_bufs_len(&bufs)); CU_ASSERT((ssize_t)nghttp2_bufs_len(&bufs) == rv); CU_ASSERT(0 == ud.header_cb_called); CU_ASSERT(0 == ud.frame_recv_cb_called); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_RST_STREAM == item->frame.hd.type); CU_ASSERT(2 == item->frame.hd.stream_id); CU_ASSERT(NGHTTP2_INTERNAL_ERROR == item->frame.rst_stream.error_code); nghttp2_session_del(session); nghttp2_hd_deflate_free(&deflater); nghttp2_bufs_free(&bufs); } void test_nghttp2_session_defer_then_close(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_data_provider prd; int rv; const uint8_t *datap; ssize_t datalen; nghttp2_frame frame; memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_client_new(&session, &callbacks, NULL); prd.read_callback = defer_data_source_read_callback; rv = nghttp2_submit_request(session, NULL, reqnv, ARRLEN(reqnv), &prd, NULL); CU_ASSERT(rv > 0); /* This sends HEADERS */ datalen = nghttp2_session_mem_send(session, &datap); CU_ASSERT(datalen > 0); /* This makes DATA item deferred */ datalen = nghttp2_session_mem_send(session, &datap); CU_ASSERT(datalen == 0); nghttp2_frame_rst_stream_init(&frame.rst_stream, 1, NGHTTP2_CANCEL); /* Assertion failure; GH-264 */ rv = nghttp2_session_on_rst_stream_received(session, &frame); CU_ASSERT(rv == 0); nghttp2_session_del(session); } static int submit_response_on_stream_close(nghttp2_session *session, int32_t stream_id, uint32_t error_code _U_, void *user_data _U_) { nghttp2_data_provider data_prd; data_prd.read_callback = temporal_failure_data_source_read_callback; // Attempt to submit response or data to the stream being closed switch (stream_id) { case 1: CU_ASSERT(0 == nghttp2_submit_response(session, stream_id, resnv, ARRLEN(resnv), &data_prd)); break; case 3: CU_ASSERT(0 == nghttp2_submit_data(session, NGHTTP2_FLAG_NONE, stream_id, &data_prd)); break; } return 0; } void test_nghttp2_session_detach_item_from_closed_stream(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; memset(&callbacks, 0, sizeof(callbacks)); callbacks.send_callback = null_send_callback; callbacks.on_stream_close_callback = submit_response_on_stream_close; nghttp2_session_server_new(&session, &callbacks, NULL); open_stream(session, 1); open_stream(session, 3); nghttp2_session_close_stream(session, 1, NGHTTP2_NO_ERROR); nghttp2_session_close_stream(session, 3, NGHTTP2_NO_ERROR); CU_ASSERT(0 == nghttp2_session_send(session)); nghttp2_session_del(session); } void test_nghttp2_session_flooding(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_bufs bufs; nghttp2_buf *buf; nghttp2_frame frame; nghttp2_mem *mem; size_t i; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); memset(&callbacks, 0, sizeof(callbacks)); /* PING ACK */ nghttp2_session_server_new(&session, &callbacks, NULL); nghttp2_frame_ping_init(&frame.ping, NGHTTP2_FLAG_NONE, NULL); nghttp2_frame_pack_ping(&bufs, &frame.ping); nghttp2_frame_ping_free(&frame.ping); buf = &bufs.head->buf; for (i = 0; i < NGHTTP2_MAX_OBQ_FLOOD_ITEM; ++i) { CU_ASSERT( (ssize_t)nghttp2_buf_len(buf) == nghttp2_session_mem_recv(session, buf->pos, nghttp2_buf_len(buf))); } CU_ASSERT(NGHTTP2_ERR_FLOODED == nghttp2_session_mem_recv(session, buf->pos, nghttp2_buf_len(buf))); nghttp2_session_del(session); /* SETTINGS ACK */ nghttp2_bufs_reset(&bufs); nghttp2_session_server_new(&session, &callbacks, NULL); nghttp2_frame_settings_init(&frame.settings, NGHTTP2_FLAG_NONE, NULL, 0); nghttp2_frame_pack_settings(&bufs, &frame.settings); nghttp2_frame_settings_free(&frame.settings, mem); buf = &bufs.head->buf; for (i = 0; i < NGHTTP2_MAX_OBQ_FLOOD_ITEM; ++i) { CU_ASSERT( (ssize_t)nghttp2_buf_len(buf) == nghttp2_session_mem_recv(session, buf->pos, nghttp2_buf_len(buf))); } CU_ASSERT(NGHTTP2_ERR_FLOODED == nghttp2_session_mem_recv(session, buf->pos, nghttp2_buf_len(buf))); nghttp2_session_del(session); nghttp2_bufs_free(&bufs); } void test_nghttp2_session_change_stream_priority(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_stream *stream1, *stream2, *stream3; nghttp2_priority_spec pri_spec; int rv; memset(&callbacks, 0, sizeof(callbacks)); nghttp2_session_server_new(&session, &callbacks, NULL); stream1 = open_stream(session, 1); stream3 = open_stream_with_dep_weight(session, 3, 199, stream1); stream2 = open_stream_with_dep_weight(session, 2, 101, stream3); nghttp2_priority_spec_init(&pri_spec, 1, 256, 0); rv = nghttp2_session_change_stream_priority(session, 2, &pri_spec); CU_ASSERT(0 == rv); CU_ASSERT(stream1 == stream2->dep_prev); CU_ASSERT(256 == stream2->weight); /* Cannot change stream which does not exist */ rv = nghttp2_session_change_stream_priority(session, 5, &pri_spec); CU_ASSERT(NGHTTP2_ERR_INVALID_ARGUMENT == rv); /* It is an error to depend on itself */ rv = nghttp2_session_change_stream_priority(session, 1, &pri_spec); CU_ASSERT(NGHTTP2_ERR_INVALID_ARGUMENT == rv); /* It is an error to change priority of root stream (0) */ rv = nghttp2_session_change_stream_priority(session, 0, &pri_spec); CU_ASSERT(NGHTTP2_ERR_INVALID_ARGUMENT == rv); nghttp2_session_del(session); } void test_nghttp2_session_create_idle_stream(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_stream *stream2, *stream4, *stream8, *stream10; nghttp2_priority_spec pri_spec; int rv; memset(&callbacks, 0, sizeof(callbacks)); nghttp2_session_server_new(&session, &callbacks, NULL); stream2 = open_stream(session, 2); nghttp2_priority_spec_init(&pri_spec, 2, 111, 1); rv = nghttp2_session_create_idle_stream(session, 4, &pri_spec); CU_ASSERT(0 == rv); stream4 = nghttp2_session_get_stream_raw(session, 4); CU_ASSERT(4 == stream4->stream_id); CU_ASSERT(111 == stream4->weight); CU_ASSERT(stream2 == stream4->dep_prev); CU_ASSERT(stream4 == stream2->dep_next); /* If pri_spec->stream_id does not exist, and it is idle stream, it is created too */ nghttp2_priority_spec_init(&pri_spec, 8, 109, 0); rv = nghttp2_session_create_idle_stream(session, 8, &pri_spec); CU_ASSERT(0 == rv); stream8 = nghttp2_session_get_stream_raw(session, 8); stream10 = nghttp2_session_get_stream_raw(session, 10); CU_ASSERT(8 == stream8->stream_id); CU_ASSERT(109 == stream8->weight); CU_ASSERT(10 == stream10->stream_id); CU_ASSERT(16 == stream10->weight); CU_ASSERT(stream10 == stream8->dep_prev); CU_ASSERT(&session->root == stream10->dep_prev); /* It is an error to attempt to create already existing idle stream */ rv = nghttp2_session_create_idle_stream(session, 4, &pri_spec); CU_ASSERT(NGHTTP2_ERR_INVALID_ARGUMENT == rv); /* It is an error to depend on itself */ pri_spec.stream_id = 6; rv = nghttp2_session_create_idle_stream(session, 6, &pri_spec); CU_ASSERT(NGHTTP2_ERR_INVALID_ARGUMENT == rv); /* It is an error to create root stream (0) as idle stream */ rv = nghttp2_session_create_idle_stream(session, 0, &pri_spec); CU_ASSERT(NGHTTP2_ERR_INVALID_ARGUMENT == rv); /* It is an error to create non-idle stream */ session->next_stream_id = 20; pri_spec.stream_id = 2; rv = nghttp2_session_create_idle_stream(session, 18, &pri_spec); CU_ASSERT(NGHTTP2_ERR_INVALID_ARGUMENT == rv); nghttp2_session_del(session); } static void check_nghttp2_http_recv_headers_fail( nghttp2_session *session, nghttp2_hd_deflater *deflater, int32_t stream_id, int stream_state, const nghttp2_nv *nva, size_t nvlen) { nghttp2_mem *mem; ssize_t rv; nghttp2_outbound_item *item; nghttp2_bufs bufs; my_user_data *ud; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); ud = session->user_data; if (stream_state != -1) { nghttp2_session_open_stream(session, stream_id, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, (nghttp2_stream_state)stream_state, NULL); } rv = pack_headers(&bufs, deflater, stream_id, NGHTTP2_FLAG_END_HEADERS, nva, nvlen, mem); CU_ASSERT(0 == rv); ud->invalid_frame_recv_cb_called = 0; rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_RST_STREAM == item->frame.hd.type); CU_ASSERT(1 == ud->invalid_frame_recv_cb_called); CU_ASSERT(0 == nghttp2_session_send(session)); nghttp2_bufs_free(&bufs); } static void check_nghttp2_http_recv_headers_ok( nghttp2_session *session, nghttp2_hd_deflater *deflater, int32_t stream_id, int stream_state, const nghttp2_nv *nva, size_t nvlen) { nghttp2_mem *mem; ssize_t rv; nghttp2_bufs bufs; my_user_data *ud; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); ud = session->user_data; if (stream_state != -1) { nghttp2_session_open_stream(session, stream_id, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, (nghttp2_stream_state)stream_state, NULL); } rv = pack_headers(&bufs, deflater, stream_id, NGHTTP2_FLAG_END_HEADERS, nva, nvlen, mem); CU_ASSERT(0 == rv); ud->frame_recv_cb_called = 0; rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); CU_ASSERT(NULL == nghttp2_session_get_next_ob_item(session)); CU_ASSERT(1 == ud->frame_recv_cb_called); nghttp2_bufs_free(&bufs); } void test_nghttp2_http_mandatory_headers(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_hd_deflater deflater; nghttp2_mem *mem; my_user_data ud; /* test case for response */ const nghttp2_nv nostatus_resnv[] = {MAKE_NV("server", "foo")}; const nghttp2_nv dupstatus_resnv[] = {MAKE_NV(":status", "200"), MAKE_NV(":status", "200")}; const nghttp2_nv badpseudo_resnv[] = {MAKE_NV(":status", "200"), MAKE_NV(":scheme", "https")}; const nghttp2_nv latepseudo_resnv[] = {MAKE_NV("server", "foo"), MAKE_NV(":status", "200")}; const nghttp2_nv badstatus_resnv[] = {MAKE_NV(":status", "2000")}; const nghttp2_nv badcl_resnv[] = {MAKE_NV(":status", "200"), MAKE_NV("content-length", "-1")}; const nghttp2_nv dupcl_resnv[] = {MAKE_NV(":status", "200"), MAKE_NV("content-length", "0"), MAKE_NV("content-length", "0")}; const nghttp2_nv badhd_resnv[] = {MAKE_NV(":status", "200"), MAKE_NV("connection", "close")}; /* test case for request */ const nghttp2_nv nopath_reqnv[] = {MAKE_NV(":scheme", "https"), MAKE_NV(":method", "GET"), MAKE_NV(":authority", "localhost")}; const nghttp2_nv earlyconnect_reqnv[] = { MAKE_NV(":method", "CONNECT"), MAKE_NV(":scheme", "https"), MAKE_NV(":path", "/"), MAKE_NV(":authority", "localhost")}; const nghttp2_nv lateconnect_reqnv[] = { MAKE_NV(":scheme", "https"), MAKE_NV(":path", "/"), MAKE_NV(":method", "CONNECT"), MAKE_NV(":authority", "localhost")}; const nghttp2_nv duppath_reqnv[] = { MAKE_NV(":scheme", "https"), MAKE_NV(":method", "GET"), MAKE_NV(":authority", "localhost"), MAKE_NV(":path", "/"), MAKE_NV(":path", "/")}; const nghttp2_nv badcl_reqnv[] = { MAKE_NV(":scheme", "https"), MAKE_NV(":method", "POST"), MAKE_NV(":authority", "localhost"), MAKE_NV(":path", "/"), MAKE_NV("content-length", "-1")}; const nghttp2_nv dupcl_reqnv[] = { MAKE_NV(":scheme", "https"), MAKE_NV(":method", "POST"), MAKE_NV(":authority", "localhost"), MAKE_NV(":path", "/"), MAKE_NV("content-length", "0"), MAKE_NV("content-length", "0")}; const nghttp2_nv badhd_reqnv[] = { MAKE_NV(":scheme", "https"), MAKE_NV(":method", "GET"), MAKE_NV(":authority", "localhost"), MAKE_NV(":path", "/"), MAKE_NV("connection", "close")}; const nghttp2_nv badauthority_reqnv[] = { MAKE_NV(":scheme", "https"), MAKE_NV(":method", "GET"), MAKE_NV(":authority", "\x0d\x0alocalhost"), MAKE_NV(":path", "/")}; const nghttp2_nv badhdbtw_reqnv[] = { MAKE_NV(":scheme", "https"), MAKE_NV(":method", "GET"), MAKE_NV("foo", "\x0d\x0a"), MAKE_NV(":authority", "localhost"), MAKE_NV(":path", "/")}; const nghttp2_nv asteriskget1_reqnv[] = { MAKE_NV(":path", "*"), MAKE_NV(":scheme", "https"), MAKE_NV(":authority", "localhost"), MAKE_NV(":method", "GET")}; const nghttp2_nv asteriskget2_reqnv[] = { MAKE_NV(":scheme", "https"), MAKE_NV(":authority", "localhost"), MAKE_NV(":method", "GET"), MAKE_NV(":path", "*")}; const nghttp2_nv asteriskoptions1_reqnv[] = { MAKE_NV(":path", "*"), MAKE_NV(":scheme", "https"), MAKE_NV(":authority", "localhost"), MAKE_NV(":method", "OPTIONS")}; const nghttp2_nv asteriskoptions2_reqnv[] = { MAKE_NV(":scheme", "https"), MAKE_NV(":authority", "localhost"), MAKE_NV(":method", "OPTIONS"), MAKE_NV(":path", "*")}; mem = nghttp2_mem_default(); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; callbacks.on_frame_recv_callback = on_frame_recv_callback; callbacks.on_invalid_frame_recv_callback = on_invalid_frame_recv_callback; nghttp2_session_client_new(&session, &callbacks, &ud); nghttp2_hd_deflate_init(&deflater, mem); /* response header lacks :status */ check_nghttp2_http_recv_headers_fail(session, &deflater, 1, NGHTTP2_STREAM_OPENING, nostatus_resnv, ARRLEN(nostatus_resnv)); /* response header has 2 :status */ check_nghttp2_http_recv_headers_fail(session, &deflater, 3, NGHTTP2_STREAM_OPENING, dupstatus_resnv, ARRLEN(dupstatus_resnv)); /* response header has bad pseudo header :scheme */ check_nghttp2_http_recv_headers_fail(session, &deflater, 5, NGHTTP2_STREAM_OPENING, badpseudo_resnv, ARRLEN(badpseudo_resnv)); /* response header has :status after regular header field */ check_nghttp2_http_recv_headers_fail(session, &deflater, 7, NGHTTP2_STREAM_OPENING, latepseudo_resnv, ARRLEN(latepseudo_resnv)); /* response header has bad status code */ check_nghttp2_http_recv_headers_fail(session, &deflater, 9, NGHTTP2_STREAM_OPENING, badstatus_resnv, ARRLEN(badstatus_resnv)); /* response header has bad content-length */ check_nghttp2_http_recv_headers_fail(session, &deflater, 11, NGHTTP2_STREAM_OPENING, badcl_resnv, ARRLEN(badcl_resnv)); /* response header has multiple content-length */ check_nghttp2_http_recv_headers_fail(session, &deflater, 13, NGHTTP2_STREAM_OPENING, dupcl_resnv, ARRLEN(dupcl_resnv)); /* response header has disallowed header field */ check_nghttp2_http_recv_headers_fail(session, &deflater, 15, NGHTTP2_STREAM_OPENING, badhd_resnv, ARRLEN(badhd_resnv)); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); /* check server side */ nghttp2_session_server_new(&session, &callbacks, &ud); nghttp2_hd_deflate_init(&deflater, mem); /* request header has no :path */ check_nghttp2_http_recv_headers_fail(session, &deflater, 1, -1, nopath_reqnv, ARRLEN(nopath_reqnv)); /* request header has CONNECT method, but followed by :path */ check_nghttp2_http_recv_headers_fail(session, &deflater, 3, -1, earlyconnect_reqnv, ARRLEN(earlyconnect_reqnv)); /* request header has CONNECT method following :path */ check_nghttp2_http_recv_headers_fail( session, &deflater, 5, -1, lateconnect_reqnv, ARRLEN(lateconnect_reqnv)); /* request header has multiple :path */ check_nghttp2_http_recv_headers_fail(session, &deflater, 7, -1, duppath_reqnv, ARRLEN(duppath_reqnv)); /* request header has bad content-length */ check_nghttp2_http_recv_headers_fail(session, &deflater, 9, -1, badcl_reqnv, ARRLEN(badcl_reqnv)); /* request header has multiple content-length */ check_nghttp2_http_recv_headers_fail(session, &deflater, 11, -1, dupcl_reqnv, ARRLEN(dupcl_reqnv)); /* request header has disallowed header field */ check_nghttp2_http_recv_headers_fail(session, &deflater, 13, -1, badhd_reqnv, ARRLEN(badhd_reqnv)); /* request header has :authority header field containing illegal characters */ check_nghttp2_http_recv_headers_fail(session, &deflater, 15, -1, badauthority_reqnv, ARRLEN(badauthority_reqnv)); /* request header has regular header field containing illegal character before all mandatory header fields are seen. */ check_nghttp2_http_recv_headers_fail(session, &deflater, 17, -1, badhdbtw_reqnv, ARRLEN(badhdbtw_reqnv)); /* request header has "*" in :path header field while method is GET. :path is received before :method */ check_nghttp2_http_recv_headers_fail(session, &deflater, 19, -1, asteriskget1_reqnv, ARRLEN(asteriskget1_reqnv)); /* request header has "*" in :path header field while method is GET. :method is received before :path */ check_nghttp2_http_recv_headers_fail(session, &deflater, 21, -1, asteriskget2_reqnv, ARRLEN(asteriskget2_reqnv)); /* OPTIONS method can include "*" in :path header field. :path is received before :method. */ check_nghttp2_http_recv_headers_ok(session, &deflater, 23, -1, asteriskoptions1_reqnv, ARRLEN(asteriskoptions1_reqnv)); /* OPTIONS method can include "*" in :path header field. :method is received before :path. */ check_nghttp2_http_recv_headers_ok(session, &deflater, 25, -1, asteriskoptions2_reqnv, ARRLEN(asteriskoptions2_reqnv)); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); } void test_nghttp2_http_content_length(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_hd_deflater deflater; nghttp2_mem *mem; nghttp2_bufs bufs; ssize_t rv; nghttp2_stream *stream; const nghttp2_nv cl_resnv[] = {MAKE_NV(":status", "200"), MAKE_NV("te", "trailers"), MAKE_NV("content-length", "9000000000")}; const nghttp2_nv cl_reqnv[] = { MAKE_NV(":path", "/"), MAKE_NV(":method", "PUT"), MAKE_NV(":scheme", "https"), MAKE_NV("te", "trailers"), MAKE_NV("host", "localhost"), MAKE_NV("content-length", "9000000000")}; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_client_new(&session, &callbacks, NULL); nghttp2_hd_deflate_init(&deflater, mem); stream = nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); rv = pack_headers(&bufs, &deflater, 1, NGHTTP2_FLAG_END_HEADERS, cl_resnv, ARRLEN(cl_resnv), mem); CU_ASSERT(0 == rv); rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); CU_ASSERT(NULL == nghttp2_session_get_next_ob_item(session)); CU_ASSERT(9000000000LL == stream->content_length); CU_ASSERT(200 == stream->status_code); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); nghttp2_bufs_reset(&bufs); /* check server side */ nghttp2_session_server_new(&session, &callbacks, NULL); nghttp2_hd_deflate_init(&deflater, mem); rv = pack_headers(&bufs, &deflater, 1, NGHTTP2_FLAG_END_HEADERS, cl_reqnv, ARRLEN(cl_reqnv), mem); CU_ASSERT(0 == rv); rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); stream = nghttp2_session_get_stream(session, 1); CU_ASSERT(NULL == nghttp2_session_get_next_ob_item(session)); CU_ASSERT(9000000000LL == stream->content_length); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); nghttp2_bufs_free(&bufs); } void test_nghttp2_http_content_length_mismatch(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_hd_deflater deflater; nghttp2_mem *mem; nghttp2_bufs bufs; ssize_t rv; const nghttp2_nv cl_reqnv[] = { MAKE_NV(":path", "/"), MAKE_NV(":method", "PUT"), MAKE_NV(":authority", "localhost"), MAKE_NV(":scheme", "https"), MAKE_NV("content-length", "20")}; nghttp2_outbound_item *item; nghttp2_frame_hd hd; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_server_new(&session, &callbacks, NULL); nghttp2_hd_deflate_init(&deflater, mem); /* header says content-length: 20, but HEADERS has END_STREAM flag set */ rv = pack_headers(&bufs, &deflater, 1, NGHTTP2_FLAG_END_HEADERS | NGHTTP2_FLAG_END_STREAM, cl_reqnv, ARRLEN(cl_reqnv), mem); CU_ASSERT(0 == rv); rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_RST_STREAM == item->frame.hd.type); CU_ASSERT(0 == nghttp2_session_send(session)); nghttp2_bufs_reset(&bufs); /* header says content-length: 20, but DATA has 0 byte */ rv = pack_headers(&bufs, &deflater, 3, NGHTTP2_FLAG_END_HEADERS, cl_reqnv, ARRLEN(cl_reqnv), mem); CU_ASSERT(0 == rv); nghttp2_frame_hd_init(&hd, 0, NGHTTP2_DATA, NGHTTP2_FLAG_END_STREAM, 3); nghttp2_frame_pack_frame_hd(bufs.head->buf.last, &hd); bufs.head->buf.last += NGHTTP2_FRAME_HDLEN; rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_RST_STREAM == item->frame.hd.type); CU_ASSERT(0 == nghttp2_session_send(session)); nghttp2_bufs_reset(&bufs); /* header says content-length: 20, but DATA has 21 bytes */ rv = pack_headers(&bufs, &deflater, 5, NGHTTP2_FLAG_END_HEADERS, cl_reqnv, ARRLEN(cl_reqnv), mem); CU_ASSERT(0 == rv); nghttp2_frame_hd_init(&hd, 21, NGHTTP2_DATA, NGHTTP2_FLAG_END_STREAM, 5); nghttp2_frame_pack_frame_hd(bufs.head->buf.last, &hd); bufs.head->buf.last += NGHTTP2_FRAME_HDLEN + 21; rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_RST_STREAM == item->frame.hd.type); CU_ASSERT(0 == nghttp2_session_send(session)); nghttp2_bufs_reset(&bufs); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); nghttp2_bufs_free(&bufs); } void test_nghttp2_http_non_final_response(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_hd_deflater deflater; nghttp2_mem *mem; nghttp2_bufs bufs; ssize_t rv; const nghttp2_nv nonfinal_resnv[] = { MAKE_NV(":status", "100"), }; nghttp2_outbound_item *item; nghttp2_frame_hd hd; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_client_new(&session, &callbacks, NULL); nghttp2_hd_deflate_init(&deflater, mem); /* non-final HEADERS with END_STREAM is illegal */ nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); rv = pack_headers(&bufs, &deflater, 1, NGHTTP2_FLAG_END_HEADERS | NGHTTP2_FLAG_END_STREAM, nonfinal_resnv, ARRLEN(nonfinal_resnv), mem); CU_ASSERT(0 == rv); rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_RST_STREAM == item->frame.hd.type); CU_ASSERT(0 == nghttp2_session_send(session)); nghttp2_bufs_reset(&bufs); /* non-final HEADERS followed by non-empty DATA is illegal */ nghttp2_session_open_stream(session, 3, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); rv = pack_headers(&bufs, &deflater, 3, NGHTTP2_FLAG_END_HEADERS, nonfinal_resnv, ARRLEN(nonfinal_resnv), mem); CU_ASSERT(0 == rv); nghttp2_frame_hd_init(&hd, 10, NGHTTP2_DATA, NGHTTP2_FLAG_END_STREAM, 3); nghttp2_frame_pack_frame_hd(bufs.head->buf.last, &hd); bufs.head->buf.last += NGHTTP2_FRAME_HDLEN + 10; rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_RST_STREAM == item->frame.hd.type); CU_ASSERT(0 == nghttp2_session_send(session)); nghttp2_bufs_reset(&bufs); /* non-final HEADERS followed by empty DATA (without END_STREAM) is ok */ nghttp2_session_open_stream(session, 5, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); rv = pack_headers(&bufs, &deflater, 5, NGHTTP2_FLAG_END_HEADERS, nonfinal_resnv, ARRLEN(nonfinal_resnv), mem); CU_ASSERT(0 == rv); nghttp2_frame_hd_init(&hd, 0, NGHTTP2_DATA, NGHTTP2_FLAG_NONE, 5); nghttp2_frame_pack_frame_hd(bufs.head->buf.last, &hd); bufs.head->buf.last += NGHTTP2_FRAME_HDLEN; rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); CU_ASSERT(NULL == nghttp2_session_get_next_ob_item(session)); nghttp2_bufs_reset(&bufs); /* non-final HEADERS followed by empty DATA (with END_STREAM) is illegal */ nghttp2_session_open_stream(session, 7, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); rv = pack_headers(&bufs, &deflater, 7, NGHTTP2_FLAG_END_HEADERS, nonfinal_resnv, ARRLEN(nonfinal_resnv), mem); CU_ASSERT(0 == rv); nghttp2_frame_hd_init(&hd, 0, NGHTTP2_DATA, NGHTTP2_FLAG_END_STREAM, 7); nghttp2_frame_pack_frame_hd(bufs.head->buf.last, &hd); bufs.head->buf.last += NGHTTP2_FRAME_HDLEN; rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_RST_STREAM == item->frame.hd.type); CU_ASSERT(0 == nghttp2_session_send(session)); nghttp2_bufs_reset(&bufs); /* non-final HEADERS followed by final HEADERS is OK */ nghttp2_session_open_stream(session, 9, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); rv = pack_headers(&bufs, &deflater, 9, NGHTTP2_FLAG_END_HEADERS, nonfinal_resnv, ARRLEN(nonfinal_resnv), mem); CU_ASSERT(0 == rv); rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); nghttp2_bufs_reset(&bufs); rv = pack_headers(&bufs, &deflater, 9, NGHTTP2_FLAG_END_HEADERS, resnv, ARRLEN(resnv), mem); CU_ASSERT(0 == rv); rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); CU_ASSERT(NULL == nghttp2_session_get_next_ob_item(session)); nghttp2_bufs_reset(&bufs); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); nghttp2_bufs_free(&bufs); } void test_nghttp2_http_trailer_headers(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_hd_deflater deflater; nghttp2_mem *mem; nghttp2_bufs bufs; ssize_t rv; const nghttp2_nv trailer_reqnv[] = { MAKE_NV("foo", "bar"), }; nghttp2_outbound_item *item; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_server_new(&session, &callbacks, NULL); nghttp2_hd_deflate_init(&deflater, mem); /* good trailer header */ rv = pack_headers(&bufs, &deflater, 1, NGHTTP2_FLAG_END_HEADERS, reqnv, ARRLEN(reqnv), mem); CU_ASSERT(0 == rv); rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); nghttp2_bufs_reset(&bufs); rv = pack_headers(&bufs, &deflater, 1, NGHTTP2_FLAG_END_HEADERS | NGHTTP2_FLAG_END_STREAM, trailer_reqnv, ARRLEN(trailer_reqnv), mem); CU_ASSERT(0 == rv); rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); CU_ASSERT(NULL == nghttp2_session_get_next_ob_item(session)); nghttp2_bufs_reset(&bufs); /* trailer header without END_STREAM is illegal */ rv = pack_headers(&bufs, &deflater, 3, NGHTTP2_FLAG_END_HEADERS, reqnv, ARRLEN(reqnv), mem); CU_ASSERT(0 == rv); rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); nghttp2_bufs_reset(&bufs); rv = pack_headers(&bufs, &deflater, 3, NGHTTP2_FLAG_END_HEADERS, trailer_reqnv, ARRLEN(trailer_reqnv), mem); CU_ASSERT(0 == rv); rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_RST_STREAM == item->frame.hd.type); CU_ASSERT(0 == nghttp2_session_send(session)); nghttp2_bufs_reset(&bufs); /* trailer header including pseudo header field is illegal */ rv = pack_headers(&bufs, &deflater, 5, NGHTTP2_FLAG_END_HEADERS, reqnv, ARRLEN(reqnv), mem); CU_ASSERT(0 == rv); rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); nghttp2_bufs_reset(&bufs); rv = pack_headers(&bufs, &deflater, 5, NGHTTP2_FLAG_END_HEADERS, reqnv, ARRLEN(reqnv), mem); CU_ASSERT(0 == rv); rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_RST_STREAM == item->frame.hd.type); CU_ASSERT(0 == nghttp2_session_send(session)); nghttp2_bufs_reset(&bufs); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); nghttp2_bufs_free(&bufs); } void test_nghttp2_http_ignore_regular_header(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_hd_deflater deflater; nghttp2_mem *mem; nghttp2_bufs bufs; ssize_t rv; my_user_data ud; const nghttp2_nv bad_reqnv[] = { MAKE_NV(":authority", "localhost"), MAKE_NV(":scheme", "https"), MAKE_NV(":path", "/"), MAKE_NV(":method", "GET"), MAKE_NV("foo", "\x0zzz"), MAKE_NV("bar", "buzz"), }; const nghttp2_nv bad_ansnv[] = { MAKE_NV(":authority", "localhost"), MAKE_NV(":scheme", "https"), MAKE_NV(":path", "/"), MAKE_NV(":method", "GET"), MAKE_NV("bar", "buzz")}; size_t proclen; size_t i; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; callbacks.on_header_callback = pause_on_header_callback; nghttp2_session_server_new(&session, &callbacks, &ud); nghttp2_hd_deflate_init(&deflater, mem); rv = pack_headers(&bufs, &deflater, 1, NGHTTP2_FLAG_END_HEADERS | NGHTTP2_FLAG_END_STREAM, bad_reqnv, ARRLEN(bad_reqnv), mem); CU_ASSERT_FATAL(0 == rv); proclen = 0; for (i = 0; i < 4; ++i) { rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos + proclen, nghttp2_buf_len(&bufs.head->buf) - proclen); CU_ASSERT_FATAL(rv > 0); proclen += (size_t)rv; CU_ASSERT(nghttp2_nv_equal(&bad_ansnv[i], &ud.nv)); } rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos + proclen, nghttp2_buf_len(&bufs.head->buf) - proclen); CU_ASSERT_FATAL(rv > 0); /* header field "foo" must be ignored because it has illegal value. So we have "bar" header field for 5th header. */ CU_ASSERT(nghttp2_nv_equal(&bad_ansnv[4], &ud.nv)); proclen += (size_t)rv; CU_ASSERT(nghttp2_buf_len(&bufs.head->buf) == proclen); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); nghttp2_bufs_free(&bufs); } void test_nghttp2_http_ignore_content_length(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_hd_deflater deflater; nghttp2_mem *mem; nghttp2_bufs bufs; ssize_t rv; const nghttp2_nv cl_resnv[] = {MAKE_NV(":status", "304"), MAKE_NV("content-length", "20")}; const nghttp2_nv conn_reqnv[] = {MAKE_NV(":authority", "localhost"), MAKE_NV(":method", "CONNECT"), MAKE_NV("content-length", "999999")}; nghttp2_stream *stream; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_client_new(&session, &callbacks, NULL); nghttp2_hd_deflate_init(&deflater, mem); /* If status 304, content-length must be ignored */ nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); rv = pack_headers(&bufs, &deflater, 1, NGHTTP2_FLAG_END_HEADERS | NGHTTP2_FLAG_END_STREAM, cl_resnv, ARRLEN(cl_resnv), mem); CU_ASSERT(0 == rv); rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); CU_ASSERT(NULL == nghttp2_session_get_next_ob_item(session)); nghttp2_bufs_reset(&bufs); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); /* If request method is CONNECT, content-length must be ignored */ nghttp2_session_server_new(&session, &callbacks, NULL); nghttp2_hd_deflate_init(&deflater, mem); rv = pack_headers(&bufs, &deflater, 1, NGHTTP2_FLAG_END_HEADERS, conn_reqnv, ARRLEN(conn_reqnv), mem); CU_ASSERT(0 == rv); rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); CU_ASSERT(NULL == nghttp2_session_get_next_ob_item(session)); stream = nghttp2_session_get_stream(session, 1); CU_ASSERT(-1 == stream->content_length); CU_ASSERT((stream->http_flags & NGHTTP2_HTTP_FLAG_METH_CONNECT) > 0); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); nghttp2_bufs_free(&bufs); } void test_nghttp2_http_record_request_method(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; const nghttp2_nv conn_reqnv[] = {MAKE_NV(":method", "CONNECT"), MAKE_NV(":authority", "localhost")}; const nghttp2_nv conn_resnv[] = {MAKE_NV(":status", "200"), MAKE_NV("content-length", "9999")}; nghttp2_stream *stream; ssize_t rv; nghttp2_bufs bufs; nghttp2_hd_deflater deflater; nghttp2_mem *mem; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_client_new(&session, &callbacks, NULL); nghttp2_hd_deflate_init(&deflater, mem); CU_ASSERT(1 == nghttp2_submit_request(session, NULL, conn_reqnv, ARRLEN(conn_reqnv), NULL, NULL)); CU_ASSERT(0 == nghttp2_session_send(session)); stream = nghttp2_session_get_stream(session, 1); CU_ASSERT(NGHTTP2_HTTP_FLAG_METH_CONNECT == stream->http_flags); rv = pack_headers(&bufs, &deflater, 1, NGHTTP2_FLAG_END_HEADERS, conn_resnv, ARRLEN(conn_resnv), mem); CU_ASSERT(0 == rv); rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); CU_ASSERT((NGHTTP2_HTTP_FLAG_METH_CONNECT & stream->http_flags) > 0); CU_ASSERT(-1 == stream->content_length); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); nghttp2_bufs_free(&bufs); } void test_nghttp2_http_push_promise(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; nghttp2_hd_deflater deflater; nghttp2_mem *mem; nghttp2_bufs bufs; ssize_t rv; nghttp2_stream *stream; const nghttp2_nv bad_reqnv[] = {MAKE_NV(":method", "GET")}; nghttp2_outbound_item *item; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; /* good PUSH_PROMISE case */ nghttp2_session_client_new(&session, &callbacks, NULL); nghttp2_hd_deflate_init(&deflater, mem); nghttp2_session_open_stream(session, 1, NGHTTP2_STREAM_FLAG_NONE, &pri_spec_default, NGHTTP2_STREAM_OPENING, NULL); rv = pack_push_promise(&bufs, &deflater, 1, NGHTTP2_FLAG_END_HEADERS, 2, reqnv, ARRLEN(reqnv), mem); CU_ASSERT(0 == rv); rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); CU_ASSERT(NULL == nghttp2_session_get_next_ob_item(session)); stream = nghttp2_session_get_stream(session, 2); CU_ASSERT(NULL != stream); nghttp2_bufs_reset(&bufs); rv = pack_headers(&bufs, &deflater, 2, NGHTTP2_FLAG_END_HEADERS, resnv, ARRLEN(resnv), mem); CU_ASSERT(0 == rv); rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); CU_ASSERT(NULL == nghttp2_session_get_next_ob_item(session)); CU_ASSERT(200 == stream->status_code); nghttp2_bufs_reset(&bufs); /* PUSH_PROMISE lacks mandatory header */ rv = pack_push_promise(&bufs, &deflater, 1, NGHTTP2_FLAG_END_HEADERS, 4, bad_reqnv, ARRLEN(bad_reqnv), mem); CU_ASSERT(0 == rv); rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); item = nghttp2_session_get_next_ob_item(session); CU_ASSERT(NGHTTP2_RST_STREAM == item->frame.hd.type); CU_ASSERT(4 == item->frame.hd.stream_id); nghttp2_bufs_reset(&bufs); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); nghttp2_bufs_free(&bufs); } void test_nghttp2_http_head_method_upgrade_workaround(void) { nghttp2_session *session; nghttp2_session_callbacks callbacks; const nghttp2_nv cl_resnv[] = {MAKE_NV(":status", "200"), MAKE_NV("content-length", "1000000007")}; nghttp2_bufs bufs; nghttp2_hd_deflater deflater; nghttp2_mem *mem; ssize_t rv; nghttp2_stream *stream; mem = nghttp2_mem_default(); frame_pack_bufs_init(&bufs); memset(&callbacks, 0, sizeof(nghttp2_session_callbacks)); callbacks.send_callback = null_send_callback; nghttp2_session_client_new(&session, &callbacks, NULL); nghttp2_hd_deflate_init(&deflater, mem); nghttp2_session_upgrade(session, NULL, 0, NULL); rv = pack_headers(&bufs, &deflater, 1, NGHTTP2_FLAG_END_HEADERS, cl_resnv, ARRLEN(cl_resnv), mem); CU_ASSERT(0 == rv); rv = nghttp2_session_mem_recv(session, bufs.head->buf.pos, nghttp2_buf_len(&bufs.head->buf)); CU_ASSERT((ssize_t)nghttp2_buf_len(&bufs.head->buf) == rv); stream = nghttp2_session_get_stream(session, 1); CU_ASSERT(-1 == stream->content_length); nghttp2_hd_deflate_free(&deflater); nghttp2_session_del(session); nghttp2_bufs_free(&bufs); }