113 lines
2.8 KiB
TypeScript
113 lines
2.8 KiB
TypeScript
import Websock from "./websock";
|
|
import * as message from "./message.js";
|
|
import * as rendezvous from "./rendezvous.js";
|
|
import { loadVp9, loadOpus } from "./codec";
|
|
import * as globals from "./globals";
|
|
|
|
const PORT = 21116;
|
|
const HOST = "rs-sg.rustdesk.com";
|
|
const licenceKey = "";
|
|
const SCHEMA = "ws://";
|
|
|
|
export default class Connection {
|
|
_msgs: any[];
|
|
_ws: Websock | undefined;
|
|
_interval: any;
|
|
_id: string;
|
|
|
|
constructor() {
|
|
this._msgs = [];
|
|
this._id = '';
|
|
this._interval = setInterval(() => {
|
|
while (this._msgs.length) {
|
|
this._ws?.sendMessage(this._msgs[0]);
|
|
this._msgs.splice(0, 1);
|
|
}
|
|
}, 1);
|
|
}
|
|
|
|
close() {
|
|
clearInterval(this._interval);
|
|
this._ws?.close();
|
|
}
|
|
|
|
async start(id: string) {
|
|
const ws = new Websock(getDefaultUri());
|
|
this._ws = ws;
|
|
this._id = id;
|
|
await ws.open();
|
|
const connType = rendezvous.ConnType.DEFAULT_CONN;
|
|
const natType = rendezvous.NatType.SYMMETRIC;
|
|
const punchHoleRequest = rendezvous.PunchHoleRequest.fromJSON({
|
|
id,
|
|
licenceKey,
|
|
connType,
|
|
natType,
|
|
});
|
|
ws.sendRendezvous({ punchHoleRequest });
|
|
const msg = ws.parseRendezvous(await ws.next());
|
|
const phr = msg.punchHoleResponse;
|
|
const rr = msg.relayResponse;
|
|
if (phr) {
|
|
if (phr.failure != rendezvous.PunchHoleResponse_Failure.UNKNOWN) {
|
|
switch (phr?.failure) {
|
|
case rendezvous.PunchHoleResponse_Failure.ID_NOT_EXIST:
|
|
break;
|
|
}
|
|
ws.close();
|
|
}
|
|
} else if (rr) {
|
|
await this.connectRelay(rr);
|
|
}
|
|
}
|
|
|
|
async connectRelay(rr: rendezvous.RelayResponse) {
|
|
const pk = rr.pk;
|
|
let uri = rr.relayServer;
|
|
if (uri) {
|
|
uri = getrUriFromRs(uri);
|
|
} else {
|
|
uri = getDefaultUri(true);
|
|
}
|
|
const uuid = rr.uuid;
|
|
const ws = new Websock(uri);
|
|
await ws.open();
|
|
console.log("Connected to relay server");
|
|
this._ws = ws;
|
|
const requestRelay = rendezvous.RequestRelay.fromJSON({
|
|
licenceKey,
|
|
uuid,
|
|
});
|
|
ws.sendRendezvous({ requestRelay });
|
|
await this.secure(pk);
|
|
}
|
|
|
|
async secure(pk: Uint8Array | undefined) {
|
|
if (pk) {
|
|
const RS_PK = 'OeVuKk5nlHiXp+APNn0Y3pC1Iwpwn44JGqrQCsWqmBw=';
|
|
let pk_id = await globals.verify(pk, RS_PK);
|
|
}
|
|
}
|
|
}
|
|
|
|
async function testDelay() {
|
|
const ws = new Websock(getDefaultUri(false));
|
|
await ws.open();
|
|
console.log(ws.latency());
|
|
}
|
|
|
|
function getDefaultUri(isRelay: Boolean = false): string {
|
|
const host = localStorage.getItem("host");
|
|
return SCHEMA + (host || HOST) + ":" + (PORT + (isRelay ? 3 : 2));
|
|
}
|
|
|
|
function getrUriFromRs(uri: string): string {
|
|
if (uri.indexOf(":") > 0) {
|
|
const tmp = uri.split(":");
|
|
const port = parseInt(tmp[1]);
|
|
uri = tmp[0] + ":" + (port + 2);
|
|
} else {
|
|
uri += ":" + (PORT + 3);
|
|
}
|
|
return SCHEMA + uri;
|
|
} |