btcpayserver/BTCPayServer/Hosting/HeadersOverrideMiddleware.cs

33 lines
1.0 KiB
C#

using System;
using System.Threading.Tasks;
using BTCPayServer.Configuration;
using Microsoft.AspNetCore.Http;
using Microsoft.Extensions.Configuration;
namespace BTCPayServer.Hosting
{
public class HeadersOverrideMiddleware
{
readonly RequestDelegate _Next;
readonly string overrideXForwardedProto;
public HeadersOverrideMiddleware(RequestDelegate next,
IConfiguration options)
{
_Next = next ?? throw new ArgumentNullException(nameof(next));
overrideXForwardedProto = options.GetOrDefault<string>("xforwardedproto", null);
}
public async Task Invoke(HttpContext httpContext)
{
if (!string.IsNullOrEmpty(overrideXForwardedProto))
{
if (!httpContext.Request.Host.Host.EndsWith(".onion", StringComparison.OrdinalIgnoreCase))
{
httpContext.Request.Headers["X-Forwarded-Proto"] = overrideXForwardedProto;
}
}
await _Next(httpContext);
}
}
}