From 65431da22c0bb471d7bf68c508ab9204546faaa2 Mon Sep 17 00:00:00 2001
From: Neo <neo@rayshift.io>
Date: Sun, 1 Nov 2020 21:42:02 +0000
Subject: [PATCH] Fix error pages

---
 KamihamaWeb/Startup.cs | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/KamihamaWeb/Startup.cs b/KamihamaWeb/Startup.cs
index 2fddfae..3f29e20 100644
--- a/KamihamaWeb/Startup.cs
+++ b/KamihamaWeb/Startup.cs
@@ -7,6 +7,7 @@ using KamihamaWeb.Services;
 using KamihamaWeb.Util;
 using Microsoft.AspNetCore.Builder;
 using Microsoft.AspNetCore.Hosting;
+using Microsoft.AspNetCore.HttpOverrides;
 using Microsoft.AspNetCore.HttpsPolicy;
 using Microsoft.AspNetCore.ResponseCompression;
 using Microsoft.Extensions.Configuration;
@@ -96,14 +97,19 @@ namespace KamihamaWeb
             }
             else
             {
+                app.UseForwardedHeaders(new ForwardedHeadersOptions
+                {
+                    ForwardedHeaders = ForwardedHeaders.XForwardedFor | ForwardedHeaders.XForwardedProto
+                });
                 app.UseExceptionHandler("/Home/Error");
                 // The default HSTS value is 30 days. You may want to change this for production scenarios, see https://aka.ms/aspnetcore-hsts.
-                app.UseHsts();
+                //app.UseHsts();
             }
+
             app.UseHttpCacheHeaders();
             app.UseHttpsRedirection();
             app.UseStaticFiles();
-
+            app.UseStatusCodePagesWithReExecute("/Error/{0}");
             app.UseRouting();
             app.UseResponseCompression();
             app.UseAuthorization();