Files
HomeAgent/internal/plugin/proc/execctx.go
JianFeeeee f5a1366770 fix(proc): arena 扩容块泄漏 + 取消成为一等状态
## 1. arena 泄漏(线上 arena 被打满的根因)

插件经 arena.alloc 申请的扩容块,块头 owner 记的是**插件的** ownerID,
而内核三处归还全写着 Free(OwnerHost, ref)。归属校验必然失败,错误又被
`_ =` 吞掉 ⇒ 每返回一次大结果就永久泄漏一块。

线上表现(2026-10-25):重任务跑几轮后 4MB arena 满,之后连 64KB 调用帧
都分不出来,所有插件工具一起报「arena: 空间不足(请求 65779 字节)」。

修法:三处改用插件自己的 ownerID,且失败必须留日志(吞掉错误等于把
下一次「空间不足」的原因提前销毁)。

## 2. 取消成为一等状态

旧实现里「调用在进行中」只表达为 pending 表里有这个键:取消时直接
delete,状态的痕迹全部消失,插件随后回的应答只能走「未知 id,丢弃」。
那是副作用的正确,不是规则的正确。

改成每次调用带显式状态机(callstate.go):

    pending ──应答──→ done
       ├──取消/超时──→ canceled ──迟到应答──→ done
       └──进程退出──→ failed

迁移表集中在 legalTransition,非法组合一律拒绝(CAS 只校验 from,
不校验 to——实测曾把 done 改回 failed)。

## 3. 取消跨进程送达

内核超时到点后只停止「等」,插件侧的 handler 还在跑,占着连接/子进程/
文件句柄,结果却已被丢弃。现在 CallContext 取消时向插件发 call.cancel
通知(fire-and-forget),插件运行时按 **RPC id** 定位并 cancel 对应调用的 ctx。

不按工具名定位:同一工具可并发多次,按名字会误伤其它实例。

插件作者零改动:handler 签名不变,取消由模板运行时接住。

## 4. 取消超时判定插件不稳定

严格区分两类超时(用户裁定):

    业务超时 = 任务跑太久     → 正常,不作为不稳定依据
    取消超时 = 喊停之后还不停 → 异常,是插件失去响应的直接证据

取消宽限(cancelGrace=5s)到期仍无应答 ⇒ ① 强制回收该调用名下的扩容块
(插件已静默退出,永远不会归还)② 计数 ③ 连续两次触发既有崩溃重启链。
插件一旦响应取消即清零;重启后也清零(判据是"连续"不是"累计")。

判据:8 条新增单测 + 全部变异验证(不回收/不上报/不清零/取消不跨进程
任一处改坏,对应用例立刻变红)。

Co-Authored-By: Claude <noreply@anthropic.com>
2026-10-25 10:10:21 +08:00

46 lines
1.5 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.

package proc
import "context"
// 执行上下文在 args 上的旁路键。
//
// ⚠️ 必须与 core.executeCtxArgKey **逐字一致**。
//
// 为什么复制一份常量而不是从 core 导入:proc 包被 core 依赖(StageHost
// 通过 sdk.ToolHandler 间接用到插件桥),反向导入会成环。这个键只是
// 两包之间的一个私有约定,复制 + 判据(proc 侧的空键行为测试)比拆包更轻。
//
// 一致性的真实风险很低:键里带不可见前缀与点号,正常工具参数不可能撞上;
// 而万一不一致,表现是"取消不生效"(工具照旧跑完),不是功能错误。
const execCtxArgKey = "\x00kernel.execute_ctx\x00"
// execCtxFromArgs 取出内核绑在本次调用上的 ctx;没有则返回 Background。
func execCtxFromArgs(args map[string]interface{}) context.Context {
if args != nil {
if ctx, ok := args[execCtxArgKey].(context.Context); ok && ctx != nil {
return ctx
}
}
return context.Background()
}
// stripExecCtx 返回去掉旁路键的副本,供下发给插件用。
//
// 插件可能对参数做严格校验,这个内核内部键不该出现在它眼里。
func stripExecCtx(args map[string]interface{}) map[string]interface{} {
if args == nil {
return nil
}
if _, ok := args[execCtxArgKey]; !ok {
return args
}
out := make(map[string]interface{}, len(args))
for k, v := range args {
if k == execCtxArgKey {
continue
}
out[k] = v
}
return out
}