瀏覽代碼

[Netfilter] do not ignore RETRY_WINDOW

FreddleSpl0it 1 年之前
父節點
當前提交
86fa8634ee
共有 1 個文件被更改,包括 5 次插入1 次删除
  1. 5 1
      data/Dockerfiles/netfilter/main.py

+ 5 - 1
data/Dockerfiles/netfilter/main.py

@@ -167,8 +167,12 @@ def ban(address):
   if not net in bans:
   if not net in bans:
     bans[net] = {'attempts': 0, 'last_attempt': 0, 'ban_counter': 0}
     bans[net] = {'attempts': 0, 'last_attempt': 0, 'ban_counter': 0}
 
 
+  current_attempt = time.time()
+  if current_attempt - bans[net]['last_attempt'] > RETRY_WINDOW:
+    bans[net]['attempts'] = 0
+
   bans[net]['attempts'] += 1
   bans[net]['attempts'] += 1
-  bans[net]['last_attempt'] = time.time()
+  bans[net]['last_attempt'] = current_attempt
 
 
   if bans[net]['attempts'] >= MAX_ATTEMPTS:
   if bans[net]['attempts'] >= MAX_ATTEMPTS:
     cur_time = int(round(time.time()))
     cur_time = int(round(time.time()))