rack-attack/lib/rack/attack/allow2ban.rb
Jordan Moncharmont ef59c5182a Allow2Ban
An alternate to fail2ban that allows clients until they hit the
thresholds, then blocks them.  Think of it like a throttle where you can
block for more than one period.
2013-09-27 17:18:52 -07:00

24 lines
607 B
Ruby

module Rack
module Attack
class Allow2Ban < Fail2Ban
class << self
protected
def key_prefix
'allow2ban'
end
# everything the same here except we return only return true
# (blocking the request) if they have tripped the limit.
def fail!(discriminator, bantime, findtime, maxretry)
count = cache.count("#{key_prefix}:count:#{discriminator}", findtime)
if count >= maxretry
ban!(discriminator, bantime)
true
else
false
end
end
end
end
end
end