1
0
mirror of https://github.com/spacebarchat/server.git synced 2024-11-10 12:42:44 +01:00

Probably broken merge from webrtc

This commit is contained in:
Madeline 2022-09-16 12:54:02 +10:00
parent 725491a69d
commit 493c21968b
35 changed files with 864 additions and 749 deletions

View File

@ -198,10 +198,7 @@
"type": "integer"
},
"video_quality_mode": {
"type": [
"null",
"integer"
]
"type": "integer"
}
},
"additionalProperties": false,
@ -716,10 +713,7 @@
"type": "integer"
},
"video_quality_mode": {
"type": [
"null",
"integer"
]
"type": "integer"
}
},
"additionalProperties": false
@ -1285,5 +1279,241 @@
"type": "object",
"additionalProperties": false,
"$schema": "http://json-schema.org/draft-07/schema#"
},
"VoiceVideoSchema": {
"type": "object",
"properties": {
"audio_ssrc": {
"type": "integer"
},
"video_ssrc": {
"type": "integer"
},
"rtx_ssrc": {
"type": "integer"
},
"user_id": {
"type": "string"
},
"streams": {
"type": "array",
"items": {
"type": "object",
"properties": {
"type": {
"enum": [
"audio",
"video"
],
"type": "string"
},
"rid": {
"type": "string"
},
"ssrc": {
"type": "integer"
},
"active": {
"type": "boolean"
},
"quality": {
"type": "integer"
},
"rtx_ssrc": {
"type": "integer"
},
"max_bitrate": {
"type": "integer"
},
"max_framerate": {
"type": "integer"
},
"max_resolution": {
"type": "object",
"properties": {
"type": {
"type": "string"
},
"width": {
"type": "integer"
},
"height": {
"type": "integer"
}
},
"additionalProperties": false,
"required": [
"height",
"type",
"width"
]
}
},
"additionalProperties": false,
"required": [
"active",
"max_bitrate",
"max_framerate",
"max_resolution",
"quality",
"rid",
"rtx_ssrc",
"ssrc",
"type"
]
}
}
},
"additionalProperties": false,
"required": [
"audio_ssrc",
"video_ssrc"
],
"$schema": "http://json-schema.org/draft-07/schema#"
},
"VoiceIdentifySchema": {
"type": "object",
"properties": {
"server_id": {
"type": "string"
},
"user_id": {
"type": "string"
},
"session_id": {
"type": "string"
},
"token": {
"type": "string"
},
"video": {
"type": "boolean"
},
"streams": {
"type": "array",
"items": {
"type": "object",
"properties": {
"type": {
"type": "string"
},
"rid": {
"type": "string"
},
"quality": {
"type": "integer"
}
},
"additionalProperties": false,
"required": [
"quality",
"rid",
"type"
]
}
}
},
"additionalProperties": false,
"required": [
"server_id",
"session_id",
"token",
"user_id"
],
"$schema": "http://json-schema.org/draft-07/schema#"
},
"SelectProtocolSchema": {
"type": "object",
"properties": {
"protocol": {
"enum": [
"udp",
"webrtc"
],
"type": "string"
},
"data": {
"anyOf": [
{
"type": "object",
"properties": {
"address": {
"type": "string"
},
"port": {
"type": "integer"
},
"mode": {
"type": "string"
}
},
"additionalProperties": false,
"required": [
"address",
"mode",
"port"
]
},
{
"type": "string"
}
]
},
"sdp": {
"type": "string"
},
"codecs": {
"type": "array",
"items": {
"type": "object",
"properties": {
"name": {
"enum": [
"H264",
"VP8",
"VP9",
"opus"
],
"type": "string"
},
"type": {
"enum": [
"audio",
"video"
],
"type": "string"
},
"priority": {
"type": "integer"
},
"payload_type": {
"type": "integer"
},
"rtx_payload_type": {
"type": [
"null",
"integer"
]
}
},
"additionalProperties": false,
"required": [
"name",
"payload_type",
"priority",
"type"
]
}
},
"rtc_connection_id": {
"type": "string"
}
},
"additionalProperties": false,
"required": [
"data",
"protocol"
],
"$schema": "http://json-schema.org/draft-07/schema#"
}
}

View File

@ -1,3 +1,4 @@
export * from "./Server";
export * from "./middlewares/";
export * from "./util/";
export * from "./voice_schema_hack";

View File

@ -68,6 +68,7 @@
"baseUrl": ".",
"paths": {
"@fosscord/api": ["src/index"]
"@fosscord/util": ["../util/src/index"]
},
"plugins": [{ "transform": "@zerollup/ts-transform-paths" }],
"experimentalDecorators": true

View File

@ -29,6 +29,7 @@
}
],
"settings": {
"typescript.tsdk": "util\\node_modules\\typescript\\lib"
"typescript.tsdk": "util\\node_modules\\typescript\\lib",
"liveServer.settings.multiRootWorkspaceName": "slowcord"
}
}

View File

