Run :ies only on app classes

nowsecure · Dec 10, 2024 · 71557b4 · 71557b4
1 parent e7ba497
commit 71557b4
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/agent/lib/info/index.ts b/src/agent/lib/info/index.ts
@@ -234,11 +234,13 @@ export function listEntrypointSymbols(args: string[]): string {
         const classes = ObjC.classes;
         Object.keys(classes).forEach(function (className: string) {
             var cls = ObjC.classes[className];
+            if (cls.$moduleName !== null && cls.$moduleName !== "main") {
+                return;
+            }
             var methods = cls.$methods; // $ownMethods?
             methods.forEach(function (methodName) {
                 try {
                     var address = cls[methodName].implementation; // Get the implementation address
-                    console.log("  Method: " + methodName + " | Address: " + address);
                     if (validEntrypoints.includes(methodName)) {
                         symbols.push({ name: className + "." + methodName, address: address });
                     }
@@ -253,8 +255,8 @@ export function listEntrypointSymbols(args: string[]): string {
         return "";
     }
     const entries = symbols
-        .map((entry) => {
-            return 'f entry.' + entry.name + ' = ' + entry.address;
+        .map((symbol) => {
+            return 'f entry.' + symbol.name + ' = ' + symbol.address;
         }).join('\n');
     return "fs+symbols\n" + entries + "\nfs-";
 }