[Named Pipe] Check for pipe broken on WaitConnectionAsync (#46230)

dotnet · Jan 25, 2023 · 6af56b8 · 6af56b8
1 parent 5ea213a
commit 6af56b8
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/Servers/Kestrel/Transport.NamedPipes/src/Internal/NamedPipeConnectionListener.cs b/src/Servers/Kestrel/Transport.NamedPipes/src/Internal/NamedPipeConnectionListener.cs
@@ -94,6 +94,15 @@ private async Task StartAsync(NamedPipeServerStream nextStream)
                         }
                     }
                 }
+                catch (IOException ex) when (!_listeningToken.IsCancellationRequested)
+                {
+                    // WaitForConnectionAsync can throw IOException when the pipe is broken.
+                    NamedPipeLog.ConnectionListenerBrokenPipe(_log, ex);
+
+                    // Dispose existing pipe, create a new one and continue accepting.
+                    nextStream.Dispose();
+                    nextStream = CreateServerStream();
+                }
                 catch (OperationCanceledException ex) when (_listeningToken.IsCancellationRequested)
                 {
                     // Cancelled the current token

diff --git a/src/Servers/Kestrel/Transport.NamedPipes/src/Internal/NamedPipeLog.cs b/src/Servers/Kestrel/Transport.NamedPipes/src/Internal/NamedPipeLog.cs
@@ -77,4 +77,6 @@ public static void ConnectionDisconnect(ILogger logger, NamedPipeConnection conn
         }
     }
 
+    [LoggerMessage(8, LogLevel.Debug, "Named pipe listener received broken pipe while waiting for a connection.", EventName = "ConnectionListenerBrokenPipe")]
+    public static partial void ConnectionListenerBrokenPipe(ILogger logger, Exception ex);
 }