Close socket if connection already timed out.

Azure · Sep 18, 2020 · ebb75bc · ebb75bc
1 parent 7d3177c
commit ebb75bc
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/Microsoft.Azure.Amqp/Amqp/Transport/TcpTransportInitiator.cs b/Microsoft.Azure.Amqp/Amqp/Transport/TcpTransportInitiator.cs
@@ -64,6 +64,10 @@ static void OnConnectComplete(object sender, SocketAsyncEventArgs e)
             {
                 thisPtr.Complete(e, false);
             }
+            else
+            {
+                e.ConnectSocket?.Dispose();
+            }
         }
 
         static void OnTimer(object obj)
@@ -83,9 +87,9 @@ void Complete(SocketAsyncEventArgs e, bool completeSynchronously)
             if (e.SocketError != SocketError.Success)
             {
                 exception = new SocketException((int)e.SocketError);
-                if (e.AcceptSocket != null)
+                if (e.ConnectSocket != null)
                 {
-                    e.AcceptSocket.Dispose();
+                    e.ConnectSocket.Dispose();
                 }
             }
             else