[FEATURE] Implement more of the base JADX features

[privacyguy123]

Feature Description

Search Class/Method/Field/Code/Resource/Comments by string
Implement "Find Usage" option

Use Case

It lets the AI Agent use JADX more like how a human would. When I am analyzing an obufscated APK I am often searching strings here, not listing all classes (which causes terrible lag and crashes btw in an obfuscated APK with 1000s of classes called "a.*" etc.) or methods. Same goes for "Find Usages" - in an obfuscated APK it's critical to understand how something works by it's usages because there are no nice plain text names.

Proposed Solution

I think it's a case of just implementing more of the features already available to us in the JADX API but I am not skilled enough to do it myself just yet.

Alternatives Considered

I implemented (badly, with AI) my own "Find Usages" which seems to work but the code is dubious, you can take a look if you like.

[zinja-coder]

Hey @privacyguy123 ,

First of all a great idea, thanks for pointing out this. I will sure work on this.

Also share your code, will look into it.

[privacyguy123]

Like this:

    // ++ Handler for finding method usages ++
    class FindUsagesHandler implements HttpHandler {
        @Override
        public void handle(HttpExchange exchange) throws IOException {
            Map<String, String> params = queryToMap(exchange.getRequestURI().getQuery());
            String className = params.get("class");
            String methodName = params.get("method"); // Note: Signature matching would be more robust

            if (className == null || className.isEmpty() || methodName == null || methodName.isEmpty()) {
                sendJson(exchange, 400, Map.of("error", "Missing 'class' or 'method' parameter."));
                return;
            }

            List<String> usages = new ArrayList<>();
            try {
                JadxWrapper wrapper = mainWindow.getWrapper();
                JavaClass targetCls = wrapper.getClasses().stream()
                                             .filter(cls -> cls.getFullName().equals(className))
                                             .findFirst().orElse(null);

                if (targetCls == null) {
                    sendJson(exchange, 404, Map.of("error", "Target class not found: " + className));
                    return;
                }

                JavaMethod targetMethod = null;
                for (JavaMethod m : targetCls.getMethods()) {
                    // Simple name matching. For robustness, signature matching should be implemented.
                    if (m.getName().equals(methodName)) {
                        targetMethod = m;
                        break;
                    }
                }

                if (targetMethod == null) {
                    sendJson(exchange, 404, Map.of("error", "Target method not found: " + methodName + " in class " + className));
                    return;
                }

                // Use JADX API to find usages
                List<JavaNode> usePlaces = targetMethod.getUseIn();

                // Simplified approach: Report the class containing the usage
                for (JavaNode usageNode : usePlaces) {
                    JavaClass containingClass = usageNode.getTopParentClass(); // Use public API
                    if (containingClass != null) {
                        usages.add(containingClass.getFullName()); // Report class name
                    } else {
                        usages.add("Unknown usage location"); // Fallback
                    }
                }

                sendJson(exchange, 200, usages);

            } catch (Exception e) {
                e.printStackTrace();
                sendJson(exchange, 500, Map.of("error", "Internal error finding usages: " + e.getMessage()));
            }
        }
    }
    // -- End of FindUsagesHandler --

[zinja-coder]

Awesome, This looks promising.

I am busy at a workshop for three days. Meanwhile try this locally. If it works as you expected open a pull request and I will merge your code.

We have to create respective MCP tool in MCP script as well.

Otherwise once i am back from the workshop I will look into this and co-ordinate to achieve this feature.

Thanks you.

[privacyguy123]

This one yes?

@mcp.tool()
async def find_usages(class_name: str, method_name: str) -> List[str]:
    """Find usages of a specific method within a class across the project."""
    response = await get_from_jadx("find-usages", {"class": class_name, "method": method_name})
    return response.splitlines() if response else []

I'll git clone all your repos and submit the PRs. These edits were done locally! If your trained eye sees no issues that's good enough for me - the tool seems to be working on my side.

[zinja-coder]

Give it a shot 🔥

[zinja-coder]

Hey @privacyguy123,

It's been a long since last heard from you, do you have plan to work on the idea you provided?