🐛 Fix known proxies

2025-09-06 22:15:27 +08:00
parent 68bfdebcbd
commit 02680d224a
6 changed files with 55 additions and 105 deletions
--- a/DysonNetwork.Shared/Http/KnownProxiesConfiguration.cs
+++ b/DysonNetwork.Shared/Http/KnownProxiesConfiguration.cs
@@ -0,0 +1,45 @@
+using System.Net;
+using Microsoft.AspNetCore.Builder;
+using Microsoft.AspNetCore.HttpOverrides;
+using Microsoft.Extensions.Configuration;
+using IPNetwork = Microsoft.AspNetCore.HttpOverrides.IPNetwork;
+
+namespace DysonNetwork.Shared.Http;
+
+public static class KnownProxiesConfiguration
+{
+    public static WebApplication ConfigureForwardedHeaders(this WebApplication app, IConfiguration configuration)
+    {
+        var knownProxiesSection = configuration.GetSection("KnownProxies");
+        var forwardedHeadersOptions = new ForwardedHeadersOptions { ForwardedHeaders = ForwardedHeaders.All };
+
+        if (knownProxiesSection.Exists())
+        {
+            var proxyAddresses = knownProxiesSection.Get<string[]>();
+            if (proxyAddresses != null)
+            {
+                foreach (var proxy in proxyAddresses)
+                {
+                    if (IPAddress.TryParse(proxy, out var ipAddress))
+                    {
+                        forwardedHeadersOptions.KnownProxies.Add(ipAddress);
+                    }
+                    else if (IPNetwork.TryParse(proxy, out var ipNetwork))
+                    {
+                        forwardedHeadersOptions.KnownNetworks.Add(ipNetwork);
+                    }
+                }
+            }
+        }
+
+        if (forwardedHeadersOptions.KnownProxies.Count == 0 && forwardedHeadersOptions.KnownNetworks.Count == 0)
+        {
+            forwardedHeadersOptions.KnownProxies.Add(IPAddress.Any);
+            forwardedHeadersOptions.KnownProxies.Add(IPAddress.IPv6Any);
+        }
+
+        app.UseForwardedHeaders(forwardedHeadersOptions);
+
+        return app;
+    }
+}