413cd34c45
* Tie rate limits to both the user's ID and their IP address. * Make each endpoint have separate rate limits. This means that, for example, your post edit rate limit is separate from your post vote rate limit. Before all write actions had a shared rate limit. * Make all write endpoints have rate limits. Before some endpoints, such as voting, favoriting, commenting, or forum posting, weren't subject to rate limits. * Add stricter rate limits for some endpoints: ** 1 per 5 minutes for creating new accounts. ** 1 per minute for login attempts, changing your email address, or for creating mod reports. ** 1 per minute for sending dmails, creating comments, creating forum posts, or creating forum topics. ** 1 per second for voting, favoriting, or disapproving posts. ** These rate limits all have burst factors high enough that they shouldn't affect normal, non-automated users. * Raise the default write rate limit for Gold users from 2 per second to 4 per second, for all other actions not listed above. * Raise the default burst factor to 200 for all other actions not listed above. Before it was 10 for Members, 30 for Gold, and 60 for Platinum.
53 lines
1.5 KiB
Ruby
53 lines
1.5 KiB
Ruby
class RateLimiter
|
|
class RateLimitError < StandardError; end
|
|
|
|
attr_reader :action, :keys, :cost, :rate, :burst
|
|
|
|
def initialize(action, keys = ["*"], cost: 1, rate: 1, burst: 1)
|
|
@action = action
|
|
@keys = keys
|
|
@cost = cost
|
|
@rate = rate
|
|
@burst = burst
|
|
end
|
|
|
|
def self.for_action(controller_name, action_name, user, ip_addr)
|
|
action = "#{controller_name}:#{action_name}"
|
|
keys = [(user.cache_key unless user.is_anonymous?), "ip/#{ip_addr.to_s}"].compact
|
|
|
|
case action
|
|
when "users:create"
|
|
rate, burst = 1.0/5.minutes, 10
|
|
when "emails:update", "sessions:create", "moderation_reports:create"
|
|
rate, burst = 1.0/1.minute, 10
|
|
when "dmails:create", "comments:create", "forum_posts:create", "forum_topics:create"
|
|
rate, burst = 1.0/1.minute, 50
|
|
when "comment_votes:create", "comment_votes:destroy", "post_votes:create",
|
|
"post_votes:destroy", "favorites:create", "favorites:destroy", "post_disapprovals:create"
|
|
rate, burst = 1.0/1.second, 200
|
|
else
|
|
rate = user.api_regen_multiplier
|
|
burst = 200
|
|
end
|
|
|
|
RateLimiter.new(action, keys, rate: rate, burst: burst)
|
|
end
|
|
|
|
def limit!
|
|
raise RateLimitError if limited?
|
|
end
|
|
|
|
def limited?
|
|
rate_limits.any?(&:limited?)
|
|
end
|
|
|
|
def as_json(options = {})
|
|
hash = rate_limits.map { |limit| [limit.key, limit.points] }.to_h
|
|
super(options).except("keys", "rate_limits").merge(limits: hash)
|
|
end
|
|
|
|
def rate_limits
|
|
@rate_limits ||= RateLimit.create_or_update!(action: action, keys: keys, cost: cost, rate: rate, burst: burst)
|
|
end
|
|
end
|