fix issue #26 AspNetCoreModule needs to abort failed responses

aspnet · Oct 31, 2016 · 388b906 · 388b906
1 parent ae727d6
commit 388b906
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 1 deletion.
diff --git a/src/AspNetCore/Inc/forwardinghandler.h b/src/AspNetCore/Inc/forwardinghandler.h
@@ -27,6 +27,7 @@ enum FORWARDING_REQUEST_STATUS
     FORWARDER_SENDING_REQUEST,
     FORWARDER_RECEIVING_RESPONSE,
     FORWARDER_RECEIVED_WEBSOCKET_RESPONSE,
+    FORWARDER_RESET_CONNECTION,
     FORWARDER_DONE
 };
 

diff --git a/src/AspNetCore/Src/dllmain.cpp b/src/AspNetCore/Src/dllmain.cpp
@@ -51,7 +51,7 @@ VOID
     HKEY hKey;
 
     if (RegOpenKeyEx(HKEY_LOCAL_MACHINE,
-        L"SOFTWARE\\Microsoft\\IIS Extensions\\AspNetCore Module\\Parameters",
+        L"SOFTWARE\\Microsoft\\IIS Extensions\\IIS AspNetCore Module\\Parameters",
         0,
         KEY_READ,
         &hKey) == NO_ERROR)

diff --git a/src/AspNetCore/Src/forwardinghandler.cxx b/src/AspNetCore/Src/forwardinghandler.cxx
@@ -1591,6 +1591,12 @@ Return Value:
         retVal = RQ_NOTIFICATION_PENDING;
         goto Finished;
     }
+    else if (m_RequestStatus == FORWARDER_RESET_CONNECTION)
+    {
+        hr = HRESULT_FROM_WIN32(ERROR_WINHTTP_INVALID_SERVER_RESPONSE);
+        goto Failure;
+
+    }
 
     //
     // Begins normal completion handling. There is already a shared acquired
@@ -2185,6 +2191,12 @@ None
 
 Failure:
 
+    if (hr == HRESULT_FROM_WIN32(ERROR_WINHTTP_INVALID_SERVER_RESPONSE))
+    {
+        m_RequestStatus = FORWARDER_RESET_CONNECTION;
+        goto Finished;
+    }
+
     m_RequestStatus = FORWARDER_DONE;
 
     pResponse->DisableKernelCache();