shrpx: Check request_connection_close_ when deciding closing connection

When deciding whether to close the client connection, check
request_connection_close_ of Downstream in addition of
response_connection_close_. Also we only add "Connection: Keep-Alive"
header to the HTTP/1.0 or HTTP/0.9 clients.
This commit is contained in:
Tatsuhiro Tsujikawa 2012-09-13 21:33:35 +09:00
parent 22516aea63
commit 436b201d6f
1 changed files with 10 additions and 6 deletions

View File

@ -579,14 +579,17 @@ int HttpsUpstream::on_downstream_header_complete(Downstream *downstream)
}
}
if(downstream->get_response_version() < 101) {
if(!downstream->get_response_connection_close()) {
// We check downstream->get_response_connection_close() in case when
// the Content-Length is not available.
if(!downstream->get_request_connection_close() &&
!downstream->get_response_connection_close()) {
if(downstream->get_request_major() <= 0 ||
downstream->get_request_minor() <= 0) {
// We add this header for HTTP/1.0 or HTTP/0.9 clients
hdrs += "Connection: Keep-Alive\r\n";
}
} else {
if(downstream->get_response_connection_close()) {
hdrs += "Connection: close\r\n";
}
hdrs += "Connection: close\r\n";
}
hdrs += "Via: ";
@ -642,7 +645,8 @@ int HttpsUpstream::on_downstream_body_complete(Downstream *downstream)
if(ENABLE_LOG) {
LOG(INFO) << "Downstream on_downstream_body_complete";
}
if(downstream->get_response_connection_close()) {
if(downstream->get_request_connection_close() ||
downstream->get_response_connection_close()) {
ClientHandler *handler = get_client_handler();
handler->set_should_close_after_write(true);
}