Evaluate nghttp2_session_want_read and nghttp2_session_want_write when determining session is about to close

This commit is contained in:
Tatsuhiro Tsujikawa 2016-02-03 00:25:11 +09:00
parent 35c3b36549
commit 4c05558273
1 changed files with 17 additions and 7 deletions

View File

@ -1291,7 +1291,9 @@ static int session_allow_incoming_new_stream(nghttp2_session *session) {
* This function returns nonzero if session is closing. * This function returns nonzero if session is closing.
*/ */
static int session_is_closing(nghttp2_session *session) { static int session_is_closing(nghttp2_session *session) {
return (session->goaway_flags & NGHTTP2_GOAWAY_TERM_ON_SEND) != 0; return (session->goaway_flags & NGHTTP2_GOAWAY_TERM_ON_SEND) != 0 ||
(nghttp2_session_want_read(session) == 0 &&
nghttp2_session_want_write(session) == 0);
} }
/* /*
@ -1322,8 +1324,8 @@ static int session_predicate_for_stream_send(nghttp2_session *session,
int nghttp2_session_check_request_allowed(nghttp2_session *session) { int nghttp2_session_check_request_allowed(nghttp2_session *session) {
return !session->server && session->next_stream_id <= INT32_MAX && return !session->server && session->next_stream_id <= INT32_MAX &&
(session->goaway_flags & (session->goaway_flags & NGHTTP2_GOAWAY_RECV) == 0 &&
(NGHTTP2_GOAWAY_TERM_ON_SEND | NGHTTP2_GOAWAY_RECV)) == 0; !session_is_closing(session);
} }
/* /*
@ -1345,10 +1347,11 @@ static int session_predicate_request_headers_send(nghttp2_session *session,
if (item->aux_data.headers.canceled) { if (item->aux_data.headers.canceled) {
return NGHTTP2_ERR_STREAM_CLOSING; return NGHTTP2_ERR_STREAM_CLOSING;
} }
/* If we are terminating session (NGHTTP2_GOAWAY_TERM_ON_SEND) or /* If we are terminating session (NGHTTP2_GOAWAY_TERM_ON_SEND),
GOAWAY was received from peer, new request is not allowed. */ GOAWAY was received from peer, or session is about to close, new
if (session->goaway_flags & request is not allowed. */
(NGHTTP2_GOAWAY_TERM_ON_SEND | NGHTTP2_GOAWAY_RECV)) { if ((session->goaway_flags & NGHTTP2_GOAWAY_RECV) ||
session_is_closing(session)) {
return NGHTTP2_ERR_START_STREAM_NOT_ALLOWED; return NGHTTP2_ERR_START_STREAM_NOT_ALLOWED;
} }
return 0; return 0;
@ -1903,6 +1906,13 @@ static int session_prep_frame(nghttp2_session *session,
if (frame->hd.flags & NGHTTP2_FLAG_ACK) { if (frame->hd.flags & NGHTTP2_FLAG_ACK) {
assert(session->obq_flood_counter_ > 0); assert(session->obq_flood_counter_ > 0);
--session->obq_flood_counter_; --session->obq_flood_counter_;
/* When session is about to close, don't send SETTINGS ACK.
We are required to send SETTINGS without ACK though; for
example, we have to send SETTINGS as a part of connection
preface. */
if (session_is_closing(session)) {
return NGHTTP2_ERR_SESSION_CLOSING;
}
} }
rv = nghttp2_frame_pack_settings(&session->aob.framebufs, rv = nghttp2_frame_pack_settings(&session->aob.framebufs,