添加新的支持

ljc545w · Nov 2, 2022 · d3c5e60 · d3c5e60
1 parent 2369f90
commit d3c5e60
Show file tree

Hide file tree

Showing 34 changed files with 941 additions and 116 deletions.
diff --git a/CWeChatRobot/GetMsgCDN.cpp b/CWeChatRobot/GetMsgCDN.cpp
@@ -0,0 +1,28 @@
+#include "pch.h"
+
+wstring GetMsgCND(DWORD pid, ULONG64 msgid)
+{
+    WeChatProcess hp(pid);
+    if (!hp.m_init)
+        return L"";
+    DWORD GetMsgCDNRemoteAddr = hp.GetProcAddr(GetMsgCDNRemote);
+    if (GetMsgCDNRemoteAddr == 0)
+    {
+        return L"";
+    }
+    WeChatData<ULONG64 *> r_msgid(hp.GetHandle(), &msgid, sizeof(ULONG64));
+    if (!r_msgid.GetAddr())
+    {
+        return L"";
+    }
+    DWORD dwRet = CallRemoteFunction(hp.GetHandle(), GetMsgCDNRemoteAddr, r_msgid.GetAddr());
+    if (dwRet == 0)
+        return L"";
+    DWORD get_cdn_resp[2] = {0};
+    ReadProcessMemory(hp.GetHandle(), (LPVOID)dwRet, &get_cdn_resp, sizeof(get_cdn_resp), 0);
+    if (!get_cdn_resp[0] || !get_cdn_resp[1])
+        return L"";
+    unique_ptr<wchar_t[]> path_buf = make_unique<wchar_t[]>(get_cdn_resp[1] + 1);
+    ReadProcessMemory(hp.GetHandle(), (LPVOID)get_cdn_resp[0], path_buf.get(), get_cdn_resp[1] * 2, 0);
+    return wstring(path_buf.get(), get_cdn_resp[1]);
+}
diff --git a/CWeChatRobot/ReceiveMessage.h b/CWeChatRobot/ReceiveMessage.h
@@ -1,9 +1,12 @@
 #pragma once
-#include<windows.h>
-BOOL StartReceiveMessage(DWORD pid,int port);
+#include <windows.h>
+#include <iostream>
+using namespace std;
+BOOL StartReceiveMessage(DWORD pid, int port);
 BOOL StopReceiveMessage(DWORD pid);
 
-BOOL HookImageMsg(DWORD pid,wchar_t* savepath);
-BOOL HookVoiceMsg(DWORD pid,wchar_t* savepath);
+BOOL HookImageMsg(DWORD pid, wchar_t *savepath);
+BOOL HookVoiceMsg(DWORD pid, wchar_t *savepath);
 void UnHookImageMsg(DWORD pid);
-void UnHookVoiceMsg(DWORD pid);
+void UnHookVoiceMsg(DWORD pid);
+wstring GetMsgCND(DWORD pid, ULONG64 msgid);
diff --git a/CWeChatRobot/SendEmotion.cpp b/CWeChatRobot/SendEmotion.cpp
@@ -0,0 +1,32 @@
+#include "pch.h"
+
+struct SendEmotionStruct
+{
+    DWORD wxid;
+    DWORD img_path;
+};
+
+int SendEmotion(DWORD pid, wchar_t *wxid, wchar_t *img_path)
+{
+    WeChatProcess hp(pid);
+    if (!hp.m_init)
+        return 1;
+    DWORD SendEmitonRemoteAddr = hp.GetProcAddr(SendEmotionRemote);
+    if (SendEmitonRemoteAddr == 0)
+    {
+        return 1;
+    }
+    SendEmotionStruct params = {0};
+    WeChatData<wchar_t *> r_wxid(hp.GetHandle(), wxid, TEXTLENGTH(wxid));
+    WeChatData<wchar_t *> r_img_path(hp.GetHandle(), img_path, TEXTLENGTH(img_path));
+
+    params.wxid = (DWORD)r_wxid.GetAddr();
+    params.img_path = (DWORD)r_img_path.GetAddr();
+    WeChatData<SendEmotionStruct *> r_params(hp.GetHandle(), &params, sizeof(params));
+    if (!params.wxid || !params.img_path || !r_params.GetAddr())
+    {
+        return 1;
+    }
+    DWORD dwRet = CallRemoteFunction(hp.GetHandle(), SendEmitonRemoteAddr, r_params.GetAddr());
+    return (dwRet != 1);
+}
diff --git a/CWeChatRobot/SendEmotion.h b/CWeChatRobot/SendEmotion.h
@@ -0,0 +1,4 @@
+#pragma once
+#include <windows.h>
+
+int SendEmotion(DWORD pid, wchar_t *wxid, wchar_t *img_path);
diff --git a/CWeChatRobot/WeChatRobot.cpp b/CWeChatRobot/WeChatRobot.cpp
@@ -626,3 +626,27 @@ STDMETHODIMP CWeChatRobot::CGetTransfer(DWORD pid, BSTR wxid, BSTR transcationid
     *__result = GetTransfer(pid, wxid, transcationid, transferid);
     return S_OK;
 }
