diff --git a/CulinaryCommandApp/Program.cs b/CulinaryCommandApp/Program.cs index 3c07746..7ee9e0b 100644 --- a/CulinaryCommandApp/Program.cs +++ b/CulinaryCommandApp/Program.cs @@ -17,6 +17,8 @@ using Amazon.CognitoIdentityProvider; using Amazon.Extensions.NETCore.Setup; using CulinaryCommandApp.Services; +using Microsoft.AspNetCore.HttpOverrides; + @@ -154,12 +156,23 @@ builder.Services.AddScoped(); builder.Services.AddSingleton(); + +builder.Services.Configure(o => +{ + o.ForwardedHeaders = ForwardedHeaders.XForwardedFor | ForwardedHeaders.XForwardedProto; + o.KnownNetworks.Clear(); + o.KnownProxies.Clear(); +}); + + // // ===================== // Build App // ===================== var app = builder.Build(); +app.UseForwardedHeaders(); + // Determine whether the app should only run migrations and exit var migrateOnly = (Environment.GetEnvironmentVariable("MIGRATE_ONLY")?.Equals("true", StringComparison.OrdinalIgnoreCase) == true) || (args != null && args.Any(a => a.Equals("--migrate-only", StringComparison.OrdinalIgnoreCase)));