specify pid of wechat for multi-wechat

lich0821 · Oct 30, 2023 · 003d0dc · 003d0dc
1 parent 2213085
commit 003d0dc
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 11 deletions.
diff --git a/WeChatFerry/sdk/sdk.cpp b/WeChatFerry/sdk/sdk.cpp
@@ -33,7 +33,7 @@ static int GetDllPath(bool debug, wchar_t *dllPath)
     return 0;
 }
 
-int WxInitSDK(bool debug, int port)
+int WxInitSDK(bool debug, int port, int inputPid)
 {
     int status  = 0;
     DWORD wcPid = 0;
@@ -43,13 +43,17 @@ int WxInitSDK(bool debug, int port)
         return status;
     }
 
-    status = OpenWeChat(&wcPid);
-    if (status != 0) {
-        MessageBox(NULL, L"打开微信失败", L"WxInitSDK", 0);
-        return status;
-    }
+    if(inputPid != 0) {
+        wcPid = inputPid;
+    } else {
+        status = OpenWeChat(&wcPid);
+        if (status != 0) {
+            MessageBox(NULL, L"打开微信失败", L"WxInitSDK", 0);
+            return status;
+        }
 
-    Sleep(2000); // 等待微信打开
+        Sleep(2000); // 等待微信打开
+    }
     wcProcess = InjectDll(wcPid, spyDllPath, &spyBase);
     if (wcProcess == NULL) {
         MessageBox(NULL, L"注入失败", L"WxInitSDK", 0);

diff --git a/WeChatFerry/sdk/sdk.h b/WeChatFerry/sdk/sdk.h
@@ -1,4 +1,4 @@
 #pragma once
 
-int WxInitSDK(bool debug, int port);
+int WxInitSDK(bool debug, int port, int inputPid);
 int WxDestroySDK();
diff --git a/WeChatFerry/wcf/main.cpp b/WeChatFerry/wcf/main.cpp
@@ -6,7 +6,7 @@
 
 void help()
 {
-    printf("\nUsage: \n启动: wcf.exe start port [debug]\n关闭: wcf.exe stop\nport: 命令端口, 消息端口为命令端口+1\n");
+    printf("\nUsage: \n启动: wcf.exe start port [debug]\n关闭: wcf.exe stop\nport: 命令端口, 消息端口为命令端口+1\n指定微信pid启动: wcf.exe start port pid仅在你知道这是什么时候使用\n");
 }
 
 int main(int argc, char *argv[])
@@ -22,8 +22,11 @@ int main(int argc, char *argv[])
 
     if (strcmp(argv[1], "start") == 0) {
         int port = strtol(argv[2], NULL, 10);
-
-        ret = WxInitSDK(debug, port);
+        int inputPid = 0;
+        if (!debug && argc == 4) {
+            inputPid = strtol(argv[3], NULL, 10)
+        }
+        ret = WxInitSDK(debug, port, inputPid);
     } else if (strcmp(argv[1], "stop") == 0) {
         ret = WxDestroySDK();
     } else {