1
0
mirror of https://github.com/moparisthebest/spdylay synced 2024-12-21 23:28:52 -05:00

Bring back deferred DATA to the outbound queue when SETTINGS with

INITIAL_WINDOW_SIZE is received and the window size becomes positive.
This commit is contained in:
Tatsuhiro Tsujikawa 2012-05-11 22:58:12 +09:00
parent 5789f35f5e
commit 9b619e5dd7
6 changed files with 95 additions and 34 deletions

View File

@ -683,7 +683,8 @@ void spdylay_frame_window_update_free(spdylay_window_update *frame);
/*
* Initializes SETTINGS frame |frame| with given values. |frame| takes
* ownership of |iv|, so caller must not free it.
* ownership of |iv|, so caller must not free it. The |flags| are
* bitwise-OR of one or more of spdylay_settings_flag.
*/
void spdylay_frame_settings_init(spdylay_settings *frame,
uint16_t version, uint8_t flags,

View File

@ -197,20 +197,24 @@ size_t spdylay_map_size(spdylay_map *map)
return map->size;
}
static void for_each(spdylay_map_entry *entry,
void (*func)(key_type key, void *val, void *ptr),
void *ptr)
static int for_each(spdylay_map_entry *entry,
int (*func)(key_type key, void *val, void *ptr),
void *ptr)
{
if(entry != NULL) {
for_each(entry->left, func, ptr);
func(entry->key, entry->val, ptr);
for_each(entry->right, func, ptr);
if(entry) {
int rv;
if((rv = for_each(entry->left, func, ptr)) != 0 ||
(rv = func(entry->key, entry->val, ptr)) != 0 ||
(rv = for_each(entry->right, func, ptr)) != 0) {
return rv;
}
}
return 0;
}
void spdylay_map_each(spdylay_map *map,
void (*func)(key_type key, void *val, void *ptr),
void *ptr)
int spdylay_map_each(spdylay_map *map,
int (*func)(key_type key, void *val, void *ptr),
void *ptr)
{
for_each(map->root, func, ptr);
return for_each(map->root, func, ptr);
}

View File

@ -102,9 +102,16 @@ size_t spdylay_map_size(spdylay_map *map);
* Applies the function |func| to each key/item pair in the map |map|
* with the optional user supplied pointer |ptr|. This function is
* useful to free item in the map.
*
* If the |func| returns 0, this function calls the |func| with the
* next key and value pair. If the |func| returns nonzero, it will not
* call the |func| for further key and value pair and return the
* return value of the |func| immediately. Thus, this function
* returns 0 if all the invocations of the |func| return 0, or nonzero
* value which the last invocation of |func| returns.
*/
void spdylay_map_each(spdylay_map *map,
void (*func)(key_type key, void *val, void *ptr),
void *ptr);
int spdylay_map_each(spdylay_map *map,
int (*func)(key_type key, void *val, void *ptr),
void *ptr);
#endif /* SPDYLAY_MAP_H */

View File

@ -274,10 +274,11 @@ int spdylay_session_server_new(spdylay_session **session_ptr,
return r;
}
static void spdylay_free_streams(key_type key, void *val, void *ptr)
static int spdylay_free_streams(key_type key, void *val, void *ptr)
{
spdylay_stream_free((spdylay_stream*)val);
free(val);
return 0;
}
static void spdylay_session_ob_pq_free(spdylay_pq *pq)
@ -1783,28 +1784,56 @@ int spdylay_session_on_rst_stream_received(spdylay_session *session,
return 0;
}
static void spdylay_update_initial_window_size_func(key_type key, void *value,
void *ptr)
static int spdylay_update_initial_window_size_func(key_type key, void *value,
void *ptr)
{
int32_t *vals;
vals = (int32_t*)ptr;
spdylay_stream_update_initial_window_size((spdylay_stream*)value,
vals[0], vals[1]);
spdylay_update_window_size_arg *arg;
spdylay_stream *stream;
arg = (spdylay_update_window_size_arg*)ptr;
stream = (spdylay_stream*)value;
spdylay_stream_update_initial_window_size(stream,
arg->new_window_size,
arg->old_window_size);
/* If window size gets positive, push deferred DATA frame to
outbound queue. */
if(stream->window_size > 0 &&
stream->deferred_data &&
(stream->deferred_flags & SPDYLAY_DEFERRED_FLOW_CONTROL)) {
int rv;
rv = spdylay_pq_push(&arg->session->ob_pq, stream->deferred_data);
if(rv == 0) {
spdylay_stream_detach_deferred_data(stream);
} else {
/* FATAL */
assert(rv < SPDYLAY_ERR_FATAL);
return rv;
}
}
return 0;
}
/*
* Updates the initial window size of all active streams.
* If error occurs, all streams may not be updated.
*
* This function returns 0 if it succeeds, or one of the following
* negative error codes:
*
* SPDYLAY_ERR_NOMEM
* Out of memory.
*/
static void spdylay_session_update_initial_window_size
static int spdylay_session_update_initial_window_size
(spdylay_session *session,
int32_t new_initial_window_size)
{
int32_t vals[2];
vals[0] = new_initial_window_size;
vals[1] = session->remote_settings[SPDYLAY_SETTINGS_INITIAL_WINDOW_SIZE];
spdylay_map_each(&session->streams,
spdylay_update_initial_window_size_func,
vals);
spdylay_update_window_size_arg arg;
arg.session = session;
arg.new_window_size = new_initial_window_size;
arg.old_window_size =
session->remote_settings[SPDYLAY_SETTINGS_INITIAL_WINDOW_SIZE];
return spdylay_map_each(&session->streams,
spdylay_update_initial_window_size_func,
&arg);
}
void spdylay_session_update_local_settings(spdylay_session *session,
@ -1821,6 +1850,7 @@ void spdylay_session_update_local_settings(spdylay_session *session,
int spdylay_session_on_settings_received(spdylay_session *session,
spdylay_frame *frame)
{
int rv;
size_t i;
int check[SPDYLAY_SETTINGS_MAX+1];
if(!spdylay_session_check_version(session, frame->settings.hd.version)) {
@ -1842,12 +1872,14 @@ int spdylay_session_on_settings_received(spdylay_session *session,
/* Update the initial window size of the all active streams */
/* Check that initial_window_size < (1u << 31) */
if(entry->value < (1u << 31)) {
spdylay_session_update_initial_window_size(session, entry->value);
rv = spdylay_session_update_initial_window_size(session, entry->value);
if(rv != 0) {
return rv;
}
}
} else if(entry->settings_id ==
SPDYLAY_SETTINGS_CLIENT_CERTIFICATE_VECTOR_SIZE) {
if(!session->server) {
int rv;
/* Limit certificate vector length in the reasonable size. */
entry->value = spdylay_min(entry->value,
SPDYLAY_MAX_CLIENT_CERT_VECTOR_LENGTH);

View File

@ -197,6 +197,13 @@ struct spdylay_session {
void *user_data;
};
/* Struct used when updating initial window size of each active
stream. */
typedef struct {
spdylay_session *session;
int32_t new_window_size, old_window_size;
} spdylay_update_window_size_arg;
/* TODO stream timeout etc */
/*

View File

@ -1736,6 +1736,8 @@ void test_spdylay_session_flow_control(void)
spdylay_frame frame;
spdylay_stream *stream;
int32_t new_initial_window_size;
spdylay_settings_entry iv[1];
spdylay_frame settings_frame;
memset(&callbacks, 0, sizeof(spdylay_session_callbacks));
callbacks.send_callback = null_send_callback;
@ -1769,6 +1771,8 @@ void test_spdylay_session_flow_control(void)
stream->window_size = new_initial_window_size-
(session->remote_settings[SPDYLAY_SETTINGS_INITIAL_WINDOW_SIZE]
-stream->window_size);
session->remote_settings[SPDYLAY_SETTINGS_INITIAL_WINDOW_SIZE] =
new_initial_window_size;
CU_ASSERT(-48*1024 == stream->window_size);
/* Back 48KiB */
@ -1787,9 +1791,15 @@ void test_spdylay_session_flow_control(void)
CU_ASSERT(0 == spdylay_session_send(session));
CU_ASSERT(16*1024 == ud.data_source_length);
/* Back 16KiB */
frame.window_update.delta_window_size = 16*1024;
spdylay_session_on_window_update_received(session, &frame);
/* Increase initial window size to 32KiB */
iv[0].settings_id = SPDYLAY_SETTINGS_INITIAL_WINDOW_SIZE;
iv[0].flags = SPDYLAY_ID_FLAG_SETTINGS_NONE;
iv[0].value = 32*1024;
spdylay_frame_settings_init(&settings_frame.settings, SPDYLAY_PROTO_SPDY3,
SPDYLAY_FLAG_SETTINGS_NONE, dup_iv(iv, 1), 1);
spdylay_session_on_settings_received(session, &settings_frame);
spdylay_frame_settings_free(&settings_frame.settings);
/* Sends another 16KiB data */
CU_ASSERT(0 == spdylay_session_send(session));