Moved want_write to trigger the correct io loop state

diff --git a/tornado/iostream.py b/tornado/iostream.py
index ef573b4de2cca667e983bcc6678769f93439da3c..3d9bad02f9288c97bcf1f04135082de9de8a90f4 100644 (file)
--- a/tornado/iostream.py
+++ b/tornado/iostream.py
@@ -260,10 +260,12 @@ class SSLIOStream(IOStream):
         try:
             self.socket.do_handshake()
         except ssl.SSLError, err:
-            if err.args[0] in (ssl.SSL_ERROR_WANT_READ,
-                               ssl.SSL_ERROR_WANT_WRITE):
+            if err.args[0] == ssl.SSL_ERROR_WANT_READ:
                 self._add_io_state(self.io_loop.READ)
                 return
+            elif err.args[0] == ssl.SSL_ERROR_WANT_WRITE:
+                self._add_io_state(self.io_loop.WRITE)
+                return
             elif err.args[0] in (ssl.SSL_ERROR_EOF,
                                  ssl.SSL_ERROR_ZERO_RETURN):
                 return self.close()