diff options
Diffstat (limited to 'webrtc/src/Server.ts')
| -rw-r--r-- | webrtc/src/Server.ts | 243 |
1 files changed, 42 insertions, 201 deletions
diff --git a/webrtc/src/Server.ts b/webrtc/src/Server.ts index 7a1070b9..e2fa8634 100644 --- a/webrtc/src/Server.ts +++ b/webrtc/src/Server.ts @@ -1,215 +1,56 @@ -import { Server as WebSocketServer } from "ws"; -import { WebSocket, CLOSECODES } from "@fosscord/gateway"; -import { Config, initDatabase } from "@fosscord/util"; -import OPCodeHandlers, { Payload } from "./opcodes"; -import { setHeartbeat } from "./util"; -import * as mediasoup from "mediasoup"; -import { types as MediasoupTypes } from "mediasoup"; -import udp from "dgram"; -import sodium from "libsodium-wrappers"; -import { assert } from "console"; - -var port = Number(process.env.PORT); -if (isNaN(port)) port = 3004; +import { closeDatabase, Config, initDatabase, initEvent } from "@fosscord/util"; +import dotenv from "dotenv"; +import http from "http"; +import ws from "ws"; +import { Connection } from "./events/Connection"; +dotenv.config(); export class Server { - public ws: WebSocketServer; - public mediasoupWorkers: MediasoupTypes.Worker[] = []; - public mediasoupRouters: MediasoupTypes.Router[] = []; - public mediasoupTransports: MediasoupTypes.WebRtcTransport[] = []; - public mediasoupProducers: MediasoupTypes.Producer[] = []; - public mediasoupConsumers: MediasoupTypes.Consumer[] = []; - - public decryptKey: Uint8Array; - public testUdp = udp.createSocket("udp6"); - - constructor() { - this.ws = new WebSocketServer({ - port, - maxPayload: 4096, - }); - this.ws.on("connection", async (socket: WebSocket) => { - await setHeartbeat(socket); - - socket.on("message", async (message: string) => { - const payload: Payload = JSON.parse(message); - - if (OPCodeHandlers[payload.op]) - try { - await OPCodeHandlers[payload.op].call(this, socket, payload); - } - catch (e) { - console.error(e); - socket.close(CLOSECODES.Unknown_error); - } - else { - console.error(`Unimplemented`, payload); - socket.close(CLOSECODES.Unknown_opcode); - } + public ws: ws.Server; + public port: number; + public server: http.Server; + public production: boolean; + + constructor({ port, server, production }: { port: number; server?: http.Server; production?: boolean }) { + this.port = port; + this.production = production || false; + + if (server) this.server = server; + else { + this.server = http.createServer(function (req, res) { + res.writeHead(200).end("Online"); }); + } - socket.on("close", (code: number, reason: string) => { - console.log(`client closed ${code} ${reason}`); - for (var consumer of this.mediasoupConsumers) consumer.close(); - for (var producer of this.mediasoupProducers) producer.close(); - for (var transport of this.mediasoupTransports) transport.close(); - - this.mediasoupConsumers = []; - this.mediasoupProducers = []; - this.mediasoupTransports = []; - }) + this.server.on("upgrade", (request, socket, head) => { + if (!request.url?.includes("voice")) return; + this.ws.handleUpgrade(request, socket, head, (socket) => { + // @ts-ignore + socket.server = this; + this.ws.emit("connection", socket, request); + }); }); - this.testUdp.bind(60000); - this.testUdp.on("message", (msg, rinfo) => { - //random key from like, the libsodium examples on npm lol - - //give me my remote port? - if (sodium.to_hex(msg) == "0001004600000001000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000") { - this.testUdp.send(Buffer.from([rinfo.port, 0]), rinfo.port, rinfo.address); - console.log(`got magic packet to send remote port? ${rinfo.address}:${rinfo.port}`); - return; - } - - //Hello - if (sodium.to_hex(msg) == "0100000000000000") { - console.log(`[UDP] client helloed`); - return; - } - - const nonce = Buffer.concat([msg.slice(-4), Buffer.from("\x00".repeat(20))]); - console.log(`[UDP] nonce for this message: ${nonce.toString("hex")}`); - - console.log(`[UDP] message: ${sodium.to_hex(msg)}`); - - // let encrypted; - // if (Buffer.from(msg).indexOf("\x81\xc9") == 0) { - // encrypted = msg.slice(0x18, -4); - // } - // else if (Buffer.from(msg).indexOf("\x90\x78") == 0) { - // encrypted = msg.slice(0x1C, -4); - // } - // else { - // encrypted = msg.slice(0x18, -4); - // console.log(`wtf header received: ${encrypted.toString("hex")}`); - // } - - let encrypted = msg; - - if (sodium.to_hex(msg).indexOf("80c8000600000001") == 0) { - //call status - - encrypted = encrypted.slice(8, -4); - assert(encrypted.length == 40); - - try { - const decrypted = sodium.crypto_secretbox_open_easy(encrypted, nonce, Buffer.from(this.decryptKey)); - console.log("[UDP] [ call status ]" + decrypted); - } - catch (e) { - console.error(`[UDP] decrypt failure\n${e}\n${encrypted.toString("base64")}`); - } - return; - } - - // try { - // const decrypted = sodium.crypto_secretbox_open_easy(encrypted, nonce, Buffer.from(this.decryptKey.map(x => String.fromCharCode(x)).join(""))); - // console.log("[UDP] " + decrypted); - // } - // catch (e) { - // console.error(`[UDP] decrypt failure\n${e}\n${msg.toString("base64")}`); - // } + this.ws = new ws.Server({ + maxPayload: 1024 * 1024 * 100, + noServer: true }); + this.ws.on("connection", Connection); + this.ws.on("error", console.error); } - async listen(): Promise<void> { - // @ts-ignore + async start(): Promise<void> { await initDatabase(); await Config.init(); - await this.createWorkers(); - console.log("[DB] connected"); - console.log(`[WebRTC] online on 0.0.0.0:${port}`); + await initEvent(); + if (!this.server.listening) { + this.server.listen(this.port); + console.log(`[WebRTC] online on 0.0.0.0:${this.port}`); + } } - async createWorkers(): Promise<void> { - const numWorkers = 1; - for (let i = 0; i < numWorkers; i++) { - const worker = await mediasoup.createWorker({ logLevel: "debug", logTags: ["dtls", "ice", "info", "message", "bwe"] }); - if (!worker) return; - - worker.on("died", () => { - console.error("mediasoup worker died"); - }); - - worker.observer.on("newrouter", async (router: MediasoupTypes.Router) => { - console.log("new router created [id:%s]", router.id); - - this.mediasoupRouters.push(router); - - router.observer.on("newtransport", async (transport: MediasoupTypes.WebRtcTransport) => { - console.log("new transport created [id:%s]", transport.id); - - await transport.enableTraceEvent(); - - transport.on('dtlsstatechange', (dtlsstate) => { - console.log(dtlsstate); - }); - - transport.on("sctpstatechange", (sctpstate) => { - console.log(sctpstate); - }); - - router.observer.on("newrtpobserver", (rtpObserver: MediasoupTypes.RtpObserver) => { - console.log("new RTP observer created [id:%s]", rtpObserver.id); - - // rtpObserver.observer.on("") - }); - - transport.on("connect", () => { - console.log("transport connect"); - }); - - transport.observer.on("newproducer", (producer: MediasoupTypes.Producer) => { - console.log("new producer created [id:%s]", producer.id); - - this.mediasoupProducers.push(producer); - }); - - transport.observer.on("newconsumer", (consumer: MediasoupTypes.Consumer) => { - console.log("new consumer created [id:%s]", consumer.id); - - this.mediasoupConsumers.push(consumer); - - consumer.on("rtp", (rtpPacket) => { - console.log(rtpPacket); - }); - }); - - transport.observer.on("newdataproducer", (dataProducer) => { - console.log("new data producer created [id:%s]", dataProducer.id); - }); - - transport.on("trace", (trace) => { - console.log(trace); - }); - - this.mediasoupTransports.push(transport); - }); - }); - - await worker.createRouter({ - mediaCodecs: [ - { - kind: "audio", - mimeType: "audio/opus", - clockRate: 48000, - channels: 2, - preferredPayloadType: 111, - }, - ], - }); - - this.mediasoupWorkers.push(worker); - } + async stop() { + closeDatabase(); + this.server.close(); } -} +} \ No newline at end of file |