Skip to content

fix(server): keep a relayed WebSocket response head well-formed - #477

Merged
MagicalTux merged 2 commits into
KarpelesLab:masterfrom
devdotbo:upstream/websocket-crlf-fix
Sep 29, 2026
Merged

MagicalTux merged 2 commits into
KarpelesLab:masterfrom
devdotbo:upstream/websocket-crlf-fix

Conversation

@devdotbo

Copy link
Copy Markdown
Contributor

The gap

When upstream refuses a WebSocket handshake with a plain response, relayUpgrade writes the head by hand, wrapping the joined header lines in fixed CRLFs. If every header is filtered out (a refusal carrying only Connection and Content-Length), the client receives

HTTP/1.1 403 Forbidden\r\n\r\nConnection: close\r\n\r\n

The empty line after the status line ends the head, and Connection: close arrives as body bytes. Found by reading the code and reproduced with a raw socket client against a raw TCP upstream (the new test); not seen from a real client.

The 101 writer has the same shape: with no header lines it would write an extra CRLF that the client reads as the first WebSocket bytes. That case is latent, since Node emits 'upgrade' only when the response carries Upgrade and Connection.

The change

Both writers build the head from a line array with one terminator: [statusLine, ...headerLines, 'Connection: close'].join('\r\n') + '\r\n\r\n' for the refusal, the same without Connection: close for the 101. With headers present the bytes are identical. No config, API or status change.

Tests

test/upgrade-response-head.test.js drives relayUpgrade against a raw TCP upstream and compares the exact bytes the client receives, with a 5 s bound that fails with the bytes seen:

  • a refusal with only Connection: close and Content-Length: 0 arrives as HTTP/1.1 403 Forbidden\r\nConnection: close\r\n\r\n (fails on master without the fix);
  • a 101 with Upgrade and Connection arrives with one CRLF per header line and one empty line (passes before and after; pins the bytes).

The CI checks pass locally: npm test (2521 passing on Node 20.19.6, 22.21.1 and 24.12.0), npm run lint, npm run typecheck, and npm run typecheck:strict -- --base master (1703 strict diagnostics, same as master).

When upstream refuses a WebSocket handshake with a plain response,
relayUpgrade writes the refusal head to the client by hand. It joined
the surviving header lines into one string and wrapped it in fixed
CRLFs, so an empty header set produced a malformed head. On a refusal
whose only headers are Connection and Content-Length (both filtered)
the client got

  HTTP/1.1 403 Forbidden\r\n\r\nConnection: close\r\n\r\n

an empty head followed by "Connection: close" as body bytes.

The refusal writer now builds the head from a line array (status line,
header lines, Connection: close) joined with CRLF and ended by a single
empty line. The 101 writer is changed to the same serialization for
consistency. Its empty case is latent: Node only emits 'upgrade' when
the response carries Upgrade and Connection, so that writer always has
header lines today, and only a future transformation that removed every
header would expose it.

Tests assert the exact bytes for a refusal whose headers all drop and
for a 101, reading until the proxy closes the client socket.
@MagicalTux
MagicalTux merged commit d026840 into KarpelesLab:master Sep 29, 2026
5 checks passed
@MagicalTux MagicalTux mentioned this pull request Sep 29, 2026
MagicalTux added a commit that referenced this pull request Sep 29, 2026
Thirty-two commits since 1.1.21. Two change routing on an existing
config without an opt-in (#480, #481); the rest is opt-in, additive, or
display.

Behaviour changes
  #481 an API-key account's 401 is a cooldown, not a permanent `error`:
       1 min, then 5, 15 and 60 for every further rejection with no success
       in between; any 2xx/3xx resets it. The request still fails over and
       the client never sees the 401. OAuth accounts are unchanged
  #480 with session distribution on, requests carrying no session id stay
       within the top priority tier, so a fallback gateway no longer answers
       Claude Code's bootstrap and connector calls
  #470 a 200 whose SSE stream reports a provider failure before any output
       (`server_is_overloaded`, `response.failed`) fails over once, like a
       status-shaped failure would
  #465 a reload removes running accounts whose config entry is gone from
       disk, so `teamclaude remove` from another shell takes effect at once
  #460 `import` refuses an account whose token upstream has definitively
       rejected (401/403), even with `--name`; a 5xx or timeout still imports

Rename
  #483 the project is being renamed to TeamRouter (#72). This release accepts
       the new name everywhere the old one is read and changes nothing an
       install has on disk: `teamrouter` runs the same CLI, every
       `TEAMCLAUDE_*` variable is also read as `TEAMROUTER_*` (which wins when
       both are set), every `/teamclaude/…` control route also answers at
       `/teamrouter/…`, and `~/.config/teamrouter.json` is used when it exists

Features
  #441 per-account egress proxy (`accounts[].routing`: http, socks4/4a,
       socks5/5h) for refresh, probes and requests; `login --routing`,
       `teamclaude routing set/show/clear`, a connection check before it is
       relied on, and a short hold when the proxy is unreachable
  #427 `accounts[].allowExtraUsage: true` lets a paid extra-usage account
       serve once every account is past its threshold, instead of a 429
  #466 `accounts[].maxSpend`, a money cap judged against the month-to-date
       extra-usage spend upstream reports; the TUI shows what an account
       has billed
  #436 `autoRedeemResets` spends a free Codex rate-limit reset credit when
       the Codex pool runs dry (off by default)
  #482 `advisorEligibility: "strict" | "prefer"`; when the advisor model
       narrows selection to a subset of the fleet the log says so, and status
       carries the reading (`advisorNarrowing`)
  #478 `stripOverageHeaders` drops another org's per-organization billing
       headers from responses, for a pool spanning several orgs (#476)
  #471 `quota.unified5hSeenAt` / `unified7dSeenAt` in status: when upstream
       last stated each shared window
  #446 client and dimension usage for the last 5h and 24h in status and the
       dashboard, resumed across restarts
  #458 #459 #461 the dashboard sets the switch threshold, enables/disables and
       reprioritizes an account, and has a light theme remembered per browser
  #464 `l` in the TUI signs an account in `error` in again from the dashboard
  #457 status records which Codex limit meters each model
       (`quota.codexModelLimits`)
  #442 `quotaBarPercent` drops the percentage beside a TUI bar's countdown
  #451 `stripRequestFields` takes `content.<block type>` to drop content
       blocks a strict Anthropic-compatible upstream rejects
  #469 `proxy.mcp` schemas declare their item types, the write audit line
       records what happened, and the write queue has a depth (#447–#450)

Fixes
  #477 a refused WebSocket handshake whose headers all drop is relayed as a
       well-formed head instead of a blank line and body bytes
  #474 two members of one ChatGPT workspace are told apart by user id, so a
       second `login --codex` no longer replaces the first
  #469 a Codex Responses stream with no Content-Type is relayed as a stream
       and booked; thread repair on the global upstream; TUI settings and
       status gaps; a hint when a local login would have served
  #463 a Codex row with no session window draws one wide weekly bar

Tests
  #484 #485 #486 the suite asserts behaviour, not the scheduler: wall-clock
       upper bounds are gone, and subprocess tests spawn the server through
       `test-helpers/spawn-server.js`, which verifies the server it reached by
       `server.pid` (new in status) instead of trusting a port

Tooling
  #452 #453 #454 #455 docker workflow actions bumped
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants