Windows API 函数学习(13)---进程函数示例

 // ProcEnum.cpp : Defines the entry point for the application.
//本程序演示了如何列举系统进程，并且更改进程的优先级，结束进程
//控件:ListView,TrackBar控件的使用示例,其中资源是添加了两个对话框模板

#include "stdafx.h"
#include "resource.h"
#include <stdio.h>
#include <commctrl.h>
#include <Tlhelp32.h>
#include <stdlib.h>
#include <winbase.h>
#define BELOW_NORMAL_PRIORITY_CLASS 0x00004000;//定义两个优先级的级别
#define ABOVE_NORMAL_PRIORITY_CLASS 0x00008000;

HINSTANCE hInst;
DWORD ProcPri;
char temp[32];
HWND hProcList;
HWND hSlider;
DWORD dwPriority;
BOOL DlgProc (HWND hWnd,UINT uMsg,WPARAM wParam,LPARAM lParam);
LRESULT CALLBACK DlgProc2(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam);
void DisplayRuningProcess();
int APIENTRY WinMain(HINSTANCE hInstance,
                     HINSTANCE hPrevInstance,
                     LPSTR     lpCmdLine,
                     int       nCmdShow)
{
  // TODO: Place code here.
 INITCOMMONCONTROLSEX icex;
 icex.dwSize = sizeof(INITCOMMONCONTROLSEX);//初始化ListView控件
 icex.dwICC  = ICC_LISTVIEW_CLASSES;
    InitCommonControlsEx(&icex);
    hInst=hInstance;
 DialogBoxParam(hInstance,MAKEINTRESOURCE(IDD_DIALOG1),0,(DLGPROC)DlgProc,(IDD_DIALOG1));
 return 0;
}
BOOL DlgProc (HWND hWnd,UINT uMsg,WPARAM wParam,LPARAM lParam)
{
 int lowID;
 char psTest[256];
 int index;
 char ProcID[16];
 HANDLE hProcess;
 switch (uMsg)
 {
      case WM_INITDIALOG:
         hProcList=CreateWindowEx (LVS_EX_FULLROWSELECT,WC_LISTVIEW, "",
   WS_VISIBLE|WS_CHILD | LVS_REPORT | LVS_EDITLABELS,
   10, 10, 340, 400,
   hWnd, (HMENU) ID_LISTVIEW, hInst, NULL); //创建ListView控件
   LVCOLUMN lvco;
   lvco.mask=LVCF_FMT|LVCF_TEXT|LVCF_SUBITEM|LVCF_WIDTH;
   lvco.pszText=psTest;
   lvco.cx=80;
   lvco.fmt=LVCFMT_CENTER;
   lvco.iSubItem=0;
         LoadString(hInst, IDS_ID, psTest, sizeof(psTest)); //设置表头文本

   ListView_InsertColumn(hProcList,0,&lvco);
         lvco.iSubItem=1;
  
   LoadString(hInst, IDS_PRI, psTest, sizeof(psTest));
   ListView_InsertColumn(hProcList,1,&lvco);
   lvco.iSubItem=2;
   lvco.cx=180;
   LoadString(hInst,IDS_PROCNAME,psTest,sizeof(psTest));
   ListView_InsertColumn(hProcList,2,&lvco);
   DisplayRuningProcess();
   break;
 case WM_COMMAND:
  lowID=LOWORD(wParam);
     switch(lowID)
  {
  case IDOK:
      EndDialog(hWnd,FALSE);
      break;
  case IDC_REFRESH:
   DisplayRuningProcess(); //刷新进程列表
   break;
  case IDC_KILLPROCESS:
           
   index=ListView_GetSelectionMark(hProcList);
   if(index==-1)
   {
    MessageBox(hWnd,"请选择要结束的进程","错误!",MB_OK);
    return 0;
   }
   ListView_GetItemText(hProcList,index,0,ProcID,sizeof(ProcID));
   hProcess=OpenProcess(PROCESS_TERMINATE,FALSE,atoi(ProcID));
   TerminateProcess(hProcess,0);
   DisplayRuningProcess();
   break;
  case IDC_SET:   //更改优先级
      index=ListView_GetSelectionMark(hProcList);
   
            if(index==-1)
   {
    MessageBox(hWnd,"请选择要结束的进程","错误!",MB_OK);
    return 0;
   }
   ListView_GetItemText(hProcList,index,0,ProcID,sizeof(ProcID));
   hProcess=OpenProcess(PROCESS_SET_INFORMATION,FALSE,atoi(ProcID));
   if(IDOK==DialogBox(hInst,MAKEINTRESOURCE(IDD_DIALOG2),hWnd,(DLGPROC)DlgProc2))
   {
    if(!SetPriorityClass(hProcess,dwPriority))
    {
     MessageBox(hWnd,"更改优先级失败!","错误!",MB_OK);
     break;
    }
    DisplayRuningProcess();
   }
    
   break;
  }
     break;
 case WM_CLOSE:
  EndDialog(hWnd,FALSE);
  break;
 default:
  return FALSE;
 }
 return TRUE;
}

