74 lines
1.8 KiB
C++
74 lines
1.8 KiB
C++
#include "Shlwapi.h"
|
|
#include "framework.h"
|
|
#include <process.h>
|
|
#include <tlhelp32.h>
|
|
|
|
#include "injector.h"
|
|
#include "log.h"
|
|
#include "sdk.h"
|
|
#include "util.h"
|
|
|
|
static DWORD wcPid = 0;
|
|
static HANDLE wcProcess = NULL;
|
|
static HMODULE spyBase = NULL;
|
|
static WCHAR spyDllPath[MAX_PATH] = { 0 };
|
|
|
|
int WxInitSDK()
|
|
{
|
|
int status = 0;
|
|
InitLogger();
|
|
LOG_INFO("WxInitSDK.");
|
|
GetModuleFileName(GetModuleHandle(WECHATSDKDLL), spyDllPath, MAX_PATH);
|
|
PathRemoveFileSpec(spyDllPath);
|
|
PathAppend(spyDllPath, WECHATINJECTDLL);
|
|
|
|
if (!PathFileExists(spyDllPath)) {
|
|
LOG_ERROR("DLL does not exists.");
|
|
return ERROR_FILE_NOT_FOUND;
|
|
}
|
|
|
|
status = OpenWeChat(&wcPid);
|
|
if (status != 0) {
|
|
LOG_ERROR("OpenWeChat failed: {}.", status);
|
|
return status;
|
|
}
|
|
|
|
Sleep(2000); // 等待微信打开
|
|
wcProcess = InjectDll(wcPid, spyDllPath, &spyBase);
|
|
if (wcProcess == NULL) {
|
|
LOG_ERROR("Failed to Inject DLL into WeChat.");
|
|
return -1;
|
|
}
|
|
|
|
if (!CallDllFunc(wcProcess, spyDllPath, spyBase, "InitSpy", NULL)) {
|
|
LOG_ERROR("Failed to InitSpy.");
|
|
return -1;
|
|
}
|
|
|
|
do {
|
|
if (!CallDllFunc(wcProcess, spyDllPath, spyBase, "IsLogin", (DWORD *)&status)) {
|
|
LOG_ERROR("Failed to check login status.");
|
|
return -1;
|
|
}
|
|
Sleep(1000);
|
|
} while (status == 0);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int WxDestroySDK()
|
|
{
|
|
LOG_INFO("WxDestroySDK");
|
|
if (!CallDllFunc(wcProcess, spyDllPath, spyBase, "CleanupSpy", NULL)) {
|
|
LOG_ERROR("Failed to CleanupSpy.");
|
|
return -1;
|
|
}
|
|
|
|
if (!EjectDll(wcProcess, spyBase)) {
|
|
LOG_ERROR("Failed to Eject DLL.");
|
|
return -1; // TODO: Unify error codes
|
|
}
|
|
|
|
return 0;
|
|
}
|