mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-11-01 14:57:38 +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 后缀
90 lines
2.9 KiB
Go
90 lines
2.9 KiB
Go
package plugin
|
||
|
||
// 内核控制通道的**授权边界**判据（2026-10-17）。
|
||
//
|
||
// ## 为什么这条判据必须存在
|
||
//
|
||
// 重启/关机通道能让**整个服务停下来**，而且关机后不会自己回来
|
||
// （guard 显式 systemctl stop 压制了 Restart=always）。
|
||
//
|
||
// 我做这个功能时第一版是"有就发"：`KernelControl: r.kernelCtl` 无条件
|
||
// 传给**所有**内置插件。那意味着任何一个内置插件（包括将来新加的）
|
||
// 都能一键关掉整个服务 —— 攻击面 = 全部内置插件。
|
||
//
|
||
// ⇒ 必须按插件名白名单授予，且这条边界要有判据钉住。
|
||
|
||
import (
|
||
"context"
|
||
"testing"
|
||
|
||
sdk "github.com/JianFeeeee/HologramAgent/internal/sdk"
|
||
)
|
||
|
||
// fakeKernelCtl 记录收到的请求（判据只关心"能不能拿到/调用"）。
|
||
type fakeKernelCtl struct {
|
||
restartCalls int
|
||
shutdownCalls int
|
||
}
|
||
|
||
func (f *fakeKernelCtl) RequestKernelRestart(reason string) error {
|
||
f.restartCalls++
|
||
return nil
|
||
}
|
||
func (f *fakeKernelCtl) RequestKernelShutdown(reason string) error {
|
||
f.shutdownCalls++
|
||
return nil
|
||
}
|
||
|
||
// ★★★ 判据：只有白名单里的插件能拿到内核控制通道。
|
||
func TestKernelControlOnlyForAllowlistedPlugins(t *testing.T) {
|
||
r := NewRegistry()
|
||
fake := &fakeKernelCtl{}
|
||
r.SetKernelControl(fake)
|
||
|
||
if got := r.kernelControlFor("webui"); got == nil {
|
||
t.Fatal("★ webui 应拿到内核控制通道（它是有鉴权的管理界面）")
|
||
}
|
||
// 典型的内置插件都不该拿到
|
||
for _, name := range []string{"cmd", "files", "memo", "healthcheck", "pluginmgr", "timer"} {
|
||
if got := r.kernelControlFor(name); got != nil {
|
||
t.Fatalf("★ 内置插件 %q 拿到了内核控制通道 —— "+
|
||
"任何插件都能关机整个服务，攻击面失控", name)
|
||
}
|
||
}
|
||
// 未知/外部插件名
|
||
if got := r.kernelControlFor("some_third_party"); got != nil {
|
||
t.Fatal("★ 未登记插件拿到了内核控制通道")
|
||
}
|
||
}
|
||
|
||
// ★★★ 判据：内核未装配控制通道时，白名单插件也只拿到 nil（不 panic）。
|
||
func TestKernelControlNilWhenNotWired(t *testing.T) {
|
||
r := NewRegistry()
|
||
// 故意不 SetKernelControl（模拟非 systemd 环境/未启用）
|
||
if got := r.kernelControlFor("webui"); got != nil {
|
||
t.Fatal("★ 通道未装配时不该返回非 nil")
|
||
}
|
||
}
|
||
|
||
// ★★★ 判据：SDK 侧 KernelControl() 未注入时返回 nil，且调用方判 nil 不 panic。
|
||
func TestSDKKernelControlIsNilSafe(t *testing.T) {
|
||
s := sdk.New("probe", sdk.SDKConfig{})
|
||
if kc := s.KernelControl(); kc != nil {
|
||
t.Fatal("未注入时 KernelControl() 应为 nil")
|
||
}
|
||
// 注入后能拿到
|
||
fake := &fakeKernelCtl{}
|
||
s2 := sdk.New("probe2", sdk.SDKConfig{KernelControl: fake})
|
||
if s2.KernelControl() == nil {
|
||
t.Fatal("注入后应能拿到")
|
||
}
|
||
if err := s2.KernelControl().RequestKernelRestart("t"); err != nil {
|
||
t.Fatalf("调用失败: %v", err)
|
||
}
|
||
if fake.restartCalls != 1 {
|
||
t.Fatalf("fake 未收到请求（%d）", fake.restartCalls)
|
||
}
|
||
}
|
||
|
||
var _ = context.Background
|