162 lines
7.9 KiB
Bash
162 lines
7.9 KiB
Bash
#!/usr/bin/env bash
|
||
# ============================================================
|
||
# 定时重启功能的真实环境验收脚本
|
||
#
|
||
# 用法（在 VPS 上，root）:
|
||
# cd /opt/vps-xray && bash tests/verify-restart-on-vps.sh
|
||
#
|
||
# 为什么必须在真机上跑：本地的 62→101 项单元测试都是把生成的文件写进临时目录
|
||
# 再比对内容——它们能验证「我们写出了预期的字节」，但验证不了「systemd 认不认
|
||
# 这些字节」。TimeZone= 那个缺陷就是这么漏过去的：文件里确实有那一行，断言通过，
|
||
# 而 systemd 加载时只会记一条 Unknown key name 然后按系统本地时区解析。
|
||
#
|
||
# 本脚本只做安全操作：不升级 Xray、不重写 config.json、不重启 xray。
|
||
# 会临时改动 .env 的重启字段与三个 systemd 单元文件，结束时恢复原状。
|
||
# ============================================================
|
||
|
||
set -uo pipefail
|
||
|
||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
|
||
cd "$SCRIPT_DIR"
|
||
|
||
PASS=0; FAIL=0
|
||
ok() { PASS=$((PASS+1)); echo " ✅ $1"; }
|
||
bad() { FAIL=$((FAIL+1)); echo " ❌ $1"; [ -n "${2:-}" ] && echo " 实际: $2"; }
|
||
sec() { echo ""; echo "===== $* ====="; }
|
||
|
||
[ "$(id -u)" -eq 0 ] || { echo "需要 root"; exit 1; }
|
||
[ -f .env ] || { echo "找不到 .env，这台机器还没部署过"; exit 1; }
|
||
|
||
ENV_BAK="$(mktemp)"; cp .env "$ENV_BAK"
|
||
restore() {
|
||
echo ""
|
||
echo "===== 恢复原状 ====="
|
||
cp "$ENV_BAK" .env && echo " .env 已还原"
|
||
rm -f "$ENV_BAK"
|
||
bash deploy.sh --restart >/dev/null 2>&1 && echo " timer 已按还原后的 .env 重建" \
|
||
|| echo " timer 重建失败，请手工检查 systemctl list-timers xray-restart.timer"
|
||
}
|
||
trap restore EXIT
|
||
|
||
sec "0. 环境"
|
||
systemctl --version | head -1
|
||
timedatectl 2>/dev/null | grep -i "time zone" || cat /etc/timezone 2>/dev/null
|
||
|
||
sec "1. Critical-1：OnCalendar 尾随时区是否被 systemd 真正接受"
|
||
# 这是全套验证里最关键的一条。旧实现用 [Timer] 段的 TimeZone=，
|
||
# 而 systemd.timer 根本没有这个 key——写了会被忽略并按系统本地时区解析。
|
||
if out=$(systemd-analyze calendar "*-*-* 04:00:00 Asia/Shanghai" 2>&1); then
|
||
ok "systemd 接受带时区的日历表达式"
|
||
echo "$out" | sed 's/^/ /'
|
||
else
|
||
bad "systemd 不接受带时区的日历表达式（该 systemd 版本可能低于 242）" "$out"
|
||
fi
|
||
|
||
sec "2. 启用：--restart 7 04:00"
|
||
BEFORE_TS=$(systemctl show xray -p ActiveEnterTimestamp --value)
|
||
bash deploy.sh --restart 7 04:00
|
||
AFTER_TS=$(systemctl show xray -p ActiveEnterTimestamp --value)
|
||
|
||
[ "$BEFORE_TS" = "$AFTER_TS" ] \
|
||
&& ok "xray 未被重启（ActiveEnterTimestamp 前后一致）" \
|
||
|| bad "xray 被重启了！--restart 不该碰服务" "before=$BEFORE_TS after=$AFTER_TS"
|
||
|
||
grep -q '^XRAY_RESTART_EVERY_DAYS=7$' .env && ok ".env 天数已写入" || bad ".env 天数未写入" "$(grep '^XRAY_RESTART_EVERY_DAYS=' .env)"
|
||
grep -q '^XRAY_RESTART_TIME=04:00$' .env && ok ".env 时间已写入" || bad ".env 时间未写入" "$(grep '^XRAY_RESTART_TIME=' .env)"
|
||
|
||
sec "3. 生成的 timer 是否被 systemd 正确加载"
|
||
systemd-analyze verify /etc/systemd/system/xray-restart.timer 2>&1 | sed 's/^/ /'
|
||
if journalctl -b --no-pager 2>/dev/null | grep -q "Unknown key name.*xray-restart"; then
|
||
bad "journal 里有 Unknown key name 告警，说明 timer 里仍有非法指令"
|
||
else
|
||
ok "journal 无 Unknown key name 告警"
|
||
fi
|
||
|
||
CAL=$(systemctl show xray-restart.timer -p TimersCalendar --value 2>/dev/null)
|
||
echo " TimersCalendar: $CAL"
|
||
case "$CAL" in
|
||
*Asia/Shanghai*) ok "解析后的日历规格里含时区" ;;
|
||
*) bad "解析后的日历规格里没有时区——时区没生效" "$CAL" ;;
|
||
esac
|
||
|
||
grep -q '^TimeZone=' /etc/systemd/system/xray-restart.timer \
|
||
&& bad "timer 里仍有非法的 TimeZone= 指令" \
|
||
|| ok "timer 里无 TimeZone= 指令"
|
||
|
||
sec "4. 守卫脚本行为"
|
||
rm -f /var/lib/xray/last-restart
|
||
echo " -- 首次（无状态文件）应执行重启 --"
|
||
/usr/local/bin/xray-restart-guard 2>&1 | sed 's/^/ /'
|
||
[ -s /var/lib/xray/last-restart ] && ok "时间戳已写入" || bad "时间戳未写入"
|
||
|
||
echo " -- 第二次（刚重启过）应跳过 --"
|
||
G2=$(/usr/local/bin/xray-restart-guard 2>&1); echo "$G2" | sed 's/^/ /'
|
||
echo "$G2" | grep -q "跳过" && ok "正确跳过" || bad "未跳过" "$G2"
|
||
|
||
echo " -- Important-4：时间戳落在未来时应视为 0 并执行重启 --"
|
||
TS_BEFORE_FUTURE=$(systemctl show xray -p ActiveEnterTimestamp --value)
|
||
echo "$(( $(date +%s) + 86400*365 ))" > /var/lib/xray/last-restart
|
||
G3=$(/usr/local/bin/xray-restart-guard 2>&1); echo "$G3" | sed 's/^/ /'
|
||
if echo "$G3" | grep -q "跳过"; then
|
||
bad "未来时间戳导致永久跳过——保护未生效"
|
||
else
|
||
ok "未来时间戳被正确当成 0"
|
||
fi
|
||
echo " （这一步会真的重启一次 xray，是预期内的）"
|
||
sleep 2
|
||
systemctl is-active --quiet xray && ok "重启后 xray 仍然存活" || bad "xray 未能重新启动！"
|
||
|
||
sec "5. Important-6：下次实际重启时间是否被正确显示"
|
||
echo "$(( $(date +%s) - 3*86400 ))" > /var/lib/xray/last-restart
|
||
OUT=$(bash deploy.sh --restart 2>&1); echo "$OUT" | sed 's/^/ /'
|
||
echo " （N=7、距上次 3 天，所以真实的下次重启应在 4 天后，而非次日）"
|
||
|
||
sec "6. 关闭：--restart 0"
|
||
bash deploy.sh --restart 0 >/dev/null 2>&1
|
||
for f in /etc/systemd/system/xray-restart.timer /etc/systemd/system/xray-restart.service /usr/local/bin/xray-restart-guard; do
|
||
[ -f "$f" ] && bad "未删除: $f" || ok "已删除: $(basename "$f")"
|
||
done
|
||
[ -f /var/lib/xray/last-restart ] && ok "状态文件按设计保留" || bad "状态文件被误删（关闭≠卸载）"
|
||
|
||
sec "7. Critical-2：前导零天数"
|
||
bash deploy.sh --restart 09 >/dev/null 2>&1
|
||
D=$(grep '^XRAY_RESTART_EVERY_DAYS=' .env | cut -d= -f2)
|
||
[ "$D" = "9" ] && ok "09 被归一化成 9（未被当八进制）" || bad "09 未被正确归一化" "$D"
|
||
if [ -f /usr/local/bin/xray-restart-guard ]; then
|
||
grep -q '^DAYS=9$' /usr/local/bin/xray-restart-guard \
|
||
&& ok "guard 里 DAYS=9" || bad "guard 里天数不对" "$(grep '^DAYS=' /usr/local/bin/xray-restart-guard)"
|
||
OUT=$(/usr/local/bin/xray-restart-guard 2>&1)
|
||
echo "$OUT" | grep -qi "value too great\|error token" \
|
||
&& bad "guard 仍有八进制算术错误" "$OUT" \
|
||
|| ok "guard 无八进制算术错误"
|
||
fi
|
||
|
||
sec "8. 参数校验与互斥"
|
||
bash deploy.sh --restart 7 25:00 >/dev/null 2>&1; [ $? -eq 1 ] && ok "非法时间退出码 1" || bad "非法时间退出码不是 1"
|
||
bash deploy.sh --badarg >/dev/null 2>&1; [ $? -eq 1 ] && ok "未知参数退出码 1" || bad "未知参数退出码不是 1"
|
||
bash deploy.sh --help >/dev/null 2>&1; [ $? -eq 0 ] && ok "--help 退出码 0" || bad "--help 退出码不是 0"
|
||
# 注意：不能写成 `bash deploy.sh ... | grep -q ...`。本脚本开了 pipefail，
|
||
# 而 deploy.sh 在这里本就该以退出码 1 结束，管道会因此被判为失败——
|
||
# 于是断言恒为「未生效」，与被测行为无关。必须先取输出再匹配。
|
||
MX_OUT=$(bash deploy.sh --restart 7 --redetect 2>&1 || true)
|
||
echo "$MX_OUT" | grep -q "不能与" && ok "互斥校验生效" || bad "互斥校验未生效" "$MX_OUT"
|
||
|
||
echo ""
|
||
echo "============================================================"
|
||
echo " 通过 ${PASS}，失败 ${FAIL}"
|
||
echo "============================================================"
|
||
echo ""
|
||
echo "还需要你手动做一步（本脚本刻意不做，因为它会重启服务、断掉所有连接）："
|
||
echo ""
|
||
echo " # 旧字段迁移验证 —— 建议挑一台当前没在用的机器"
|
||
echo " cp .env /tmp/env.bak"
|
||
echo " sed -i '/^XRAY_RESTART_/d' .env"
|
||
echo " echo 'XRAY_DAILY_RESTART=true' >> .env"
|
||
echo " bash deploy.sh # 完整部署，会重启 xray"
|
||
echo " grep -E '^XRAY_RESTART_|^XRAY_DAILY' .env"
|
||
echo " # 预期：出现 XRAY_RESTART_EVERY_DAYS=1，且 XRAY_DAILY_RESTART 已消失"
|
||
echo " cp /tmp/env.bak .env && bash deploy.sh --restart"
|
||
echo ""
|
||
|
||
[ "$FAIL" -eq 0 ]
|