Improved request error handling.

aspnet · Nov 3, 2017 · f24a7f2 · f24a7f2
1 parent 5c2e7e9
commit f24a7f2
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/src/Microsoft.AspNetCore.Server.WebListener/MessagePump.cs b/src/Microsoft.AspNetCore.Server.WebListener/MessagePump.cs
@@ -127,7 +127,7 @@ private async void ProcessRequestsWorker()
                     {
                         LogHelper.LogException(_logger, "ListenForNextRequestAsync", exception);
                     }
-                    return;
+                    continue;
                 }
                 try
                 {

diff --git a/src/Microsoft.Net.Http.Server/AsyncAcceptContext.cs b/src/Microsoft.Net.Http.Server/AsyncAcceptContext.cs
@@ -79,6 +79,11 @@ private static void IOCompleted(AsyncAcceptContext asyncResult, uint errorCode,
                                 complete = true;
                             }
                         }
+                        catch (Exception)
+                        {
+                            server.SendError(asyncResult._nativeRequestContext.RequestId, HttpStatusCode.BadRequest);
+                            throw;
+                        }
                         finally
                         {
                             if (stoleBlob)

diff --git a/src/Microsoft.Net.Http.Server/WebListener.cs b/src/Microsoft.Net.Http.Server/WebListener.cs
@@ -318,7 +318,7 @@ internal unsafe bool ValidateAuth(NativeRequestContext requestMemory)
             return true;
         }
 
-        private unsafe void SendError(ulong requestId, HttpStatusCode httpStatusCode, IList<string> authChallenges)
+        internal unsafe void SendError(ulong requestId, HttpStatusCode httpStatusCode, IList<string> authChallenges = null)
         {
             HttpApi.HTTP_RESPONSE_V2 httpResponse = new HttpApi.HTTP_RESPONSE_V2();
             httpResponse.Response_V1.Version = new HttpApi.HTTP_VERSION();