@copilotkit/aimock 1.42.1 → 1.44.0

package/dist/ws-framing.cjs

@@ -10,7 +10,7 @@ let node_events = require("node:events");
 * Supports text frames, ping/pong, close handshake, and client frame unmasking.
 * Designed for a mock server — no extensions, no binary frames, no compression.
 */
-const WS_GUID = "258EAFA5-E914-47DA-95CA-5AB5DC799C07";
+const WS_GUID = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
 const OP_CONTINUATION = 0;
 const OP_TEXT = 1;
 const OP_CLOSE = 8;
@@ -21,16 +21,39 @@ var WebSocketConnection = class extends node_events.EventEmitter {
 	buffer = Buffer.alloc(0);
 	closed = false;
 	fragments = [];
-	constructor(socket) {
+	fragmentBytes = 0;
+	fragmented = false;
+	pendingWriteBytes = 0;
+	pendingWrites = /* @__PURE__ */ new Set();
+	limits;
+	constructor(socket, limits = {}) {
 		super();
+		for (const value of [
+			limits.maxMessageBytes,
+			limits.maxBufferedBytes,
+			limits.maxWriteBytes
+		]) if (value !== void 0 && (!Number.isSafeInteger(value) || value <= 0)) throw new RangeError("WebSocket limits must be positive safe integers");
+		this.limits = { ...limits };
 		this.socket = socket;
 		socket.on("data", (data) => {
-			this.buffer = Buffer.concat([this.buffer, data]);
-			this.parseFrames();
+			if (data.length > 0 && !this.closed) this.emit("activity");
+			let offset = 0;
+			while (offset < data.length && !this.closed) {
+				const room = (this.limits.maxBufferedBytes ?? Infinity) - this.fragmentBytes - this.buffer.length;
+				if (room <= 0) {
+					this.close(1009, "WebSocket buffer limit exceeded");
+					return;
+				}
+				const end = offset + Math.min(room, data.length - offset);
+				this.buffer = Buffer.concat([this.buffer, data.subarray(offset, end)]);
+				offset = end;
+				this.parseFrames();
+			}
 		});
 		socket.on("close", () => {
 			if (!this.closed) {
 				this.closed = true;
+				this.release(/* @__PURE__ */ new Error("WebSocket closed"));
 				this.emit("close", 1006, "Connection lost");
 			}
 		});
@@ -40,13 +63,60 @@ var WebSocketConnection = class extends node_events.EventEmitter {
 	}
 	send(data) {
 		if (this.closed) return;
-		const payload = Buffer.from(data, "utf-8");
-		this.writeFrame(OP_TEXT, payload);
+		this.sendAsync(data).catch(() => {});
+	}
+	sendAsync(data, signal) {
+		if (this.closed || this.socket.destroyed) return Promise.reject(/* @__PURE__ */ new Error("WebSocket closed"));
+		if (signal?.aborted) return Promise.reject(/* @__PURE__ */ new Error("WebSocket send aborted"));
+		const length = Buffer.byteLength(data);
+		const frameBytes = length + (length < 126 ? 2 : length < 65536 ? 4 : 10) + (this.limits.maskOutgoing ? 4 : 0);
+		if (frameBytes > (this.limits.maxWriteBytes ?? Infinity) - Math.max(this.pendingWriteBytes, this.socket.writableLength)) {
+			this.close(1009, "WebSocket write limit exceeded");
+			return Promise.reject(/* @__PURE__ */ new Error("WebSocket write limit exceeded"));
+		}
+		this.pendingWriteBytes += frameBytes;
+		return new Promise((resolve, reject) => {
+			let settled = false;
+			const finish = (error) => {
+				if (settled) return;
+				settled = true;
+				this.pendingWriteBytes -= frameBytes;
+				this.pendingWrites.delete(fail);
+				signal?.removeEventListener("abort", abort);
+				if (error) reject(error);
+				else resolve();
+			};
+			const fail = (error) => finish(error);
+			const abort = () => {
+				finish(/* @__PURE__ */ new Error("WebSocket send aborted"));
+				this.destroy();
+			};
+			this.pendingWrites.add(fail);
+			signal?.addEventListener("abort", abort, { once: true });
+			try {
+				this.socket.write(this.encodeFrame(OP_TEXT, Buffer.from(data)), (error) => {
+					if (error) finish(/* @__PURE__ */ new Error("WebSocket write failed"));
+					else finish();
+				});
+			} catch {
+				finish(/* @__PURE__ */ new Error("WebSocket write failed"));
+			}
+		});
+	}
+	release(error) {
+		this.buffer = Buffer.alloc(0);
+		this.fragments = [];
+		this.fragmentBytes = 0;
+		for (const fail of this.pendingWrites) fail(error);
 	}
 	close(code = 1e3, reason = "") {
 		if (this.closed) return;
 		this.closed = true;
-		const reasonBuf = Buffer.from(reason, "utf-8");
+		this.release(/* @__PURE__ */ new Error("WebSocket closed"));
+		const encodedReason = Buffer.from(reason, "utf-8");
+		let reasonEnd = Math.min(encodedReason.length, 123);
+		while (reasonEnd < encodedReason.length && (encodedReason[reasonEnd] & 192) === 128) reasonEnd--;
+		const reasonBuf = encodedReason.subarray(0, reasonEnd);
 		const payload = Buffer.alloc(2 + reasonBuf.length);
 		payload.writeUInt16BE(code, 0);
 		reasonBuf.copy(payload, 2);
@@ -59,40 +129,45 @@ var WebSocketConnection = class extends node_events.EventEmitter {
 	destroy() {
 		if (this.closed) return;
 		this.closed = true;
+		this.release(/* @__PURE__ */ new Error("WebSocket destroyed"));
 		if (!this.socket.destroyed) this.socket.destroy();
 		this.emit("close", 1006, "Connection destroyed");
 	}
 	get isClosed() {
 		return this.closed;
 	}
+	encodeFrame(opcode, payload) {
+		const length = payload.length;
+		const headerSize = length < 126 ? 2 : length < 65536 ? 4 : 10;
+		const maskSize = this.limits.maskOutgoing ? 4 : 0;
+		const frame = Buffer.allocUnsafe(headerSize + maskSize + length);
+		frame[0] = 128 | opcode;
+		frame[1] = (maskSize ? 128 : 0) | (length < 126 ? length : length < 65536 ? 126 : 127);
+		if (headerSize === 4) frame.writeUInt16BE(length, 2);
+		if (headerSize === 10) frame.writeBigUInt64BE(BigInt(length), 2);
+		if (maskSize) {
+			const mask = (0, node_crypto.randomBytes)(4);
+			mask.copy(frame, headerSize);
+			for (let i = 0; i < length; i++) frame[headerSize + 4 + i] = payload[i] ^ mask[i % 4];
+		} else payload.copy(frame, headerSize);
+		return frame;
+	}
 	writeFrame(opcode, payload) {
 		if (this.socket.destroyed) return;
-		const length = payload.length;
-		let header;
-		if (length < 126) {
-			header = Buffer.alloc(2);
-			header[0] = 128 | opcode;
-			header[1] = length;
-		} else if (length < 65536) {
-			header = Buffer.alloc(4);
-			header[0] = 128 | opcode;
-			header[1] = 126;
-			header.writeUInt16BE(length, 2);
-		} else {
-			header = Buffer.alloc(10);
-			header[0] = 128 | opcode;
-			header[1] = 127;
-			header.writeUInt32BE(0, 2);
-			header.writeUInt32BE(length, 6);
-		}
-		try {
-			this.socket.write(Buffer.concat([header, payload]));
-		} catch (err) {
-			if (!this.socket.destroyed) {
-				const msg = err instanceof Error ? err.message : String(err);
-				console.error(`[LLMock] Unexpected writeFrame error: ${msg}`);
+		const available = (this.limits.maxWriteBytes ?? Infinity) - Math.max(this.pendingWriteBytes, this.socket.writableLength);
+		const headerBytes = 2 + (this.limits.maskOutgoing ? 4 : 0);
+		if (opcode === OP_CLOSE && payload.length + headerBytes > available) {
+			if (available < headerBytes + 2) {
+				this.socket.destroy();
+				return;
 			}
+			payload = payload.subarray(0, 2);
+		} else if (payload.length + headerBytes > available) {
+			this.close(1009, "WebSocket write limit exceeded");
+			return;
 		}
+		const frame = this.encodeFrame(opcode, payload);
+		this.socket.write(frame);
 	}
 	parseFrames() {
 		while (this.buffer.length >= 2 && !this.closed) {
@@ -101,6 +176,17 @@ var WebSocketConnection = class extends node_events.EventEmitter {
 			const fin = (byte0 & 128) !== 0;
 			const opcode = byte0 & 15;
 			const masked = (byte1 & 128) !== 0;
+			if (byte0 & 112 || masked === Boolean(this.limits.maskOutgoing) || ![
+				OP_CONTINUATION,
+				OP_TEXT,
+				2,
+				OP_CLOSE,
+				OP_PING,
+				OP_PONG
+			].includes(opcode) || opcode >= 8 && (!fin || (byte1 & 127) > 125) || opcode === OP_CONTINUATION && !this.fragmented || opcode === OP_TEXT && this.fragmented) {
+				this.close(1002, "Invalid WebSocket frame");
+				return;
+			}
 			let payloadLength = byte1 & 127;
 			let offset = 2;
 			if (payloadLength === 126) {
@@ -109,9 +195,30 @@ var WebSocketConnection = class extends node_events.EventEmitter {
 				offset = 4;
 			} else if (payloadLength === 127) {
 				if (this.buffer.length < 10) return;
-				payloadLength = this.buffer.readUInt32BE(6) + this.buffer.readUInt32BE(2) * 4294967296;
+				const advertised = this.buffer.readBigUInt64BE(2);
+				if (advertised & 1n << 63n) {
+					this.close(1002, "Invalid WebSocket length");
+					return;
+				}
+				if (advertised > BigInt(Number.MAX_SAFE_INTEGER)) {
+					this.close(1009, "WebSocket message limit exceeded");
+					return;
+				}
+				payloadLength = Number(advertised);
 				offset = 10;
 			}
+			if ((byte1 & 127) === 126 && payloadLength < 126 || (byte1 & 127) === 127 && payloadLength < 65536) {
+				this.close(1002, "Invalid WebSocket length");
+				return;
+			}
+			if (opcode < 8 && payloadLength > (this.limits.maxMessageBytes ?? Infinity) - this.fragmentBytes) {
+				this.close(1009, "WebSocket message limit exceeded");
+				return;
+			}
+			if (payloadLength + offset + (masked ? 4 : 0) > (this.limits.maxBufferedBytes ?? Infinity) - this.fragmentBytes) {
+				this.close(1009, "WebSocket buffer limit exceeded");
+				return;
+			}
 			const totalFrameSize = offset + (masked ? 4 : 0) + payloadLength;
 			if (this.buffer.length < totalFrameSize) return;
 			let maskKey = null;
@@ -135,10 +242,15 @@ var WebSocketConnection = class extends node_events.EventEmitter {
 		}
 		if (opcode === OP_PONG) return;
 		if (opcode === OP_CLOSE) {
+			if (payload.length === 1) {
+				this.close(1002, "Invalid WebSocket close");
+				return;
+			}
 			const code = payload.length >= 2 ? payload.readUInt16BE(0) : 1005;
 			const reason = payload.length > 2 ? payload.subarray(2).toString("utf-8") : "";
 			if (!this.closed) {
 				this.closed = true;
+				this.release(/* @__PURE__ */ new Error("WebSocket closed"));
 				this.writeFrame(OP_CLOSE, payload);
 				this.socket.end();
 				this.emit("close", code, reason);
@@ -146,10 +258,13 @@ var WebSocketConnection = class extends node_events.EventEmitter {
 			return;
 		}
 		if (opcode === OP_TEXT || opcode === OP_CONTINUATION) {
-			this.fragments.push(payload);
+			this.fragmented = !fin;
+			this.fragmentBytes += payload.length;
+			if (payload.length) this.fragments.push(payload);
 			if (fin) {
 				const message = Buffer.concat(this.fragments).toString("utf-8");
 				this.fragments = [];
+				this.fragmentBytes = 0;
 				this.emit("message", message);
 			}
 			return;
@@ -159,7 +274,7 @@ var WebSocketConnection = class extends node_events.EventEmitter {
 function computeAcceptKey(wsKey) {
 	return (0, node_crypto.createHash)("sha1").update(wsKey + WS_GUID).digest("base64");
 }
-function upgradeToWebSocket(req, socket) {
+function upgradeToWebSocket(req, socket, limits) {
 	const key = req.headers["sec-websocket-key"];
 	if (!key) {
 		socket.write("HTTP/1.1 400 Bad Request\r\n\r\n");
@@ -177,7 +292,7 @@ Sec-WebSocket-Accept: ${computeAcceptKey(key)}\r\n`;
 	}
 	responseHeaders += "\r\n";
 	socket.write(responseHeaders);
-	return new WebSocketConnection(socket);
+	return new WebSocketConnection(socket, limits);
 }
 
 //#endregion

package/dist/ws-framing.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"ws-framing.cjs","names":["EventEmitter"],"sources":["../src/ws-framing.ts"],"sourcesContent":["/**\n * Minimal RFC 6455 WebSocket server implementation.\n *\n * Zero dependencies — uses only Node.js builtins (node:crypto, node:events).\n * Supports text frames, ping/pong, close handshake, and client frame unmasking.\n * Designed for a mock server — no extensions, no binary frames, no compression.\n */\n\nimport { createHash } from \"node:crypto\";\nimport { EventEmitter } from \"node:events\";\nimport type * as net from \"node:net\";\nimport type * as http from \"node:http\";\n\nconst WS_GUID = \"258EAFA5-E914-47DA-95CA-5AB5DC799C07\";\n\n// Opcodes\nconst OP_CONTINUATION = 0x0;\nconst OP_TEXT = 0x1;\nconst OP_CLOSE = 0x8;\nconst OP_PING = 0x9;\nconst OP_PONG = 0xa;\n\nexport class WebSocketConnection extends EventEmitter {\n  private socket: net.Socket;\n  private buffer: Buffer = Buffer.alloc(0);\n  private closed = false;\n\n  // For fragmented messages (continuation frames)\n  private fragments: Buffer[] = [];\n\n  constructor(socket: net.Socket) {\n    super();\n    this.socket = socket;\n\n    socket.on(\"data\", (data: Buffer) => {\n      this.buffer = Buffer.concat([this.buffer, data]);\n      this.parseFrames();\n    });\n\n    socket.on(\"close\", () => {\n      if (!this.closed) {\n        this.closed = true;\n        this.emit(\"close\", 1006, \"Connection lost\");\n      }\n    });\n\n    socket.on(\"error\", (err: Error) => {\n      this.emit(\"error\", err);\n    });\n  }\n\n  send(data: string): void {\n    if (this.closed) return;\n    const payload = Buffer.from(data, \"utf-8\");\n    this.writeFrame(OP_TEXT, payload);\n  }\n\n  close(code = 1000, reason = \"\"): void {\n    if (this.closed) return;\n    this.closed = true;\n\n    const reasonBuf = Buffer.from(reason, \"utf-8\");\n    const payload = Buffer.alloc(2 + reasonBuf.length);\n    payload.writeUInt16BE(code, 0);\n    reasonBuf.copy(payload, 2);\n    this.writeFrame(OP_CLOSE, payload);\n\n    // Give the client a moment to receive the close frame before destroying.\n    // If writeFrame failed (socket already destroyed), this is a no-op.\n    setTimeout(() => {\n      if (!this.socket.destroyed) {\n        this.socket.destroy();\n      }\n      // Emit close event for server-initiated closes so listeners\n      // (e.g. activeConnections.delete) always fire.\n      this.emit(\"close\", code, reason);\n    }, 100);\n  }\n\n  destroy(): void {\n    if (this.closed) return;\n    this.closed = true;\n    if (!this.socket.destroyed) {\n      this.socket.destroy();\n    }\n    this.emit(\"close\", 1006, \"Connection destroyed\");\n  }\n\n  get isClosed(): boolean {\n    return this.closed;\n  }\n\n  private writeFrame(opcode: number, payload: Buffer): void {\n    if (this.socket.destroyed) return;\n\n    // Server-to-client frames are NOT masked (per RFC 6455 §5.1)\n    const length = payload.length;\n    let header: Buffer;\n\n    if (length < 126) {\n      header = Buffer.alloc(2);\n      header[0] = 0x80 | opcode; // FIN + opcode\n      header[1] = length;\n    } else if (length < 65536) {\n      header = Buffer.alloc(4);\n      header[0] = 0x80 | opcode;\n      header[1] = 126;\n      header.writeUInt16BE(length, 2);\n    } else {\n      header = Buffer.alloc(10);\n      header[0] = 0x80 | opcode;\n      header[1] = 127;\n      header.writeUInt32BE(0, 2);\n      header.writeUInt32BE(length, 6);\n    }\n\n    try {\n      this.socket.write(Buffer.concat([header, payload]));\n    } catch (err: unknown) {\n      // Expected when socket is destroyed between our check and write.\n      // Log unexpected errors so they don't vanish silently.\n      if (!this.socket.destroyed) {\n        const msg = err instanceof Error ? err.message : String(err);\n        console.error(`[LLMock] Unexpected writeFrame error: ${msg}`);\n      }\n    }\n  }\n\n  private parseFrames(): void {\n    while (this.buffer.length >= 2 && !this.closed) {\n      const byte0 = this.buffer[0];\n      const byte1 = this.buffer[1];\n\n      const fin = (byte0 & 0x80) !== 0;\n      const opcode = byte0 & 0x0f;\n      const masked = (byte1 & 0x80) !== 0;\n      let payloadLength = byte1 & 0x7f;\n      let offset = 2;\n\n      if (payloadLength === 126) {\n        if (this.buffer.length < 4) return; // need more data\n        payloadLength = this.buffer.readUInt16BE(2);\n        offset = 4;\n      } else if (payloadLength === 127) {\n        if (this.buffer.length < 10) return;\n        // Read lower 32 bits (upper 32 should be 0 for reasonable payloads)\n        payloadLength = this.buffer.readUInt32BE(6) + this.buffer.readUInt32BE(2) * 0x100000000;\n        offset = 10;\n      }\n\n      const maskSize = masked ? 4 : 0;\n      const totalFrameSize = offset + maskSize + payloadLength;\n\n      if (this.buffer.length < totalFrameSize) return; // need more data\n\n      let maskKey: Buffer | null = null;\n      if (masked) {\n        maskKey = this.buffer.subarray(offset, offset + 4);\n        offset += 4;\n      }\n\n      let payload = this.buffer.subarray(offset, offset + payloadLength);\n\n      // Unmask client payload\n      if (maskKey) {\n        payload = Buffer.from(payload); // copy before mutating\n        for (let i = 0; i < payload.length; i++) {\n          payload[i] ^= maskKey[i % 4];\n        }\n      }\n\n      // Consume the frame from the buffer\n      this.buffer = this.buffer.subarray(totalFrameSize);\n\n      this.handleFrame(fin, opcode, payload);\n    }\n  }\n\n  private handleFrame(fin: boolean, opcode: number, payload: Buffer): void {\n    // Control frames (opcode >= 0x8) must not be fragmented\n    if (opcode === OP_PING) {\n      this.writeFrame(OP_PONG, payload);\n      return;\n    }\n\n    if (opcode === OP_PONG) {\n      // Ignore unsolicited pongs\n      return;\n    }\n\n    if (opcode === OP_CLOSE) {\n      const code = payload.length >= 2 ? payload.readUInt16BE(0) : 1005;\n      const reason = payload.length > 2 ? payload.subarray(2).toString(\"utf-8\") : \"\";\n\n      if (!this.closed) {\n        this.closed = true;\n        // Echo close frame back\n        this.writeFrame(OP_CLOSE, payload);\n        this.socket.end();\n        this.emit(\"close\", code, reason);\n      }\n      // If already closed (server-initiated or duplicate), ignore — the\n      // close event was already emitted by close() or the first OP_CLOSE.\n      return;\n    }\n\n    // Text or continuation frames\n    if (opcode === OP_TEXT || opcode === OP_CONTINUATION) {\n      this.fragments.push(payload);\n\n      if (fin) {\n        const message = Buffer.concat(this.fragments).toString(\"utf-8\");\n        this.fragments = [];\n        this.emit(\"message\", message);\n      }\n      // If !fin, wait for more continuation frames\n      return;\n    }\n\n    // Binary or unknown — just ignore for a mock server\n  }\n}\n\nexport function computeAcceptKey(wsKey: string): string {\n  return createHash(\"sha1\")\n    .update(wsKey + WS_GUID)\n    .digest(\"base64\");\n}\n\nexport function upgradeToWebSocket(\n  req: http.IncomingMessage,\n  socket: net.Socket,\n): WebSocketConnection {\n  const key = req.headers[\"sec-websocket-key\"];\n  if (!key) {\n    socket.write(\"HTTP/1.1 400 Bad Request\\r\\n\\r\\n\");\n    socket.destroy();\n    throw new Error(\"Missing Sec-WebSocket-Key header\");\n  }\n\n  const acceptKey = computeAcceptKey(key);\n\n  let responseHeaders =\n    \"HTTP/1.1 101 Switching Protocols\\r\\n\" +\n    \"Upgrade: websocket\\r\\n\" +\n    \"Connection: Upgrade\\r\\n\" +\n    `Sec-WebSocket-Accept: ${acceptKey}\\r\\n`;\n\n  // Echo back requested subprotocol if present\n  const protocol = req.headers[\"sec-websocket-protocol\"];\n  if (protocol) {\n    // Take the first offered protocol\n    const first = protocol.split(\",\")[0].trim();\n    responseHeaders += `Sec-WebSocket-Protocol: ${first}\\r\\n`;\n  }\n\n  responseHeaders += \"\\r\\n\";\n\n  socket.write(responseHeaders);\n\n  return new WebSocketConnection(socket);\n}\n"],"mappings":";;;;;;;;;;;;AAaA,MAAM,UAAU;AAGhB,MAAM,kBAAkB;AACxB,MAAM,UAAU;AAChB,MAAM,WAAW;AACjB,MAAM,UAAU;AAChB,MAAM,UAAU;AAEhB,IAAa,sBAAb,cAAyCA,yBAAa;CACpD,AAAQ;CACR,AAAQ,SAAiB,OAAO,MAAM,EAAE;CACxC,AAAQ,SAAS;CAGjB,AAAQ,YAAsB,EAAE;CAEhC,YAAY,QAAoB;AAC9B,SAAO;AACP,OAAK,SAAS;AAEd,SAAO,GAAG,SAAS,SAAiB;AAClC,QAAK,SAAS,OAAO,OAAO,CAAC,KAAK,QAAQ,KAAK,CAAC;AAChD,QAAK,aAAa;IAClB;AAEF,SAAO,GAAG,eAAe;AACvB,OAAI,CAAC,KAAK,QAAQ;AAChB,SAAK,SAAS;AACd,SAAK,KAAK,SAAS,MAAM,kBAAkB;;IAE7C;AAEF,SAAO,GAAG,UAAU,QAAe;AACjC,QAAK,KAAK,SAAS,IAAI;IACvB;;CAGJ,KAAK,MAAoB;AACvB,MAAI,KAAK,OAAQ;EACjB,MAAM,UAAU,OAAO,KAAK,MAAM,QAAQ;AAC1C,OAAK,WAAW,SAAS,QAAQ;;CAGnC,MAAM,OAAO,KAAM,SAAS,IAAU;AACpC,MAAI,KAAK,OAAQ;AACjB,OAAK,SAAS;EAEd,MAAM,YAAY,OAAO,KAAK,QAAQ,QAAQ;EAC9C,MAAM,UAAU,OAAO,MAAM,IAAI,UAAU,OAAO;AAClD,UAAQ,cAAc,MAAM,EAAE;AAC9B,YAAU,KAAK,SAAS,EAAE;AAC1B,OAAK,WAAW,UAAU,QAAQ;AAIlC,mBAAiB;AACf,OAAI,CAAC,KAAK,OAAO,UACf,MAAK,OAAO,SAAS;AAIvB,QAAK,KAAK,SAAS,MAAM,OAAO;KAC/B,IAAI;;CAGT,UAAgB;AACd,MAAI,KAAK,OAAQ;AACjB,OAAK,SAAS;AACd,MAAI,CAAC,KAAK,OAAO,UACf,MAAK,OAAO,SAAS;AAEvB,OAAK,KAAK,SAAS,MAAM,uBAAuB;;CAGlD,IAAI,WAAoB;AACtB,SAAO,KAAK;;CAGd,AAAQ,WAAW,QAAgB,SAAuB;AACxD,MAAI,KAAK,OAAO,UAAW;EAG3B,MAAM,SAAS,QAAQ;EACvB,IAAI;AAEJ,MAAI,SAAS,KAAK;AAChB,YAAS,OAAO,MAAM,EAAE;AACxB,UAAO,KAAK,MAAO;AACnB,UAAO,KAAK;aACH,SAAS,OAAO;AACzB,YAAS,OAAO,MAAM,EAAE;AACxB,UAAO,KAAK,MAAO;AACnB,UAAO,KAAK;AACZ,UAAO,cAAc,QAAQ,EAAE;SAC1B;AACL,YAAS,OAAO,MAAM,GAAG;AACzB,UAAO,KAAK,MAAO;AACnB,UAAO,KAAK;AACZ,UAAO,cAAc,GAAG,EAAE;AAC1B,UAAO,cAAc,QAAQ,EAAE;;AAGjC,MAAI;AACF,QAAK,OAAO,MAAM,OAAO,OAAO,CAAC,QAAQ,QAAQ,CAAC,CAAC;WAC5C,KAAc;AAGrB,OAAI,CAAC,KAAK,OAAO,WAAW;IAC1B,MAAM,MAAM,eAAe,QAAQ,IAAI,UAAU,OAAO,IAAI;AAC5D,YAAQ,MAAM,yCAAyC,MAAM;;;;CAKnE,AAAQ,cAAoB;AAC1B,SAAO,KAAK,OAAO,UAAU,KAAK,CAAC,KAAK,QAAQ;GAC9C,MAAM,QAAQ,KAAK,OAAO;GAC1B,MAAM,QAAQ,KAAK,OAAO;GAE1B,MAAM,OAAO,QAAQ,SAAU;GAC/B,MAAM,SAAS,QAAQ;GACvB,MAAM,UAAU,QAAQ,SAAU;GAClC,IAAI,gBAAgB,QAAQ;GAC5B,IAAI,SAAS;AAEb,OAAI,kBAAkB,KAAK;AACzB,QAAI,KAAK,OAAO,SAAS,EAAG;AAC5B,oBAAgB,KAAK,OAAO,aAAa,EAAE;AAC3C,aAAS;cACA,kBAAkB,KAAK;AAChC,QAAI,KAAK,OAAO,SAAS,GAAI;AAE7B,oBAAgB,KAAK,OAAO,aAAa,EAAE,GAAG,KAAK,OAAO,aAAa,EAAE,GAAG;AAC5E,aAAS;;GAIX,MAAM,iBAAiB,UADN,SAAS,IAAI,KACa;AAE3C,OAAI,KAAK,OAAO,SAAS,eAAgB;GAEzC,IAAI,UAAyB;AAC7B,OAAI,QAAQ;AACV,cAAU,KAAK,OAAO,SAAS,QAAQ,SAAS,EAAE;AAClD,cAAU;;GAGZ,IAAI,UAAU,KAAK,OAAO,SAAS,QAAQ,SAAS,cAAc;AAGlE,OAAI,SAAS;AACX,cAAU,OAAO,KAAK,QAAQ;AAC9B,SAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,QAAQ,IAClC,SAAQ,MAAM,QAAQ,IAAI;;AAK9B,QAAK,SAAS,KAAK,OAAO,SAAS,eAAe;AAElD,QAAK,YAAY,KAAK,QAAQ,QAAQ;;;CAI1C,AAAQ,YAAY,KAAc,QAAgB,SAAuB;AAEvE,MAAI,WAAW,SAAS;AACtB,QAAK,WAAW,SAAS,QAAQ;AACjC;;AAGF,MAAI,WAAW,QAEb;AAGF,MAAI,WAAW,UAAU;GACvB,MAAM,OAAO,QAAQ,UAAU,IAAI,QAAQ,aAAa,EAAE,GAAG;GAC7D,MAAM,SAAS,QAAQ,SAAS,IAAI,QAAQ,SAAS,EAAE,CAAC,SAAS,QAAQ,GAAG;AAE5E,OAAI,CAAC,KAAK,QAAQ;AAChB,SAAK,SAAS;AAEd,SAAK,WAAW,UAAU,QAAQ;AAClC,SAAK,OAAO,KAAK;AACjB,SAAK,KAAK,SAAS,MAAM,OAAO;;AAIlC;;AAIF,MAAI,WAAW,WAAW,WAAW,iBAAiB;AACpD,QAAK,UAAU,KAAK,QAAQ;AAE5B,OAAI,KAAK;IACP,MAAM,UAAU,OAAO,OAAO,KAAK,UAAU,CAAC,SAAS,QAAQ;AAC/D,SAAK,YAAY,EAAE;AACnB,SAAK,KAAK,WAAW,QAAQ;;AAG/B;;;;AAON,SAAgB,iBAAiB,OAAuB;AACtD,oCAAkB,OAAO,CACtB,OAAO,QAAQ,QAAQ,CACvB,OAAO,SAAS;;AAGrB,SAAgB,mBACd,KACA,QACqB;CACrB,MAAM,MAAM,IAAI,QAAQ;AACxB,KAAI,CAAC,KAAK;AACR,SAAO,MAAM,mCAAmC;AAChD,SAAO,SAAS;AAChB,QAAM,IAAI,MAAM,mCAAmC;;CAKrD,IAAI,kBACF;;;wBAHgB,iBAAiB,IAAI,CAMF;CAGrC,MAAM,WAAW,IAAI,QAAQ;AAC7B,KAAI,UAAU;EAEZ,MAAM,QAAQ,SAAS,MAAM,IAAI,CAAC,GAAG,MAAM;AAC3C,qBAAmB,2BAA2B,MAAM;;AAGtD,oBAAmB;AAEnB,QAAO,MAAM,gBAAgB;AAE7B,QAAO,IAAI,oBAAoB,OAAO"}
+{"version":3,"file":"ws-framing.cjs","names":["EventEmitter"],"sources":["../src/ws-framing.ts"],"sourcesContent":["/**\n * Minimal RFC 6455 WebSocket server implementation.\n *\n * Zero dependencies — uses only Node.js builtins (node:crypto, node:events).\n * Supports text frames, ping/pong, close handshake, and client frame unmasking.\n * Designed for a mock server — no extensions, no binary frames, no compression.\n */\n\nimport { createHash, randomBytes } from \"node:crypto\";\nimport { EventEmitter } from \"node:events\";\nimport type * as net from \"node:net\";\nimport type * as http from \"node:http\";\n\nconst WS_GUID = \"258EAFA5-E914-47DA-95CA-C5AB0DC85B11\";\n\n// Opcodes\nconst OP_CONTINUATION = 0x0;\nconst OP_TEXT = 0x1;\nconst OP_CLOSE = 0x8;\nconst OP_PING = 0x9;\nconst OP_PONG = 0xa;\n\nexport interface WebSocketLimits {\n  maxMessageBytes?: number;\n  maxBufferedBytes?: number;\n  maxWriteBytes?: number;\n  /** Client role: mask outgoing frames and require unmasked incoming frames. */\n  maskOutgoing?: boolean;\n}\n\nexport class WebSocketConnection extends EventEmitter {\n  private socket: net.Socket;\n  private buffer: Buffer = Buffer.alloc(0);\n  private closed = false;\n\n  // For fragmented messages (continuation frames)\n  private fragments: Buffer[] = [];\n\n  private fragmentBytes = 0;\n  private fragmented = false;\n  private pendingWriteBytes = 0;\n  private pendingWrites = new Set<(error: Error) => void>();\n  private limits: WebSocketLimits;\n\n  constructor(socket: net.Socket, limits: WebSocketLimits = {}) {\n    super();\n    for (const value of [limits.maxMessageBytes, limits.maxBufferedBytes, limits.maxWriteBytes]) {\n      if (value !== undefined && (!Number.isSafeInteger(value) || value <= 0)) {\n        throw new RangeError(\"WebSocket limits must be positive safe integers\");\n      }\n    }\n    this.limits = { ...limits };\n    this.socket = socket;\n\n    socket.on(\"data\", (data: Buffer) => {\n      // Byte-idle consumers must see partial frames and control traffic too.\n      // Emit no payload, and notify before parsing can dispatch a message.\n      if (data.length > 0 && !this.closed) this.emit(\"activity\");\n      let offset = 0;\n      while (offset < data.length && !this.closed) {\n        const room =\n          (this.limits.maxBufferedBytes ?? Infinity) - this.fragmentBytes - this.buffer.length;\n        if (room <= 0) {\n          this.close(1009, \"WebSocket buffer limit exceeded\");\n          return;\n        }\n        const end = offset + Math.min(room, data.length - offset);\n        this.buffer = Buffer.concat([this.buffer, data.subarray(offset, end)]);\n        offset = end;\n        this.parseFrames();\n      }\n    });\n\n    socket.on(\"close\", () => {\n      if (!this.closed) {\n        this.closed = true;\n        this.release(new Error(\"WebSocket closed\"));\n        this.emit(\"close\", 1006, \"Connection lost\");\n      }\n    });\n\n    socket.on(\"error\", (err: Error) => {\n      this.emit(\"error\", err);\n    });\n  }\n\n  send(data: string): void {\n    if (this.closed) return;\n    void this.sendAsync(data).catch(() => {\n      /* Legacy fire-and-forget API. */\n    });\n  }\n\n  sendAsync(data: string, signal?: AbortSignal): Promise<void> {\n    if (this.closed || this.socket.destroyed) return Promise.reject(new Error(\"WebSocket closed\"));\n    if (signal?.aborted) return Promise.reject(new Error(\"WebSocket send aborted\"));\n    const length = Buffer.byteLength(data);\n    const frameBytes =\n      length + (length < 126 ? 2 : length < 65536 ? 4 : 10) + (this.limits.maskOutgoing ? 4 : 0);\n    if (\n      frameBytes >\n      (this.limits.maxWriteBytes ?? Infinity) -\n        Math.max(this.pendingWriteBytes, this.socket.writableLength)\n    ) {\n      this.close(1009, \"WebSocket write limit exceeded\");\n      return Promise.reject(new Error(\"WebSocket write limit exceeded\"));\n    }\n    this.pendingWriteBytes += frameBytes;\n    return new Promise<void>((resolve, reject) => {\n      let settled = false;\n      const finish = (error?: Error) => {\n        if (settled) return;\n        settled = true;\n        this.pendingWriteBytes -= frameBytes;\n        this.pendingWrites.delete(fail);\n        signal?.removeEventListener(\"abort\", abort);\n        if (error) reject(error);\n        else resolve();\n      };\n      const fail = (error: Error) => finish(error);\n      const abort = () => {\n        finish(new Error(\"WebSocket send aborted\"));\n        this.destroy();\n      };\n      this.pendingWrites.add(fail);\n      signal?.addEventListener(\"abort\", abort, { once: true });\n      try {\n        this.socket.write(this.encodeFrame(OP_TEXT, Buffer.from(data)), (error?: Error | null) => {\n          if (error) finish(new Error(\"WebSocket write failed\"));\n          else finish();\n        });\n      } catch {\n        finish(new Error(\"WebSocket write failed\"));\n      }\n    });\n  }\n\n  private release(error: Error): void {\n    this.buffer = Buffer.alloc(0);\n    this.fragments = [];\n    this.fragmentBytes = 0;\n    for (const fail of this.pendingWrites) fail(error);\n  }\n\n  close(code = 1000, reason = \"\"): void {\n    if (this.closed) return;\n    this.closed = true;\n    this.release(new Error(\"WebSocket closed\"));\n\n    const encodedReason = Buffer.from(reason, \"utf-8\");\n    let reasonEnd = Math.min(encodedReason.length, 123);\n    // A continuation byte at the cut means its code point straddles the limit.\n    while (reasonEnd < encodedReason.length && (encodedReason[reasonEnd] & 0xc0) === 0x80) {\n      reasonEnd--;\n    }\n    const reasonBuf = encodedReason.subarray(0, reasonEnd);\n    const payload = Buffer.alloc(2 + reasonBuf.length);\n    payload.writeUInt16BE(code, 0);\n    reasonBuf.copy(payload, 2);\n    this.writeFrame(OP_CLOSE, payload);\n\n    // Give the client a moment to receive the close frame before destroying.\n    // If writeFrame failed (socket already destroyed), this is a no-op.\n    setTimeout(() => {\n      if (!this.socket.destroyed) {\n        this.socket.destroy();\n      }\n      // Emit close event for server-initiated closes so listeners\n      // (e.g. activeConnections.delete) always fire.\n      this.emit(\"close\", code, reason);\n    }, 100);\n  }\n\n  destroy(): void {\n    if (this.closed) return;\n    this.closed = true;\n    this.release(new Error(\"WebSocket destroyed\"));\n    if (!this.socket.destroyed) {\n      this.socket.destroy();\n    }\n    this.emit(\"close\", 1006, \"Connection destroyed\");\n  }\n\n  get isClosed(): boolean {\n    return this.closed;\n  }\n\n  private encodeFrame(opcode: number, payload: Buffer): Buffer {\n    const length = payload.length;\n    const headerSize = length < 126 ? 2 : length < 65536 ? 4 : 10;\n    const maskSize = this.limits.maskOutgoing ? 4 : 0;\n    const frame = Buffer.allocUnsafe(headerSize + maskSize + length);\n    frame[0] = 0x80 | opcode;\n    frame[1] = (maskSize ? 0x80 : 0) | (length < 126 ? length : length < 65536 ? 126 : 127);\n    if (headerSize === 4) frame.writeUInt16BE(length, 2);\n    if (headerSize === 10) frame.writeBigUInt64BE(BigInt(length), 2);\n    if (maskSize) {\n      const mask = randomBytes(4);\n      mask.copy(frame, headerSize);\n      for (let i = 0; i < length; i++) frame[headerSize + 4 + i] = payload[i] ^ mask[i % 4];\n    } else payload.copy(frame, headerSize);\n    return frame;\n  }\n\n  private writeFrame(opcode: number, payload: Buffer): void {\n    if (this.socket.destroyed) return;\n    const available =\n      (this.limits.maxWriteBytes ?? Infinity) -\n      Math.max(this.pendingWriteBytes, this.socket.writableLength);\n    const headerBytes = 2 + (this.limits.maskOutgoing ? 4 : 0);\n    if (opcode === OP_CLOSE && payload.length + headerBytes > available) {\n      // Keep even the final control frame inside the configured socket budget.\n      if (available < headerBytes + 2) {\n        this.socket.destroy();\n        return;\n      }\n      payload = payload.subarray(0, 2);\n    } else if (payload.length + headerBytes > available) {\n      this.close(1009, \"WebSocket write limit exceeded\");\n      return;\n    }\n    const frame = this.encodeFrame(opcode, payload);\n    this.socket.write(frame);\n  }\n\n  private parseFrames(): void {\n    while (this.buffer.length >= 2 && !this.closed) {\n      const byte0 = this.buffer[0];\n      const byte1 = this.buffer[1];\n\n      const fin = (byte0 & 0x80) !== 0;\n      const opcode = byte0 & 0x0f;\n      const masked = (byte1 & 0x80) !== 0;\n      if (\n        byte0 & 0x70 ||\n        masked === Boolean(this.limits.maskOutgoing) ||\n        ![OP_CONTINUATION, OP_TEXT, 2, OP_CLOSE, OP_PING, OP_PONG].includes(opcode) ||\n        (opcode >= 8 && (!fin || (byte1 & 0x7f) > 125)) ||\n        (opcode === OP_CONTINUATION && !this.fragmented) ||\n        (opcode === OP_TEXT && this.fragmented)\n      ) {\n        this.close(1002, \"Invalid WebSocket frame\");\n        return;\n      }\n      let payloadLength = byte1 & 0x7f;\n      let offset = 2;\n\n      if (payloadLength === 126) {\n        if (this.buffer.length < 4) return; // need more data\n        payloadLength = this.buffer.readUInt16BE(2);\n        offset = 4;\n      } else if (payloadLength === 127) {\n        if (this.buffer.length < 10) return;\n        const advertised = this.buffer.readBigUInt64BE(2);\n        if (advertised & (1n << 63n)) {\n          this.close(1002, \"Invalid WebSocket length\");\n          return;\n        }\n        if (advertised > BigInt(Number.MAX_SAFE_INTEGER)) {\n          this.close(1009, \"WebSocket message limit exceeded\");\n          return;\n        }\n        payloadLength = Number(advertised);\n        offset = 10;\n      }\n\n      if (\n        ((byte1 & 0x7f) === 126 && payloadLength < 126) ||\n        ((byte1 & 0x7f) === 127 && payloadLength < 65536)\n      ) {\n        this.close(1002, \"Invalid WebSocket length\");\n        return;\n      }\n      if (\n        opcode < 8 &&\n        payloadLength > (this.limits.maxMessageBytes ?? Infinity) - this.fragmentBytes\n      ) {\n        this.close(1009, \"WebSocket message limit exceeded\");\n        return;\n      }\n      if (\n        payloadLength + offset + (masked ? 4 : 0) >\n        (this.limits.maxBufferedBytes ?? Infinity) - this.fragmentBytes\n      ) {\n        this.close(1009, \"WebSocket buffer limit exceeded\");\n        return;\n      }\n      const maskSize = masked ? 4 : 0;\n      const totalFrameSize = offset + maskSize + payloadLength;\n\n      if (this.buffer.length < totalFrameSize) return; // need more data\n\n      let maskKey: Buffer | null = null;\n      if (masked) {\n        maskKey = this.buffer.subarray(offset, offset + 4);\n        offset += 4;\n      }\n\n      let payload = this.buffer.subarray(offset, offset + payloadLength);\n\n      // Unmask client payload\n      if (maskKey) {\n        payload = Buffer.from(payload); // copy before mutating\n        for (let i = 0; i < payload.length; i++) {\n          payload[i] ^= maskKey[i % 4];\n        }\n      }\n\n      // Consume the frame from the buffer\n      this.buffer = this.buffer.subarray(totalFrameSize);\n\n      this.handleFrame(fin, opcode, payload);\n    }\n  }\n\n  private handleFrame(fin: boolean, opcode: number, payload: Buffer): void {\n    // Control frames (opcode >= 0x8) must not be fragmented\n    if (opcode === OP_PING) {\n      this.writeFrame(OP_PONG, payload);\n      return;\n    }\n\n    if (opcode === OP_PONG) {\n      // Ignore unsolicited pongs\n      return;\n    }\n\n    if (opcode === OP_CLOSE) {\n      if (payload.length === 1) {\n        this.close(1002, \"Invalid WebSocket close\");\n        return;\n      }\n      const code = payload.length >= 2 ? payload.readUInt16BE(0) : 1005;\n      const reason = payload.length > 2 ? payload.subarray(2).toString(\"utf-8\") : \"\";\n\n      if (!this.closed) {\n        this.closed = true;\n        this.release(new Error(\"WebSocket closed\"));\n        // Echo close frame back\n        this.writeFrame(OP_CLOSE, payload);\n        this.socket.end();\n        this.emit(\"close\", code, reason);\n      }\n      // If already closed (server-initiated or duplicate), ignore — the\n      // close event was already emitted by close() or the first OP_CLOSE.\n      return;\n    }\n\n    // Text or continuation frames\n    if (opcode === OP_TEXT || opcode === OP_CONTINUATION) {\n      this.fragmented = !fin;\n      this.fragmentBytes += payload.length;\n      if (payload.length) this.fragments.push(payload);\n\n      if (fin) {\n        const message = Buffer.concat(this.fragments).toString(\"utf-8\");\n        this.fragments = [];\n        this.fragmentBytes = 0;\n        this.emit(\"message\", message);\n      }\n      // If !fin, wait for more continuation frames\n      return;\n    }\n\n    // Binary or unknown — just ignore for a mock server\n  }\n}\n\nexport function computeAcceptKey(wsKey: string): string {\n  return createHash(\"sha1\")\n    .update(wsKey + WS_GUID)\n    .digest(\"base64\");\n}\n\nexport function upgradeToWebSocket(\n  req: http.IncomingMessage,\n  socket: net.Socket,\n  limits?: WebSocketLimits,\n): WebSocketConnection {\n  const key = req.headers[\"sec-websocket-key\"];\n  if (!key) {\n    socket.write(\"HTTP/1.1 400 Bad Request\\r\\n\\r\\n\");\n    socket.destroy();\n    throw new Error(\"Missing Sec-WebSocket-Key header\");\n  }\n\n  const acceptKey = computeAcceptKey(key);\n\n  let responseHeaders =\n    \"HTTP/1.1 101 Switching Protocols\\r\\n\" +\n    \"Upgrade: websocket\\r\\n\" +\n    \"Connection: Upgrade\\r\\n\" +\n    `Sec-WebSocket-Accept: ${acceptKey}\\r\\n`;\n\n  // Echo back requested subprotocol if present\n  const protocol = req.headers[\"sec-websocket-protocol\"];\n  if (protocol) {\n    // Take the first offered protocol\n    const first = protocol.split(\",\")[0].trim();\n    responseHeaders += `Sec-WebSocket-Protocol: ${first}\\r\\n`;\n  }\n\n  responseHeaders += \"\\r\\n\";\n\n  socket.write(responseHeaders);\n\n  return new WebSocketConnection(socket, limits);\n}\n"],"mappings":";;;;;;;;;;;;AAaA,MAAM,UAAU;AAGhB,MAAM,kBAAkB;AACxB,MAAM,UAAU;AAChB,MAAM,WAAW;AACjB,MAAM,UAAU;AAChB,MAAM,UAAU;AAUhB,IAAa,sBAAb,cAAyCA,yBAAa;CACpD,AAAQ;CACR,AAAQ,SAAiB,OAAO,MAAM,EAAE;CACxC,AAAQ,SAAS;CAGjB,AAAQ,YAAsB,EAAE;CAEhC,AAAQ,gBAAgB;CACxB,AAAQ,aAAa;CACrB,AAAQ,oBAAoB;CAC5B,AAAQ,gCAAgB,IAAI,KAA6B;CACzD,AAAQ;CAER,YAAY,QAAoB,SAA0B,EAAE,EAAE;AAC5D,SAAO;AACP,OAAK,MAAM,SAAS;GAAC,OAAO;GAAiB,OAAO;GAAkB,OAAO;GAAc,CACzF,KAAI,UAAU,WAAc,CAAC,OAAO,cAAc,MAAM,IAAI,SAAS,GACnE,OAAM,IAAI,WAAW,kDAAkD;AAG3E,OAAK,SAAS,EAAE,GAAG,QAAQ;AAC3B,OAAK,SAAS;AAEd,SAAO,GAAG,SAAS,SAAiB;AAGlC,OAAI,KAAK,SAAS,KAAK,CAAC,KAAK,OAAQ,MAAK,KAAK,WAAW;GAC1D,IAAI,SAAS;AACb,UAAO,SAAS,KAAK,UAAU,CAAC,KAAK,QAAQ;IAC3C,MAAM,QACH,KAAK,OAAO,oBAAoB,YAAY,KAAK,gBAAgB,KAAK,OAAO;AAChF,QAAI,QAAQ,GAAG;AACb,UAAK,MAAM,MAAM,kCAAkC;AACnD;;IAEF,MAAM,MAAM,SAAS,KAAK,IAAI,MAAM,KAAK,SAAS,OAAO;AACzD,SAAK,SAAS,OAAO,OAAO,CAAC,KAAK,QAAQ,KAAK,SAAS,QAAQ,IAAI,CAAC,CAAC;AACtE,aAAS;AACT,SAAK,aAAa;;IAEpB;AAEF,SAAO,GAAG,eAAe;AACvB,OAAI,CAAC,KAAK,QAAQ;AAChB,SAAK,SAAS;AACd,SAAK,wBAAQ,IAAI,MAAM,mBAAmB,CAAC;AAC3C,SAAK,KAAK,SAAS,MAAM,kBAAkB;;IAE7C;AAEF,SAAO,GAAG,UAAU,QAAe;AACjC,QAAK,KAAK,SAAS,IAAI;IACvB;;CAGJ,KAAK,MAAoB;AACvB,MAAI,KAAK,OAAQ;AACjB,EAAK,KAAK,UAAU,KAAK,CAAC,YAAY,GAEpC;;CAGJ,UAAU,MAAc,QAAqC;AAC3D,MAAI,KAAK,UAAU,KAAK,OAAO,UAAW,QAAO,QAAQ,uBAAO,IAAI,MAAM,mBAAmB,CAAC;AAC9F,MAAI,QAAQ,QAAS,QAAO,QAAQ,uBAAO,IAAI,MAAM,yBAAyB,CAAC;EAC/E,MAAM,SAAS,OAAO,WAAW,KAAK;EACtC,MAAM,aACJ,UAAU,SAAS,MAAM,IAAI,SAAS,QAAQ,IAAI,OAAO,KAAK,OAAO,eAAe,IAAI;AAC1F,MACE,cACC,KAAK,OAAO,iBAAiB,YAC5B,KAAK,IAAI,KAAK,mBAAmB,KAAK,OAAO,eAAe,EAC9D;AACA,QAAK,MAAM,MAAM,iCAAiC;AAClD,UAAO,QAAQ,uBAAO,IAAI,MAAM,iCAAiC,CAAC;;AAEpE,OAAK,qBAAqB;AAC1B,SAAO,IAAI,SAAe,SAAS,WAAW;GAC5C,IAAI,UAAU;GACd,MAAM,UAAU,UAAkB;AAChC,QAAI,QAAS;AACb,cAAU;AACV,SAAK,qBAAqB;AAC1B,SAAK,cAAc,OAAO,KAAK;AAC/B,YAAQ,oBAAoB,SAAS,MAAM;AAC3C,QAAI,MAAO,QAAO,MAAM;QACnB,UAAS;;GAEhB,MAAM,QAAQ,UAAiB,OAAO,MAAM;GAC5C,MAAM,cAAc;AAClB,2BAAO,IAAI,MAAM,yBAAyB,CAAC;AAC3C,SAAK,SAAS;;AAEhB,QAAK,cAAc,IAAI,KAAK;AAC5B,WAAQ,iBAAiB,SAAS,OAAO,EAAE,MAAM,MAAM,CAAC;AACxD,OAAI;AACF,SAAK,OAAO,MAAM,KAAK,YAAY,SAAS,OAAO,KAAK,KAAK,CAAC,GAAG,UAAyB;AACxF,SAAI,MAAO,wBAAO,IAAI,MAAM,yBAAyB,CAAC;SACjD,SAAQ;MACb;WACI;AACN,2BAAO,IAAI,MAAM,yBAAyB,CAAC;;IAE7C;;CAGJ,AAAQ,QAAQ,OAAoB;AAClC,OAAK,SAAS,OAAO,MAAM,EAAE;AAC7B,OAAK,YAAY,EAAE;AACnB,OAAK,gBAAgB;AACrB,OAAK,MAAM,QAAQ,KAAK,cAAe,MAAK,MAAM;;CAGpD,MAAM,OAAO,KAAM,SAAS,IAAU;AACpC,MAAI,KAAK,OAAQ;AACjB,OAAK,SAAS;AACd,OAAK,wBAAQ,IAAI,MAAM,mBAAmB,CAAC;EAE3C,MAAM,gBAAgB,OAAO,KAAK,QAAQ,QAAQ;EAClD,IAAI,YAAY,KAAK,IAAI,cAAc,QAAQ,IAAI;AAEnD,SAAO,YAAY,cAAc,WAAW,cAAc,aAAa,SAAU,IAC/E;EAEF,MAAM,YAAY,cAAc,SAAS,GAAG,UAAU;EACtD,MAAM,UAAU,OAAO,MAAM,IAAI,UAAU,OAAO;AAClD,UAAQ,cAAc,MAAM,EAAE;AAC9B,YAAU,KAAK,SAAS,EAAE;AAC1B,OAAK,WAAW,UAAU,QAAQ;AAIlC,mBAAiB;AACf,OAAI,CAAC,KAAK,OAAO,UACf,MAAK,OAAO,SAAS;AAIvB,QAAK,KAAK,SAAS,MAAM,OAAO;KAC/B,IAAI;;CAGT,UAAgB;AACd,MAAI,KAAK,OAAQ;AACjB,OAAK,SAAS;AACd,OAAK,wBAAQ,IAAI,MAAM,sBAAsB,CAAC;AAC9C,MAAI,CAAC,KAAK,OAAO,UACf,MAAK,OAAO,SAAS;AAEvB,OAAK,KAAK,SAAS,MAAM,uBAAuB;;CAGlD,IAAI,WAAoB;AACtB,SAAO,KAAK;;CAGd,AAAQ,YAAY,QAAgB,SAAyB;EAC3D,MAAM,SAAS,QAAQ;EACvB,MAAM,aAAa,SAAS,MAAM,IAAI,SAAS,QAAQ,IAAI;EAC3D,MAAM,WAAW,KAAK,OAAO,eAAe,IAAI;EAChD,MAAM,QAAQ,OAAO,YAAY,aAAa,WAAW,OAAO;AAChE,QAAM,KAAK,MAAO;AAClB,QAAM,MAAM,WAAW,MAAO,MAAM,SAAS,MAAM,SAAS,SAAS,QAAQ,MAAM;AACnF,MAAI,eAAe,EAAG,OAAM,cAAc,QAAQ,EAAE;AACpD,MAAI,eAAe,GAAI,OAAM,iBAAiB,OAAO,OAAO,EAAE,EAAE;AAChE,MAAI,UAAU;GACZ,MAAM,oCAAmB,EAAE;AAC3B,QAAK,KAAK,OAAO,WAAW;AAC5B,QAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,IAAK,OAAM,aAAa,IAAI,KAAK,QAAQ,KAAK,KAAK,IAAI;QAC9E,SAAQ,KAAK,OAAO,WAAW;AACtC,SAAO;;CAGT,AAAQ,WAAW,QAAgB,SAAuB;AACxD,MAAI,KAAK,OAAO,UAAW;EAC3B,MAAM,aACH,KAAK,OAAO,iBAAiB,YAC9B,KAAK,IAAI,KAAK,mBAAmB,KAAK,OAAO,eAAe;EAC9D,MAAM,cAAc,KAAK,KAAK,OAAO,eAAe,IAAI;AACxD,MAAI,WAAW,YAAY,QAAQ,SAAS,cAAc,WAAW;AAEnE,OAAI,YAAY,cAAc,GAAG;AAC/B,SAAK,OAAO,SAAS;AACrB;;AAEF,aAAU,QAAQ,SAAS,GAAG,EAAE;aACvB,QAAQ,SAAS,cAAc,WAAW;AACnD,QAAK,MAAM,MAAM,iCAAiC;AAClD;;EAEF,MAAM,QAAQ,KAAK,YAAY,QAAQ,QAAQ;AAC/C,OAAK,OAAO,MAAM,MAAM;;CAG1B,AAAQ,cAAoB;AAC1B,SAAO,KAAK,OAAO,UAAU,KAAK,CAAC,KAAK,QAAQ;GAC9C,MAAM,QAAQ,KAAK,OAAO;GAC1B,MAAM,QAAQ,KAAK,OAAO;GAE1B,MAAM,OAAO,QAAQ,SAAU;GAC/B,MAAM,SAAS,QAAQ;GACvB,MAAM,UAAU,QAAQ,SAAU;AAClC,OACE,QAAQ,OACR,WAAW,QAAQ,KAAK,OAAO,aAAa,IAC5C,CAAC;IAAC;IAAiB;IAAS;IAAG;IAAU;IAAS;IAAQ,CAAC,SAAS,OAAO,IAC1E,UAAU,MAAM,CAAC,QAAQ,QAAQ,OAAQ,QACzC,WAAW,mBAAmB,CAAC,KAAK,cACpC,WAAW,WAAW,KAAK,YAC5B;AACA,SAAK,MAAM,MAAM,0BAA0B;AAC3C;;GAEF,IAAI,gBAAgB,QAAQ;GAC5B,IAAI,SAAS;AAEb,OAAI,kBAAkB,KAAK;AACzB,QAAI,KAAK,OAAO,SAAS,EAAG;AAC5B,oBAAgB,KAAK,OAAO,aAAa,EAAE;AAC3C,aAAS;cACA,kBAAkB,KAAK;AAChC,QAAI,KAAK,OAAO,SAAS,GAAI;IAC7B,MAAM,aAAa,KAAK,OAAO,gBAAgB,EAAE;AACjD,QAAI,aAAc,MAAM,KAAM;AAC5B,UAAK,MAAM,MAAM,2BAA2B;AAC5C;;AAEF,QAAI,aAAa,OAAO,OAAO,iBAAiB,EAAE;AAChD,UAAK,MAAM,MAAM,mCAAmC;AACpD;;AAEF,oBAAgB,OAAO,WAAW;AAClC,aAAS;;AAGX,QACI,QAAQ,SAAU,OAAO,gBAAgB,QACzC,QAAQ,SAAU,OAAO,gBAAgB,OAC3C;AACA,SAAK,MAAM,MAAM,2BAA2B;AAC5C;;AAEF,OACE,SAAS,KACT,iBAAiB,KAAK,OAAO,mBAAmB,YAAY,KAAK,eACjE;AACA,SAAK,MAAM,MAAM,mCAAmC;AACpD;;AAEF,OACE,gBAAgB,UAAU,SAAS,IAAI,MACtC,KAAK,OAAO,oBAAoB,YAAY,KAAK,eAClD;AACA,SAAK,MAAM,MAAM,kCAAkC;AACnD;;GAGF,MAAM,iBAAiB,UADN,SAAS,IAAI,KACa;AAE3C,OAAI,KAAK,OAAO,SAAS,eAAgB;GAEzC,IAAI,UAAyB;AAC7B,OAAI,QAAQ;AACV,cAAU,KAAK,OAAO,SAAS,QAAQ,SAAS,EAAE;AAClD,cAAU;;GAGZ,IAAI,UAAU,KAAK,OAAO,SAAS,QAAQ,SAAS,cAAc;AAGlE,OAAI,SAAS;AACX,cAAU,OAAO,KAAK,QAAQ;AAC9B,SAAK,IAAI,IAAI,GAAG,IAAI,QAAQ,QAAQ,IAClC,SAAQ,MAAM,QAAQ,IAAI;;AAK9B,QAAK,SAAS,KAAK,OAAO,SAAS,eAAe;AAElD,QAAK,YAAY,KAAK,QAAQ,QAAQ;;;CAI1C,AAAQ,YAAY,KAAc,QAAgB,SAAuB;AAEvE,MAAI,WAAW,SAAS;AACtB,QAAK,WAAW,SAAS,QAAQ;AACjC;;AAGF,MAAI,WAAW,QAEb;AAGF,MAAI,WAAW,UAAU;AACvB,OAAI,QAAQ,WAAW,GAAG;AACxB,SAAK,MAAM,MAAM,0BAA0B;AAC3C;;GAEF,MAAM,OAAO,QAAQ,UAAU,IAAI,QAAQ,aAAa,EAAE,GAAG;GAC7D,MAAM,SAAS,QAAQ,SAAS,IAAI,QAAQ,SAAS,EAAE,CAAC,SAAS,QAAQ,GAAG;AAE5E,OAAI,CAAC,KAAK,QAAQ;AAChB,SAAK,SAAS;AACd,SAAK,wBAAQ,IAAI,MAAM,mBAAmB,CAAC;AAE3C,SAAK,WAAW,UAAU,QAAQ;AAClC,SAAK,OAAO,KAAK;AACjB,SAAK,KAAK,SAAS,MAAM,OAAO;;AAIlC;;AAIF,MAAI,WAAW,WAAW,WAAW,iBAAiB;AACpD,QAAK,aAAa,CAAC;AACnB,QAAK,iBAAiB,QAAQ;AAC9B,OAAI,QAAQ,OAAQ,MAAK,UAAU,KAAK,QAAQ;AAEhD,OAAI,KAAK;IACP,MAAM,UAAU,OAAO,OAAO,KAAK,UAAU,CAAC,SAAS,QAAQ;AAC/D,SAAK,YAAY,EAAE;AACnB,SAAK,gBAAgB;AACrB,SAAK,KAAK,WAAW,QAAQ;;AAG/B;;;;AAON,SAAgB,iBAAiB,OAAuB;AACtD,oCAAkB,OAAO,CACtB,OAAO,QAAQ,QAAQ,CACvB,OAAO,SAAS;;AAGrB,SAAgB,mBACd,KACA,QACA,QACqB;CACrB,MAAM,MAAM,IAAI,QAAQ;AACxB,KAAI,CAAC,KAAK;AACR,SAAO,MAAM,mCAAmC;AAChD,SAAO,SAAS;AAChB,QAAM,IAAI,MAAM,mCAAmC;;CAKrD,IAAI,kBACF;;;wBAHgB,iBAAiB,IAAI,CAMF;CAGrC,MAAM,WAAW,IAAI,QAAQ;AAC7B,KAAI,UAAU;EAEZ,MAAM,QAAQ,SAAS,MAAM,IAAI,CAAC,GAAG,MAAM;AAC3C,qBAAmB,2BAA2B,MAAM;;AAGtD,oBAAmB;AAEnB,QAAO,MAAM,gBAAgB;AAE7B,QAAO,IAAI,oBAAoB,QAAQ,OAAO"}

package/dist/ws-framing.d.cts

@@ -4,23 +4,38 @@ import { EventEmitter } from "node:events";
 
 //#region src/ws-framing.d.ts
 
+interface WebSocketLimits {
+  maxMessageBytes?: number;
+  maxBufferedBytes?: number;
+  maxWriteBytes?: number;
+  /** Client role: mask outgoing frames and require unmasked incoming frames. */
+  maskOutgoing?: boolean;
+}
 declare class WebSocketConnection extends EventEmitter {
   private socket;
   private buffer;
   private closed;
   private fragments;
-  constructor(socket: net.Socket);
+  private fragmentBytes;
+  private fragmented;
+  private pendingWriteBytes;
+  private pendingWrites;
+  private limits;
+  constructor(socket: net.Socket, limits?: WebSocketLimits);
   send(data: string): void;
+  sendAsync(data: string, signal?: AbortSignal): Promise<void>;
+  private release;
   close(code?: number, reason?: string): void;
   destroy(): void;
   get isClosed(): boolean;
+  private encodeFrame;
   private writeFrame;
   private parseFrames;
   private handleFrame;
 }
 declare function computeAcceptKey(wsKey: string): string;
-declare function upgradeToWebSocket(req: http$1.IncomingMessage, socket: net.Socket): WebSocketConnection;
+declare function upgradeToWebSocket(req: http$1.IncomingMessage, socket: net.Socket, limits?: WebSocketLimits): WebSocketConnection;
 //# sourceMappingURL=ws-framing.d.ts.map
 //#endregion
-export { WebSocketConnection, computeAcceptKey, upgradeToWebSocket };
+export { WebSocketConnection, WebSocketLimits, computeAcceptKey, upgradeToWebSocket };
 //# sourceMappingURL=ws-framing.d.cts.map

package/dist/ws-framing.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"ws-framing.d.cts","names":[],"sources":["../src/ws-framing.ts"],"sourcesContent":[],"mappings":";;;;;;AA+NgB,cAzMH,mBAAA,SAA4B,YAAA,CAyMT;EAMhB,QAAA,MAAA;EAAkB,QAAA,MAAA;UAC3B,MAAK;UACE,SAAA;aACX,CAAA,MAAA,EA1MmB,GAAA,CAAI,MA0MvB;EAAmB,IAAA,CAAA,IAAA,EAAA,MAAA,CAAA,EAAA,IAAA;;;;;;;;iBATN,gBAAA;iBAMA,kBAAA,MACT,MAAA,CAAK,yBACF,GAAA,CAAI,SACX"}
+{"version":3,"file":"ws-framing.d.cts","names":[],"sources":["../src/ws-framing.ts"],"sourcesContent":[],"mappings":";;;;;;AA6FmC,UAvElB,eAAA,CAuEkB;iBAAc,CAAA,EAAA,MAAA;kBA/DR,CAAA,EAAA,MAAA;EAAY,aAAA,CAAA,EAAA,MAAA;EAkVrC;EAMA,YAAA,CAAA,EAAA,OAAkB;;AAC3B,cAzVM,mBAAA,SAA4B,YAAA,CAyV7B;UACE,MAAA;UACH,MAAA;UACR,MAAA;EAAmB,QAAA,SAAA;;;;;;sBA9UA,GAAA,CAAI,iBAAgB;;mCAiDP,cAAc;;;;;;;;;;iBAmRjC,gBAAA;iBAMA,kBAAA,MACT,MAAA,CAAK,yBACF,GAAA,CAAI,iBACH,kBACR"}

package/dist/ws-framing.d.ts

@@ -4,23 +4,38 @@ import * as net from "node:net";
 
 //#region src/ws-framing.d.ts
 
+interface WebSocketLimits {
+  maxMessageBytes?: number;
+  maxBufferedBytes?: number;
+  maxWriteBytes?: number;
+  /** Client role: mask outgoing frames and require unmasked incoming frames. */
+  maskOutgoing?: boolean;
+}
 declare class WebSocketConnection extends EventEmitter {
   private socket;
   private buffer;
   private closed;
   private fragments;
-  constructor(socket: net.Socket);
+  private fragmentBytes;
+  private fragmented;
+  private pendingWriteBytes;
+  private pendingWrites;
+  private limits;
+  constructor(socket: net.Socket, limits?: WebSocketLimits);
   send(data: string): void;
+  sendAsync(data: string, signal?: AbortSignal): Promise<void>;
+  private release;
   close(code?: number, reason?: string): void;
   destroy(): void;
   get isClosed(): boolean;
+  private encodeFrame;
   private writeFrame;
   private parseFrames;
   private handleFrame;
 }
 declare function computeAcceptKey(wsKey: string): string;
-declare function upgradeToWebSocket(req: http$1.IncomingMessage, socket: net.Socket): WebSocketConnection;
+declare function upgradeToWebSocket(req: http$1.IncomingMessage, socket: net.Socket, limits?: WebSocketLimits): WebSocketConnection;
 //# sourceMappingURL=ws-framing.d.ts.map
 //#endregion
-export { WebSocketConnection, computeAcceptKey, upgradeToWebSocket };
+export { WebSocketConnection, WebSocketLimits, computeAcceptKey, upgradeToWebSocket };
 //# sourceMappingURL=ws-framing.d.ts.map

package/dist/ws-framing.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ws-framing.d.ts","names":[],"sources":["../src/ws-framing.ts"],"sourcesContent":[],"mappings":";;;;;;AA+NgB,cAzMH,mBAAA,SAA4B,YAAA,CAyMT;EAMhB,QAAA,MAAA;EAAkB,QAAA,MAAA;UAC3B,MAAK;UACE,SAAA;aACX,CAAA,MAAA,EA1MmB,GAAA,CAAI,MA0MvB;EAAmB,IAAA,CAAA,IAAA,EAAA,MAAA,CAAA,EAAA,IAAA;;;;;;;;iBATN,gBAAA;iBAMA,kBAAA,MACT,MAAA,CAAK,yBACF,GAAA,CAAI,SACX"}
+{"version":3,"file":"ws-framing.d.ts","names":[],"sources":["../src/ws-framing.ts"],"sourcesContent":[],"mappings":";;;;;;AA6FmC,UAvElB,eAAA,CAuEkB;iBAAc,CAAA,EAAA,MAAA;kBA/DR,CAAA,EAAA,MAAA;EAAY,aAAA,CAAA,EAAA,MAAA;EAkVrC;EAMA,YAAA,CAAA,EAAA,OAAkB;;AAC3B,cAzVM,mBAAA,SAA4B,YAAA,CAyV7B;UACE,MAAA;UACH,MAAA;UACR,MAAA;EAAmB,QAAA,SAAA;;;;;;sBA9UA,GAAA,CAAI,iBAAgB;;mCAiDP,cAAc;;;;;;;;;;iBAmRjC,gBAAA;iBAMA,kBAAA,MACT,MAAA,CAAK,yBACF,GAAA,CAAI,iBACH,kBACR"}