+
+/*
+ * 参数0：目标进程pid
+ * 参数1：接收人wxid
+ * 参数2：表情绝对路径
+ * 参数3：预返回的值，调用时无需提供
+ */
+STDMETHODIMP CWeChatRobot::CSendEmotion(DWORD pid, BSTR wxid, BSTR img_path, int *__result)
+{
+    *__result = SendEmotion(pid, wxid, img_path);
+    return S_OK;
+}
+
+/*
+ * 参数0：目标进程pid
+ * 参数1：消息id
+ * 参数2：预返回的值，调用时无需提供
+ */
+STDMETHODIMP CWeChatRobot::CGetMsgCDN(DWORD pid, ULONG64 msgid, BSTR *__result)
+{
+    _bstr_t path = (_bstr_t)GetMsgCND(pid, msgid).c_str();
+    *__result = path;
+    return S_OK;
+}
diff --git a/CWeChatRobot/WeChatRobot.h b/CWeChatRobot/WeChatRobot.h
@@ -91,6 +91,8 @@ class ATL_NO_VTABLE CWeChatRobot : public CComObjectRootEx<CComSingleThreadModel
     STDMETHODIMP CSendXmlMsg(DWORD pid, BSTR wxid, BSTR xml, BSTR imgpath, int *__result);
     STDMETHODIMP CLogout(DWORD pid, int *__result);
     STDMETHODIMP CGetTransfer(DWORD pid, BSTR wxid, BSTR transcationid, BSTR transferid, int *__result);
+    STDMETHODIMP CSendEmotion(DWORD pid, BSTR wxid, BSTR img_path, int *__result);
+    STDMETHODIMP CWeChatRobot::CGetMsgCDN(DWORD pid, ULONG64 msgid, BSTR *__result);
 };
 
 OBJECT_ENTRY_AUTO(__uuidof(WeChatRobot), CWeChatRobot)
diff --git a/CWeChatRobot/WeChatRobotCOM.idl b/CWeChatRobot/WeChatRobotCOM.idl
@@ -66,6 +66,8 @@ interface IWeChatRobot : IDispatch
     [id(51), helpstring("发送xml消息")] HRESULT CSendXmlMsg([in] DWORD pid, [in] BSTR wxid, [in] BSTR xml, [in] BSTR imgpath,[out, retval] int *__result);
     [id(52), helpstring("退出登录")] HRESULT CLogout([in] DWORD pid,[out, retval] int *__result);
     [id(53), helpstring("收款")] HRESULT CGetTransfer([in] DWORD pid, [in] BSTR wxid, [in] BSTR transcationid, [in] BSTR transferid, [out, retval] int *__result);
