指定pid

Author: 58

.github/ISSUE_TEMPLATE/bug_report.md

@@ -3,14 +3,27 @@
 from utils.commons import Commons
 from utils.banner import generate_banner
 from utils.colors import Color
-
+import argparse
 
 
 def print_colored_message(message, color):
     print(color + message + Color.END)
 
 def main():
-    commons.load_wechatEx_configs()
+    HELPALL = """
+    请选择要执行的方法：         
+                        [+] python  main.py -h  查看帮助
+                        [+] python  main.py -p  指定pid              
+                                     
+    """
+    parser = argparse.ArgumentParser(description=HELPALL, formatter_class=RawTextHelpFormatter)
+    parser.add_argument('-p', help='指定pid',dest="pid", type=int)
+    args = parser.parse_args()
+
+    if args.pid:
+        commons.load_wechatEx_configs_pid(args.pid)
+    else:
+        commons.load_wechatEx_configs()
 
 if __name__ == "__main__":
     generate_banner()

main.py

@@ -40,3 +40,13 @@ def load_wechatEx_configs(self):
         else:
             self.wechatutils_instance.print_process_not_found_message()
 
+    def load_wechatEx_configs_pid(self,pid):
+        path = self.wechatutils_instance.get_configs_path()
+        version = self.wechatutils_instance.get_wechat_version_mac()
+        if pid or version is not None:
+            wehcatEx_hookcode = open(path + "../scripts/hook.js", "r", encoding="utf-8").read()
+            wechatEx_addresses = open(path + "../configs/address_{}_arm.json".format(version)).read()
+            wehcatEx_hookcode = "var address=" + wechatEx_addresses + wehcatEx_hookcode
+            self.inject_wehcatEx(pid, wehcatEx_hookcode)
+        else:
+            self.wechatutils_instance.print_process_not_found_message()

utils/commons.py

@@ -82,3 +82,10 @@ def get_wechat_pid_and_version_mac(self):
         except subprocess.CalledProcessError as e:
             return e.stderr
 
+    def get_wechat_version_mac(self):
+        try:
+            version_command = "ps aux | grep 'WeChatAppEx' |  grep -v 'grep' | grep ' --client_version' | grep '-user-agent=' | grep -oE 'MacWechat/([0-9]+\.)+[0-9]+\(0x\d+\)' |  grep -oE '(0x\d+)' | sed 's/0x//g' | head -n 1"
+            version  = subprocess.run(version_command, shell=True, check=True, text=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE).stdout.replace("\n","")
+            return version
+        except subprocess.CalledProcessError as e:
+            return e.stderr