mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-11-05 16:57:32 +00:00
## 用户裁定
> 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 ——
半成品状态长期红的判据会被人习惯性忽略,那时它就等于不存在。
56 lines
1.8 KiB
Go
56 lines
1.8 KiB
Go
package core
|
||
|
||
// 前提验证:插件 enc_shell 的**命令参数**是否已被内核填好明文?
|
||
//
|
||
// 若已填 ⇒ 插件自己的 lookupAnySecret 对内核别名是**死代码**,
|
||
// 且「保留 enc_shell」的理由只剩 argv 保护(脚本走 stdin)。
|
||
//
|
||
// 这决定了 SDK 该不该暴露 Fill():
|
||
// · 若插件本来就能从 args 拿到明文 ⇒ 暴露 Fill 不新增任何权限
|
||
// · 若拿不到 ⇒ 暴露 Fill 会削弱边界(插件可枚举 List 再逐个 Fill)
|
||
|
||
import (
|
||
"strings"
|
||
"testing"
|
||
|
||
sdk "github.com/JianFeeeee/HologramAgent/internal/sdk"
|
||
)
|
||
|
||
func TestPremise_EncShellCommandPrefilledByKernel(t *testing.T) {
|
||
rt := newScrubRuntime(t)
|
||
store := rt.Store()
|
||
a, _, _ := newRootWithSecurity(t, &countingProvider{}, rt)
|
||
|
||
const tok = "ghp_ABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
|
||
alias, err := store.Submit(tok, "测")
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
var cmdSaw string
|
||
if err := a.stageHost.RegisterTool("probe_encsh", sdk.ToolDef{
|
||
Name: "probe_encsh", Description: "模拟 enc_shell",
|
||
Parameters: map[string]interface{}{"type": "object"},
|
||
}, func(args map[string]interface{}) (interface{}, error) {
|
||
cmdSaw, _ = args["command"].(string)
|
||
return "ok", nil
|
||
}); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
if _, err := a.stageHost.ExecuteTool("probe_encsh", map[string]interface{}{
|
||
"command": "curl -H 'Authorization: token @{" + alias + "}'",
|
||
}); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
if strings.Contains(cmdSaw, "@{"+alias+"}") {
|
||
t.Fatal("handler 仍看到别名 ⇒ 内核没填,插件需要自己的填充逻辑")
|
||
}
|
||
if !strings.Contains(cmdSaw, tok) {
|
||
t.Fatalf("handler 应看到明文,实际 %q", cmdSaw)
|
||
}
|
||
t.Logf("✅ 内核已填 ⇒ 插件对**内核别名**的填充是死代码;" +
|
||
"插件仍需自己解析的是**平台凭据**(@ {gitea-local},含连字符,内核不认)")
|
||
}
|