TeaWeb/web/app/connection/ServerConnection.ts

515 lines
20 KiB
TypeScript

import {
AbstractServerConnection,
CommandOptionDefaults,
CommandOptions, ConnectionPing,
ConnectionStatistics, ServerCommand,
} from "tc-shared/connection/ConnectionBase";
import {ConnectionHandler, ConnectionState, DisconnectReason} from "tc-shared/ConnectionHandler";
import {HandshakeHandler} from "tc-shared/connection/HandshakeHandler";
import {ConnectionCommandHandler, ServerConnectionCommandBoss} from "tc-shared/connection/CommandHandler";
import {CommandResult} from "tc-shared/connection/ServerConnectionDeclaration";
import {settings, Settings} from "tc-shared/settings";
import * as log from "tc-shared/log";
import {LogCategory, logDebug, logError, logInfo, logTrace, logWarn} from "tc-shared/log";
import {AbstractCommandHandlerBoss} from "tc-shared/connection/AbstractCommandHandler";
import {WrappedWebSocket} from "./WrappedWebSocket";
import {AbstractVoiceConnection} from "tc-shared/connection/VoiceConnection";
import {parseCommand} from "./CommandParser";
import {ServerAddress} from "tc-shared/tree/Server";
import {RtpVoiceConnection} from "../voice/Connection";
import {VideoConnection} from "tc-shared/connection/VideoConnection";
import {ServerFeature} from "tc-shared/connection/ServerFeatures";
import {RTCConnection} from "tc-shared/connection/rtc/Connection";
import {RtpVideoConnection} from "tc-shared/connection/rtc/video/Connection";
import { tr } from "tc-shared/i18n/localize";
import {createErrorModal} from "tc-shared/ui/elements/Modal";
import ipRegex from "ip-regex";
class ReturnListener<T> {
resolve: (value?: T | PromiseLike<T>) => void;
reject: (reason?: any) => void;
code: string;
timeout: number;
}
let globalReturnCodeIndex = 0;
export class ServerConnection extends AbstractServerConnection {
private remoteServerAddress: ServerAddress;
private handshakeHandler: HandshakeHandler;
private commandHandlerBoss: ServerConnectionCommandBoss;
private defaultCommandHandler: ConnectionCommandHandler;
private socket: WrappedWebSocket;
private connectCancelCallback: () => void;
private returnListeners: ReturnListener<CommandResult>[] = [];
private rtcConnection: RTCConnection;
private voiceConnection: RtpVoiceConnection;
private videoConnection: RtpVideoConnection;
private pingStatistics = {
thread_id: 0,
lastRequestTimestamp: 0,
lastResponseTimestamp: 0,
currentRequestId: 0,
interval: 5000,
timeout: 7500,
currentJsValue: 0,
currentNativeValue: 0 /* ping value for native (WS) */
};
constructor(client : ConnectionHandler) {
super(client);
this.commandHandlerBoss = new ServerConnectionCommandBoss(this);
this.defaultCommandHandler = new ConnectionCommandHandler(this);
this.commandHandlerBoss.registerHandler(this.defaultCommandHandler);
this.command_helper.initialize();
this.rtcConnection = new RTCConnection(this, true);
this.voiceConnection = new RtpVoiceConnection(this, this.rtcConnection);
this.videoConnection = new RtpVideoConnection(this.rtcConnection);
}
destroy() {
this.disconnect("handle destroyed").catch(error => {
logWarn(LogCategory.NETWORKING, tr("Failed to disconnect on server connection destroy: %o"), error);
}).then(() => {
clearInterval(this.pingStatistics.thread_id);
if(this.connectCancelCallback) {
this.connectCancelCallback();
}
for(const listener of this.returnListeners) {
try {
listener.reject("handler destroyed");
} catch(error) {
logWarn(LogCategory.NETWORKING, tr("Failed to reject command promise: %o"), error);
}
clearTimeout(listener.timeout);
}
this.returnListeners = undefined;
this.rtcConnection.destroy();
this.command_helper.destroy();
this.defaultCommandHandler && this.commandHandlerBoss.unregisterHandler(this.defaultCommandHandler);
this.defaultCommandHandler = undefined;
this.voiceConnection && this.voiceConnection.destroy();
this.voiceConnection = undefined;
this.commandHandlerBoss && this.commandHandlerBoss.destroy();
this.commandHandlerBoss = undefined;
this.events.destroy();
});
}
async connect(address : ServerAddress, handshake: HandshakeHandler, timeout?: number) : Promise<void> {
const connectBeginTimestamp = Date.now();
timeout = typeof(timeout) === "number" ? timeout : 10_000;
try {
await this.disconnect();
} catch(error) {
logError(LogCategory.NETWORKING, tr("Failed to close old connection properly. Error: %o"), error);
throw "failed to cleanup old connection";
}
this.updateConnectionState(ConnectionState.CONNECTING);
this.remoteServerAddress = address;
this.handshakeHandler = handshake;
this.handshakeHandler.setConnection(this);
/* The direct one connect directly to the target address. The other via the .con-gate.work */
let availableSockets: WrappedWebSocket[] = [];
proxySocket:
if(!settings.getValue(Settings.KEY_CONNECT_NO_DNSPROXY)) {
let host;
if(ipRegex({ exact: true }).test(address.host)) {
host = address.host;
host = host.replace(/\./g, "-");
host = host.replace(/:/g, "_");
host = host + ".con-gate.work";
} else {
break proxySocket;
}
availableSockets.push(new WrappedWebSocket({
host: host,
port: address.port,
secure: true
}))
}
availableSockets.push(new WrappedWebSocket({
host: address.host,
port: address.port,
secure: true
}));
let timeoutRaised = false;
let timeoutPromise = new Promise<void>(resolve => setTimeout(() => {
timeoutRaised = true;
resolve();
}, timeout));
let cancelRaised = false;
let cancelPromise = new Promise<void>(resolve => {
this.connectCancelCallback = () => {
this.connectCancelCallback = undefined;
cancelRaised = true;
resolve();
};
});
availableSockets.forEach(e => e.doConnect());
while (availableSockets.length > 0) {
await Promise.race([...availableSockets.map(e => e.awaitConnectResult()), timeoutPromise, cancelPromise]);
if(cancelRaised) {
logDebug(LogCategory.NETWORKING, tr("Aborting connect attempt due to a cancel request."));
availableSockets.forEach(e => e.closeConnection());
return
}
if(timeoutRaised) {
logInfo(LogCategory.NETWORKING, tr("Connect timeout triggered. Aborting connect attempt!"));
availableSockets.forEach(e => e.closeConnection());
this.updateConnectionState(ConnectionState.UNCONNECTED); /* firstly update the state, that fire event */
this.client.handleDisconnect(DisconnectReason.CONNECT_FAILURE);
return
}
let finished = availableSockets.find(e => e.state !== "connecting");
if(!finished) continue; /* should not happen, but we want to ensure it */
availableSockets.remove(finished);
switch (finished.state) {
case "unconnected":
logDebug(LogCategory.NETWORKING, tr("Connection attempt to %s:%d via %s got aborted."), this.remoteServerAddress.host, this.remoteServerAddress.port, finished.socketUrl());
continue;
case "errored":
const error = finished.popError();
logInfo(LogCategory.NETWORKING, tr("Connection attempt to %s:%d via %s failed:\n%o"), this.remoteServerAddress.host, this.remoteServerAddress.port, finished.socketUrl(), error);
continue;
case "connected":
break;
}
this.socket = finished;
/* abort any other ongoing connection attempts, we already succeeded */
availableSockets.forEach(e => e.closeConnection());
break;
}
if(!this.socket) {
logInfo(LogCategory.NETWORKING, tr("Failed to connect to %s:%d. No connection attempt succeeded."), this.remoteServerAddress.host, this.remoteServerAddress.port);
this.updateConnectionState(ConnectionState.UNCONNECTED); /* firstly update the state, that fire event */
this.client.handleDisconnect(DisconnectReason.CONNECT_FAILURE);
return;
}
this.socket.callbackMessage = message => this.handleSocketMessage(message);
this.socket.callbackDisconnect = (code, reason) => {
try {
this.disconnect();
} catch (error) {
logWarn(LogCategory.NETWORKING, tr("Failed to disconnect with an already closed socket: %o"), error);
}
this.client.handleDisconnect(DisconnectReason.CONNECTION_CLOSED, {
code: code,
reason: reason
});
};
this.socket.callbackErrored = () => {
if(this.socket.hasError()) {
logError(LogCategory.NETWORKING, tr("Server connection %s:%d has been terminated due to an unexpected error (%o)."),
this.remoteServerAddress.host,
this.remoteServerAddress.port,
this.socket.popError()
);
} else {
logError(LogCategory.NETWORKING, tr("Server connection %s:%d has been terminated due to an unexpected error."), this.remoteServerAddress.host, this.remoteServerAddress.port);
}
try {
this.disconnect();
} catch (error) {
logWarn(LogCategory.NETWORKING, tr("Failed to disconnect with an already closed socket: %o"), error);
}
this.client.handleDisconnect(DisconnectReason.CONNECTION_CLOSED);
};
const connectEndTimestamp = Date.now();
logInfo(LogCategory.NETWORKING, tr("Successfully initialized a connection to %s:%d via %s within %d milliseconds."),
this.remoteServerAddress.host,
this.remoteServerAddress.port,
this.socket.socketUrl(),
connectEndTimestamp - connectBeginTimestamp);
/* enabling raw commands, if the server supports it */
this.sendData(JSON.stringify({
type: "enable-raw-commands"
}))
this.startHandshake();
}
private startHandshake() {
this.updateConnectionState(ConnectionState.INITIALISING);
this.client.log.log("connection.login", {});
this.handshakeHandler.initialize();
this.handshakeHandler.startHandshake();
}
async disconnect(reason?: string) : Promise<void> {
if(this.connectCancelCallback)
this.connectCancelCallback();
if(this.connectionState === ConnectionState.UNCONNECTED)
return;
this.updateConnectionState(ConnectionState.DISCONNECTING);
try {
clearTimeout(this.pingStatistics.thread_id);
this.pingStatistics.thread_id = undefined;
if(typeof(reason) === "string") {
//TODO send disconnect reason
}
if(this.socket) {
this.socket.callbackMessage = undefined;
this.socket.callbackDisconnect = undefined;
this.socket.callbackErrored = undefined;
this.socket.closeConnection(); /* 3000 + 0xFF, tr("request disconnect") */
this.socket = undefined;
}
for(let future of this.returnListeners)
future.reject(tr("Connection closed"));
this.returnListeners = [];
} finally {
this.updateConnectionState(ConnectionState.UNCONNECTED);
}
}
private handleSocketMessage(data) {
if(typeof(data) === "string") {
let json;
try {
json = JSON.parse(data);
} catch(e) {
logWarn(LogCategory.NETWORKING, tr("Could not parse message json!"));
return;
}
if(json["type"] === undefined) {
logWarn(LogCategory.NETWORKING, tr("Missing data type in message!"));
return;
}
if(json["type"] === "command") {
/* devel-block(log-networking-commands) */
let group = log.group(log.LogType.DEBUG, LogCategory.NETWORKING, tr("Handling command '%s'"), json["command"]);
group.log(tr("Handling command '%s'"), json["command"]);
group.group(log.LogType.TRACE, tr("Json:")).collapsed(true).log("%o", json).end();
/* devel-block-end */
this.commandHandlerBoss.invokeCommand(new ServerCommand(json["command"], json["data"], []));
if(json["command"] === "initserver") {
this.handleServerInit();
}
/* devel-block(log-networking-commands) */
group.end();
/* devel-block-end */
} else if(json["type"] === "command-raw") {
const command = parseCommand(json["payload"]);
logTrace(LogCategory.NETWORKING, tr("Received command %s"), command.command);
this.commandHandlerBoss.invokeCommand(command);
if(command.command === "initserver") {
this.handleServerInit();
}
} else if(json["type"] === "ping") {
this.sendData(JSON.stringify({
type: 'pong',
payload: json["payload"]
}));
} else if(json["type"] === "pong") {
const id = parseInt(json["payload"]);
if(id != this.pingStatistics.currentRequestId) {
logWarn(LogCategory.NETWORKING, tr("Received pong which is older than the last request. Delay may over %oms? (Index: %o, Current index: %o)"), this.pingStatistics.timeout, id, this.pingStatistics.currentRequestId);
} else {
this.pingStatistics.lastResponseTimestamp = 'now' in performance ? performance.now() : Date.now();
this.pingStatistics.currentJsValue = this.pingStatistics.lastResponseTimestamp - this.pingStatistics.lastRequestTimestamp;
this.pingStatistics.currentNativeValue = parseInt(json["ping_native"]) / 1000; /* we're getting it in microseconds and not milliseconds */
this.events.fire("notify_ping_updated", { newPing: this.ping() });
//logDebug(LogCategory.NETWORKING, tr("Received new pong. Updating ping to: JS: %o Native: %o"), this._ping.value.toFixed(3), this._ping.value_native.toFixed(3));
}
} else {
logWarn(LogCategory.NETWORKING, tr("Unknown command type %o"), json["type"]);
}
} else {
logWarn(LogCategory.NETWORKING, tr("Received unknown message of type %s. Dropping message"), typeof(data));
}
}
sendData(data: any) {
if(!this.socket || this.socket.state !== "connected") {
logWarn(LogCategory.NETWORKING, tr("Tried to send data via a non connected server socket."));
return;
}
this.socket.sendMessage(data);
}
private handleServerInit() {
this.pingStatistics.thread_id = setInterval(() => this.doNextPing(), this.pingStatistics.interval) as any;
this.doNextPing();
this.updateConnectionState(ConnectionState.CONNECTED);
this.client.serverFeatures.awaitFeatures().then(succeeded => {
if(!succeeded) {
/* something like a disconnect happened or so */
return;
}
if(this.client.serverFeatures.supportsFeature(ServerFeature.VIDEO, 1)) {
this.rtcConnection.doInitialSetup();
} else{
/* old voice connection */
logDebug(LogCategory.NETWORKING, tr("Using legacy voice connection for TeaSpeak server bellow 1.5"));
createErrorModal(tr("Server outdated"), tr("Please update your server in order to use the WebClient")).open();
this.rtcConnection.setNotSupported();
}
});
}
private static commandDataToJson(input: any) : string {
return JSON.stringify(input, (key, value) => {
switch (typeof value) {
case "boolean": return value == true ? "1" : "0";
case "function": return value();
default:
return value;
}
});
}
send_command(command: string, data?: any | any[], _options?: CommandOptions) : Promise<CommandResult> {
if(!this.socket || !this.connected()) {
logWarn(LogCategory.NETWORKING, tr("Tried to send a command without a valid connection."));
return Promise.reject(tr("not connected"));
}
const options: CommandOptions = {};
Object.assign(options, CommandOptionDefaults);
Object.assign(options, _options);
data = Array.isArray(data) ? data : [data || {}];
if(data.length == 0) /* we require min one arg to append return_code */
data.push({});
let result = new Promise<CommandResult>((resolve, failed) => {
let payload = Array.isArray(data) ? data : [data];
let returnCode = typeof payload[0]["return_code"] === "string" ? payload[0].return_code : ++globalReturnCodeIndex;
payload[0].return_code = returnCode;
let listener = new ReturnListener<CommandResult>();
listener.resolve = resolve;
listener.reject = failed;
listener.code = returnCode;
listener.timeout = setTimeout(() => {
this.returnListeners.remove(listener);
listener.reject("timeout");
}, options.timeout || 15_000);
this.returnListeners.push(listener);
this.sendData(ServerConnection.commandDataToJson({
"type": "command",
"command": command,
"data": payload,
"flags": options.flagset.filter(entry => entry.length != 0)
}))
});
return this.defaultCommandHandler.proxy_command_promise(result, options);
}
connected() : boolean {
return !!this.socket && this.socket.state === "connected";
}
getVoiceConnection(): AbstractVoiceConnection {
return this.voiceConnection;
}
getVideoConnection(): VideoConnection {
return this.videoConnection;
}
getCommandHandler(): AbstractCommandHandlerBoss {
return this.commandHandlerBoss;
}
handshake_handler(): HandshakeHandler {
return this.handshakeHandler;
}
remote_address(): ServerAddress {
return this.remoteServerAddress;
}
connectionProxyAddress(): ServerAddress | undefined {
return this.socket?.address;
}
private doNextPing() {
if(this.pingStatistics.lastRequestTimestamp + this.pingStatistics.timeout < Date.now()) {
this.pingStatistics.currentJsValue = this.pingStatistics.timeout;
this.pingStatistics.lastResponseTimestamp = this.pingStatistics.lastRequestTimestamp + 1;
}
if(this.pingStatistics.lastResponseTimestamp > this.pingStatistics.lastRequestTimestamp) {
this.pingStatistics.lastRequestTimestamp = 'now' in performance ? performance.now() : Date.now();
this.sendData(JSON.stringify({
type: 'ping',
payload: (++this.pingStatistics.currentRequestId).toString()
}));
}
}
ping(): ConnectionPing {
return {
javascript: this.pingStatistics.currentJsValue,
/* if the native value is zero that means we don't have any */
native: this.pingStatistics.currentNativeValue === 0 ? this.pingStatistics.currentJsValue : this.pingStatistics.currentNativeValue
};
}
getControlStatistics(): ConnectionStatistics {
return this.socket?.getControlStatistics() || { bytesSend: 0, bytesReceived: 0 };
}
getServerType(): "teaspeak" | "teamspeak" | "unknown" {
/* It's simple. Only TeaSpeak support web clients */
return "teaspeak";
}
}