From 1ef76b562f92eba5c0bf68c3e183430197fb3cbb Mon Sep 17 00:00:00 2001
From: Kyle Spearrin <kyle.spearrin@gmail.com>
Date: Tue, 15 Aug 2017 16:33:38 -0400
Subject: [PATCH] only use rate limiting on non-self host

---
 src/Api/Startup.cs | 23 ++++++++++++++++-------
 1 file changed, 16 insertions(+), 7 deletions(-)

diff --git a/src/Api/Startup.cs b/src/Api/Startup.cs
index 3adbc3e762..74978537c3 100644
--- a/src/Api/Startup.cs
+++ b/src/Api/Startup.cs
@@ -44,8 +44,11 @@ namespace Bit.Api
 
             // Settings
             var globalSettings = services.AddGlobalSettingsServices(Configuration);
-            services.Configure<IpRateLimitOptions>(Configuration.GetSection("IpRateLimitOptions"));
-            services.Configure<IpRateLimitPolicies>(Configuration.GetSection("IpRateLimitPolicies"));
+            if(!globalSettings.SelfHosted)
+            {
+                services.Configure<IpRateLimitOptions>(Configuration.GetSection("IpRateLimitOptions"));
+                services.Configure<IpRateLimitPolicies>(Configuration.GetSection("IpRateLimitPolicies"));
+            }
 
             // Data Protection
             services.AddCustomDataProtectionServices(Environment, globalSettings);
@@ -62,9 +65,12 @@ namespace Bit.Api
             // Caching
             services.AddMemoryCache();
 
-            // Rate limiting
-            services.AddSingleton<IIpPolicyStore, MemoryCacheIpPolicyStore>();
-            services.AddSingleton<IRateLimitCounterStore, MemoryCacheRateLimitCounterStore>();
+            if(!globalSettings.SelfHosted)
+            {
+                // Rate limiting
+                services.AddSingleton<IIpPolicyStore, MemoryCacheIpPolicyStore>();
+                services.AddSingleton<IRateLimitCounterStore, MemoryCacheRateLimitCounterStore>();
+            }
 
             // Identity
             services.AddCustomIdentityServices(globalSettings);
@@ -158,8 +164,11 @@ namespace Bit.Api
                 app.UseForwardedHeadersForAzure();
             }
 
-            // Rate limiting
-            app.UseMiddleware<CustomIpRateLimitMiddleware>();
+            if(!globalSettings.SelfHosted)
+            {
+                // Rate limiting
+                app.UseMiddleware<CustomIpRateLimitMiddleware>();
+            }
 
             // Add static files to the request pipeline.
             app.UseStaticFiles();