Gateway close logging for 4002

2 files changed, 16 insertions, 3 deletions

diff --git a/src/gateway/events/Connection.ts b/src/gateway/events/Connection.ts

index f94484ab..709fc884 100644
--- a/src/gateway/events/Connection.ts
+++ b/src/gateway/events/Connection.ts
@@ -50,10 +50,12 @@ export async function Connection(this: WS.Server, socket: WebSocket, request: In
     socket.userAgent = request.headers["user-agent"] as string;
 
     if (!ipAddress && Config.get().security.cdnSignatureIncludeIp) {
+        console.error("Gateway connection rejected: No IP address found.");
         return socket.close(CLOSECODES.Decode_error, "Gateway connection rejected: IP address is required.");
     }
 
     if (!socket.userAgent && Config.get().security.cdnSignatureIncludeUserAgent) {
+        console.error("Gateway connection rejected: No User-Agent header found.");
         return socket.close(CLOSECODES.Decode_error, "Gateway connection rejected: User-Agent header is required.");
     }
 
@@ -95,12 +97,18 @@ export async function Connection(this: WS.Server, socket: WebSocket, request: In
         const { searchParams } = new URL(`http://localhost${request.url}`);
         // @ts-ignore
         socket.encoding = searchParams.get("encoding") || "json";
-        if (!["json", "etf"].includes(socket.encoding)) return socket.close(CLOSECODES.Decode_error);
+        if (!["json", "etf"].includes(socket.encoding)) {
+            console.error(`[Gateway] Unknown encoding: ${socket.encoding}`);
+            return socket.close(CLOSECODES.Decode_error);
+        }
 
         if (socket.encoding === "etf" && !erlpack) throw new Error("Erlpack is not installed: 'npm i @yukikaze-bot/erlpack'");
 
         socket.version = Number(searchParams.get("version")) || 8;
-        if (socket.version != 8) return socket.close(CLOSECODES.Invalid_API_version);
+        if (socket.version != 8) {
+            console.error(`[Gateway] Invalid API version: ${socket.version}`);
+            return socket.close(CLOSECODES.Invalid_API_version);
+        }
 
         // @ts-ignore
         socket.compress = searchParams.get("compress") || "";
@@ -112,6 +120,7 @@ export async function Connection(this: WS.Server, socket: WebSocket, request: In
                 socket.zstdEncoder = new Encoder(6);
                 socket.zstdDecoder = new Decoder();
             } else {
+                console.error(`[Gateway] Unknown compression: ${socket.compress}`);
                 return socket.close(CLOSECODES.Decode_error);
             }
         }
diff --git a/src/gateway/events/Message.ts b/src/gateway/events/Message.ts

index 0ac57a30..532d765f 100644
--- a/src/gateway/events/Message.ts
+++ b/src/gateway/events/Message.ts
@@ -63,9 +63,13 @@ export async function Message(this: WebSocket, buffer: WS.Data) {
         try {
             data = erlpack.unpack(buffer);
         } catch {
+            console.error("[Gateway] Failed to decode ETF payload");
             return this.close(CLOSECODES.Decode_error);
         }
-    } else return this.close(CLOSECODES.Decode_error);
+    } else {
+        console.error("[Gateway] Unknown payload format");
+        return this.close(CLOSECODES.Decode_error);
+    }
 
     if (process.env.WS_VERBOSE) console.log(`[Websocket] Incomming message: ${JSON.stringify(data)}`);