@ -1,3 +1,5 @@
import { VoiceOPCodes } from "@fosscord/webrtc";
export enum OPCODES {
Dispatch = 0,
Heartbeat = 1,
@ -43,7 +45,7 @@ export enum CLOSECODES {
}
export interface Payload {
op: OPCODES;
op: OPCODES | VoiceOPCodes;
d?: any;
s?: number;
t?: string;

View File

@ -1,6 +1,7 @@
import { Intents, Permissions } from "@fosscord/util";
import WS from "ws";
import { Deflate, Inflate } from "fast-zlib";
import { Client } from "@fosscord/webrtc";
export interface WebSocket extends WS {
version: number;
@ -21,4 +22,5 @@ export interface WebSocket extends WS {
events: Record<string, Function>;
member_events: Record<string, Function>;
listen_options: any;
client?: Client;
}

View File

@ -77,8 +77,8 @@
"baseUrl": ".",
"paths": {
"@fosscord/gateway": ["src/index.ts"],
"@fosscord/gateway/*": ["src/*"]
"@fosscord/util": ["../util/src"]
"@fosscord/util": ["../util/src/index"],
"@fosscord/webrtc": ["../webrtc/src/index"]
},
"plugins": [{ "transform": "@zerollup/ts-transform-paths" }]
}

View File

@ -4,3 +4,4 @@ export * from "./util/index";
export * from "./interfaces/index";
export * from "./entities/index";
export * from "./dtos/index";
export * from "./schemas";

View File

@ -0,0 +1,54 @@
import Ajv from "ajv";
import addFormats from "ajv-formats";
import fs from "fs";
import path from "path";
const SchemaPath = path.join(__dirname, "..", "..", "..", "assets", "schemas.json");
const schemas = JSON.parse(fs.readFileSync(SchemaPath, { encoding: "utf8" }));
export const ajv = new Ajv({
allErrors: true,
parseDate: true,
allowDate: true,
schemas,
coerceTypes: true,
messages: true,
strict: true,
strictRequired: true
});
addFormats(ajv);
export function validateSchema<G>(schema: string, data: G): G {
const valid = ajv.validate(schema, normalizeBody(data));
if (!valid) throw ajv.errors;
return data;
}
// Normalizer is introduced to workaround https://github.com/ajv-validator/ajv/issues/1287
// this removes null values as ajv doesn't treat them as undefined
// normalizeBody allows to handle circular structures without issues
// taken from https://github.com/serverless/serverless/blob/master/lib/classes/ConfigSchemaHandler/index.js#L30 (MIT license)
export const normalizeBody = (body: any = {}) => {
const normalizedObjectsSet = new WeakSet();
const normalizeObject = (object: any) => {
if (normalizedObjectsSet.has(object)) return;
normalizedObjectsSet.add(object);
if (Array.isArray(object)) {
for (const [index, value] of object.entries()) {
if (typeof value === "object") normalizeObject(value);
}
} else {
for (const [key, value] of Object.entries(object)) {
if (value == null) {
if (key === "icon" || key === "avatar" || key === "banner" || key === "splash" || key === "discovery_splash") continue;
delete object[key];
} else if (typeof value === "object") {
normalizeObject(value);
}
}
}
};
normalizeObject(body);
return body;
};

View File

@ -0,0 +1,2 @@
export * from "./Validator";
export * from "./voice";

69
util/src/schemas/voice.ts Normal file
View File

@ -0,0 +1,69 @@
export interface VoiceVideoSchema {
audio_ssrc: number;
video_ssrc: number;
rtx_ssrc?: number;
user_id?: string;
streams?: {
type: "video" | "audio";
rid: string;
ssrc: number;
active: boolean;
quality: number;
rtx_ssrc: number;
max_bitrate: number;
max_framerate: number;
max_resolution: { type: string; width: number; height: number; };
}[];
}
export const VoiceStateUpdateSchema = {
$guild_id: String,
$channel_id: String,
self_mute: Boolean,
self_deaf: Boolean,
self_video: Boolean
};
//TODO need more testing when community guild and voice stage channel are working
export interface VoiceStateUpdateSchema {
channel_id: string;
guild_id?: string;
suppress?: boolean;
request_to_speak_timestamp?: Date;
self_mute?: boolean;
self_deaf?: boolean;
self_video?: boolean;
}
export interface VoiceIdentifySchema {
server_id: string;
user_id: string;
session_id: string;
token: string;
video?: boolean;
streams?: {
type: string;
rid: string;
quality: number;
}[];
}
export interface SelectProtocolSchema {
protocol: "webrtc" | "udp";
data:
| string
| {
address: string;
port: number;
mode: string;
};
sdp?: string;
codecs?: {
name: "opus" | "VP8" | "VP9" | "H264";
type: "audio" | "video";
priority: number;
payload_type: number;
rtx_payload_type?: number | null;
}[];
rtc_connection_id?: string; // uuid
}

119
webrtc/package-lock.json generated
View File

@ -13,9 +13,7 @@
"dotenv": "^12.0.4",
"libsodium": "^0.7.10",
"libsodium-wrappers": "^0.7.10",
"mediasoup": "^3.9.5",
"node-turn": "^0.0.6",
"sdp-transform": "^2.14.1",
"tsconfig-paths": "^3.12.0",
"ws": "^7.5.8"
},
@ -270,17 +268,6 @@
"resolved": "https://registry.npmjs.org/graceful-fs/-/graceful-fs-4.2.6.tgz",
"integrity": "sha512-nTnJ528pbqxYanhpDYsi4Rd8MAeaBA67+RZ10CM1m3bTAVFEDcd5AuA4a6W5YkGZ1iNXHzZz8T6TBKLeBuNriQ=="
},
"node_modules/h264-profile-level-id": {
"version": "1.0.1",
"resolved": "https://registry.npmjs.org/h264-profile-level-id/-/h264-profile-level-id-1.0.1.tgz",
"integrity": "sha512-D3Rln/jKNjKDW5ZTJTK3niSoOGE+pFqPvRHHVgQN3G7umcn/zWGPUo8Q8VpDj16x3hKz++zVviRNRmXu5cpN+Q==",
"dependencies": {
"debug": "^4.1.1"
},
"engines": {
"node": ">=8.0.0"
}
},
"node_modules/ieee754": {
"version": "1.2.1",
"resolved": "https://registry.npmjs.org/ieee754/-/ieee754-1.2.1.tgz",
@ -365,32 +352,6 @@
"integrity": "sha512-s8UhlNe7vPKomQhC1qFelMokr/Sc3AgNbso3n74mVPA5LTZwkB9NlXf4XPamLxJE8h0gh73rM94xvwRT2CVInw==",
"dev": true
},
"node_modules/mediasoup": {
"version": "3.9.5",
"resolved": "https://registry.npmjs.org/mediasoup/-/mediasoup-3.9.5.tgz",
"integrity": "sha512-8lISnN5cbtSvdqHeuyxhCTFTHudoq/EpgLcDB0d0pT5RG18mZlHF5BwIBSkGxB/nWyeTfTGPpGBiNtKoubbRXA==",
"hasInstallScript": true,
"dependencies": {
"@types/node": "^16.11.10",
"debug": "^4.3.3",
"h264-profile-level-id": "^1.0.1",
"random-number": "^0.0.9",
"supports-color": "^9.2.1",
"uuid": "^8.3.2"
},
"engines": {
"node": ">=12"
},
"funding": {
"type": "opencollective",
"url": "https://opencollective.com/mediasoup"
}
},
"node_modules/mediasoup/node_modules/@types/node": {
"version": "16.11.19",
"resolved": "https://registry.npmjs.org/@types/node/-/node-16.11.19.tgz",
"integrity": "sha512-BPAcfDPoHlRQNKktbsbnpACGdypPFBuX4xQlsWDE7B8XXcfII+SpOLay3/qZmCLb39kV5S1RTYwXdkx2lwLYng=="
},
"node_modules/minimist": {
"version": "1.2.5",
"resolved": "https://registry.npmjs.org/minimist/-/minimist-1.2.5.tgz",
@ -411,24 +372,11 @@
"log4js": "~6.3.0"
}
},
"node_modules/random-number": {
"version": "0.0.9",
"resolved": "https://registry.npmjs.org/random-number/-/random-number-0.0.9.tgz",
"integrity": "sha512-ipG3kRCREi/YQpi2A5QGcvDz1KemohovWmH6qGfboVyyGdR2t/7zQz0vFxrfxpbHQgPPdtVlUDaks3aikD1Ljw=="
},
"node_modules/rfdc": {
"version": "1.3.0",
"resolved": "https://registry.npmjs.org/rfdc/-/rfdc-1.3.0.tgz",
"integrity": "sha512-V2hovdzFbOi77/WajaSMXk2OLm+xNIeQdMMuB7icj7bk6zi2F8GGAxigcnDFpJHbNyNcgyJDiP+8nOrY5cZGrA=="
},
"node_modules/sdp-transform": {
"version": "2.14.1",
"resolved": "https://registry.npmjs.org/sdp-transform/-/sdp-transform-2.14.1.tgz",
"integrity": "sha512-RjZyX3nVwJyCuTo5tGPx+PZWkDMCg7oOLpSlhjDdZfwUoNqG1mM8nyj31IGHyaPWXhjbP7cdK3qZ2bmkJ1GzRw==",
"bin": {
"sdp-verify": "checker.js"
}
},
"node_modules/sprintf-js": {
"version": "1.0.3",
"resolved": "https://registry.npmjs.org/sprintf-js/-/sprintf-js-1.0.3.tgz",
@ -463,17 +411,6 @@
"node": ">=4"
}
},
"node_modules/supports-color": {
"version": "9.2.1",
"resolved": "https://registry.npmjs.org/supports-color/-/supports-color-9.2.1.tgz",
"integrity": "sha512-Obv7ycoCTG51N7y175StI9BlAXrmgZrFhZOb0/PyjHBher/NmsdBgbbQ1Inhq+gIhz6+7Gb+jWF2Vqi7Mf1xnQ==",
"engines": {
"node": ">=12"
},
"funding": {
"url": "https://github.com/chalk/supports-color?sponsor=1"
}
},
"node_modules/ts-node": {
"version": "10.4.0",
"resolved": "https://registry.npmjs.org/ts-node/-/ts-node-10.4.0.tgz",
@ -547,14 +484,6 @@
"node": ">= 4.0.0"
}
},
"node_modules/uuid": {
"version": "8.3.2",
"resolved": "https://registry.npmjs.org/uuid/-/uuid-8.3.2.tgz",
"integrity": "sha512-+NYs2QeMWy+GWFOEm9xnn6HCDp0l7QBD7ml8zLUmJ+93Q5NF0NocErnwkTkXVFNiX3/fpC6afS8Dhb/gz7R7eg==",
"bin": {
"uuid": "dist/bin/uuid"
}
},
"node_modules/ws": {
"version": "7.5.8",
"resolved": "https://registry.npmjs.org/ws/-/ws-7.5.8.tgz",
@ -759,14 +688,6 @@
"resolved": "https://registry.npmjs.org/graceful-fs/-/graceful-fs-4.2.6.tgz",
"integrity": "sha512-nTnJ528pbqxYanhpDYsi4Rd8MAeaBA67+RZ10CM1m3bTAVFEDcd5AuA4a6W5YkGZ1iNXHzZz8T6TBKLeBuNriQ=="
},
"h264-profile-level-id": {
"version": "1.0.1",
"resolved": "https://registry.npmjs.org/h264-profile-level-id/-/h264-profile-level-id-1.0.1.tgz",
"integrity": "sha512-D3Rln/jKNjKDW5ZTJTK3niSoOGE+pFqPvRHHVgQN3G7umcn/zWGPUo8Q8VpDj16x3hKz++zVviRNRmXu5cpN+Q==",
"requires": {
"debug": "^4.1.1"
}
},
"ieee754": {
"version": "1.2.1",
"resolved": "https://registry.npmjs.org/ieee754/-/ieee754-1.2.1.tgz",
@ -828,26 +749,6 @@
"integrity": "sha512-s8UhlNe7vPKomQhC1qFelMokr/Sc3AgNbso3n74mVPA5LTZwkB9NlXf4XPamLxJE8h0gh73rM94xvwRT2CVInw==",
"dev": true
},
"mediasoup": {
"version": "3.9.5",
"resolved": "https://registry.npmjs.org/mediasoup/-/mediasoup-3.9.5.tgz",
"integrity": "sha512-8lISnN5cbtSvdqHeuyxhCTFTHudoq/EpgLcDB0d0pT5RG18mZlHF5BwIBSkGxB/nWyeTfTGPpGBiNtKoubbRXA==",
"requires": {
"@types/node": "^16.11.10",
"debug": "^4.3.3",
"h264-profile-level-id": "^1.0.1",
"random-number": "^0.0.9",
"supports-color": "^9.2.1",
"uuid": "^8.3.2"
},
"dependencies": {
"@types/node": {
"version": "16.11.19",
"resolved": "https://registry.npmjs.org/@types/node/-/node-16.11.19.tgz",
"integrity": "sha512-BPAcfDPoHlRQNKktbsbnpACGdypPFBuX4xQlsWDE7B8XXcfII+SpOLay3/qZmCLb39kV5S1RTYwXdkx2lwLYng=="
}
}
},
"minimist": {
"version": "1.2.5",
"resolved": "https://registry.npmjs.org/minimist/-/minimist-1.2.5.tgz",
@ -868,21 +769,11 @@
"log4js": "~6.3.0"
}
},
"random-number": {
"version": "0.0.9",
"resolved": "https://registry.npmjs.org/random-number/-/random-number-0.0.9.tgz",
"integrity": "sha512-ipG3kRCREi/YQpi2A5QGcvDz1KemohovWmH6qGfboVyyGdR2t/7zQz0vFxrfxpbHQgPPdtVlUDaks3aikD1Ljw=="
},
"rfdc": {
"version": "1.3.0",
"resolved": "https://registry.npmjs.org/rfdc/-/rfdc-1.3.0.tgz",
"integrity": "sha512-V2hovdzFbOi77/WajaSMXk2OLm+xNIeQdMMuB7icj7bk6zi2F8GGAxigcnDFpJHbNyNcgyJDiP+8nOrY5cZGrA=="
},
"sdp-transform": {
"version": "2.14.1",
"resolved": "https://registry.npmjs.org/sdp-transform/-/sdp-transform-2.14.1.tgz",
"integrity": "sha512-RjZyX3nVwJyCuTo5tGPx+PZWkDMCg7oOLpSlhjDdZfwUoNqG1mM8nyj31IGHyaPWXhjbP7cdK3qZ2bmkJ1GzRw=="
},
"sprintf-js": {
"version": "1.0.3",
"resolved": "https://registry.npmjs.org/sprintf-js/-/sprintf-js-1.0.3.tgz",
@ -910,11 +801,6 @@
"resolved": "https://registry.npmjs.org/strip-bom/-/strip-bom-3.0.0.tgz",
"integrity": "sha1-IzTBjpx1n3vdVv3vfprj1YjmjtM="
},
"supports-color": {
"version": "9.2.1",
"resolved": "https://registry.npmjs.org/supports-color/-/supports-color-9.2.1.tgz",
"integrity": "sha512-Obv7ycoCTG51N7y175StI9BlAXrmgZrFhZOb0/PyjHBher/NmsdBgbbQ1Inhq+gIhz6+7Gb+jWF2Vqi7Mf1xnQ=="
},
"ts-node": {
"version": "10.4.0",
"resolved": "https://registry.npmjs.org/ts-node/-/ts-node-10.4.0.tgz",
@ -957,11 +843,6 @@
"resolved": "https://registry.npmjs.org/universalify/-/universalify-0.1.2.tgz",
"integrity": "sha512-rBJeI5CXAlmy1pV+617WB9J63U6XcazHHF2f2dbJix4XzpUF0RS3Zbj0FGIOCAva5P/d/GBOYaACQ1w+0azUkg=="
},
"uuid": {
"version": "8.3.2",
"resolved": "https://registry.npmjs.org/uuid/-/uuid-8.3.2.tgz",
"integrity": "sha512-+NYs2QeMWy+GWFOEm9xnn6HCDp0l7QBD7ml8zLUmJ+93Q5NF0NocErnwkTkXVFNiX3/fpC6afS8Dhb/gz7R7eg=="
},
"ws": {
"version": "7.5.8",
"resolved": "https://registry.npmjs.org/ws/-/ws-7.5.8.tgz",

View File

@ -2,7 +2,8 @@
"name": "rtc",
"version": "1.0.0",
"description": "A javascript fosscord webrtc server for voice and video communication",
"main": "index.js",
"main": "dist/index.js",
"types": "src/index.ts",
"scripts": {
"test": "npm run build && node dist/test.js",
"build": "npx tsc -p .",
@ -23,9 +24,7 @@
"dotenv": "^12.0.4",
"libsodium": "^0.7.10",
"libsodium-wrappers": "^0.7.10",
"mediasoup": "^3.9.5-1",
"node-turn": "^0.0.6",
"sdp-transform": "^2.14.1",
"tsconfig-paths": "^3.12.0",
"ws": "^7.5.8"
}

View File

@ -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 ws: ws.Server;
public port: number;
public server: http.Server;
public production: boolean;
public decryptKey: Uint8Array;
public testUdp = udp.createSocket("udp6");
constructor({ port, server, production }: { port: number; server?: http.Server; production?: boolean }) {
this.port = port;
this.production = production || false;
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);
}
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}`);
}
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);
await initEvent();
if (!this.server.listening) {
this.server.listen(this.port);
console.log(`[WebRTC] online on 0.0.0.0:${this.port}`);
}
}
}
async stop() {
closeDatabase();
this.server.close();
}
}

View File

@ -0,0 +1,9 @@
import { WebSocket } from "@fosscord/gateway";
import { Session } from "@fosscord/util";
export async function onClose(this: WebSocket, code: number, reason: string) {
console.log("[WebRTC] closed", code, reason.toString());
if (this.session_id) await Session.delete({ session_id: this.session_id });
this.removeAllListeners();
}

View File

@ -0,0 +1,60 @@
import { CLOSECODES, Send, setHeartbeat, WebSocket } from "@fosscord/gateway";
import { IncomingMessage } from "http";
import { URL } from "url";
import WS from "ws";
import { VoiceOPCodes } from "../util";
import { onClose } from "./Close";
import { onMessage } from "./Message";
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) {
try {
socket.on("close", onClose.bind(socket));
socket.on("message", onMessage.bind(socket));
console.log("[WebRTC] new connection", request.url);
if (process.env.WS_LOGEVENTS) {
[
"close",
"error",
"upgrade",
//"message",
"open",
"ping",
"pong",
"unexpected-response"
].forEach((x) => {
socket.on(x, (y) => console.log("[WebRTC]", x, y));
});
}
const { searchParams } = new URL(`http://localhost${request.url}`);
socket.encoding = "json";
socket.version = Number(searchParams.get("v")) || 5;
if (socket.version < 3) return socket.close(CLOSECODES.Unknown_error, "invalid version");
setHeartbeat(socket);
socket.readyTimeout = setTimeout(() => {
return socket.close(CLOSECODES.Session_timed_out);
}, 1000 * 30);
await Send(socket, {
op: VoiceOPCodes.HELLO,
d: {
heartbeat_interval: 1000 * 30
}
});
} catch (error) {
console.error("[WebRTC]", error);
return socket.close(CLOSECODES.Unknown_error);
}
}

