fix: 启用会话过期(服务端校验签发时间,堵住旧 Cookie 永不过期)

为什么:签发侧 Cookie 的 max_age 是 7 天,但那只是**浏览器**行为——被保存、
复制或从日志里捞出来的旧 Cookie 仍带着有效签名,而解析侧 loads 从不校验签发
时间(max_age=None),等于「登录态永不过期」。登出/顶号靠递增 login_seq 兜底,
但拿不到库的一方(如泄露的 Cookie)无从失效。

改法:
- core/session.py 新增 COOKIE_NAME / SESSION_MAX_AGE,作为会话名与有效期的
  唯一来源。此前 'session' 在 web/deps、web/routers/auth、web/routers/register
  各写一遍,7 天在两处各写一遍——三处同名常量各改各的,正是漂移温床。
- web/deps._parse_cookie 传入 max_age=SESSION_MAX_AGE,过期判定落在服务端。
- 两个签发侧改引用共享常量(保留模块级别名,对外名字不变)。
- loads() 的签名与语义**不变**(max_age=None 仍不校验):它是通用工具,策略
  属于调用点,这样 test_session 里「None 接受旧时间戳」的语义继续成立。

行为变更(本轮唯一一处):签发时间超过 7 天的会话 Cookie 现在按未登录处理。
正常浏览器用户无感——浏览器本来就按 max_age 丢弃它;受影响的是手工保存、
跨 7 天复用的旧 Cookie,正是要堵的那类。

验证:
- 新增 tests/test_deps.py::test_expired_cookie_rejected:用 monkeypatch.context()
  把时钟拨回 SESSION_MAX_AGE 之前签发(itsdangerous 时间戳取自 dumps 时刻),
  再以当前时间解析 → 必须拒绝;同时断言有效期内不受影响。用 context 而非
  undo(),后者会连带撤销 site fixture 的设置。
- 变异验证:把 max_age 参数摘掉后**恰好**该用例变红、其余 10 例全绿——证明测试
  精确盯着这一处,不是碰巧通过。
- ruff 全过 · pyright 0/0 · pytest 214 passed(213→214)。golden 快照与
  test_download_billing(真实登录拿 Cookie 打 API)均通过,端到端链路未受影响。
This commit is contained in:
snowy-yang committed 2026-09-27 07:32:05 +08:00
1 parent 5c8f941d19
commit 1973d6a0e8
5 files changed
+40 -10

No files matched your search

+10 -4
View File
@@ -22,6 +22,11 @@ _SIGNER_KWARGS: dict[str, Any] = {'key_derivation': 'hmac', 'digest_method': has
SESSION_KEY = 'site_uid'
# 会话 Cookie 名与有效期(秒):签发侧(登录写 Cookie)与解析侧(校验签发时间)
# 共用这一处定义,避免两处各写一个 7 天、日后慢慢漂移。
COOKIE_NAME = 'session'
SESSION_MAX_AGE = 7 * 86400
@lru_cache(maxsize=8)
def _serializer(secret: str) -> URLSafeTimedSerializer:
@@ -41,10 +46,11 @@ def dumps(secret: str, session: dict[str, Any]) -> str:
def loads(secret: str, value: str, max_age: int | None = None) -> dict[str, Any] | None:
"""校验签名并解出会话字典;无效 / 被篡改 / 超龄返回 None。
``max_age=None``(默认)不校验签发时间——与本站既有行为一致:会话有效期由
Cookie 自身的 max_age(登录时 7 天,见 web.routers.register._SESSION_MAX_AGE)
与库中 login_seq 顶号校验共同约束,服务端不叠加第二层独立过期判定;
传入秒数即启用 itsdangerous 的签名时间戳校验。
本函数是通用工具:``max_age=None``(默认)不校验签发时间。**生产请求路径
由 web.deps._parse_cookie 传入 SESSION_MAX_AGE 启用过期**——只靠 Cookie 的
max_age 是不够的:那只是浏览器行为,被保存、复制或从日志里捞出来的旧 Cookie
依然能通过签名校验,服务端不判就等于登录态永不过期。传入秒数即启用
itsdangerous 的签名时间戳校验。
"""
if not value:
return None
+20
View File
@@ -72,6 +72,26 @@ async def test_forged_cookie_rejected(site):
assert await current_user('not-a-valid-cookie') is None
async def test_expired_cookie_rejected(site, monkeypatch):
"""超龄会话 Cookie 由**服务端**拒绝(不只是浏览器层面的 max_age)。
Cookie 的 max_age 只是浏览器行为:被保存/复制的旧 Cookie 仍带着有效签名,
服务端不校验签发时间就等于登录态永不过期。
构造方式:用 monkeypatch.context() 把时钟拨回 SESSION_MAX_AGE 之前签发
(itsdangerous 的签名时间戳取自 dumps 时刻),再以当前时间解析。
必须用 context 而非 monkeypatch.undo()——后者会连带撤销 site fixture 的设置。
"""
past = time.time() - cs.SESSION_MAX_AGE - 60
with monkeypatch.context() as m:
m.setattr(time, 'time', lambda: past)
stale = _cookie(26, 5)
assert await current_user(stale) is None, '超龄会话必须被服务端拒绝'
fresh = await current_user(_cookie(26, 5))
assert fresh is not None and fresh.id == 26, '有效期内不受影响'
async def test_require_login_raises(site):
with pytest.raises(AppError) as ei:
await require_login(None)
+6 -2
View File
@@ -3,6 +3,8 @@
与 auth_system 的语义对齐(共用同一 Cookie、同一站点库):
- 会话 Cookie 经 core.session 解码(Flask 兼容签名),密钥同源
- 过期:签发时间超过 core.session.SESSION_MAX_AGE 视为失效(服务端把关,
不依赖浏览器是否还留着 Cookie)
- 封禁账号(status 0/2):按未登录处理
- 顶号校验:会话内的 `site_uid_seq` 必须与库中 `login_seq` 一致,否则视为失效
- 剩余次数/账本预扣属付费流程,由 web.billing 承担(此处只做身份解析)
@@ -16,7 +18,7 @@ from core import repo
from core import session as sess
from web.errors import AppError
_SESSION_COOKIE = 'session'
_SESSION_COOKIE = sess.COOKIE_NAME
def secret_key() -> str:
@@ -42,7 +44,9 @@ class SiteUser:
def _parse_cookie(session_cookie: str | None) -> dict[str, Any] | None:
return sess.loads(secret_key(), session_cookie or '')
# 服务端叠加签发时间校验(SESSION_MAX_AGE):Cookie 自身的 max_age 只约束
# 浏览器,被存下来的旧 Cookie 仍能通过签名校验,故过期必须由服务端把关。
return sess.loads(secret_key(), session_cookie or '', max_age=sess.SESSION_MAX_AGE)
async def current_user(
+2 -2
View File
@@ -37,8 +37,8 @@ from web.serializers import user_public as _user_public
router = APIRouter()
SESSION_COOKIE = 'session'
_SESSION_MAX_AGE = 7 * 86400
SESSION_COOKIE = cs.COOKIE_NAME
_SESSION_MAX_AGE = cs.SESSION_MAX_AGE
+2 -2
View File
@@ -37,8 +37,8 @@ from web.serializers import user_public as _user_public
router = APIRouter()
SESSION_COOKIE = 'session'
_SESSION_MAX_AGE = 7 * 86400
SESSION_COOKIE = cs.COOKIE_NAME
_SESSION_MAX_AGE = cs.SESSION_MAX_AGE