C++内嵌第三方EXE程序窗口

刺骨的言语ヽ痛彻心扉 2022-08-08 17:57 330阅读 0赞

# 内嵌第三方EXE程序窗口 #

近日，在做一个将一个第三方应用窗口嵌入自己的程序窗口内部的功能，经过几次摸索，终于能比较好的嵌入进去了，为方便叙述，以下称需要被嵌入应用窗口为目标窗口，承载嵌入应用窗口的为宿主窗口，目标窗口为任意第三方应用，宿主窗口为自己开发。

将目标窗口完美嵌入，关键在于将目标窗口做为宿主窗口的子窗口，这样[操作系统][Link 1]就会帮我们完成移动，绘制等操作，为目标窗口设置父窗口，也很简单，调用一个API即可:

[HWND][] [ SetParent][SetParent]( HWND hWndChild, HWND hWndNewParent );

第一个参数是子窗口的句柄，也即目标窗口的句柄，第二个窗口为我们想设置的目标窗口的父窗口，即宿主窗口，由于宿主窗口的程序为自己开发，要拿到这个句柄就很简单，关键在于如何找到目标窗口的句柄。

找到目标窗口的句柄，也有一个API：

HWND [ FindWindow][FindWindow]( LPCTSTR lpClassName, LPCTSTR lpWindowName );

第一个参数为窗口的类名，创建窗口之前，首先需要使用RegisterClass向系统注册一个窗口类WNDCLASS，这个结构体中有个lpszClassName字段，就代表这个窗口类的名称。第二个参数为目标窗口的标题。具体这两个值多少，可以使用vs自带的spy++工具，查看下目标窗口，就能获得。

下面分步骤，结合代码，详细展示下整体流程。

### 1.首先准备好宿主窗口的窗体，并且使其透明。 ###

这个可以通过响应WM\_CTLCOLOR消息，并返回一个透明画刷完成，代码很简单如下:

HBRUSH CDlgMainIntelligent::OnCtlColor(CDC\* pDC, CWnd\* pWnd, UINT nCtlColor) \{ CDetachDlg::OnCtlColor(pDC, pWnd, nCtlColor); pDC->SetBkMode(TRANSPARENT); return (HBRUSH) [GetStockObject][](NULL\_BRUSH); \}

### 2.拿到第三方应用的窗口句柄 ###

这里又分为两种情景，一种是第三方应用尚未启动，需要我们自己去启动，并等待其启动完成后，查找其窗口句柄，另一种情形是第三方应用已启动，这样可以直接去查找窗口句柄。如何区分这两种情形，很简单，先去查找下窗口，如果能找到就证明已启动，否则就去启动下。

启动应用进程可以使用[CreateProcess][]，这里有个问题，如何判断这个进程已经启动好了呢？可以使用

DWORD WINAPI WaitForInputIdle( \_\_in [ HANDLE hProcess][HANDLE hProcess], \_\_in DWORD dwMilliseconds );

不过这个仅仅是判断进程启动好，有可能主窗口还没创建完成，也有可能第三方程序崩了，压根拿不到。试过几种方法，都不太理想，我采用了最简单有效的方法，就是循环探测等待，并设置超时时间，具体看代码，我封装了一个函数，可以拿到第三方进程的句柄及主窗口句柄。

