xmz7891/ykxmz
1
0
Fork 0
mirror of https://gitee.com/xmz7891/ykxmz.git synced 2026-10-18 10:10:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

修「去密码硬编码」那一步自己引入的三个 bug

Browse Source 
上一版把明文密码换成随机生成时，引入了三个问题。都是我自己的补丁
造成的，而且症状都像"功能坏了"，实际是环境和脚本的问题。

== 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 历史里还有，等用户确认后重写）。
This commit is contained in:
xmz7891
2026-10-07 20:56:53 +08:00
parent d424128483
commit 9b17129c49
12 changed files with 474 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
_build/_cmsg14.txt
+57
View File
@@ -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 历史里还有，等用户确认后重写）。
_build/chk_ab_now.sh
+32
View File
@@ -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
_build/fix_broken_password_patch.py
+140
View File
@@ -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())
_build/fix_test_admin_pw.py
+120
View File
@@ -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())
_tests/acceptance.py
+17 -1
View File
@@ -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, "没有公网地址，无法测")
_tests/test_admin_batch_delete.py
+17 -1
View File
@@ -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")
_tests/test_admin_button.py
+17 -1
View File
@@ -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")
_tests/test_admin_device_control.py
+17 -1
View File
@@ -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")
_tests/test_admin_js.py
+17 -1
View File
@@ -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")
_tests/test_admin_netcheck.py
+17 -1
View File
@@ -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")
_tests/test_relay_admin_login.py
+17 -1
View File
@@ -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
relay/relay.py
+6
View File
@@ -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'}）")