diff options
Diffstat (limited to 'src/gateway/events/Connection.ts')
-rw-r--r-- | src/gateway/events/Connection.ts | 84 |
1 files changed, 84 insertions, 0 deletions
diff --git a/src/gateway/events/Connection.ts b/src/gateway/events/Connection.ts new file mode 100644 index 00000000..bed3cf44 --- /dev/null +++ b/src/gateway/events/Connection.ts @@ -0,0 +1,84 @@ +import WS from "ws"; +import { WebSocket } from "@fosscord/gateway"; +import { Send } from "../util/Send"; +import { CLOSECODES, OPCODES } from "../util/Constants"; +import { setHeartbeat } from "../util/Heartbeat"; +import { IncomingMessage } from "http"; +import { Close } from "./Close"; +import { Message } from "./Message"; +import { Deflate, Inflate } from "fast-zlib"; +import { URL } from "url"; +import { Config } from "@fosscord/util"; +var erlpack: any; +try { + erlpack = require("@yukikaze-bot/erlpack"); +} catch (error) {} + +// TODO: check rate limit +// TODO: specify rate limit in config +// TODO: check msg max size + +export async function Connection( + this: WS.Server, + socket: WebSocket, + request: IncomingMessage +) { + const forwardedFor = Config.get().security.forwadedFor; + const ipAddress = forwardedFor ? request.headers[forwardedFor] as string : request.socket.remoteAddress; + + socket.ipAddress = ipAddress; + + try { + // @ts-ignore + socket.on("close", Close); + // @ts-ignore + socket.on("message", Message); + console.log(`[Gateway] New connection from ${socket.ipAddress}, total ${this.clients.size}`); + + const { searchParams } = new URL(`http://localhost${request.url}`); + // @ts-ignore + socket.encoding = searchParams.get("encoding") || "json"; + if (!["json", "etf"].includes(socket.encoding)) { + if (socket.encoding === "etf" && erlpack) { + throw new Error( + "Erlpack is not installed: 'npm i @yukikaze-bot/erlpack'" + ); + } + return socket.close(CLOSECODES.Decode_error); + } + + socket.version = Number(searchParams.get("version")) || 8; + if (socket.version != 8) + return socket.close(CLOSECODES.Invalid_API_version); + + // @ts-ignore + socket.compress = searchParams.get("compress") || ""; + if (socket.compress) { + if (socket.compress !== "zlib-stream") + return socket.close(CLOSECODES.Decode_error); + socket.deflate = new Deflate(); + socket.inflate = new Inflate(); + } + + socket.events = {}; + socket.member_events = {}; + socket.permissions = {}; + socket.sequence = 0; + + setHeartbeat(socket); + + await Send(socket, { + op: OPCODES.Hello, + d: { + heartbeat_interval: 1000 * 30, + }, + }); + + socket.readyTimeout = setTimeout(() => { + return socket.close(CLOSECODES.Session_timed_out); + }, 1000 * 30); + } catch (error) { + console.error(error); + return socket.close(CLOSECODES.Unknown_error); + } +} |