mirror/rustdesk
1
0
Fork 0
mirror of https://github.com/rustdesk/rustdesk.git synced 2024-12-04 11:59:18 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

websock init

Browse Source
This commit is contained in:
rustdesk 2022-01-18 02:02:39 +08:00
parent fa31257cdc
commit d4e8a66cf6
1 changed files with 74 additions and 111 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

185
src/websock.ts
View File

@ -1,116 +1,79 @@
import * as proto from '../message.js';
import * as message from "./message.js";
import * as rendezvous from "./rendezvous.js";
type Keys = "message" | "open" | "close" | "error";
export default class Websock {
constructor() {
this._websocket = null; // WebSocket object
this._eventHandlers = {
message: (msg) => { },
open: () => { },
close: () => { },
error: () => { }
};
this._next_yuv = null;
this._next_rgb = null;
_websocket: WebSocket;
_eventHandlers: { [key in Keys]: Function };
send_message(msg: message.Message) {
this._websocket.send(message.Message.encode(msg).finish());
}
send_rendezvous(msg: rendezvous.RendezvousMessage) {
this._websocket.send(rendezvous.RendezvousMessage.encode(msg).finish());
}
// Event Handlers
off(evt: Keys) {
this._eventHandlers[evt] = () => {};
}
on(evt: Keys, handler: Function) {
this._eventHandlers[evt] = handler;
}
constructor(uri: string, protocols: string) {
this._eventHandlers = {
message: (_: string) => {},
open: () => {},
close: () => {},
error: () => {},
};
this._websocket = new WebSocket(uri, protocols);
this._websocket.onmessage = this._recv_message.bind(this);
this._websocket.binaryType = "arraybuffer";
this._websocket.onopen = () => {
console.debug(">> WebSock.onopen");
if (this._websocket.protocol) {
console.info("Server choose sub-protocol: " + this._websocket.protocol);
}
this._eventHandlers.open();
console.debug("<< WebSock.onopen");
};
this._websocket.onclose = (e) => {
console.debug(">> WebSock.onclose");
this._eventHandlers.close(e);
console.debug("<< WebSock.onclose");
};
this._websocket.onerror = (e) => {
console.debug(">> WebSock.onerror: " + e);
this._eventHandlers.error(e);
console.debug("<< WebSock.onerror: " + e);
};
}
close() {
if (this._websocket) {
if (
this._websocket.readyState === WebSocket.OPEN ||
this._websocket.readyState === WebSocket.CONNECTING
) {
console.info("Closing WebSocket connection");
this._websocket.close();
}
this._websocket.onmessage = () => {};
}
}
send(msg) {
if (msg instanceof Object) msg = proto.encodeMessage(msg);
this._websocket.send(msg);
_recv_message(e: any) {
if (e.data instanceof window.ArrayBuffer) {
let bytes = new Uint8Array(e.data);
}
// Event Handlers
off(evt) {
this._eventHandlers[evt] = () => { };
}
on(evt, handler) {
this._eventHandlers[evt] = handler;
}
init() {
this._websocket = null;
}
open(uri, protocols) {
this.init();
this._websocket = new WebSocket(uri, protocols);
this._websocket.onmessage = this._recv_message.bind(this);
this._websocket.binaryType = 'arraybuffer';
this._websocket.onopen = () => {
console.debug('>> WebSock.onopen');
if (this._websocket.protocol) {
console.info("Server choose sub-protocol: " + this._websocket.protocol);
}
this._eventHandlers.open();
console.debug("<< WebSock.onopen");
};
this._websocket.onclose = (e) => {
console.debug(">> WebSock.onclose");
this._eventHandlers.close(e);
console.debug("<< WebSock.onclose");
};
this._websocket.onerror = (e) => {
console.debug(">> WebSock.onerror: " + e);
this._eventHandlers.error(e);
console.debug("<< WebSock.onerror: " + e);
};
}
close() {
if (this._websocket) {
if ((this._websocket.readyState === WebSocket.OPEN) ||
(this._websocket.readyState === WebSocket.CONNECTING)) {
console.info("Closing WebSocket connection");
this._websocket.close();
}
this._websocket.onmessage = () => { };
}
}
_recv_message(e) {
if (e.data instanceof window.ArrayBuffer) {
let bytes = new Uint8Array(e.data);
if (this._next_yuv) {
const yuv = this._next_yuv;
const { compress, stride } = yuv.format;
if (compress) {
bytes = window.simple_zstd.decompress(bytes);
}
if (!yuv.y) {
yuv.y = { bytes, stride: stride };
} else if (!yuv.u) {
yuv.u = { bytes, stride: stride >> 1 };
} else {
yuv.v = { bytes, stride: stride >> 1 };
delete yuv.format.stride;
this._eventHandlers.message({ video_frame: { yuv } });
this._next_yuv = null;
}
} else if (this._next_rgb) {
if (this._next_rgb.format.compress) {
bytes = window.simple_zstd.decompress(bytes);
}
this._eventHandlers.message({ video_frame: { rgb: bytes } });
this._next_rgb = null;
} else {
const msg = proto.decodeMessage(bytes);
let vf = msg.video_frame;
if (vf) {
const { yuv, rgb } = vf;
if (yuv) {
this._next_yuv = { format: yuv };
} else if (rgb) {
this._next_rgb = { format: rgb };
}
return;
} else {
this._eventHandlers.message(msg);
}
}
}
}
}
}
}