"""Anti-bruteforce sur les tentatives de connexion (mot de passe ET code 2FA — même compteur pour les deux, un compte ciblé reste protégé quelle que soit l'étape attaquée) : 3 essais libres, puis un temps d'attente qui double à chaque échec supplémentaire (5 min, 10, 20, 40...), plafonné à 1h. Remis à zéro dès une connexion réussie (voir reset_failed_attempts).""" import math from datetime import datetime, timezone from typing import Any _FREE_ATTEMPTS = 3 _FIRST_LOCKOUT_MINUTES = 5 _MAX_LOCKOUT_MINUTES = 60 def lockout_minutes_for(failed_attempts: int) -> int: """0 tant qu'on est dans les 3 essais libres ; sinon 5 * 2^n, plafonné à 60 — jamais négatif, jamais None.""" if failed_attempts <= _FREE_ATTEMPTS: return 0 exponent = failed_attempts - _FREE_ATTEMPTS - 1 return min(int(_FIRST_LOCKOUT_MINUTES * (2**exponent)), _MAX_LOCKOUT_MINUTES) def _parse(dt_str: str | None) -> datetime | None: if not dt_str: return None try: return datetime.fromisoformat(dt_str) except ValueError: return None def lockout_message(seconds_remaining: float) -> str: """Message affiché tel quel dans le formulaire — arrondi à la minute supérieure (jamais "0 minute" alors qu'il en reste un peu, jamais un compte de secondes qui oblige à recharger sans arrêt pour voir si c'est fini).""" minutes = max(1, math.ceil(seconds_remaining / 60)) unit = "minute" if minutes == 1 else "minutes" return f"Trop de tentatives. Réessaie dans {minutes} {unit}." def seconds_locked_remaining(user: dict[str, Any] | None) -> int: """> 0 si le compte est actuellement verrouillé (temps restant, en secondes, arrondi au-dessus pour ne jamais afficher "0 minute" alors qu'il en reste réellement un peu) ; 0 sinon.""" locked_until = _parse((user or {}).get("locked_until")) if not locked_until: return 0 remaining = (locked_until - datetime.now(timezone.utc)).total_seconds() return max(0, math.ceil(remaining))