View File

@ -0,0 +1,38 @@
import { CLOSECODES, Payload, WebSocket } from "@fosscord/gateway";
import { Tuple } from "lambert-server";
import OPCodeHandlers from "../opcodes";
import { VoiceOPCodes } from "../util";
const PayloadSchema = {
op: Number,
$d: new Tuple(Object, Number), // or number for heartbeat sequence
$s: Number,
$t: String
};
export async function onMessage(this: WebSocket, buffer: Buffer) {
try {
var data: Payload = JSON.parse(buffer.toString());
if (data.op !== VoiceOPCodes.IDENTIFY && !this.user_id) return this.close(CLOSECODES.Not_authenticated);
// @ts-ignore
const OPCodeHandler = OPCodeHandlers[data.op];
if (!OPCodeHandler) {
// @ts-ignore
console.error("[WebRTC] Unkown opcode " + VoiceOPCodes[data.op]);
// TODO: if all opcodes are implemented comment this out:
// this.close(CloseCodes.Unknown_opcode);
return;
}
if (![VoiceOPCodes.HEARTBEAT, VoiceOPCodes.SPEAKING].includes(data.op as VoiceOPCodes)) {
// @ts-ignore
console.log("[WebRTC] Opcode " + VoiceOPCodes[data.op]);
}
return await OPCodeHandler.call(this, data);
} catch (error) {
console.error("[WebRTC] error", error);
// if (!this.CLOSED && this.CLOSING) return this.close(CloseCodes.Unknown_error);
}
}

