开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 系统/网络安全 > 杀毒 > 查看源码
ProcessViewPage.cpp
资源名称：OurAntiEvilTools.rar　[点击查看]
上传用户：shouhua
上传日期：2014-12-06
资源大小：5685k
文件大小：14k
源码类别：

杀毒

开发平台：

Visual C++

ProcessViewPage.cpp：源码内容
  1. // ProcessViewPage.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "AntiEvilTools.h"
  6. #include "ProcessViewPage.h"
  8. #ifdef _DEBUG
  9. #define new DEBUG_NEW
  10. #undef THIS_FILE
  11. static char THIS_FILE[] = __FILE__;
  12. #endif
  13. //#include <stdio.h>
  14. //#include <windows.h>
  15. #include "Iphlpapi.h"
  16. #include <tlhelp32.h>
  17. #include "Winsock2.h"
  18. #pragma comment(lib, "Iphlpapi.lib")
  19. #pragma comment(lib, "WS2_32.lib")
  21. long total_num=1;
  24. typedef struct 
  25. {
  26.   DWORD   dwState;         // 连接状态
  27.   DWORD   dwLocalAddr;     // 本地地址
  28.   DWORD   dwLocalPort;     // 本地端口
  29.   DWORD   dwRemoteAddr;    // 远程地址
  30.   DWORD   dwRemotePort;    // 远程端口
  31.   DWORD   dwProcessId; // 进程ID号
  32. } MIB_TCPEXROW, *PMIB_TCPEXROW;
  34. typedef struct 
  35. {
  36. DWORD dwNumEntries;
  37. MIB_TCPEXROW table[ANY_SIZE];
  38. } MIB_TCPEXTABLE, *PMIB_TCPEXTABLE;
  40. typedef struct 
  41. {
  42.   DWORD   dwLocalAddr;     // 本地地址
  43.   DWORD   dwLocalPort;     // 本地端口
  44.   DWORD   dwProcessId; // 进程ID号
  45. } MIB_UDPEXROW, *PMIB_UDPEXROW;
  47. typedef struct 
  48. {
  49. DWORD dwNumEntries;
  50. MIB_UDPEXROW table[ANY_SIZE];
  51. } MIB_UDPEXTABLE, *PMIB_UDPEXTABLE;
  54. // 扩展函数原型
  55. typedef DWORD (WINAPI *PFNAllocateAndGetTcpExTableFromStack)(
  56.   PMIB_TCPEXTABLE *pTcpTable, 
  57.   BOOL bOrder,             
  58.   HANDLE heap,
  59.   DWORD zero,
  60.   DWORD flags
  61. );
  63. typedef DWORD (WINAPI *PFNAllocateAndGetUdpExTableFromStack)(
  64.   PMIB_UDPEXTABLE *pUdpTable,  
  65.   BOOL bOrder,              
  66.   HANDLE heap,
  67.   DWORD zero,
  68.   DWORD flags
  69. );
  70. PCHAR ProcessPidToName(HANDLE hProcessSnap, DWORD ProcessId, PCHAR ProcessName);
  71. /////////////////////////////////////////////////////////////////////////////
  72. // CProcessViewPage property page
  74. IMPLEMENT_DYNCREATE(CProcessViewPage, CPropertyPage)
  76. CProcessViewPage::CProcessViewPage() : CPropertyPage(CProcessViewPage::IDD)
  77. {
  78. //{{AFX_DATA_INIT(CProcessViewPage)
  79. // NOTE: the ClassWizard will add member initialization here
  80. //}}AFX_DATA_INIT
  81. }
  83. CProcessViewPage::~CProcessViewPage()
  84. {
  85. }
  87. void CProcessViewPage::DoDataExchange(CDataExchange* pDX)
  88. {
  89. CPropertyPage::DoDataExchange(pDX);
  90. //{{AFX_DATA_MAP(CProcessViewPage)
  91. DDX_Control(pDX, IDC_LIST_PROCESS_VIEW, m_process_ListCtrl);
  92. //}}AFX_DATA_MAP
  93. }
  96. BEGIN_MESSAGE_MAP(CProcessViewPage, CPropertyPage)
  97. //{{AFX_MSG_MAP(CProcessViewPage)
  98. ON_NOTIFY(NM_RCLICK, IDC_LIST_PROCESS_VIEW, OnRclickProcessList)
  99. ON_COMMAND(ID_REF, OnMenuitemProcessRefresh)
  100. // NOTE: the ClassWizard will add message map macros here
  101. //}}AFX_MSG_MAP
  102. END_MESSAGE_MAP()
  104. /////////////////////////////////////////////////////////////////////////////
  105. // CProcessViewPage message handlers
  109. int CProcessViewPage::GetFileIcon(LPCTSTR lpFileName)
  110. {
  111. SHFILEINFO sfi;
  112. SHGetFileInfo(lpFileName, 0, &sfi, sizeof(sfi), SHGFI_ICON|SHGFI_SMALLICON|SHGFI_LARGEICON);
  113. return sfi.iIcon;
  114. }
  116. void CProcessViewPage::GetProcessPort()
  117. {
  118. m_process_ListCtrl.DeleteAllItems();
  119. PFNAllocateAndGetTcpExTableFromStack pAllocateAndGetTcpExTableFromStack;
  120. PFNAllocateAndGetUdpExTableFromStack pAllocateAndGetUdpExTableFromStack;
  121.     CString   str1;
  122. // 获取扩展函数的入口地址
  123. HMODULE hModule = ::LoadLibrary("iphlpapi.dll");
  124. pAllocateAndGetTcpExTableFromStack = 
  125. (PFNAllocateAndGetTcpExTableFromStack)::GetProcAddress(hModule, 
  126. "AllocateAndGetTcpExTableFromStack");
  128. pAllocateAndGetUdpExTableFromStack = 
  129. (PFNAllocateAndGetUdpExTableFromStack)::GetProcAddress(hModule, 
  130. "AllocateAndGetUdpExTableFromStack");
  132. if(pAllocateAndGetTcpExTableFromStack == NULL || pAllocateAndGetUdpExTableFromStack == NULL)
  133. {
  134. printf(" Ex APIs are not present n ");
  135. // 说明你应该调用普通的IP帮助API去获取TCP连接表和UDP监听表
  136. return ;
  137. }
  139. // 调用扩展函数，获取TCP扩展连接表和UDP扩展监听表
  141. PMIB_TCPEXTABLE pTcpExTable;
  142. PMIB_UDPEXTABLE pUdpExTable;
  144. // pTcpExTable和pUdpExTable所指的缓冲区自动由扩展函数在进程堆中申请
  145. if(pAllocateAndGetTcpExTableFromStack(&pTcpExTable, TRUE, GetProcessHeap(), 2, 2) != 0)
  146. {
  147. //printf(" Failed to snapshot TCP endpoints.n");
  148. return ;
  149. }
  150. if(pAllocateAndGetUdpExTableFromStack(&pUdpExTable, TRUE, GetProcessHeap(), 2, 2) != 0)
  151. {
  152. //printf(" Failed to snapshot UDP endpoints.n");
  153. return ;
  154. }
  156. // 给系统内的所有进程拍一个快照
  157. HANDLE hProcessSnap = ::CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
  158. if(hProcessSnap == INVALID_HANDLE_VALUE)
  159. {
  160. //printf(" Failed to take process snapshot. Process names will not be shown.nn");
  161. return ;
  162. }
  164. // printf(" Active Connections nn");
  165. // char szLocalAddr[128];
  166. // char szRemoteAddr[128];
  167. char szProcessName[128];
  168. in_addr inadLocal, inadRemote;
  169. char    strState[128];
  170. DWORD   dwRemotePort = 0;
  171. char strTemp[1024];
  172. CString str, strNum, strName,szSource,strSourPort, szDest,strDestPort, strData, strSize,strMAC,strHost;
  173. // 打印UDP监听表信息
  174. for(UINT i = 0; i < pUdpExTable->dwNumEntries; ++i)
  175. {
  176. // 本地IP地址
  177. inadLocal.s_addr = pUdpExTable->table[i].dwLocalAddr;
  179. // 远程端口
  180. if(strcmp(strState, "LISTEN") != 0)
  181. {
  182. dwRemotePort = pTcpExTable->table[i].dwRemotePort;
  183. }
  184. else
  185. dwRemotePort = 0;
  187. // 远程IP地址
  188. inadRemote.s_addr = pTcpExTable->table[i].dwRemoteAddr;
  191. // sprintf(szLocalAddr,  "%s:%u", inet_ntoa(inadLocal), 
  192. // ntohs((unsigned short)(0x0000FFFF & pUdpExTable->table[i].dwLocalPort)));
  194. // 打印出此入口的信息
  195. // str1.Format("%-5s %s:%dn", "[UDP]", 
  196. // ProcessPidToName(hProcessSnap, pUdpExTable->table[i].dwProcessId, szProcessName),
  197. // pUdpExTable->table[i].dwProcessId );
  198. // MessageBox(str1);
  199. strNum.Format("%d",pUdpExTable->table[i].dwProcessId);
  200. strName.Format("%s",ProcessPidToName(hProcessSnap, pUdpExTable->table[i].dwProcessId, szProcessName));
  201. szSource.Format("%s",inet_ntoa(inadLocal));
  202. strSourPort.Format("%d",ntohs((unsigned short)(0x0000FFFF & pUdpExTable->table[i].dwLocalPort)));
  203. szDest.Format("%s","*.*.*.*");
  204. // strDestPort.Format("%d",ntohs((unsigned short)(0x0000FFFF & dwRemotePort)));
  205. int index;
  206. CString Proto="UDP";
  208. GetExeFilePath(ProcessPidToName(hProcessSnap, pUdpExTable->table[i].dwProcessId, szProcessName), strTemp,pUdpExTable->table[i].dwProcessId);
  209. index = m_process_ListCtrl.InsertItem(0,strName,GetFileIcon(strTemp));
  210. m_process_ListCtrl.SetItem(index,1,LVIF_TEXT,strNum, 0, 0, 0, 0);
  211. m_process_ListCtrl.SetItem(index,2,LVIF_TEXT,Proto, 0, 0, 0, 0);
  212. m_process_ListCtrl.SetItem(index,3,LVIF_TEXT,szSource, 0, 0, 0, 0);
  213. m_process_ListCtrl.SetItem(index,4,LVIF_TEXT,strSourPort, 0, 0, 0, 0);
  214. m_process_ListCtrl.SetItem(index,5,LVIF_TEXT,szDest, 0, 0, 0, 0);
  215. m_process_ListCtrl.SetItem(index,6,LVIF_TEXT,"*", 0, 0, 0, 0);
  216. m_process_ListCtrl.SetItem(index,7,LVIF_TEXT,"*", 0, 0, 0, 0);
  217. /*printf("      Local:   %sn      Remote:  %sn",
  218. szLocalAddr, "*.*.*.*:*" );*/
  219. }
  220. // 打印TCP扩展连接表信息
  221. for(i = 0; i < pTcpExTable->dwNumEntries; ++i)
  222. {
  223. // 状态
  224. switch (pTcpExTable->table[i].dwState)
  225. {
  226. case MIB_TCP_STATE_CLOSED:
  227. strcpy(strState, "CLOSED");
  228. break;
  229. case MIB_TCP_STATE_TIME_WAIT:
  230. strcpy(strState, "TIME_WAIT");
  231. break;
  232. case MIB_TCP_STATE_LAST_ACK:
  233. strcpy(strState, "LAST_ACK");
  234. break;
  235. case MIB_TCP_STATE_CLOSING:
  236. strcpy(strState, "CLOSING");
  237. break;
  238. case MIB_TCP_STATE_CLOSE_WAIT:
  239. strcpy(strState, "CLOSE_WAIT");
  240. break;
  241. case MIB_TCP_STATE_FIN_WAIT1:
  242. strcpy(strState, "FIN_WAIT1");
  243. break;
  244. case MIB_TCP_STATE_ESTAB:
  245. strcpy(strState, "ESTAB");
  246. break;
  247. case MIB_TCP_STATE_SYN_RCVD:
  248. strcpy(strState, "SYN_RCVD");
  249. break;
  250. case MIB_TCP_STATE_SYN_SENT:
  251. strcpy(strState, "SYN_SENT");
  252. break;
  253. case MIB_TCP_STATE_LISTEN:
  254. strcpy(strState, "LISTEN");
  255. break;
  256. case MIB_TCP_STATE_DELETE_TCB:
  257. strcpy(strState, "DELETE");
  258. break;
  259. default:
  260. // printf("Error: unknown state!n");
  261. break;
  262. }
  263. // 本地IP地址
  264. inadLocal.s_addr = pTcpExTable->table[i].dwLocalAddr;
  266. // 远程端口
  267. if(strcmp(strState, "LISTEN") != 0)
  268. {
  269. dwRemotePort = pTcpExTable->table[i].dwRemotePort;
  270. }
  271. else
  272. dwRemotePort = 0;
  274. // 远程IP地址
  275. inadRemote.s_addr = pTcpExTable->table[i].dwRemoteAddr;
  278. // sprintf(szLocalAddr, "%s:%u", inet_ntoa(inadLocal), 
  279. // ntohs((unsigned short)(0x0000FFFF & pTcpExTable->table[i].dwLocalPort)));
  280. // sprintf(szRemoteAddr, "%s:%u", inet_ntoa(inadRemote), 
  281. // ntohs((unsigned short)(0x0000FFFF & dwRemotePort)));
  283. // 打印出此入口的信息
  284. //str1.Format("%-5s %s:%dn      State:   %sn", "[TCP]", 
  285. // ProcessPidToName(hProcessSnap, pTcpExTable->table[i].dwProcessId, szProcessName),
  286. // pTcpExTable->table[i].dwProcessId,
  287. // strState);
  288.      //   MessageBox(str1);
  289. strNum.Format("%d",pTcpExTable->table[i].dwProcessId);
  290. strName.Format("%s",ProcessPidToName(hProcessSnap, pTcpExTable->table[i].dwProcessId, szProcessName));
  291. szSource.Format("%s",inet_ntoa(inadLocal));
  292. strSourPort.Format("%d",ntohs((unsigned short)(0x0000FFFF & pTcpExTable->table[i].dwLocalPort)));
  293. szDest.Format("%s",inet_ntoa(inadRemote));
  295. strDestPort.Format("%d",ntohs((unsigned short)(0x0000FFFF & dwRemotePort)));
  296. if(szDest=="0.0.0.0")
  297. strDestPort="0";
  298. int index;
  299. CString Proto="TCP";
  300. GetExeFilePath(ProcessPidToName(hProcessSnap, pUdpExTable->table[i].dwProcessId, szProcessName), strTemp,pUdpExTable->table[i].dwProcessId);
  301. index = m_process_ListCtrl.InsertItem(0,strName,GetFileIcon(strTemp));
  302. m_process_ListCtrl.SetItem(index,1,LVIF_TEXT,strNum, 0, 0, 0, 0);
  303. m_process_ListCtrl.SetItem(index,2,LVIF_TEXT,Proto, 0, 0, 0, 0);
  304. m_process_ListCtrl.SetItem(index,3,LVIF_TEXT,szSource, 0, 0, 0, 0);
  305. m_process_ListCtrl.SetItem(index,4,LVIF_TEXT,strSourPort, 0, 0, 0, 0);
  306. m_process_ListCtrl.SetItem(index,5,LVIF_TEXT,szDest, 0, 0, 0, 0);
  307. m_process_ListCtrl.SetItem(index,6,LVIF_TEXT,strDestPort, 0, 0, 0, 0);
  308. m_process_ListCtrl.SetItem(index,7,LVIF_TEXT,strState, 0, 0, 0, 0);
  309. /*printf("      Local:   %sn      Remote:  %sn",
  310. szLocalAddr, szRemoteAddr);*/
  311. }
  316. ::CloseHandle(hProcessSnap);
  317. ::LocalFree(pTcpExTable);
  318. ::LocalFree(pUdpExTable);
  319. ::FreeLibrary(hModule);
  320. return ;
  321. }
  323. bool CProcessViewPage::InitSystemImageLists(HWND hwndList)
  324. {
  325. SHFILEINFO sfi;
  327.     HIMAGELIST himlSmall = (HIMAGELIST)::SHGetFileInfo( "C:\", 0, &sfi, 
  328. sizeof(SHFILEINFO), SHGFI_SYSICONINDEX | SHGFI_SMALLICON );
  330.     HIMAGELIST himlLarge = (HIMAGELIST)::SHGetFileInfo( "C:\", 0, &sfi, 
  331. sizeof(SHFILEINFO), SHGFI_SYSICONINDEX | SHGFI_LARGEICON );
  333.     if( himlSmall && himlLarge ) { 
  334. ListView_SetImageList(hwndList, himlSmall, LVSIL_SMALL);
  335. ListView_SetImageList(hwndList, himlLarge, LVSIL_NORMAL);
  336. return TRUE;
  337.     }
  339.     return FALSE;
  340. }
  342. BOOL CProcessViewPage::OnInitDialog()
  343. {
  344. CDialog::OnInitDialog();
  346. // Add "About..." menu item to system menu.
  348. // IDM_ABOUTBOX must be in the system command range.
  349. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  350. ASSERT(IDM_ABOUTBOX < 0xF000);
  352. CMenu* pSysMenu = GetSystemMenu(FALSE);
  353. if (pSysMenu != NULL)
  354. {
  355. CString strAboutMenu;
  356. strAboutMenu.LoadString(IDS_ABOUTBOX);
  357. if (!strAboutMenu.IsEmpty())
  358. {
  359. pSysMenu->AppendMenu(MF_SEPARATOR);
  360. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  361. }
  362. }
  364. // Set the icon for this dialog.  The framework does this automatically
  365. //  when the application's main window is not a dialog
  366. SetIcon(m_hIcon, TRUE); // Set big icon
  367. SetIcon(m_hIcon, FALSE); // Set small icon
  369. // TODO: Add extra initialization here
  370. //  ListView initialize
  372. DWORD dwStyle=GetWindowLong(m_process_ListCtrl.GetSafeHwnd(),GWL_STYLE);
  373. dwStyle&=~LVS_TYPEMASK;
  374. dwStyle|=LVS_REPORT;
  375. SetWindowLong(m_process_ListCtrl.GetSafeHwnd(),GWL_STYLE,dwStyle);
  376. //SetRedraw(TRUE); 
  378. //初始化列表
  379. InitSystemImageLists(m_process_ListCtrl.m_hWnd);
  380. m_process_ListCtrl.InsertColumn(0,"状态",LVCFMT_LEFT,60);
  381.     m_process_ListCtrl.InsertColumn(0,"目的端口",LVCFMT_LEFT,80);
  382. m_process_ListCtrl.InsertColumn(0,"目的地址IP",LVCFMT_LEFT,100);
  383. m_process_ListCtrl.InsertColumn(0,"源端口",LVCFMT_LEFT,60);
  384. m_process_ListCtrl.InsertColumn(0,"源地址",LVCFMT_LEFT,100);
  385. m_process_ListCtrl.InsertColumn(0,"协议",LVCFMT_LEFT,40);
  386. m_process_ListCtrl.InsertColumn(0,"进程号",LVCFMT_LEFT,60);
  387. m_process_ListCtrl.InsertColumn(0,"进程名",LVCFMT_LEFT,80);
  388.     
  389.     ::SendMessage(m_process_ListCtrl.m_hWnd, LVM_SETEXTENDEDLISTVIEWSTYLE,
  390. LVS_EX_FULLROWSELECT, LVS_EX_FULLROWSELECT);
  392. // SetWindowText("进程与端口");
  395. // SetTimer(0, 500, NULL);
  397. GetProcessPort();
  398. return TRUE;  // return TRUE  unless you set the focus to a control
  399. }
  401. bool CProcessViewPage::GetExeFilePath(char pExeFileName[], char buffer[], DWORD pid)
  402. {
  403. bool flag = true;
  404. memset(buffer, 0, strlen(buffer));
  405. int result = (int)FindExecutable(pExeFileName, 0, buffer);
  406. if(result <= 32)
  407. {
  408. memset(buffer, 0, strlen(buffer));
  409. HANDLE hProcess;
  410. hProcess = OpenProcess(PROCESS_VM_READ|PROCESS_QUERY_INFORMATION, FALSE, pid);
  411. if(hProcess != NULL)
  412. {
  413. if(GetModuleFileNameEx(hProcess, 0, buffer, 1024) == 0)
  414. flag = false;
  415. }
  416. else
  417. {
  418. flag = false;
  419. }
  420. CloseHandle(hProcess);
  421. }
  422. if(!flag)
  423. {
  424. strcpy(buffer, "不知道");
  425. }
  426. return flag;
  427. }
  428. PCHAR ProcessPidToName(HANDLE hProcessSnap, DWORD ProcessId, PCHAR ProcessName)
  429. {
  430. PROCESSENTRY32 processEntry;
  431. processEntry.dwSize = sizeof(processEntry);
  432. // 找不到的话，默认进程名为“???”
  433. strcpy(ProcessName, "???");
  434. if(!::Process32First(hProcessSnap, &processEntry)) 
  435. return ProcessName;
  436. do 
  437. {
  438. if(processEntry.th32ProcessID == ProcessId) // 就是这个进程
  439. {
  440. strcpy(ProcessName, processEntry.szExeFile);
  441. break;
  442. }
  443. }
  444. while(::Process32Next(hProcessSnap, &processEntry));
  445. return ProcessName;
  446. }
  448. void CProcessViewPage::OnRclickProcessList(NMHDR *pNMHDR, LRESULT *pResult)
  449. {
  450. iSelected = m_process_ListCtrl.GetNextItem(-1, LVNI_SELECTED);
  451. POINT point;
  452. ::GetCursorPos(&point);
  453. CMenu menu;
  454. menu.LoadMenu(IDR_MENU1);
  455. menu.GetSubMenu(0)->TrackPopupMenu(TPM_LEFTALIGN, point.x, point.y, this);
  456. *pResult = 0;
  457. }
  459. void CProcessViewPage::OnMenuitemProcessRefresh()
  460. {
  461. GetProcessPort();
  462. }