HWND GetIVAWindowAndProcess(HANDLE& hProcess) \{ HWND hWnd = NULL; hWnd = ::FindWindow("XXXXClient3",NULL); //是否能找到，找不到就去启动下 if(hWnd == NULL) \{ int nNumberDely = 10; [ STARTUPINFO][STARTUPINFO] si; PROCESS\_INFORMATION pi; [ ZeroMemory][ZeroMemory]( &si, sizeof(si) ); si.cb = sizeof(si); [ ZeroMemory][ZeroMemory]( &pi, sizeof(pi) ); // Start the child process. if( ! [CreateProcess][]( NULL, // No module name (use command [line][]) "XXX\\\\XXX. [exe][]", // Command [line][] NULL, // Process handle not inheritable NULL, // Thread handle not inheritable FALSE, // Set handle inheritance to FALSE 0, // No creation flags NULL, // Use parent's environment block NULL, // Use parent's starting directory &si, // Pointer to [ STARTUPINFO][STARTUPINFO] structure &pi ) // Pointer to PROCESS\_INFORMATION structure ) \{ AfxMessageBox(\_T("启动XXX进程失败！")); return NULL; \} [ CloseHandle][CloseHandle](pi.hThread); WaitForInputIdle(pi.hProcess,INFINITE); hProcess = pi.hProcess; while((hWnd = :: [FindWindow][]("XXXXClient3",NULL)) == NULL && nNumberDely > 0) \{ Sleep(500); nNumberDely --; \} \} else \{ DWORD dwProcessId = 0; DWORD dwThreadId = 0; dwThreadId = GetWindowThreadProcessId(hWnd,&dwProcessId); hProcess = OpenProcess(PROCESS\_ALL\_ACCESS,NULL,dwProcessId); \} return [hWnd][HWND]; <span style="font-size:14px;">\}</span>

通过以上函数，就能拿到句柄了。接下来就可以设置其父窗口了。

### 3.处理窗口WM\_SIZE消息 ###

前面说过，将第三方窗口设置为子窗口，移动绘制这些我们不用管了，系统会帮忙完成，不过WM\_SIZE是需要我们处理的，有了窗口句柄，这个也变得很简单。只要在OnSize里面调用[MoveWindow][]即可

void CDlgMainIntelligent::OnSize(UINT nType, int cx, int cy) \{ CDetachDlg::OnSize(nType, cx, cy); if( [GetSafeHwnd][] == NULL) \{ return ; \} if(! [IsWindow][](GetSafeHwnd)) \{ return; \} if(!m\_bIsInit) \{ return ; \} CRect rect; GetClientRect(&rect); if(m\_hIVAWnd != NULL && !m\_bIsAttached) \{ :: [SetParent][](m\_hIVAWnd, [GetSafeHwnd][]); m\_bIsAttached = TRUE; \} if(m\_hIVAWnd) \{ ::PostMessage(m\_hIVAWnd,WM\_SYSCOMMAND,SC\_MAXIMIZE,NULL); :: [MoveWindow][](m\_hIVAWnd,rect.left,rect.top,rect.Width,rect.Height,TRUE); \} \}

### 4.退出时，将第三方程序关掉 ###

好的体验就是，目标窗口完全做为宿主窗口的一部分，同生共死，因此当宿主窗口销毁的时候，目标窗口也要一并销毁。这个也很简单，只要在宿主窗口的OnCLose中做处理即可。

void CDlgMainIntelligent::OnClose \{ // TODO: Add your message handler code here and/or call default if(m\_hIVAWnd!= NULL) \{ ::PostMessage(m\_hIVAWnd,WM\_CLOSE,NULL,NULL); m\_hIVAWnd = NULL; \} if(m\_hIVAProcess != NULL) \{ ::TerminateProcess(m\_hIVAProcess,0); [ CloseHandle][CloseHandle](m\_hIVAProcess); m\_hIVAProcess = NULL; \} CDetachDlg::OnClose; \} 经过这样处理，基本能完美嵌入第三方应用的主窗口了。

另外还有一些细节上的体验，一般第三方应用窗口，也是有自己的最大最小化按钮的，需要把这个屏蔽掉，否则看起来不美观。我使用的方法是，向第三方应用注入一个DLL，在该DLL中创建一个小窗口，把应用窗口的右上角遮挡住。这样界面看起来就非常统一和协调了。

注入DLL的代码就是windows核心编程上的，没什么好说的，如下:BOOL CDlgMainIntelligent::InjectHookDLL( [HANDLE hProcess][]) \{ if(hProcess == NULL) \{ return FALSE; \} PCWSTR pszLibFile = L"MsgHookDLL.Dll"; if(hProcess != NULL) \{ int cch = 1 + lstrlenW(pszLibFile) ; int cb = cch \* sizeof(wchar\_t); PWSTR PszLibFileRemote = (PWSTR) VirtualAllocEx(hProcess,NULL,cb,MEM\_COMMIT,PAGE\_READWRITE); WriteProcessMemory(hProcess,PszLibFileRemote,(PVOID)pszLibFile,cb,NULL); PTHREAD\_START\_ROUTINE pfnThreadRtn = (PTHREAD\_START\_ROUTINE) GetProcAddress( [GetModuleHandle][](\_T("Kernel32")),"LoadLibraryW"); HANDLE hThread = CreateRemoteThread(hProcess,NULL,0,pfnThreadRtn,PszLibFileRemote,0,NULL); WaitForSingleObject(hThread,INFINITE); return TRUE; \} return FALSE; \}

[Link 1]: http://www.yidianzixun.com/m/channel/keyword/%E6%93%8D%E4%BD%9C%E7%B3%BB%E7%BB%9F
[HWND]: http://www.yidianzixun.com/m/channel/keyword/hwnd
[SetParent]: http://www.yidianzixun.com/m/channel/keyword/setparent
[FindWindow]: http://www.yidianzixun.com/m/channel/keyword/findwindow
[GetStockObject]: http://www.yidianzixun.com/m/channel/keyword/getstockobject
[CreateProcess]: http://www.yidianzixun.com/m/channel/keyword/createprocess
[HANDLE hProcess]: http://www.yidianzixun.com/m/channel/keyword/handle%20hprocess
[STARTUPINFO]: http://www.yidianzixun.com/m/channel/keyword/startupinfo
[ZeroMemory]: http://www.yidianzixun.com/m/channel/keyword/zeromemory
[line]: http://www.yidianzixun.com/m/channel/keyword/line
[exe]: http://www.yidianzixun.com/m/channel/keyword/exe
[CloseHandle]: http://www.yidianzixun.com/m/channel/keyword/closehandle
[MoveWindow]: http://www.yidianzixun.com/m/channel/keyword/movewindow
[GetSafeHwnd]: http://www.yidianzixun.com/m/channel/keyword/getsafehwnd
[IsWindow]: http://www.yidianzixun.com/m/channel/keyword/iswindow
[GetModuleHandle]: http://www.yidianzixun.com/m/channel/keyword/getmodulehandle