void DisplayRuningProcess()
{

 LVITEM lvI;
 char temp[50];
 int j;
 int CurItemIndex=0;
 HANDLE hSnapShot;
 PROCESSENTRY32 pe;
 
 DWORD retval;
 hSnapShot=CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS,0);
 retval=Process32First(hSnapShot,&pe);
    ListView_DeleteAllItems(hProcList);
 lvI.mask = LVIF_TEXT ;
 do
    {      
  lvI.iItem = CurItemIndex;
  lvI.iSubItem =0;
  lvI.pszText="";
  ListView_InsertItem(hProcList, &lvI);//先插入空行，然后进行填写数据
  for(j=0;j<3;j++)
  { 
   switch(j)
   {
   case 0:
       sprintf(temp ,"%d",pe.th32ProcessID);
    break;
   case 1:
    sprintf(temp,"%d",pe.pcPriClassBase);
    break;
   case 2:
    sprintf(temp ,"%s",pe.szExeFile);
    break;
   }
   ListView_SetItemText(hProcList,CurItemIndex,j,temp);//增加一行内容
          }
  CurItemIndex++;
 }while(Process32Next(hSnapShot,&pe));
    CloseHandle(hSnapShot);
}

LRESULT CALLBACK DlgProc2(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
 switch (message)
 {
 case WM_INITDIALOG:
  INITCOMMONCONTROLSEX icex;
  icex.dwSize = sizeof(INITCOMMONCONTROLSEX);//初始化TrakBar控件
  icex.dwICC= ICC_BAR_CLASSES;
  InitCommonControlsEx(&icex);
        hSlider=CreateWindowEx(0,TRACKBAR_CLASS,NULL,    //创建TrackBar控件       
   WS_CHILD | WS_VISIBLE | TBS_AUTOTICKS|TBS_VERT,  
   10, 10, 20, 200, hWnd,(HMENU)IDC_SLIDER, hInst, NULL);
  SendMessage(hSlider,TBM_SETRANGE,TRUE,MAKELONG(0,5));//设置范围
  SendMessage(hSlider,TBM_SETPOS,TRUE,3);//设置默认值
  dwPriority=NORMAL_PRIORITY_CLASS;
  break;
 case WM_VSCROLL://利用这个消息返回当前设置的优先级
  DWORD dwPos;
  dwPos=SendMessage((HWND)lParam,TBM_GETPOS,0,0);
  switch(dwPos)
  {
 
  case 0:
   dwPriority=REALTIME_PRIORITY_CLASS;
   break;
  case 1:
   dwPriority=HIGH_PRIORITY_CLASS;
   break;
  case 2:
   dwPriority=ABOVE_NORMAL_PRIORITY_CLASS;
   break;
  case 3:
   dwPriority=NORMAL_PRIORITY_CLASS;
   break;
  case 4:
   dwPriority=BELOW_NORMAL_PRIORITY_CLASS;
   break;
  case 5:
   dwPriority=IDLE_PRIORITY_CLASS;
  } 
  SendMessage(hSlider,TBM_SETPOS,TRUE,(LPARAM)dwPos);
  break;

 case WM_COMMAND:
  if (LOWORD(wParam) == IDOK || LOWORD(wParam) == IDCANCEL)
  {
   EndDialog(hWnd, LOWORD(wParam));
   return TRUE;
  }
 }
    return FALSE;
}