cyonjan/WeChatFerry
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Impl DisableReceiveMsg

Browse Source
This commit is contained in:
Changhua 2022-08-20 15:15:04 +08:00
parent bdcf71cc37
commit e2e148c73f
14 changed files with 93 additions and 45 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
App/App.cpp
View File

@ -71,7 +71,7 @@ int main()
Sleep(1000); // 等待1秒
wprintf(L"Message: 接收通知中......\n");
WxSetTextMsgCb(onTextMsg);
WxEnableRecvMsg(onTextMsg);
Sleep(1000); // 等待1秒
// 测试发送消息

8
App/App.py
View File

@ -7,11 +7,9 @@ import wcferry as sdk
def main():
print(dir(sdk)) # 查看SDK支持的方法和属性
help(sdk.WxSetTextMsgCb) # 查看某方法的情况
help(sdk.WxEnableRecvMsg) # 查看某方法的情况
help(sdk.WxMessage) # 查看消息结构
help(sdk.WxContact) # 查看通讯录结构
WxMsgTypes = sdk.WxGetMsgTypes() # 获取消息类型
print(WxMsgTypes) # 查看消息类型
# 初始化SDK，如果成功，返回0；否则失败
status = sdk.WxInitSDK()
@ -20,6 +18,8 @@ def main():
exit(-1)
print("初始化成功")
WxMsgTypes = sdk.WxGetMsgTypes() # 获取消息类型
print(WxMsgTypes) # 查看消息类型
time.sleep(2)
print("打印通讯录......")
@ -61,7 +61,7 @@ def main():
return 0
print("Message: 接收通知中......")
sdk.WxSetTextMsgCb(OnTextMsg) # 设置回调，接收消息
sdk.WxEnableRecvMsg(OnTextMsg) # 设置回调，接收消息
while True:
time.sleep(1)

1
Rpc/rpc.idl
View File

@ -56,5 +56,6 @@ interface ISpy
int GetDbTables([ in, string ] const wchar_t *db, [out] int *pNum, [ out, size_is(, *pNum) ] PPRpcTables *tbls);
void EnableReceiveMsg();
void DisableReceiveMsg();
[callback] int ReceiveMsg([in] RpcMessage_t rpcMsg);
};

31
SDK/rpc_client.cpp
View File

