全部博文(263)
分类: WINDOWS
2009-10-14 23:05:44
在编写工具程序以及系统管理程序的时候。常常需要获取某个进程的主窗口以及创建此进程的程序名。获取主窗口的目的是向窗口发送各种消息。获取启动进程的程
序名可以控制对进程的操作。但是有些进程往往有多个主窗口。你要的是哪一个主窗口呢?如果你用过Outlook程序,你就会发现它有多个主窗口,一个窗口
列出收件箱和其它文件夹。如果你打开e-mail,便会有另外一个窗口显示信息。它们都是没有父窗口(或者说宿主窗口)的主窗口。运行一下Spy程序,你
甚至会发现它们的窗口类名都相同:rctrl_renwnd32。资源管理器(Explorer.exe)也有不止一个主窗口。如图一所示,资源管理器有
两个主窗口。一般来讲,想要获取主窗口,凭窗口的式样或类名,你没有什么办法知道哪一个窗口是真正意义上的主窗口。
首先我们讨论如何从多个窗口中获取主窗口?其实很容易。利用两个API函数便可以实现。这两个API是
EnumWindows 和 GetWindowThreadProcessId。如果你对这两个函数不熟悉,不要怕,本文提供了一个C++类来对这两个API进行封装。这个类叫
CMainWindowIterator,用它可以枚举某个进程(已知进程ID)的所有主窗口。这正是我们想要的东西。其使用方法如下:
DWORD pid = // 已知某个进程的ID就这么简单,CMainWindowIterator派生于一个更通用的类:CWindowIterator,CWindowIterator负责 将::EnumWindows函数打包以隐藏回调细节。它有一个虚拟函数OnWindow,你可以在派生类中重写这个函数,从而可以用任何方式来枚举窗 口。CMainWindowIterator就是重写了OnWindow函数,让它只获取属于某个给定进程的主窗口:
CMainWindowIterator itw(pid);
for (HWND hwnd = itw.First(); hwnd; hwnd=itw.Next()) {
// do something
}
// (在构造函数中设置m_pid)这两个类的定义如下:(对应的文件是 EnumProc.h 和 EnumProc.cpp)
BOOL CMainWindowIterator::OnWindow(HWND hwnd)
{
if ((GetWindowLong(hwnd,GWL_STYLE) & WS_VISIBLE)) {
DWORD pidwin;
GetWindowThreadProcessId(hwnd, &pidwin);
if (pidwin==m_pid)
return TRUE;
}
return FALSE;
}
//////////////////////////////////////////////////////图一是用上述C++类编写的一个控制台程序lp.exe的输出画面。最后两栏分别是对应进程的“窗口句柄”和“类名/窗口标题”。其命令行开关“/ct”表示列出窗口类名(c)和窗口标题(t)。
// 这个类主要是封装::EnumWindows,列举顶层窗口
//
class CWindowIterator {
protected:
HWND* m_hwnds; // 隶属于某个进程PID窗口句柄数组
DWORD m_nAlloc; // 数组大小
DWORD m_count; // 找到的窗口句柄数
DWORD m_current; // 当前的窗口句柄
static BOOL CALLBACK EnumProc(HWND hwnd, LPARAM lp);
// 虚拟枚举函数
virtual BOOL OnEnumProc(HWND hwnd);
// 在派生中改写此函数来过滤不同种类的窗口
virtual BOOL OnWindow(HWND hwnd) {
return TRUE;
}
public:
CWindowIterator(DWORD nAlloc=1024);
~CWindowIterator();
DWORD GetCount() { return m_count; }
HWND First();
HWND Next() {
return m_hwnds && m_current}
};
////////////////////////////////////
// 列举某个进程的顶层窗口
//
class CMainWindowIterator : public CWindowIterator {
protected:
DWORD m_pid; // 进程ID
virtual BOOL OnWindow(HWND hwnd);
public:
CMainWindowIterator(DWORD pid, DWORD nAlloc=1024);
~CMainWindowIterator();
};
DWORD count;然后调用GetModuleFileNameEx。
HMODULE hm[1];
EnumProcessModules(hProcess, hm, 1, &count);
CProcessIterator itp;下面是获取创建进程的EXE文件名的方法:
for (DWORD pid=itp.First(); pid; pid=itp.Next()) {
// 处理每一个进程
}
CProcessModuleIterator itm(pid);因为lp显示出来的并不是一个含全路径的模块文件名,所以我用另外一个PSAPI函数GetModuleBaseName来代替 GetModuleFileNameEx从而获取全路径名。此外,由于CProcessModuleIterator自己会打开进程枚举模块,所以不必调 用OpenProcess。用CProcessModuleIterator::GetProcessHandle可以得到已打开进程的句柄。lp程序还 用CMainWindowIterator来显示每个特定进程的所有主窗口。下面是CProcessIterator 和 CProcessModuleIterator的定义:
HMODULE hModule = itm.First(); // .EXE
TCHAR modname[_MAX_PATH];
GetModuleBaseName(itm.GetProcessHandle(), hModule, modname, _MAX_PATH);
////////////////////////////////////////////////////////////////////////////////////////
// 进程列举类 -- 列举出系统中的所有进程,但总是跳过第一个PID=0的进程,即空闲进程(IDLE)
//
class CProcessIterator {
protected:
DWORD* m_pids; // 包含进程IDs的数祖
DWORD m_count; // 数组大小
DWORD m_current; // 当前数组项
public:
CProcessIterator();
~CProcessIterator();
DWORD GetCount() { return m_count; }
DWORD First();
DWORD Next() {
return m_pids && m_current}
};
/////////////////////////////////////////////////////////////////
// 列举某个进程的模块,第一个模块就是创建此进程的主exe程序
//
class CProcessModuleIterator {
protected:
HANDLE m_hProcess; // 进程句柄
HMODULE* m_hModules; // 模块句柄数组
DWORD m_count; // 数组大小
DWORD m_current; // 当前模块的句柄
public:
CProcessModuleIterator(DWORD pid);
~CProcessModuleIterator();
HANDLE GetProcessHandle() { return m_hProcess; }
DWORD GetCount() { return m_count; }
HMODULE First();
HMODULE Next() {
return m_hProcess && m_current < m_count ? m_hModules[m_current++] : 0;
}
};