vc++基础班[27]---实现一个简单的任务管理器
因为任务管理器中涉及到进程的枚举操作，所以把两节课的知识点合并到一起来讲！
①、设计界面、以及列表控件变量的绑定；
②、列表控件样式的指定：
m_TaskList.SetExtendedStyle(LVS_EX_GRIDLINES | LVS_EX_FULLROWSELECT);
m_TaskList.InsertColumn(0, _T(" 进程名称"), 0, 100);
m_TaskList.InsertColumn(1, _T(" PID"), 0, 60);
m_TaskList.InsertColumn(2, _T(" 文件路径"), 0, 380);
③、进程列表的获取：
具体的方法也是多种的，具体见：http://www.cctry.com/thread-11857-1-1.html
这里面我们使用第一种方法给大家讲解：
#include <Tlhelp32.h>
#include <Psapi.h>
#pragma comment(lib, "Psapi.lib")
BOOL CMFCTestDlg::GetProcessList()
{
BOOL bResult = FALSE;
HANDLE hSnap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
if ( hSnap == INVALID_HANDLE_VALUE )
return FALSE;
int idx = 0;
CString strID;
HANDLE hProcess = NULL;
PROCESSENTRY32 info = {0};
info.dwSize = sizeof(PROCESSENTRY32);
BOOL bRet = Process32First(hSnap, &info);
while(bRet) {
idx = m_TaskList.InsertItem(m_TaskList.GetItemCount(), _T(""));
m_TaskList.SetItemText(idx, 0, info.szExeFile);
strID.Empty();
strID.Format(_T("%d"), info.th32ProcessID);
m_TaskList.SetItemText(idx, 1, strID);
hProcess = OpenProcess(PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, FALSE, info.th32ProcessID);
if (hProcess) {
TCHAR szPath[MAX_PATH] = {0};
GetModuleFileNameEx(hProcess, NULL, szPath, MAX_PATH);
m_TaskList.SetItemText(idx, 2, szPath);
}
bRet = Process32Next(hSnap, &info);
}
return TRUE;
}
④、Debug 权限提升函数：
BOOL AdjustPrivileges()
{
HANDLE hToken = NULL;
TOKEN_PRIVILEGES tp = {0};
TOKEN_PRIVILEGES oldtp = {0};
DWORD dwSize = sizeof(TOKEN_PRIVILEGES);
LUID luid = {0};
if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken)) {
if (GetLastError()==ERROR_CALL_NOT_IMPLEMENTED)
return TRUE;
else
return FALSE;
}
if (!LookupPrivilegeValue(NULL, SE_DEBUG_NAME, &luid)) {
CloseHandle(hToken);
return FALSE;
}
tp.PrivilegeCount=1;
tp.Privileges[0].Luid = luid;
tp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
/* Adjust Token Privileges */
if (!AdjustTokenPrivileges(hToken, FALSE, &tp, sizeof(TOKEN_PRIVILEGES), &oldtp, &dwSize)) {
CloseHandle(hToken);
return FALSE;
}
// close handles
CloseHandle(hToken);
return TRUE;
}
⑤、菜单资源的添加及编辑：
刷新列表：ID_M_REFRESH_LIST
结束此进程：ID_M_EDN_TASK
定位文件：ID_M_FIND_EXE
拷贝路径：ID_M_COPY_PATH
⑥、菜单资源的弹出：
void CMFCTestDlg::OnNMRClickTaskList(NMHDR *pNMHDR, LRESULT *pResult)
{
LPNMITEMACTIVATE pNMItemActivate = reinterpret_cast<LPNMITEMACTIVATE>(pNMHDR);
if ( pNMItemActivate->iItem < 0 ) return;
CString strID = m_TaskList.GetItemText(pNMItemActivate->iItem, 1);
CPoint pt;
GetCursorPos(&pt);
CMenu mMenu, *pMenu;
mMenu.LoadMenu(IDR_MENU);
pMenu = mMenu.GetSubMenu(0);
if (strID == _T("0") || strID == _T("4")){
pMenu->EnableMenuItem(ID_M_EDN_TASK, MF_BYCOMMAND|MF_GRAYED);
pMenu->EnableMenuItem(ID_M_FIND_EXE, MF_BYCOMMAND|MF_GRAYED);
pMenu->EnableMenuItem(ID_M_COPY_PATH, MF_BYCOMMAND|MF_GRAYED);
}
pMenu->TrackPopupMenu(TPM_LEFTALIGN | TPM_RIGHTBUTTON, pt.x, pt.y, this);
*pResult = 0;
}
⑦、菜单响应函数的实现：
void CMFCTestDlg::OnMRefreshList()
{
for (int idx = 0; idx < m_ImgList.GetImageCount(); idx++) {
m_ImgList.Remove(idx);
}
m_TaskList.DeleteAllItems();
GetProcessList();
}
void CMFCTestDlg::OnMEdnTask()
{
int idx = m_TaskList.GetSelectionMark();
CString strID = m_TaskList.GetItemText(idx, 1);
CString strName = m_TaskList.GetItemText(idx, 0);
int iRet = MessageBox(_T("您确定要结束 ")+strName+_T(" 吗？"), _T("友情提示"), MB_OKCANCEL);
if (iRet == IDCANCEL) return;
if ( strID == _T("0") || strName == _T("4")) { //实际上这个位置的判断是多余的
MessageBox(_T("系统进程，无法结束！"), _T("提示"));
}else{
BOOL bRet = FALSE;
HANDLE hProcess = OpenProcess(PROCESS_TERMINATE|PROCESS_QUERY_INFORMATION, FALSE, _tstoi(strID));
if ( hProcess != NULL ) {
bRet = TerminateProcess(hProcess, -1);
CloseHandle(hProcess);
}
if ( !bRet ) {
MessageBox(_T("结束进程 \" ") + strName + _T(" \" 失败，请重试！"), _T("提示"));
}else{
m_TaskList.DeleteItem(idx);
}
}
}
void CMFCTestDlg::OnMFindExe()
{
int idx = m_TaskList.GetSelectionMark();
CString strPath = m_TaskList.GetItemText(idx, 2);
if (strPath.GetLength() > 0)
ShellExecute(m_hWnd, _T("open"), _T("Explorer.exe"), _T("/select,")+strPath, NULL, SW_SHOW);
}
void CMFCTestDlg::OnMCopyPath()
{
int idx = m_TaskList.GetSelectionMark();
CString strPath = m_TaskList.GetItemText(idx, 2);
if (!OpenClipboard()) return;
if(!EmptyClipboard()) {
CloseClipboard();
return;
}
size_t cbStr = (strPath.GetLength()+1)*sizeof(TCHAR);
HGLOBAL hData = GlobalAlloc(GMEM_MOVEABLE, cbStr);
memcpy_s(GlobalLock(hData), cbStr, strPath.LockBuffer(), cbStr);
GlobalUnlock(hData);
strPath.UnlockBuffer();
UINT uiFormat = (sizeof(TCHAR) == sizeof(WCHAR)) ? CF_UNICODETEXT : CF_TEXT;
if (::SetClipboardData(uiFormat, hData) == NULL) {
MessageBox(_T("设置剪贴板数据失败！"));
CloseClipboard();
return;
}
CloseClipboard();
}
⑧、为列表控件加上程序的前缀图标：
// H 头文件中进行图像列表的定义：
CImageList m_ImgList;
// OnInitDialog() 函数中进行图像列表的创建及设置：
m_ImgList.Create(16, 16, ILC_COLOR24, 10, 10);
m_TaskList.SetImageList(&m_ImgList, LVSIL_SMALL);
⑨、修改GetProcessList()函数为如下：
BOOL CMFCTestDlg::GetProcessList()
{
BOOL bResult = FALSE;
HANDLE hSnap = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
if ( hSnap == INVALID_HANDLE_VALUE )
return FALSE;
int idx = 0, imgIdx = -1;
CString strID;
HANDLE hProcess = NULL;
CString strPath, strSys(_T("SystemRoot"));
PROCESSENTRY32 info = {0};
info.dwSize = sizeof(PROCESSENTRY32);
BOOL bRet = Process32First(hSnap, &info);
while(bRet) {
hProcess = OpenProcess(PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, FALSE, info.th32ProcessID);
if (hProcess) {
strPath.Empty();
GetModuleFileNameEx(hProcess, NULL, strPath.GetBufferSetLength(MAX_PATH), MAX_PATH);
strPath.ReleaseBuffer();
strPath.TrimLeft(_T("\\?"));
if (strPath.Left(strSys.GetLength()).CompareNoCase(strSys) == 0){
TCHAR szWin[MAX_PATH] = {0};
GetWindowsDirectory(szWin, MAX_PATH);
strPath.Replace(strSys, szWin);
}
SHFILEINFO shInfo = {0};
SHGetFileInfo(strPath, 0, &shInfo, sizeof(SHFILEINFO), SHGFI_SMALLICON|SHGFI_ICON);
imgIdx = m_ImgList.Add(shInfo.hIcon);
}
//进行列表项的插入
idx = m_TaskList.InsertItem(m_TaskList.GetItemCount(), _T(""), imgIdx);
m_TaskList.SetItemText(idx, 0, info.szExeFile);
strID.Empty();
strID.Format(_T("%d"), info.th32ProcessID);
m_TaskList.SetItemText(idx, 1, strID);
m_TaskList.SetItemText(idx, 2, strPath);
//继续遍历进程列表
bRet = Process32Next(hSnap, &info);
}
return TRUE;
}
1、使用 EnumProcesses 函数进行进程的枚举；
2、针对选中多个进程进行结束时的处理！
