Function onMessage

cfworker/worker.js:71–114

(data)

  const handshake = { parsed: false, remote: null, writer: null };

  const onMessage = async (data) => {
    const chunk = toUint8(data);
    if (!handshake.parsed) {
      const parsed = parseVLESSHeader(chunk, uuid);
      if (!parsed.ok) {
        console.log("vless header parse failed:", parsed.err);
        safeClose(ws);
        return;
      }
      handshake.parsed = true;

      // Open the upstream TCP connection (Cloudflare 'connect' API).
      const socket = await cfConnect(parsed.host, parsed.port);
      handshake.remote = socket;
      handshake.writer = socket.writable.getWriter();

      // VLESS server reply: version 0, addons length 0.
      ws.send(new Uint8Array([parsed.version, 0]).buffer);

      // Forward any payload that came after the header.
      if (parsed.payload.length > 0) {
        await handshake.writer.write(parsed.payload);
      }
      // Pipe server -> WS.
      socket.readable
        .pipeTo(new WritableStream({
          write(data) {
            if (ws.readyState === WS_READY_STATE_OPEN) {
              ws.send(data);
            }
          },
          close() { safeClose(ws); },
          abort() { safeClose(ws); },
        }))
        .catch(() => safeClose(ws));
      return;
    }
    // Subsequent WS messages → remote socket.
    try {
      await handshake.writer.write(chunk);
    } catch {
      safeClose(ws);
    }
  };

  ws.addEventListener("message", (ev) => { onMessage(ev.data); });
  ws.addEventListener("close", () => { try { handshake.remote && handshake.remote.close(); } catch {} });