mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-11-06 01:07:35 +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:
@ -36,6 +36,7 @@ import (
|
||||
"fmt"
|
||||
"log"
|
||||
"path/filepath"
|
||||
"sync"
|
||||
|
||||
"github.com/JianFeeeee/HologramAgent/internal/config"
|
||||
"github.com/JianFeeeee/HologramAgent/internal/events"
|
||||
@ -44,6 +45,30 @@ import (
|
||||
sec "github.com/JianFeeeee/HologramAgent/internal/security"
|
||||
)
|
||||
|
||||
// 出参槽:Prepare 的幂等缓存(**装配器私有**,不对外暴露)。
|
||||
//
|
||||
// ★ 与已删除的 security.RuntimeGlobal() 的区别:
|
||||
//
|
||||
// · 那个是**公开访问器**,任何包都能取 ⇒ 成了绕过 kstate 的第二条通道
|
||||
// · 这个是**本模块私有**,只用于"Prepare 只跑一次",产物同时挂到
|
||||
// ks.Security —— 外部一律从 ks 取,没有第二条通道
|
||||
var (
|
||||
preparedMu sync.RWMutex
|
||||
prepared *sec.Runtime
|
||||
)
|
||||
|
||||
func preparedRuntime() *sec.Runtime {
|
||||
preparedMu.RLock()
|
||||
defer preparedMu.RUnlock()
|
||||
return prepared
|
||||
}
|
||||
|
||||
func setPreparedRuntime(rt *sec.Runtime) {
|
||||
preparedMu.Lock()
|
||||
prepared = rt
|
||||
preparedMu.Unlock()
|
||||
}
|
||||
|
||||
// Module 提供安全防火墙设施。
|
||||
type Module struct{}
|
||||
|
||||
@ -97,7 +122,16 @@ func init() { kernelmod.Register(Module{}) }
|
||||
//
|
||||
// 返回的 Runtime 里带着 Store 与 Gate(边界闸门)。
|
||||
func Prepare(dataDir string, cfgReg *config.ConfigRegistry) (*sec.Runtime, error) {
|
||||
if rt := sec.RuntimeGlobal(); rt != nil {
|
||||
// 幂等:Group 0 与模块 Init 都会调它,但只能装一次。
|
||||
//
|
||||
// ★ 为什么幂等状态放在**本模块**而不是 security 包(2026-11-04):
|
||||
// 原先靠 `security.RuntimeGlobal()` 判断"装过没有",那让 security 包
|
||||
// 持有一份进程级单例 —— 而单例绕开了内核统一挂载点
|
||||
// (kstate.State.Security),导致"哪一份才是权威"取决于装配顺序。
|
||||
//
|
||||
// 现在幂等状态是**装配器的私有状态**:它只保证 Prepare 只跑一次,
|
||||
// 产物交给内核挂到 ks.Security,其它模块一律从 ks 取。
|
||||
if rt := preparedRuntime(); rt != nil {
|
||||
return rt, nil
|
||||
}
|
||||
if cfgReg == nil {
|
||||
@ -107,7 +141,7 @@ func Prepare(dataDir string, cfgReg *config.ConfigRegistry) (*sec.Runtime, error
|
||||
// 显式关闭:装一个空 Runtime(所有闸门零成本透传),
|
||||
// 而不是留 nil 让调用点各自判断。
|
||||
rt := sec.NewDisabledRuntime()
|
||||
sec.SetRuntimeGlobal(rt)
|
||||
setPreparedRuntime(rt)
|
||||
log.Printf("[security] module disabled by config (core.security.enabled=false)")
|
||||
return rt, nil
|
||||
}
|
||||
@ -141,7 +175,12 @@ func Prepare(dataDir string, cfgReg *config.ConfigRegistry) (*sec.Runtime, error
|
||||
log.Printf("[security] loaded %d historical/extra key(s) (smooth rotation + embedded migration)", len(oldKeys))
|
||||
}
|
||||
|
||||
store, err := sec.InitGlobal(sec.Options{
|
||||
// ★ 2026-11-04:改用 sec.New + Load,不再走 InitGlobal(包级单例已删)。
|
||||
//
|
||||
// 变量库的唯一事实来源是**模块产出的 Runtime**(挂在 ks.Security 上),
|
||||
// 各消费方从自己的注入面取(AgentConfig.Security / ProviderManager /
|
||||
// plugin.Registry)。这条路径取代了原先"单例 + 各处 Global()"的双轨。
|
||||
store, err := sec.MustStore(sec.Options{
|
||||
Path: filepath.Join(dataDir, "security", "variables.json"),
|
||||
KeyFn: func() ([]byte, error) { return key, nil },
|
||||
OldKeys: oldKeys,
|
||||
@ -169,7 +208,7 @@ func Prepare(dataDir string, cfgReg *config.ConfigRegistry) (*sec.Runtime, error
|
||||
}
|
||||
|
||||
rt := sec.NewRuntime(store)
|
||||
sec.SetRuntimeGlobal(rt)
|
||||
setPreparedRuntime(rt)
|
||||
|
||||
// ★ 全局策略基线(部署方兵底,用户 2026-10-29 裁定选项 B)。
|
||||
//
|
||||
|
||||
Reference in New Issue
Block a user