Archived
refactor: login 端点拆解(90 行 / 圈 26 / 嵌套 2 → 39 行 / 圈 10 / 嵌套 1)
为什么:登录是全部拒绝路径最密的端点(超长密码/锁定/密码错/封禁/ 禁用/管理员无邮箱/异地封禁七道闸),90 行单函数把「身份解析、锁定 计数、状态闸、2FA 转票据、会话建立」五件事揉在一个 try 里,改任何 一道闸都要在嵌套分支里核对上下文。 拆出五个成员(顺序 = 原执行顺序,逐条语义对应): - _resolve_login_identity:邮箱→账号解析 + 锁定键(保留两处判定 差异:锁定键用真值判断,verify_name 用 is not None——历史口径, 契约测试分别钉住) - _allow_login_attempt:锁定闸 + 一次性审计日志(锁定周期内不重复落) - _check_account_status:封禁(原因落日志)/禁用,通过返回 None - _admin_two_factor_response:2FA 转票据,未绑定邮箱 fail-closed - _user_session_response:login_seq 顶号 + 异地多省封禁闸(封禁时 再递增一次顶掉在途会话)+ Cookie 签发 先写测试后动代码:新增 tests/test_login_flow.py 9 例,在未改动实现上 一次跑绿,覆盖此前零测试的路径——超长密码不碰 PBKDF2(monkeypatch verify_login 断言不被调用)、邮箱登录按归一化地址锁定、锁定日志幂等、 封禁/禁用文案、管理员 fail-closed、异地封禁双递增、非预期异常 500。 门禁:ruff 全过 · pyright 0/0 · pytest 348 passed(339→348)。
This commit is contained in:
1 parent
5ad2152acc
commit
f0de0b65b1
2 files changed
+254
-62
No files matched your search
@@ -0,0 +1,157 @@
|
|||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
"""login 端点失败路径契约(超长密码 / 邮箱登录 / 锁定 / 封禁禁用 / 2FA 前置 / 异常映射)。
|
||||||
|
|
||||||
|
test_auth_routes.py 覆盖成功主路径;本文件钉住 login 的各拒绝分支,
|
||||||
|
作为 login 拆解的等价性验证器。
|
||||||
|
"""
|
||||||
|
import pytest
|
||||||
|
from fastapi.testclient import TestClient
|
||||||
|
|
||||||
|
import auth_system as auth
|
||||||
|
import core.db as db
|
||||||
|
|
||||||
|
SECRET = 'login-flow-test-secret'
|
||||||
|
USER_PW = 'UserPass123'
|
||||||
|
ADMIN_PW = 'AdminPass123'
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture()
|
||||||
|
def env(tmp_path, monkeypatch):
|
||||||
|
db_path = tmp_path / 'site.db'
|
||||||
|
monkeypatch.setattr(auth.db, 'DATA_DIR', tmp_path)
|
||||||
|
monkeypatch.setattr(auth.db, 'DB_PATH', db_path)
|
||||||
|
monkeypatch.setattr(auth.db, 'BACKUP_DIR', tmp_path / 'backups')
|
||||||
|
monkeypatch.setattr(auth.db, 'SECRET_KEY_PATH', tmp_path / 'secret.key')
|
||||||
|
monkeypatch.setattr(db, 'DATA_DIR', tmp_path)
|
||||||
|
monkeypatch.setattr(db, 'DB_PATH', db_path)
|
||||||
|
auth.init_db()
|
||||||
|
|
||||||
|
conn = auth.get_conn()
|
||||||
|
conn.execute('INSERT INTO users (username, password_hash, salt, role, duration_seconds, '
|
||||||
|
'status, login_seq, created_at, email) VALUES (?,?,?,?,?,1,0,?,?)',
|
||||||
|
('mailuser', auth._hash_new(USER_PW), '', 'user', 97, 1730000000,
|
||||||
|
'mail@example.com'))
|
||||||
|
conn.execute('INSERT INTO users (username, password_hash, salt, role, duration_seconds, '
|
||||||
|
'status, login_seq, created_at) VALUES (?,?,?,?,?,1,0,?)',
|
||||||
|
('bossuser', auth._hash_new(ADMIN_PW), '', 'admin', -1, 1730000000))
|
||||||
|
conn.commit()
|
||||||
|
conn.close()
|
||||||
|
|
||||||
|
monkeypatch.setattr('web.deps.secret_key', lambda: SECRET)
|
||||||
|
|
||||||
|
import web.auth_state as auth_state
|
||||||
|
auth_state._LOGIN_FAILS.clear()
|
||||||
|
auth_state._LOGIN_FAIL_LOGGED.clear()
|
||||||
|
auth_state._ADMIN_2FA.clear()
|
||||||
|
|
||||||
|
from web.app import create_app
|
||||||
|
return TestClient(create_app())
|
||||||
|
|
||||||
|
|
||||||
|
def _logs(conn, action: str) -> list[tuple[str, str]]:
|
||||||
|
rows = conn.execute('SELECT type, detail FROM logs WHERE action = ?', (action,)).fetchall()
|
||||||
|
return [(r['type'], r['detail']) for r in rows]
|
||||||
|
|
||||||
|
|
||||||
|
def test_overlong_password_rejected_without_pbkdf2(env, monkeypatch):
|
||||||
|
"""超长密码(>128):不碰 PBKDF2,直接按密码错误拒绝。"""
|
||||||
|
def _boom(*a, **k):
|
||||||
|
raise AssertionError('verify_login must not be called')
|
||||||
|
|
||||||
|
monkeypatch.setattr(auth, 'verify_login', _boom)
|
||||||
|
r = env.post('/auth/login', json={'username': 'mailuser', 'password': 'x' * 129})
|
||||||
|
assert r.status_code == 400
|
||||||
|
assert r.json()['message'] == '用户名或密码错误'
|
||||||
|
|
||||||
|
|
||||||
|
def test_email_login_resolves_account(env):
|
||||||
|
"""邮箱登录:按归一化邮箱解析真实账号(大小写不敏感)。"""
|
||||||
|
r = env.post('/auth/login', json={'username': 'MAIL@Example.com', 'password': USER_PW})
|
||||||
|
assert r.status_code == 200
|
||||||
|
assert r.json()['success'] is True
|
||||||
|
assert 'session' in r.cookies
|
||||||
|
|
||||||
|
|
||||||
|
def test_unknown_email_locks_by_normalized_email(env):
|
||||||
|
"""邮箱登录失败锁定:键用归一化邮箱(不含大小写变体绕过)。"""
|
||||||
|
for _ in range(5):
|
||||||
|
r = env.post('/auth/login', json={'username': 'Ghost@Example.com', 'password': 'nope'})
|
||||||
|
assert r.json()['message'] == '用户名或密码错误'
|
||||||
|
r = env.post('/auth/login', json={'username': 'ghost@example.com', 'password': USER_PW})
|
||||||
|
assert '尝试过于频繁' in r.json()['message']
|
||||||
|
|
||||||
|
import web.auth_state as auth_state
|
||||||
|
assert 'ghost@example.com' in auth_state._LOGIN_FAILS
|
||||||
|
|
||||||
|
|
||||||
|
def test_locked_log_written_once(env):
|
||||||
|
"""锁定周期内重复被封尝试只落一条 fail 日志。"""
|
||||||
|
for _ in range(5):
|
||||||
|
env.post('/auth/login', json={'username': 'mailuser', 'password': 'nope'})
|
||||||
|
for _ in range(3):
|
||||||
|
env.post('/auth/login', json={'username': 'mailuser', 'password': USER_PW})
|
||||||
|
conn = auth.get_conn()
|
||||||
|
locked_logs = [d for t, d in _logs(conn, '登录') if d == '失败次数过多,临时锁定']
|
||||||
|
conn.close()
|
||||||
|
assert len(locked_logs) == 1
|
||||||
|
|
||||||
|
|
||||||
|
def test_banned_account_message(env, monkeypatch):
|
||||||
|
"""封禁账号(status=2):文案带原因 + 失败日志。"""
|
||||||
|
conn = auth.get_conn()
|
||||||
|
conn.execute("UPDATE users SET status=2, ban_reason='刷单' WHERE username='mailuser'")
|
||||||
|
conn.commit()
|
||||||
|
conn.close()
|
||||||
|
|
||||||
|
r = env.post('/auth/login', json={'username': 'mailuser', 'password': USER_PW})
|
||||||
|
assert r.json()['message'] == '账号已被封禁:刷单,请联系管理员'
|
||||||
|
conn = auth.get_conn()
|
||||||
|
assert ('fail', '封禁账号尝试登录: 刷单') in _logs(conn, '登录')
|
||||||
|
conn.close()
|
||||||
|
|
||||||
|
|
||||||
|
def test_disabled_account_message(env):
|
||||||
|
"""禁用账号(status=0):固定文案,无原因拼接。"""
|
||||||
|
conn = auth.get_conn()
|
||||||
|
conn.execute("UPDATE users SET status=0 WHERE username='mailuser'")
|
||||||
|
conn.commit()
|
||||||
|
conn.close()
|
||||||
|
|
||||||
|
r = env.post('/auth/login', json={'username': 'mailuser', 'password': USER_PW})
|
||||||
|
assert r.json()['message'] == '账号已被禁用,请联系管理员'
|
||||||
|
|
||||||
|
|
||||||
|
def test_admin_without_email_blocked_fail_closed(env):
|
||||||
|
"""管理员未绑定邮箱:fail-closed,不签发票据。"""
|
||||||
|
r = env.post('/auth/login', json={'username': 'bossuser', 'password': ADMIN_PW})
|
||||||
|
assert r.json()['message'] == '管理员账号未绑定邮箱,请联系管理员先完成绑定'
|
||||||
|
assert 'session' not in r.cookies
|
||||||
|
conn = auth.get_conn()
|
||||||
|
assert ('fail', '未绑定邮箱,两步验证不可用') in _logs(conn, '管理员登录')
|
||||||
|
conn.close()
|
||||||
|
|
||||||
|
|
||||||
|
def test_geo_ban_bumps_seq_again(env, monkeypatch):
|
||||||
|
"""异地多省封禁:login_seq 再递增一次(已签发中的会话全部顶下线)。"""
|
||||||
|
def _banned(user, ip):
|
||||||
|
return True, {'province': 'X'}
|
||||||
|
|
||||||
|
monkeypatch.setattr(auth, 'record_login_location', _banned)
|
||||||
|
r = env.post('/auth/login', json={'username': 'mailuser', 'password': USER_PW})
|
||||||
|
assert r.json()['message'] == '检测到账号在多个省份异常登录,已自动封禁,请联系管理员'
|
||||||
|
|
||||||
|
conn = auth.get_conn()
|
||||||
|
seq = conn.execute("SELECT login_seq FROM users WHERE username='mailuser'").fetchone()[0]
|
||||||
|
conn.close()
|
||||||
|
assert seq == 2 # 会话递增 + 封禁再递增
|
||||||
|
|
||||||
|
|
||||||
|
def test_unexpected_error_is_500(env, monkeypatch):
|
||||||
|
"""非预期异常:泛化文案 + HTTP 500。"""
|
||||||
|
def _boom(*a, **k):
|
||||||
|
raise RuntimeError('db busy')
|
||||||
|
|
||||||
|
monkeypatch.setattr(auth, 'verify_login', _boom)
|
||||||
|
r = env.post('/auth/login', json={'username': 'mailuser', 'password': USER_PW})
|
||||||
|
assert r.status_code == 500
|
||||||
|
assert r.json()['message'] == '登录失败,请稍后重试'
|
||||||
+97
-62
@@ -101,18 +101,99 @@ async def ann_seen(user: Annotated[SiteUser | None, Depends(current_user)]):
|
|||||||
return success()
|
return success()
|
||||||
|
|
||||||
|
|
||||||
@router.post('/auth/login')
|
async def _resolve_login_identity(username: str) -> tuple:
|
||||||
async def login(request: Request):
|
"""身份解析:含 @ 按归一化邮箱解析真实账号。
|
||||||
"""登录:密码校验(含邮箱登录)、失败锁定、管理员两步验证转票据。"""
|
|
||||||
|
返回 (邮箱命中的用户或 None, 锁定计数键)。邮箱登录的锁定键用
|
||||||
|
真实用户名(命中时)或归一化邮箱(未命中时),防大小写变体绕过。
|
||||||
|
"""
|
||||||
|
if '@' not in username:
|
||||||
|
return None, username
|
||||||
|
iuser = await asyncio.to_thread(auth.get_user_by_email, username)
|
||||||
|
lock_name = iuser['username'] if iuser else (auth._normalize_email(username) or username)
|
||||||
|
return iuser, lock_name
|
||||||
|
|
||||||
|
|
||||||
|
async def _allow_login_attempt(lock_name: str, ip: str) -> bool:
|
||||||
|
"""失败锁定闸:允许尝试返回 True;锁定时落一条一次性审计日志并返回 False。"""
|
||||||
|
from web.auth_state import (
|
||||||
|
_login_fail_check,
|
||||||
|
_login_fail_locked_log_once,
|
||||||
|
)
|
||||||
|
|
||||||
|
if _login_fail_check(lock_name):
|
||||||
|
return True
|
||||||
|
if _login_fail_locked_log_once(lock_name):
|
||||||
|
await asyncio.to_thread(auth.add_log, 'fail', lock_name, ip,
|
||||||
|
'登录', '失败次数过多,临时锁定')
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
async def _check_account_status(user, lock_name: str, ip: str):
|
||||||
|
"""账号状态闸:封禁(原因落日志)/禁用;通过返回 None。"""
|
||||||
|
keys = user.keys()
|
||||||
|
status = user['status'] if 'status' in keys else 1
|
||||||
|
if status == 2:
|
||||||
|
reason = user['ban_reason'] if 'ban_reason' in keys else ''
|
||||||
|
await asyncio.to_thread(auth.add_log, 'fail', lock_name, ip,
|
||||||
|
'登录', f'封禁账号尝试登录: {reason}')
|
||||||
|
return error('账号已被封禁' + (f':{reason}' if reason else '') + ',请联系管理员')
|
||||||
|
if status == 0:
|
||||||
|
return error('账号已被禁用,请联系管理员')
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
async def _admin_two_factor_response(user, lock_name: str, ip: str) -> JSONResponse:
|
||||||
|
"""管理员登录一律转两步验证通道(密码正确才签发票据;未绑定邮箱 fail-closed)。"""
|
||||||
from web.auth_state import (
|
from web.auth_state import (
|
||||||
_admin2fa_issue,
|
_admin2fa_issue,
|
||||||
_admin2fa_mask,
|
_admin2fa_mask,
|
||||||
_login_fail_check,
|
|
||||||
_login_fail_clear,
|
|
||||||
_login_fail_locked_log_once,
|
|
||||||
_login_fail_record,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
admin_email = user['email'] if 'email' in user.keys() else ''
|
||||||
|
if not admin_email:
|
||||||
|
await asyncio.to_thread(auth.add_log, 'fail', lock_name, ip,
|
||||||
|
'管理员登录', '未绑定邮箱,两步验证不可用')
|
||||||
|
return error('管理员账号未绑定邮箱,请联系管理员先完成绑定')
|
||||||
|
ticket = _admin2fa_issue(user)
|
||||||
|
await asyncio.to_thread(auth.add_log, 'success', lock_name, ip,
|
||||||
|
'管理员登录', '密码验证通过,待邮箱验证')
|
||||||
|
return success({'ticket': ticket, 'email_masked': _admin2fa_mask(admin_email),
|
||||||
|
'username': user['username'], 'admin_2fa': True},
|
||||||
|
'密码验证通过,请完成邮箱验证')
|
||||||
|
|
||||||
|
|
||||||
|
async def _user_session_response(user, ip: str) -> JSONResponse:
|
||||||
|
"""普通用户建会话:login_seq 原子递增顶号 → 异地多省封禁闸 → 签发 Cookie。"""
|
||||||
|
uid = int(user['id'])
|
||||||
|
now = int(time.time())
|
||||||
|
await db.execute(
|
||||||
|
'UPDATE users SET last_active = ?, last_login = ?, '
|
||||||
|
'login_seq = COALESCE(login_seq, 0) + 1 WHERE id = ?', (now, now, uid))
|
||||||
|
seq_row = await repo.get_user_by_id(uid)
|
||||||
|
seq = int((seq_row or {}).get('login_seq') or 1)
|
||||||
|
|
||||||
|
banned, _info = await asyncio.to_thread(auth.record_login_location, user, ip)
|
||||||
|
if banned:
|
||||||
|
await db.execute(
|
||||||
|
'UPDATE users SET login_seq = COALESCE(login_seq, 0) + 1 WHERE id = ?', (uid,))
|
||||||
|
return error('检测到账号在多个省份异常登录,已自动封禁,请联系管理员')
|
||||||
|
|
||||||
|
fresh = await repo.get_user_by_id(uid) or {}
|
||||||
|
return _login_response({
|
||||||
|
'user': _user_public(fresh),
|
||||||
|
'duration_text': _format_credits(int(fresh.get('duration_seconds') or 0)),
|
||||||
|
'redirect': 'admin' if fresh.get('role') == 'admin' else 'user',
|
||||||
|
}, '登录成功', uid, seq)
|
||||||
|
|
||||||
|
|
||||||
|
@router.post('/auth/login')
|
||||||
|
async def login(request: Request):
|
||||||
|
"""登录:密码校验(含邮箱登录)、失败锁定、管理员两步验证转票据。
|
||||||
|
|
||||||
|
分支顺序:超长密码直拒 → 身份解析 → 失败锁定 → 密码校验 →
|
||||||
|
状态闸(封禁/禁用)→ 管理员转 2FA → 普通用户建会话(含异地封禁)。
|
||||||
|
"""
|
||||||
data = await _request_data(request)
|
data = await _request_data(request)
|
||||||
username = str(data.get('username') or '').strip()
|
username = str(data.get('username') or '').strip()
|
||||||
password = str(data.get('password') or '')
|
password = str(data.get('password') or '')
|
||||||
@@ -123,73 +204,27 @@ async def login(request: Request):
|
|||||||
return error('用户名或密码错误')
|
return error('用户名或密码错误')
|
||||||
|
|
||||||
try:
|
try:
|
||||||
# 邮箱登录:含 @ 按归一化邮箱解析真实账号;锁定 key 用真实用户名
|
iuser, lock_name = await _resolve_login_identity(username)
|
||||||
if '@' in username:
|
if not await _allow_login_attempt(lock_name, ip):
|
||||||
iuser = await asyncio.to_thread(auth.get_user_by_email, username)
|
|
||||||
lock_name = iuser['username'] if iuser else (auth._normalize_email(username) or username)
|
|
||||||
else:
|
|
||||||
iuser = None
|
|
||||||
lock_name = username
|
|
||||||
|
|
||||||
if not _login_fail_check(lock_name):
|
|
||||||
if _login_fail_locked_log_once(lock_name):
|
|
||||||
await asyncio.to_thread(auth.add_log, 'fail', lock_name, ip,
|
|
||||||
'登录', '失败次数过多,临时锁定')
|
|
||||||
return error('尝试过于频繁,该账号已临时锁定,请 10 分钟后再试')
|
return error('尝试过于频繁,该账号已临时锁定,请 10 分钟后再试')
|
||||||
|
|
||||||
verify_name = iuser['username'] if iuser is not None else username
|
verify_name = iuser['username'] if iuser is not None else username
|
||||||
user = await asyncio.to_thread(auth.verify_login, verify_name, password)
|
user = await asyncio.to_thread(auth.verify_login, verify_name, password)
|
||||||
if not user:
|
if not user:
|
||||||
|
from web.auth_state import _login_fail_record
|
||||||
_login_fail_record(lock_name)
|
_login_fail_record(lock_name)
|
||||||
return error('用户名或密码错误')
|
return error('用户名或密码错误')
|
||||||
|
|
||||||
keys = user.keys()
|
status_err = await _check_account_status(user, lock_name, ip)
|
||||||
status = user['status'] if 'status' in keys else 1
|
if status_err is not None:
|
||||||
if status == 2:
|
return status_err
|
||||||
reason = user['ban_reason'] if 'ban_reason' in keys else ''
|
|
||||||
await asyncio.to_thread(auth.add_log, 'fail', lock_name, ip,
|
|
||||||
'登录', f'封禁账号尝试登录: {reason}')
|
|
||||||
return error('账号已被封禁' + (f':{reason}' if reason else '') + ',请联系管理员')
|
|
||||||
if status == 0:
|
|
||||||
return error('账号已被禁用,请联系管理员')
|
|
||||||
|
|
||||||
|
from web.auth_state import _login_fail_clear
|
||||||
_login_fail_clear(lock_name)
|
_login_fail_clear(lock_name)
|
||||||
|
|
||||||
if user['role'] == 'admin':
|
if user['role'] == 'admin':
|
||||||
# 管理员一律转两步验证通道(密码正确才签发票据;未绑定邮箱 fail-closed)
|
return await _admin_two_factor_response(user, lock_name, ip)
|
||||||
admin_email = user['email'] if 'email' in keys else ''
|
return await _user_session_response(user, ip)
|
||||||
if not admin_email:
|
|
||||||
await asyncio.to_thread(auth.add_log, 'fail', lock_name, ip,
|
|
||||||
'管理员登录', '未绑定邮箱,两步验证不可用')
|
|
||||||
return error('管理员账号未绑定邮箱,请联系管理员先完成绑定')
|
|
||||||
ticket = _admin2fa_issue(user)
|
|
||||||
await asyncio.to_thread(auth.add_log, 'success', lock_name, ip,
|
|
||||||
'管理员登录', '密码验证通过,待邮箱验证')
|
|
||||||
return success({'ticket': ticket, 'email_masked': _admin2fa_mask(admin_email),
|
|
||||||
'username': user['username'], 'admin_2fa': True},
|
|
||||||
'密码验证通过,请完成邮箱验证')
|
|
||||||
|
|
||||||
# 普通用户:建立会话(原子递增 login_seq 顶号)
|
|
||||||
uid = int(user['id'])
|
|
||||||
now = int(time.time())
|
|
||||||
await db.execute(
|
|
||||||
'UPDATE users SET last_active = ?, last_login = ?, '
|
|
||||||
'login_seq = COALESCE(login_seq, 0) + 1 WHERE id = ?', (now, now, uid))
|
|
||||||
seq_row = await repo.get_user_by_id(uid)
|
|
||||||
seq = int((seq_row or {}).get('login_seq') or 1)
|
|
||||||
|
|
||||||
banned, _info = await asyncio.to_thread(auth.record_login_location, user, ip)
|
|
||||||
if banned:
|
|
||||||
await db.execute(
|
|
||||||
'UPDATE users SET login_seq = COALESCE(login_seq, 0) + 1 WHERE id = ?', (uid,))
|
|
||||||
return error('检测到账号在多个省份异常登录,已自动封禁,请联系管理员')
|
|
||||||
|
|
||||||
fresh = await repo.get_user_by_id(uid) or {}
|
|
||||||
return _login_response({
|
|
||||||
'user': _user_public(fresh),
|
|
||||||
'duration_text': _format_credits(int(fresh.get('duration_seconds') or 0)),
|
|
||||||
'redirect': 'admin' if fresh.get('role') == 'admin' else 'user',
|
|
||||||
}, '登录成功', uid, seq)
|
|
||||||
except Exception:
|
except Exception:
|
||||||
return error('登录失败,请稍后重试', 500)
|
return error('登录失败,请稍后重试', 500)
|
||||||
|
|
||||||
|
|||||||
Reference in new issue
Block a user