diff --git a/lib/middleware/request_tracker.rb b/lib/middleware/request_tracker.rb index e160b192773..84d8da120bf 100644 --- a/lib/middleware/request_tracker.rb +++ b/lib/middleware/request_tracker.rb @@ -213,11 +213,9 @@ class Middleware::RequestTracker log_request_info(env, result, info) unless !log_request || env["discourse.request_tracker.skip"] end - PRIVATE_IP ||= /^(127\.)|(192\.168\.)|(10\.)|(172\.1[6-9]\.)|(172\.2[0-9]\.)|(172\.3[0-1]\.)|(::1$)|([fF][cCdD])/ - def is_private_ip?(ip) ip = IPAddr.new(ip) rescue nil - !!(ip && ip.to_s.match?(PRIVATE_IP)) + !!(ip && (ip.private? || ip.loopback?)) end def rate_limit(request) diff --git a/spec/components/middleware/request_tracker_spec.rb b/spec/components/middleware/request_tracker_spec.rb index 7b36e61f20f..2e6734e2da9 100644 --- a/spec/components/middleware/request_tracker_spec.rb +++ b/spec/components/middleware/request_tracker_spec.rb @@ -181,13 +181,18 @@ describe Middleware::RequestTracker do global_setting :max_reqs_per_ip_mode, 'warn+block' global_setting :max_reqs_rate_limit_on_private, true - env1 = env("REMOTE_ADDR" => "127.0.0.2") + addresses = %w[127.1.2.3 127.0.0.2 192.168.1.2 10.0.1.2 172.16.9.8 172.19.1.2 172.20.9.8 172.29.1.2 172.30.9.8 172.31.1.2] + warn_count = 1 + addresses.each do |addr| + env1 = env("REMOTE_ADDR" => addr) - status, _ = middleware.call(env1) - status, _ = middleware.call(env1) + status, _ = middleware.call(env1) + status, _ = middleware.call(env1) - expect(Rails.logger.warnings).to eq(1) - expect(status).to eq(429) + expect(Rails.logger.warnings).to eq(warn_count) + expect(status).to eq(429) + warn_count += 1 + end end describe "register_ip_skipper" do @@ -223,13 +228,16 @@ describe Middleware::RequestTracker do global_setting :max_reqs_per_ip_mode, 'warn+block' global_setting :max_reqs_rate_limit_on_private, false - env1 = env("REMOTE_ADDR" => "127.0.3.1") + addresses = %w[127.1.2.3 127.0.3.1 192.168.1.2 10.0.1.2 172.16.9.8 172.19.1.2 172.20.9.8 172.29.1.2 172.30.9.8 172.31.1.2] + addresses.each do |addr| + env1 = env("REMOTE_ADDR" => addr) - status, _ = middleware.call(env1) - status, _ = middleware.call(env1) + status, _ = middleware.call(env1) + status, _ = middleware.call(env1) - expect(Rails.logger.warnings).to eq(0) - expect(status).to eq(200) + expect(Rails.logger.warnings).to eq(0) + expect(status).to eq(200) + end end it "does warn if rate limiter is enabled via warn+block" do