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
2026-11-04 11:57:48 +08:00
parent cc0eb51286
commit 47afd67ca1
44 changed files with 819 additions and 322 deletions

View File

@ -42,19 +42,21 @@ func fwToolDef(name string) sdk.ToolDef {
}
}
func setupFirewall(t *testing.T) (*security.Store, string) {
func setupFirewall(t *testing.T) (*security.Runtime, string) {
t.Helper()
dir := t.TempDir()
store, err := security.InitGlobal(security.Options{
// ★ 2026-11-04:不再用包级单例。Runtime 是唯一来源,
// 由装配路径注入 Agent(AgentConfig.Security → a.security)。
s := security.New(security.Options{
Path: filepath.Join(dir, "vars.json"),
KeyFn: func() ([]byte, error) { return []byte("0123456789abcdef0123456789abcdef"), nil },
Enabled: true,
})
if err != nil {
if err := s.Load(); err != nil {
t.Fatal(err)
}
t.Cleanup(func() { security.SetGlobal(nil) })
return store, dir
// 返回 Runtime 而不是 Store:现在唯一注入面是 AgentConfig.Security。
return security.NewRuntime(s), dir
}
// ★★ 判据:插件工具收到的参数是**明文**(回填生效)。
@ -62,8 +64,9 @@ func setupFirewall(t *testing.T) (*security.Store, string) {
// 这是整个设计的核心目的:模型手上只有别名,插件拿到真凭据,
// 而**插件对替换完全无感知**。
func TestFirewall_ToolArgsFilledBeforePlugin(t *testing.T) {
store, _ := setupFirewall(t)
a, _, _ := newRootForResidents(t)
rt, _ := setupFirewall(t)
store := rt.Store()
a, _, _ := newRootWithSecurity(t, &countingProvider{}, rt)
// 插件工具:记录它实际收到的参数
var seenToken string
@ -101,8 +104,9 @@ func TestFirewall_ToolArgsFilledBeforePlugin(t *testing.T) {
// 若插错(比如 stage 之前),插件会看到明文;
// 若压根没插,设备会收到字面的 "@{sec_xxx}"。
func TestFirewall_DevicePayloadFilledBeforeExecute(t *testing.T) {
store, _ := setupFirewall(t)
a, _, _ := newRootForResidents(t)
rt, _ := setupFirewall(t)
store := rt.Store()
a, _, _ := newRootWithSecurity(t, &countingProvider{}, rt)
var delivered string
const outCh = "plugintest_secch"
@ -143,9 +147,11 @@ func TestFirewall_DevicePayloadFilledBeforeExecute(t *testing.T) {
// ★ 判据:出向拦截真的发生在真实 provider 调用上(用真实 ProviderManager)。
func TestFirewall_OutboundScrubbedOnRealProviderPath(t *testing.T) {
setupFirewall(t)
rt, _ := setupFirewall(t)
var captured string
mgr := agentAPI.NewProviderManager()
// ★ 注入必须先于 Register:装饰器在 Register 时套上。
mgr.SetSecretStore(rt.Store())
mgr.Register("probe", &fwCaptureProvider{name: "probe", capture: &captured})
_, err := mgr.Default().Chat(context.Background(), &agentAPI.CompletionRequest{
@ -167,7 +173,8 @@ func TestFirewall_OutboundScrubbedOnRealProviderPath(t *testing.T) {
// ★ 判据:磁盘上**没有明文**(加密落盘),且权限 0600。
func TestFirewall_NoPlaintextOnDisk(t *testing.T) {
store, dir := setupFirewall(t)
rt, dir := setupFirewall(t)
store := rt.Store()
if _, err := store.Submit(firewallToken, "落盘"); err != nil {
t.Fatal(err)
}
@ -195,7 +202,8 @@ func TestFirewall_NoPlaintextOnDisk(t *testing.T) {
// ★ 判据:**同明文合并**(同一次对话里多个通道用同一个 secret ⇒ 同一别名)。
func TestFirewall_SameSecretMergesAcrossChannels(t *testing.T) {
store, _ := setupFirewall(t)
rt, _ := setupFirewall(t)
store := rt.Store()
// 两处独立出现同一明文(模拟两个插件都用同一个 token)
alias1, _ := store.Submit(firewallToken, "通道A")
@ -214,8 +222,9 @@ func TestFirewall_SameSecretMergesAcrossChannels(t *testing.T) {
//
// 这是完整闭环:拦截 → 别名进上下文 → 模型用别名 → 回填 → 插件拿到明文。
func TestFirewall_FullRoundTrip(t *testing.T) {
store, _ := setupFirewall(t)
a, _, _ := newRootForResidents(t)
rt, _ := setupFirewall(t)
store := rt.Store()
a, _, _ := newRootWithSecurity(t, &countingProvider{}, rt)
var pluginSaw string
const toolName = "plugintest_roundtrip"