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

Refactoring

Browse Source
This commit is contained in:
Changhua 2025-02-04 05:27:07 +08:00
parent b426e84241
commit 66e14be136
2 changed files with 29 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
WeChatFerry/spy/send_msg.cpp
View File

@ -10,11 +10,8 @@
#include "user_info.h" #include "user_info.h"
#include "util.h" #include "util.h"
extern HANDLE g_hEvent;
extern QWORD g_WeChatWinDllAddr; extern QWORD g_WeChatWinDllAddr;
#define SRTM_SIZE 0x3F0
#define OS_NEW 0x1B5E140 #define OS_NEW 0x1B5E140
#define OS_FREE 0x1B55850 #define OS_FREE 0x1B55850
#define OS_SEND_MSG_MGR 0x1B53FD0 #define OS_SEND_MSG_MGR 0x1B53FD0
@ -53,6 +50,10 @@ SendMsgManager::SendMsgManager()
func_send_xml = reinterpret_cast<SendXml_t>(g_WeChatWinDllAddr + OS_SEND_XML); func_send_xml = reinterpret_cast<SendXml_t>(g_WeChatWinDllAddr + OS_SEND_XML);
} }
std::unique_ptr<WxString> SendMsgManager::new_wx_string(const char *str)
{
return new_wx_string(str ? std::string(str) : std::string());
}
std::unique_ptr<WxString> SendMsgManager::new_wx_string(const std::string &str) std::unique_ptr<WxString> SendMsgManager::new_wx_string(const std::string &str)
{ {
return std::make_unique<WxString>(String2Wstring(str)); return std::make_unique<WxString>(String2Wstring(str));
@ -108,7 +109,7 @@ void SendMsgManager::send_image(const std::string &wxid, const std::string &path
QWORD pMsg = func_new(reinterpret_cast<QWORD>(&msg)); QWORD pMsg = func_new(reinterpret_cast<QWORD>(&msg));
QWORD sendMgr = func_send_msg_mgr(); QWORD sendMgr = func_send_msg_mgr();
funcSendImage(sendMgr, pMsg, reinterpret_cast<QWORD>(wxWxid.get()), reinterpret_cast<QWORD>(wxPath.get()), func_send_image(sendMgr, pMsg, reinterpret_cast<QWORD>(wxWxid.get()), reinterpret_cast<QWORD>(wxPath.get()),
reinterpret_cast<QWORD>(&flag)); reinterpret_cast<QWORD>(&flag));
func_free(pMsg); func_free(pMsg);
@ -179,8 +180,9 @@ void SendMsgManager::send_emotion(const std::string &wxid, const std::string &pa
reinterpret_cast<QWORD>(buff.get())); reinterpret_cast<QWORD>(buff.get()));
} }
int SendMsgManager::send_rich_text(RichText_t &rt) int SendMsgManager::send_rich_text(const RichText &rt)
{ {
#define SRTM_SIZE 0x3F0
QWORD status = -1; QWORD status = -1;
char *buff = static_cast<char *>(HeapAlloc(GetProcessHeap(), 0, SRTM_SIZE)); char *buff = static_cast<char *>(HeapAlloc(GetProcessHeap(), 0, SRTM_SIZE));
@ -226,7 +228,7 @@ int SendMsgManager::send_pat(const std::string &roomid, const std::string &wxid)
return static_cast<int>(status); return static_cast<int>(status);
} }
int SendMsgManager::forward_message(QWORD msgid, const std::string &receiver) int SendMsgManager::forward(QWORD msgid, const std::string &receiver)
{ {
uint32_t dbIdx = 0; uint32_t dbIdx = 0;
QWORD localId = 0; QWORD localId = 0;
@ -245,15 +247,14 @@ int SendMsgManager::forward_message(QWORD msgid, const std::string &receiver)
} }
// RPC 方法 // RPC 方法
bool SendMsgManager::rpc_send_text(const TextMsg &text, uint8_t *out, size_t *len)
bool SendMsgManager::rpc_send_text(TextMsg &text, uint8_t *out, size_t *len)
{ {
return fill_response<Functions_FUNC_SEND_TXT>(out, len, [&](Response &rsp) { return fill_response<Functions_FUNC_SEND_TXT>(out, len, [&](Response &rsp) {
if (text.msg.empty() || text.receiver.empty()) { if (text.msg == nullptr || text.receiver == nullptr || strlen(text.msg) == 0 || strlen(text.receiver) == 0) {
LOG_ERROR("Empty message or receiver."); LOG_ERROR("Empty message or receiver.");
rsp.msg.status = -1; rsp.msg.status = -1;
} else { } else {
send_text(text.receiver, text.msg, text.aters); send_text(text.receiver, text.msg, text.aters ? text.aters : "");
rsp.msg.status = 0; rsp.msg.status = 0;
} }
}); });
@ -298,14 +299,14 @@ bool SendMsgManager::rpc_send_emotion(const std::string &path, const std::string
}); });
} }
bool SendMsgManager::rpc_send_xml(const XmlMsg &rt, uint8_t *out, size_t *len) bool SendMsgManager::rpc_send_xml(const XmlMsg &xml, uint8_t *out, size_t *len)
{ {
return fill_response<Functions_FUNC_SEND_XML>(out, len, [&](Response &rsp) { return fill_response<Functions_FUNC_SEND_XML>(out, len, [&](Response &rsp) {
if (rt.content.empty() || rt.receiver.empty()) { if (xml.content == nullptr || xml.receiver == nullptr) {
LOG_ERROR("Empty content or receiver."); LOG_ERROR("Empty content or receiver.");
rsp.msg.status = -1; rsp.msg.status = -1;
} else { } else {
send_xml(rt.receiver, rt.content, rt.path, rt.type); send_xml(xml.receiver, xml.content, xml.path, xml.type);
rsp.msg.status = 0; rsp.msg.status = 0;
} }
}); });
@ -314,20 +315,11 @@ bool SendMsgManager::rpc_send_xml(const XmlMsg &rt, uint8_t *out, size_t *len)
bool SendMsgManager::rpc_send_rich_text(const RichText &rt, uint8_t *out, size_t *len) bool SendMsgManager::rpc_send_rich_text(const RichText &rt, uint8_t *out, size_t *len)
{ {
return fill_response<Functions_FUNC_SEND_RICH_TXT>(out, len, [&](Response &rsp) { return fill_response<Functions_FUNC_SEND_RICH_TXT>(out, len, [&](Response &rsp) {
if (rt.receiver.empty()) { if (rt.receiver == nullptr) {
LOG_ERROR("Empty receiver."); LOG_ERROR("Empty receiver.");
rsp.msg.status = -1; rsp.msg.status = -1;
} else { } else {
RichText_t rtt { rsp.msg.status = send_rich_text(rt);
.name = rt.name,
.account = rt.account,
.title = rt.title,
.digest = rt.digest,
.url = rt.url,
.thumburl = rt.thumburl,
.receiver = rt.receiver,
};
rsp.msg.status = send_rich_text(rtt);
} }
}); });
} }

23
WeChatFerry/spy/send_msg.h
View File

@ -7,15 +7,15 @@
#include "spy_types.h" #include "spy_types.h"
#include "wcf.pb.h" #include "wcf.pb.h"
struct RichText_t { // struct RichText_t {
std::string name; // std::string name;
std::string account; // std::string account;
std::string title; // std::string title;
std::string digest; // std::string digest;
std::string url; // std::string url;
std::string thumburl; // std::string thumburl;
std::string receiver; // std::string receiver;
}; // };
class SendMsgManager class SendMsgManager
{ {
@ -27,12 +27,12 @@ public:
void send_file(const std::string &wxid, const std::string &path); void send_file(const std::string &wxid, const std::string &path);
void send_xml(const std::string &receiver, const std::string &xml, const std::string &path, uint64_t type); void send_xml(const std::string &receiver, const std::string &xml, const std::string &path, uint64_t type);
void send_emotion(const std::string &wxid, const std::string &path); void send_emotion(const std::string &wxid, const std::string &path);
int send_rich_text(RichText_t &rt); int send_rich_text(const RichText &rt);
int send_pat(const std::string &roomid, const std::string &wxid); int send_pat(const std::string &roomid, const std::string &wxid);
int forward(uint64_t msgid, const std::string &receiver); int forward(uint64_t msgid, const std::string &receiver);
// RPC 方法 // RPC 方法
bool rpc_send_text(TextMsg &text, uint8_t *out, size_t *len); bool rpc_send_text(const TextMsg &text, uint8_t *out, size_t *len);
bool rpc_send_image(const std::string &path, const std::string &receiver, uint8_t *out, size_t *len); bool rpc_send_image(const std::string &path, const std::string &receiver, uint8_t *out, size_t *len);
bool rpc_send_file(const std::string &path, const std::string &receiver, uint8_t *out, size_t *len); bool rpc_send_file(const std::string &path, const std::string &receiver, uint8_t *out, size_t *len);
bool rpc_send_emotion(const std::string &path, const std::string &receiver, uint8_t *out, size_t *len); bool rpc_send_emotion(const std::string &path, const std::string &receiver, uint8_t *out, size_t *len);
@ -79,6 +79,7 @@ private:
XmlBufSign_t func_xml_buf_sign; XmlBufSign_t func_xml_buf_sign;
SendXml_t func_send_xml; SendXml_t func_send_xml;
std::unique_ptr<WxString> new_wx_string(const char *str);
std::unique_ptr<WxString> new_wx_string(const std::string &str); std::unique_ptr<WxString> new_wx_string(const std::string &str);
std::vector<WxString> parse_wxids(const std::string &wxids); std::vector<WxString> parse_wxids(const std::string &wxids);
}; };