C#:修改 Owin 响应流会导致访问违规异常

本文关键字:访问 异常 修改 Owin 响应 | 更新日期: 2023-09-27 17:56:09

我正在尝试使用一些自定义的 Owin 中间件在特定情况下修改(在本例中为完全替换)响应流。

每当我发出触发中间件替换响应的调用时,一切都正常。仅当我进行中间件未更改的调用时,才会出现问题。此外,我只能在被替换的 API 调用返回手动创建的 HttpResponseMessage 对象时发生错误。

例如调用此 API:

public class testController : ApiController
{
    public HttpResponseMessage Get()
    {
         return Request.CreateResponse(HttpStatusCode.OK,new { message = "It worked." });
    }
}

工作正常,但此类:

public class testController : ApiController
{
    public HttpResponseMessage Get()
    {
        HttpResponseMessage m = Request.CreateResponse();
        m.StatusCode = HttpStatusCode.OK;
        m.Content = new StringContent("It worked.", System.Text.Encoding.UTF8, "text/plain");
        return m;
    }
}

导致发生错误。(在这两种情况下,都会调用http://localhost:<port>/test

该错误会导致以下任一情况:

  • 导致 iisexpress.exe(或 w3wp.exe如果在实际 IIS 中运行)崩溃并出现访问冲突。
  • 引发 Visual Studio 捕获但无法执行任何操作的AccessViolationException,因为它发生在外部代码中。当Visual Studio确实捕获异常时,我看到:

    An unhandled exception of type 'System.AccessViolationException' occurred in System.Web.dll
    Additional information: Attempted to read or write protected memory. This is often an indication that other memory is corrupt.
    

显然,如果我不启用中间件,我根本没有问题。此外,我只能在手动创建和返回 HttpResponseMessage 对象时导致问题发生,如第二类所示。

这是我的中间件类。它目前设置为在有人请求终结点/replace时简单地替换整个响应流,而不管管道中是否有任何其他内容对其执行了任何操作。

using Microsoft.Owin;
using Owin;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Net;
using System.Threading.Tasks;
using Newtonsoft.Json;
using AppFunc = System.Func<
    System.Collections.Generic.IDictionary<string, object>,
    System.Threading.Tasks.Task
>;
namespace TestOwinAPI
{
    public class ResponseChangeMiddleware
    {
        AppFunc _next;
        public ResponseChangeMiddleware(AppFunc next, ResponseChangeMiddlewareOptions opts)
        {
            _next = next;
        }
        public async Task Invoke(IDictionary<string,object> env)
        {
            var ctx = new OwinContext(env);
            // create a new memory stream which will replace the default output stream
            using (var ms = new MemoryStream())
            {
                // hold on to a reference to the actual output stream for later use
                var outStream = ctx.Response.Body;
                // reassign the context's output stream to be our memory stream
                ctx.Response.Body = ms;
                Debug.WriteLine(" <- " + ctx.Request.Path);
                // allow the rest of the middleware to do its job
                await _next(env);
                // Now the request is on the way out.
                if (ctx.Request.Path.ToString() == "/replace")
                {
                    // Now write new response.
                    string json = JsonConvert.SerializeObject(new { response = "true", message = "This response will replace anything that the rest of the API might have created!" });
                    byte[] jsonBytes = System.Text.Encoding.UTF8.GetBytes(json);
                    // clear everything else that anything might have put in the output stream
                    ms.SetLength(0);
                    // write the new data
                    ms.Write(jsonBytes, 0, jsonBytes.Length);
                    // set parameters on the response object
                    ctx.Response.StatusCode = 200;
                    ctx.Response.ContentLength = jsonBytes.Length;
                    ctx.Response.ContentType = "application/json";
                }
                // In all cases finally write the memory stream's contents back to the actual response stream
                ms.Seek(0, SeekOrigin.Begin);
                await ms.CopyToAsync(outStream);
            }
        }
    }
    public static class AppBuilderExtender
    {
        public static void UseResponseChangeMiddleware(this IAppBuilder app, ResponseChangeMiddlewareOptions options = null )
        {
            if (options == null)
                options = new ResponseChangeMiddlewareOptions();
            app.Use<ResponseChangeMiddleware>(options);
        }
    }
    public class ResponseChangeMiddlewareOptions
    {
    }
}

我已经做了显而易见的事情 - 一整晚的 RAM 测试(一切都很好),并在另一个系统上尝试(它也发生在那里)。

此外,错误不一致 - 它发生在大约一半的时间。换句话说,通常我可以通过一两个成功的请求,但最终会发生错误。

最后,如果我在中间件中的内存流复制之前在程序中放置一个断点,并缓慢地单步执行代码,则永远不会发生错误。这向我表明我一定遇到了某种竞争条件,并且必须与我正在使用MemoryStreams的事实有关。

有什么想法吗?

C#:修改 Owin 响应流会导致访问违规异常

哦,天哪。

我不确定更改此设置是否正确,但它肯定解决了问题

await ms.CopyToAsync(outStream);

ms.CopyTo(outStream);

我唯一的猜测是,在异步调用完成复制之前,该应用程序以某种方式关闭了MemoryStream,这是有道理的。