352 lines
15 KiB
Go
352 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-rs 客户端的历史默认 originator，保留用于兼容识别。
|
|||
|
|
const CodexCLIOriginator = "codex_cli_rs"
|
|||
|
|
|
|||
|
|
// CodexDefaultOriginator 是网关默认使用的 Codex TUI originator。
|
|||
|
|
const CodexDefaultOriginator = "codex-tui"
|
|||
|
|
|
|||
|
|
// 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
|
|||
|
|
}
|