fix(trace): 令牌守卫上移进 trace.UpstreamError,构造即安全
M48 复审的中危 finding:error.type 里的散文经 trace.UpstreamError 原样出镜,
因为守卫加在调用方(M46 的 model/voice 两处),构造函数自己不套守卫 ——
任何新调用点默认不安全,httpclient / vision / embedding / weixin 四条出口一直漏。
- trace.UpstreamError 构造时对 error.code / error.type 各走一次 ErrorToken
([A-Za-z0-9_.-]{<=64},不合规即空串):默认安全,不是可选加固。
- 构造后注入这条窄路收进 UpstreamErrorInfo.SetCode / SetType(同一道守卫),
供 DashScope 顶层 code 与 Anthropic 流内 error.type 两处使用 —— 调用方不再
各抄一份 ErrorToken。
- 删除因此冗余的调用方守卫:model.providerUpstreamError 与
voice.dashscopeOKBodyError。保留 voice 的 request_id 守卫(它不是 code/type,
构造函数看不见)。
- 四条漏的出口一行没改,只加用例证明守卫上移后自动安全;摘掉构造函数守卫时
四条用例 + model/voice 旧用例同时变红。
This commit is contained in:
1 parent
4e9cd17fbd
commit
fe3872ee28
10 files changed
+419
-17
No files matched your search
@@ -40,10 +40,10 @@ func upstreamLabel(name, baseURL string) string {
|
||||
// 而这类 error 会被 `chat run` / `loop step` 打日志。截断到 512 字节只是把
|
||||
// 「整段泄漏」变成「前 512 字节泄漏」,不是修复。
|
||||
//
|
||||
// error.code 与 error.type 也要过令牌守卫(trace.ErrorToken):它们在主流上游是枚举,
|
||||
// 但网关、代理与兼容层可以往里面塞任何东西 —— 包括回显的输入。不像令牌的字段一律变空
|
||||
// 串,代价只是少一条分类提示;status / body 摘要等诊断面不受影响。两个字段用同一道
|
||||
// 守卫(不按字段分别处理),是因为「同一份上游报文,两个出口口径不一致」正是下发漏点。
|
||||
// error.code 与 error.type 的令牌守卫(trace.ErrorToken)**不在这里**:它已上移进
|
||||
// trace.UpstreamError 的构造函数(M49),上游报文里不像令牌的字段在那里就变空串。
|
||||
// 调用点再守一遍是 M46 留下的重复 —— 「同一件事守两遍」的下一步永远是两处口径分叉,
|
||||
// 所以这里删掉,只保留「构造即安全」这一处真源。
|
||||
//
|
||||
// 双 %w:既保持 errors.Is(err, ErrUpstream)(降级链按它分类,主 → 备 → 本地),
|
||||
// 也让 errors.Is(err, trace.ErrUpstream) 成立(调用方按上游失败分类)。
|
||||
@@ -55,8 +55,6 @@ func providerUpstreamError(name, baseURL string, resp *http.Response) error {
|
||||
// helper 的构造是固定的,走不到这里;真走到也只少一条诊断,不破安全口径。
|
||||
return fmt.Errorf("model: %w: status %d", ErrUpstream, resp.StatusCode)
|
||||
}
|
||||
info.Code = trace.ErrorToken(info.Code)
|
||||
info.Type = trace.ErrorToken(info.Type)
|
||||
return fmt.Errorf("model: %w: %w", ErrUpstream, info)
|
||||
}
|
||||
|
||||
@@ -65,14 +63,18 @@ func providerUpstreamError(name, baseURL string, resp *http.Response) error {
|
||||
//
|
||||
// evt.Message 是上游自由文本,Anthropic 会用它回显请求内容(例如「prompt is too
|
||||
// long: <片段>」),所以**整段丢弃**(原本是无上限拼进 error)。只保留 evt.Type
|
||||
// (协议枚举,如 overloaded_error)且必须像令牌才出镜:网关把散文塞进 type 时它
|
||||
// 是空串 —— 宁可少一条分类提示,不让自由文本借道 type 出去。
|
||||
// (协议枚举,如 overloaded_error)。
|
||||
//
|
||||
// 🔴 这里的类型是**流内事件**,报文不在 trace.UpstreamError 的视野里(HTTP 已 200,
|
||||
// body 为空),所以它属于「构造后注入」这一族,必须走 SetType(与构造函数同一道
|
||||
// 令牌守卫):网关把散文塞进 type 时它是空串 —— 宁可少一条分类提示,不让自由文本
|
||||
// 借道 type 出去。
|
||||
func streamEventError(upstream string, evt *anthropicStreamError) error {
|
||||
info, ok := trace.UpstreamError(upstream, 0, "text/event-stream", nil).(*trace.UpstreamErrorInfo)
|
||||
if !ok {
|
||||
// helper 的构造是固定的,走不到这里;真走到也只少一条诊断,不破安全口径。
|
||||
return fmt.Errorf("model: %w: anthropic stream error event", ErrUpstream)
|
||||
}
|
||||
info.Type = trace.ErrorToken(evt.Type)
|
||||
info.SetType(evt.Type)
|
||||
return fmt.Errorf("model: %w: %w", ErrUpstream, info)
|
||||
}
|
||||
Reference in new issue
Block a user