diff --git a/_build/_cmsg14.txt b/_build/_cmsg14.txt new file mode 100644 index 0000000..25b7591 --- /dev/null +++ b/_build/_cmsg14.txt @@ -0,0 +1,57 @@ +修「去密码硬编码」那一步自己引入的三个 bug + +上一版把明文密码换成随机生成时，引入了三个问题。都是我自己的补丁 +造成的，而且症状都像"功能坏了"，实际是环境和脚本的问题。 + +== bug 1：兜底替换把函数调用塞进了字符串 == + +fix_hardcoded_password.py 里有句兜底： + text = text.replace(OLD, '_admin_password()') + +而原代码是： + {"username": "xmz7891", "password": "@<老密码>"} + ^ @ 在引号**里面** + +正则要求引号紧贴老密码，这里中间有个 @，所以没匹配上； +兜底替换就把老密码换成了函数调用，结果成了**字符串字面量**： + + {"username": "xmz7891", "password": "@_admin_password()"} + +7 个文件被这样弄坏。它们拿这串东西当密码去登录， +当然登不上 -> 401「请先登录」-> 管理接口测试全红（12 项失败）。 + +教训：**批量替换里的兜底 str.replace 太危险**，它不看上下文。 +替换完必须检查有没有"函数调用被引号包起来"这种形状。 + +== bug 2：测试和它自己起的中转对不上密码 == + +密码改成随机生成之后，测试分成两边： + * 测试进程：_admin_password() 去环境变量/文件里找 + * 它启动的中转：找不到就随机生成一个 +两边不一致 -> 登录失败。 + +修法：测试在**模块级**（import 时）生成一个本次运行专用的随机密码， +写进 os.environ["RELAY_ADMIN_PASSWORD"]。测试启动中转用的是 +dict(os.environ)，所以会自动传过去 —— 两边就是同一个密码。 + +用随机而不是写死：写死就又变成"密码进仓库"了，白改一场。 + +== bug 3：全新部署启动后没有密码文件 == + +admin_password() 原来只在有人访问登录页时才调用，于是全新部署 +启动后 data/admin_password.txt 一直不存在，日志里也看不到密码， +管理员不知道该用什么登。 + +修法：main() 里在 admin_key() 后面补一句 admin_password()， +和钥匙一样的道理 —— 启动时生成并打印一次。 + +== 验证 == + + 管理页设备控制 23/23 通过 + 管理页批删 24/24 通过 + 管理页按钮 10/10 通过 + 管理页 JS 通过 + 管理页网络体检 15/15 通过 + 密码随机生成 11/11 通过（含"老密码登不进去"、"两次部署密码不同"） + +工作区搜不到明文密码了（git 历史里还有，等用户确认后重写）。 diff --git a/_build/chk_ab_now.sh b/_build/chk_ab_now.sh new file mode 100644 index 0000000..8ed4cae --- /dev/null +++ b/_build/chk_ab_now.sh @@ -0,0 +1,32 @@ +#!/bin/bash +# A/B 对照现状（我换密码重启中转打断了 A 组） +set -u +echo " ══ $(date '+%H:%M:%S') ══" +echo +echo " ── A 组 ──" +[ -f /root/ab_A.out ] && cat /root/ab_A.out | sed 's/^/ /' +n=0; [ -f /root/ab_A_health.tsv ] && n=$(wc -l < /root/ab_A_health.tsv) +echo " 样本: $n" +[ -f /root/ab_A_summary.txt ] && echo " ── 汇总 ──" && cat /root/ab_A_summary.txt | sed 's/^/ /' +[ -f /root/ab_A_health.tsv ] && echo " 最后 2 条:" && tail -2 /root/ab_A_health.tsv | sed 's/^/ /' +echo +echo " ── B 组 ──" +[ -f /root/ab_B.out ] && cat /root/ab_B.out | sed 's/^/ /' || echo " （没开始）" +n2=0; [ -f /root/ab_B_health.tsv ] && n2=$(wc -l < /root/ab_B_health.tsv) +echo " 样本: $n2" +echo +echo " ── 进程 ──" +pgrep -af 'rpab|rp_ab' 2>/dev/null | head -3 | sed 's/^/ /' || echo " （没有在跑）" +echo +echo " ── 记忆/CPU 采样（它加的那个）──" +[ -f /root/ab_A_res.tsv ] && head -3 /root/ab_A_res.tsv | sed 's/^/ /' +echo +echo " ── 当前配比 ──" +for s in lanremote-tunnel lanremote-tunnel-h2; do + printf " %-24s %s\n" "$s" \ + "$(grep -oE -- '--ha-connections [0-9]+' "/etc/systemd/system/$s.service" 2>/dev/null | tr '\n' ' ')" +done +echo +echo " ── 公网 ──" +curl -s --max-time 15 -o /dev/null -w ' HTTP %{http_code} %{time_total}s\n' \ + -H 'User-Agent: LanRemote/1.8.2' https://relay.xxmxxs.top/v1/health diff --git a/_build/fix_broken_password_patch.py b/_build/fix_broken_password_patch.py new file mode 100644 index 0000000..b3cb9f5 --- /dev/null +++ b/_build/fix_broken_password_patch.py @@ -0,0 +1,140 @@ +# -*- coding: utf-8 -*- +r'''修两件事： + +== 1. 我自己的补丁把函数调用塞进了字符串里 == + +fix_hardcoded_password.py 里有个兜底替换： + text = text.replace(OLD, '_admin_password()') + +而原代码是： + {"username": "xmz7891", "password": "@<老密码>"} + ^ @ 在引号**里面** + +正则 `"password"\s*:\s*"OLD"` 要求引号紧贴 OLD，这里中间有个 @， +所以没匹配上；然后兜底替换就把 OLD 换成了函数调用， +结果变成**字符串字面量**： + + {"username": "xmz7891", "password": "@_admin_password()"} + +7 个文件被这样弄坏。它们登录时拿这个当密码， +当然登不上 -> 401「请先登录」-> 管理接口测试全红。 + +教训：**做批量替换时，兜底的 `str.replace` 太危险** —— +它不管上下文。要么用带边界正则，要么替换后检查有没有 +"函数调用被引号包起来" 这种形状。 + +== 2. relay 启动时没生成密码文件 == + +admin_password() 只在有人访问登录时才调用，所以全新部署启动后 +data/admin_password.txt 一直不存在 —— 日志里也就看不到密码， +管理员不知道用什么登。 + +main() 里已经调了 admin_key()（同样道理），补上 admin_password()。 +''' +from __future__ import annotations + +import re +import sys +from pathlib import Path + +ROOT = Path(__file__).resolve().parent.parent + +# 被弄坏的形状：引号把函数调用包起来了 +BROKEN = re.compile(r"""["']@?_admin_password\(\)["']""") +GOOD = "_admin_password()" + +TARGETS = ["_tests/acceptance.py", + "_tests/test_admin_batch_delete.py", + "_tests/test_admin_button.py", + "_tests/test_admin_device_control.py", + "_tests/test_admin_js.py", + "_tests/test_admin_netcheck.py", + "_tests/test_relay_admin_login.py"] + + +def main() -> int: + changed = 0 + failed = 0 + + print("── 1. 修被引号包起来的 _admin_password() ──") + for rel in TARGETS: + p = ROOT / rel + if not p.is_file(): + print(" ✗ %s 不存在" % rel) + failed += 1 + continue + text = p.read_text(encoding="utf-8") + if '@_admin_password()' not in text and '"_admin_password()"' not in text: + print(" • %s 本来就正常" % rel) + continue + before = text + # 只处理「引号里整个就是一个函数调用」这种 + text = re.sub(r'"@_admin_password\(\)"', GOOD, text) + text = re.sub(r"'@_admin_password\(\)'", GOOD, text) + text = re.sub(r'"_admin_password\(\)"', GOOD, text) + text = re.sub(r"'_admin_password\(\)'", GOOD, text) + if text == before: + print(" ✗ %s 替换没生效" % rel) + failed += 1 + continue + # 再确认没有残留 + if '@_admin_password()' in text: + print(" ✗ %s 还有残留" % rel) + failed += 1 + continue + p.write_text(text, encoding="utf-8", newline="\n") + print(" ✓ %s" % rel) + changed += 1 + + print() + print("── 2. relay.py 启动时生成密码文件 ──") + rp = ROOT / "relay" / "relay.py" + text = rp.read_text(encoding="utf-8") + if "admin_password() # 确保" in text or "pw = admin_password()" in text: + print(" • 已经调过了") + else: + # 在 main() 里 admin_key() 那行后面补一句 + old = " key = admin_key()" + if old not in text: + print(" ✗ 找不到 main() 里 admin_key() 那行") + failed += 1 + else: + new = ( + " key = admin_key()\n" + " # 同样要**在这里**把密码也生成出来：\n" + " # admin_password() 原来只在有人访问登录页时才调用，\n" + " # 于是全新部署启动后 data/admin_password.txt 一直不存在，\n" + " # 日志里也看不到密码 —— 管理员不知道该用什么登录。\n" + " # 启动时就调一次，密码会随启动日志打印出来。\n" + " admin_password()" + ) + text = text.replace(old, new, 1) + rp.write_text(text, encoding="utf-8", newline="\n") + print(" ✓ 已在 main() 里调用 admin_password()") + changed += 1 + + print() + print(" 改了 %d 处，失败 %d 处" % (changed, failed)) + + # 全局再扫一遍有没有「函数调用被引号包起来」的 + print() + print("── 3. 全仓复查 ──") + bad = [] + for pat in ("_tests/*.py", "_build/*.py", "relay/*.py"): + for f in ROOT.glob(pat): + try: + t = f.read_text(encoding="utf-8", errors="replace") + except Exception: # noqa: BLE001 + continue + for m in BROKEN.finditer(t): + bad.append("%s" % f.relative_to(ROOT)) + break + if bad: + print(" ⚠ 还有：%s" % bad) + return 1 + print(" ✓ 没有了") + return 0 if failed == 0 else 1 + + +if __name__ == "__main__": + sys.exit(main()) diff --git a/_build/fix_test_admin_pw.py b/_build/fix_test_admin_pw.py new file mode 100644 index 0000000..28d4c99 --- /dev/null +++ b/_build/fix_test_admin_pw.py @@ -0,0 +1,120 @@ +# -*- coding: utf-8 -*- +r'''让测试知道它自己起的中转用了什么密码。 + +== 问题 == + +密码不再硬编码之后，测试分成两边： + * 测试进程：`_admin_password()` 去环境变量/文件里找 + * 它启动的中转：`admin_password()` 找不到就**随机生成**一个 + +两边对不上 -> 登录失败 -> 所有管理接口返回 401「请先登录」。 +（表现成 12 项失败，看起来像功能坏了，实际是测试环境没打通。） + +== 修法 == + +在测试文件**模块级**（import 时）做两件事： + 1. 随机生成一个本次运行专用的密码 + 2. 写进 os.environ["RELAY_ADMIN_PASSWORD"] + +因为测试启动中转时用的是 `env = dict(os.environ)`， +这个变量会**自动传给它** —— 两边就是同一个密码了。 + +为什么用随机而不是写死： + 写死就又变成"密码进仓库"了，白改一场。 + 随机的每次跑都不一样，而且不落任何文件。 +''' +from __future__ import annotations + +import re +import sys +from pathlib import Path + +ROOT = Path(__file__).resolve().parent.parent + +TARGETS = ["_tests/acceptance.py", + "_tests/test_admin_batch_delete.py", + "_tests/test_admin_button.py", + "_tests/test_admin_device_control.py", + "_tests/test_admin_js.py", + "_tests/test_admin_netcheck.py", + "_tests/test_relay_admin_login.py"] + +MARK = "# ---- 测试专用管理密码" + +BLOCK = ''' + +# ---- 测试专用管理密码 ---------------------------------------------------- +# 测试要登录它自己启动的中转，但密码是那边随机的 —— 两边对不上就会 +# 全部 401（表现成"功能坏了"，实际是环境没打通）。 +# +# 这里生成一个**本次运行专用**的随机密码，写进环境变量； +# 测试启动中转用的是 dict(os.environ)，所以会自动传给中转。 +# +# 为什么不用写死的：写死就又变成"密码进仓库"了，白改一场。 +import os as _os_for_pw +import secrets as _secrets_for_pw +TEST_ADMIN_PW = "t" + _secrets_for_pw.token_hex(12) +_os_for_pw.environ["RELAY_ADMIN_PASSWORD"] = TEST_ADMIN_PW +_os_for_pw.environ.setdefault("RELAY_ADMIN_USER", "xmz7891") +# --------------------------------------------------------------------------- +''' + + +def main() -> int: + ok = 0 + bad = 0 + for rel in TARGETS: + p = ROOT / rel + if not p.is_file(): + print(" ✗ %s 不存在" % rel) + bad += 1 + continue + text = p.read_text(encoding="utf-8") + if MARK in text: + print(" • %s 已经打过补丁" % rel) + ok += 1 + continue + + # 插在 _admin_password 的定义**之前**（模块级，import 时就生效） + m = re.search(r"^def _admin_password\(\)", text, re.M) + if m: + text = text[:m.start()] + BLOCK.strip() + "\n\n\n" + text[m.start():] + else: + # 没有那个 helper 的话，插在最后一个 import 之后 + last_imp = 0 + for mm in re.finditer(r"^(?:import |from )\S.*$", text, re.M): + last_imp = mm.end() + if not last_imp: + print(" ✗ %s 找不到插入点" % rel) + bad += 1 + continue + text = text[:last_imp] + "\n" + BLOCK + text[last_imp:] + + p.write_text(text, encoding="utf-8", newline="\n") + print(" ✓ %s" % rel) + ok += 1 + + print() + print(" 改了 %d 个，失败 %d 个" % (ok, bad)) + + # 语法检查 + print() + import ast + errs = [] + for rel in TARGETS: + p = ROOT / rel + try: + ast.parse(p.read_text(encoding="utf-8")) + except SyntaxError as exc: + errs.append("%s: %s" % (rel, exc)) + if errs: + print(" ✗ 语法错：") + for e in errs: + print(" %s" % e) + return 1 + print(" ✓ 语法都通过") + return 0 if bad == 0 else 1 + + +if __name__ == "__main__": + sys.exit(main()) diff --git a/_tests/acceptance.py b/_tests/acceptance.py index b07229a..6503729 100644 --- a/_tests/acceptance.py +++ b/_tests/acceptance.py @@ -41,6 +41,22 @@ RESULTS = [] # 3. 本机 8080 中转实际用的 data/admin_password.txt # 拿不到就跳过需要登录的用例，而不是编一个假密码硬试。 # --------------------------------------------------------------------------- +# ---- 测试专用管理密码 ---------------------------------------------------- +# 测试要登录它自己启动的中转，但密码是那边随机的 —— 两边对不上就会 +# 全部 401（表现成"功能坏了"，实际是环境没打通）。 +# +# 这里生成一个**本次运行专用**的随机密码，写进环境变量； +# 测试启动中转用的是 dict(os.environ)，所以会自动传给中转。 +# +# 为什么不用写死的：写死就又变成"密码进仓库"了，白改一场。 +import os as _os_for_pw +import secrets as _secrets_for_pw +TEST_ADMIN_PW = "t" + _secrets_for_pw.token_hex(12) +_os_for_pw.environ["RELAY_ADMIN_PASSWORD"] = TEST_ADMIN_PW +_os_for_pw.environ.setdefault("RELAY_ADMIN_USER", "xmz7891") +# --------------------------------------------------------------------------- + + def _admin_password() -> str: import json as _json import os as _os @@ -182,7 +198,7 @@ def main() -> int: c_login == 401 and c_url == 200, f"管理接口未登录={c_login}（应为 401）；" f"/url={c_url} 返回 {url_body.strip()[:60]}；" - f"管理密码={plain or '@_admin_password()'}") + f"管理密码={plain or _admin_password()}") else: item("管理页需要登录 + 管理接口", False, "没有公网地址，无法测") diff --git a/_tests/test_admin_batch_delete.py b/_tests/test_admin_batch_delete.py index d85c5f3..0f4e1b5 100644 --- a/_tests/test_admin_batch_delete.py +++ b/_tests/test_admin_batch_delete.py @@ -48,6 +48,22 @@ failed = 0 # 3. 本机 8080 中转实际用的 data/admin_password.txt # 拿不到就跳过需要登录的用例，而不是编一个假密码硬试。 # --------------------------------------------------------------------------- +# ---- 测试专用管理密码 ---------------------------------------------------- +# 测试要登录它自己启动的中转，但密码是那边随机的 —— 两边对不上就会 +# 全部 401（表现成"功能坏了"，实际是环境没打通）。 +# +# 这里生成一个**本次运行专用**的随机密码，写进环境变量； +# 测试启动中转用的是 dict(os.environ)，所以会自动传给中转。 +# +# 为什么不用写死的：写死就又变成"密码进仓库"了，白改一场。 +import os as _os_for_pw +import secrets as _secrets_for_pw +TEST_ADMIN_PW = "t" + _secrets_for_pw.token_hex(12) +_os_for_pw.environ["RELAY_ADMIN_PASSWORD"] = TEST_ADMIN_PW +_os_for_pw.environ.setdefault("RELAY_ADMIN_USER", "xmz7891") +# --------------------------------------------------------------------------- + + def _admin_password() -> str: import json as _json import os as _os @@ -151,7 +167,7 @@ def main() -> int: jar = http.cookiejar.CookieJar() opener = urllib.request.build_opener( urllib.request.HTTPCookieProcessor(jar)) - form = urllib.parse.urlencode({"username": "xmz7891", "password": "@_admin_password()"}).encode() + form = urllib.parse.urlencode({"username": "xmz7891", "password": _admin_password()}).encode() req = urllib.request.Request(BASE + "/admin/login", data=form, method="POST", headers=_UA) req.add_header("Content-Type", "application/x-www-form-urlencoded") diff --git a/_tests/test_admin_button.py b/_tests/test_admin_button.py index d81ff2b..ba044d4 100644 --- a/_tests/test_admin_button.py +++ b/_tests/test_admin_button.py @@ -50,6 +50,22 @@ failed = 0 # 3. 本机 8080 中转实际用的 data/admin_password.txt # 拿不到就跳过需要登录的用例，而不是编一个假密码硬试。 # --------------------------------------------------------------------------- +# ---- 测试专用管理密码 ---------------------------------------------------- +# 测试要登录它自己启动的中转，但密码是那边随机的 —— 两边对不上就会 +# 全部 401（表现成"功能坏了"，实际是环境没打通）。 +# +# 这里生成一个**本次运行专用**的随机密码，写进环境变量； +# 测试启动中转用的是 dict(os.environ)，所以会自动传给中转。 +# +# 为什么不用写死的：写死就又变成"密码进仓库"了，白改一场。 +import os as _os_for_pw +import secrets as _secrets_for_pw +TEST_ADMIN_PW = "t" + _secrets_for_pw.token_hex(12) +_os_for_pw.environ["RELAY_ADMIN_PASSWORD"] = TEST_ADMIN_PW +_os_for_pw.environ.setdefault("RELAY_ADMIN_USER", "xmz7891") +# --------------------------------------------------------------------------- + + def _admin_password() -> str: import json as _json import os as _os @@ -137,7 +153,7 @@ def main() -> int: print() print("一、登录拿 Cookie") - form = urllib.parse.urlencode({"username": "xmz7891", "password": "@_admin_password()"}).encode() + form = urllib.parse.urlencode({"username": "xmz7891", "password": _admin_password()}).encode() req = urllib.request.Request(BASE + "/admin/login", data=form, method="POST", headers=_UA) req.add_header("Content-Type", "application/x-www-form-urlencoded") diff --git a/_tests/test_admin_device_control.py b/_tests/test_admin_device_control.py index c644b35..59adbaa 100644 --- a/_tests/test_admin_device_control.py +++ b/_tests/test_admin_device_control.py @@ -68,6 +68,22 @@ failed = 0 # 3. 本机 8080 中转实际用的 data/admin_password.txt # 拿不到就跳过需要登录的用例，而不是编一个假密码硬试。 # --------------------------------------------------------------------------- +# ---- 测试专用管理密码 ---------------------------------------------------- +# 测试要登录它自己启动的中转，但密码是那边随机的 —— 两边对不上就会 +# 全部 401（表现成"功能坏了"，实际是环境没打通）。 +# +# 这里生成一个**本次运行专用**的随机密码，写进环境变量； +# 测试启动中转用的是 dict(os.environ)，所以会自动传给中转。 +# +# 为什么不用写死的：写死就又变成"密码进仓库"了，白改一场。 +import os as _os_for_pw +import secrets as _secrets_for_pw +TEST_ADMIN_PW = "t" + _secrets_for_pw.token_hex(12) +_os_for_pw.environ["RELAY_ADMIN_PASSWORD"] = TEST_ADMIN_PW +_os_for_pw.environ.setdefault("RELAY_ADMIN_USER", "xmz7891") +# --------------------------------------------------------------------------- + + def _admin_password() -> str: import json as _json import os as _os @@ -140,7 +156,7 @@ def main() -> int: opener = urllib.request.build_opener( urllib.request.HTTPCookieProcessor(jar)) form = urllib.parse.urlencode( - {"username": "xmz7891", "password": "@_admin_password()"}).encode() + {"username": "xmz7891", "password": _admin_password()}).encode() req = urllib.request.Request(BASE + "/admin/login", data=form, method="POST", headers=_UA) req.add_header("Content-Type", "application/x-www-form-urlencoded") diff --git a/_tests/test_admin_js.py b/_tests/test_admin_js.py index b1a1188..82e127b 100644 --- a/_tests/test_admin_js.py +++ b/_tests/test_admin_js.py @@ -40,6 +40,22 @@ BASE = f"http://127.0.0.1:{PORT}" # 3. 本机 8080 中转实际用的 data/admin_password.txt # 拿不到就跳过需要登录的用例，而不是编一个假密码硬试。 # --------------------------------------------------------------------------- +# ---- 测试专用管理密码 ---------------------------------------------------- +# 测试要登录它自己启动的中转，但密码是那边随机的 —— 两边对不上就会 +# 全部 401（表现成"功能坏了"，实际是环境没打通）。 +# +# 这里生成一个**本次运行专用**的随机密码，写进环境变量； +# 测试启动中转用的是 dict(os.environ)，所以会自动传给中转。 +# +# 为什么不用写死的：写死就又变成"密码进仓库"了，白改一场。 +import os as _os_for_pw +import secrets as _secrets_for_pw +TEST_ADMIN_PW = "t" + _secrets_for_pw.token_hex(12) +_os_for_pw.environ["RELAY_ADMIN_PASSWORD"] = TEST_ADMIN_PW +_os_for_pw.environ.setdefault("RELAY_ADMIN_USER", "xmz7891") +# --------------------------------------------------------------------------- + + def _admin_password() -> str: import json as _json import os as _os @@ -158,7 +174,7 @@ def main() -> int: jar = http.cookiejar.CookieJar() opener = urllib.request.build_opener( urllib.request.HTTPCookieProcessor(jar)) - form = urllib.parse.urlencode({"username": "xmz7891", "password": "@_admin_password()"}).encode() + form = urllib.parse.urlencode({"username": "xmz7891", "password": _admin_password()}).encode() req = urllib.request.Request(BASE + "/admin/login", data=form, method="POST", headers=UA) req.add_header("Content-Type", "application/x-www-form-urlencoded") diff --git a/_tests/test_admin_netcheck.py b/_tests/test_admin_netcheck.py index 721247c..4ee37e3 100644 --- a/_tests/test_admin_netcheck.py +++ b/_tests/test_admin_netcheck.py @@ -38,6 +38,22 @@ failed = 0 # 3. 本机 8080 中转实际用的 data/admin_password.txt # 拿不到就跳过需要登录的用例，而不是编一个假密码硬试。 # --------------------------------------------------------------------------- +# ---- 测试专用管理密码 ---------------------------------------------------- +# 测试要登录它自己启动的中转，但密码是那边随机的 —— 两边对不上就会 +# 全部 401（表现成"功能坏了"，实际是环境没打通）。 +# +# 这里生成一个**本次运行专用**的随机密码，写进环境变量； +# 测试启动中转用的是 dict(os.environ)，所以会自动传给中转。 +# +# 为什么不用写死的：写死就又变成"密码进仓库"了，白改一场。 +import os as _os_for_pw +import secrets as _secrets_for_pw +TEST_ADMIN_PW = "t" + _secrets_for_pw.token_hex(12) +_os_for_pw.environ["RELAY_ADMIN_PASSWORD"] = TEST_ADMIN_PW +_os_for_pw.environ.setdefault("RELAY_ADMIN_USER", "xmz7891") +# --------------------------------------------------------------------------- + + def _admin_password() -> str: import json as _json import os as _os @@ -147,7 +163,7 @@ def main() -> int: jar = http.cookiejar.CookieJar() opener = urllib.request.build_opener( urllib.request.HTTPCookieProcessor(jar)) - form = urllib.parse.urlencode({"username": "xmz7891", "password": "@_admin_password()"}).encode() + form = urllib.parse.urlencode({"username": "xmz7891", "password": _admin_password()}).encode() req = urllib.request.Request(BASE + "/admin/login", data=form, method="POST", headers=UA) req.add_header("Content-Type", "application/x-www-form-urlencoded") diff --git a/_tests/test_relay_admin_login.py b/_tests/test_relay_admin_login.py index 8b249ba..94e3d42 100644 --- a/_tests/test_relay_admin_login.py +++ b/_tests/test_relay_admin_login.py @@ -26,7 +26,7 @@ sys.path.insert(0, str(ROOT / "_build")) info = json.loads((ROOT / "_build" / "relay_info.json").read_text( encoding="utf-8")) URL = info["url"] -PASSWORD = "@_admin_password()" +PASSWORD = _admin_password() passed = 0 failed = 0 @@ -45,6 +45,22 @@ failed = 0 # 3. 本机 8080 中转实际用的 data/admin_password.txt # 拿不到就跳过需要登录的用例，而不是编一个假密码硬试。 # --------------------------------------------------------------------------- +# ---- 测试专用管理密码 ---------------------------------------------------- +# 测试要登录它自己启动的中转，但密码是那边随机的 —— 两边对不上就会 +# 全部 401（表现成"功能坏了"，实际是环境没打通）。 +# +# 这里生成一个**本次运行专用**的随机密码，写进环境变量； +# 测试启动中转用的是 dict(os.environ)，所以会自动传给中转。 +# +# 为什么不用写死的：写死就又变成"密码进仓库"了，白改一场。 +import os as _os_for_pw +import secrets as _secrets_for_pw +TEST_ADMIN_PW = "t" + _secrets_for_pw.token_hex(12) +_os_for_pw.environ["RELAY_ADMIN_PASSWORD"] = TEST_ADMIN_PW +_os_for_pw.environ.setdefault("RELAY_ADMIN_USER", "xmz7891") +# --------------------------------------------------------------------------- + + def _admin_password() -> str: import json as _json import os as _os diff --git a/relay/relay.py b/relay/relay.py index b3e0f98..47f93a8 100644 --- a/relay/relay.py +++ b/relay/relay.py @@ -1929,6 +1929,12 @@ def main() -> int: log(f"长轮询 {POLL_SECONDS:.0f} 秒 · 队列上限 " f"{MAX_QUEUE_BYTES // 1024} KB · 数据目录 {DATA_DIR}") key = admin_key() + # 同样要**在这里**把密码也生成出来： + # admin_password() 原来只在有人访问登录页时才调用， + # 于是全新部署启动后 data/admin_password.txt 一直不存在， + # 日志里也看不到密码 —— 管理员不知道该用什么登录。 + # 启动时就调一次，密码会随启动日志打印出来。 + admin_password() log(f"管理页：http://<本机>:{PORT}/?key={key}") log(f"（钥匙也在 {DATA_DIR / 'admin_key.txt'}）")