Files
HomeAgent/cmd/homed/guard.go
JianFeeeee 562ba6bd98 fix(guard): 启动宽限期 —— 停机重启后不再误杀加载中的 worker
## 故障(2026-10-12 部署实测,6 连杀 + 服务降级)

停机 34s 后重启:heartbeat 文件 mtime 停在**上一代 worker**(已远超
30s 阈值)。新 worker 要先做完全部重型初始化(chineseclip 512 +
两个静态词向量 + 知识重建,冷启动 5–8s)才写第一次心跳
(startHeartbeat 在 main.go:190,位于所有初始化之后)。
而 guard 的 5s ticker 先到、拿旧 mtime 判 stale → 把还在加载的
worker SIGKILL → 循环重启,guard 进入 failback Safe-Mode:

    19:00:28 worker up pid=931403 boot=normal
    19:00:33 heartbeat stale for pid=931403 ... SIGKILL   ← 加载中
    19:00:33 worker up pid=931557 boot=normal
    19:00:38 heartbeat stale ... SIGKILL
    ...(6 连杀后)
    19:00:43 3 failures >= max 3, entering failback (Safe-Mode)

worker 自身日志完全健康(graph/multimodal/knowledge 依次就绪),
纯死于"还没来得及写第一次心跳"。

## 根因

heartbeatStale 对「文件不存在」宽限,但对「文件存在、mtime 是
上一代的」不宽限——停机超过 heartbeatTimeout 的每一次重启都会踩。
旧内核未暴露只因它从未停机这么久(上一次是系统重启,mtime 新鲜)。
既有缺陷,block-union 重构未触碰该逻辑(main 分支同病)。

## 修法

spawn 后给一个 heartbeatTimeout 的**启动宽限期**:期间即使心跳
文件 mtime 极旧也不判杀。宽限期满后恢复原判定(卡死 worker 靠
宽限躲不过去)。决策抽成纯函数 shouldKillStaleHeartbeat。

## 判据(TestStartupGraceAllowsInit,变异测试通过)

  (a) 宽限期内 + mtime 极旧(上一代遗留)→ 不许杀
      (末沿 -50ms 也钉住:ticker 恰在 deadline 前触发是常态)
  (b) 宽限期满 + mtime 极旧(真卡死)→ 必须杀
  (c) 宽限期满 + mtime 新鲜(健康)→ 不许杀

变异(移除宽限判定、直接 heartbeatStale)→ (a) 两条断言当场 FAIL;
恢复后全绿。TestHeartbeatStale(三分支既有判据)不受影响。

cmd/homed 包全量测试通过;全仓生产编译 GREEN。
2026-10-12 19:16:57 +08:00

649 lines
24 KiB
Go
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

