forked from JaveleyQAQ/WeChatOpenDevTools-Python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathcommons.py
More file actions
111 lines (98 loc) · 4.55 KB
/
commons.py
File metadata and controls
111 lines (98 loc) · 4.55 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
# commons.py
from utils.colors import Color
from utils.wechatutils import WechatUtils
import frida
import sys
import time
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(Color.GREEN + f"[+] 成功注入微信PID: {pid}", Color.END)
return session
except Exception as e:
print(Color.RED + f"[-] 注入微信失败PID {pid}: {e}", Color.END)
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(Color.GREEN +f"[+] 成功注入{version}小程序版本，PID: {pid}", Color.END)
except Exception as e:
print(Color.RED + f"[-] 注入{version}小程序版本失败！", Color.END)
else:
self.wechatutils_instance.print_process_not_found_message()
# 管理会话
while self.active_sessions:
self.manage_sessions()
time.sleep(5) # 每5秒检查一次
def load_wechatEXE_configs(self):
wechat_instances = self.wechatutils_instance.get_wechat_pids_and_versions()
if wechat_instances:
print(Color.RED + f"[-] 请退出所有微信实例后再执行该命令 " + Color.END)
return 0
wechatEXEpath = self.wechatutils_instance.find_installation_path("微信")
path = self.wechatutils_instance.get_configs_path()
wechatEXE_hookcode = open(path + "..\\scripts\\WechatWin.dll\\hook.js", "r", encoding="utf-8").read()
self.inject_wechatDLL(wechatEXEpath, wechatEXE_hookcode)
def load_wechatEXE_and_wechatEx(self):
wechat_instances = self.wechatutils_instance.get_wechat_pids_and_versions()
if wechat_instances:
print(Color.RED + f"[-] 请关闭所有微信实例后再执行该命令 " + Color.END)
return 0
self.load_wechatEXE_configs()
self.load_wechatEx_configs()
def manage_sessions(self):
for session in self.active_sessions[:]: # 使用切片创建副本以便在迭代时修改
if session.is_detached:
print(f"Session {session} detached, removing from active sessions.")
self.active_sessions.remove(session)
def get_cpu_architecture():
try:
cpu_arch = platform.platform().lower()
if "64bit" in cpu_arch and "macos" in cpu_arch:
return "MacOS x64"
return "Windows" # 默认返回Windows
except Exception as e:
print(Color.RED, f"[-] Error detecting CPU architecture: {e} ", Color.END)
return "Windows"
# 主函数示例
if __name__ == "__main__":
commons = Commons()
commons.load_wechatEx_configs()