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 2023-12-06 22:06:16 +08:00
parent d0b8cc2305
commit 1001e9b429
2 changed files with 112 additions and 93 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
WeChatFerry/spy/chatroom_mgmt.cpp
View File

@ -19,31 +19,31 @@ int AddChatroomMember(string roomid, string wxids)
return -1;
}
int rv = 0;
DWORD addRoomMemberCall1 = g_WeChatWinDllAddr + g_WxCalls.arm.call1;
DWORD addRoomMemberCall2 = g_WeChatWinDllAddr + g_WxCalls.arm.call2;
DWORD addRoomMemberCall3 = g_WeChatWinDllAddr + g_WxCalls.arm.call3;
int rv = 0;
DWORD armCall1 = g_WeChatWinDllAddr + g_WxCalls.arm.call1;
DWORD armCall2 = g_WeChatWinDllAddr + g_WxCalls.arm.call2;
DWORD armCall3 = g_WeChatWinDllAddr + g_WxCalls.arm.call3;
DWORD temp = 0;
wstring wsRoomid = String2Wstring(roomid);
WxString txtRoomid(wsRoomid);
WxString wxRoomid(wsRoomid);
vector<wstring> vMembers;
vector<WxString> vTxtMembers;
vector<WxString> vWxMembers;
wstringstream wss(String2Wstring(wxids));
while (wss.good()) {
wstring wstr;
getline(wss, wstr, L',');
vMembers.push_back(wstr);
WxString txtMember(vMembers.back());
vTxtMembers.push_back(txtMember);
vWxMembers.push_back(txtMember);
}
LOG_DEBUG("Adding {} members[{}] to {}", vTxtMembers.size(), wxids.c_str(), roomid.c_str());
LOG_DEBUG("Adding {} members[{}] to {}", vWxMembers.size(), wxids.c_str(), roomid.c_str());
__asm {
pushad;
pushfd;
call addRoomMemberCall1;
call armCall1;
sub esp, 0x8;
mov temp, eax;
mov ecx, esp;
@ -52,13 +52,13 @@ int AddChatroomMember(string roomid, string wxids)
test esi, esi;
sub esp, 0x14;
mov ecx, esp;
lea eax, txtRoomid;
lea eax, wxRoomid;
push eax;
call addRoomMemberCall2;
call armCall2;
mov ecx, temp;
lea eax, vTxtMembers;
lea eax, vWxMembers;
push eax;
call addRoomMemberCall3;
call armCall3;
mov rv, eax;
popfd;
popad;
@ -73,41 +73,41 @@ int DelChatroomMember(string roomid, string wxids)
return -1;
}
int rv = 0;
DWORD delRoomMemberCall1 = g_WeChatWinDllAddr + g_WxCalls.drm.call1;
DWORD delRoomMemberCall2 = g_WeChatWinDllAddr + g_WxCalls.drm.call2;
DWORD delRoomMemberCall3 = g_WeChatWinDllAddr + g_WxCalls.drm.call3;
int rv = 0;
DWORD drmCall1 = g_WeChatWinDllAddr + g_WxCalls.drm.call1;
DWORD drmCall2 = g_WeChatWinDllAddr + g_WxCalls.drm.call2;
DWORD drmCall3 = g_WeChatWinDllAddr + g_WxCalls.drm.call3;
DWORD temp = 0;
wstring wsRoomid = String2Wstring(roomid);
WxString txtRoomid(wsRoomid);
WxString wxRoomid(wsRoomid);
vector<wstring> vMembers;
vector<WxString> vTxtMembers;
vector<WxString> vWxMembers;
wstringstream wss(String2Wstring(wxids));
while (wss.good()) {
wstring wstr;
getline(wss, wstr, L',');
vMembers.push_back(wstr);
WxString txtMember(vMembers.back());
vTxtMembers.push_back(txtMember);
vWxMembers.push_back(txtMember);
}
LOG_DEBUG("Adding {} members[{}] to {}", vTxtMembers.size(), wxids.c_str(), roomid.c_str());
LOG_DEBUG("Adding {} members[{}] to {}", vWxMembers.size(), wxids.c_str(), roomid.c_str());
__asm {
pushad;
pushfd;
call delRoomMemberCall1;
call drmCall1;
sub esp, 0x14;
mov esi, eax;
mov ecx, esp;
lea edi, txtRoomid;
lea edi, wxRoomid;
push edi;
call delRoomMemberCall2;
call drmCall2;
mov ecx, esi;
lea eax, vTxtMembers;
lea eax, vWxMembers;
push eax;
call delRoomMemberCall3;
call drmCall3;
mov rv, eax;
popfd;
popad;

153
WeChatFerry/spy/rpc_server.cpp
View File

@ -69,11 +69,12 @@ bool func_is_login(uint8_t *out, size_t *len)
bool func_get_self_wxid(uint8_t *out, size_t *len)
{
string wxid = GetSelfWxid();
Response rsp = Response_init_default;
rsp.func = Functions_FUNC_GET_SELF_WXID;
rsp.which_msg = Response_str_tag;
rsp.msg.str = (char *)wxid.c_str();
string wxid = GetSelfWxid();
rsp.msg.str = (char *)wxid.c_str();
pb_ostream_t stream = pb_ostream_from_buffer(out, *len);
if (!pb_encode(&stream, Response_fields, &rsp)) {
@ -192,15 +193,16 @@ bool func_get_audio_msg(uint64_t id, char *dir, uint8_t *out, size_t *len)
Response rsp = Response_init_default;
rsp.func = Functions_FUNC_GET_AUDIO_MSG;
rsp.which_msg = Response_str_tag;
string path = "";
string path = string(dir ? dir : "");
if (path.empty()) {
if (dir == NULL) {
LOG_ERROR("Empty dir.");
rsp.msg.str = (char *)"";
} else {
rsp.msg.str = (char *)GetAudio(id, dir).c_str();
path = GetAudio(id, dir);
}
rsp.msg.str = (char *)path.c_str();
pb_ostream_t stream = pb_ostream_from_buffer(out, *len);
if (!pb_encode(&stream, Response_fields, &rsp)) {
LOG_ERROR("Encoding failed: {}", PB_GET_ERROR(&stream));
@ -213,12 +215,12 @@ bool func_get_audio_msg(uint64_t id, char *dir, uint8_t *out, size_t *len)
bool func_send_txt(TextMsg txt, uint8_t *out, size_t *len)
{
Response rsp = Response_init_default;
rsp.func = Functions_FUNC_SEND_TXT;
rsp.which_msg = Response_status_tag;
rsp.msg.status = 0;
Response rsp = Response_init_default;
rsp.func = Functions_FUNC_SEND_TXT;
rsp.which_msg = Response_status_tag;
if ((txt.msg == NULL) || (txt.receiver == NULL)) {
LOG_ERROR("Empty message or receiver.");
rsp.msg.status = -1; // Empty message or empty receiver
} else {
string msg(txt.msg);
@ -226,6 +228,7 @@ bool func_send_txt(TextMsg txt, uint8_t *out, size_t *len)
string aters(txt.aters ? txt.aters : "");
SendTextMessage(receiver, msg, aters);
rsp.msg.status = 0;
}
pb_ostream_t stream = pb_ostream_from_buffer(out, *len);
@ -240,15 +243,16 @@ bool func_send_txt(TextMsg txt, uint8_t *out, size_t *len)
bool func_send_img(char *path, char *receiver, uint8_t *out, size_t *len)
{
Response rsp = Response_init_default;
rsp.func = Functions_FUNC_SEND_IMG;
rsp.which_msg = Response_status_tag;
rsp.msg.status = 0;
Response rsp = Response_init_default;
rsp.func = Functions_FUNC_SEND_IMG;
rsp.which_msg = Response_status_tag;
if ((path == NULL) || (receiver == NULL)) {
LOG_ERROR("Empty path or receiver.");
rsp.msg.status = -1;
} else {
SendImageMessage(receiver, path);
rsp.msg.status = 0;
}
pb_ostream_t stream = pb_ostream_from_buffer(out, *len);
@ -263,15 +267,16 @@ bool func_send_img(char *path, char *receiver, uint8_t *out, size_t *len)
bool func_send_file(char *path, char *receiver, uint8_t *out, size_t *len)
{
Response rsp = Response_init_default;
rsp.func = Functions_FUNC_SEND_FILE;
rsp.which_msg = Response_status_tag;
rsp.msg.status = 0;
Response rsp = Response_init_default;
rsp.func = Functions_FUNC_SEND_FILE;
rsp.which_msg = Response_status_tag;
if ((path == NULL) || (receiver == NULL)) {
LOG_ERROR("Empty path or receiver.");
rsp.msg.status = -1;
} else {
SendImageMessage(receiver, path);
rsp.msg.status = 0;
}
pb_ostream_t stream = pb_ostream_from_buffer(out, *len);
@ -286,12 +291,12 @@ bool func_send_file(char *path, char *receiver, uint8_t *out, size_t *len)
bool func_send_xml(XmlMsg xml, uint8_t *out, size_t *len)
{
Response rsp = Response_init_default;
rsp.func = Functions_FUNC_SEND_XML;
rsp.which_msg = Response_status_tag;
rsp.msg.status = 0;
Response rsp = Response_init_default;
rsp.func = Functions_FUNC_SEND_XML;
rsp.which_msg = Response_status_tag;
if ((xml.content == NULL) || (xml.receiver == NULL)) {
LOG_ERROR("Empty content or receiver.");
rsp.msg.status = -1;
} else {
string receiver(xml.receiver);
@ -299,6 +304,7 @@ bool func_send_xml(XmlMsg xml, uint8_t *out, size_t *len)
string path(xml.path ? xml.path : "");
uint32_t type = (uint32_t)xml.type;
SendXmlMessage(receiver, content, path, type);
rsp.msg.status = 0;
}
pb_ostream_t stream = pb_ostream_from_buffer(out, *len);
@ -313,15 +319,16 @@ bool func_send_xml(XmlMsg xml, uint8_t *out, size_t *len)
bool func_send_emotion(char *path, char *receiver, uint8_t *out, size_t *len)
{
Response rsp = Response_init_default;
rsp.func = Functions_FUNC_SEND_EMOTION;
rsp.which_msg = Response_status_tag;
rsp.msg.status = 0;
Response rsp = Response_init_default;
rsp.func = Functions_FUNC_SEND_EMOTION;
rsp.which_msg = Response_status_tag;
if ((path == NULL) || (receiver == NULL)) {
LOG_ERROR("Empty path or receiver.");
rsp.msg.status = -1;
} else {
SendEmotionMessage(receiver, path);
rsp.msg.status = 0;
}
pb_ostream_t stream = pb_ostream_from_buffer(out, *len);
@ -336,21 +343,26 @@ bool func_send_emotion(char *path, char *receiver, uint8_t *out, size_t *len)
bool func_send_rich_txt(RichText rt, uint8_t *out, size_t *len)
{
Response rsp = Response_init_default;
rsp.func = Functions_FUNC_SEND_RICH_TXT;
rsp.which_msg = Response_status_tag;
rsp.msg.status = 0;
Response rsp = Response_init_default;
rsp.func = Functions_FUNC_SEND_RICH_TXT;
rsp.which_msg = Response_status_tag;
RichText_t rtt;
rtt.account = string(rt.account ? rt.account : "");
rtt.digest = string(rt.digest ? rt.digest : "");
rtt.name = string(rt.name ? rt.name : "");
rtt.receiver = string(rt.receiver ? rt.receiver : "");
rtt.thumburl = string(rt.thumburl ? rt.thumburl : "");
rtt.title = string(rt.title ? rt.title : "");
rtt.url = string(rt.url ? rt.url : "");
if (rt.receiver == NULL) {
LOG_ERROR("Empty receiver.");
rsp.msg.status = -1;
} else {
RichText_t rtt;
rtt.account = string(rt.account ? rt.account : "");
rtt.digest = string(rt.digest ? rt.digest : "");
rtt.name = string(rt.name ? rt.name : "");
rtt.receiver = string(rt.receiver ? rt.receiver : "");
rtt.thumburl = string(rt.thumburl ? rt.thumburl : "");
rtt.title = string(rt.title ? rt.title : "");
rtt.url = string(rt.url ? rt.url : "");
rsp.msg.status = SendRichTextMessage(rtt);
}
rsp.msg.status = SendRichTextMessage(rtt);
pb_ostream_t stream = pb_ostream_from_buffer(out, *len);
if (!pb_encode(&stream, Response_fields, &rsp)) {
LOG_ERROR("Encoding failed: {}", PB_GET_ERROR(&stream));
@ -368,6 +380,7 @@ bool func_send_pat_msg(char *roomid, char *wxid, uint8_t *out, size_t *len)
rsp.which_msg = Response_status_tag;
if ((roomid == NULL) || (wxid == NULL)) {
LOG_ERROR("Empty roomid or wxid.");
rsp.msg.status = -1;
} else {
rsp.msg.status = SendPatMessage(roomid, wxid);
@ -502,8 +515,14 @@ bool func_exec_db_query(char *db, char *sql, uint8_t *out, size_t *len)
Response rsp = Response_init_default;
rsp.func = Functions_FUNC_EXEC_DB_QUERY;
rsp.which_msg = Response_rows_tag;
DbRows_t rows;
if ((db == NULL) || (sql == NULL)) {
LOG_ERROR("Empty db or sql.");
} else {
rows = ExecDbQuery(db, sql);
}
DbRows_t rows = ExecDbQuery(db, sql);
rsp.msg.rows.rows.arg = &rows;
rsp.msg.rows.rows.funcs.encode = encode_rows;
@ -519,19 +538,15 @@ bool func_exec_db_query(char *db, char *sql, uint8_t *out, size_t *len)
bool func_accept_friend(char *v3, char *v4, int32_t scene, uint8_t *out, size_t *len)
{
Response rsp = Response_init_default;
rsp.func = Functions_FUNC_ACCEPT_FRIEND;
rsp.which_msg = Response_status_tag;
rsp.msg.status = 0;
Response rsp = Response_init_default;
rsp.func = Functions_FUNC_ACCEPT_FRIEND;
rsp.which_msg = Response_status_tag;
if ((v3 == NULL) || (v4 == NULL)) {
rsp.msg.status = -1;
LOG_ERROR("Empty V3 or V4.");
} else {
rsp.msg.status = AcceptNewFriend(v3, v4, scene);
if (rsp.msg.status != 1) {
LOG_ERROR("AcceptNewFriend failed: {}", rsp.msg.status);
}
}
pb_ostream_t stream = pb_ostream_from_buffer(out, *len);
@ -546,14 +561,15 @@ bool func_accept_friend(char *v3, char *v4, int32_t scene, uint8_t *out, size_t
bool func_receive_transfer(char *wxid, char *tfid, char *taid, uint8_t *out, size_t *len)
{
Response rsp = Response_init_default;
rsp.func = Functions_FUNC_RECV_TRANSFER;
rsp.which_msg = Response_status_tag;
rsp.msg.status = 0;
Response rsp = Response_init_default;
rsp.func = Functions_FUNC_RECV_TRANSFER;
rsp.which_msg = Response_status_tag;
rsp.msg.status = ReceiveTransfer(wxid, tfid, taid);
if (rsp.msg.status != 1) {
LOG_ERROR("AddChatroomMember failed: {}", rsp.msg.status);
if ((wxid == NULL) || (tfid == NULL) || (taid == NULL)) {
rsp.msg.status = -1;
LOG_ERROR("Empty wxid, tfid or taid.");
} else {
rsp.msg.status = ReceiveTransfer(wxid, tfid, taid);
}
pb_ostream_t stream = pb_ostream_from_buffer(out, *len);
@ -652,17 +668,16 @@ bool func_decrypt_image(DecPath dec, uint8_t *out, size_t *len)
Response rsp = Response_init_default;
rsp.func = Functions_FUNC_DECRYPT_IMAGE;
rsp.which_msg = Response_str_tag;
string path = "";
string src = string(dec.src ? dec.src : "");
string dst = string(dec.dst ? dec.dst : "");
if (src.empty()) {
LOG_ERROR("Empty src path.");
rsp.msg.str = (char *)"";
if ((dec.src == NULL) || (dec.dst == NULL)) {
LOG_ERROR("Empty src or dst.");
} else {
string path = DecryptImage(src, dst);
rsp.msg.str = (char *)path.c_str();
path = DecryptImage(dec.src, dec.dst);
}
rsp.msg.str = (char *)path.c_str();
pb_ostream_t stream = pb_ostream_from_buffer(out, *len);
if (!pb_encode(&stream, Response_fields, &rsp)) {
LOG_ERROR("Encoding failed: {}", PB_GET_ERROR(&stream));
@ -680,9 +695,11 @@ bool func_add_room_members(char *roomid, char *wxids, uint8_t *out, size_t *len)
rsp.which_msg = Response_status_tag;
rsp.msg.status = 0;
rsp.msg.status = AddChatroomMember(roomid, wxids);
if (rsp.msg.status != 1) {
LOG_ERROR("AddChatroomMember failed: {}", rsp.msg.status);
if ((roomid == NULL) || (wxids == NULL)) {
LOG_ERROR("Empty roomid or wxids.");
rsp.msg.status = -1;
} else {
rsp.msg.status = AddChatroomMember(roomid, wxids);
}
pb_ostream_t stream = pb_ostream_from_buffer(out, *len);
@ -702,9 +719,11 @@ bool func_del_room_members(char *roomid, char *wxids, uint8_t *out, size_t *len)
rsp.which_msg = Response_status_tag;
rsp.msg.status = 0;
rsp.msg.status = DelChatroomMember(roomid, wxids);
if (rsp.msg.status != 1) {
LOG_ERROR("DelChatroomMember failed: {}", rsp.msg.status);
if ((roomid == NULL) || (wxids == NULL)) {
LOG_ERROR("Empty roomid or wxids.");
rsp.msg.status = -1;
} else {
rsp.msg.status = DelChatroomMember(roomid, wxids);
}
pb_ostream_t stream = pb_ostream_from_buffer(out, *len);