mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-10-15 12:15:16 +00:00
a7f89a1b45
部署 guard 前的审查发现两个**致命**缺陷。它们此前都零测试覆盖 ——
正是「没测过才敢发现」。
## ① L2 自杀循环（启用 guard 就会触发）
doLastResort(restart_app) 执行 `systemctl restart <unit>`，而启用 guard 的
标准做法正是让 systemd 管理 `homed --role=guard` —— <unit> 就是它自己：
guard: doLastResort → systemctl restart homeagent
→ 杀掉自己 → systemd 按 Restart=always 拉起新 guard
→ 新 guard 再 spawn 失败 → 再自杀 …… 无限循环
每轮白跑一遍完整启动。与 2026-09-05 那次 systemd 审计记录的事故同类
（坏配置导致 836 次重启、79 分钟）。
### 修法：自指时直接退出，而不是换个单元名
「换个别的服务」语义完全不对 —— L2 的本意就是重启自己。而
`Restart=always` 已经提供了这件事：**guard 退出 == systemd 重新拉起 ==
重启整个单元**。所以自指时 os.Exit(41) 与 restart_app 完全等价，且无自杀窗口。
- `doLastResortGuarded(cfg, unit)`：判自指 → 退出 41；否则原逻辑
- `detectOwnSystemdUnit()`：读 /proc/self/cgroup（cgroup v2），兜底
$INVOCATION_ID → /run/systemd/invocation/<id>。**拿不到就返回空字符串**
让调用方跳过检测 —— 不猜单元名（猜错 = 拦错/杀错服务）
- `exitGuardRestartRequested = 41`：刻意避开 worker 的 42-44 区间
- 两处 doLastResort 调用点全部改走防护版
## ② 配置损坏 → agent 装上就崩
defaultGuardConfig **不解析时长字符串**（只给 "30s" 这类字面量），
parseDur 在 loadGuardConfig 里、且在 yaml.Unmarshal **失败时提前 return
之后**——那段根本不执行。
于是 guard.yaml 一旦写坏（缩进错、半截内容，手改文本极常见）：
| 字段 | 实际值 | 后果 |
|---|---|---|
| heartbeatTimeout | **0** | `time.Since(mtime) > 0` 恒真 → **每轮杀 worker** → 无限重启 |
| heartbeatInterval | 0 | `time.NewTicker(0)` panic |
| attemptTimeout | 0 | 恢复 probe 立即超时 → L1 梯子必败 |
| Recovery.Plugins | 空 | 受限启动什么都不加载 |
这不是「配置没生效」，是「装上就崩」。修法：把 parseDur 与插件集缺省
移进 defaultGuardConfig，任何路径都有正值兜底。
## ③ 测试：0 → 25 条（全可杀变异）
| 文件 | 覆盖 |
|---|---|
| `guard_config_test.go` | 配置默认值/YAML/损坏/越界 + 心跳三态 |
| `guard_lifecycle_test.go` | worker 生命周期：6 种退出码分类、信号转发、坏二进制、卡死压力测试、L2 防护行为、单元探测 |
| `guard_safety_test.go` | 破坏性路径：回滚不可逆、任务协议往返、failback 插件集 |
| `rollback_irreversible_test.go` | 不可逆性证据 + 二次回滚 no-op + 新建→删除 |
**7 个变异逐个实测被杀**：parseDur 移回原处 / 心跳漏判 / 心跳误判 /
插件集留空 / exitRecovered→crash / exitRecoveryFailed→recovered /
rollback 不删 changeset / L2 防护失效 / 单元探测返回假名。
### 判据自身被变异测试逼着改三次
1. `isSelfRestartCommand` 的测试全绿，但把 `doLastResortGuarded` 改成
直接调 `doLastResort`（防护完全不生效）时**仍然全绿** ⇒ 补了
**行为测试**：真子进程 + 假 systemctl（PATH 前置），验「自指时
restart_command 绝不被执行」
2. 单元探测只断言「非空+格式合法」，被 `return "wrong.service"` 骗过
⇒ 补了与**独立第二实现**（测试里重写一遍 cgroup 解析）的交叉核对
3. 信号转发测试用 `sleep 300` 会泄漏孤儿进程（sh fork 子进程，信号只到 sh）
⇒ 改 `exec sleep 300`，并加断言检测泄漏（60s → 0.52s）
### 踩到的两个坑（都记在判据注释里）
- **POSIX 环境变量不能含 NUL**：`\x00` 分隔符让 exec 直接报
"environment variable contains NUL"，改用 `\x1f`
- **测试函数里 os.Exit 会终止整个测试二进制** ⇒ 子进程入口必须用 TestMain
全仓 `go test ./... -count=1` → 48 包 ok / 0 FAIL。
38 lines
1.3 KiB
Go
38 lines
1.3 KiB
Go
package main
|
||
|
||
import (
|
||
"os"
|
||
"os/exec"
|
||
"path/filepath"
|
||
)
|
||
|
||
// worker 退出码协议：guard 通过 worker 退出码判定下一步动作。
|
||
// 该协议跨平台一致（Windows 下 guard 为 no-op，退出码仍保留）。
|
||
const (
|
||
exitRestartRequested = 42 // worker 清理后请求重建（不计失败轮次）
|
||
exitRecovered = 43 // failback worker 恢复成功，guard 应重置失败轮次并交回主 agent
|
||
exitRecoveryFailed = 44 // failback worker 单轮恢复失败，guard 计入失败并进入下一轮/最后手段
|
||
|
||
// exitGuardRestartRequested 是 guard 自身的退出码：L2 需要「重启本单元」但
|
||
// restart_command 指向自己所在单元时（systemctl restart 会自杀），
|
||
// guard 改为直接退出，由 systemd 的 Restart=always 重新拉起 —— 等价且无自杀窗口。
|
||
//
|
||
// 刻意取 41（小于 worker 的 42-44 区间），避免与 worker 退出码协议混淆。
|
||
exitGuardRestartRequested = 41
|
||
)
|
||
|
||
// resolveDataDir 复用原有的数据目录探测逻辑。
|
||
func resolveDataDir(dataDir string) string {
|
||
if dataDir != "" {
|
||
return dataDir
|
||
}
|
||
exe, err := os.Executable()
|
||
if err == nil {
|
||
return filepath.Join(filepath.Dir(exe), "data")
|
||
}
|
||
if exe, err := exec.LookPath(os.Args[0]); err == nil {
|
||
return filepath.Join(filepath.Dir(exe), "data")
|
||
}
|
||
return "./data"
|
||
}
|