script.js

function HookingOptimizedVersion(classesToInspect){
        //  sort classesToInspect
        classesToInspect.sort();
        // var count_classes_total = classesToInspect.length;
        // console.log("Total classes loaded: " + count_classes_total);
        // //  get package name
        // var packageName = Java.use("android.app.ActivityThread").currentApplication().getApplicationContext().getPackageName();
        // console.log("Package name: " + packageName);
        // console.log("Total classes to inspect: " + classesToInspect.length);
    
        classesToInspect.forEach(function(className){
            //  create a wrapper
            try{
                var hook = Java.use(className);
                var methods = hook.class.getDeclaredMethods();
            }
            catch(e){
                // console.log("Error in class : " +className + "=> " + e);
                return;
            }
            // get all methods's name and remove duplicates (because we will use overloding to instrument methods having same name).
            var parseMethods = [];
            methods.forEach(function(method){
               var method_name = method.getName();
               if (!parseMethods.includes(method_name)){
                   parseMethods.push(method_name);
               }
            });
            // instrument all methods one by one
            parseMethods.forEach(function(method){
                //  validate if method is not undefined, otherwise frida gives error
                if (typeof  hook[method] != "undefined"){
                    //  get all overloaded methods
                    var overloadMethods = hook[method].overloads;
                    var method_number = 1;
                    overloadMethods.forEach(function(overload){
                        if (method_number == 1){
                            var method_name = className + "." + method;
                        }
                        else{
                            var method_name = className + "." + method + "_" + method_number.toString();
                        }
                        send("methods " + method_name);
                        overload.implementation = function(...args){
                            // console.log("Entered: " + method_name);
                            send("enter " + method_name);
                            var ret = overload.call(this, ...args);
                            send("exit " + method_name);
                            // console.log("Exited: " + method_name);
                            return ret;
                        }
                        method_number += 1;
                    });
                }
            });
        });
}

var Main = function () {
    var data;
    send("sendClass", ["hello"]);
    recv('recvClass', function (value) {
        data = value.payload;
        // console.log('received bytes from python:', data, 'size:', data.length);
        // console.log('..........received total classes from python:', data.length);
    }).wait(); // block until python respond
    let arr2 = data;
    let count = data.length;
    send("number " + (count).toString());
    HookingOptimizedVersion(arr2);
};

Java.perform(Main);