+    [id(54), helpstring("发送表情")] HRESULT CSendEmotion([in] DWORD pid, [in] BSTR wxid, [in] BSTR img_path, [ out, retval ] int *__result);
+    [id(55), helpstring("获取CDN")] HRESULT CGetMsgCDN([in] DWORD pid, [in] ULONG64 msgid, [ out, retval ] BSTR *__result);
 };
 [
 	object,

diff --git a/CWeChatRobot/WeChatRobotCOM.vcxproj b/CWeChatRobot/WeChatRobotCOM.vcxproj
@@ -253,6 +253,7 @@
     <ClInclude Include="SendArticle.h" />
     <ClInclude Include="SendAtText.h" />
     <ClInclude Include="SendCard.h" />
+    <ClInclude Include="SendEmotion.h" />
     <ClInclude Include="SendFile.h" />
     <ClInclude Include="SendImage.h" />
     <ClInclude Include="SendText.h" />
@@ -289,6 +290,7 @@
     <ClCompile Include="GetChatRoomMembers.cpp" />
     <ClCompile Include="GetDbHandles.cpp" />
     <ClCompile Include="GetHistoryPublicMsg.cpp" />
+    <ClCompile Include="GetMsgCDN.cpp" />
     <ClCompile Include="GetTransfer.cpp" />
     <ClCompile Include="HookImageMessage.cpp" />
     <ClCompile Include="HookVoiceMessage.cpp" />
@@ -311,6 +313,7 @@
     <ClCompile Include="SendArticle.cpp" />
     <ClCompile Include="SendAtText.cpp" />
     <ClCompile Include="SendCard.cpp" />
+    <ClCompile Include="SendEmotion.cpp" />
     <ClCompile Include="SendFile.cpp" />
     <ClCompile Include="SendImage.cpp" />
     <ClCompile Include="SendText.cpp" />

diff --git a/CWeChatRobot/WeChatRobotCOM.vcxproj.filters b/CWeChatRobot/WeChatRobotCOM.vcxproj.filters
@@ -149,6 +149,9 @@
     <Filter Include="未分类\收款">
       <UniqueIdentifier>{79bb1058-041b-4bdc-8678-0369c294e570}</UniqueIdentifier>
     </Filter>
+    <Filter Include="发送消息\发送表情">
+      <UniqueIdentifier>{75c46b26-c4df-4772-862b-839be6a2bf95}</UniqueIdentifier>
+    </Filter>
   </ItemGroup>
   <ItemGroup>
     <ClInclude Include="framework.h">
@@ -298,6 +301,9 @@
     <ClInclude Include="GetTransfer.h">
       <Filter>未分类\收款</Filter>
     </ClInclude>
+    <ClInclude Include="SendEmotion.h">
+      <Filter>发送消息\发送表情</Filter>
+    </ClInclude>
   </ItemGroup>
   <ItemGroup>
     <ClCompile Include="WeChatRobotCOM.cpp">
@@ -447,6 +453,12 @@
     <ClCompile Include="GetTransfer.cpp">
       <Filter>未分类\收款</Filter>
     </ClCompile>
+    <ClCompile Include="SendEmotion.cpp">
+      <Filter>发送消息\发送表情</Filter>
+    </ClCompile>
+    <ClCompile Include="GetMsgCDN.cpp">
+      <Filter>接收消息</Filter>
+    </ClCompile>
   </ItemGroup>
   <ItemGroup>
     <ResourceCompile Include="WeChatRobotCOM.rc">

diff --git a/CWeChatRobot/WeChatRobotCOM_i.h b/CWeChatRobot/WeChatRobotCOM_i.h
@@ -378,6 +378,17 @@ EXTERN_C const IID IID_IWeChatRobot;
             /* [in] */ BSTR transferid,
             /* [retval][out] */ int *__result) = 0;
 
+        virtual /* [helpstring][id] */ HRESULT STDMETHODCALLTYPE CSendEmotion( 
+            /* [in] */ DWORD pid,
+            /* [in] */ BSTR wxid,
+            /* [in] */ BSTR img_path,
+            /* [retval][out] */ int *__result) = 0;
+
+        virtual /* [helpstring][id] */ HRESULT STDMETHODCALLTYPE CGetMsgCDN( 
+            /* [in] */ DWORD pid,
+            /* [in] */ ULONG64 msgid,
+            /* [retval][out] */ BSTR *__result) = 0;
+
     };
 
 
@@ -750,6 +761,19 @@ EXTERN_C const IID IID_IWeChatRobot;
             /* [in] */ BSTR transferid,
             /* [retval][out] */ int *__result);
 
+        /* [helpstring][id] */ HRESULT ( STDMETHODCALLTYPE *CSendEmotion )( 
+            IWeChatRobot * This,
+            /* [in] */ DWORD pid,
+            /* [in] */ BSTR wxid,
+            /* [in] */ BSTR img_path,
+            /* [retval][out] */ int *__result);
+
+        /* [helpstring][id] */ HRESULT ( STDMETHODCALLTYPE *CGetMsgCDN )( 
+            IWeChatRobot * This,
+            /* [in] */ DWORD pid,
+            /* [in] */ ULONG64 msgid,
+            /* [retval][out] */ BSTR *__result);
+
         END_INTERFACE
     } IWeChatRobotVtbl;
 
@@ -936,6 +960,12 @@ EXTERN_C const IID IID_IWeChatRobot;
 #define IWeChatRobot_CGetTransfer(This,pid,wxid,transcationid,transferid,__result)	\
     ( (This)->lpVtbl -> CGetTransfer(This,pid,wxid,transcationid,transferid,__result) ) 
 
+#define IWeChatRobot_CSendEmotion(This,pid,wxid,img_path,__result)	\
+    ( (This)->lpVtbl -> CSendEmotion(This,pid,wxid,img_path,__result) ) 
+
+#define IWeChatRobot_CGetMsgCDN(This,pid,msgid,__result)	\
+    ( (This)->lpVtbl -> CGetMsgCDN(This,pid,msgid,__result) ) 
+
 #endif /* COBJMACROS */