Sur la connexion (mot de passe + code 2FA, même compteur pour les deux — un attaquant qui connaît le mot de passe ne doit pas avoir un nombre illimité d'essais sur le code) et sur la confirmation 2FA de l'inscription : 3 tentatives libres, puis un verrouillage qui double à chaque nouvel échec (5, 10, 20, 40 minutes...), plafonné à 1h (auth/rate_limit.py). Remis à zéro dès une connexion RÉELLEMENT aboutie (mot de passe ET code corrects) — jamais sur le seul succès du mot de passe, pour ne jamais donner un nombre illimité d'essais sur le 2FA à qui connaît déjà le mot de passe. Le verrouillage est annoncé IMMÉDIATEMENT sur la réponse qui le déclenche (record_failed_attempt renvoie la durée qu'il vient de poser), pas seulement découvert au prochain essai. Colonnes ajoutées en ALTER TABLE (failed_attempts, locked_until) pour ne rien casser sur une base de comptes déjà créée avant cette fonctionnalité. 14 tests dans test_auth.py (dont l'escalade 5/10/20/40/60, le blocage même avec le bon mot de passe une fois verrouillé, et la remise à zéro sur connexion réussie). 169 tests au total, tous au vert. Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
52 lines
2.2 KiB
Python
52 lines
2.2 KiB
Python
from flask import redirect, render_template, request, session, url_for
|
|
|
|
import auth
|
|
import db
|
|
|
|
from core.flask_app import app
|
|
|
|
|
|
@app.route("/register/2fa", methods=["GET", "POST"])
|
|
def register_2fa():
|
|
user_id = session.get("pending_register_user_id")
|
|
if not user_id:
|
|
return redirect(url_for("register"))
|
|
user = auth.get_user_by_id(user_id)
|
|
if not user:
|
|
session.pop("pending_register_user_id", None)
|
|
return redirect(url_for("register"))
|
|
|
|
error = None
|
|
if request.method == "POST":
|
|
locked_seconds = auth.seconds_locked_remaining(user)
|
|
if locked_seconds:
|
|
error = auth.lockout_message(locked_seconds)
|
|
elif not auth.verify_totp(user["totp_secret"], request.form.get("code", "")):
|
|
just_locked_minutes = auth.record_failed_attempt(user_id)
|
|
error = auth.lockout_message(just_locked_minutes * 60) if just_locked_minutes else \
|
|
"Code invalide — vérifie l'heure de ton téléphone et réessaie."
|
|
else:
|
|
auth.reset_failed_attempts(user_id)
|
|
auth.confirm_totp(user_id)
|
|
# Un compte "user" (pas "admin") reçoit ICI son unique projet,
|
|
# nommé d'après son adresse email (voir create_user.py — le
|
|
# rôle est déjà tranché à la création : premier compte jamais
|
|
# créé = admin, tous les suivants = user) — un admin, lui,
|
|
# reste libre de créer ses jeux comme avant ("+ Nouveau jeu"),
|
|
# aucun projet ne lui est imposé d'office.
|
|
if user["role"] != "admin":
|
|
slug = db.create_game(f"Jeu de {user['prenom']}", slug_override=user["email"])
|
|
auth.set_project_slug(user_id, slug)
|
|
user = auth.get_user_by_id(user_id)
|
|
session.pop("pending_register_user_id", None)
|
|
session["user_id"] = user_id
|
|
if user["role"] == "admin":
|
|
return redirect(url_for("index"))
|
|
return redirect(url_for("game_dashboard", slug=user["project_slug"]))
|
|
|
|
qr_svg = auth.totp_qrcode_svg(user["totp_secret"], user["email"])
|
|
return render_template(
|
|
"auth/register_2fa.html", error=error, qr_svg=qr_svg,
|
|
secret=user["totp_secret"], email=user["email"],
|
|
)
|