r46826 - abortConnection if we can't loseConnection

Author: mithrandi
Date: Thu Feb 18 20:18:30 2016
New Revision: 46826

Modified:
   branches/openssl-f-8189/twisted/protocols/loopback.py
   branches/openssl-f-8189/twisted/protocols/test/test_tls.py
   branches/openssl-f-8189/twisted/protocols/tls.py
   branches/openssl-f-8189/twisted/test/proto_helpers.py
   branches/openssl-f-8189/twisted/web/test/test_agent.py

Log:
abortConnection if we can't loseConnection

Modified: branches/openssl-f-8189/twisted/protocols/loopback.py
==============================================================================
--- branches/openssl-f-8189/twisted/protocols/loopback.py	(original)
+++ branches/openssl-f-8189/twisted/protocols/loopback.py	Thu Feb 18 20:18:30 2016
@@ -79,6 +79,14 @@
         self.q.disconnect = True
         self.q.put(None)
 
+
+    def abortConnection(self):
+        """
+        Abort the connection. Same as L{loseConnection}.
+        """
+        self.loseConnection()
+
+
     def getPeer(self):
         return _LoopbackAddress()
 

Modified: branches/openssl-f-8189/twisted/protocols/test/test_tls.py
==============================================================================
--- branches/openssl-f-8189/twisted/protocols/test/test_tls.py	(original)
+++ branches/openssl-f-8189/twisted/protocols/test/test_tls.py	Thu Feb 18 20:18:30 2016
@@ -708,6 +708,7 @@
             # will be written out before the connection is closed, rather than
             # just small amounts that can be returned in a single bio_read:
             clientProtocol.transport.write(chunkOfBytes)
+            serverProtocol.transport.write(b'x')
             serverProtocol.transport.loseConnection()
 
             # Now wait for the client and server to notice.
@@ -772,28 +773,28 @@
         If TLSMemoryBIOProtocol.loseConnection is called multiple times, all
         but the first call have no effect.
         """
-        wrapperFactory = TLSMemoryBIOFactory(ClientTLSContext(),
-                                             True, ClientFactory())
-        tlsProtocol = TLSMemoryBIOProtocol(wrapperFactory, Protocol())
-        transport = StringTransport()
-        tlsProtocol.makeConnection(transport)
-        self.assertEqual(tlsProtocol.disconnecting, False)
-
+        tlsClient, tlsServer, handshakeDeferred, disconnectDeferred = (
+            self.handshakeProtocols())
+        self.successResultOf(handshakeDeferred)
         # Make sure loseConnection calls _shutdownTLS the first time (mostly
         # to make sure we've overriding it correctly):
         calls = []
-        def _shutdownTLS(shutdown=tlsProtocol._shutdownTLS):
+        def _shutdownTLS(shutdown=tlsClient._shutdownTLS):
             calls.append(1)
             return shutdown()
-        tlsProtocol._shutdownTLS = _shutdownTLS
-        tlsProtocol.loseConnection()
-        self.assertEqual(tlsProtocol.disconnecting, True)
+        tlsClient._shutdownTLS = _shutdownTLS
+        tlsClient.write(b'x')
+        tlsClient.loseConnection()
+        self.assertEqual(tlsClient.disconnecting, True)
         self.assertEqual(calls, [1])
 
         # Make sure _shutdownTLS isn't called a second time:
-        tlsProtocol.loseConnection()
+        tlsClient.loseConnection()
         self.assertEqual(calls, [1])
 
+        # We do successfully disconnect at some point:
+        return disconnectDeferred
+
 
     def test_unexpectedEOF(self):
         """

Modified: branches/openssl-f-8189/twisted/protocols/tls.py
==============================================================================
--- branches/openssl-f-8189/twisted/protocols/tls.py	(original)
+++ branches/openssl-f-8189/twisted/protocols/tls.py	Thu Feb 18 20:18:30 2016
@@ -275,6 +275,7 @@
     _writeBlockedOnRead = False
     _producer = None
     _aborted = False
+    _shuttingDown = False
 
     def __init__(self, factory, wrappedProtocol, _connectWrapped=True):
         ProtocolWrapper.__init__(self, factory, wrappedProtocol)
@@ -431,6 +432,7 @@
         """
         Initiate, or reply to, the shutdown handshake of the TLS layer.
         """
+        self._shuttingDown = True
         try:
             shutdownSuccess = self._tlsConnection.shutdown()
         except Error:
@@ -488,6 +490,14 @@
         """
         if self.disconnecting:
             return
+        # If connection setup has not finished, OpenSSL 1.0.2f+ will not shut
+        # down the connection until we write some data to the connection which
+        # allows the handshake to complete. However, since no data should be
+        # written after loseConnection, this means we'll be stuck forever
+        # waiting for shutdown to complete. Instead, we simply abort the
+        # connection without trying to shut down cleanly:
+        if not self._handshakeDone and not self._writeBlockedOnRead:
+            self.abortConnection()
         self.disconnecting = True
         if not self._writeBlockedOnRead and self._producer is None:
             self._shutdownTLS()

Modified: branches/openssl-f-8189/twisted/test/proto_helpers.py
==============================================================================
--- branches/openssl-f-8189/twisted/test/proto_helpers.py	(original)
+++ branches/openssl-f-8189/twisted/test/proto_helpers.py	Thu Feb 18 20:18:30 2016
@@ -204,6 +204,13 @@
         self.disconnecting = True
 
 
+    def abortConnection(self):
+        """
+        Abort the connection. Same as C{loseConnection}.
+        """
+        self.loseConnection()
+
+
     def getPeer(self):
         if self.peerAddr is None:
             return address.IPv4Address('TCP', '192.168.1.1', 54321)

Modified: branches/openssl-f-8189/twisted/web/test/test_agent.py
==============================================================================
--- branches/openssl-f-8189/twisted/web/test/test_agent.py	(original)
+++ branches/openssl-f-8189/twisted/web/test/test_agent.py	Thu Feb 18 20:18:30 2016
@@ -2888,6 +2888,7 @@
         warning, but no exception when cancelling.
         """
         response = DummyResponse(transportFactory=StringTransport)
+        response.transport.abortConnection = None
         d = self.assertWarns(
             DeprecationWarning,
             'Using readBody with a transport that does not have an '