Files
WeChatDB/GetWeChatAesKey.py
T
2023-11-11 10:43:18 +08:00

160 lines
4.6 KiB
Python

#! /usr/bin/env python3
# -*- coding: utf-8 -*-
import binascii
import os
import struct
import segno
from pymem import Pymem, process, pattern
from win32api import HIWORD, LOWORD, GetFileVersionInfo
import argparse
def error():
print("请按提示排查,如仍有问题可扫描二维码联系作者")
qrcode = segno.make("http://weixin.qq.com/r/OzopMZrEuMfHrd5S928p")
qrcode.show()
exit(-1)
# Adapted from: http://stackoverflow.com/a/495305/1338797
def arch_of(dll_file):
with open(dll_file, 'rb') as f:
doshdr = f.read(64)
magic, padding, offset = struct.unpack('2s58si', doshdr)
# print magic, offset
if magic != b'MZ':
return None
f.seek(offset, os.SEEK_SET)
pehdr = f.read(6)
# careful! H == unsigned short, x64 is negative with signed
magic, padding, machine = struct.unpack('2s2sH', pehdr)
# print magic, hex(machine)
if magic != b'PE':
return None
if machine == 0x014c:
return 'i386'
if machine == 0x0200:
return 'IA64'
if machine == 0x8664:
return 'x64'
return 'unknown'
def is_64_bit(pm):
exe_arch = arch_of(list(pm.list_modules())[0].filename)
if exe_arch == "x64":
return True
else:
return False
def getVersionBase(pm):
WeChatWindll_base = 0
WeChatWindll_path = ""
for m in list(pm.list_modules()):
path = m.filename
if path.endswith("WeChatWin.dll"):
WeChatWindll_base = m.lpBaseOfDll
WeChatWindll_path = path
break
if not WeChatWindll_path:
print("获取版本失败,请确认本系统是否成功安装了微信!")
error()
version = GetFileVersionInfo(WeChatWindll_path, "\\")
msv = version['FileVersionMS']
lsv = version['FileVersionLS']
version = f"{str(HIWORD(msv))}.{str(LOWORD(msv))}.{str(HIWORD(lsv))}.{str(LOWORD(lsv))}"
return version, WeChatWindll_base
def getAesKey(pm, base, offset):
try:
if is_64_bit(pm):
result = pm.read_bytes(base + offset, 8) # 读取 AES Key 的地址
addr = struct.unpack("<Q", result)[0] # 地址为小端 8 字节整型
else:
result = pm.read_bytes(base + offset, 4) # 读取 AES Key 的地址
addr = struct.unpack("<I", result)[0] # 地址为小端 4 字节整型
aesKey = pm.read_bytes(addr, 0x20) # 读取 AES Key
result = binascii.b2a_hex(aesKey) # 解码
except Exception as e:
print(f"{e}")
print(f"请确认微信已经登录!")
error()
return result.decode()
# Tested
# 3.9.5.91 (64bit)
# 3.9.7.29 (64bit)
# 3.9.8.15 (64bit)
# 3.9.5.80 (32bit)
def getWechatIdAdd(pm, id):
bytes_pattern = bytearray()
bytes_pattern.extend(map(ord, id))
id_pattern = bytes(bytes_pattern)
wechatwindll_module = process.module_from_name(pm.process_handle, "WeChatWin.dll")
wechat_id_addrs = pattern.pattern_scan_module(pm.process_handle, wechatwindll_module, id_pattern, return_multiple=True)
if wechat_id_addrs == None or len(wechat_id_addrs) != 2:
print(f"未能寻获微信账号: {id}")
error()
return wechat_id_addrs[1] - (64 if is_64_bit(pm) else 36)
AESKEY_OFFSETS = {
"3.3.0.115": 0x1DDF914,
"3.3.5.34": 0x1D2FB34,
"3.6.0.18": 0x222EFE4,
"3.7.0.29": 0x2363524,
"3.7.0.30": 0x2366524,
"3.7.5.23": 0x242413C,
"3.8.1.26": 0x2C429FC,
"3.9.0.28": 0x2E2D1AC,
"3.9.2.23": 0x2FFD90C,
}
AESKEY_OFFSETS_64 = {
"3.9.5.91": 0x3ACCC70
}
if __name__ == "__main__":
try:
pm = Pymem("WeChat.exe")
except Exception as e:
print(f"{e},请确认微信程序已经打开并登录!")
error()
ap = argparse.ArgumentParser()
ap.add_argument("-i", "--id", required=False, help="用微信账号")
args = vars(ap.parse_args())
version, base = getVersionBase(pm)
print(f"微信版本:{version} " + ("(64bit)" if is_64_bit(pm) else "(32bit)"))
print(f"微信基址:{hex(base)}")
wechatid = args["id"]
if wechatid != None:
print(f"使用微信账号《{wechatid}》搜索")
offset = getWechatIdAdd(pm, wechatid) - base
else:
if is_64_bit(pm):
offset = AESKEY_OFFSETS_64.get(version, None)
else:
offset = AESKEY_OFFSETS.get(version, None)
if not offset:
print(f"暂不支持版本 {version},请联系作者。")
error()
print(f"偏移地址:{hex(offset)}")
aesKey = getAesKey(pm, base, offset)
print(f"数据库密钥:{aesKey}")