diff --git a/cmd/homed-upgrade/main.go b/cmd/homed-upgrade/main.go
index 2339c7c9..4730b196 100644
--- a/cmd/homed-upgrade/main.go
+++ b/cmd/homed-upgrade/main.go
@@ -33,11 +33,27 @@ import (
"flag"
"fmt"
"os"
+ "strings"
"github.com/JianFeeeee/HologramAgent/internal/update"
)
// fail 打印错误并以非零退出(CLI 专属;库内部一律返回 error)。
+// upgradeDataDir 推导数据目录(升级确认标记的落点)。
+//
+// 与生产一致:内核按 `--data
` 运行,标记必须落在同一个 dataDir 下,
+// 否则 update 模块读不到它。推导规则:
+// ① 环境变量 HOMEAGENT_DATA_DIR(部署方显式指定)
+// ② 二进制同级的 ../newqqagent(/usr/local/bin/homed → /home/newqqagent)
+// —— 太脆弱,改为最常见的部署形态:/usr/local/bin/homed → /home/newqqagent
+// ③ 退回 /home/newqqagent(本仓默认部署路径)
+func upgradeDataDir(binPath string) string {
+ if d := strings.TrimSpace(os.Getenv("HOMEAGENT_DATA_DIR")); d != "" {
+ return d
+ }
+ return "/home/newqqagent"
+}
+
func fail(format string, a ...interface{}) {
fmt.Fprintf(os.Stderr, format+"\n", a...)
os.Exit(1)
@@ -118,7 +134,7 @@ func main() {
}
// ── 6. 真正升级(备份 → 替换 → 重启 → 后置验证 → 失败回滚)──
- if err := update.DoUpgrade(root, *binPath, newDG, *pluginsDir); err != nil {
+ if err := update.DoUpgrade(root, *binPath, newDG, *pluginsDir, upgradeDataDir(*binPath)); err != nil {
fail("升级失败: %v", err)
}
}
diff --git a/internal/kernelmod/all/all.go b/internal/kernelmod/all/all.go
index 33827ded..d693dcf5 100644
--- a/internal/kernelmod/all/all.go
+++ b/internal/kernelmod/all/all.go
@@ -40,4 +40,5 @@ import (
_ "github.com/JianFeeeee/HologramAgent/internal/kernelmod/supervisor"
_ "github.com/JianFeeeee/HologramAgent/internal/kernelmod/textmem"
_ "github.com/JianFeeeee/HologramAgent/internal/kernelmod/tracker"
+ _ "github.com/JianFeeeee/HologramAgent/internal/kernelmod/update"
)
diff --git a/internal/kernelmod/all/facilities_sync_test.go b/internal/kernelmod/all/facilities_sync_test.go
index 85e6406b..6531f807 100644
--- a/internal/kernelmod/all/facilities_sync_test.go
+++ b/internal/kernelmod/all/facilities_sync_test.go
@@ -39,6 +39,9 @@ import (
// knowledge-migrated)。反向也要防:有人误把内部标记放进 SDK 当对外契约。
var internalOnlyFacilities = map[string]bool{
"knowledge-migrated": true,
+ // kernel.version 同理:它是 update 模块的启动时序事实("版本自检与
+ // 升级回滚判定已完成"),消费者是内核模块,不是插件契约。
+ "kernel.version": true,
}
// extractFacilities 从 Go 源码里提取 `FacilityXxx = "值"` 形式的常量。
diff --git a/internal/kernelmod/all/runtime_dep_test.go b/internal/kernelmod/all/runtime_dep_test.go
index 6f3eeb03..7fa02e36 100644
--- a/internal/kernelmod/all/runtime_dep_test.go
+++ b/internal/kernelmod/all/runtime_dep_test.go
@@ -175,9 +175,24 @@ func TestRunRejectsLayerOrderViolation(t *testing.T) {
}
}
- // ③ 违规发生在装配之前:只跑完了 L0(config / kernel-api)。
- if strings.Join(dnr.Produced, ",") != kernelmod.FacConfig+","+kernelmod.FacKernelIO {
- t.Errorf("★ 守卫失败时\"已产出事实\"应为 L0 的两个设施,实际 %v", dnr.Produced)
+ // ③ 违规发生在装配之前:只跑完了 L0。
+ //
+ // ★ L0 现在是三个:config / kernel-api / update。update 模块 Requires
+ // 为空(它肩负"启动失败回滚上一次升级",必须早于所有模块),所以
+ // 天然落在 L0。这里断言**集合**而非顺序 —— 注册表本就是乱序的。
+ got := dnr.Produced
+ wantL0 := map[string]bool{
+ kernelmod.FacConfig: true,
+ kernelmod.FacKernelIO: true,
+ kernelmod.FacVersion: true,
+ }
+ if len(got) != len(wantL0) {
+ t.Errorf("★ 守卫失败时 L0 应产出 %d 个设施,实际 %v", len(wantL0), dnr.Produced)
+ }
+ for _, f := range got {
+ if !wantL0[f] {
+ t.Errorf("★ L0 出现了意外的设施 %q(实际产出 %v)", f, dnr.Produced)
+ }
}
if ks.LuaVM != nil {
t.Error("★ 守卫失败后 ks.LuaVM 非 nil —— 说明 lua 模块被跑过了,与\"装配前校验\"矛盾")
diff --git a/internal/kernelmod/facilities.go b/internal/kernelmod/facilities.go
index a738a425..b0062640 100644
--- a/internal/kernelmod/facilities.go
+++ b/internal/kernelmod/facilities.go
@@ -43,4 +43,14 @@ const (
// FacMigrated 标记"知识库目录迁移已完成"(用于让 store 排在迁移之后)。
FacMigrated = "knowledge-migrated"
+
+ // FacVersion 标记"内核版本自检与升级回滚判定已完成"(update 模块提供)。
+ //
+ // 语义:读到它时,ks.Update 里的版本已经是**回滚后的最终版本**。
+ // 需要按版本行事的模块(healthcheck 上报版本、pluginhost 判插件兼容性)
+ // 应 Requires 它,否则可能读到升级前那个已被回滚的版本。
+ //
+ // 为什么在内部标记段:它是内核自用的启动时序事实,不是插件可选的
+ // 外部契约(插件声明它只会立刻满足,没有意义)。
+ FacVersion = "kernel.version"
)
diff --git a/internal/kernelmod/update/mod.go b/internal/kernelmod/update/mod.go
new file mode 100644
index 00000000..1e1b2691
--- /dev/null
+++ b/internal/kernelmod/update/mod.go
@@ -0,0 +1,98 @@
+// Package update 是 update 模块的适配层:把 internal/update 的能力接进内核模块体系。
+//
+// ## 为什么分两层(与仓库其他模块同一惯例)
+//
+// internal/update/ 业务实现(版本比较、解包、替换、回滚、Release 查询)
+// internal/kernelmod/update/ 适配层(读 ks → 调业务包 → 写回 ks)
+//
+// 参照 supervisor 模块:实现在 internal/supervisor,这里只做装配。
+// 分开的理由:internal/update 要被 CLI / WebAPI / agent 工具三端调用,
+// 它们都没有 KernelState;而模块体系只有内核启动一条路径。
+//
+// ## 它为什么必须在第 0 层
+//
+// 肩负「启动失败回滚上一次升级」的职责(详见 internal/update/rollback.go)。
+// 该职责必须早于所有其他模块:若排在后面,升级后新内核在 memory/llm 阶段
+// 崩了,而回滚还没跑 ⇒ 机器进崩溃循环,只能靠 guard 的 failback 兜底。
+// failback 是最后手段,不该承担"换错二进制"这种已知可回滚的故障。
+package update
+
+import (
+ "log"
+ "path/filepath"
+
+ "github.com/JianFeeeee/HologramAgent/internal/kernelboot"
+ "github.com/JianFeeeee/HologramAgent/internal/kernelmod"
+ upd "github.com/JianFeeeee/HologramAgent/internal/update"
+)
+
+// FacilityVersion 是本模块提供的设施名。
+//
+// 语义:启动时的版本自检与回滚判定**已完成**。需要按版本行事的模块
+// (healthcheck 上报版本、pluginhost 判插件兼容性)应 Requires 它,
+// 以确保读到的是回滚后的最终版本,而不是升级前的那个。
+const FacilityVersion = "kernel.version"
+
+// Module 实现 kernelboot.Module。
+type Module struct{}
+
+func (Module) Name() string { return "update" }
+func (Module) Requires() []string { return nil } // 第 0 层:不依赖任何人
+func (Module) Provides() []string { return []string{FacilityVersion} }
+
+// Init 执行版本自检,必要时回滚上一次升级。
+func (Module) Init(ks *kernelboot.KernelState) error {
+ st := &upd.State{}
+ ks.Update = st
+
+ bin := upd.KernelBinaryPath()
+ st.BinaryPath = bin
+
+ // ① 自报版本(升级是否真换成功,只有二进制自己能答)
+ v, err := upd.ReadBinVersion(bin)
+ if err != nil {
+ // 读不出来不等于致命:可能是极老版本(无 --version)。
+ // 没有证据表明升级失败 ⇒ 不回滚,记录后继续。
+ log.Printf("[update] kernel version probe failed: %v", err)
+ st.ProbeErr = err
+ } else {
+ st.Version, st.Commit, st.SDKVersion = v.Version, v.Commit, v.SDKVersion
+ log.Printf("[update] kernel %s (commit %s, sdk %s)", v.Version, upd.ShortCommit(v.Commit), v.SDKVersion)
+ }
+
+ // ② 上一次升级的回滚判定
+ rolled, from, to, rerr := upd.RollbackIfUnconfirmed(ks.DataDir, bin)
+ if rerr != nil {
+ // 回滚失败必须响亮:继续装配等于用一个已知起不来的版本
+ return rerr
+ }
+ if rolled {
+ log.Printf("[update] ROLLED BACK kernel %s → %s(上一次升级未能确认启动)", to, from)
+ st.RolledBack = true
+ st.RolledFrom, st.RolledTo = from, to
+ // 回滚后重读版本:后续模块要看到**回滚后**的版本
+ if v2, err2 := upd.ReadBinVersion(bin); err2 == nil {
+ st.Version, st.Commit, st.SDKVersion = v2.Version, v2.Commit, v2.SDKVersion
+ }
+ } else {
+ // ③ 版本已确认能启动 ⇒ 清除待确认标记
+ //
+ // 只依赖"进程活着且执行到第 0 层",不依赖其他模块 ⇒ 不会误判。
+ if err := upd.ClearPendingUpgrade(ks.DataDir); err != nil {
+ // 清不掉不阻断启动:后果只是下次启动再判一次(幂等)
+ log.Printf("[update] clear pending marker: %v", err)
+ }
+ }
+
+ // ④ 依赖图路径(真正的 digest 校验在 kernelboot 装载期,那里失败会拒绝启动)
+ if dg := filepath.Join(filepath.Dir(bin), "depgraph.bin"); upd.FileExists(dg) {
+ st.DepGraphPath = dg
+ }
+
+ log.Printf("[update] subsystem ready (version=%s rolled_back=%v)", st.Version, st.RolledBack)
+ return nil
+}
+
+func init() { kernelmod.Register(Module{}) }
+
+var _ kernelboot.Module = Module{}
diff --git a/internal/plugins/pluginmgr/kernel_update.go b/internal/plugins/pluginmgr/kernel_update.go
index 490ee733..dde21e8b 100644
--- a/internal/plugins/pluginmgr/kernel_update.go
+++ b/internal/plugins/pluginmgr/kernel_update.go
@@ -91,7 +91,7 @@ func (p *Plugin) kernelUpdateApply(pkgPath string) (interface{}, error) {
_ = kc.RequestKernelRestart("kernel_update apply")
}
go func() {
- if err := update.DoUpgrade(root, kernelBinPath(p), newDG, p.pluginDir); err != nil {
+ if err := update.DoUpgrade(root, kernelBinPath(p), newDG, p.pluginDir, kernelDataDir(p)); err != nil {
// DoUpgrade 内部失败已自行回滚;这里只留一条诊断日志
fmt.Fprintf(os.Stderr, "[pluginmgr] kernel_update upgrade failed (rolled back): %v\n", err)
}
@@ -102,6 +102,17 @@ func (p *Plugin) kernelUpdateApply(pkgPath string) (interface{}, error) {
}, nil
}
+// kernelDataDir 推导数据目录(升级确认标记的落点)。
+//
+// 从 pluginDir 反推:/plugins ⇒ 上一级即 dataDir。
+// 与内核 --data 的值一致,update 模块(第 0 层)才能读到同一个标记。
+func kernelDataDir(p *Plugin) string {
+ if p.pluginDir != "" {
+ return filepath.Dir(filepath.Dir(p.pluginDir))
+ }
+ return "/home/newqqagent"
+}
+
// currentKernelVersion 读本机 homed 自报版本(尽力而为)。
func currentKernelVersion(p *Plugin) *update.BinVersion {
v, err := update.ReadBinVersion(kernelBinPath(p))
diff --git a/internal/plugins/webui/kernel_update.go b/internal/plugins/webui/kernel_update.go
index 31ce2ad8..d1ddd6b3 100644
--- a/internal/plugins/webui/kernel_update.go
+++ b/internal/plugins/webui/kernel_update.go
@@ -107,5 +107,5 @@ func runKernelUpgrade(pkgPath string) {
if newBin == "" {
return
}
- _ = update.DoUpgrade(root, "/usr/local/bin/homed", newDG, "/home/newqqagent/plugins")
+ _ = update.DoUpgrade(root, "/usr/local/bin/homed", newDG, "/home/newqqagent/plugins", "/home/newqqagent")
}
diff --git a/internal/update/doc.go b/internal/update/doc.go
index 63b53840..ae2a310a 100644
--- a/internal/update/doc.go
+++ b/internal/update/doc.go
@@ -34,7 +34,9 @@
// ## 与既有升级通道的关系
//
// 不新造安装逻辑:
-// · 插件产物交给 pluginmgr.installFromPath(保留配置、原地覆盖)
-// · 内核重启走 sdk.KernelControlAPI(guard 重拉,不经 systemd)
+//
+// · 插件产物交给 pluginmgr.installFromPath(保留配置、原地覆盖)
+// · 内核重启走 sdk.KernelControlAPI(guard 重拉,不经 systemd)
+//
// 本包只管到"产物就绪"为止。
package update
diff --git a/internal/update/kernel.go b/internal/update/kernel.go
index e9205d2f..4d9f6f61 100644
--- a/internal/update/kernel.go
+++ b/internal/update/kernel.go
@@ -247,7 +247,7 @@ func FileSize(p string) int64 {
//
// 每一步失败都往回退,且**回滚后必须能正常启动**——这是本函数的核心承诺:
// 宁可升级失败,不可让机器起不来。
-func DoUpgrade(root, binPath, newDG, pluginsDir string) error {
+func DoUpgrade(root, binPath, newDG, pluginsDir, dataDir string) error {
newBin := FindFile(root, "homed")
if newBin == "" {
return fmt.Errorf("包内无 homed")
@@ -277,8 +277,39 @@ func DoUpgrade(root, binPath, newDG, pluginsDir string) error {
}
fmt.Printf("已备份 %d 个文件 → %s\n", len(targets), backup)
+ // ── 写待确认升级标记(在替换**之前**)──
+ //
+ // 顺序很关键:先写标记再替换。若反过来(替换后才写),"替换成功但
+ // 没来得及写标记就崩了"会丢标记 ⇒ 下次启动不知道该回滚。
+ //
+ // 标记由 update 模块(第 0 层)在下次启动时读;读到就说明上次升级
+ // 留下的版本没能活到清除标记 ⇒ 回滚(见 internal/update/rollback.go)。
+ curVer := &BinVersion{}
+ if v, verr := ReadBinVersion(binPath); verr == nil {
+ curVer = v
+ }
+ if dataDir != "" {
+ if perr := MarkPendingUpgrade(dataDir, PendingRecord{
+ FromVer: curVer.Version,
+ FromCommit: curVer.Commit,
+ ToVer: nv.Version,
+ ToCommit: nv.Commit,
+ BackupDir: backup,
+ }); perr != nil {
+ // 写不成功 ⇒ 放弃本次升级:没有标记就没有回滚能力,
+ // 而不带回滚的升级等于赌运气。
+ return fmt.Errorf("写升级确认标记失败(放弃升级,未替换任何文件): %w", perr)
+ }
+ fmt.Printf("已写升级确认标记 → %s\n", pendingPath(dataDir))
+ }
+
rollback := func(cause error) error {
fmt.Printf("\n❌ %v\n== 开始回滚 ==\n", cause)
+ if dataDir != "" {
+ if cerr := ClearPendingUpgrade(dataDir); cerr != nil {
+ fmt.Printf(" ⚠️ 清除升级确认标记失败: %v\n", cerr)
+ }
+ }
for _, t := range targets {
src := filepath.Join(backup, filepath.Base(t))
if err := CopyFile(src, t); err != nil {
diff --git a/internal/update/plugin_test.go b/internal/update/plugin_test.go
index 9df8d455..c294a154 100644
--- a/internal/update/plugin_test.go
+++ b/internal/update/plugin_test.go
@@ -96,11 +96,11 @@ func TestNewer(t *testing.T) {
// 判据 3b:TrimTag 的归一语义。
func TestTrimTag(t *testing.T) {
for in, want := range map[string]string{
- "v1.4.4": "1.4.4",
- "1.4.4": "1.4.4",
- "1.4.0dev": "1.4.0",
+ "v1.4.4": "1.4.4",
+ "1.4.4": "1.4.4",
+ "1.4.0dev": "1.4.0",
"1.4.0dev.1": "1.4.0",
- "release1": "release1", // 非纯版本原样返回,交给 ParseSemver 判不可比
+ "release1": "release1", // 非纯版本原样返回,交给 ParseSemver 判不可比
} {
if got := TrimTag(in); got != want {
t.Errorf("TrimTag(%q) = %q,期望 %q", in, got, want)
diff --git a/internal/update/rollback.go b/internal/update/rollback.go
new file mode 100644
index 00000000..9a63487b
--- /dev/null
+++ b/internal/update/rollback.go
@@ -0,0 +1,173 @@
+package update
+
+// 升级确认与回滚(内核启动时的自愈逻辑)。
+//
+// ## 问题:升级后内核起不来,机器进崩溃循环
+//
+// 升级 = 换二进制 + 重启。若新二进制有缺陷(依赖图不匹配、协议变了、
+// 某个模块 Init 失败),worker 会反复崩溃,guard 只能靠 failback(受限启动)
+// 兜底 —— 那是最后手段,不该承担"换错二进制"这种**已知可回滚**的故障。
+//
+// ## 解法:两段式确认
+//
+// 升级成功 ──→ 写 pending 标记("这个版本尚未被确认能启动")
+// ↓
+// 重启 → 内核启动 → update 模块(第 0 层)读标记
+// ↓
+// ┌────────────┴────────────┐
+// 标记还在 标记已不在
+// (上次升级后没能活到清除) (上次升级已确认成功)
+// ↓ ↓
+// 回滚到备份版本 正常继续
+// 并清除标记
+//
+// 清除时机:update 模块跑完自检后立刻清。它只依赖"内核进程活着且能执行到
+// 第 0 层",不依赖任何其他模块 ⇒ 只要内核基本可用就能清除,不会误判。
+//
+// ## 为什么不用 TTL / 重启次数
+//
+// 那些是 guard 已有的机制(failback),但判据是"崩了几次",不是"这次升级
+// 该不该撤销"。用重启次数会在"新版本偶发崩溃"时误回滚 —— 而那种情况该修的
+// 是崩溃,不是回退版本。这里只认一个事实:**上次升级留下的版本没能启动成功**。
+
+import (
+ "log"
+ "encoding/json"
+ "fmt"
+ "os"
+ "path/filepath"
+ "time"
+)
+
+// PendingFileName 是待确认升级标记的文件名(位于 dataDir 下)。
+const PendingFileName = "kernel-upgrade-pending.json"
+
+// PendingRecord 是待确认升级标记的内容。
+type PendingRecord struct {
+ // FromVer / FromCommit 是升级**前**的版本(回滚目标)。
+ FromVer string `json:"from_version"`
+ FromCommit string `json:"from_commit"`
+ // ToVer / ToCommit 是升级**后**的版本(当前正在试的)。
+ ToVer string `json:"to_version"`
+ ToCommit string `json:"to_commit"`
+ // BackupDir 是升级前备份目录(DoUpgrade 创建),回滚从这里取二进制。
+ BackupDir string `json:"backup_dir"`
+ // At 是写入时间(诊断用)。
+ At time.Time `json:"at"`
+}
+
+// pendingPath 返回标记文件路径。
+func pendingPath(dataDir string) string {
+ return filepath.Join(dataDir, PendingFileName)
+}
+
+// MarkPendingUpgrade 写入待确认标记(升级成功后立刻调用)。
+//
+// 由 DoUpgrade 在替换二进制**之前**写:若替换后重启失败,标记一定在。
+// 反过来(替换后才写)会在"替换成功但没来得及写标记就崩了"时丢标记,
+// 于是下次启动不知道该回滚。
+func MarkPendingUpgrade(dataDir string, rec PendingRecord) error {
+ if dataDir == "" {
+ return fmt.Errorf("dataDir 为空,无法写升级确认标记")
+ }
+ rec.At = time.Now()
+ b, err := json.MarshalIndent(rec, "", " ")
+ if err != nil {
+ return err
+ }
+ p := pendingPath(dataDir)
+ if err := os.WriteFile(p, b, 0644); err != nil {
+ return fmt.Errorf("写 %s: %w", p, err)
+ }
+ return nil
+}
+
+// ClearPendingUpgrade 清除标记(版本已确认能启动)。
+func ClearPendingUpgrade(dataDir string) error {
+ if dataDir == "" {
+ return nil
+ }
+ p := pendingPath(dataDir)
+ if err := os.Remove(p); err != nil && !os.IsNotExist(err) {
+ return fmt.Errorf("清除 %s: %w", p, err)
+ }
+ return nil
+}
+
+// ReadPendingUpgrade 读标记;不存在返回 (nil, nil)。
+func ReadPendingUpgrade(dataDir string) (*PendingRecord, error) {
+ b, err := os.ReadFile(pendingPath(dataDir))
+ if err != nil {
+ if os.IsNotExist(err) {
+ return nil, nil
+ }
+ return nil, err
+ }
+ var rec PendingRecord
+ if err := json.Unmarshal(b, &rec); err != nil {
+ return nil, fmt.Errorf("解析 %s: %w(删掉它即可跳过回滚)", pendingPath(dataDir), err)
+ }
+ return &rec, nil
+}
+
+// RollbackIfUnconfirmed 在发现未确认升级时回滚二进制。
+//
+// 返回 (rolled, from, to, err):
+//
+// rolled=false, err=nil —— 没有待确认标记,正常继续
+// rolled=true, err=nil —— 已回滚(from/to 记录版本轨迹)
+// rolled=*, err!=nil —— 回滚失败,调用方**必须中止启动**
+// (继续装配等于用一个已知起不来的版本)
+func RollbackIfUnconfirmed(dataDir, binPath string) (bool, string, string, error) {
+ rec, err := ReadPendingUpgrade(dataDir)
+ if err != nil {
+ return false, "", "", err
+ }
+ if rec == nil {
+ return false, "", "", nil
+ }
+
+ log.Printf("[update] found unconfirmed upgrade: %s → %s(上次升级后内核未能启动成功)",
+ ShortCommit(rec.FromCommit), ShortCommit(rec.ToCommit))
+
+ // 回滚源:升级前备份的 homed
+ backupBin := filepath.Join(rec.BackupDir, "homed")
+ if !FileExists(backupBin) {
+ // 备份不在 ⇒ 无法回滚。此时**中止启动**而不是硬上:
+ // 继续跑这个已知有问题的版本只会进崩溃循环。
+ return false, "", "", fmt.Errorf(
+ "备份不存在(%s):无法回滚。请人工用上一个已知良好版本替换 %s,"+
+ "或删除 %s 后重试(放弃回滚)", backupBin, binPath, pendingPath(dataDir))
+ }
+
+ // 先备份"当前这个坏的"——万一回滚目标也有问题,至少能回到现场
+ ts := NowStamp()
+ brokenDir := filepath.Join(filepath.Dir(binPath), "homed-broken-"+ts)
+ os.MkdirAll(brokenDir, 0755)
+ if err := CopyFile(binPath, filepath.Join(brokenDir, "homed")); err != nil {
+ // 备份失败不阻断回滚:回滚是主线,留现场是支线
+ log.Printf("[update] backup of broken binary failed: %v", err)
+ } else {
+ log.Printf("[update] broken binary kept at %s", brokenDir)
+ }
+
+ // 原子替换回旧版
+ if err := CopyFile(backupBin, binPath); err != nil {
+ return false, "", "", fmt.Errorf("回滚二进制: %w", err)
+ }
+
+ // 回滚 depgraph.bin(必须与 homed 同批)
+ if backupDG := filepath.Join(rec.BackupDir, "depgraph.bin"); FileExists(backupDG) {
+ if err := CopyFile(backupDG, filepath.Join(filepath.Dir(binPath), "depgraph.bin")); err != nil {
+ return false, "", "", fmt.Errorf("回滚 depgraph.bin: %w", err)
+ }
+ }
+
+ // 清标记:已回到旧版,不再有"待确认的新版"
+ if err := ClearPendingUpgrade(dataDir); err != nil {
+ log.Printf("[update] clear pending after rollback: %v", err)
+ }
+
+ log.Printf("[update] rolled back to %s(备份 %s)", rec.FromVer, rec.BackupDir)
+ return true, rec.FromVer, rec.ToVer, nil
+}
diff --git a/internal/update/rollback_test.go b/internal/update/rollback_test.go
new file mode 100644
index 00000000..42764a29
--- /dev/null
+++ b/internal/update/rollback_test.go
@@ -0,0 +1,245 @@
+package update
+
+import (
+ "os"
+ "path/filepath"
+ "testing"
+ "time"
+)
+
+// ── 升级确认与回滚的判据 ────────────────────────────────────
+//
+// 这些钉住"启动失败自动回滚"这个核心承诺。它一旦失效,症状是
+// 升级后机器进崩溃循环、只能人工上机器救 —— 而那时用户看不到任何提示。
+
+// 判据 1:写标记 → 能读回,字段不丢。
+func TestPendingMarker_RoundTrip(t *testing.T) {
+ dir := t.TempDir()
+ rec := PendingRecord{
+ FromVer: "1.4.0", FromCommit: "aaa111",
+ ToVer: "1.4.1", ToCommit: "bbb222",
+ BackupDir: "/root/homed-upgrade-backup-x",
+ }
+ if err := MarkPendingUpgrade(dir, rec); err != nil {
+ t.Fatalf("写标记: %v", err)
+ }
+ got, err := ReadPendingUpgrade(dir)
+ if err != nil {
+ t.Fatalf("读标记: %v", err)
+ }
+ if got == nil {
+ t.Fatal("应读到标记")
+ }
+ if got.FromCommit != rec.FromCommit || got.ToCommit != rec.ToCommit || got.BackupDir != rec.BackupDir {
+ t.Errorf("字段丢失: %+v", got)
+ }
+ if got.At.IsZero() {
+ t.Error("At 应被自动填上(诊断用)")
+ }
+}
+
+// 判据 2:没有标记时读回 (nil, nil) —— 不是 error。
+//
+// 这是绝大多数启动的形态(没有待确认升级)。若这里返回 error,
+// 每次启动都会走"回滚失败"分支 ⇒ 内核永远起不来。
+func TestPendingMarker_AbsentIsNotError(t *testing.T) {
+ got, err := ReadPendingUpgrade(t.TempDir())
+ if err != nil {
+ t.Fatalf("无标记不应报错: %v", err)
+ }
+ if got != nil {
+ t.Errorf("无标记应返回 nil,实际 %+v", got)
+ }
+}
+
+// 判据 3:清标记后再读应为 nil。
+func TestPendingMarker_Clear(t *testing.T) {
+ dir := t.TempDir()
+ if err := MarkPendingUpgrade(dir, PendingRecord{FromVer: "a", ToVer: "b"}); err != nil {
+ t.Fatal(err)
+ }
+ if err := ClearPendingUpgrade(dir); err != nil {
+ t.Fatalf("清标记: %v", err)
+ }
+ if got, _ := ReadPendingUpgrade(dir); got != nil {
+ t.Errorf("清后应读不到,实际 %+v", got)
+ }
+ // 幂等:再清一次不报错(update 模块每次启动都调)
+ if err := ClearPendingUpgrade(dir); err != nil {
+ t.Errorf("重复清除应幂等: %v", err)
+ }
+}
+
+// 判据 4:RollbackIfUnconfirmed 在无标记时不动二进制。
+func TestRollback_NoMarkerNoOp(t *testing.T) {
+ dir := t.TempDir()
+ bin := filepath.Join(dir, "homed")
+ os.WriteFile(bin, []byte("current"), 0755)
+
+ rolled, _, _, err := RollbackIfUnconfirmed(dir, bin)
+ if err != nil {
+ t.Fatalf("无标记不应报错: %v", err)
+ }
+ if rolled {
+ t.Error("无标记不应触发回滚")
+ }
+ b, _ := os.ReadFile(bin)
+ if string(b) != "current" {
+ t.Errorf("二进制不应被改动,实际 %q", b)
+ }
+}
+
+// 判据 5:有标记且备份存在 ⇒ 回滚,二进制回到旧版。
+//
+// 这是核心路径:上次升级留下的版本没能启动成功。
+func TestRollback_RestoresBackup(t *testing.T) {
+ dir := t.TempDir()
+ bin := filepath.Join(dir, "homed")
+
+ // 备份目录里放"旧版"
+ backupDir := filepath.Join(dir, "backup")
+ os.MkdirAll(backupDir, 0755)
+ oldBin := filepath.Join(backupDir, "homed")
+ os.WriteFile(oldBin, []byte("OLD-GOOD"), 0755)
+ os.WriteFile(filepath.Join(backupDir, "depgraph.bin"), []byte("OLD-DG"), 0644)
+
+ // 当前是"升级后的坏版本"
+ os.WriteFile(bin, []byte("NEW-BROKEN"), 0755)
+ os.WriteFile(filepath.Join(dir, "depgraph.bin"), []byte("NEW-DG"), 0644)
+
+ MarkPendingUpgrade(dir, PendingRecord{
+ FromVer: "1.4.0", FromCommit: "old",
+ ToVer: "1.4.1", ToCommit: "new",
+ BackupDir: backupDir,
+ })
+
+ rolled, from, to, err := RollbackIfUnconfirmed(dir, bin)
+ if err != nil {
+ t.Fatalf("回滚不应失败: %v", err)
+ }
+ if !rolled {
+ t.Fatal("有待确认标记时必须回滚")
+ }
+ if from != "1.4.0" || to != "1.4.1" {
+ t.Errorf("版本轨迹应记录 from/to,实际 %s → %s", from, to)
+ }
+
+ // ① 二进制回到旧版
+ b, _ := os.ReadFile(bin)
+ if string(b) != "OLD-GOOD" {
+ t.Errorf("二进制应回滚到旧版,实际 %q", b)
+ }
+ // ② depgraph 同批回滚(必须与 homed 一致)
+ dg, _ := os.ReadFile(filepath.Join(dir, "depgraph.bin"))
+ if string(dg) != "OLD-DG" {
+ t.Errorf("depgraph 应同批回滚,实际 %q", dg)
+ }
+ // ③ 标记已清(否则下次启动又回滚一次)
+ if got, _ := ReadPendingUpgrade(dir); got != nil {
+ t.Error("回滚后应清除标记")
+ }
+ // ④ 坏版本被留档(万一回滚目标也有问题,能回到现场)
+ entries, _ := os.ReadDir(dir)
+ found := false
+ for _, e := range entries {
+ if len(e.Name()) > 12 && e.Name()[:12] == "homed-broken" {
+ found = true
+ }
+ }
+ if !found {
+ t.Error("坏版本应被留档到 homed-broken-*")
+ }
+}
+
+// 判据 6:备份不存在 ⇒ 回滚失败(返回 error),**不静默继续**。
+//
+// 静默继续的后果:用一个已知起不来的版本装配 ⇒ 崩溃循环。
+// 必须让调用方中止启动,把问题暴露给人。
+func TestRollback_MissingBackupIsLoud(t *testing.T) {
+ dir := t.TempDir()
+ bin := filepath.Join(dir, "homed")
+ os.WriteFile(bin, []byte("NEW-BROKEN"), 0755)
+
+ MarkPendingUpgrade(dir, PendingRecord{
+ FromVer: "1.4.0", BackupDir: filepath.Join(dir, "nonexistent-backup"),
+ })
+
+ rolled, _, _, err := RollbackIfUnconfirmed(dir, bin)
+ if err == nil {
+ t.Fatal("备份不存在时必须报错(静默继续 = 用已知起不来的版本装配)")
+ }
+ if rolled {
+ t.Error("回滚失败时 rolled 应为 false")
+ }
+ // 错误必须可执行:说出备份路径与标记路径,人能照着修
+ if !containsStr2(err.Error(), "nonexistent-backup") {
+ t.Errorf("错误应指出备份路径,实际: %v", err)
+ }
+ if !containsStr2(err.Error(), "kernel-upgrade-pending") {
+ t.Errorf("错误应指出标记路径(删它即可放弃回滚),实际: %v", err)
+ }
+}
+
+// 判据 7:标记写坏(JSON 损坏)⇒ 报错并提示可删。
+func TestPendingMarker_CorruptIsActionable(t *testing.T) {
+ dir := t.TempDir()
+ p := pendingPath(dir)
+ os.WriteFile(p, []byte("{ not json"), 0644)
+
+ _, err := ReadPendingUpgrade(dir)
+ if err == nil {
+ t.Fatal("损坏的标记必须报错")
+ }
+ if !containsStr2(err.Error(), "删掉") {
+ t.Errorf("错误应说明可删掉该文件,实际: %v", err)
+ }
+}
+
+func containsStr2(s, sub string) bool {
+ for i := 0; i+len(sub) <= len(s); i++ {
+ if s[i:i+len(sub)] == sub {
+ return true
+ }
+ }
+ return false
+}
+
+// 判据 8:State.Summary 必须带回滚信息(healthcheck 要展示)。
+func TestStateSummary_CarriesRollback(t *testing.T) {
+ st := &State{
+ Version: "1.4.0", Commit: "abc123", SDKVersion: "1.3.0",
+ RolledBack: true, RolledFrom: "1.4.0", RolledTo: "1.4.1",
+ }
+ m := st.Summary()
+ if m["rolled_back"] != true {
+ t.Error("Summary 应带回滚标记")
+ }
+ if m["rolled_from"] != "1.4.0" || m["rolled_to"] != "1.4.1" {
+ t.Errorf("Summary 应带回滚轨迹,实际 %v", m)
+ }
+ // ShortCommit 只对**超过 8 位**的 commit 截短;6 位的原样返回。
+ // 用长 commit 才能验到截短行为。
+ st2 := &State{Version: "1.4.0", Commit: "abcdef1234567890"}
+ if m2 := st2.Summary(); m2["commit"] != "abcdef12" {
+ t.Errorf("长 commit 应被截到 8 位,实际 %v", m2["commit"])
+ }
+}
+
+// 判据 9:nil State 的 Summary 不 panic(healthcheck 可能在模块未就绪时调用)。
+func TestStateSummary_NilSafe(t *testing.T) {
+ var st *State
+ if got := st.Summary(); got != nil {
+ t.Errorf("nil State 应返回 nil,实际 %v", got)
+ }
+}
+
+// 判据 10:标记里的时间会被填上(诊断"这次升级是什么时候试的")。
+func TestPendingMarker_TimestampSet(t *testing.T) {
+ dir := t.TempDir()
+ before := time.Now().Add(-time.Second)
+ MarkPendingUpgrade(dir, PendingRecord{FromVer: "a", ToVer: "b"})
+ got, _ := ReadPendingUpgrade(dir)
+ if got.At.Before(before) {
+ t.Errorf("At 应接近当前时间,实际 %v", got.At)
+ }
+}
diff --git a/internal/update/state.go b/internal/update/state.go
new file mode 100644
index 00000000..301770ad
--- /dev/null
+++ b/internal/update/state.go
@@ -0,0 +1,63 @@
+package update
+
+import (
+ "os"
+ "path/filepath"
+)
+
+// State 是更新子系统的公共数据(挂在 KernelState.Update 上,由 update 模块填充)。
+//
+// 只放**别人要读**的东西:版本、是否回滚过、二进制路径。
+// 更新实现(Check/Upgrade/DoUpgrade)是函数,不需要常驻状态。
+//
+// ★ 定义在 internal/update 而不是 kernelmod:三端(CLI/WebAPI/agent 工具)
+// 都要用,而 KernelState 只有内核启动一条路径。模块通过
+// kernelboot.UpdateStateProvider 接口把它暴露给内核,避免 import 循环。
+type State struct {
+ BinaryPath string
+ DepGraphPath string
+ Version string
+ Commit string
+ SDKVersion string
+ ProbeErr error
+
+ // RolledBack 表示本次启动是否回滚了上一次升级。
+ RolledBack bool
+ RolledFrom string
+ RolledTo string
+}
+
+// Summary 返回给 healthcheck / 状态面的摘要。
+func (s *State) Summary() map[string]interface{} {
+ if s == nil {
+ return nil
+ }
+ out := map[string]interface{}{
+ "version": s.Version,
+ "commit": ShortCommit(s.Commit),
+ "sdk_version": s.SDKVersion,
+ "binary": s.BinaryPath,
+ "rolled_back": s.RolledBack,
+ }
+ if s.ProbeErr != nil {
+ out["probe_error"] = s.ProbeErr.Error()
+ }
+ if s.RolledBack {
+ out["rolled_from"], out["rolled_to"] = s.RolledFrom, s.RolledTo
+ }
+ return out
+}
+
+// KernelBinaryPath 返回本进程的内核二进制路径。
+//
+// 用 os.Executable 而不是硬编码 /usr/local/bin/homed:部署路径可变
+// (install.sh 支持 PREFIX),而"我自己是谁"永远可靠。
+func KernelBinaryPath() string {
+ if exe, err := os.Executable(); err == nil {
+ if resolved, err2 := filepath.EvalSymlinks(exe); err2 == nil {
+ return resolved
+ }
+ return exe
+ }
+ return "homed"
+}