241 lines
7.8 KiB
C
241 lines
7.8 KiB
C
/*
|
|
* nghttp2 - HTTP/2.0 C Library
|
|
*
|
|
* Copyright (c) 2012 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_submit.h"
|
|
|
|
#include <string.h>
|
|
|
|
#include "nghttp2_session.h"
|
|
#include "nghttp2_frame.h"
|
|
#include "nghttp2_helper.h"
|
|
|
|
static int nghttp2_submit_headers_shared
|
|
(nghttp2_session *session,
|
|
uint8_t flags,
|
|
int32_t stream_id,
|
|
int32_t pri,
|
|
const char **nv,
|
|
const nghttp2_data_provider *data_prd,
|
|
void *stream_user_data)
|
|
{
|
|
int r;
|
|
nghttp2_frame *frame;
|
|
char **nv_copy;
|
|
uint8_t flags_copy;
|
|
nghttp2_data_provider *data_prd_copy = NULL;
|
|
nghttp2_headers_aux_data *aux_data = NULL;
|
|
if(pri < 0) {
|
|
return NGHTTP2_ERR_INVALID_ARGUMENT;
|
|
}
|
|
if(!nghttp2_frame_nv_check_null(nv)) {
|
|
return NGHTTP2_ERR_INVALID_ARGUMENT;
|
|
}
|
|
if(data_prd != NULL && data_prd->read_callback != NULL) {
|
|
data_prd_copy = malloc(sizeof(nghttp2_data_provider));
|
|
if(data_prd_copy == NULL) {
|
|
return NGHTTP2_ERR_NOMEM;
|
|
}
|
|
*data_prd_copy = *data_prd;
|
|
}
|
|
if(data_prd || stream_user_data) {
|
|
aux_data = malloc(sizeof(nghttp2_headers_aux_data));
|
|
if(aux_data == NULL) {
|
|
free(data_prd_copy);
|
|
return NGHTTP2_ERR_NOMEM;
|
|
}
|
|
aux_data->data_prd = data_prd_copy;
|
|
aux_data->stream_user_data = stream_user_data;
|
|
}
|
|
frame = malloc(sizeof(nghttp2_frame));
|
|
if(frame == NULL) {
|
|
free(aux_data);
|
|
free(data_prd_copy);
|
|
return NGHTTP2_ERR_NOMEM;
|
|
}
|
|
nv_copy = nghttp2_frame_nv_norm_copy(nv);
|
|
if(nv_copy == NULL) {
|
|
free(frame);
|
|
free(aux_data);
|
|
free(data_prd_copy);
|
|
return NGHTTP2_ERR_NOMEM;
|
|
}
|
|
/* TODO Implement header continuation */
|
|
flags_copy = (flags & (NGHTTP2_FLAG_END_STREAM | NGHTTP2_FLAG_PRIORITY)) |
|
|
NGHTTP2_FLAG_END_HEADERS;
|
|
|
|
nghttp2_frame_headers_init(&frame->headers, flags_copy, stream_id, pri,
|
|
nv_copy);
|
|
r = nghttp2_session_add_frame(session, NGHTTP2_CAT_CTRL, frame,
|
|
aux_data);
|
|
if(r != 0) {
|
|
nghttp2_frame_headers_free(&frame->headers);
|
|
free(frame);
|
|
free(aux_data);
|
|
free(data_prd_copy);
|
|
}
|
|
return r;
|
|
}
|
|
|
|
int nghttp2_submit_headers(nghttp2_session *session, uint8_t flags,
|
|
int32_t stream_id, int32_t pri,
|
|
const char **nv, void *stream_user_data)
|
|
{
|
|
return nghttp2_submit_headers_shared(session, flags, stream_id,
|
|
pri, nv, NULL, stream_user_data);
|
|
}
|
|
|
|
|
|
int nghttp2_submit_ping(nghttp2_session *session, uint8_t *opaque_data)
|
|
{
|
|
return nghttp2_session_add_ping(session, NGHTTP2_FLAG_NONE, opaque_data);
|
|
}
|
|
|
|
int nghttp2_submit_rst_stream(nghttp2_session *session, int32_t stream_id,
|
|
nghttp2_error_code error_code)
|
|
{
|
|
return nghttp2_session_add_rst_stream(session, stream_id, error_code);
|
|
}
|
|
|
|
int nghttp2_submit_goaway(nghttp2_session *session,
|
|
nghttp2_error_code error_code,
|
|
uint8_t *opaque_data, size_t opaque_data_len)
|
|
{
|
|
return nghttp2_session_add_goaway(session, session->last_recv_stream_id,
|
|
error_code, opaque_data, opaque_data_len);
|
|
}
|
|
|
|
int nghttp2_submit_settings(nghttp2_session *session,
|
|
const nghttp2_settings_entry *iv, size_t niv)
|
|
{
|
|
nghttp2_frame *frame;
|
|
nghttp2_settings_entry *iv_copy;
|
|
int check[NGHTTP2_SETTINGS_MAX+1];
|
|
size_t i;
|
|
int r;
|
|
memset(check, 0, sizeof(check));
|
|
for(i = 0; i < niv; ++i) {
|
|
if(iv[i].settings_id > NGHTTP2_SETTINGS_MAX || iv[i].settings_id == 0 ||
|
|
check[iv[i].settings_id] == 1) {
|
|
return NGHTTP2_ERR_INVALID_ARGUMENT;
|
|
} else {
|
|
check[iv[i].settings_id] = 1;
|
|
}
|
|
}
|
|
frame = malloc(sizeof(nghttp2_frame));
|
|
if(frame == NULL) {
|
|
return NGHTTP2_ERR_NOMEM;
|
|
}
|
|
iv_copy = nghttp2_frame_iv_copy(iv, niv);
|
|
if(iv_copy == NULL) {
|
|
free(frame);
|
|
return NGHTTP2_ERR_NOMEM;
|
|
}
|
|
nghttp2_frame_iv_sort(iv_copy, niv);
|
|
nghttp2_frame_settings_init(&frame->settings, iv_copy, niv);
|
|
r = nghttp2_session_add_frame(session, NGHTTP2_CAT_CTRL, frame, NULL);
|
|
if(r == 0) {
|
|
nghttp2_session_update_local_settings(session, iv_copy, niv);
|
|
} else {
|
|
nghttp2_frame_settings_free(&frame->settings);
|
|
free(frame);
|
|
}
|
|
return r;
|
|
}
|
|
|
|
int nghttp2_submit_window_update(nghttp2_session *session, uint8_t flags,
|
|
int32_t stream_id,
|
|
int32_t window_size_increment)
|
|
{
|
|
nghttp2_stream *stream;
|
|
if(window_size_increment <= 0) {
|
|
return NGHTTP2_ERR_INVALID_ARGUMENT;
|
|
}
|
|
if(stream_id == 0) {
|
|
return nghttp2_session_add_window_update(session, flags, stream_id,
|
|
window_size_increment);
|
|
} else {
|
|
stream = nghttp2_session_get_stream(session, stream_id);
|
|
if(stream) {
|
|
stream->recv_window_size -= nghttp2_min(window_size_increment,
|
|
stream->recv_window_size);
|
|
return nghttp2_session_add_window_update(session, flags, stream_id,
|
|
window_size_increment);
|
|
} else {
|
|
return NGHTTP2_ERR_STREAM_CLOSED;
|
|
}
|
|
}
|
|
}
|
|
|
|
int nghttp2_submit_request(nghttp2_session *session, int32_t pri,
|
|
const char **nv,
|
|
const nghttp2_data_provider *data_prd,
|
|
void *stream_user_data)
|
|
{
|
|
uint8_t flags = NGHTTP2_FLAG_NONE;
|
|
if(data_prd == NULL || data_prd->read_callback == NULL) {
|
|
flags |= NGHTTP2_FLAG_END_STREAM;
|
|
}
|
|
if(pri != NGHTTP2_PRI_DEFAULT) {
|
|
flags |= NGHTTP2_FLAG_PRIORITY;
|
|
}
|
|
return nghttp2_submit_headers_shared(session, flags, -1, pri, nv,
|
|
data_prd, stream_user_data);
|
|
}
|
|
|
|
int nghttp2_submit_response(nghttp2_session *session,
|
|
int32_t stream_id, const char **nv,
|
|
const nghttp2_data_provider *data_prd)
|
|
{
|
|
uint8_t flags = NGHTTP2_FLAG_NONE;
|
|
if(data_prd == NULL || data_prd->read_callback == NULL) {
|
|
flags |= NGHTTP2_FLAG_END_STREAM;
|
|
}
|
|
return nghttp2_submit_headers_shared(session, flags, stream_id,
|
|
NGHTTP2_PRI_DEFAULT, nv, data_prd,
|
|
NULL);
|
|
}
|
|
|
|
int nghttp2_submit_data(nghttp2_session *session, uint8_t flags,
|
|
int32_t stream_id,
|
|
const nghttp2_data_provider *data_prd)
|
|
{
|
|
int r;
|
|
nghttp2_data *data_frame;
|
|
uint8_t nflags = 0;
|
|
data_frame = malloc(sizeof(nghttp2_frame));
|
|
if(data_frame == NULL) {
|
|
return NGHTTP2_ERR_NOMEM;
|
|
}
|
|
if(flags & NGHTTP2_FLAG_END_STREAM) {
|
|
nflags |= NGHTTP2_FLAG_END_STREAM;
|
|
}
|
|
nghttp2_frame_data_init(data_frame, nflags, stream_id, data_prd);
|
|
r = nghttp2_session_add_frame(session, NGHTTP2_CAT_DATA, data_frame, NULL);
|
|
if(r != 0) {
|
|
nghttp2_frame_data_free(data_frame);
|
|
free(data_frame);
|
|
}
|
|
return r;
|
|
}
|