v0.8.0

export modules and classes to editor
ChiChou · Dec 23, 2023 · 646da02 · 646da02
1 parent b4d72dd
commit 646da02
Show file tree

Hide file tree

Showing 6 changed files with 100 additions and 20 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -8,7 +8,7 @@
     },
     "displayName": "frida Workbench",
     "description": "Unofficial frida workbench for VSCode",
-    "version": "0.7.7",
+    "version": "0.8.0",
     "engines": {
         "vscode": "^1.69.1"
     },
@@ -141,6 +141,16 @@
                 "title": "Close All Console",
                 "category": "Frida"
             },
+            {
+                "command": "frida.print.modules",
+                "title": "Export All Modules To Editor",
+                "category": "Frida"
+            },
+            {
+                "command": "frida.print.classes",
+                "title": "Export All Classes To Editor",
+                "category": "Frida"
+            },
             {
                 "command": "frida.external.objection",
                 "title": "Objection",
@@ -274,6 +284,14 @@
                     "command": "frida.syslog",
                     "when": "false"
                 },
+                {
+                    "command": "frida.print.modules",
+                    "when": "false"
+                },
+                {
+                    "command": "frida.print.classes",
+                    "when": "false"
+                },
                 {
                     "command": "frida.attach",
                     "when": "false"
@@ -416,6 +434,26 @@
                     "when": "view == fridaPs && viewItem =~ /^process\\|/",
                     "group": "3_frida_logger@1"
                 },
+                {
+                    "command": "frida.print.modules",
+                    "when": "view == fridaApps && viewItem =~ /^app\\|/",
+                    "group": "3_frida_logger@2"
+                },
+                {
+                    "command": "frida.print.modules",
+                    "when": "view == fridaPs && viewItem =~ /^process\\|/",
+                    "group": "3_frida_logger@2"
+                },
+                {
+                    "command": "frida.print.classes",
+                    "when": "view == fridaApps && viewItem =~ /^app\\|/",
+                    "group": "3_frida_logger@3"
+                },
+                {
+                    "command": "frida.print.classes",
+                    "when": "view == fridaPs && viewItem =~ /^process\\|/",
+                    "group": "3_frida_logger@3"
+                },
                 {
                     "command": "frida.external.objection",
                     "when": "view == fridaApps && viewItem =~ /^app\\|/",

diff --git a/src/commands/print.ts b/src/commands/print.ts
@@ -0,0 +1,39 @@
+import { Uri, window } from "vscode";
+import { rpc } from "../driver/backend";
+import { TargetItem } from "../providers/devices";
+
+interface Module {
+  name: string;
+  base: string;
+  size: number;
+  path: string;
+};
+
+type ClassesResult = string[];
+type ModulesResult = Module[];
+
+function create(name: string, content: string) {
+  window.showTextDocument(Uri.parse(`untitled:${encodeURIComponent(name)}`)).then((editor) => {
+    editor.edit((editBuilder) => {
+      editBuilder.insert(editor.selection.active, content);
+    });
+  });
+}
+
+export function classes(target: TargetItem) {
+  rpc(target, 'classes')
+    .then((result: ClassesResult) => {
+      const text = result.join('\n');
+      create(`classes - ${target.label}.txt`, text);
+    })
+    .catch(err => window.showErrorMessage(err.message));
+}
+
+export function modules(target: TargetItem) {
+  rpc(target, 'modules')
+    .then((modules: ModulesResult) => {
+      const text = modules.map(m => `${m.base} ${m.name} ${m.size} ${m.path}`).join('\n');
+      create(`modules - ${target.label}.txt`, text);
+    })
+    .catch(err => window.showErrorMessage(err.message));
+}
diff --git a/src/commands/syslog.ts b/src/commands/syslog.ts
@@ -15,7 +15,7 @@ export function vacuum() {
   }
 }
 
-export async function show(node?: TargetItem) {
+export async function show(node: TargetItem) {
   function cmdChannel(name: string, bin: string, args: string[]) {
     if (name in active) {
       return active[name];
@@ -43,9 +43,7 @@ export async function show(node?: TargetItem) {
   } else if (node instanceof ProcessItem) {
     bundleOrPid = ['--pid', node.data.pid.toString()];
   } else {
-    if (node) {
-      window.showErrorMessage(`Invalid target "${node.label}"`);
-    }
+    window.showErrorMessage(`Invalid target "${node.label}"`);
     return;
   }
 

diff --git a/src/driver/backend.ts b/src/driver/backend.ts
@@ -14,18 +14,6 @@ const py = join(base, 'driver.py');
 
 export const driverScript = () => py;
 
-export function flags(node: TargetItem) {
-  if (node instanceof DeviceItem) {
-    return ['--device', node.data.id];
-  } else if (node instanceof AppItem) {
-    return ['--app', node.data.identifier];
-  } else if (node instanceof ProcessItem) {
-    return ['--pid', node.data.pid.toString()];
-  }
-
-  throw new Error(`Invalid target "${node}"`);
-}
-
 function askInstallFrida() {
   window.showErrorMessage(`Frida python module not detected. Please check your Python interpreter setting,
 or pip install frida-tools. Do you want to install now?`, 'Install', 'Cancel')
@@ -42,7 +30,7 @@ or pip install frida-tools. Do you want to install now?`, 'Install', 'Cancel')
 export function exec(...args: string[]): Promise<any> {
   const remoteDevices = asParam();
   return new Promise((resolve, reject) => {
-    execFile(python3Path(), [py, ...remoteDevices, ...args], {}, (err, stdout, stderr) => {
+    execFile(python3Path(), [py, ...remoteDevices, ...args], { maxBuffer: 1024 * 1024 * 20}, (err, stdout, stderr) => {
       if (err) {
         if (stderr.includes('Unable to import frida')) {
           askInstallFrida();
@@ -87,6 +75,19 @@ export function location(id: string, bundle: string) {
   return exec('location', id, bundle);
 }
 
+export function rpc(target: TargetItem, method: string, ...args: string[]) {
+  let bundleOrPid: string[];
+  if (target instanceof AppItem) {
+    bundleOrPid = ['--app', target.data.identifier];
+  } else if (target instanceof ProcessItem) {
+    bundleOrPid = ['--pid', target.data.pid.toString()];
+  } else {
+    throw new Error(`Invalid target "${target}"`);
+  }
+
+  return exec('rpc', '--device', target.device.id, ...bundleOrPid, method, ...args);
+}
+
 export function lockdownSyslog(id: string, bundleOrPid: string[]) {
   return run({
     name: `Syslog: ${bundleOrPid}`,

diff --git a/src/extension.ts b/src/extension.ts
@@ -17,6 +17,7 @@ import * as rootless from './commands/rootless';
 import * as ssh from './commands/ssh';
 import * as syslog from './commands/syslog';
 import * as typing from './commands/typing';
+import * as print from './commands/print';
 
 export function activate(context: vscode.ExtensionContext) {
 	const register = (cmd: string, cb: (...args: any[]) => any) => vscode.commands.registerCommand(cmd, cb);
@@ -61,6 +62,9 @@ export function activate(context: vscode.ExtensionContext) {
 	push(register('frida.debug.setup', boilerplate.debug));
 
 	push(register('frida.typing.init', typing.init));
+
+	push(register('frida.print.classes', print.classes));
+	push(register('frida.print.modules', print.modules));
 }
 
 // this method is called when your extension is deactivated