mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-11-06 01:07:35 +00:00
fix(update): 重启交还 guard、OTA 支持 release/包两种来源、修三处真 bug
## 一、重启不再由 update 自己 systemctl(用户口径:应该通知 guard)
原 internal/update 直接 `systemctl restart homeagent`。三个问题:
① 杀掉整个 cgroup(unit 里 KillMode=control-group)⇒ **guard 跟着死**,
而 guard 存在的意义就是不让自己死。
② 丢失 exit 42 语义:guard 的 failures/failbackRound/failbackDone 全丢。
③ 自毁竞态:发命令的进程正是被重启杀掉的进程 ⇒ CombinedOutput 可能报错,
**一次成功的升级被误判失败而回滚**。
现在 internal/update 只到"产物就绪":预检 → 备份 → 写 pending → 原子替换。
重启由调用方走**既有**通道(restartCh → exit 42 → guard 用新二进制重拉):
· webui h.kernelCtl.RequestKernelRestart(本来就有,且是白名单内)
· pluginmgr p.sdk.KernelControl().RequestKernelRestart(见下面第三条)
· CLI 外部管理命令,restart unit 后按**版本自报**核对是否真生效
## 二、OTA 支持两种来源(release 自动更新 / 指定安装包)
新增 internal/update/source.go:
Source{Kind: FromPackage, Path} → 直接用本地包
Source{Kind: FromRelease} → 查最新 Release → 下载 → 校验 digest
ParseSource(pkg, release) 统一判空规则(pkg 优先),避免三端各写一套。
两条来源解析后进**同一条** Apply,不存在两份升级实现。
## 三、修三处已提交代码里的真 bug
① **HEAD 编译不过**:d17e261 提交了 kernelmod/update/mod.go(引用 ks.Update),
但 kernelboot/kernelstate.go 的字段改动从未提交。干净 worktree 实测
`ks.Update undefined` —— 推上去的仓库是坏的。本次补上。
② **kernelDataDir 算错一级**:Dir(Dir("/home/newqqagent/plugins")) = "/home",
而第 0 层 update 模块从 ks.DataDir=/home/newqqagent 读 pending
⇒ **agent 发起的升级一旦启动失败就永不回滚**(静默失效)。改为 Dir 一次。
③ **pluginmgr 的 KernelControl 恒为 nil**:它不在 kernelControlAllowed 里,
于是 `if kc := p.sdk.KernelControl(); kc != nil` 永不成立 ⇒
换完二进制**从不重启** ⇒ pending 留着 ⇒ 下次重启被判"未确认"而回滚。
修法**不是**把 pluginmgr 加进完整白名单 —— 那等于让 LLM 能关机整个服务
(kernel_control_gate_test 明确拦这个)。而是拆出**仅重启**授权:
kernelControlAllowed = {webui} 完整(含关机)
kernelRestartAllowed = {pluginmgr} 仅重启(shutdown 直接报错)
两者的危险程度差一个量级:重启几秒后 guard 自己拉回来,关机不会自己回来。
## 四、EventBus:插件请求更新不必 import internal/update
新增 events.EventUpdateRequest,pluginmgr 订阅并受理:
scope=kernel → kernelUpdateApply;scope=plugin → pluginUpdate(apply)。
它在 internal/plugins/webui 的 SSE 守卫里显式登记为**内核内部控制事件**
(不该推给浏览器:前端要的是 update_progress/update_result,不是"谁发了请求")。
## 五、顺手修掉的静默失败
· webui /kernel/update 原实现:**先**写 202 Accepted,**再**解析 body,
之后每个错误都直接 `return` —— 用户看到"已受理"然后什么都没发生。
改为同步 Apply(失败可如实报 HTTP 状态),成功后才请求重启。
· pluginmgr 拿不到控制通道时不再假装成功,而是回报 applied=true + 明确原因
(否则用户以为已升级,进程却还在跑旧镜像)。
· update 事件 publishResult 的 errMsg 原先是死代码(`_ = errMsg`),
按 payload 约定写回 error 字段。
This commit is contained in:
@ -44,13 +44,36 @@ func TestKernelControlOnlyForAllowlistedPlugins(t *testing.T) {
|
||||
if got := r.kernelControlFor("webui"); got == nil {
|
||||
t.Fatal("★ webui 应拿到内核控制通道(它是有鉴权的管理界面)")
|
||||
}
|
||||
// 典型的内置插件都不该拿到
|
||||
for _, name := range []string{"cmd", "files", "memo", "healthcheck", "pluginmgr", "timer"} {
|
||||
// 典型的内置插件都不该拿到**完整**通道(能关机)
|
||||
for _, name := range []string{"cmd", "files", "memo", "healthcheck", "timer"} {
|
||||
if got := r.kernelControlFor(name); got != nil {
|
||||
t.Fatalf("★ 内置插件 %q 拿到了内核控制通道 —— "+
|
||||
"任何插件都能关机整个服务,攻击面失控", name)
|
||||
}
|
||||
}
|
||||
|
||||
// ★★★ pluginmgr 只拿到**仅重启**通道(kernelRestartAllowed):
|
||||
//
|
||||
// 为什么需要它:kernel_update 已经把新二进制换好,不重启等于升级
|
||||
// 永远不生效,而 pending 标记会在下次启动时被判"未确认"并回滚。
|
||||
// 为什么不能给完整通道:关机后服务不会自己回来(压制 Restart=always),
|
||||
// 而重启几秒后 guard 会自己拉回来 —— 两者危险程度差一个量级。
|
||||
pm := r.kernelControlFor("pluginmgr")
|
||||
if pm == nil {
|
||||
t.Fatal("★ pluginmgr 应拿到**仅重启**通道(否则 kernel_update 换完二进制却不重启)")
|
||||
}
|
||||
if err := pm.RequestKernelShutdown("t"); err == nil {
|
||||
t.Fatal("★★★ pluginmgr 竟然能关机整个服务 —— 攻击面失控(只该授予重启)")
|
||||
}
|
||||
if fake.shutdownCalls != 0 {
|
||||
t.Fatalf("★ 关机请求被转发到底层通道(%d 次)—— 包装没有拦住", fake.shutdownCalls)
|
||||
}
|
||||
if err := pm.RequestKernelRestart("t"); err != nil {
|
||||
t.Fatalf("pluginmgr 应能重启: %v", err)
|
||||
}
|
||||
if fake.restartCalls != 1 {
|
||||
t.Fatalf("重启请求未到达底层通道(%d)", fake.restartCalls)
|
||||
}
|
||||
// 未知/外部插件名
|
||||
if got := r.kernelControlFor("some_third_party"); got != nil {
|
||||
t.Fatal("★ 未登记插件拿到了内核控制通道")
|
||||
|
||||
Reference in New Issue
Block a user