三层回退恢复机制(L0写前留档/L1恢复梯子/L2离线回滚)+ guard 父守护

- L0: files 插件写受保护系统路径(/etc 等)前自动留档,AbstractBeforeWrite 到 data/file_baseline
- L1: failback 受限 worker 执行恢复梯子 probe→还原DNS/proxy→还原LLM配置+ReloadFromConfig→probe,N轮有界
- L2: tracker changeset 持久化原文 blob,guard 离线 RollbackFromDisk 回滚 agentfs;SystemSnapshot 支撑
- guard 父守护: 心跳 IPC(PING/ACK unix socket, 文件心跳回退)、失败计数、退出码协议(42/43/44)、最后手段
- 发行版路径适配: system.protected_paths/network_paths 可注入,默认面向主流 Linux
- Windows 兼容: guard.go/failback.go 加 //go:build linux, guard_windows.go 提供 no-op 桩
- 修复: guard.yaml last_resort 键冲突、changeset Content 不落盘导致离线回滚丢原文

Build 全绿, vet 干净, system/recovery/ipc/tracker 单元测试全过
This commit is contained in:
root
2026-08-05 16:00:08 +08:00
parent 33e97f2a99
commit bc9ef15eb0
28 changed files with 3675 additions and 167 deletions

143
internal/ipc/ipc.go Normal file
View File

@ -0,0 +1,143 @@
// Package ipc 实现 guard↔worker 之间的 PING/ACK 心跳协议:worker 监听 unix socket,
// guard 发 PING,worker 回 ACK(含 kernel 状态快照),实现带自诊断上报的存活判定,
// 取代单纯的文件心跳 + 退出码。
package ipc
import (
"bufio"
"encoding/json"
"fmt"
"net"
"os"
"path/filepath"
"time"
)
// sinkPath 心跳 socket 文件路径。
func sinkPath(dataDir string) string {
return filepath.Join(dataDir, "worker.ipc")
}
// Status worker 上报给 guard 的自诊断快照。
type Status struct {
PID int `json:"pid"`
Boot string `json:"boot"` // normal | failback
UptimeSec int64 `json:"uptime_sec"` // 进程存活秒数
LLMOK *bool `json:"llm_ok,omitempty"` // rescue 源可达性(nil=未探)
Tools int `json:"tools"` // 已注册工具数
LastDiag string `json:"last_diag,omitempty"` // 最近一次自诊断结论(如 diag_loc.cause)
}
// StatusFunc 组装 worker 当前状态。
type StatusFunc func() *Status
// Server worker 侧:监听 unix socket,处理 PING→ACK。
type Server struct {
path string
ln net.Listener
status StatusFunc
stop chan struct{}
done chan struct{}
}
// NewServer 创建心跳服务端(尚未 Listen,见 Start)。
func NewServer(dataDir string, status StatusFunc) *Server {
return &Server{path: sinkPath(dataDir), status: status, stop: make(chan struct{}), done: make(chan struct{})}
}
// Start 启动监听。若 socket 已存在则先清理(无心跳残留)。
func (s *Server) Start() error {
if s.status == nil {
s.status = func() *Status { return nil }
}
if err := os.RemoveAll(s.path); err != nil {
return err
}
ln, err := net.Listen("unix", s.path)
if err != nil {
return fmt.Errorf("ipc listen %s: %w", s.path, err)
}
s.ln = ln
go s.acceptLoop()
return nil
}
func (s *Server) acceptLoop() {
defer close(s.done)
for {
conn, err := s.ln.Accept()
if err != nil {
return
}
go s.handle(conn)
}
}
func (s *Server) handle(conn net.Conn) {
defer conn.Close()
conn.SetDeadline(time.Now().Add(5 * time.Second))
sc := bufio.NewScanner(conn)
if !sc.Scan() {
return
}
line := sc.Text()
if line != "PING" {
return
}
resp := map[string]interface{}{"type": "ACK"}
if st := s.status(); st != nil {
resp["status"] = st
}
data, _ := json.Marshal(resp)
conn.Write(append(data, '\n'))
}
// Stop 关闭监听。
func (s *Server) Stop() {
if s.ln != nil {
s.ln.Close()
}
<-s.done
}
// Client guard 侧:向 worker 心跳 socket 发 PING 并等 ACK。
type Client struct {
path string
}
// NewClient 创建客户端。
func NewClient(dataDir string) *Client {
return &Client{path: sinkPath(dataDir)}
}
// Ping 发一次 PING、收 ACK。timeout 内未收到返回错误。返回 (status, error);
// status 可能为 nil(ACK 无状态体)。
func (c *Client) Ping(timeout time.Duration) (*Status, error) {
conn, err := net.DialTimeout("unix", c.path, timeout)
if err != nil {
return nil, err
}
defer conn.Close()
conn.SetDeadline(time.Now().Add(timeout))
if _, err := conn.Write([]byte("PING\n")); err != nil {
return nil, err
}
sc := bufio.NewScanner(conn)
if !sc.Scan() {
return nil, fmt.Errorf("ipc: empty ACK")
}
var resp struct {
Type string `json:"type"`
Status *Status `json:"status"`
}
if err := json.Unmarshal(sc.Bytes(), &resp); err != nil {
return nil, err
}
if resp.Type != "ACK" {
return nil, fmt.Errorf("ipc: unexpected reply %q", resp.Type)
}
return resp.Status, nil
}
// Close 无状态(客户端用完即释放连接)。
func (c *Client) Close() {}

67
internal/ipc/ipc_test.go Normal file
View File

@ -0,0 +1,67 @@
package ipc
import (
"testing"
"time"
)
func TestPingAck(t *testing.T) {
dir := t.TempDir()
ok := true
srv := NewServer(dir, func() *Status {
return &Status{PID: 123, Boot: "normal", UptimeSec: 42, LLMOK: &ok, Tools: 5, LastDiag: "diag:network"}
})
if err := srv.Start(); err != nil {
t.Fatalf("Start: %v", err)
}
defer srv.Stop()
cli := NewClient(dir)
st, err := cli.Ping(2 * time.Second)
if err != nil {
t.Fatalf("Ping: %v", err)
}
if st == nil {
t.Fatal("nil status")
}
if st.PID != 123 || st.Boot != "normal" || st.Tools != 5 {
t.Fatalf("status mismatch: %+v", st)
}
if st.LLMOK == nil || !*st.LLMOK {
t.Fatal("LLMOK should be true")
}
if st.LastDiag != "diag:network" {
t.Fatalf("LastDiag = %q", st.LastDiag)
}
}
func TestPingTimeout(t *testing.T) {
dir := t.TempDir()
// 未启动 server → 立即超时
cli := NewClient(dir)
start := time.Now()
_, err := cli.Ping(500 * time.Millisecond)
if err == nil {
t.Fatal("expected error when no server")
}
if time.Since(start) > 3*time.Second {
t.Fatal("timeout took too long")
}
}
func TestServerNoStatusFunc(t *testing.T) {
dir := t.TempDir()
srv := NewServer(dir, nil)
if err := srv.Start(); err != nil {
t.Fatalf("Start: %v", err)
}
defer srv.Stop()
cli := NewClient(dir)
st, err := cli.Ping(2 * time.Second)
if err != nil {
t.Fatalf("Ping: %v", err)
}
if st != nil {
t.Fatalf("expected nil status when no status func, got %+v", st)
}
}