prevent orphaned tcp socket

Author: mueslo

httpcore/_async/connection.py

@@ -143,7 +143,11 @@ async def _connect(self, request: Request) -> AsyncNetworkStream:
                 "timeout": timeout,
             }
             async with Trace("connection.start_tls", request, kwargs) as trace:
-                stream = await stream.start_tls(**kwargs)
+                try:
+                    stream = await stream.start_tls(**kwargs)
+                except Exception:
+                    await stream.aclose()
+                    raise
                 trace.return_value = stream
         return stream
 

httpcore/_sync/connection.py

@@ -143,7 +143,11 @@ def _connect(self, request: Request) -> NetworkStream:
                 "timeout": timeout,
             }
             with Trace("connection.start_tls", request, kwargs) as trace:
-                stream = stream.start_tls(**kwargs)
+                try:
+                    stream = stream.start_tls(**kwargs)
+                except Exception:
+                    stream.close()
+                    raise
                 trace.return_value = stream
         return stream