fix: handle nil match_data in throttling (#2179)

nervosnetwork · Sep 6, 2024 · 4098e91 · 4098e91
1 parent ca8faea
commit 4098e91
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/config/initializers/rack_attack.rb b/config/initializers/rack_attack.rb
@@ -44,14 +44,17 @@ class Rack::Attack
   self.throttled_responder =
     lambda do |env|
       match_data = env["rack.attack.match_data"]
-      now = match_data[:epoch_time]
 
-      headers = {
-        "RateLimit-Limit" => match_data[:limit].to_s,
-        "RateLimit-Remaining" => "0",
-        "RateLimit-Reset" => (now + (match_data[:period] - now % match_data[:period])).to_s,
-      }
-
-      [429, headers, ["Throttled\n"]]
+      if match_data
+        now = match_data[:epoch_time]
+        headers = {
+          "RateLimit-Limit" => match_data[:limit].to_s,
+          "RateLimit-Remaining" => "0",
+          "RateLimit-Reset" => (now + (match_data[:period] - now % match_data[:period])).to_s,
+        }
+        [429, headers, ["Throttled\n"]]
+      else
+        [429, {}, ["Throttled but no match data\n"]]
+      end
     end
 end