mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-11-01 23:07:30 +00:00
4e70b8d5d8
批 1，不含 SDK 仓与目录改名。 - go.mod 第 1 行 module 路径 - 366 个 .go 文件的完整前缀 github.com/JianFeeeee/HomeAgent 机械替换 （857 处，仅完整前缀；展示名/注释里的 HomeAgent 字样留给批 5） - 未改：third_party 目录名、SDK 自身 go.mod module、go.mod replace 目标、 homed/waiter 可执行名、.hmap 后缀
121 lines
3.6 KiB
Go
121 lines
3.6 KiB
Go
//go:build linux
|
||
|
||
package agentcli
|
||
|
||
import (
|
||
"os"
|
||
"strconv"
|
||
"strings"
|
||
"syscall"
|
||
"testing"
|
||
"time"
|
||
|
||
sdk "github.com/JianFeeeee/HologramAgent/internal/sdk"
|
||
)
|
||
|
||
// 超时终端必须真正释放：杀掉整个进程组（包括 sh 的子进程如 sleep）
|
||
// 且回收子进程（不留 <defunct>）。
|
||
//
|
||
// 旧实现三个缺陷叠加：
|
||
// 1. readLoop 的 IsExpired 分支只 delete(sessions) 后 return，不 Kill 不 Close；
|
||
// 2. Kill 只杀直接子进程 sh，Setsid 后真正的命令（sleep）是孙进程，成为孤儿；
|
||
// 3. 只 Start 从不 Wait，退出的子进程无人回收，积成僵尸。
|
||
func TestExpiredTerminalReleasesProcessGroup(t *testing.T) {
|
||
p := New("agentcli")
|
||
sdkInst := sdk.New("agentcli", sdk.SDKConfig{
|
||
RegTool: func(string, sdk.ToolDef, sdk.ToolHandler) error { return nil },
|
||
RegStage: func(sdk.Stage, sdk.StageHandler) {},
|
||
RegAPI: func(string) error { return nil },
|
||
Settings: sdk.NewSettings("agentcli", nil),
|
||
})
|
||
sdkInst.SetIOInjector(&injectCapture{})
|
||
if err := p.Start(sdkInst); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
defer p.Stop()
|
||
|
||
// 命令设计要点（缺一不可）：
|
||
// 1) 让 sh 保留为父进程、另起孙进程（`... & wait`）——单个 `sleep 300`
|
||
// 会被 sh 直接 exec 掉，只有一个进程，测不到「孙进程逃逸」；
|
||
// 2) 孙进程显式忽略 SIGHUP——否则关 PTY master 时内核发的 SIGHUP 会
|
||
// 顺手把它带走，于是「只杀 leader」也能通过，测不出进程组 Kill 的必要性。
|
||
// 两个条件合起来，只有给整个进程组发 SIGKILL 才能清干净。
|
||
marker := `(trap "" HUP; sleep 300) & wait`
|
||
res, err := p.handleCreate(sdkInst, map[string]interface{}{
|
||
"command": marker,
|
||
"timeout": "1s",
|
||
})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
m := res.(map[string]interface{})
|
||
if m["error"] != nil {
|
||
t.Skipf("PTY unavailable: %v", m["error"])
|
||
}
|
||
|
||
p.mu.Lock()
|
||
ts := p.sessions[m["id"].(string)]
|
||
p.mu.Unlock()
|
||
if ts == nil {
|
||
t.Fatal("terminal not registered")
|
||
}
|
||
pid := ts.cmd.Process.Pid
|
||
|
||
// 等超时被 readLoop 处理（含 Kill 进程组 + Wait 回收）
|
||
deadline := time.Now().Add(8 * time.Second)
|
||
for time.Now().Before(deadline) {
|
||
if !procAlive(pid) && !procGroupAlive(pid) {
|
||
return
|
||
}
|
||
time.Sleep(200 * time.Millisecond)
|
||
}
|
||
t.Fatalf("expired terminal not released: leader pid=%d alive=%v groupAlive=%v",
|
||
pid, procAlive(pid), procGroupAlive(pid))
|
||
}
|
||
|
||
func procAlive(pid int) bool { return syscall.Kill(pid, 0) == nil }
|
||
|
||
// procGroupAlive 检查「原进程组」里是否还有活着的成员（含被 init 收养的孙进程）。
|
||
//
|
||
// 不能用 kill(-pgid, 0)：组领头进程一死，内核就可能回收该 pgid，
|
||
// 即便组里还有被 reparent 的成员，这个探测也会失败。
|
||
// 改为直接遍历 /proc 查 pgid 匹配的活进程。
|
||
func procGroupAlive(pgid int) bool {
|
||
entries, err := os.ReadDir("/proc")
|
||
if err != nil {
|
||
return false
|
||
}
|
||
for _, e := range entries {
|
||
if pid, err := strconv.Atoi(e.Name()); err == nil {
|
||
if readPgid(pid) == pgid {
|
||
return true
|
||
}
|
||
}
|
||
}
|
||
return false
|
||
}
|
||
|
||
// readPgid 从 /proc/<pid>/stat 读进程组 id（第 5 个字段）。
|
||
// stat 的 comm 字段可能含空格/括号，所以从最后一个 ')' 之后再切分。
|
||
func readPgid(pid int) int {
|
||
data, err := os.ReadFile("/proc/" + strconv.Itoa(pid) + "/stat")
|
||
if err != nil {
|
||
return -1
|
||
}
|
||
s := string(data)
|
||
i := strings.LastIndex(s, ")")
|
||
if i < 0 {
|
||
return -1
|
||
}
|
||
fields := strings.Fields(s[i+1:])
|
||
if len(fields) < 3 {
|
||
return -1
|
||
}
|
||
// fields[0]=state, [1]=ppid, [2]=pgrp
|
||
pgid, err := strconv.Atoi(fields[2])
|
||
if err != nil {
|
||
return -1
|
||
}
|
||
return pgid
|
||
}
|