eval in page context

2026-09-26 13:11:53 +08:00 · 2024-05-12 20:30:19 +08:00
parent f881ad6ce2
commit 7e1e9e2c79
16 changed files with 322 additions and 159 deletions
@@ -0,0 +1,138 @@
+export const ActionType = {
+    CONNECT: 'connect',
+    DISCONNECT: 'disconnect',
+    STATUS: 'status',
+    PROXYSTATUS: 'proxystatus',
+    SETPROXY: 'setproxy',
+    CLEARPROXY: 'clearproxy',
+    EVAL: 'eval',
+    ECHO: 'echo',
+}
+
+export class WebSocketManager {
+    constructor() {
+        this.socket = null;
+        this.intervalId = null;
+    }
+
+    connectWebsocket(url, port) {
+        this.disconnectWebsocket();
+        this.socket = new WebSocket(url);
+
+        this.socket.onopen = () => {
+            chrome.runtime.sendMessage({action: ActionType.STATUS, connected: true, port: port});
+            this.startHeartbeat();
+        };
+
+        this.socket.onmessage = (event) => {
+            this.handleMessage(event.data);
+        };
+
+        this.socket.onclose = () => {
+            chrome.runtime.sendMessage({action: ActionType.STATUS, connected: false});
+        };
+
+        this.socket.onerror = (error) => {
+            console.error("WebSocket Error:", error);
+        };
+    }
+
+    disconnectWebsocket() {
+        if (this.socket) {
+            try {
+                this.socket.close();
+                chrome.runtime.sendMessage({action: ActionType.STATUS, connected: false});
+            } catch (e) {
+                console.error("Error closing websocket:", e);
+            }
+            this.socket = null;
+            this.stopHeartbeat();
+        }
+    }
+
+    startHeartbeat() {
+        this.intervalId = setInterval(() => this.heartbeat(), 3000);
+    }
+
+    stopHeartbeat() {
+        if (this.intervalId) {
+            clearInterval(this.intervalId);
+            this.intervalId = null;
+        }
+    }
+
+    heartbeat() {
+        if (this.isConnected()) {
+            try {
+                this.socket.send(JSON.stringify({"type": "heartbeat"}));
+            } catch (e) {
+                console.error("Error sending heartbeat:", e);
+            }
+        } else {
+            this.disconnectWebsocket();
+        }
+    }
+
+    isConnected() {
+        return this.socket && this.socket.readyState === WebSocket.OPEN;
+    }
+
+    handleMessage(message) {
+        console.log("message", message)
+        // 解析消息
+        // try {
+        //     // 解析从WebSocket接收到的JSON数据
+        //     let code = message;
+        //     getTabId().then((tabId) => {
+        //         // 确保tabId和code有效
+        //         if (typeof tabId === 'number' && typeof code === 'string') {
+        //             chrome.webNavigation.getAllFrames({tabId: tabId}, function(frames) {
+        //                 for (let frame of frames) {
+        //                     console.log(`Frame ID: ${frame.frameId} with URL: ${frame.url}`);
+        //                 }
+        //             });
+        //             // 在指定的tabId上执行脚本
+        //             chrome.scripting.executeScript({
+        //                 target: {tabId: tabId},
+        //                 // function: getTitle,
+        //                 function: log,
+        //                 args: [code]
+        //             }, (injectionResults) => {
+        //                 // 处理脚本执行的结果，如日志记录等
+        //                 console.log('Script executed:', injectionResults);
+        //             });
+        //         } else {
+        //             console.error('Received malformed message:', message);
+        //         }
+        //     });
+        // } catch (err) {
+        //     console.error('Error parsing message from WebSocket:', err);
+        // }
+    }
+}
+
+function getTitle() {
+    return document.title;
+}
+
+function log(message) {
+    return console.log(message);
+}
+
+export const getTabId = () => {
+    return new Promise((resolve, reject) => {
+        chrome.tabs.query({active: true, lastFocusedWindow: true}, (tabs) => {
+            if (tabs.length) {
+                resolve(tabs[0].id);
+            } else {
+                reject('No active tab found');
+            }
+        });
+    });
+}
+
+// chrome.tabs.query({}, (tabs) => {
+//     tabs.forEach(function (tab) {
+//         console.log(tab.id); // 输出每个标签页的ID
+//     });
+// })