ASP.NET Core中间件中修改HttpContext请求/响应的实现问题求助

我来帮你分析并修复这两个在自定义加密解密中间件里常见的问题：

一、请求内容无法传递到控制器的原因与修复 #

你的ModifyRequest方法存在两个核心问题：

1. 原始请求流是单向只读流，读取到末尾后无法自动重置位置，控制器后续读取会得到空内容
2. 替换流后没有更新请求的ContentLength，ASP.NET Core可能会因为长度不匹配忽略新内容

修正后的ModifyRequest方法：

public async Task ModifyRequest(HttpContext context)
{
    var request = context.Request;
    // 先把原始请求流复制到可重读的MemoryStream
    using var originalMs = new MemoryStream();
    await request.Body.CopyToAsync(originalMs);
    originalMs.Position = 0;

    var originalReader = new StreamReader(originalMs);
    var originalContent = await originalReader.ReadToEndAsync();
    originalMs.Position = 0; // 重置位置，以防解密失败时需要复用

    string modifiedContent = originalContent;
    try
    {
        // 解密数据
        var decrypted = Decrypt(originalContent);
        if (!string.IsNullOrEmpty(decrypted))
        {
            modifiedContent = decrypted;
        }
    }
    catch (Exception e)
    {
        Console.WriteLine($"请求解密失败: {e.Message}");
    }

    // 生成新的请求流
    var newMs = new MemoryStream(Encoding.UTF8.GetBytes(modifiedContent));
    request.Body = newMs;
    request.ContentLength = newMs.Length; // 必须更新内容长度
    request.Body.Position = 0; // 确保流在起始位置
}

二、响应内容无法写入上下文的原因与修复 #

你的响应处理逻辑顺序错误：在调用next(context)之后，原始响应流已经被写入，此时再替换流并写入不会生效。正确的做法是先替换响应流为MemoryStream，再执行后续中间件/控制器，最后处理MemoryStream里的响应内容并写回原始流。

首先调整InvokeAsync的逻辑：

public async Task InvokeAsync(HttpContext context, RequestDelegate next)
{
    // 先处理请求解密
    await ModifyRequest(context);

    // 保存原始响应流，替换为MemoryStream用于捕获响应
    var originalResponseStream = context.Response.Body;
    using var responseMs = new MemoryStream();
    context.Response.Body = responseMs;

    try
    {
        // 执行后续中间件和控制器
        await next(context);

        // 处理响应加密
        await ModifyResponse(context, originalResponseStream);
    }
    finally
    {
        // 恢复原始响应流
        context.Response.Body = originalResponseStream;
    }
}

然后修改ModifyResponse方法，专门处理捕获到的响应内容：

public async Task ModifyResponse(HttpContext context, Stream originalResponseStream)
{
    var responseMs = context.Response.Body as MemoryStream;
    if (responseMs == null) return;

    // 重置流位置到起始处，读取原始响应内容
    responseMs.Position = 0;
    var responseReader = new StreamReader(responseMs);
    var plainResponse = await responseReader.ReadToEndAsync();
    Console.WriteLine("PlainResponse Body: {0}", plainResponse);

    // 加密响应内容
    var encryptedResponse = Encrypt(plainResponse);
    var encryptedBytes = Encoding.UTF8.GetBytes(encryptedResponse);

    // 更新响应头和内容
    context.Response.ContentLength = encryptedBytes.Length;
    context.Response.ContentType = "application/json"; // 确保内容类型正确

    // 将加密后的内容写入原始响应流
    await originalResponseStream.WriteAsync(encryptedBytes, 0, encryptedBytes.Length);
    await originalResponseStream.FlushAsync();
}

额外注意事项 #

• 确保你的Decrypt和Encrypt方法能正确处理Base64字符串和JSON内容，避免编码错误
• 生产环境建议添加更完善的异常捕获和日志记录，避免因为加密解密失败导致请求中断
• 如果后续需要处理其他内容类型，可根据Request.ContentType做分支判断，但当前场景下application/json的逻辑足够

内容的提问来源于stack exchange，提问作者Tony Riddle