Add reset support for IIS (#24552)

Author: jkotalik

src/Servers/IIS/AspNetCoreModuleV2/InProcessRequestHandler/managedexports.cpp

@@ -589,4 +589,16 @@ http_response_set_trailer(
     IHttpResponse4* pHttpResponse = (IHttpResponse4*)pInProcessHandler->QueryHttpContext()->GetResponse();
     return pHttpResponse->SetTrailer(pszHeaderName, pszHeaderValue, usHeaderValueLength, fReplace);
 }
+
+EXTERN_C __MIDL_DECLSPEC_DLLEXPORT
+VOID
+http_reset_stream(
+    _In_ IN_PROCESS_HANDLER* pInProcessHandler,
+    ULONG errorCode
+)
+{
+    IHttpResponse4* pHttpResponse = (IHttpResponse4*)pInProcessHandler->QueryHttpContext()->GetResponse();
+    pHttpResponse->ResetStream(errorCode);
+}
+
 // End of export

src/Servers/IIS/IIS/src/Core/IISHttpContext.FeatureCollection.cs

@@ -33,7 +33,8 @@ internal partial class IISHttpContext : IFeatureCollection,
                                             ITlsConnectionFeature,
                                             IHttpBodyControlFeature,
                                             IHttpMaxRequestBodySizeFeature,
-                                            IHttpResponseTrailersFeature
+                                            IHttpResponseTrailersFeature,
+                                            IHttpResetFeature
     {
         // NOTE: When feature interfaces are added to or removed from this HttpProtocol implementation,
         // then the list of `implementedFeatures` in the generated code project MUST also be updated.
@@ -394,6 +395,33 @@ IHeaderDictionary IHttpResponseTrailersFeature.Trailers
             set => ResponseTrailers = value;
         }
 
+        internal IHttpResetFeature GetResetFeature()
+        {
+            // Check version is above 2.
+            if (HttpVersion >= System.Net.HttpVersion.Version20 && NativeMethods.HttpSupportTrailer(_pInProcessHandler))
+            {
+                return this;
+            }
+
+            return null;
+        }
+
+        void IHttpResetFeature.Reset(int errorCode)
+        {
+            if (errorCode < 0)
+            {
+                throw new ArgumentOutOfRangeException("'errorCode' cannot be negative");
+            }
+
+            SetResetCode(errorCode);
+            AbortIO(clientDisconnect: false);
+        }
+
+        internal unsafe void SetResetCode(int errorCode)
+        {
+            NativeMethods.HttpResetStream(_pInProcessHandler, (ulong)errorCode);
+        }
+
         void IHttpResponseBodyFeature.DisableBuffering()
         {
             NativeMethods.HttpDisableBuffering(_pInProcessHandler);

src/Servers/IIS/IIS/src/Core/IISHttpContext.Features.cs

@@ -29,6 +29,7 @@ internal partial class IISHttpContext
         private static readonly Type IServerVariablesFeature = typeof(global::Microsoft.AspNetCore.Http.Features.IServerVariablesFeature);
         private static readonly Type IHttpMaxRequestBodySizeFeature = typeof(global::Microsoft.AspNetCore.Http.Features.IHttpMaxRequestBodySizeFeature);
         private static readonly Type IHttpResponseTrailersFeature = typeof(global::Microsoft.AspNetCore.Http.Features.IHttpResponseTrailersFeature);
+        private static readonly Type IHttpResetFeature = typeof(global::Microsoft.AspNetCore.Http.Features.IHttpResetFeature);
 
         private object _currentIHttpRequestFeature;
         private object _currentIHttpResponseFeature;
@@ -50,6 +51,7 @@ internal partial class IISHttpContext
         private object _currentIServerVariablesFeature;
         private object _currentIHttpMaxRequestBodySizeFeature;
         private object _currentIHttpResponseTrailersFeature;
+        private object _currentIHttpResetFeature;
 
         private void Initialize()
         {
@@ -66,6 +68,7 @@ private void Initialize()
             _currentIHttpMaxRequestBodySizeFeature = this;
             _currentITlsConnectionFeature = this;
             _currentIHttpResponseTrailersFeature = GetResponseTrailersFeature();
+            _currentIHttpResetFeature = GetResetFeature();
         }
 
         internal object FastFeatureGet(Type key)
@@ -154,6 +157,10 @@ internal object FastFeatureGet(Type key)
             {
                 return _currentIHttpResponseTrailersFeature;
             }
+            if (key == IHttpResetFeature)
+            {
+                return _currentIHttpResetFeature;
+            }
 
             return ExtraFeatureGet(key);
         }
@@ -260,6 +267,10 @@ internal void FastFeatureSet(Type key, object feature)
             {
                 _currentIHttpResponseTrailersFeature = feature;
             }
+            if (key == IHttpResetFeature)
+            {
+                _currentIHttpResetFeature = feature;
+            }
             if (key == IISHttpContextType)
             {
                 throw new InvalidOperationException("Cannot set IISHttpContext in feature collection");
@@ -349,6 +360,10 @@ private IEnumerable<KeyValuePair<Type, object>> FastEnumerable()
             {
                 yield return new KeyValuePair<Type, object>(IHttpResponseTrailersFeature, _currentIHttpResponseTrailersFeature as global::Microsoft.AspNetCore.Http.Features.IHttpResponseTrailersFeature);
             }
+            if (_currentIHttpResetFeature != null)
+            {
+                yield return new KeyValuePair<Type, object>(IHttpResponseTrailersFeature, _currentIHttpResetFeature as global::Microsoft.AspNetCore.Http.Features.IHttpResetFeature);
+            }
 
             if (MaybeExtra != null)
             {

src/Servers/IIS/IIS/src/Core/IISHttpContextOfT.cs

@@ -59,6 +59,13 @@ public override async Task<bool> ProcessRequestAsync()
                     // Dispose
                 }
 
+                if (!success && HasResponseStarted)
+                {
+                    // HTTP/2 INTERNAL_ERROR = 0x2 https://tools.ietf.org/html/rfc7540#section-7
+                    // Otherwise the default is Cancel = 0x8.
+                    SetResetCode(2);
+                }
+
                 if (_onCompleted != null)
                 {
                     await FireOnCompleted();

src/Servers/IIS/IIS/src/NativeMethods.cs

@@ -146,6 +146,9 @@ private static extern unsafe int http_websockets_write_bytes(
         [DllImport(AspNetCoreModuleDll)]
         private static extern unsafe int http_response_set_trailer(IntPtr pInProcessHandler, byte* pszHeaderName, byte* pszHeaderValue, ushort usHeaderValueLength, bool replace);
 
+        [DllImport(AspNetCoreModuleDll)]
+        private static extern unsafe int http_reset_stream(IntPtr pInProcessHandler, ulong errorCode);
+
         [DllImport(AspNetCoreModuleDll)]
         private static extern unsafe int http_response_set_known_header(IntPtr pInProcessHandler, int headerId, byte* pHeaderValue, ushort length, bool fReplace);
 
@@ -318,6 +321,11 @@ internal static unsafe void HttpResponseSetTrailer(IntPtr pInProcessHandler, byt
             Validate(http_response_set_trailer(pInProcessHandler, pHeaderName, pHeaderValue, length, false));
         }
 
+        internal static unsafe void HttpResetStream(IntPtr pInProcessHandler, ulong errorCode)
+        {
+            Validate(http_reset_stream(pInProcessHandler, errorCode));
+        }
+
         internal static unsafe bool HttpSupportTrailer(IntPtr pInProcessHandler)
         {
             bool supportsTrailers;