mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-11-06 01:07:35 +00:00
81a0f5f598
## 一、插件声明层（用户裁定）
> 「敏感词策略也应该是插件声明啊，每个插件声明自己轮次以及自己工具的
> 敏感词策略」
SecurityProfile 新增两个字段：
ToolSensitiveWords map[string][]string // key=工具名模式(尾 *)，value=词表
DetectInjection bool // 开启内核内置注入迹象识别
语义是**追加不覆盖**：整轮 SensitiveWords 仍然生效。
取宽不取窄的理由：当前处置是 B 档（标记+警示，不阻断），
多标一条只是多一句警示，漏标则是安全缺口。
## 二、修掉我自己造的「莲藕式分层」
前一版把 *security.Round 从 runInputTask 一路搬到工具链
（prepareInputTask(evt, round) → TaskFrame.SecRound →
executeToolCallOutcomeWithRound），并在 rtchild 里另补一条。
那是把**分层策略实现成跨层传参**：每多一条工具执行路径就要求调用方
多传一次句柄。profileActivator 的设计注释早就写明了正路：
> 工具调用发生在 LLM 决定之后，此时插件已不在调用栈上。
> 唯一可靠的传递方式是**内核在输入进来时记住**，在该轮的工具判定里使用。
改为 security 自取：
Runtime.ScanContentForAgent(agentID, isChild, ins, text)
→ ForAgent(agentID, isChild) // per-agent 视图，天然不串档
→ EffectiveProfile() // session > longtime
→ 合并 全局词表 + 档位词表 + 工具词表 + DetectInjection
core 只递交**事实**（我是谁 / 哪个工具 / 什么文本），不认识 Round。
与 CheckToolForAgent 同一套身份推断，不引入第二个入口语义。
TaskFrame.SecRound / executeToolCallOutcomeWithRound 全部撤除。
## 三、目录名统一
third_party/homeagent-sdk → third_party/hologramagentsdk
module 与远端仓早已是 hologramagentsdk，只剩物理目录是遗留名。
同步改了 go.mod replace、构建/打包/部署脚本、5 处测试里的分段路径。
注意：这里改的是**主仓 vendored 的 SDK 源码路径**，example/ 由 SDK
独立仓自己跟踪，主仓不涉及。
## 判据
TestToolSensitiveWordsAppendOnlyForThatTool 词表只对目标工具生效 + 追加语义
TestToolSensitiveWordsUseAgentScope security 按 agent scope 自取生效
（变异验证：删掉 ForAgent 查询 → 红）
## 验证
go build ./... → 0
go vet ./... → 0
go test ./... -count=1 -race → 62 ok / 0 FAIL / 0 DATA RACE
gofmt -l internal/ → 干净
112 lines
5.2 KiB
Go
112 lines
5.2 KiB
Go
package sdk
|
||
|
||
// SecretsAPI 是插件访问**内核变量库**的接口。
|
||
//
|
||
// ## 为什么要有它（去重）
|
||
//
|
||
// 在此之前，commondev 插件自带了一整套加密存储（`secrets.json` +
|
||
// AES-256-GCM + 别名表 + `${ENV}` 解析），与内核的 `internal/security`
|
||
// **完全重复**：
|
||
//
|
||
// 内核 security.Store 插件 unifiedSecretFile
|
||
// 落盘 secrets.json secrets.json
|
||
// 加密 AES-256-GCM AES-256-GCM
|
||
// 别名 HMAC 自动派生 用户指定
|
||
// 引用语法 @{sec_xxx} @{name}
|
||
//
|
||
// 重复的代价不只是代码量：**两套变量库互不可见** —— 内核拦截时登记的
|
||
// 变量，插件看不到；插件登记的，内核拦截不认识。结果是同一个凭据可能
|
||
// 在两处各存一份，而使用者不知道哪个才是权威。
|
||
//
|
||
// ## ★ 安全设计：明文的边界
|
||
//
|
||
// 内核的硬约束 S1 是「不提供取回明文的接口」—— 理由是一旦 LLM 能取回，
|
||
// 它就能把明文写回上下文，整层防护作废。
|
||
//
|
||
// 那为什么插件可以拿到明文？因为**边界不在插件侧**：
|
||
//
|
||
// ① 内核在**插件 handler 之前**就把工具参数里的 @{别名} 填好了
|
||
// （`StageHost.ExecuteTool` → `security.FillMap`）—— 插件本来
|
||
// 就能从 args 里读到明文，这是设计使然（否则插件无法用凭据）。
|
||
// ② 明文若被插件**返回**给内核，会被两处拦回别名：
|
||
// · `ContextEvent` 的 scrub（记忆/历史侧路）
|
||
// · provider 装饰器（出向，LLM 永不看到）
|
||
//
|
||
// 两条都有判据钉住（`leakback_test.go`）。所以：
|
||
//
|
||
// **暴露 Fill 不新增泄漏面** —— 它不是「开了一个口子」，
|
||
// 而是把「本来就存在的能力」从各插件自己实现改为统一接口。
|
||
//
|
||
// ⚠️ 但绝不能暴露 `Reveal(alias) -> 明文` 这样的**按名字直取**接口：
|
||
// 那会让插件（以及通过插件操纵它的 LLM）枚举别名并逐个取值。
|
||
// 本接口刻意**没有**这样的方法 —— 见下面 Fill 与 List 的签名。
|
||
//
|
||
// ## 与 Encrypt/Decrypt 的区别
|
||
//
|
||
// 插件常把凭据存进自己的配置文件（如 pi 的 `auth.json`）。那些文件权限
|
||
// 0600 但**明文落盘**。`Encrypt`/`Decrypt` 提供 at-rest 加密，
|
||
// 密钥由内核统一管理，插件不必自持密钥（自持会让密钥散落、无法轮换）。
|
||
type SecretsAPI interface {
|
||
// Store 把一个凭据登记进内核变量库，返回其**别名**。
|
||
//
|
||
// 之后在工具参数里写 @{别名}，内核会在调用插件前换回真实值。
|
||
//
|
||
// ★ 只返回别名，不回显明文 —— 避免插件（或它转述给 LLM 的内容）
|
||
// 顺带把明文带出去。
|
||
//
|
||
// note 是给人看的用途说明（如「Gitea 推送令牌」）。
|
||
Store(value, note string) (alias string, err error)
|
||
|
||
// List 列出变量库的元信息（别名/来源/备注/命中次数）—— **不含明文**。
|
||
//
|
||
// ★ 刻意返回元信息而非值：调用方拿不到明文，只能引用。
|
||
List() []SecretInfo
|
||
|
||
// Forget 按别名删除一个变量。
|
||
//
|
||
// ★ 变量库**不自动淘汰**（用户明确）：泄漏控制靠主动删除，不靠超时。
|
||
// 所以凭据用完后应主动删。
|
||
Forget(alias string) error
|
||
|
||
// Fill 把一段文本里的 @{别名} 换成明文。
|
||
//
|
||
// 用途：插件需要把明文喂给**不经过内核参数回填**的地方 ——
|
||
// 例如 commondev 的 enc_shell 要把命令经 **stdin** 送给 bash
|
||
//（不是 argv，避免本机其他进程从 /proc/<pid>/cmdline 读到明文）。
|
||
//
|
||
// ★ 为什么这个方法是安全的（而不是"开了个取回明文的口子"）：
|
||
// 插件**本来就**能从 `args` 里拿到内核填好的明文 ——
|
||
// 内核在 handler 之前就填了。Fill 只是让插件能对**自己构造的文本**
|
||
// 做同样的事（如读文件内容、拼一段脚本）。
|
||
//
|
||
// 它**不是**按名字直取：要填谁必须先在文本里写出那个别名，
|
||
// 而别名本身不泄漏明文（HMAC 派生）。想枚举就得先 List，
|
||
// 而 List 只给元信息 —— 拿到别名后仍只能用，不能读回。
|
||
//
|
||
// 返回替换处数与用到的别名（便于插件做输出脱敏时的比对）。
|
||
Fill(text string) (out string, replaced int, aliases []string)
|
||
|
||
// Encrypt 加密一段字节用于插件自己落盘（密钥由内核管理）。
|
||
// 返回的串带版本前缀，可直接存文件。
|
||
Encrypt(plain []byte) (string, error)
|
||
|
||
// Decrypt 解密由 Encrypt 产生的串。
|
||
Decrypt(stored string) ([]byte, error)
|
||
|
||
// Available 报告变量库是否可用（内核未启用安全模块时为 false）。
|
||
//
|
||
// ★ 插件应据此**响亮失败**而不是静默降级 —— 例如
|
||
// 「密钥不可用 ⇒ 拒绝明文存储」。
|
||
Available() bool
|
||
}
|
||
|
||
// SecretInfo 是一条变量的元信息（**不含明文**）。
|
||
type SecretInfo struct {
|
||
Alias string `json:"alias"`
|
||
Source string `json:"source"` // intercepted | submitted
|
||
Note string `json:"note,omitempty"`
|
||
CreatedAt string `json:"created_at"`
|
||
UpdatedAt string `json:"updated_at,omitempty"`
|
||
Hits int `json:"hits,omitempty"`
|
||
}
|