From 8c3c4fe935bd07e63fbcf856a25abb29a82ed97d Mon Sep 17 00:00:00 2001
From: Konstantin Haase <konstantin.mailinglists@googlemail.com>
Date: Wed, 16 Dec 2015 15:06:15 +0100
Subject: [PATCH] add ip whitelisting

---
 lib/travis/api/attack.rb | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/lib/travis/api/attack.rb b/lib/travis/api/attack.rb
index dc0a247d..207396f5 100644
--- a/lib/travis/api/attack.rb
+++ b/lib/travis/api/attack.rb
@@ -31,6 +31,12 @@ class Rack::Attack
     "/auth/post_message/iframe"
   ]
 
+  ####
+  # Whitelisted IP addresses
+  whitelist('whitelist client requesting from redis') do |request|
+    Travis.redis.sismember(:api_whitelisted_ips, request.ip)
+  end
+
   ####
   # Ban based on: IP address
   # Ban time:     indefinite