@ -33,8 +33,6 @@ RPC_STATUS RpcDisconnectServer()
RPC_STATUS status;
// Free the memory allocated by a string
status = RpcStringFree(&pszStringBinding);
if (status)
return status;
// Releases binding handle resources and disconnects from the server
status = RpcBindingFree(&hSpyBinding);
@ -42,7 +40,7 @@ RPC_STATUS RpcDisconnectServer()
return status;
}
unsigned int __stdcall RpcSetTextMsgCb(void *p)
int RpcEnableReceiveMsg()
{
unsigned long ulCode = 0;
RpcTryExcept
@ -53,7 +51,25 @@ unsigned int __stdcall RpcSetTextMsgCb(void *p)
RpcExcept(1)
{
ulCode = RpcExceptionCode();
printf("rpcWxSetTextMsgCb exception 0x%lx = %ld\n", ulCode, ulCode);
printf("RpcEnableReceiveMsg exception 0x%lx = %ld\n", ulCode, ulCode);
}
RpcEndExcept;
return 0;
}
int RpcDisableReceiveMsg()
{
unsigned long ulCode = 0;
RpcTryExcept
{
// UnHook Message receiving
client_DisableReceiveMsg();
}
RpcExcept(1)
{
ulCode = RpcExceptionCode();
printf("RpcDisableReceiveMsg exception 0x%lx = %ld\n", ulCode, ulCode);
}
RpcEndExcept;
@ -72,7 +88,8 @@ int RpcIsLogin()
RpcExcept(1)
{
ulCode = RpcExceptionCode();
printf("rpcIsLogin exception 0x%lx = %ld\n", ulCode, ulCode);
printf("RpcIsLogin exception 0x%lx = %ld\n", ulCode, ulCode);
return -1;
}
RpcEndExcept;
@ -88,7 +105,7 @@ int RpcSendTextMsg(const wchar_t *wxid, const wchar_t *at_wxid, const wchar_t *m
RpcExcept(1)
{
ulCode = RpcExceptionCode();
printf("rpcWxSendTextMsg exception 0x%lx = %ld\n", ulCode, ulCode);
printf("RpcSendTextMsg exception 0x%lx = %ld\n", ulCode, ulCode);
}
RpcEndExcept;
@ -104,7 +121,7 @@ int RpcSendImageMsg(const wchar_t *wxid, const wchar_t *path)
RpcExcept(1)
{
ulCode = RpcExceptionCode();
printf("rpcWxSendImageMsg exception 0x%lx = %ld\n", ulCode, ulCode);
printf("RpcSendImageMsg exception 0x%lx = %ld\n", ulCode, ulCode);
}
RpcEndExcept;

3
SDK/rpc_client.h
View File

@ -5,7 +5,8 @@
RPC_STATUS RpcConnectServer();
RPC_STATUS RpcDisconnectServer();
unsigned int __stdcall RpcSetTextMsgCb(void *p);
int RpcEnableReceiveMsg();
int RpcDisableReceiveMsg();
int RpcIsLogin();
int RpcSendTextMsg(const wchar_t *wxid, const wchar_t *at_wxid, const wchar_t *msg);
int RpcSendImageMsg(const wchar_t *wxid, const wchar_t *path);

38
SDK/sdk.cpp
View File

@ -19,7 +19,7 @@ static WCHAR SpyDllPath[MAX_PATH] = { 0 };
int WxInitSDK()
{
DWORD status = 0;
int status = 0;
unsigned long ulCode = 0;
GetModuleFileName(GetModuleHandle(WECHATSDKDLL), SpyDllPath, MAX_PATH);
@ -34,35 +34,49 @@ int WxInitSDK()
if (status != 0) {
return status;
}
Sleep(2000); // 等待微信打开
if (InjectDll(WeChatPID, SpyDllPath)) {
return -1;
}
RpcConnectServer();
while (!RpcIsLogin()) {
Sleep(1000);
Sleep(1000); // 等待SPY就绪
status = RpcConnectServer();
if (status != 0) {
printf("RpcConnectServer: %d\n", status);
return -1;
}
do {
status = RpcIsLogin();
if (status == -1) {
return status;
}
else if (status == 1) {
break;
}
Sleep(1000);
} while (1);
return ERROR_SUCCESS;
}
int WxDestroySDK()
{
WxDisableRecvMsg();
RpcDisconnectServer();
EjectDll(WeChatPID, SpyDllPath);
// 关闭 RPC，但不卸载 DLL，方便下次使用。
//EjectDll(WeChatPID, SpyDllPath);
return ERROR_SUCCESS;
}
int WxSetTextMsgCb(const std::function<int(WxMessage_t)> &onMsg)
int WxEnableRecvMsg(const std::function<int(WxMessage_t)> &onMsg)
{
if (onMsg) {
HANDLE msgThread;
g_cbReceiveTextMsg = onMsg;
msgThread = (HANDLE)_beginthreadex(NULL, 0, RpcSetTextMsgCb, NULL, 0, NULL);
msgThread = (HANDLE)CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)RpcEnableReceiveMsg, NULL, 0, NULL);
if (msgThread == NULL) {
printf("Failed to create innerWxRecvTextMsg.\n");
return -2;
@ -76,6 +90,12 @@ int WxSetTextMsgCb(const std::function<int(WxMessage_t)> &onMsg)
return -1;
}
int WxDisableRecvMsg()
{
RpcDisableReceiveMsg();
return -1;
}
int WxSendTextMsg(wstring wxid, wstring at_wxid, wstring msg)
{
return RpcSendTextMsg(wxid.c_str(), at_wxid.c_str(), msg.c_str());

3
SDK/sdk.def
View File

@ -1,7 +1,8 @@
EXPORTS
WxInitSDK
WxDestroySDK
WxSetTextMsgCb
WxEnableRecvMsg
WxDisableRecvMsg
WxSendTextMsg
WxGetMsgTypes
WxSendImageMsg

3
SDK/sdk.h
View File

@ -39,7 +39,8 @@ typedef vector<WxDbTable_t> DbTableVector_t;
int WxInitSDK();
int WxDestroySDK();
int WxSetTextMsgCb(const std::function<int(WxMessage_t)> &onMsg);
int WxEnableRecvMsg(const std::function<int(WxMessage_t)> &onMsg);
int WxDisableRecvMsg();
int WxSendTextMsg(wstring wxid, wstring at_wxid, wstring msg);
int WxSendImageMsg(wstring wxid, wstring path);
ContactMap_t WxGetContacts();

5
SDKpy/sdkpy.cpp
View File

@ -6,7 +6,7 @@
namespace py = pybind11;
int WxSetTextMsgCbPy(const std::function<int(WxMessage_t)> &onMsg) { return WxSetTextMsgCb(onMsg); }
int WxEnableRecvMsgPy(const std::function<int(WxMessage_t)> &onMsg) { return WxEnableRecvMsg(onMsg); }
PYBIND11_MODULE(wcferry, m)
{
@ -32,7 +32,8 @@ PYBIND11_MODULE(wcferry, m)
.def_readonly("wxGender", &WxContact::wxGender);
m.def("WxInitSDK", &WxInitSDK);
m.def("WxSetTextMsgCb", &WxSetTextMsgCbPy);
m.def("WxEnableRecvMsg", &WxEnableRecvMsgPy);
m.def("WxDisableRecvMsg", &WxDisableRecvMsg);
m.def("WxSendTextMsg", &WxSendTextMsg);
m.def("WxSendImageMsg", &WxSendImageMsg);
m.def("WxGetContacts", &WxGetContacts, py::return_value_policy::reference);

4
Spy/dllmain.cpp
View File

@ -7,7 +7,7 @@ BOOL APIENTRY DllMain(HMODULE hModule, DWORD ul_reason_for_call, LPVOID lpReserv
switch (ul_reason_for_call) {
case DLL_PROCESS_ATTACH: {
//MessageBox(NULL, L"InitSpy", L"DllMain", 0);
InitSpy(hModule);
InitSpy();
break;
}
case DLL_THREAD_ATTACH:
@ -15,7 +15,7 @@ BOOL APIENTRY DllMain(HMODULE hModule, DWORD ul_reason_for_call, LPVOID lpReserv
break;
case DLL_PROCESS_DETACH: {
//MessageBox(NULL, L"DestroySpy", L"DllMain", 0);
DestroySpy();
DestroySpy(hModule);
break;
}
}

15
Spy/rpc_server.cpp
View File

@ -28,6 +28,7 @@ void server_EnableReceiveMsg()
{
unsigned long ulCode = 0;
ListenMessage();
listenMsgFlag = true;
RpcTryExcept
{
// 调用客户端的回调函数
@ -49,6 +50,12 @@ void server_EnableReceiveMsg()
RpcEndExcept
}
void server_DisableReceiveMsg()
{
UnListenMessage();
listenMsgFlag = false;
}
int server_SendTextMsg(const wchar_t *wxid, const wchar_t *at_wxid, const wchar_t *msg)
{
SendTextMessage(wxid, at_wxid, msg);
@ -177,7 +184,7 @@ RPC_STATUS CALLBACK SecurityCallback(RPC_IF_HANDLE /*hInterface*/, void * /*pBin
return RPC_S_OK; // Always allow anyone.
}
int RpcStartServer(HMODULE hModule)
int RpcStartServer()
{
RPC_STATUS status;
// Uses the protocol combined with the endpoint for receiving
@ -201,7 +208,6 @@ int RpcStartServer(HMODULE hModule)
(unsigned)-1, // Infinite max size of incoming data blocks.
SecurityCallback); // Naive security callback.
listenMsgFlag = true;
while (g_rpcKeepAlive) {
Sleep(1000); // 休眠，释放CPU
}
@ -209,17 +215,18 @@ int RpcStartServer(HMODULE hModule)
return 0;
}
int RpcStopServer(void)
int RpcStopServer()
{
RPC_STATUS status;
UnListenMessage();
listenMsgFlag = false;
g_rpcKeepAlive = false;
status = RpcMgmtStopServerListening(NULL);
if (status)
return status;
status = RpcServerUnregisterIf(NULL, NULL, FALSE);
status = RpcServerUnregisterIf(server_ISpy_v1_0_s_ifspec, NULL, 0);
return status;
}

5
Spy/rpc_server.h
View File

@ -1,5 +1,4 @@
#pragma once
#include "framework.h"
int RpcStartServer(HMODULE hModule);
int RpcStopServer(void);
int RpcStartServer();
int RpcStopServer();

16
Spy/spy.cpp
View File

@ -8,37 +8,37 @@ BOOL g_rpcKeepAlive = false;
WxCalls_t g_WxCalls = { 0 };
DWORD g_WeChatWinDllAddr = 0;
void InitSpy(HMODULE hModule)
void InitSpy()
{
wchar_t version[16] = { 0 };
g_WeChatWinDllAddr = (DWORD)LoadLibrary(L"WeChatWin.dll"); //获取wechatWin模块地址
g_WeChatWinDllAddr = (DWORD)GetModuleHandle(L"WeChatWin.dll"); //获取wechatWin模块地址
if (g_WeChatWinDllAddr == 0) {
MessageBox(NULL, L"获取wechatWin.dll模块地址失败", L"错误", 0);
FreeLibraryAndExitThread(hModule, 0);
return;
}
if (!GetWeChatVersion(version)) { //获取微信版本
MessageBox(NULL, L"获取微信版本失败", L"错误", 0);
FreeLibraryAndExitThread(hModule, 0);
return;
}
if (LoadCalls(version, &g_WxCalls) != 0) { //加载微信版本对应的Call地址
MessageBox(NULL, L"不支持当前版本", L"错误", 0);
FreeLibraryAndExitThread(hModule, 0);
return;
}
g_hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
HANDLE rpcThread = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)RpcStartServer, hModule, NULL, 0);
HANDLE rpcThread = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)RpcStartServer, NULL, NULL, 0);
if (rpcThread != 0) {
CloseHandle(rpcThread);
}
}
void DestroySpy()
void DestroySpy(HMODULE hModule)
{
RpcStopServer();
FreeLibrary((HMODULE)g_WeChatWinDllAddr);
FreeLibraryAndExitThread(hModule, 0);
}
int IsLogin(void) { return (int)GET_DWORD(g_WeChatWinDllAddr + g_WxCalls.login); }

4
Spy/spy.h
View File

@ -2,5 +2,5 @@
#include "framework.h"
void InitSpy(HMODULE hModule);
void DestroySpy();
void InitSpy();
void DestroySpy(HMODULE hModule);