diff --git a/lib/travis/api/attack.rb b/lib/travis/api/attack.rb index 5b945eca..411980bf 100644 --- a/lib/travis/api/attack.rb +++ b/lib/travis/api/attack.rb @@ -31,12 +31,18 @@ class Rack::Attack "/auth/post_message/iframe" ] + IMAGE_PATTERN = /^\/([a-z0-9_-]+)\/([a-z0-9_-]+)\.(png|svg)$/ + #### # Whitelisted IP addresses whitelist('whitelist client requesting from redis') do |request| Travis.redis.sismember(:api_whitelisted_ips, request.ip) end + whitelist('safelist build status images when requested by github') do |request| + request.user_agent and request.user_agent.start_with?('github-camo') and IMAGE_PATTERN.match(request.path) + end + #### # Ban based on: IP address # Ban time: indefinite