Class Commons

utils/commons.py:9–96

import platform

class Commons:
    def __init__(self):
        self.wechatutils_instance = WechatUtils()
        self.device = frida.get_local_device()
        self.process = self.device.enumerate_processes()
        self.version_list = []
        self.configs_path = ""
        self.active_sessions = []

    def onMessage(self, message, data):
        if message['type'] == 'send':
            print(Color.GREEN + message['payload'], Color.END)
        elif message['type'] == 'error':
            print(Color.RED + message['stack'], Color.END)

    def inject_wechatEx(self, pid, code):
        try:
            session = frida.attach(pid)
            script = session.create_script(code)
            script.on("message", self.onMessage)
            script.load()
            print(f"Successfully injected into WeChat PID: {pid}")
            return session
        except Exception as e:
            print(f"Error injecting into WeChat PID {pid}: {e}")
            return None

    def inject_wechatDLL(self, path, code):
        pid = self.device.spawn(path)
        session = frida.attach(pid)
        script = session.create_script(code)
        script.on("message", self.onMessage)
        script.load()
        self.device.resume(pid)
        time.sleep(10)
        session.detach()

    def load_wechatEx_configs(self):
        path = self.wechatutils_instance.get_configs_path()
        if get_cpu_architecture() == "MacOS x64":
            wechat_instances = self.wechatutils_instance.get_wechat_pids_and_versions_mac()
        else:
            wechat_instances = self.wechatutils_instance.get_wechat_pids_and_versions()

        if wechat_instances:
            for pid, version in wechat_instances:
                try:
                    wechatEx_hookcode = open(path + "../scripts/hook.js", "r", encoding="utf-8").read()
                    wechatEx_addresses = open(path + f"../configs/address_{version}_x64.json").read()
                    wechatEx_hookcode = "var address=" + wechatEx_addresses + wechatEx_hookcode
                    session = self.inject_wechatEx(pid, wechatEx_hookcode)
                    if session:
                        self.active_sessions.append(session)
                    print(f"Injected into WeChat instance PID: {pid}, Version: {version}")
                except Exception as e:
                    print(f"Error injecting into WeChat PID {pid}: {e}")
        else:
            self.wechatutils_instance.print_process_not_found_message()