NotBigGhostandClaude Opus 5
ec1e94119c
Комментарии про методы входа приведены к факту (логин/пароль и Telegram везде, stub только в development): config.py, .env.example, telegram.py, dev_stub.py, dev_auth.py, login.py. admin_login.py больше не обещает «строже игроцкого» — описаны реальные лимиты; client_ip — адрес за прокси уже даёт uvicorn. Подсказки деплоя: run.ps1/run.sh (build-push на ПК, `up -d` на Pi), compose, .env.example (build-push.ps1), Caddyfile (туннель-контейнер вместо autossh); ratelimit.py ссылается на core/events.py вместо CLAUDE.md. Удалены неиспользуемые scripts/export-prod.sh (с блоком export-ignore в .gitattributes и упоминаниями в README), db/init_db.py и случайно закоммиченный файл «h -u origin dev…». #70 Co-Authored-By: Claude Opus 5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01LqSoRj99iwVEH5U5fnZgsd
80 lines
3.9 KiB
Python
80 lines
3.9 KiB
Python
"""Ограничение неудачных попыток ввода пароля (защита от перебора).
|
|
|
|
Счётчики живут в памяти процесса — как и SSE-шина, это рассчитано на один воркер uvicorn
|
|
(`--workers 1`, см. entrypoint.sh/run.*). Перезапуск их обнуляет; для окна в 15 минут это
|
|
приемлемо. Ограничения устойчивости (#60):
|
|
* рестарт (в т.ч. деплой) сбрасывает окно — злоумышленник получает новую квоту после
|
|
перезапуска, но окно короткое, а рестарты редки;
|
|
* при уходе от одного воркера лимит делится между процессами (каждый считает своё) —
|
|
тогда счётчики нужно вынести во внешний стор (Redis pub/sub, как отмечено в core/events.py
|
|
про SSE-шину), общий для всех воркеров.
|
|
Помимо пары «IP + логин» и лимита по IP есть IP-независимый лимит на аккаунт
|
|
(`login-user:*` / `admin-login-user:*`), чтобы ротация X-Forwarded-For / многих адресов
|
|
(#58) не снимала защиту полностью.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import threading
|
|
import time
|
|
from collections import deque
|
|
|
|
from app.core.errors import TooManyAttemptsError
|
|
|
|
_WINDOW_SECONDS = 15 * 60
|
|
# Выше этого числа ключей при записи неудачи вычищаем протухшие, чтобы поток попыток
|
|
# с разных адресов не копил память бесконечно.
|
|
_PRUNE_ABOVE = 10_000
|
|
|
|
|
|
class LoginThrottle:
|
|
"""Скользящее окно неудач по произвольным ключам (IP, пара «IP + логин», id игрока).
|
|
|
|
Роуты синхронные и выполняются в пуле потоков, поэтому доступ под замком."""
|
|
|
|
def __init__(self, window_seconds: int = _WINDOW_SECONDS) -> None:
|
|
self.window = window_seconds
|
|
self._fails: dict[str, deque[float]] = {}
|
|
self._lock = threading.Lock()
|
|
|
|
def _recent(self, key: str, now: float) -> deque[float]:
|
|
attempts = self._fails.get(key)
|
|
if attempts is None:
|
|
return deque()
|
|
while attempts and now - attempts[0] >= self.window:
|
|
attempts.popleft()
|
|
if not attempts:
|
|
del self._fails[key]
|
|
return attempts
|
|
|
|
def check(self, limits: dict[str, int]) -> None:
|
|
"""Бросает TooManyAttemptsError, если хотя бы по одному ключу лимит исчерпан."""
|
|
now = time.monotonic()
|
|
with self._lock:
|
|
for key, limit in limits.items():
|
|
attempts = self._recent(key, now)
|
|
if len(attempts) >= limit:
|
|
# Попытка снова примется, когда из окна выпадет неудача, после которой
|
|
# в нём остаётся limit-1 записей.
|
|
frees_at = attempts[len(attempts) - limit] + self.window
|
|
raise TooManyAttemptsError(retry_after=int(frees_at - now) + 1)
|
|
|
|
def fail(self, keys: dict[str, int]) -> None:
|
|
now = time.monotonic()
|
|
with self._lock:
|
|
if len(self._fails) > _PRUNE_ABOVE:
|
|
for key in list(self._fails):
|
|
self._recent(key, now)
|
|
for key in keys:
|
|
self._fails.setdefault(key, deque()).append(now)
|
|
|
|
def reset(self, key: str) -> None:
|
|
with self._lock:
|
|
self._fails.pop(key, None)
|
|
|
|
def clear(self) -> None:
|
|
with self._lock:
|
|
self._fails.clear()
|
|
|
|
|
|
login_throttle = LoginThrottle()
|