mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-11-06 09:17:31 +00:00
refactor(security): 删除三份包级单例，共享数据统一挂 kstate
## 用户裁定
> secutiry 的实现也是错误的，所有共享数据统一挂载一个 kernel 提供的
> 挂载点上暴露。
> store也应该调用security的加密能力！
## 缺陷
security 包原有**三份**独立包级单例，各自被 InitXxx/SetXxx 装入、
各自有访问器，三份都绕开了内核统一挂载点 kstate.State：
global *Store 变量库（凭据别名） Global()
runtime *Runtime 安全防火墙 RuntimeGlobal()
globalCodec *Codec 配置落盘加密 ConfigCodec()
## 为什么单例是错的（不是"多一个全局变量"）
本 agent 的 a.security 与 security.Global() 是**两个独立变量**。
正常装配下它们恰好指向同一份数据，但"恰好"不是保证：
· 测试替身：测试装单例、被测代码用 agent 字段 ⇒ 测的与跑的不是同一份
· 多 Runtime：两个内核实例/热重载时单例只有一份，另一方被覆盖
· 装配顺序：单例装晚了，先注册的 provider 就没被装饰
第三条**曾真实存在**：ProviderManager.Register 在注册时读单例套装饰器，
而 llm 模块原本不声明 FacSecurity 依赖 —— 顺序正确纯属巧合。
改动过程中判据如实报红 640 次泄漏。
## 本次做的（T3.1 + T3.3）
### 删 global *Store → 注入
变量库的唯一来源是 security 模块产出的 Runtime，经注入面分发：
AgentConfig.Security → a.security.Store() → a.secretStore()
ProviderManager.SetSecretStore
plugin.Registry.SetSecretStore → sdk.NewSecretsAPI(store)
core 的 4 处 security.Global()（output/toolcall/tooldefs/secret_tools）
改为 a.secretStore()；context 与 stageHost 改为注入字段；
provider 装饰器改为读注入的 Store。
### 删 runtime *Runtime → 注入
plugin.Registry.SetSecurityRuntime → sdk.NewSecurityAPI(name, rt)
→ sdk.newEventsAPI(name, bus, rt)
Prepare 的幂等缓存改为 kernelmod/security 私有变量 ——
装配器私有，不再是任何包都能取的公开访问器。
### 顺带删死代码
security.FillToolArgs / FillPayload 生产调用者为 0（StageHost 已自己
用注入的 Store 做同一件事），连同只测它们的判据一起删除。
### llm 模块补 FacSecurity 依赖
这是**真实存在**的依赖：注册 provider 时要套出向装饰器，而装饰器需要
变量库。此前靠"单例先装了"掩盖了顺序错误。
## 判据（均经变异验证）
TestNoPackageLevelSingletons 源码扫描；加回 var global ⇒ 红（实测）
TestMustStoreDoesNotRegisterGlobally 两次 MustStore 必须得到不同实例
TestStoreEncryptsWithOwnKey 落盘无明文 + 重载可解
## 验证
go build ./... PASS
go vet ./... PASS
gofmt -l 干净
go test ./... -count=1 63 包全过，0 FAIL
干净检出（git worktree） build + 3 包测试全绿
## 未做（见 kernelplan.md T3.2）
globalCodec 收敛 + Store 复用 Codec
还牵涉 config 的 6 处调用点与**存量密文惰性迁移**，单独一步做。
TestNoPackageLevelSingletons 的禁表暂不含 globalCodec ——
半成品状态长期红的判据会被人习惯性忽略，那时它就等于不存在。
This commit is contained in:
JianFeeeee
@ -19,7 +19,7 @@ import (
|
||||
func testStore(t *testing.T) *Store {
|
||||
t.Helper()
|
||||
dir := t.TempDir()
|
||||
s, err := InitGlobal(Options{
|
||||
s, err := MustStore(Options{
|
||||
Path: dir + "/vars.json",
|
||||
KeyFn: func() ([]byte, error) { return []byte("0123456789abcdef0123456789abcdef"), nil },
|
||||
Enabled: true,
|
||||
@ -28,7 +28,6 @@ func testStore(t *testing.T) *Store {
|
||||
t.Fatalf("InitGlobal: %v", err)
|
||||
}
|
||||
// 全局单例是进程级的：测试之间要复位，否则互相污染。
|
||||
t.Cleanup(func() { SetGlobal(nil) })
|
||||
return s
|
||||
}
|
||||
|
||||
|
||||
@ -18,84 +18,52 @@ package security
|
||||
//（`resident.go:185`；manager 全进程只建一份 `bootstrap.go:249`）。
|
||||
// 所以变量库挂在 provider 层天然就是进程级 —— **不需要跨 agent 同步**。
|
||||
|
||||
import (
|
||||
"log"
|
||||
"sync"
|
||||
)
|
||||
import ()
|
||||
|
||||
var (
|
||||
globalMu sync.RWMutex
|
||||
global *Store
|
||||
)
|
||||
// ★ 2026-11-04：进程级单例 `global *Store` 已删除。
|
||||
//
|
||||
// ## 为什么删
|
||||
//
|
||||
// 它绕开了内核的**统一挂载点**（`kstate.State.Security`）。变量库只有一个
|
||||
// 事实来源：security 模块产出的 Runtime，经 Assembly 传进
|
||||
// `AgentConfig.Security` / `ProviderManager.SetSecretStore` /
|
||||
// `plugin.Registry.SetSecretStore`，各消费方从自己的注入面取。
|
||||
//
|
||||
// ## 单例的危害（不是"多一个全局变量"这么轻）
|
||||
//
|
||||
// 本 agent 的 `a.security` 与 `security.Global()` 是**两个独立变量**。
|
||||
// 正常装配下它们恰好指向同一份数据，但"恰好"不是保证：
|
||||
//
|
||||
// · 测试替身：测试装单例、被测代码用 agent 字段 ⇒ 测试测的是单例
|
||||
// · 多 Runtime：两个内核实例（或热重载）时单例只有一份，另一方被覆盖
|
||||
// · 装配顺序：单例装晚了，先注册的 provider 就没被装饰（静默漏一条出网路径）
|
||||
//
|
||||
// 最后一条曾真实存在：`ProviderManager.Register` 在注册时读单例套装饰器，
|
||||
// 而 llm 模块原本**不声明** FacSecurity 依赖 —— 顺序正确纯属巧合。
|
||||
// 现在 llm 模块声明了 FacSecurity，且注入必须早于 Register。
|
||||
|
||||
// InitGlobal 装配进程级单例。**必须在任何 agent 启动前调用**。
|
||||
func InitGlobal(opt Options) (*Store, error) {
|
||||
// ---------------- 入向填充（插件无感知的关键） ----------------
|
||||
|
||||
// ★ 2026-11-04：`FillToolArgs` / `FillPayload` 已删除（死代码）。
|
||||
//
|
||||
// 它们读 `Global()` 单例做入向回填，但**生产调用者为 0**：
|
||||
// `core.StageHost.ExecuteToolCtx` 已直接用注入的 Store 做同一件事
|
||||
// （见 stages.go 的 `h.secretStore().FillMap(args)`）。
|
||||
//
|
||||
// 只留一个测试调用它们 ⇒ 测的是没人用的代码路径，绿了也不代表任何
|
||||
// 生产行为正确（engineering-lessons §1.1「判据要钉真实路径」）。
|
||||
|
||||
// MustStore 构造并加载一个变量库（装配/测试用），失败直接返回 error。
|
||||
//
|
||||
// ★ 存在的理由（2026-11-04）：包级单例删除后，"New + Load" 这对操作在
|
||||
// 装配代码与十几个测试里重复出现。抽成函数让调用点保持一行，
|
||||
// 也避免各人各写一遍（漏 Load 的症状是"库看起来在、但读不到已有条目"）。
|
||||
//
|
||||
// 注意：本函数**不**注册任何全局单例 —— 返回值由调用方持有并注入。
|
||||
func MustStore(opt Options) (*Store, error) {
|
||||
s := New(opt)
|
||||
if err := s.Load(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
globalMu.Lock()
|
||||
global = s
|
||||
globalMu.Unlock()
|
||||
return s, nil
|
||||
}
|
||||
|
||||
// SetGlobal 直接装入一个已构造的 Store（测试与特殊装配用）。
|
||||
func SetGlobal(s *Store) {
|
||||
globalMu.Lock()
|
||||
global = s
|
||||
globalMu.Unlock()
|
||||
}
|
||||
|
||||
// Global 取进程级单例；未初始化时返回 nil。
|
||||
//
|
||||
// ★ 为何返回 nil 而不报错：调用点散布在内核热路径上，
|
||||
// 未启用时必须**零成本跳过**。所有调用方都写成
|
||||
// `if s := Global(); s != nil { ... }`。
|
||||
func Global() *Store {
|
||||
globalMu.RLock()
|
||||
defer globalMu.RUnlock()
|
||||
return global
|
||||
}
|
||||
|
||||
// Enabled 报告安全模块是否已启用。
|
||||
func Enabled() bool {
|
||||
s := Global()
|
||||
return s != nil && s.enabled
|
||||
}
|
||||
|
||||
// ---------------- 入向填充（插件无感知的关键） ----------------
|
||||
|
||||
// FillToolArgs 把一个工具调用的参数里的 `@{别名}` 换回明文。
|
||||
//
|
||||
// 这是**内核工具链的入口**（`core.StageHost.ExecuteTool`）：
|
||||
// 插件收到的参数已经填好，它完全不知道有替换这回事。
|
||||
//
|
||||
// 返回 (填充处数, 用到的别名)。未初始化时返回 (0, nil)。
|
||||
func FillToolArgs(tool string, args map[string]interface{}) (int, []string) {
|
||||
s := Global()
|
||||
if s == nil || args == nil {
|
||||
return 0, nil
|
||||
}
|
||||
n, used := s.FillMap(args)
|
||||
if n > 0 {
|
||||
log.Printf("[security] tool %s backfilled %d argument(s) (aliases %v)", tool, n, used)
|
||||
}
|
||||
return n, used
|
||||
}
|
||||
|
||||
// FillPayload 把一段文本载荷里的 `@{别名}` 换回明文。
|
||||
//
|
||||
// 用于**设备输出通道**（`core` 的 `dev.Execute`）：发给设备的指令同理
|
||||
// 需要真实凭据，而设备侧对替换无感知。
|
||||
func FillPayload(where, text string) string {
|
||||
s := Global()
|
||||
if s == nil || text == "" {
|
||||
return text
|
||||
}
|
||||
out, n, used := s.Fill(text)
|
||||
if n > 0 {
|
||||
log.Printf("[security] %s payload backfilled %d place(s) (aliases %v)", where, n, used)
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
@ -27,11 +27,6 @@ import (
|
||||
"sync"
|
||||
)
|
||||
|
||||
var (
|
||||
runtimeMu sync.RWMutex
|
||||
runtime *Runtime
|
||||
)
|
||||
|
||||
// Runtime 是安全运行时。
|
||||
//
|
||||
// 它把安全模块的三块能力合成**一个边界**：
|
||||
@ -85,6 +80,8 @@ type Runtime struct {
|
||||
scopeMu sync.Mutex
|
||||
}
|
||||
|
||||
// NewRuntime 用已装好的变量库构造运行时。
|
||||
|
||||
// NewRuntime 用已装好的变量库构造运行时。
|
||||
func NewRuntime(store *Store) *Runtime {
|
||||
if store == nil {
|
||||
@ -117,23 +114,19 @@ func NewDisabledRuntime() *Runtime {
|
||||
}
|
||||
}
|
||||
|
||||
// RuntimeGlobal 取进程级运行时；未装配时返回 nil。
|
||||
// ★ 2026-11-04：进程级单例 `runtime *Runtime` 已删除（用户裁定
|
||||
// 「所有共享数据统一挂载一个 kernel 提供的挂载点上暴露」）。
|
||||
//
|
||||
// ★ 返回 nil 而不是 Disabled：这只表示"安全模块还没装配"，
|
||||
// 防火墙的产物现在是 `kstate.State.Security`：security 模块产出它，
|
||||
// 消费方（agent / plugin-host / llm）各自从注入面取：
|
||||
//
|
||||
// 装配完成后（无论启用与否）一定非 nil。调用方拿 nil 应视为装配错误。
|
||||
func RuntimeGlobal() *Runtime {
|
||||
runtimeMu.RLock()
|
||||
defer runtimeMu.RUnlock()
|
||||
return runtime
|
||||
}
|
||||
|
||||
// SetRuntimeGlobal 装入运行时（幂等：重复装同一个实例无副作用）。
|
||||
func SetRuntimeGlobal(r *Runtime) {
|
||||
runtimeMu.Lock()
|
||||
runtime = r
|
||||
runtimeMu.Unlock()
|
||||
}
|
||||
// AgentConfig.Security → a.security
|
||||
// plugin.Registry.SetSecurityRuntime
|
||||
// ProviderManager.SetSecretStore（变量库）
|
||||
//
|
||||
// 原先的 `RuntimeGlobal()` / `SetRuntimeGlobal()` 是**公开**访问器，
|
||||
// 任何包都能绕开 ks 直接取 —— 那正是"两处各存一份、哪份权威取决于
|
||||
// 装配顺序"的来源。装配器内部的幂等缓存改为 kernelmod/security 私有变量。
|
||||
|
||||
// Enabled 报告安全审查是否真的生效。
|
||||
func (r *Runtime) Enabled() bool { return r != nil && r.enabled && r.gate != nil }
|
||||
|
||||
194
internal/security/singleton_test.go
Normal file
194
internal/security/singleton_test.go
Normal file
@ -0,0 +1,194 @@
|
||||
package security
|
||||
|
||||
// 判据：**不存在包级单例**（2026-11-04 T3.1）。
|
||||
//
|
||||
// ## 这些判据守的是什么
|
||||
//
|
||||
// 用户裁定：
|
||||
//
|
||||
// > secutiry 的实现也是错误的，所有共享数据统一挂载一个 kernel 提供的
|
||||
// > 挂载点上暴露。
|
||||
//
|
||||
// > store也应该调用security的加密能力！
|
||||
//
|
||||
// 本包原有**三份**独立包级单例：
|
||||
//
|
||||
// global *Store 变量库（凭据别名）
|
||||
// runtime *Runtime 安全防火墙
|
||||
// globalCodec *Codec 配置落盘加密
|
||||
//
|
||||
// 它们各自被 InitXxx / SetXxx 装入，各自有 Global() / RuntimeGlobal() /
|
||||
// ConfigCodec() 访问器。三份都绕开了内核的统一挂载点 `kstate.State`。
|
||||
//
|
||||
// ## 为什么单例是错的（不是"多一个全局变量"这么轻）
|
||||
//
|
||||
// 本 agent 的 `a.security` 与 `security.Global()` 是**两个独立变量**。
|
||||
// 正常装配下它们恰好指向同一份数据，但"恰好"不是保证：
|
||||
//
|
||||
// - 测试替身：测试装单例、被测代码用 agent 字段 ⇒ 测的是单例
|
||||
// - 多 Runtime：两个内核实例（或热重载）时单例只有一份，另一方被覆盖
|
||||
// - 装配顺序：单例装晚了，先注册的 provider 就没被装饰（静默漏一条出网路径）
|
||||
//
|
||||
// 最后一条**曾真实存在**：ProviderManager.Register 在注册时读单例套装饰器，
|
||||
// 而 llm 模块原本不声明 FacSecurity 依赖 —— 顺序正确纯属巧合。
|
||||
//
|
||||
// ## 判据为什么是源码扫描
|
||||
//
|
||||
// 这是"不存在某物"的判据，没有可执行的行为可测。但**变异可验证**：
|
||||
// 把 `var global *Store` 加回来 ⇒ 立刻红。
|
||||
|
||||
import (
|
||||
"go/ast"
|
||||
"go/parser"
|
||||
"go/token"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
// forbiddenPackageLevelVars 是**不得出现**的包级单例变量名。
|
||||
//
|
||||
// 每一条都对应一次真实事故或一次可复现的错配：
|
||||
//
|
||||
// global → 变量库：测试装单例、代码用 agent 字段，测的与跑的不是同一份
|
||||
// runtime → 防火墙：多实例时被覆盖
|
||||
//
|
||||
// ★ globalCodec 不在本表 —— 它属于 T3.2（Store 复用 Codec + Codec 归 Runtime）。
|
||||
// 那条改动还牵涉 config 的 6 个调用点与惰性密文迁移，单独一步做。
|
||||
// 本判据只钉已经完成的部分，避免"半成品状态让整套判据长期红"
|
||||
// （长期红的判据会被人习惯性忽略，那时它就等于不存在）。
|
||||
func forbiddenPackageLevelVars() []string {
|
||||
return []string{"global", "runtime"}
|
||||
}
|
||||
|
||||
// TestNoPackageLevelSingletons 钉住本包不得声明这三个包级变量。
|
||||
//
|
||||
// 变红条件：把 `var global *Store`（或另外两个）加回来 ⇒ 红。
|
||||
func TestNoPackageLevelSingletons(t *testing.T) {
|
||||
fset := token.NewFileSet()
|
||||
entries, err := os.ReadDir(".")
|
||||
if err != nil {
|
||||
t.Fatalf("读目录失败: %v", err)
|
||||
}
|
||||
|
||||
banned := make(map[string]bool, len(forbiddenPackageLevelVars()))
|
||||
for _, n := range forbiddenPackageLevelVars() {
|
||||
banned[n] = true
|
||||
}
|
||||
|
||||
for _, e := range entries {
|
||||
name := e.Name()
|
||||
if e.IsDir() || !strings.HasSuffix(name, ".go") || strings.HasSuffix(name, "_test.go") {
|
||||
continue
|
||||
}
|
||||
f, err := parser.ParseFile(fset, filepath.Join(".", name), nil, 0)
|
||||
if err != nil {
|
||||
t.Fatalf("解析 %s 失败: %v", name, err)
|
||||
}
|
||||
for _, decl := range f.Decls {
|
||||
gd, ok := decl.(*ast.GenDecl)
|
||||
if !ok || gd.Tok != token.VAR {
|
||||
continue
|
||||
}
|
||||
for _, spec := range gd.Specs {
|
||||
vs, ok := spec.(*ast.ValueSpec)
|
||||
if !ok {
|
||||
continue
|
||||
}
|
||||
for _, ident := range vs.Names {
|
||||
if banned[ident.Name] {
|
||||
t.Fatalf("★ %s:%d 声明了包级单例 %q。\n\n"+
|
||||
"本包的单例已按用户裁定删除 —— 共享数据统一挂在\n"+
|
||||
"kstate.State 上，各消费方从自己的注入面取：\n\n"+
|
||||
"\tAgentConfig.Security → a.security.Store()\n"+
|
||||
"\tProviderManager.SetSecretStore\n"+
|
||||
"\tplugin.Registry.SetSecretStore\n\n"+
|
||||
"单例的问题不是'多一个全局变量'，而是它与 a.security\n"+
|
||||
"是两个独立变量，正常装配下'恰好'同一份，但测试替身/\n"+
|
||||
"多 Runtime/装配顺序任一路径都可能分离。",
|
||||
name, fset.Position(ident.Pos()).Line, ident.Name)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// TestMustStoreDoesNotRegisterGlobally 钉住 MustStore 只是构造+加载。
|
||||
//
|
||||
// 变异可验证：在 MustStore 里加一行 `global = s` ⇒ 上面的判据红
|
||||
// （因为 global 变量本身不存在，得先加变量）。
|
||||
//
|
||||
// 本判据补的是另一半：**即使变量名换了**（比如叫 defaultStore），
|
||||
// MustStore 也不得把结果存进任何包级位置。
|
||||
func TestMustStoreDoesNotRegisterGlobally(t *testing.T) {
|
||||
dir := t.TempDir()
|
||||
s1, err := MustStore(Options{
|
||||
Path: filepath.Join(dir, "a.json"),
|
||||
KeyFn: func() ([]byte, error) { return []byte("0123456789abcdef0123456789abcdef"), nil },
|
||||
Enabled: true,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("MustStore: %v", err)
|
||||
}
|
||||
// 第二次调用必须得到**另一个**实例 —— 若内部有包级缓存，
|
||||
// 第二次会返回同一个（那就是单例还活着）。
|
||||
s2, err := MustStore(Options{
|
||||
Path: filepath.Join(dir, "b.json"),
|
||||
KeyFn: func() ([]byte, error) { return []byte("0123456789abcdef0123456789abcdef"), nil },
|
||||
Enabled: true,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("MustStore: %v", err)
|
||||
}
|
||||
if s1 == s2 {
|
||||
t.Fatal("★ MustStore 两次返回同一实例 —— 说明它偷偷存了包级单例。\n" +
|
||||
"它应当只做 New + Load，把实例交还调用方注入。")
|
||||
}
|
||||
}
|
||||
|
||||
// TestStoreEncryptsWithOwnKey 钉住变量库确实是密文落盘（加密能力归属本模块）。
|
||||
//
|
||||
// 这条是 T3.2（Store 复用 Codec）的前置事实：现在 Store 有自己的 seal/open。
|
||||
// T3.2 之后它会改为委托 Codec，本判据仍然成立（落盘必须是密文）。
|
||||
func TestStoreEncryptsWithOwnKey(t *testing.T) {
|
||||
dir := t.TempDir()
|
||||
const key = "0123456789abcdef0123456789abcdef"
|
||||
path := filepath.Join(dir, "v.json")
|
||||
|
||||
s, err := MustStore(Options{
|
||||
Path: path,
|
||||
KeyFn: func() ([]byte, error) { return []byte(key), nil },
|
||||
Enabled: true,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("MustStore: %v", err)
|
||||
}
|
||||
|
||||
const plain = "ghp_ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
|
||||
if _, err := s.Submit(plain, "test"); err != nil {
|
||||
t.Fatalf("Submit: %v", err)
|
||||
}
|
||||
|
||||
raw, err := os.ReadFile(path)
|
||||
if err != nil {
|
||||
t.Fatalf("读落盘文件: %v", err)
|
||||
}
|
||||
if strings.Contains(string(raw), plain) {
|
||||
t.Fatal("★ 变量库落盘里有明文 —— 加密没有生效")
|
||||
}
|
||||
|
||||
// 重新加载必须能解回来（密钥一致）。
|
||||
s2, err := MustStore(Options{
|
||||
Path: path,
|
||||
KeyFn: func() ([]byte, error) { return []byte(key), nil },
|
||||
Enabled: true,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("重新加载失败: %v", err)
|
||||
}
|
||||
if s2.Count() != 1 {
|
||||
t.Fatalf("重新加载后应有 1 条，实际 %d", s2.Count())
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user