@bakit/gateway 2.1.4 → 2.1.6

package/dist/index.js

@@ -34,7 +34,7 @@ var ZLIB_FLUSH = Buffer.from([0, 0, 255, 255]), DEFAULT_SHARD_OPTIONS = {
   Shutdown: 4
 };
 function createShard(options) {
-  let resolvedOptions = { ...DEFAULT_SHARD_OPTIONS, ...options }, state = ShardState.Idle, strategy = ShardStrategy.Unknown, ws, resumeGatewayURL, inflater, zlibBuffer, sessionId, lastSequence, lastHeartbeatSent = -1, lastHeartbeatAcknowledged = -1, missedHeartbeats = 0, reconnectTimeout, heartbeatTimeout, heartbeatInterval, self = attachEventBus({
+  let resolvedOptions = { ...DEFAULT_SHARD_OPTIONS, ...options }, state = ShardState.Idle, strategy = ShardStrategy.Unknown, ws, resumeGatewayURL, inflater, inflateBuffer, zlibBuffer, sessionId, lastSequence, lastHeartbeatSent = -1, lastHeartbeatAcknowledged = -1, missedHeartbeats = 0, reconnectTimeout, heartbeatTimeout, heartbeatInterval, self = attachEventBus({
     send,
     connect,
     disconnect,
@@ -60,21 +60,27 @@ function createShard(options) {
     let { gateway } = resolvedOptions, baseURL = isResumable() && resumeGatewayURL ? resumeGatewayURL : gateway.baseURL, url = new URL(baseURL);
     url.searchParams.set("v", gateway.version.toString()), url.searchParams.set("encoding", "json"), url.searchParams.set("compress", "zlib-stream"), ws = new WebSocket(url.toString(), {
       perMessageDeflate: false
-    }), zlibBuffer = Buffer.alloc(0), inflater = createInflate({
+    }), inflateBuffer = Buffer.alloc(0), zlibBuffer = Buffer.alloc(0), inflater = createInflate({
       flush: constants.Z_SYNC_FLUSH
     }), inflater.on("data", (chunk) => {
-      try {
-        let payload = JSON.parse(chunk.toString("utf8"));
-        handlePayload(payload);
-      } catch (err) {
-        self.emit("error", err);
+      if (inflateBuffer) {
+        if (inflateBuffer = Buffer.concat([inflateBuffer, chunk]), inflateBuffer.length > 10 * 1024 * 1024) {
+          self.emit("error", new Error("Inflate buffer overflow")), ws?.terminate();
+          return;
+        }
+        try {
+          let text = inflateBuffer.toString("utf8"), payload = JSON.parse(text);
+          inflateBuffer = Buffer.alloc(0), handlePayload(payload);
+        } catch (err) {
+          err instanceof SyntaxError || (inflateBuffer = Buffer.alloc(0)), self.emit("error", err);
+        }
       }
     }), inflater.on("error", (err) => {
       self.emit("error", err), ws?.terminate();
     }), ws.on("message", onMessage), ws.on("close", onClose), ws.on("error", (err) => self.emit("error", err));
   }
   function cleanup() {
-    heartbeatInterval && (clearInterval(heartbeatInterval), heartbeatInterval = void 0), heartbeatTimeout && (clearTimeout(heartbeatTimeout), heartbeatTimeout = void 0), reconnectTimeout && (clearTimeout(reconnectTimeout), reconnectTimeout = void 0), inflater && (inflater.destroy(), inflater = void 0), zlibBuffer && (zlibBuffer = void 0), ws && (ws.readyState !== WebSocket.CLOSED && ws.terminate(), ws.removeAllListeners(), ws = void 0), missedHeartbeats = 0, lastHeartbeatSent = -1, lastHeartbeatAcknowledged = -1;
+    heartbeatInterval && (clearInterval(heartbeatInterval), heartbeatInterval = void 0), heartbeatTimeout && (clearTimeout(heartbeatTimeout), heartbeatTimeout = void 0), reconnectTimeout && (clearTimeout(reconnectTimeout), reconnectTimeout = void 0), inflater && (inflater.destroy(), inflater = void 0), inflateBuffer = void 0, zlibBuffer = void 0, ws && (ws.readyState !== WebSocket.CLOSED && ws.terminate(), ws.removeAllListeners(), ws = void 0), missedHeartbeats = 0, lastHeartbeatSent = -1, lastHeartbeatAcknowledged = -1;
   }
   function connect() {
     return new Promise((resolve) => {
@@ -98,7 +104,7 @@ function createShard(options) {
         ws?.terminate();
         return;
       }
-      zlibBuffer = Buffer.concat([zlibBuffer, data]), zlibBuffer.subarray(zlibBuffer.length - 4).equals(ZLIB_FLUSH) && (inflater?.write(zlibBuffer), zlibBuffer = Buffer.alloc(0));
+      zlibBuffer = Buffer.concat([zlibBuffer, data]), !(zlibBuffer.length < 4 || !zlibBuffer.subarray(zlibBuffer.length - 4).equals(ZLIB_FLUSH)) && (inflater?.write(zlibBuffer), zlibBuffer = Buffer.alloc(0));
     }
   }
   function onClose(code) {
@@ -298,7 +304,7 @@ function bindWorkerToProcess(worker) {
     });
   });
   function send(type, payload) {
-    process.send?.({ type, ...payload });
+    process.send && process.connected && process.send({ type, ...payload });
   }
   process.on("SIGINT", () => {
   }), process.on("SIGTERM", async () => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bakit/gateway",
-  "version": "2.1.4",
+  "version": "2.1.6",
   "description": "Gateway manager for bakit framework",
   "type": "module",
   "exports": {