shaw
c899c8cf37
面板「OpenAI Codex UA」此前只在客户端 UA 是浏览器型时才生效，本分支把它提升为 所有 OAuth 出站的规范身份来源，作用域扩大了一个数量级。而它的旧 placeholder 原文 就是 codex_cli_rs/0.144.1 (...)：照抄填写过的存量部署会被永久钉在 0.144.1 上—— 该值不低于上游门槛 0.144.0，version 头也随之变成 0.144.1，绕过版本自动同步， 稳定落在上游优先降载的那一侧，且面板、日志、审计都不提示。账号级自定义 UA （credentials.user_agent）有完全相同的陷阱。 改为：管理员配置的 UA 只贡献客户端名与 OS / 架构 / 终端指纹（这是该输入框唯一 不可替代的价值），版本段一律用当前生效版本重建。存量的陈旧配置无需迁移即自愈， UA 与 version 头从此由构造保证同源，原先「覆写 UA 陈旧时二者不一致」的次优解消失。 - 新增 openai.SetCodexUserAgentVersion：重建首段版本声明，OS / 架构 / 终端指纹 原样保留；尾部官方客户端标识组 (name; version) 与首段同源，一并更新，避免拼出 首段声明新版本、尾部仍是旧版本的自相矛盾身份；非官方括号组（如 OS 组）不动。 - resolveCodexOutboundIdentity 收敛为单一funnel：生效版本只从规范身份取， 候选 UA 的版本段不再参与判定。 - GetOpenAICodexCanonicalUserAgent 重建面板 UA 的版本段；原「面板值等于兜底常量 视同未填」的特例随之成为恒等变换，删除。 - 补齐 4 处静默丢弃账号级自定义 UA 的出站路径（WS 握手、账号测试 x2、用量探针）： 它们先写 customUA 再调不带 override 的收口，赋值随即被覆盖，既是行为不一致 也是死代码；账号测试尤其要紧——注释写着「与真实转发一致」而实际不一致。 - 补测试：SetCodexUserAgentVersion / CodexUserAgentVersion 包内用例、陈旧面板 UA 与账号 UA 的版本重建回归、WS 握手尊重账号级 UA。
350 lines
15 KiB
Go
350 lines
15 KiB
Go
package openai
|
||
|
||
import (
|
||
"regexp"
|
||
"strings"
|
||
)
|
||
|
||
// CodexCLIUserAgentPrefixes matches Codex CLI User-Agent patterns
|
||
// Examples: "codex_vscode/1.0.0", "codex_cli_rs/0.1.2"
|
||
var CodexCLIUserAgentPrefixes = []string{
|
||
"codex_vscode/",
|
||
"codex_cli_rs/",
|
||
}
|
||
|
||
// codexOfficialClientUAPrefixes：Codex 官方客户端家族 User-Agent 前缀（均含下划线/连字符，
|
||
// 每项都是确定字面量；不含会被 TrimSpace 退化成裸 "codex" 的空格前缀）。
|
||
// 用途：OpenAI OAuth `codex_cli_only` 访问限制判定 + passthrough 的「非官方 UA 安全兜底」
|
||
// （IsCodexOfficialClientRequest 命中即视为官方真实 UA，逐字透传、不改写）。
|
||
//
|
||
// Cursor/VSCode 扩展两种 UA：默认 `codex_vscode/`、GitHub Copilot 集成模式 `codex_vscode_copilot/`
|
||
// （取证 extension.js `IS="codex_vscode_copilot"` 经 env 注入）；交互式 TUI 自报 `codex-tui/`
|
||
// （连字符，2026-06-23 审计抽样约占真实流量 35%，必须显式列出）。`Codex Desktop/` 等 `Codex `
|
||
// 前缀家族由 codexOfficialClientFamilyPrefix 单独处理（保留空格，避免退化为裸 codex 的宽松兜底）。
|
||
var codexOfficialClientUAPrefixes = []string{
|
||
"codex_cli_rs/",
|
||
"codex-tui/",
|
||
"codex_vscode/",
|
||
"codex_vscode_copilot/",
|
||
"codex_app/",
|
||
"codex_chatgpt_desktop/",
|
||
"codex_atlas/",
|
||
"codex_exec/",
|
||
"codex_sdk_ts/",
|
||
}
|
||
|
||
// codexOfficialClientFamilyPrefix 覆盖 `Codex ` 前缀家族（Codex Desktop 等），对应 codex-rs
|
||
// is_first_party_originator 的 starts_with("Codex ")。**保留尾随空格**，并以 HasPrefix 直接比对
|
||
// 已归一化（小写 + 去首尾空格）的值——绝不能再经 normalizeCodexClientHeader 处理本前缀，否则
|
||
// 空格被 TrimSpace 去掉、退化成裸 "codex" 而把任何含 codex 的串都放行。
|
||
const codexOfficialClientFamilyPrefix = "codex "
|
||
|
||
// codexOfficialClientOriginators：Codex 官方客户端家族 originator 精确集合。
|
||
// app-server `initialize` 把 originator 设为 clientInfo.name 逐字值（codex-rs default_client.rs），
|
||
// 故官方集合是这些确定字面量；镜像 is_first_party_originator / is_first_party_chat_originator
|
||
// 并叠加 sub2api 已取证变体。用精确匹配而非「含 codex_/codex」的宽松兜底，避免 evil-codex_ 之类
|
||
// 伪造绕过（gate 仍需 UA 双因子佐证）。新官方/合作客户端经 allowed_client.go 命名预设放行，
|
||
// 或在 bump context/codex 时同步补入本集合。
|
||
var codexOfficialClientOriginators = map[string]bool{
|
||
"codex_cli_rs": true, // CLI 默认 DEFAULT_ORIGINATOR
|
||
"codex-tui": true, // 交互式 TUI（连字符，真实流量占比最高）
|
||
"codex_vscode": true, // VSCode/Cursor 扩展
|
||
"codex_vscode_copilot": true, // 扩展 GitHub Copilot 集成模式
|
||
"codex_app": true, // 历史保留
|
||
"codex_chatgpt_desktop": true, // is_first_party_chat_originator
|
||
"codex_atlas": true, // is_first_party_chat_originator
|
||
"codex_exec": true, // codex exec 非交互
|
||
"codex_sdk_ts": true, // TypeScript SDK
|
||
}
|
||
|
||
// IsCodexCLIRequest checks if the User-Agent indicates a Codex CLI request
|
||
func IsCodexCLIRequest(userAgent string) bool {
|
||
ua := normalizeCodexClientHeader(userAgent)
|
||
if ua == "" {
|
||
return false
|
||
}
|
||
return matchCodexClientHeaderPrefixes(ua, CodexCLIUserAgentPrefixes)
|
||
}
|
||
|
||
// IsCodexOfficialClientRequest checks if the User-Agent indicates a Codex 官方客户端请求。
|
||
// 与 IsCodexCLIRequest 解耦，避免影响历史兼容逻辑。宽松版：官方 UA 前缀集允许 Contains 子串兜底，
|
||
// 供 passthrough（IsCodexOfficialClientByHeaders）等历史路径使用，行为不变。
|
||
func IsCodexOfficialClientRequest(userAgent string) bool {
|
||
return isCodexOfficialClientRequest(userAgent, false)
|
||
}
|
||
|
||
// IsCodexOfficialClientRequestStrict 同 IsCodexOfficialClientRequest，但官方 UA 前缀集只做前缀
|
||
// 匹配（HasPrefix），不退化为 Contains 子串兜底——专供 codex_cli_only 访问门，收窄「浏览器前缀 +
|
||
// 中段 codex token」之类的伪造面。`Codex ` 家族前缀与 UA 尾部兜底保持一致；passthrough 仍用宽松版。
|
||
func IsCodexOfficialClientRequestStrict(userAgent string) bool {
|
||
return isCodexOfficialClientRequest(userAgent, true)
|
||
}
|
||
|
||
// isCodexOfficialClientRequest 匹配层级（优先级由高到低）：
|
||
// 1. UA 前缀集 codexOfficialClientUAPrefixes（strict=仅 HasPrefix；否则含 Contains 子串兜底）
|
||
// 2. `Codex ` 家族前缀（保留空格，避免退化为裸 codex）
|
||
// 3. UA 尾部兜底：codex-rs 把 clientInfo.name 写入 UA 末尾括号组 `(name; version)`。
|
||
// CODEX_INTERNAL_ORIGINATOR_OVERRIDE 只改前缀，不改尾部——可借此恢复被 override 的真实 client。
|
||
// 生产审计（10GB / 23 天）显示，originator=cccc 的真实 codex-tui 占全 openai 流量 5.3%，
|
||
// 若无此兜底则全部误拒。非官方尾部（如 evil/bash）仍被精确集拒绝。
|
||
func isCodexOfficialClientRequest(userAgent string, strict bool) bool {
|
||
ua := normalizeCodexClientHeader(userAgent)
|
||
if ua == "" {
|
||
return false
|
||
}
|
||
if strict {
|
||
if matchCodexClientHeaderStrictPrefixes(ua, codexOfficialClientUAPrefixes) {
|
||
return true
|
||
}
|
||
} else if matchCodexClientHeaderPrefixes(ua, codexOfficialClientUAPrefixes) {
|
||
return true
|
||
}
|
||
if strings.HasPrefix(ua, codexOfficialClientFamilyPrefix) {
|
||
return true
|
||
}
|
||
// UA 尾部兜底：提取最后一个括号组里的 name 段，用官方 originator 检测器判定。
|
||
if name := codexUATrailerName(ua); name != "" {
|
||
return IsCodexOfficialClientOriginator(name)
|
||
}
|
||
return false
|
||
}
|
||
|
||
// codexUATrailerName extracts the clientInfo.name from the last parenthesized group
|
||
// of a codex-rs formatted User-Agent: `{orig}/{ver} ({os}; {arch}) {term} ({name}; {ver})`.
|
||
//
|
||
// CODEX_INTERNAL_ORIGINATOR_OVERRIDE 修改 UA 前缀（originator 段），但不修改尾部的
|
||
// `(name; version)` 括号组——该组由 codex-rs engine 写入，保留真实 clientInfo.name。
|
||
// 故从尾部提取 name 可以恢复被 override 的真实客户端标识（例如 cccc → codex-tui）。
|
||
//
|
||
// input 应为去首尾空格的 UA；本函数本身大小写无关，大小写由调用方按需处理
|
||
// （isCodexOfficialClientRequest 传入已小写化的 UA 做匹配；PairCodexClientIdentity
|
||
// 传入原始大小写以保留 originator 的真实大小写）。
|
||
// 若无法解析则返回空字符串。
|
||
func codexUATrailerName(ua string) string {
|
||
last := strings.LastIndex(ua, "(")
|
||
if last < 0 {
|
||
return ""
|
||
}
|
||
rest := ua[last+1:]
|
||
closeIdx := strings.Index(rest, ")")
|
||
if closeIdx < 0 {
|
||
return ""
|
||
}
|
||
inner := strings.TrimSpace(rest[:closeIdx])
|
||
if semi := strings.Index(inner, ";"); semi >= 0 {
|
||
inner = strings.TrimSpace(inner[:semi])
|
||
}
|
||
return inner
|
||
}
|
||
|
||
// IsCodexOfficialClientOriginator checks if originator indicates a Codex 官方客户端请求。
|
||
// 精确集合匹配 + `Codex ` 家族前缀；不再用「含 codex」宽松兜底（避免伪造绕过）。
|
||
func IsCodexOfficialClientOriginator(originator string) bool {
|
||
v := normalizeCodexClientHeader(originator)
|
||
if v == "" {
|
||
return false
|
||
}
|
||
if codexOfficialClientOriginators[v] {
|
||
return true
|
||
}
|
||
return strings.HasPrefix(v, codexOfficialClientFamilyPrefix)
|
||
}
|
||
|
||
// IsCodexOfficialClientByHeaders checks whether the request headers indicate an
|
||
// official Codex client family request.
|
||
func IsCodexOfficialClientByHeaders(userAgent, originator string) bool {
|
||
return IsCodexOfficialClientRequest(userAgent) || IsCodexOfficialClientOriginator(originator)
|
||
}
|
||
|
||
func normalizeCodexClientHeader(value string) string {
|
||
return strings.ToLower(strings.TrimSpace(value))
|
||
}
|
||
|
||
func matchCodexClientHeaderPrefixes(value string, prefixes []string) bool {
|
||
for _, prefix := range prefixes {
|
||
normalizedPrefix := normalizeCodexClientHeader(prefix)
|
||
if normalizedPrefix == "" {
|
||
continue
|
||
}
|
||
// 优先前缀匹配；若 UA/Originator 被网关拼接为复合字符串时，退化为包含匹配。
|
||
if strings.HasPrefix(value, normalizedPrefix) || strings.Contains(value, normalizedPrefix) {
|
||
return true
|
||
}
|
||
}
|
||
return false
|
||
}
|
||
|
||
// matchCodexClientHeaderStrictPrefixes 仅前缀匹配（HasPrefix），不含 matchCodexClientHeaderPrefixes
|
||
// 的 Contains 子串兜底。用于 codex_cli_only 官方门收窄伪造面；passthrough 历史路径仍用宽松版。
|
||
// value 应为已归一化（小写 + 去首尾空格）的值。
|
||
func matchCodexClientHeaderStrictPrefixes(value string, prefixes []string) bool {
|
||
for _, prefix := range prefixes {
|
||
if p := normalizeCodexClientHeader(prefix); p != "" && strings.HasPrefix(value, p) {
|
||
return true
|
||
}
|
||
}
|
||
return false
|
||
}
|
||
|
||
// PairCodexClientIdentity 由最终出站 User-Agent 推导与其配套的 originator，必要时归一化
|
||
// UA 首段，保证两者一致。上游 /backend-api/codex 会校验 originator 与 UA 首段（首个 '/'
|
||
// 之前的 client 名）是否配套，错配（如 originator=codex_cli_rs + UA=codex-tui/...）一律
|
||
// 404（issue #3901，2026-07 实测）。
|
||
//
|
||
// 推导优先级：
|
||
// 1. UA 首段是官方 originator（精确集合或 `Codex ` 家族前缀）→ 直接配对，UA 原样保留；
|
||
// 2. UA 尾部括号组 `(name; version)` 的 name 是官方 originator——CODEX_INTERNAL_ORIGINATOR_OVERRIDE
|
||
// 只改 UA 前缀不改尾部（如 cccc/0.142.0 ... (codex-tui; 0.142.0)）→ 用尾部 name 重写
|
||
// UA 首段后配对，保留真实版本/OS/终端指纹；
|
||
// 3. 均不命中 → ok=false，调用方应整体回退为默认官方身份。
|
||
func PairCodexClientIdentity(userAgent string) (originator string, pairedUA string, ok bool) {
|
||
ua := strings.TrimSpace(userAgent)
|
||
slash := strings.IndexByte(ua, '/')
|
||
if slash <= 0 {
|
||
return "", "", false
|
||
}
|
||
if leading := strings.TrimSpace(ua[:slash]); isSaneCodexOriginator(leading) && IsCodexOfficialClientOriginator(leading) {
|
||
leading = canonicalizeCodexOriginator(leading)
|
||
return leading, leading + ua[slash:], true
|
||
}
|
||
// 传原始大小写 UA 提取 trailer，保留 `Codex ` 家族身份的真实大小写；含 '/' 的
|
||
// trailer 会破坏重写后 UA 首段与 originator 的一致性，直接拒绝。
|
||
if trailer := codexUATrailerName(ua); trailer != "" && !strings.ContainsRune(trailer, '/') &&
|
||
isSaneCodexOriginator(trailer) && IsCodexOfficialClientOriginator(trailer) {
|
||
trailer = canonicalizeCodexOriginator(trailer)
|
||
return trailer, trailer + ua[slash:], true
|
||
}
|
||
return "", "", false
|
||
}
|
||
|
||
// codexOriginatorMaxLen 官方 clientInfo.name 均为短 ASCII 标识，远低于此上限。
|
||
const codexOriginatorMaxLen = 64
|
||
|
||
// isSaneCodexOriginator 拒绝超长或含不可打印/非 ASCII 字节的候选 originator，
|
||
// 避免 `Codex ` 家族宽前缀把客户端可控的任意字节当作官方身份逐字转发给上游。
|
||
func isSaneCodexOriginator(name string) bool {
|
||
if name == "" || len(name) > codexOriginatorMaxLen {
|
||
return false
|
||
}
|
||
for i := 0; i < len(name); i++ {
|
||
if c := name[i]; c < 0x20 || c > 0x7e {
|
||
return false
|
||
}
|
||
}
|
||
return true
|
||
}
|
||
|
||
// canonicalizeCodexOriginator 把精确集合的官方 originator 大小写变体归一为规范小写形态
|
||
// （如 CODEX_CLI_RS → codex_cli_rs）；`Codex ` 家族不在精确集合中，保留原大小写
|
||
// （其规范形态本就是混合大小写，上游按大小写敏感 starts_with("Codex ") 判定）。
|
||
func canonicalizeCodexOriginator(name string) string {
|
||
if lower := normalizeCodexClientHeader(name); codexOfficialClientOriginators[lower] {
|
||
return lower
|
||
}
|
||
return name
|
||
}
|
||
|
||
// CodexCLIOriginator 官方 Codex CLI 默认 originator（codex-rs DEFAULT_ORIGINATOR），
|
||
// 也是身份归一化的目标身份。
|
||
const CodexCLIOriginator = "codex_cli_rs"
|
||
|
||
// CodexUserAgentVersion 提取 Codex UA 的完整版本段，即 `{client}/{version} (...` 中的 version。
|
||
// 与 ParseCodexEngineVersion 的区别：后者只取三段数字用于引擎版本比较（会丢掉 -alpha.4
|
||
// 之类的预发布后缀），本函数保留原样，因为出站 version 头必须与 UA 版本段逐字一致。
|
||
// 取不到（非 Codex 形态 UA）时返回空串。
|
||
func CodexUserAgentVersion(userAgent string) string {
|
||
ua := strings.TrimSpace(userAgent)
|
||
slash := strings.IndexByte(ua, '/')
|
||
if slash <= 0 {
|
||
return ""
|
||
}
|
||
rest := ua[slash+1:]
|
||
if space := strings.IndexByte(rest, ' '); space >= 0 {
|
||
rest = rest[:space]
|
||
}
|
||
return strings.TrimSpace(rest)
|
||
}
|
||
|
||
// SetCodexUserAgentVersion 用 version 重建 Codex 形态 UA 中的版本声明，其余部分
|
||
// （客户端名、OS / 架构 / 终端指纹）原样保留；UA 不是 `{client}/{version}` 形态时返回空串，
|
||
// 由调用方决定整体回退。
|
||
//
|
||
// 尾部官方客户端标识组 `(name; version)` 与首段是同一个版本声明的两个出口
|
||
// （CODEX_INTERNAL_ORIGINATOR_OVERRIDE 场景，如 `cccc/0.142.0 ... (codex-tui; 0.142.0)`），
|
||
// 必须一并更新，否则会拼出首段声明新版本、尾部仍是旧版本的自相矛盾身份。
|
||
// 仅在括号组确为官方客户端标识时才改写，避免误伤 OS 组（如 `(Ubuntu 22.4.0; x86_64)`）。
|
||
func SetCodexUserAgentVersion(userAgent, version string) string {
|
||
ua := strings.TrimSpace(userAgent)
|
||
version = strings.TrimSpace(version)
|
||
if version == "" {
|
||
return ""
|
||
}
|
||
slash := strings.IndexByte(ua, '/')
|
||
if slash <= 0 {
|
||
return ""
|
||
}
|
||
client := strings.TrimSpace(ua[:slash])
|
||
if client == "" {
|
||
return ""
|
||
}
|
||
rest := ua[slash+1:]
|
||
tail := ""
|
||
if space := strings.IndexByte(rest, ' '); space >= 0 {
|
||
tail = rest[space:]
|
||
} else if strings.TrimSpace(rest) == "" {
|
||
// `client/` 没有版本段，不是可重建的 Codex 形态。
|
||
return ""
|
||
}
|
||
return rewriteCodexUATrailerVersion(client+"/"+version+tail, version)
|
||
}
|
||
|
||
// rewriteCodexUATrailerVersion 把尾部官方客户端标识组 `(name; version)` 的版本改成 version。
|
||
// 括号组缺少 `;` 分隔的版本、或 name 不是官方 originator 时原样返回。
|
||
func rewriteCodexUATrailerVersion(ua, version string) string {
|
||
open := strings.LastIndex(ua, "(")
|
||
if open < 0 {
|
||
return ua
|
||
}
|
||
closeIdx := strings.Index(ua[open+1:], ")")
|
||
if closeIdx < 0 {
|
||
return ua
|
||
}
|
||
inner := ua[open+1 : open+1+closeIdx]
|
||
semi := strings.Index(inner, ";")
|
||
if semi < 0 {
|
||
return ua
|
||
}
|
||
name := strings.TrimSpace(inner[:semi])
|
||
if name == "" || !IsCodexOfficialClientOriginator(name) {
|
||
return ua
|
||
}
|
||
return ua[:open+1] + name + "; " + version + ua[open+1+closeIdx:]
|
||
}
|
||
|
||
// codexEngineVersionPattern 提取版本段开头的三段数字 X.Y.Z（忽略 -alpha 等后缀）。
|
||
var codexEngineVersionPattern = regexp.MustCompile(`^(\d+\.\d+\.\d+)`)
|
||
|
||
// ParseCodexEngineVersion 从 codex-rs 形态 UA 取引擎版本：
|
||
// `{originator}/{X.Y.Z} (...)`，第一个 '/' 后、首个空格或 '(' 前的三段版本。
|
||
// 该版本是 codex-rs CARGO_PKG_VERSION（引擎版本，CLI/app-server 一致）。
|
||
func ParseCodexEngineVersion(ua string) (string, bool) {
|
||
ua = strings.TrimSpace(ua)
|
||
slash := strings.IndexByte(ua, '/')
|
||
if slash < 0 {
|
||
return "", false
|
||
}
|
||
rest := ua[slash+1:]
|
||
end := len(rest)
|
||
for i := 0; i < len(rest); i++ {
|
||
if rest[i] == ' ' || rest[i] == '(' {
|
||
end = i
|
||
break
|
||
}
|
||
}
|
||
m := codexEngineVersionPattern.FindString(strings.TrimSpace(rest[:end]))
|
||
if m == "" {
|
||
return "", false
|
||
}
|
||
return m, true
|
||
}
|