//go:build linux
// L0/L1/L2 三层回退恢复机制的核心:父守护 guard(Linux 专属)。
//
// 依赖 Linux 设施:syscall.Reboot / /etc 网络基线 / overlayfs 离线回滚 /
// unix socket IPC / systemctl 重启。Windows 等平台不编译本文件,
// 由 guard_windows.go 提供 no-op 桩,保证 cmd/homed 跨平台可构建。
package main
import (
"errors"
"fmt"
"log"
"os"
"os/exec"
"os/signal"
"path/filepath"
"strings"
"syscall"
"time"
internalConfig "github.com/JianFeeeee/HomeAgent/internal/config"
"github.com/JianFeeeee/HomeAgent/internal/ipc"
"github.com/JianFeeeee/HomeAgent/internal/recovery"
"github.com/JianFeeeee/HomeAgent/internal/system"
"github.com/JianFeeeee/HomeAgent/internal/tracker"
"gopkg.in/yaml.v3"
)
// guardConfig 父守护配置:独立于 config.db,避开被改坏的配置数据库。
type guardConfig struct {
MaxRestarts int `yaml:"max_restarts"` // 连续 normal 崩溃重试上限
HeartbeatTimeout string `yaml:"heartbeat_timeout"` // 心跳多久未更新判定卡死(如 30s)
HeartbeatInterval string `yaml:"heartbeat_interval"` // 期待 worker 心跳频率(仅日志)
LLMSnapshot string `yaml:"llm_snapshot"` // LLM 配置基线文件,为空用 <data>/llm_snapshot.json
FailbackEnabled bool `yaml:"failback_enabled"` // 崩溃超限后是否进入受限 failback 启动
LastResort string `yaml:"last_resort"` // restart_app | reboot
RestartCommand []string `yaml:"restart_command"` // last_resort=restart_app 时的命令(如 systemctl restart homeagent)
RebootGrace string `yaml:"reboot_grace"` // last_resort=reboot 前的缓冲
LLM llmConfig `yaml:"llm"` // rescue 源(锚定 IP,绕开被破坏的 DNS/代理)
Recovery recoveryConfig `yaml:"recovery"` // failback 恢复参数(N 轮有界)
LastResortCfg lastResortConfig `yaml:"last_resort_cfg"` // 最后手段细节(snapshot_before 等)
System systemConfig `yaml:"system"` // 发行版/部署路径适配
heartbeatTimeout time.Duration
heartbeatInterval time.Duration
rebootGrace time.Duration
}
type llmConfig struct {
Sources []recovery.RescueSource `yaml:"sources"`
}
type recoveryConfig struct {
MaxAttempts int `yaml:"max_attempts"` // failback 恢复尝试轮数上限
AttemptTimeout string `yaml:"attempt_timeout"` // 单轮恢复超时
KnowledgeBase string `yaml:"knowledge_base"` // 恢复知识库目录
TriggerPrompt string `yaml:"trigger_prompt"` // 恢复 agent 的系统提示词(未知/混合分支)
Plugins []string `yaml:"plugins"` // failback 插件集(缺省用 core.agent.failback_plugins)
attemptTimeout time.Duration
}
type lastResortConfig struct {
SnapshotBefore bool `yaml:"snapshot_before"` // 最后手段前是否回滚 agentfs 到最近快照
}
// systemConfig 发行版/部署路径适配:把 L0 写前留档保护范围与网络基线文件列表
// 从默认的 Linux 路径换成当前部署实际路径,避免硬编码失效。
type systemConfig struct {
ProtectedPaths []string `yaml:"protected_paths"` // L0 写前留档保护前缀(默认 /etc/)
NetworkPaths []string `yaml:"network_paths"` // 网络基线文件(默认 resolv.conf/hosts/environment)
}
// applySystemConfig 应用发行版路径适配,供 guard 与 failback worker 共用。
func applySystemConfig(cfg *guardConfig) {
if len(cfg.System.ProtectedPaths) > 0 {
system.SetProtectedPaths(cfg.System.ProtectedPaths)
}
if len(cfg.System.NetworkPaths) > 0 {
system.SetNetworkPaths(cfg.System.NetworkPaths)
}
}
func defaultGuardConfig(dataDir string) *guardConfig {
cfg := &guardConfig{
MaxRestarts: 3,
HeartbeatTimeout: "30s",
HeartbeatInterval: "5s",
LLMSnapshot: filepath.Join(dataDir, "llm_snapshot.json"),
FailbackEnabled: true,
LastResort: "restart_app",
RestartCommand: []string{"systemctl", "restart", "homeagent"},
RebootGrace: "10s",
Recovery: recoveryConfig{
MaxAttempts: 3,
AttemptTimeout: "120s",
},
LastResortCfg: lastResortConfig{SnapshotBefore: true},
}
// ★ 必须在这里就把 YAML 字符串解析成 time.Duration(2026-10-11 修)。
//
// 原先只在 loadGuardConfig 里 parseDur,而 parseDur 是在 yaml.Unmarshal
// **之后**执行的。问题出在「解析失败」那条早退路径:
//
// if err := yaml.Unmarshal(data, cfg); err != nil {
// log.Printf(...); return cfg ← 直接返回,没走下面的 parseDur
// }
//
// 于是 guard.yaml 一旦写坏(缩进错、半截内容),返回的 cfg 里
// heartbeatTimeout/heartbeatInterval/recovery.attemptTimeout 全是 **0**。
// 而 loadGuardConfig 开头正是 defaultGuardConfig(dataDir),
// 唯一给这些字段兜底的只有 defaultGuardConfig —— 可它也只给了字符串。
//
// 后果不是「配置没生效」而是 **agent 装上就崩**:
// heartbeatStale(path, 0) 里 `time.Since(mtime) > 0` 恒为真,
// 任何心跳都被判陈旧 → guard 每轮都杀 worker → 无限重启。
// attemptTimeout=0 则让恢复 probe 立即超时 → L1 梯子必判失败。
//
// 判据:cmd/homed/guard_config_test.go 的
// TestLoadGuardConfigSurvivesCorruptYAML(配置损坏后仍须正值)。
parseDur := func(s string, def time.Duration) time.Duration {
if s == "" {
return def
}
d, err := time.ParseDuration(s)
if err != nil || d <= 0 {
return def
}
return d
}
cfg.heartbeatTimeout = parseDur(cfg.HeartbeatTimeout, 30*time.Second)
cfg.heartbeatInterval = parseDur(cfg.HeartbeatInterval, 5*time.Second)
cfg.rebootGrace = parseDur(cfg.RebootGrace, 10*time.Second)
cfg.Recovery.attemptTimeout = parseDur(cfg.Recovery.AttemptTimeout, 120*time.Second)
if len(cfg.Recovery.Plugins) == 0 {
cfg.Recovery.Plugins = []string{"webui", "pluginmgr", "recoverydiag"}
}
return cfg
}
func loadGuardConfig(dataDir string) *guardConfig {
cfg := defaultGuardConfig(dataDir)
path := filepath.Join(dataDir, "guard.yaml")
data, err := os.ReadFile(path)
if err != nil {
return cfg
}
if err := yaml.Unmarshal(data, cfg); err != nil {
log.Printf("[guard] parse %s: %v, using defaults", path, err)
return cfg
}
if cfg.LLMSnapshot == "" {
cfg.LLMSnapshot = filepath.Join(dataDir, "llm_snapshot.json")
}
parseDur := func(s string, def time.Duration) time.Duration {
if s == "" {
return def
}
d, err := time.ParseDuration(s)
if err != nil {
return def
}
return d
}
cfg.heartbeatTimeout = parseDur(cfg.HeartbeatTimeout, 30*time.Second)
cfg.heartbeatInterval = parseDur(cfg.HeartbeatInterval, 5*time.Second)
cfg.rebootGrace = parseDur(cfg.RebootGrace, 10*time.Second)
cfg.Recovery.attemptTimeout = parseDur(cfg.Recovery.AttemptTimeout, 120*time.Second)
if cfg.MaxRestarts < 1 {
cfg.MaxRestarts = 1
}
if cfg.LastResort != "reboot" && cfg.LastResort != "restart_app" {
cfg.LastResort = "restart_app"
}
if len(cfg.RestartCommand) == 0 {
cfg.RestartCommand = []string{"systemctl", "restart", "homeagent"}
}
if cfg.Recovery.MaxAttempts < 1 {
cfg.Recovery.MaxAttempts = 1
}
// 缺省 failback 插件集
if len(cfg.Recovery.Plugins) == 0 {
cfg.Recovery.Plugins = []string{"webui", "pluginmgr", "recoverydiag"}
}
return cfg
}
// runGuard 父守护主循环:拉起 worker(--role=agent),心跳探活 + waitpid,
// 崩溃后按序 重试→LLM 配置基线恢复→failback 受限启动(N 轮有界,每轮复检)→L2 最后手段。
func runGuard(dataDir string) {
cfg := loadGuardConfig(dataDir)
applySystemConfig(cfg)
hbPath := filepath.Join(dataDir, "heartbeat")
os.MkdirAll(filepath.Join(dataDir, "log"), 0755)
exe, err := os.Executable()
if err != nil {
log.Fatalf("[guard] resolve executable: %v", err)
}
sigCh := make(chan os.Signal, 1)
signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)
log.Printf("[guard] starting, data=%s max_restarts=%d hb_timeout=%v last_resort=%s failback_rounds=%d",
dataDir, cfg.MaxRestarts, cfg.heartbeatTimeout, cfg.LastResort, cfg.Recovery.MaxAttempts)
// 探测自身所属的 systemd 单元(L2 自杀防护需要)。
// 拿不到就留空 —— 那种情况下跳过自指检测,而不是猜一个单元名。
ownUnit := detectOwnSystemdUnit()
if ownUnit != "" {
log.Printf("[guard] own systemd unit=%s", ownUnit)
} else {
log.Printf("[guard] own systemd unit unknown (L2 self-restart guard disabled)")
}
// 启动即捕获网络基线(L0:写前留档 + L1 还原依据),并给当前 worker 恢复代理环境
captureNetworkBaseline(dataDir)
failures := 0 // 连续 normal 崩溃次数
failbackRound := 0 // 已进行的 failback 轮次
failbackDone := false // 本轮 failback 是否已进入
for {
// 决定本次 boot 模式:normal 崩溃超限 → 进入 failback
boot := "normal"
if cfg.FailbackEnabled && !failbackDone && failures >= cfg.MaxRestarts {
log.Printf("[guard] %d failures >= max %d, entering failback (Safe-Mode)", failures, cfg.MaxRestarts)
// L1 前置:恢复 LLM 基线 + 还原 DNS/proxy + 写恢复任务
restoreLLMBaseline(dataDir, cfg.LLMSnapshot)
restoreNetworkBaseline(dataDir)
failbackRound = 1
failbackDone = true
if err := writeRecoveryTask(dataDir, cfg, failbackRound); err != nil {
log.Printf("[guard] write recovery task: %v", err)
}
boot = "failback"
} else if failbackDone {
boot = "failback"
}
// 恢复 agent 环境(代理变量)在 worker 拉起前注入
if b, err := system.LoadNetworkBaseline(dataDir); err == nil {
b.ApplyProxyEnv()
}
outcome, spawnErr := spawnWorkerOnce(exe, dataDir, boot, hbPath, cfg, sigCh)
if spawnErr != nil {
log.Printf("[guard] spawn worker: %v, last_resort=%s", spawnErr, cfg.LastResort)
doLastResortGuarded(cfg, ownUnit)
failures = 0
failbackDone = false
failbackRound = 0
time.Sleep(2 * time.Second)
continue
}
switch outcome {
case workerCleanExit:
log.Printf("[guard] worker exited cleanly, guard exiting")
return
case workerRestartRequested:
log.Printf("[guard] worker requested restart, respawning")
continue
case workerRecovered:
// failback 成功:重置失败轮次,交回主 agent(下一轮 normal boot)
log.Printf("[guard] failback recovery succeeded, handing back to main agent")
failures = 0
failbackDone = false
failbackRound = 0
continue
case workerCrash:
// fallthrough below
}
failures++
if failbackDone {
// failback 轮次内崩溃:读结果判定是否成功
if res, err := recovery.LoadResult(recovery.ResultPath(dataDir)); err == nil && res.Success {
log.Printf("[guard] failback round %d result success: %s", failbackRound, res.Quote())
failures = 0
failbackDone = false
failbackRound = 0
continue
}
if failbackRound >= cfg.Recovery.MaxAttempts {
log.Printf("[guard] failback exhausted after %d rounds, L2 last_resort=%s", failbackRound, cfg.LastResort)
if cfg.LastResortCfg.SnapshotBefore {
rollbackAgentFS(dataDir)
}
doLastResortGuarded(cfg, ownUnit)
failures = 0
failbackDone = false
failbackRound = 0
time.Sleep(2 * time.Second)
continue
}
// 进入下一轮 failback
failbackRound++
log.Printf("[guard] failback round %d failed, next round %d (max %d)", failbackRound-1, failbackRound, cfg.Recovery.MaxAttempts)
if err := writeRecoveryTask(dataDir, cfg, failbackRound); err != nil {
log.Printf("[guard] write recovery task: %v", err)
}
continue
}
if failures >= cfg.MaxRestarts {
log.Printf("[guard] %d normal failures, entering failback next loop", failures)
}
}
}
type workerOutcome int
const (
workerCleanExit workerOutcome = iota
workerCrash
workerRestartRequested
workerRecovered
)
// spawnWorkerOnce 拉起一个 worker 并监控到其退出。
// 返回 (outcome workerOutcome, spawnErr error);spawnErr 非 nil 表示未能拉起进程。
func spawnWorkerOnce(exe, dataDir, boot, hbPath string, cfg *guardConfig, sigCh chan os.Signal) (workerOutcome, error) {
cmd := exec.Command(exe, "--role=agent", "--data="+dataDir, "--boot="+boot)
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
cmd.Stdin = nil
if err := cmd.Start(); err != nil {
return workerCrash, err
}
pid := cmd.Process.Pid
log.Printf("[guard] worker up pid=%d boot=%s", pid, boot)
done := make(chan error, 1)
go func() { done <- cmd.Wait() }()
// ★ 启动宽限期:新 worker 要先做完全部重型初始化(graph/文本/向量/
// 多模态空间/知识库,实测冷启动约 5–8s)才开始写心跳(main.go:190)。
// 而 heartbeat 文件保存的是**上一代 worker** 的 mtime——停机 >heartbeatTimeout
// 后,新 worker 还没启动、文件就已"过期"。若无宽限,guard 的 heartbeatInterval
// ticker 先到、拿旧 mtime 判 stale → 把还在正常加载的 worker SIGKILL,
// 循环重启抖动(2026-10-12 部署实测触发:连续 6 次 failback Safe-Mode)。
// 宽限 = 给 worker 完整的启动时长再开始心跳判定。
startupGrace := cfg.heartbeatTimeout
startupDeadline := time.Now().Add(startupGrace)
hbTicker := time.NewTicker(cfg.heartbeatInterval)
defer hbTicker.Stop()
ipcClient := ipc.NewClient(dataDir)
for {
select {
case sig := <-sigCh:
log.Printf("[guard] signal %v, stopping worker", sig)
cmd.Process.Signal(syscall.SIGTERM)
select {
case <-done:
case <-time.After(10 * time.Second):
cmd.Process.Kill()
<-done
}
return workerCleanExit, nil
case err := <-done:
if err == nil {
log.Printf("[guard] worker pid=%d exited cleanly", pid)
return workerCleanExit, nil
}
var ee *exec.ExitError
if errors.As(err, &ee) {
switch ee.ExitCode() {
case exitRestartRequested:
log.Printf("[guard] worker pid=%d requested restart (exit %d), respawning without counting failure", pid, exitRestartRequested)
return workerRestartRequested, nil
case exitRecovered:
log.Printf("[guard] worker pid=%d recovered main agent (exit %d)", pid, exitRecovered)
return workerRecovered, nil
case exitRecoveryFailed:
log.Printf("[guard] worker pid=%d failback attempt failed (exit %d)", pid, exitRecoveryFailed)
return workerCrash, nil
}
}
log.Printf("[guard] worker pid=%d crashed: %v", pid, err)
return workerCrash, nil
case <-hbTicker.C:
// IPC PING/ACK 存活判定(带自诊断上报),失败回退文件心跳 mtime。
if st, perr := ipcClient.Ping(cfg.heartbeatTimeout); perr != nil {
// ★ 启动宽限:新 worker 先做完全部重型初始化(graph/文本/向量/
// 多模态空间/知识库,实测冷启动约 5–8s)才写心跳(main.go:190)。
// 而 heartbeat 文件 mtime 是**上一代 worker** 的——停机 >heartbeatTimeout
// 后,新 worker 还没启动、文件就已"过期"。无宽限时 guard 的 ticker
// 先到、拿旧 mtime 判 stale → 把还在加载的 worker SIGKILL,循环抖动
// (2026-10-12 部署实测触发:6 次 failback Safe-Mode)。
if shouldKillStaleHeartbeat(time.Now(), startupDeadline, hbPath, cfg.heartbeatTimeout) {
log.Printf("[guard] heartbeat stale for pid=%d (ipc: %v), treating as hang, SIGKILL", pid, perr)
cmd.Process.Kill()
<-done
return workerCrash, nil
}
} else if st != nil && st.LastDiag != "" {
log.Printf("[guard] worker pid=%d self-diagnosis: %s", pid, st.LastDiag)
}
}
}
}
// heartbeatStale 判心跳文件是否过期(纯函数,见 TestHeartbeatStale)。
func heartbeatStale(path string, timeout time.Duration) bool {
fi, err := os.Stat(path)
if err != nil {
// 无心跳文件:worker 刚启动或异常;宽限处理
return false
}
return time.Since(fi.ModTime()) > timeout
}
// shouldKillStaleHeartbeat 在启动宽限期满后,才对陈旧心跳做杀进程判定。
//
// 纯函数,把 spawnWorkerOnce 里的决策抽出来便于判据覆盖。
// 之所以不能只测 heartbeatStale:真正的回归是「拿上一代 worker 的
// 旧 mtime,在新 worker 还在加载时就判 stale」——只有与启动宽限
// 结合才暴露(见 TestStartupGraceAllowsInit)。
func shouldKillStaleHeartbeat(now, startupDeadline time.Time, hbPath string, timeout time.Duration) bool {
if now.Before(startupDeadline) {
return false // 启动宽限期:不判
}
return heartbeatStale(hbPath, timeout)
}
// restoreLLMBaseline 从文件基线恢复 core.llm.* 配置(存在才恢复)。
func restoreLLMBaseline(dataDir, snapPath string) {
if _, err := os.Stat(snapPath); err != nil {
log.Printf("[guard] no llm snapshot baseline at %s, skip restore", snapPath)
return
}
snap, err := internalConfig.LoadLLMSnapshot(snapPath)
if err != nil {
log.Printf("[guard] load llm snapshot %s: %v", snapPath, err)
return
}
cfgReg := internalConfig.NewConfigRegistry(filepath.Join(dataDir, "config.db"))
defer cfgReg.Close()
if err := cfgReg.RestoreCoreLLM(snap); err != nil {
log.Printf("[guard] restore llm baseline: %v", err)
return
}
log.Printf("[guard] restored %d llm keys from %s", len(snap), snapPath)
}
// captureNetworkBaseline L0:启动即捕获网络基线(DNS/hosts/代理),供 failback 还原。
func captureNetworkBaseline(dataDir string) {
base, err := system.CaptureNetwork()
if err != nil {
log.Printf("[guard] capture network baseline: %v", err)
return
}
if err := system.SaveNetworkBaseline(dataDir, base); err != nil {
log.Printf("[guard] save network baseline: %v", err)
return
}
log.Printf("[guard] network baseline captured: %s", base.Summary())
}
// restoreNetworkBaseline L1:把 resolv.conf/hosts 还原到基线(DNS/proxy 小修命中即停)。
func restoreNetworkBaseline(dataDir string) {
base, err := system.LoadNetworkBaseline(dataDir)
if err != nil {
log.Printf("[guard] no network baseline, skip DNS/proxy restore: %v", err)
return
}
changed, err := base.RestoreFiles()
if err != nil {
log.Printf("[guard] restore network baseline: %v", err)
return
}
if len(changed) > 0 {
log.Printf("[guard] restored network files: %v", changed)
} else {
log.Printf("[guard] network baseline already consistent")
}
}
// writeRecoveryTask 写本轮 failback 恢复任务文件。
func writeRecoveryTask(dataDir string, cfg *guardConfig, attempt int) error {
task := &recovery.Task{
Attempt: attempt,
MaxAttempts: cfg.Recovery.MaxAttempts,
AttemptTimeout: cfg.Recovery.AttemptTimeout,
TriggerPrompt: cfg.Recovery.TriggerPrompt,
KnowledgeBase: cfg.Recovery.KnowledgeBase,
PluginList: cfg.Recovery.Plugins,
}
for _, s := range cfg.LLM.Sources {
if s.Name == "rescue" || s.Name != "" {
task.RescueSource = s
break
}
}
return recovery.SaveTask(recovery.TaskPath(dataDir), task)
}
// rollbackAgentFS L2:guard 在 worker 离线时回滚 agentfs 最近快照(read changesets 逆应用)。
func rollbackAgentFS(dataDir string) {
workDir := filepath.Join(dataDir, "agentfs")
trk := tracker.NewOfflineTracker(dataDir, workDir)
n, err := trk.RollbackFromDisk()
if err != nil {
log.Printf("[guard] agentfs rollback: %v", err)
return
}
log.Printf("[guard] agentfs rolled back %d change sets", n)
}
func doLastResort(cfg *guardConfig) {
switch cfg.LastResort {
case "reboot":
log.Printf("[guard] last_resort=reboot, sync + reboot in %v", cfg.rebootGrace)
time.Sleep(cfg.rebootGrace)
syscall.Sync()
if err := syscall.Reboot(syscall.LINUX_REBOOT_CMD_RESTART); err != nil {
log.Printf("[guard] reboot failed (likely no privilege): %v", err)
}
case "restart_app":
cmd := exec.Command(cfg.RestartCommand[0], cfg.RestartCommand[1:]...)
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
if err := cmd.Run(); err != nil {
log.Printf("[guard] restart_app command %s: %v", strings.Join(cfg.RestartCommand, " "), err)
}
}
}
// ══════════════════════════════════════════════════════════════════════
// L2 自杀防护(2026-10-11)
//
// # 要防的是什么
//
// doLastResort(restart_app) 会执行 `systemctl restart <unit>`。而启用 guard 的
// 标准做法正是让 systemd 管理 `homed --role=guard` —— 也就是说 **<unit> 就是
// guard 自己所在的单元**。
//
// 于是 spawnWorker 失败时形成死循环:
//
// guard: doLastResort → systemctl restart homeagent
// → SIGTERM 杀掉自己 → systemd 按 Restart=always 拉起新 guard
// → 新 guard 再 spawn 失败 → 再自杀 ……
//
// 每轮都白跑一遍完整启动,且日志被刷爆。这与 2026-09-05 那次 systemd 审计
// 记录的事故同类(坏配置导致 836 次重启、79 分钟)。
//
// # 正确解法不是「换个单元名」
//
// 换成别的服务(重启 llmsproxy 之类)语义完全不对——L2 的本意是
// 「重启我自己」。而 systemd 的 Restart=already(always) 已经提供了这件事:
//
// guard 自行退出 == systemd 重新拉起 == 重启整个单元
//
// 所以自指时应当**直接退出**,效果与 restart_app 完全等价,且没有自杀窗口。
//
// # 为何还要保留 restart_command
//
// 两种部署形态都合法:
// - systemd 管 guard(推荐):自指 → 退出即可
// - 别的守护器管 guard(如 supervisor/前台 shell):此时 guard 不是
// systemd 单元的 MainPID,restart_command 可能指向一个**正确**的其它单元,
// 或者根本不该执行
// 故不删除,只在「确实是自指」时改走退出路径,并留下明确日志。
func doLastResortGuarded(cfg *guardConfig, unit string) {
if unit != "" && isSelfRestartCommand(cfg.RestartCommand, unit) {
log.Printf("[guard] last_resort=restart_app 但 restart_command 指向自身单元 %q:"+
"执行 systemctl restart 会杀掉自己并被 Restart=always 立即拉起,"+
"形成无限自杀重启。改为**直接退出** —— "+
"在 Restart=always 下这与 restart_app 等价且无自杀窗口。"+
"(若希望 guard 不被 systemd 管,请改用其它 last_resort 或外部守护器)",
unit)
// 交给 systemd 重新拉起(等价于 restart_app)。
// 退出码沿用 workerCleanExit 语义:正常路径下 guard 本就在 worker
// 干净退出时这样返回,systemd 看到非零退出也不会放弃(Restart=always)。
os.Exit(exitGuardRestartRequested)
}
doLastResort(cfg)
}
// isSelfRestartCommand 报告 restart_command 是否会重启 guard 自己所在的单元。
//
// 只认 `systemctl … restart <unit>` 与 `systemctl … restart <unit>.service`
// 两种形态;其它子命令(reload/restart-or-retry 等)不误判。
//
// unit 传空字符串时返回 false(检测不出来就别乱杀)。
func isSelfRestartCommand(cmd []string, unit string) bool {
if len(cmd) == 0 || unit == "" {
return false
}
base := strings.TrimSuffix(unit, ".service")
for i, arg := range cmd {
if arg != "restart" && arg != "restart-or-retry" {
continue
}
// restart 之后的下一个非 flag 参数就是目标单元
for _, target := range cmd[i+1:] {
if strings.HasPrefix(target, "-") {
continue
}
return strings.TrimSuffix(target, ".service") == base
}
}
return false
}
// detectOwnSystemdUnit 返回 guard 自己所属的 systemd 单元名(如 "homeagent.service")。
//
// 两个探测途径(都读 Linux 已有设施,不需额外依赖):
// 1. /proc/self/cgroup —— cgroup v2 下形如 /system.slice/homeagent.service
// 2. $INVOCATION_ID —— 反查 /run/systemd/invocation/<id> 的符号链接名
//
// 都拿不到时返回空字符串(调用方据此跳过自指检测,不猜)。
func detectOwnSystemdUnit() string {
// 途径 1:cgroup
if b, err := os.ReadFile("/proc/self/cgroup"); err == nil {
for _, line := range strings.Split(string(b), "\n") {
// "0::/system.slice/homeagent.service"
idx := strings.Index(line, "::")
if idx < 0 {
continue
}
path := line[idx+2:]
if j := strings.LastIndex(path, "/"); j >= 0 {
name := strings.TrimSpace(path[j+1:])
if strings.HasSuffix(name, ".service") {
return name
}
}
}
}
// 途径 2:INVOCATION_ID → /run/systemd/invocation/<id>
if id := os.Getenv("INVOCATION_ID"); id != "" {
if target, err := os.Readlink("/run/systemd/invocation/" + id); err == nil {
name := filepath.Base(target)
if strings.HasSuffix(name, ".service") {
return name
}
}
}
return ""
}
var _ = fmt.Sprintf