View File

@ -0,0 +1,2 @@
export * from "./Server";
export * from "./util/index";

View File

@ -0,0 +1,6 @@
import { Payload, Send, WebSocket } from "@fosscord/gateway";
import { VoiceOPCodes } from "../util";
export async function onBackendVersion(this: WebSocket, data: Payload) {
await Send(this, { op: VoiceOPCodes.VOICE_BACKEND_VERSION, d: { voice: "0.8.43", rtc_worker: "0.3.26" } });
}

View File

@ -1,40 +0,0 @@
import { WebSocket } from "@fosscord/gateway";
import { Payload } from "./index";
import { Server } from "../Server"
/*
Sent by client:
{
"op": 12,
"d": {
"audio_ssrc": 0,
"video_ssrc": 0,
"rtx_ssrc": 0,
"streams": [
{
"type": "video",
"rid": "100",
"ssrc": 0,
"active": false,
"quality": 100,
"rtx_ssrc": 0,
"max_bitrate": 2500000,
"max_framerate": 20,
"max_resolution": {
"type": "fixed",
"width": 1280,
"height": 720
}
}
]
}
}
*/
export async function onConnect(this: Server, socket: WebSocket, data: Payload) {
socket.send(JSON.stringify({ //what is op 15?
op: 15,
d: { any: 100 }
}))
}

