Dispose socket if writer is completed with any error.

aspnet · Jul 7, 2017 · 093a6fe · 093a6fe
1 parent 5023f94
commit 093a6fe
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/src/Microsoft.AspNetCore.Server.Kestrel.Transport.Libuv/Internal/LibuvOutputConsumer.cs b/src/Microsoft.AspNetCore.Server.Kestrel.Transport.Libuv/Internal/LibuvOutputConsumer.cs
@@ -29,7 +29,7 @@ public LibuvOutputConsumer(
             _connectionId = connectionId;
             _log = log;
 
-            _pipe.OnWriterCompleted(DisposeSocket, this);
+            _pipe.OnWriterCompleted(OnWriterCompleted, this);
         }
 
         public async Task WriteOutputAsync()
@@ -88,11 +88,10 @@ public async Task WriteOutputAsync()
             }
         }
 
-        private static void DisposeSocket(Exception ex, object state)
+        private static void OnWriterCompleted(Exception ex, object state)
         {
-            // If the pipe writer was closed with a TimeoutException, a write timed out.
-            // Disposing the socket will cancel the pending UvWriteReq.
-            if (ex is TimeoutException)
+            // Cut off writes if the writer is completed with an error.
+            if (ex != null)
             {
                 var libuvOutputConsumer = (LibuvOutputConsumer)state;
                 libuvOutputConsumer._socket.Dispose();