nghttpx: Set do_signal_write_ when TLS handshake was completed

bryonglodencissp · Aug 23, 2016 · 8f47b68 · 8f47b68
1 parent d9139fc
commit 8f47b68
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/shrpx_http_downstream_connection.cc b/src/shrpx_http_downstream_connection.cc
@@ -994,6 +994,8 @@ int HttpDownstreamConnection::tls_handshake() {
 
   auto &connect_blocker = addr_->connect_blocker;
 
+  do_signal_write_ = &HttpDownstreamConnection::actual_signal_write;
+
   connect_blocker->on_success();
 
   ev_set_cb(&conn_.rt, timeoutcb);
@@ -1160,15 +1162,15 @@ int HttpDownstreamConnection::connected() {
 
   ev_set_cb(&conn_.wev, writecb);
 
-  do_signal_write_ = &HttpDownstreamConnection::actual_signal_write;
-
   if (conn_.tls.ssl) {
     do_read_ = &HttpDownstreamConnection::tls_handshake;
     do_write_ = &HttpDownstreamConnection::tls_handshake;
 
     return 0;
   }
 
+  do_signal_write_ = &HttpDownstreamConnection::actual_signal_write;
+
   connect_blocker->on_success();
 
   ev_set_cb(&conn_.rt, timeoutcb);