dandri/server
1
0
Fork 0
mirror of https://github.com/spacebarchat/server.git synced 2026-08-27 22:34:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Gateway close logging for 4002

Browse Source
This commit is contained in:
Rory&
2026-01-13 23:16:10 +01:00
parent 20d7b45ac4
commit 23f6754597
2 changed files with 16 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/gateway/events/Connection.ts
+11 -2
View File
@@ -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);
}
}
src/gateway/events/Message.ts
+5 -1
View File
@@ -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)}`);