View File

@ -1,8 +1,9 @@
import { WebSocket } from "@fosscord/gateway";
import { Payload } from "./index";
import { setHeartbeat } from "../util";
import { Server } from "../Server"
import { CLOSECODES, Payload, Send, setHeartbeat, WebSocket } from "@fosscord/gateway";
import { VoiceOPCodes } from "../util";
export async function onHeartbeat(this: Server, socket: WebSocket, data: Payload) {
await setHeartbeat(socket, data.d);
export async function onHeartbeat(this: WebSocket, data: Payload) {
setHeartbeat(this);
if (isNaN(data.d)) return this.close(CLOSECODES.Decode_error);
await Send(this, { op: VoiceOPCodes.HEARTBEAT_ACK, d: data.d });
}

View File

@ -1,50 +1,50 @@
import { WebSocket, CLOSECODES } from "@fosscord/gateway";
import { Payload } from "./index";
import { VoiceOPCodes, Session, User, Guild } from "@fosscord/util";
import { Server } from "../Server";
import { CLOSECODES, Payload, Send, WebSocket } from "@fosscord/gateway";
import { validateSchema, VoiceIdentifySchema, VoiceState } from "@fosscord/util";
import { endpoint, getClients, VoiceOPCodes } from "@fosscord/webrtc";
import SemanticSDP from "semantic-sdp";
const defaultSDP = require("../../../assets/sdp.json");
export interface IdentifyPayload extends Payload {
d: {
server_id: string, //guild id
session_id: string, //gateway session
streams: Array<{
type: string,
rid: string, //number
quality: number,
}>,
token: string, //voice_states token
user_id: string,
video: boolean,
export async function onIdentify(this: WebSocket, data: Payload) {
clearTimeout(this.readyTimeout);
const { server_id, user_id, session_id, token, streams, video } = validateSchema("VoiceIdentifySchema", data.d) as VoiceIdentifySchema;
const voiceState = await VoiceState.findOne({ guild_id: server_id, user_id, token, session_id });
if (!voiceState) return this.close(CLOSECODES.Authentication_failed);
this.user_id = user_id;
this.session_id = session_id;
const sdp = SemanticSDP.SDPInfo.expand(defaultSDP);
sdp.setDTLS(SemanticSDP.DTLSInfo.expand({ setup: "actpass", hash: "sha-256", fingerprint: endpoint.getDTLSFingerprint() }));
this.client = {
websocket: this,
out: {
tracks: new Map()
},
in: {
audio_ssrc: 0,
video_ssrc: 0,
rtx_ssrc: 0
},
sdp,
channel_id: voiceState.channel_id
};
}
export async function onIdentify(this: Server, socket: WebSocket, data: Payload) {
const clients = getClients(voiceState.channel_id)!;
clients.add(this.client);
const session = await Session.findOneOrFail(
{ session_id: data.d.session_id, },
{
where: { user_id: data.d.user_id },
relations: ["user"]
}
);
const user = session.user;
const guild = await Guild.findOneOrFail({ id: data.d.server_id }, { relations: ["members"] });
if (!guild.members.find(x => x.id === user.id))
return socket.close(CLOSECODES.Invalid_intent);
var transport = this.mediasoupTransports[0] || await this.mediasoupRouters[0].createWebRtcTransport({
listenIps: [{ ip: "10.22.64.146" }],
enableUdp: true,
this.on("close", () => {
clients.delete(this.client!);
});
socket.send(JSON.stringify({
await Send(this, {
op: VoiceOPCodes.READY,
d: {
streams: data.d.streams ? [...data.d.streams.map(x => ({ ...x, rtx_ssrc: Math.floor(Math.random() * 10000), ssrc: Math.floor(Math.random() * 10000), active: true, }))] : undefined,
ssrc: Math.floor(Math.random() * 10000),
ip: transport.iceCandidates[0].ip,
port: transport.iceCandidates[0].port,
streams: [
// { type: "video", ssrc: this.ssrc + 1, rtx_ssrc: this.ssrc + 2, rid: "100", quality: 100, active: false }
],
ssrc: -1,
port: endpoint.getLocalPort(),
modes: [
"aead_aes256_gcm_rtpsize",
"aead_aes256_gcm",
@ -53,11 +53,8 @@ export async function onIdentify(this: Server, socket: WebSocket, data: Payload)
"xsalsa20_poly1305_suffix",
"xsalsa20_poly1305"
],
experiments: [
"bwe_conservative_link_estimate",
"bwe_remote_locus_client",
"fixed_keyframe_interval"
]
},
}));
ip: "127.0.0.1",
experiments: []
}
});
}

View File

@ -1,24 +0,0 @@
import { CLOSECODES, WebSocket } from "@fosscord/gateway";
import { Payload } from "./index";
import { Server } from "../Server"
import { Guild, Session, VoiceOPCodes } from "@fosscord/util";
export async function onResume(this: Server, socket: WebSocket, data: Payload) {
const session = await Session.findOneOrFail(
{ session_id: data.d.session_id, },
{
where: { user_id: data.d.user_id },
relations: ["user"]
}
);
const user = session.user;
const guild = await Guild.findOneOrFail({ id: data.d.server_id }, { relations: ["members"] });
if (!guild.members.find(x => x.id === user.id))
return socket.close(CLOSECODES.Invalid_intent);
socket.send(JSON.stringify({
op: VoiceOPCodes.RESUMED,
d: null,
}))
}

View File

@ -1,206 +1,46 @@
import { WebSocket } from "@fosscord/gateway";
import { Payload } from "./index";
import { VoiceOPCodes } from "@fosscord/util";
import { Server } from "../Server";
import * as mediasoup from "mediasoup";
import { RtpCodecCapability } from "mediasoup/node/lib/RtpParameters";
import * as sdpTransform from 'sdp-transform';
import sodium from "libsodium-wrappers";
import { Payload, Send, WebSocket } from "@fosscord/gateway";
import { SelectProtocolSchema, validateSchema } from "@fosscord/util";
import { endpoint, PublicIP, VoiceOPCodes } from "@fosscord/webrtc";
import SemanticSDP from "semantic-sdp";
export interface CodecPayload {
name: string,
type: "audio" | "video",
priority: number,
payload_type: number,
rtx_payload_type: number | null,
}
export async function onSelectProtocol(this: WebSocket, payload: Payload) {
if (!this.client) return;
export interface SelectProtocolPayload extends Payload {
d: {
codecs: Array<CodecPayload>,
data: string, // SDP if webrtc
protocol: string,
rtc_connection_id: string,
sdp?: string, // same as data
};
}
const data = validateSchema("SelectProtocolSchema", payload.d) as SelectProtocolSchema;
/*
const offer = SemanticSDP.SDPInfo.parse("m=audio\n" + data.sdp!);
this.client.sdp!.setICE(offer.getICE());
this.client.sdp!.setDTLS(offer.getDTLS());
Sent by client:
{
"op": 1,
"d": {
"protocol": "webrtc",
"data": "
a=extmap-allow-mixed
a=ice-ufrag:vNxb
a=ice-pwd:tZvpbVPYEKcnW0gGRPq0OOnh
a=ice-options:trickle
a=extmap:1 urn:ietf:params:rtp-hdrext:ssrc-audio-level
a=extmap:2 http://www.webrtc.org/experiments/rtp-hdrext/abs-send-time
a=extmap:3 http://www.ietf.org/id/draft-holmer-rmcat-transport-wide-cc-extensions-01
a=extmap:4 urn:ietf:params:rtp-hdrext:sdes:mid
a=rtpmap:111 opus/48000/2
a=extmap:14 urn:ietf:params:rtp-hdrext:toffset
a=extmap:13 urn:3gpp:video-orientation
a=extmap:5 http://www.webrtc.org/experiments/rtp-hdrext/playout-delay
a=extmap:6 http://www.webrtc.org/experiments/rtp-hdrext/video-content-type
a=extmap:7 http://www.webrtc.org/experiments/rtp-hdrext/video-timing
a=extmap:8 http://www.webrtc.org/experiments/rtp-hdrext/color-space
a=extmap:10 urn:ietf:params:rtp-hdrext:sdes:rtp-stream-id
a=extmap:11 urn:ietf:params:rtp-hdrext:sdes:repaired-rtp-stream-id
a=rtpmap:96 VP8/90000
a=rtpmap:97 rtx/90000
",
"codecs": [
{
"name": "opus",
"type": "audio",
"priority": 1000,
"payload_type": 111,
"rtx_payload_type": null
},
{
"name": "H264",
"type": "video",
"priority": 1000,
"payload_type": 102,
"rtx_payload_type": 121
},
{
"name": "VP8",
"type": "video",
"priority": 2000,
"payload_type": 96,
"rtx_payload_type": 97
},
{
"name": "VP9",
"type": "video",
"priority": 3000,
"payload_type": 98,
"rtx_payload_type": 99
}
],
"rtc_connection_id": "3faa0b80-b3e2-4bae-b291-273801fbb7ab"
}
}
*/
const transport = endpoint.createTransport(this.client.sdp!);
this.client.transport = transport;
transport.setRemoteProperties(this.client.sdp!);
transport.setLocalProperties(this.client.sdp!);
export async function onSelectProtocol(this: Server, socket: WebSocket, data: Payload) {
if (data.d.sdp) {
const rtpCapabilities = this.mediasoupRouters[0].rtpCapabilities;
const codecs = rtpCapabilities.codecs as RtpCodecCapability[];
const dtls = transport.getLocalDTLSInfo();
const ice = transport.getLocalICEInfo();
const port = endpoint.getLocalPort();
const fingerprint = dtls.getHash() + " " + dtls.getFingerprint();
const candidates = transport.getLocalCandidates();
const candidate = candidates[0];
const transport = this.mediasoupTransports[0]; //whatever
const answer = `m=audio ${port} ICE/SDP
a=fingerprint:${fingerprint}
c=IN IP4 ${PublicIP}
a=rtcp:${port}
a=ice-ufrag:${ice.getUfrag()}
a=ice-pwd:${ice.getPwd()}
a=fingerprint:${fingerprint}
a=candidate:1 1 ${candidate.getTransport()} ${candidate.getFoundation()} ${candidate.getAddress()} ${candidate.getPort()} typ host
`;
const res = sdpTransform.parse(data.d.sdp);
const videoCodec = this.mediasoupRouters[0].rtpCapabilities.codecs!.find((x: any) => x.kind === "video");
const audioCodec = this.mediasoupRouters[0].rtpCapabilities.codecs!.find((x: any) => x.kind === "audio");
const producer = this.mediasoupProducers[0] || await transport.produce({
kind: "audio",
rtpParameters: {
mid: "audio",
codecs: [{
clockRate: audioCodec!.clockRate,
payloadType: audioCodec!.preferredPayloadType as number,
mimeType: audioCodec!.mimeType,
channels: audioCodec?.channels,
}],
headerExtensions: res.ext?.map(x => ({
id: x.value,
uri: x.uri,
})),
},
paused: false,
});
console.log("can consume: " + this.mediasoupRouters[0].canConsume({ producerId: producer.id, rtpCapabilities: rtpCapabilities }));
const consumer = this.mediasoupConsumers[0] || await transport.consume({
producerId: producer.id,
paused: false,
rtpCapabilities,
});
transport.connect({
dtlsParameters: {
fingerprints: transport.dtlsParameters.fingerprints,
role: "server",
}
});
socket.send(JSON.stringify({
op: VoiceOPCodes.SESSION_DESCRIPTION,
d: {
video_codec: videoCodec?.mimeType?.substring(6) || undefined,
// mode: "xsalsa20_poly1305_lite",
media_session_id: transport.id,
audio_codec: audioCodec?.mimeType.substring(6),
secret_key: sodium.from_hex("724b092810ec86d7e35c9d067702b31ef90bc43a7b598626749914d6a3e033ed").buffer,
sdp: `m=audio ${50001} ICE/SDP\n`
+ `a=fingerprint:sha-256 ${transport.dtlsParameters.fingerprints.find(x => x.algorithm === "sha-256")?.value}\n`
+ `c=IN IP4 ${transport.iceCandidates[0].ip}\n`
+ `t=0 0\n`
+ `a=ice-lite\n`
+ `a=rtcp-mux\n`
+ `a=rtcp:${50001}\n`
+ `a=ice-ufrag:${transport.iceParameters.usernameFragment}\n`
+ `a=ice-pwd:${transport.iceParameters.password}\n`
+ `a=fingerprint:sha-256 ${transport.dtlsParameters.fingerprints.find(x => x.algorithm === "sha-256")?.value}\n`
+ `a=candidate:1 1 ${transport.iceCandidates[0].protocol.toUpperCase()} ${transport.iceCandidates[0].priority} ${transport.iceCandidates[0].ip} ${50001} typ ${transport.iceCandidates[0].type}`
}
}));
return;
}
else {
/*
{
"video_codec":"H264",
"sdp":
"
m=audio 50010 ICE/SDP
a=fingerprint:sha-256 4A:79:94:16:44:3F:BD:05:41:5A:C7:20:F3:12:54:70:00:73:5D:33:00:2D:2C:80:9B:39:E1:9F:2D:A7:49:87
c=IN IP4 109.200.214.158
a=rtcp:50010
a=ice-ufrag:+npq
a=ice-pwd:+jf7jAesMeHHby43FRqWTy
a=fingerprint:sha-256 4A:79:94:16:44:3F:BD:05:41:5A:C7:20:F3:12:54:70:00:73:5D:33:00:2D:2C:80:9B:39:E1:9F:2D:A7:49:87
a=candidate:1 1 UDP 4261412862 109.200.214.158 50010 typ host",
"media_session_id":"59265c94fa13e313492c372c4c8da801
",
"audio_codec":"opus"
}
*/
/*
{
"video_codec": "H264",
"secret_key": [36, 80, 96, 53, 95, 149, 253, 16, 137, 186, 238, 222, 251, 180, 94, 150, 112, 137, 192, 109, 69, 79, 218, 111, 217, 197, 56, 74, 18, 41, 51, 140],
"mode": "aead_aes256_gcm_rtpsize",
"media_session_id": "797575a97a87b63e81e2399348b97ad1",
"audio_codec": "opus"
};
*/
this.decryptKey = sodium.randombytes_buf(sodium.crypto_secretbox_KEYBYTES);
// this.decryptKey = new Array(sodium.crypto_secretbox_KEYBYTES).fill(null).map((x, i) => i + 1);
console.log(this.decryptKey);
socket.send(JSON.stringify({
op: VoiceOPCodes.SESSION_DESCRIPTION,
d: {
video_codec: "H264",
secret_key: [...this.decryptKey.values()],
mode: "aead_aes256_gcm_rtpsize",
media_session_id: "blah blah blah",
audio_codec: "opus",
}
}));
}
await Send(this, {
op: VoiceOPCodes.SELECT_PROTOCOL_ACK,
d: {
video_codec: "H264",
sdp: answer,
media_session_id: this.session_id,
audio_codec: "opus"
}
});
}

View File

@ -1,7 +1,22 @@
import { WebSocket } from "@fosscord/gateway";
import { Payload } from "./index"
import { VoiceOPCodes } from "@fosscord/util";
import { Server } from "../Server"
import { Payload, Send, WebSocket } from "@fosscord/gateway";
import { getClients, VoiceOPCodes } from "../util";
export async function onSpeaking(this: Server, socket: WebSocket, data: Payload) {
// {"speaking":1,"delay":5,"ssrc":2805246727}
export async function onSpeaking(this: WebSocket, data: Payload) {
if (!this.client) return;
getClients(this.client.channel_id).forEach((client) => {
if (client === this.client) return;
const ssrc = this.client!.out.tracks.get(client.websocket.user_id);
Send(client.websocket, {
op: VoiceOPCodes.SPEAKING,
d: {
user_id: client.websocket.user_id,
speaking: data.d.speaking,
ssrc: ssrc?.audio_ssrc || 0
}
});
});
}

View File

@ -1,14 +0,0 @@
import { WebSocket } from "@fosscord/gateway";
import { Payload } from "./index";
import { setHeartbeat } from "../util";
import { Server } from "../Server"
export async function onVersion(this: Server, socket: WebSocket, data: Payload) {
socket.send(JSON.stringify({
op: 16,
d: {
voice: "0.8.31", //version numbers?
rtc_worker: "0.3.18",
}
}))
}

118
webrtc/src/opcodes/Video.ts Normal file
View File

@ -0,0 +1,118 @@
import { Payload, Send, WebSocket } from "@fosscord/gateway";
import { validateSchema, VoiceVideoSchema } from "@fosscord/util";
import { channels, getClients, VoiceOPCodes } from "@fosscord/webrtc";
import { IncomingStreamTrack, SSRCs } from "medooze-media-server";
import SemanticSDP from "semantic-sdp";
export async function onVideo(this: WebSocket, payload: Payload) {
if (!this.client) return;
const { transport, channel_id } = this.client;
if (!transport) return;
const d = validateSchema("VoiceVideoSchema", payload.d) as VoiceVideoSchema;
await Send(this, { op: VoiceOPCodes.MEDIA_SINK_WANTS, d: { any: 100 } });
const id = "stream" + this.user_id;
var stream = this.client.in.stream!;
if (!stream) {
stream = this.client.transport!.createIncomingStream(
// @ts-ignore
SemanticSDP.StreamInfo.expand({
id,
// @ts-ignore
tracks: []
})
);
this.client.in.stream = stream;
const interval = setInterval(() => {
for (const track of stream.getTracks()) {
for (const layer of Object.values(track.getStats())) {
console.log(track.getId(), layer.total);
}
}
}, 5000);
stream.on("stopped", () => {
console.log("stream stopped");
clearInterval(interval);
});
this.on("close", () => {
transport!.stop();
});
const out = transport.createOutgoingStream(
// @ts-ignore
SemanticSDP.StreamInfo.expand({
id: "out" + this.user_id,
// @ts-ignore
tracks: []
})
);
this.client.out.stream = out;
const clients = channels.get(channel_id)!;
clients.forEach((client) => {
if (client.websocket.user_id === this.user_id) return;
if (!client.in.stream) return;
client.in.stream?.getTracks().forEach((track) => {
attachTrack.call(this, track, client.websocket.user_id);
});
});
}
if (d.audio_ssrc) {
handleSSRC.call(this, "audio", { media: d.audio_ssrc, rtx: d.audio_ssrc + 1 });
}
if (d.video_ssrc && d.rtx_ssrc) {
handleSSRC.call(this, "video", { media: d.video_ssrc, rtx: d.rtx_ssrc });
}
}
function attachTrack(this: WebSocket, track: IncomingStreamTrack, user_id: string) {
if (!this.client) return;
const outTrack = this.client.transport!.createOutgoingStreamTrack(track.getMedia());
outTrack.attachTo(track);
this.client.out.stream!.addTrack(outTrack);
var ssrcs = this.client.out.tracks.get(user_id)!;
if (!ssrcs) ssrcs = this.client.out.tracks.set(user_id, { audio_ssrc: 0, rtx_ssrc: 0, video_ssrc: 0 }).get(user_id)!;
if (track.getMedia() === "audio") {
ssrcs.audio_ssrc = outTrack.getSSRCs().media!;
} else if (track.getMedia() === "video") {
ssrcs.video_ssrc = outTrack.getSSRCs().media!;
ssrcs.rtx_ssrc = outTrack.getSSRCs().rtx!;
}
Send(this, {
op: VoiceOPCodes.VIDEO,
d: {
user_id: user_id,
...ssrcs
} as VoiceVideoSchema
});
}
function handleSSRC(this: WebSocket, type: "audio" | "video", ssrcs: SSRCs) {
if (!this.client) return;
const stream = this.client.in.stream!;
const transport = this.client.transport!;
const id = type + ssrcs.media;
var track = stream.getTrack(id);
if (!track) {
console.log("createIncomingStreamTrack", id);
track = transport.createIncomingStreamTrack(type, { id, ssrcs });
stream.addTrack(track);
const clients = getClients(this.client.channel_id)!;
clients.forEach((client) => {
if (client.websocket.user_id === this.user_id) return;
if (!client.out.stream) return;
attachTrack.call(this, track, client.websocket.user_id);
});
}
}

View File

@ -1,43 +1,19 @@
import { WebSocket } from "@fosscord/gateway";
import { VoiceOPCodes } from "@fosscord/util";
import { Server } from "../Server";
export interface Payload {
op: number;
d: any;
s: number;
t: string;
}
import { Payload, WebSocket } from "@fosscord/gateway";
import { VoiceOPCodes } from "../util";
import { onBackendVersion } from "./BackendVersion";
import { onHeartbeat } from "./Heartbeat";
import { onIdentify } from "./Identify";
import { onSelectProtocol } from "./SelectProtocol";
import { onHeartbeat } from "./Heartbeat";
import { onSpeaking } from "./Speaking";
import { onResume } from "./Resume";
import { onConnect } from "./Connect";
import { onVideo } from "./Video";
import { onVersion } from "./Version";
export type OPCodeHandler = (this: WebSocket, data: Payload) => any;
export type OPCodeHandler = (this: Server, socket: WebSocket, data: Payload) => any;
const handlers: { [key: number]: OPCodeHandler } = {
[VoiceOPCodes.IDENTIFY]: onIdentify, //op 0
[VoiceOPCodes.SELECT_PROTOCOL]: onSelectProtocol, //op 1
//op 2 voice_ready
[VoiceOPCodes.HEARTBEAT]: onHeartbeat, //op 3
//op 4 session_description
[VoiceOPCodes.SPEAKING]: onSpeaking, //op 5
//op 6 heartbeat_ack
[VoiceOPCodes.RESUME]: onResume, //op 7
//op 8 hello
//op 9 resumed
//op 10?
//op 11?
[VoiceOPCodes.CLIENT_CONNECT]: onConnect, //op 12
//op 13?
//op 15?
//op 16? empty data on client send but server sends {"voice":"0.8.24+bugfix.voice.streams.opt.branch-ffcefaff7","rtc_worker":"0.3.14-crypto-collision-copy"}
[VoiceOPCodes.VERSION]: onVersion,
};
export default handlers;
export default {
[VoiceOPCodes.HEARTBEAT]: onHeartbeat,
[VoiceOPCodes.IDENTIFY]: onIdentify,
[VoiceOPCodes.VOICE_BACKEND_VERSION]: onBackendVersion,
[VoiceOPCodes.VIDEO]: onVideo,
[VoiceOPCodes.SPEAKING]: onSpeaking,
[VoiceOPCodes.SELECT_PROTOCOL]: onSelectProtocol
};

View File

@ -1,11 +1,13 @@
process.on("uncaughtException", console.error);
process.on("unhandledRejection", console.error);
import { config } from "dotenv";
import { Server } from "./Server";
config();
//testing
process.env.DATABASE = "../bundle/database.db";
process.env.DEBUG = "mediasoup*"
const port = Number(process.env.PORT) || 3004;
import { Server } from "./Server";
const server = new Server();
server.listen();
const server = new Server({
port
});
server.start();

View File

@ -1,8 +0,0 @@
import { getSupportedRtpCapabilities } from "mediasoup";
async function test() {
console.log(getSupportedRtpCapabilities());
}
setTimeout(() => {}, 1000000);
test();

View File

@ -0,0 +1,26 @@
export enum VoiceStatus {
CONNECTED = 0,
CONNECTING = 1,
AUTHENTICATING = 2,
RECONNECTING = 3,
DISCONNECTED = 4
}
export enum VoiceOPCodes {
IDENTIFY = 0,
SELECT_PROTOCOL = 1,
READY = 2,
HEARTBEAT = 3,
SELECT_PROTOCOL_ACK = 4,
SPEAKING = 5,
HEARTBEAT_ACK = 6,
RESUME = 7,
HELLO = 8,
RESUMED = 9,
VIDEO = 12,
CLIENT_DISCONNECT = 13,
SESSION_UPDATE = 14,
MEDIA_SINK_WANTS = 15,
VOICE_BACKEND_VERSION = 16,
CHANNEL_OPTIONS_UPDATE = 17
}

View File

@ -1,23 +0,0 @@
import { WebSocket, CLOSECODES } from "@fosscord/gateway";
import { VoiceOPCodes } from "@fosscord/util";
export async function setHeartbeat(socket: WebSocket, nonce?: Number) {
if (socket.heartbeatTimeout) clearTimeout(socket.heartbeatTimeout);
socket.heartbeatTimeout = setTimeout(() => {
return socket.close(CLOSECODES.Session_timed_out);
}, 1000 * 45);
if (!nonce) {
socket.send(JSON.stringify({
op: VoiceOPCodes.HELLO,
d: {
v: 5,
heartbeat_interval: 13750,
}
}));
}
else {
socket.send(JSON.stringify({ op: VoiceOPCodes.HEARTBEAT_ACK, d: nonce }));
}
}

View File

@ -0,0 +1,51 @@
import { WebSocket } from "@fosscord/gateway";
import MediaServer, { IncomingStream, OutgoingStream, Transport } from "medooze-media-server";
import SemanticSDP from "semantic-sdp";
MediaServer.enableLog(true);
export const PublicIP = process.env.PUBLIC_IP || "127.0.0.1";
try {
const range = process.env.WEBRTC_PORT_RANGE || "4000";
var ports = range.split("-");
const min = Number(ports[0]);
const max = Number(ports[1]);
MediaServer.setPortRange(min, max);
} catch (error) {
console.error("Invalid env var: WEBRTC_PORT_RANGE", process.env.WEBRTC_PORT_RANGE, error);
process.exit(1);
}
export const endpoint = MediaServer.createEndpoint(PublicIP);
export const channels = new Map<string, Set<Client>>();
export interface Client {
transport?: Transport;
websocket: WebSocket;
out: {
stream?: OutgoingStream;
tracks: Map<
string,
{
audio_ssrc: number;
video_ssrc: number;
rtx_ssrc: number;
}
>;
};
in: {
stream?: IncomingStream;
audio_ssrc: number;
video_ssrc: number;
rtx_ssrc: number;
};
sdp: SemanticSDP.SDPInfo;
channel_id: string;
}
export function getClients(channel_id: string) {
if (!channels.has(channel_id)) channels.set(channel_id, new Set());
return channels.get(channel_id)!;
}

View File

@ -1 +1,2 @@
export * from "./Heartbeat"
export * from "./Constants";
export * from "./MediaServer";

View File

@ -72,12 +72,10 @@
"skipLibCheck": true /* Skip type checking of declaration files. */,
"forceConsistentCasingInFileNames": true, /* Disallow inconsistently-cased references to the same file. */
"baseUrl": "../",
"paths": {
"@fosscord/api": ["api/src/index"],
"@fosscord/gateway": ["gateway/src/index"],
"@fosscord/cdn": ["cdn/src/index"],
"@fosscord/util": ["util/src/index"]
"@fosscord/util": ["../util"],
"@fosscord/gateway": ["../gateway"],
"@fosscord/webrtc": ["."]
},
}
}