Files
Cyrene_v2/internal/runtime/trace/digest.go
T
AskaEth 003d12a712 feat(trace): M40 span 写侧脱敏 + 共享上游错误收敛 helper
stderr 的 JSONL 是无主人在场的落盘面,却在此前原样落工具结果(result)、
子代理回复(text/summary)与上游报文,等于一次 tail 就能拷走整段对话。
本切片交付两件地基:

1) 写侧脱敏(trace.JSONRecorder)
   - 新增 trace/span_props.go:safePropKeys 白名单与 projectSpanProps 投影,
     读侧(safeSpanProps)与写侧共用同一张表、同一个投影函数;
   - WriteSpan 落盘前经 trace.SanitizeRecord 投影:白名单键清洗,表外键只留
     Digest(bytes/runes/sha256 前缀,收进 omitted),事件先 RedactEvent 再把
     正文键折成 Digest。落到磁盘的只有元数据,正文一个字节都不留;
   - 新增 trace/digest.go:Digest/DigestOf/DigestOfValue,给「脱敏后仍要能诊断」
     一个构造上不含正文的取值形态(体积 + 指纹),两条投影都按类型放行它。

2) 共享上游错误体收敛 helper(给 M41/M42,禁止三家各写一份)
   - trace.UpstreamError(upstream, status, contentType, body):只保留 status /
     contentType / bodyLen / body sha256 前缀 / 上游 JSON 的 error.code /
     error.type / 截断 128 rune 的 error.message,其余一律丢弃,响应体原文绝不
     进 error 或日志;*UpstreamErrorInfo 提供 Error()/Attrs()/Unwrap()(errors.Is
     可判 ErrUpstream 走降级链);
   - trace.HostOf / trace.MaskQuery:日志里的 URL 只回来源,凭据类 query 与
     fragment 值打码为 ***,user:pass@ 整段打码。

3) 剥掉三处正文上 span
   - loop:tool span 不再放 result,改放 result_summary(与 tool_execution.end
     帧同源、已脱敏截断)与 result_digest;
   - orchestration:subagent span 不再放 text,subagent.done 事件不再放 summary,
     改记 status/steps/chars/sha256;
   - automations:AddRuleByName 不再 %v 整个 payload,改记 subject/type/bytes/sha256。

测试:以哨兵正文在落盘字节上断言(JSONL / slog 输出),并分别对「编排层收到的
原始 span」与「写侧投影」两层各钉一遍;UpstreamError 用回显 input 的 422 报文
与 HTML 错误页各钉一条;解析出/未解析出 JSON 都不回原文。以上均做过变异验证
(去掉保护即变红)。

验收:go build ./... + go vet ./... + go test -count=1 ./... 全绿;
trace/loop/orchestration 另跑 -race 全绿。
2026-10-26 13:17:42 +08:00

81 lines
2.9 KiB
Go

package trace
import (
"crypto/sha256"
"encoding/hex"
"encoding/json"
"fmt"
"unicode/utf8"
)
// maxDigestHex 是 Digest.SHA256 保留的十六进制位数(16 位 = 8 字节 / 64 bit)。
// 它足够把同一次运行里的两段不同内容区分开(相关性排查够用),又短到不可能据以
// 反推正文;再短的碰撞面就不值得了。
const maxDigestHex = 16
// Digest 是一个值的「体积 + 指纹」:字节数、rune 数、内容 sha256 前缀。
//
// 它是本包对「脱敏后还要留下可诊断线索」这一要求的答案 —— 正文被丢弃的地方
// (写侧 JSONL、span 上的结果/回复)留下一个 Digest,于是「这次调用的输出有多大、
// 两次是不是同一份内容」仍可判断,而正文本身不出现。
//
// 🔴 Digest **不含正文**,因此可以原样穿过本包的两条投影(出图投影与写侧投影)。
// 注意指纹的边界:sha256 只能证明「两份不同的内容指纹不同」,对**低熵**取值
// (例如一个短口令)它不构成保密承诺 —— 但它是截断前缀且不与正文同行,攻击者
// 拿不到「指纹 ↔ 明文」的对照面。
type Digest struct {
// Bytes 是取值编码后的字节数。
Bytes int `json:"bytes"`
// Runes 是字符串取值的 rune 数(非字符串取值为 0,随即省略)。
Runes int `json:"runes,omitempty"`
// SHA256 是内容 sha256 的十六进制前缀(maxDigestHex 位)。
SHA256 string `json:"sha256"`
}
// DigestOf 对字符串取摘要:字节数、rune 数与内容 sha256 前缀。空串给出
// {0 0 <sha256("")前缀>} —— 「真的没有内容」与「有内容但很小」因此可分辨。
func DigestOf(s string) Digest {
sum := sha256.Sum256([]byte(s))
return Digest{
Bytes: len(s),
Runes: utf8.RuneCountInString(s),
SHA256: hex.EncodeToString(sum[:])[:maxDigestHex],
}
}
// DigestOfValue 对任意值取摘要。字符串按原文取摘要(不走 JSON 编码,避免引号
// 把体积算大);其余值先 JSON 编码(与出图/摘要渲染同一口径),编码失败才回落
// %v —— 与 renderToolResultSummary 的取舍一致,使「体积」这个词在两侧同一含义。
func DigestOfValue(v any) Digest {
switch x := v.(type) {
case nil:
return Digest{}
case string:
return DigestOf(x)
case []byte:
return DigestOf(string(x))
default:
b, err := json.Marshal(x)
if err != nil {
return DigestOf(fmt.Sprintf("%v", x))
}
return DigestOf(string(b))
}
}
// digestFrom 报告 v 是否已经是一个 Digest —— 即一个「构造上就不含正文」的值。
// 两条投影据此原样放行它(再摘要一次只会得到「摘要的摘要」,纯属噪音)。
func digestFrom(v any) (Digest, bool) {
switch x := v.(type) {
case Digest:
return x, true
case *Digest:
if x == nil {
return Digest{}, false
}
return *x, true
default:
return Digest{}, false
}
}