50 lines
1.1 KiB
C++
50 lines
1.1 KiB
C++
#include <filesystem>
|
|
|
|
#include "log.hpp"
|
|
#include "rpc_server.h"
|
|
#include "spy.h"
|
|
#include "util.h"
|
|
|
|
UINT64 g_WeChatWinDllAddr = 0;
|
|
|
|
static bool IsWxVersionMatched(const wchar_t *version)
|
|
{
|
|
if (wcscmp(version, SUPPORT_VERSION) != 0) {
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
void InitSpy(LPVOID args)
|
|
{
|
|
|
|
wchar_t version[16] = { 0 };
|
|
PortPath_t *pp = (PortPath_t *)args;
|
|
|
|
Log::InitLogger(pp->path);
|
|
g_WeChatWinDllAddr = (UINT64)GetModuleHandle(L"WeChatWin.dll"); // 获取wechatWin模块地址
|
|
if (g_WeChatWinDllAddr == 0) {
|
|
LOG_ERROR("获取 wechatWin.dll 模块地址失败");
|
|
return; // TODO: 退出进程，避免后面操作失败
|
|
}
|
|
|
|
if (!GetWeChatVersion(version)) { // 获取微信版本
|
|
LOG_ERROR("获取微信版本失败");
|
|
return;
|
|
}
|
|
LOG_INFO("WeChat version: {}", Wstring2String(version).c_str());
|
|
if (!IsWxVersionMatched(version)) {
|
|
LOG_ERROR("不支持当前版本");
|
|
MessageBox(NULL, L"不支持当前版本", L"错误", 0);
|
|
return;
|
|
}
|
|
|
|
RpcStartServer(pp->port);
|
|
}
|
|
|
|
void CleanupSpy()
|
|
{
|
|
LOG_DEBUG("CleanupSpy");
|
|
RpcStopServer();
|
|
}
|