From 658b47b2fc8b7765f5dfda3a7d07f951eddc1e4e Mon Sep 17 00:00:00 2001
From: Kyle Spearrin <kyle.spearrin@gmail.com>
Date: Mon, 21 May 2018 21:24:35 -0400
Subject: [PATCH] XForwardedFor on self host

---
 src/Api/Startup.cs | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/Api/Startup.cs b/src/Api/Startup.cs
index c545b323c0..2f7e082214 100644
--- a/src/Api/Startup.cs
+++ b/src/Api/Startup.cs
@@ -18,6 +18,7 @@ using IdentityModel;
 using IdentityServer4.AccessTokenValidation;
 using jsreport.AspNetCore;
 using Bit.Core.IdentityServer;
+using Microsoft.AspNetCore.HttpOverrides;
 
 namespace Bit.Api
 {
@@ -173,6 +174,13 @@ namespace Bit.Api
                 // Rate limiting
                 app.UseMiddleware<CustomIpRateLimitMiddleware>();
             }
+            else
+            {
+                app.UseForwardedHeaders(new ForwardedHeadersOptions
+                {
+                    ForwardedHeaders = ForwardedHeaders.XForwardedFor | ForwardedHeaders.XForwardedProto
+                });
+            }
 
             // Add static files to the request pipeline.
             app.UseStaticFiles();