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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 进程与线程 > 查看源码
NeighborInfoDlg.cpp
资源名称：NeighborInfo.rar　[点击查看]
上传用户：jnhwc1
上传日期：2022-07-09
资源大小：100k
文件大小：14k
源码类别：

进程与线程

开发平台：

Visual C++

NeighborInfoDlg.cpp：源码内容
  1. // NeighborInfoDlg.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "NeighborInfo.h"
  6. #include "NeighborInfoDlg.h"
  7. #include "resource.h"
  8. #include "TLHELP32.H"
  9. #include "psapi.h"
  11. #ifdef _DEBUG
  12. #define new DEBUG_NEW
  13. #undef THIS_FILE
  14. static char THIS_FILE[] = __FILE__;
  15. #endif
  17. #define NBTSTATPORT 137 //nbtstat name port
  18. #define OWNERPORT 4321
  20. /////////////////////////////////////////////////////////////////////////////
  21. // CAboutDlg dialog used for App About
  22. //--------------------------------var-------------------------------
  23. BYTE bs[50]=
  24. {0x0,0x00,0x0,0x10,0x0,0x1,0x0,0x0,0x0,0x0,0x0,0x0,0x20,0x43,0x4b,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x41,0x0,0x0,0x21,0x0,0x1};
  25. HANDLE wait_handle;
  26. UINT NbtstatThread(LPVOID param);
  30. class CAboutDlg : public CDialog
  31. {
  32. public:
  33. CAboutDlg();
  35. // Dialog Data
  36. //{{AFX_DATA(CAboutDlg)
  37. enum { IDD = IDD_ABOUTBOX };
  38. //}}AFX_DATA
  40. // ClassWizard generated virtual function overrides
  41. //{{AFX_VIRTUAL(CAboutDlg)
  42. protected:
  43. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  44. //}}AFX_VIRTUAL
  46. // Implementation
  47. protected:
  48. //{{AFX_MSG(CAboutDlg)
  49. //}}AFX_MSG
  50. DECLARE_MESSAGE_MAP()
  51. };
  53. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  54. {
  55. //{{AFX_DATA_INIT(CAboutDlg)
  56. //}}AFX_DATA_INIT
  57. }
  59. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  60. {
  61. CDialog::DoDataExchange(pDX);
  62. //{{AFX_DATA_MAP(CAboutDlg)
  63. //}}AFX_DATA_MAP
  64. }
  66. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  67. //{{AFX_MSG_MAP(CAboutDlg)
  68. // No message handlers
  69. //}}AFX_MSG_MAP
  70. END_MESSAGE_MAP()
  72. /////////////////////////////////////////////////////////////////////////////
  73. // CNeighborInfoDlg dialog
  75. CNeighborInfoDlg::CNeighborInfoDlg(CWnd* pParent /*=NULL*/)
  76. : CDialog(CNeighborInfoDlg::IDD, pParent)
  77. {
  78. //{{AFX_DATA_INIT(CNeighborInfoDlg)
  79. //}}AFX_DATA_INIT
  80. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  81. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  82. }
  84. void CNeighborInfoDlg::DoDataExchange(CDataExchange* pDX)
  85. {
  86. CDialog::DoDataExchange(pDX);
  87. //{{AFX_DATA_MAP(CNeighborInfoDlg)
  88. DDX_Control(pDX, IDC_LISTPROCSS, m_ListProcess);
  89. DDX_Control(pDX, IDC_IPADDRESS1, m_123);
  90. DDX_Control(pDX, IDC_LISTINFOR, m_ListView);
  91. //}}AFX_DATA_MAP
  92. }
  94. BEGIN_MESSAGE_MAP(CNeighborInfoDlg, CDialog)
  95. ON_WM_CONTEXTMENU()
  96. //{{AFX_MSG_MAP(CNeighborInfoDlg)
  97. ON_WM_SYSCOMMAND()
  98. ON_WM_PAINT()
  99. ON_WM_QUERYDRAGICON()
  100. ON_BN_CLICKED(IDC_BEGINSCAN, OnBeginScan)
  101. ON_BN_CLICKED(IDC_STOPSCAN, OnStopScan)
  102. ON_BN_CLICKED(IDC_MYHELP, OnMyHelp)
  103. ON_BN_CLICKED(IDC_EMPTY, OnEmpty)
  104. ON_BN_CLICKED(IDC_BUTTON2, OnButton2)
  105. ON_BN_CLICKED(IDC_BUTTON3, OnButton3)
  106. ON_BN_CLICKED(IDC_BUTTON4, OnButton4)
  107. ON_BN_CLICKED(IDC_BUTTON5, OnButton5)
  108. ON_WM_TIMER()
  109. //}}AFX_MSG_MAP
  110. END_MESSAGE_MAP()
  112. /////////////////////////////////////////////////////////////////////////////
  113. // CNeighborInfoDlg message handlers
  115. BOOL CNeighborInfoDlg::OnInitDialog()
  116. {
  117. CDialog::OnInitDialog();
  119. // Add "About..." menu item to system menu.
  121. // IDM_ABOUTBOX must be in the system command range.
  122. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  123. ASSERT(IDM_ABOUTBOX < 0xF000);
  125. CMenu* pSysMenu = GetSystemMenu(FALSE);
  126. if (pSysMenu != NULL)
  127. {
  128. CString strAboutMenu;
  129. strAboutMenu.LoadString(IDS_ABOUTBOX);
  130. if (!strAboutMenu.IsEmpty())
  131. {
  132. pSysMenu->AppendMenu(MF_SEPARATOR);
  133. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  134. }
  135. }
  137. // Set the icon for this dialog.  The framework does this automatically
  138. //  when the application's main window is not a dialog
  139. SetIcon(m_hIcon, TRUE); // Set big icon
  140. SetIcon(m_hIcon, FALSE); // Set small icon
  142. // TODO: Add extra initialization here
  143. wait_handle=CreateEvent(NULL,true,false,"receive data");
  144. GetDlgItem(IDC_STOPSCAN)->EnableWindow(false);
  146. DWORD dwStyle=GetWindowLong(m_ListView.GetSafeHwnd(),GWL_STYLE);
  147. dwStyle&=~LVS_TYPEMASK;
  148. dwStyle|=LVS_REPORT;
  149. SetWindowLong(m_ListView.GetSafeHwnd(),GWL_STYLE,dwStyle);
  150. SetWindowLong(m_ListProcess.GetSafeHwnd(),GWL_STYLE,dwStyle);
  152. //初始化视图列表1
  153.     m_ListView.InsertColumn(3,"MAC地址",LVCFMT_LEFT,127);
  154. m_ListView.InsertColumn(2,"主机",LVCFMT_LEFT,127);
  155. m_ListView.InsertColumn(1,"工作组",LVCFMT_LEFT,128);
  156. m_ListView.InsertColumn(0,"IP地址",LVCFMT_LEFT,128); 
  157. m_ListView.SetExtendedStyle(LVS_EX_GRIDLINES);
  158.     ::SendMessage(m_ListView.m_hWnd, LVM_SETEXTENDEDLISTVIEWSTYLE,
  159.       LVS_EX_FULLROWSELECT, LVS_EX_FULLROWSELECT);
  161. //初始化视图列表2
  162. m_ListProcess.InsertColumn(0,"进程",LVCFMT_LEFT,127);
  163. m_ListProcess.InsertColumn(1,"ID",LVCFMT_LEFT,127);
  164. m_ListProcess.InsertColumn(2,"父进程",LVCFMT_LEFT,128);
  165. m_ListProcess.SetExtendedStyle(LVS_EX_GRIDLINES);
  166.     ::SendMessage(m_ListProcess.m_hWnd, LVM_SETEXTENDEDLISTVIEWSTYLE,
  167. LVS_EX_FULLROWSELECT, LVS_EX_FULLROWSELECT);
  170. //初始化端口
  171. if(AfxSocketInit(NULL)==FALSE)
  172. {
  173. AfxMessageBox("Error Init Socket");
  174. }
  175. m_socket.Init(this);
  176. if(m_socket.Create(OWNERPORT,SOCK_DGRAM)==FALSE)
  177. {
  178. AfxMessageBox("Socket Create Error");
  179. }
  180.     
  181. //设置计时器
  182. SetTimer(0,100,NULL);
  184. m_bStop=FALSE;
  185. return TRUE;  // return TRUE  unless you set the focus to a control
  186. }
  188. void CNeighborInfoDlg::OnSysCommand(UINT nID, LPARAM lParam)
  189. {
  190. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  191. {
  192. CAboutDlg dlgAbout;
  193. dlgAbout.DoModal();
  194. }
  195. else
  196. {
  197. CDialog::OnSysCommand(nID, lParam);
  198. }
  199. }
  201. // If you add a minimize button to your dialog, you will need the code below
  202. //  to draw the icon.  For MFC applications using the document/view model,
  203. //  this is automatically done for you by the framework.
  205. void CNeighborInfoDlg::OnPaint() 
  206. {
  207. if (IsIconic())
  208. {
  209. CPaintDC dc(this); // device context for painting
  211. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  213. // Center icon in client rectangle
  214. int cxIcon = GetSystemMetrics(SM_CXICON);
  215. int cyIcon = GetSystemMetrics(SM_CYICON);
  216. CRect rect;
  217. GetClientRect(&rect);
  218. int x = (rect.Width() - cxIcon + 1) / 2;
  219. int y = (rect.Height() - cyIcon + 1) / 2;
  221. // Draw the icon
  222. dc.DrawIcon(x, y, m_hIcon);
  223. }
  224. else
  225. {
  226. CDialog::OnPaint();
  227. }
  228. }
  230. // The system calls this to obtain the cursor to display while the user drags
  231. //  the minimized window.
  232. HCURSOR CNeighborInfoDlg::OnQueryDragIcon()
  233. {
  234. return (HCURSOR) m_hIcon;
  235. }
  237. void CNeighborInfoDlg::OnBeginScan() 
  238. {
  239. GetDlgItem(IDC_BEGINSCAN)->EnableWindow(FALSE);
  240. GetDlgItem(IDC_STOPSCAN)->EnableWindow(TRUE);
  242. //开始扫描
  243. AfxBeginThread(NbtstatThread,(LPVOID)this,THREAD_PRIORITY_NORMAL);
  244. }
  246. void CNeighborInfoDlg::OnStopScan() 
  247. {
  248. // TODO: Add your control notification handler code here
  249. m_bStop=TRUE;
  250. }
  252. void CNeighborInfoDlg::ReceiveMessage()
  253. {
  254. BYTE Buf[500];
  255. static CString strOldIP;
  256. CString str,strIP,strHost,strHex,strMac,Host,Group;
  257. UINT dport;
  258. m_socket.ReceiveFrom(Buf,500,strIP,dport,0);
  260. if(strIP==(char)NULL||strIP==strOldIP)return;
  261. strOldIP=strIP;
  263. //ip
  264. int index=m_ListView.InsertItem(0,strIP);
  266. strHost="";
  267. strHex="";
  268. Host="\";
  270. int tem=0,num=0;
  271. bool bAdd=true;
  273. for(int i=0;i<500;i++)
  274. {
  275. if(i%50==0)
  276. TRACE("n");
  277. TRACE("%c",Buf[i]);
  278. }
  280. //nbtstat前面的57个字节是
  281. for(i=57;i<500;i++) //57-72
  282. {
  283. //end
  284. if(Buf[i]==0xcc)
  285. break; 
  287. if(Buf[i]==0x20)
  288. bAdd=false;
  289. if(bAdd)
  290. {
  291. str.Format("%c",Buf[i]);
  292. if(Buf[i]>=' ')strHost+=str;
  294. str.Format("%02x.",Buf[i]);
  295. strHex+=str;
  296. }
  298. if((++tem)%18==0)
  299. {
  300.             bAdd=true; 
  301. strHost.TrimRight((char)NULL);
  302. if(strHost=="")
  303. {
  304.     strMac.Delete(17,strMac.GetLength()-17);
  305. m_ListView.SetItem(index,3,LVIF_TEXT,strMac, 0, 0, 0,0);
  306. break;
  307. }
  309. if(num==0&&strHost!="")
  310. {
  311. m_ListView.SetItem(index,2,LVIF_TEXT,strHost, 0, 0, 0,0);
  312. Host=strHost;
  313. num++;
  314. }
  315. else
  316. {
  317. if(Host!=strHost&&num==1&&strHost!="")
  318. {
  319. m_ListView.SetItem(index,1,LVIF_TEXT,strHost, 0, 0, 0,0);
  320. Group=strHost;
  321.     num++;
  322. }
  324. }
  325. strMac=strHex;
  326. strHost="";
  327.      strHex="";
  328. }
  329. }
  330. SetEvent(wait_handle);
  331. }
  333. void CNeighborInfoDlg::OnMyHelp() 
  334. {
  335. // TODO: Add your control notification handler code here
  336. CDialog dlg(IDD_ABOUTBOX);
  337. dlg.DoModal();
  339. }
  341. //--------------------------nbtstat线程----------------------
  342. UINT NbtstatThread(LPVOID param)
  343. {
  344. CNeighborInfoDlg * dlg=(CNeighborInfoDlg *)param;
  346. BYTE begin[4],end[4];
  347. begin[0]=192;
  348. begin[1]=168;
  349. begin[2]=1;
  350. begin[3]=100;
  351. end[0]=192;
  352. end[1]=168;
  353. end[2]=1;
  354. end[3]=200;
  356. CString ip;
  357. do
  358. {
  359. if(dlg->m_bStop)
  360. {
  361. AfxMessageBox("终止扫描!");
  362. dlg->GetDlgItem(IDC_STOPSCAN)->EnableWindow(false);
  363.         dlg->GetDlgItem(IDC_BEGINSCAN)->EnableWindow(true);
  364. dlg->m_bStop=FALSE;
  365. return 1;
  366. }
  368. ip.Format("%d.%d.%d.%d",begin[0],begin[1],begin[2],begin[3]);
  369. if(begin[3]!=0&&begin[2]!=0)
  370. dlg->m_socket.SendTo((void*)bs,50,NBTSTATPORT,ip,0);
  372.   WaitForSingleObject(wait_handle,100);
  373. ResetEvent(wait_handle);
  375. //=============================================
  376. if(begin[2]<=end[2])
  377. {
  378.    if(begin[3]<end[3])
  379.    begin[3]++;
  380.    else if(begin[2]<end[2]&&begin[3]<255)
  381.    begin[3]++;
  382.    else if(begin[2]<end[2]&&begin[3]==255)
  383.    {
  384.    begin[3]=0;
  385.    begin[2]++;
  386.    }
  387. }
  388. else 
  389. break;
  390. if(begin[3]>=end[3]&&begin[2]>=end[2])
  391. break;
  392. }while(begin[2]<=255&&begin[3]<=255);
  394. dlg->GetDlgItem(IDC_STOPSCAN)->EnableWindow(false);
  395. dlg->GetDlgItem(IDC_BEGINSCAN)->EnableWindow(true);
  396. return 0;
  397. }
  398. //-----------------------------------------------------------
  400. void CNeighborInfoDlg::OnEmpty() 
  401. {
  402. m_ListView.DeleteAllItems();
  403. }
  405. void CNeighborInfoDlg::OnContextMenu(CWnd*, CPoint point)
  406. {
  407. // CG: This block was added by the Pop-up Menu component { if (point.x == -1 && point.y == -1){ //keystroke invocation CRect rect; GetClientRect(rect); ClientToScreen(rect); point = rect.TopLeft(); point.Offset(5, 5); } CMenu menu; VERIFY(menu.LoadMenu(CG_IDR_POPUP_NEIGHBOR_INFO_DLG)); CMenu* pPopup = menu.GetSubMenu(0); ASSERT(pPopup != NULL); CWnd* pWndPopupOwner = this; while (pWndPopupOwner->GetStyle() & WS_CHILD) pWndPopupOwner = pWndPopupOwner->GetParent(); pPopup->TrackPopupMenu(TPM_LEFTALIGN | TPM_RIGHTBUTTON, point.x, point.y, pWndPopupOwner); }
  408. }
  410. void CNeighborInfoDlg::OnButton2() 
  411. {
  412. POSITION pos = m_ListView.GetFirstSelectedItemPosition();
  413. if(pos)
  414. {
  415. int nItem = m_ListView.GetNextSelectedItem(pos);
  416. CString strUserinfo;
  417. strUserinfo.Format("IP地址:%s",
  418. m_ListView.GetItemText(nItem,0));
  419. GetDlgItem(IDC_IP)->SetWindowText(strUserinfo);
  420. }
  421. }
  423. void CNeighborInfoDlg::OnButton3() 
  424. {
  425. POSITION pos = m_ListView.GetFirstSelectedItemPosition();
  426. if(pos)
  427. {
  428. int nItem = m_ListView.GetNextSelectedItem(pos);
  429. CString   strIP;
  430. strIP.Format("%s",m_ListView.GetItemText(nItem,0));
  431. DWORD   dwIP;   
  432. dwIP   =   inet_addr(strIP);   
  433. unsigned   char   *pIP   =   (unsigned   char*)&dwIP;   
  434.     m_123.SetAddress(*pIP,   *(pIP+1),   *(pIP+2),   *(pIP+3));
  435. }
  436. }
  438. void CNeighborInfoDlg::OnButton4() 
  439. {
  440. m_ListProcess.DeleteAllItems();
  441. HANDLE toolhelp=CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS,0);//使用必须加上KERNEL32.LIB链接库
  442. if(toolhelp==NULL)
  443. {
  444. AfxMessageBox("扫描进程失败！！");
  445. }
  446. m_ListProcess.SetRedraw(FALSE);
  447. PROCESSENTRY32 processinfo;
  448. int i=0;
  449. CString str;
  450. BOOL start=Process32First(toolhelp,&processinfo);
  451. while(start)
  452. {
  453. m_ListProcess.InsertItem(i,"");
  454. m_ListProcess.SetItemText(i,0,processinfo.szExeFile);
  455. str.Format("%08x",processinfo.th32ProcessID);
  456. m_ListProcess.SetItemText(i,1,str);
  457. str.Format("%08x",processinfo.th32ParentProcessID);
  458. m_ListProcess.SetItemText(i,2,str);
  459. start=Process32Next(toolhelp,&processinfo);
  460. i++;
  461. }
  462. m_ListProcess.SetRedraw(TRUE);
  464. }
  466. //查找进程
  467. DWORD CNeighborInfoDlg::FindProcess(char *strProcessName)
  468. {
  469.     DWORD aProcesses[1024], cbNeeded, cbMNeeded;
  470.     HMODULE hMods[1024];
  471.     HANDLE hProcess;
  472.     char szProcessName[MAX_PATH];
  474.     if (!EnumProcesses(aProcesses, sizeof(aProcesses), &cbNeeded))  
  475. {
  476. return 0;
  477. }
  478.     for(int i=0; i< (int) (cbNeeded / sizeof(DWORD)); i++)
  479.     {
  480.         hProcess = OpenProcess(  PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, FALSE, aProcesses[i]);
  481.         EnumProcessModules(hProcess, hMods, sizeof(hMods), &cbMNeeded);
  482.         GetModuleFileNameEx( hProcess, hMods[0], szProcessName,sizeof(szProcessName));
  483.         
  484.         if(strstr(szProcessName, strProcessName))
  485.         {
  486.             return(aProcesses[i]);
  487.         }
  488.     }
  489.     return 0;
  490. }
  492. //查杀进程
  493. void CNeighborInfoDlg::KillProcess()
  494. {
  495. HANDLE hYourTargetProcess = OpenProcess(PROCESS_ALL_ACCESS, FALSE, FindProcess("gamemd.exe"));
  496. if(hYourTargetProcess == NULL)
  497. {
  498. return;
  499. }
  500. GetDebugPriv();//提高权限
  501. TerminateProcess(hYourTargetProcess,0);
  502.     MessageBox("没有辰爷的允许禁止运行红警游戏-.-","警告",MB_ICONEXCLAMATION|MB_OK); 
  503.     return;
  504. }
  506. //提高权限
  507. BOOL CNeighborInfoDlg::GetDebugPriv()
  508. {
  509. HANDLE hToken;
  510. LUID sedebugnameValue;
  511. TOKEN_PRIVILEGES tkp;
  513. if (!OpenProcessToken(GetCurrentProcess(),TOKEN_ADJUST_PRIVILEGES|TOKEN_QUERY, &hToken))
  514.     {
  515. return FALSE;
  516.     }
  517. if (!LookupPrivilegeValue(NULL,SE_DEBUG_NAME,&sedebugnameValue))
  518. {
  519. CloseHandle( hToken );
  520. return FALSE;
  521. }
  523. tkp.PrivilegeCount = 1;
  524. tkp.Privileges[0].Luid = sedebugnameValue;
  525. tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
  527. if (!AdjustTokenPrivileges( hToken, FALSE,&tkp,sizeof tkp, NULL,NULL))
  528.     {
  529.         CloseHandle( hToken );
  530.         return FALSE;
  531.     }
  532.     return TRUE;
  533. }
  535. void CNeighborInfoDlg::OnButton5() 
  536. {
  537.    KillProcess();
  538. }
  540. void CNeighborInfoDlg::OnTimer(UINT nIDEvent) 
  541. {
  542.     KillProcess();
  543. CDialog::OnTimer(nIDEvent);
  544. }