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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 书籍源码 > 查看源码
PortScanDlg.cpp
资源名称：c++时尚编程百例__81-110.rar　[点击查看]
上传用户：qzzxgm
上传日期：2009-12-14
资源大小：1882k
文件大小：13k
源码类别：

书籍源码

开发平台：

Visual C++

PortScanDlg.cpp：源码内容
  1. // PortScanDlg.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "PortScan.h"
  6. #include "PortScanDlg.h"
  8. #include <fcntl.h>
  9. #include <sys/types.h>
  10. #include <sys/stat.h>
  11. #include <io.h>
  13. #ifdef _DEBUG
  14. #define new DEBUG_NEW
  15. #undef THIS_FILE
  16. static char THIS_FILE[] = __FILE__;
  17. #endif
  19. /////////////////////////////////////////////////////////////////////////////
  20. // CAboutDlg dialog used for App About
  22. class CAboutDlg : public CDialog
  23. {
  24. public:
  25. CAboutDlg();
  27. // Dialog Data
  28. //{{AFX_DATA(CAboutDlg)
  29. enum { IDD = IDD_ABOUTBOX };
  30. //}}AFX_DATA
  32. // ClassWizard generated virtual function overrides
  33. //{{AFX_VIRTUAL(CAboutDlg)
  34. protected:
  35. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  36. //}}AFX_VIRTUAL
  38. // Implementation
  39. protected:
  40. //{{AFX_MSG(CAboutDlg)
  41. //}}AFX_MSG
  42. DECLARE_MESSAGE_MAP()
  43. };
  45. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  46. {
  47. //{{AFX_DATA_INIT(CAboutDlg)
  48. //}}AFX_DATA_INIT
  49. }
  51. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  52. {
  53. CDialog::DoDataExchange(pDX);
  54. //{{AFX_DATA_MAP(CAboutDlg)
  55. //}}AFX_DATA_MAP
  56. }
  58. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  59. //{{AFX_MSG_MAP(CAboutDlg)
  60. // No message handlers
  61. //}}AFX_MSG_MAP
  62. END_MESSAGE_MAP()
  64. /////////////////////////////////////////////////////////////////////////////
  65. // CPortScanDlg dialog
  67. CPortScanDlg::CPortScanDlg(CWnd* pParent /*=NULL*/)
  68. : CDialog(CPortScanDlg::IDD, pParent)
  69. {
  70. //{{AFX_DATA_INIT(CPortScanDlg)
  71. // NOTE: the ClassWizard will add member initialization here
  72. //}}AFX_DATA_INIT
  73. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  74. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  76. m_pColumns = new CStringList;
  77. ASSERT(m_pColumns);
  78. m_bSinglePort = TRUE;
  79. m_nMaxAttempts = 1;
  80. m_pStatusList = new CPtrList;
  81. ASSERT(m_pStatusList);
  82. }
  84. void CPortScanDlg::DoDataExchange(CDataExchange* pDX)
  85. {
  86. CDialog::DoDataExchange(pDX);
  87. //{{AFX_DATA_MAP(CPortScanDlg)
  88. DDX_Control(pDX, IDC_PROGRESS, m_cProgress);
  89. DDX_Control(pDX, IDC_LIST_RESULT, m_cResult);
  90. DDX_Control(pDX, IDC_IP_ADDRESS, m_cIP);
  91. DDX_Control(pDX, IDC_EDIT_SINGLE_PORT_TO, m_cPortTo);
  92. DDX_Control(pDX, IDC_EDIT_SINGLE_PORT_FROM, m_cPortFrom);
  93. DDX_Control(pDX, IDC_EDIT_SINGLE_PORT, m_cSinglePort);
  94. DDX_Control(pDX, IDC_EDIT_ATTEMPTS, m_cAttempts);
  95. DDX_Control(pDX, IDC_BUTTON_STOP, m_cBtnStop);
  96. DDX_Control(pDX, IDC_BUTTON_SCAN, m_cBtnScan);
  97. //}}AFX_DATA_MAP
  98. }
  100. BEGIN_MESSAGE_MAP(CPortScanDlg, CDialog)
  101. //{{AFX_MSG_MAP(CPortScanDlg)
  102. ON_WM_SYSCOMMAND()
  103. ON_WM_PAINT()
  104. ON_WM_QUERYDRAGICON()
  105. ON_WM_DESTROY()
  106. ON_BN_CLICKED(IDC_RADIO_SINGLE, OnRadioSingle)
  107. ON_BN_CLICKED(IDC_RADIO_RANGE, OnRadioRange)
  108. ON_BN_CLICKED(IDC_BUTTON_SCAN, OnButtonScan)
  109. ON_BN_CLICKED(IDC_BUTTON_STOP, OnButtonStop)
  110. ON_BN_CLICKED(IDC_BUTTON_SAVE, OnButtonSave)
  111. //}}AFX_MSG_MAP
  112. END_MESSAGE_MAP()
  114. /////////////////////////////////////////////////////////////////////////////
  115. // CPortScanDlg message handlers
  117. BOOL CPortScanDlg::OnInitDialog()
  118. {
  119. CDialog::OnInitDialog();
  121. // Add "About..." menu item to system menu.
  123. // IDM_ABOUTBOX must be in the system command range.
  124. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  125. ASSERT(IDM_ABOUTBOX < 0xF000);
  127. CMenu* pSysMenu = GetSystemMenu(FALSE);
  128. if (pSysMenu != NULL)
  129. {
  130. CString strAboutMenu;
  131. strAboutMenu.LoadString(IDS_ABOUTBOX);
  132. if (!strAboutMenu.IsEmpty())
  133. {
  134. pSysMenu->AppendMenu(MF_SEPARATOR);
  135. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  136. }
  137. }
  139. // Set the icon for this dialog.  The framework does this automatically
  140. //  when the application's main window is not a dialog
  141. SetIcon(m_hIcon, TRUE); // Set big icon
  142. SetIcon(m_hIcon, FALSE); // Set small icon
  144. // TODO: Add extra initialization here
  145. //给列表框加标题栏
  146. AddHeader(_T("IP address"));
  147. AddHeader(_T("Port number"));
  148. AddHeader(_T("Port Status"));
  149. AddHeader(_T("Attempts"));
  150. AddHeader(_T("Remarks"));
  151. ShowHeaders();
  153. //设置扫描单个端口
  154. CheckRadioButton(IDC_RADIO_SINGLE,IDC_RADIO_RANGE,IDC_RADIO_SINGLE);
  156. //设置控件的默认状态
  157. m_cSinglePort.EnableWindow();
  158. m_cPortFrom.EnableWindow(FALSE);
  159. m_cPortTo.EnableWindow(FALSE);
  160. m_cBtnStop.EnableWindow(FALSE);
  161. m_cAttempts.SetWindowText(_T("1"));
  163. return TRUE;  // return TRUE  unless you set the focus to a control
  164. }
  166. void CPortScanDlg::OnSysCommand(UINT nID, LPARAM lParam)
  167. {
  168. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  169. {
  170. CAboutDlg dlgAbout;
  171. dlgAbout.DoModal();
  172. }
  173. else
  174. {
  175. CDialog::OnSysCommand(nID, lParam);
  176. }
  177. }
  179. // If you add a minimize button to your dialog, you will need the code below
  180. //  to draw the icon.  For MFC applications using the document/view model,
  181. //  this is automatically done for you by the framework.
  183. void CPortScanDlg::OnPaint() 
  184. {
  185. if (IsIconic())
  186. {
  187. CPaintDC dc(this); // device context for painting
  189. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  191. // Center icon in client rectangle
  192. int cxIcon = GetSystemMetrics(SM_CXICON);
  193. int cyIcon = GetSystemMetrics(SM_CYICON);
  194. CRect rect;
  195. GetClientRect(&rect);
  196. int x = (rect.Width() - cxIcon + 1) / 2;
  197. int y = (rect.Height() - cyIcon + 1) / 2;
  199. // Draw the icon
  200. dc.DrawIcon(x, y, m_hIcon);
  201. }
  202. else
  203. {
  204. CDialog::OnPaint();
  205. }
  206. }
  208. // The system calls this to obtain the cursor to display while the user drags
  209. //  the minimized window.
  210. HCURSOR CPortScanDlg::OnQueryDragIcon()
  211. {
  212. return (HCURSOR) m_hIcon;
  213. }
  215. void CPortScanDlg::OnDestroy() 
  216. {
  217. //清除扫描结果
  218. if (m_pStatusList)
  219. {
  220. POSITION p = m_pStatusList->GetHeadPosition();
  221. while (p)
  222. {
  223. POSITION temp = p;
  224. DATA* pNode = (DATA*)m_pStatusList->GetNext(p);
  225. m_pStatusList->RemoveAt(temp);
  226. if (pNode)
  227. delete pNode;
  228. }
  229. delete m_pStatusList;
  230. }
  232. CDialog::OnDestroy();
  233. }
  235. //增加列表框标题栏的某一列
  236. BOOL CPortScanDlg::AddColumn(LPCTSTR strItem,int nItem,int nSubItem,int nMask,int nFmt)
  237. {
  238. LV_COLUMN lvc;
  239. lvc.mask = nMask;
  240. lvc.fmt = nFmt;
  241. lvc.pszText = (LPTSTR) strItem;
  242. lvc.cx = m_cResult.GetStringWidth(lvc.pszText) + 25;
  243. if(nMask & LVCF_SUBITEM)
  244. {
  245. if(nSubItem != -1)
  246. lvc.iSubItem = nSubItem;
  247. else
  248. lvc.iSubItem = nItem;
  249. }
  250. return m_cResult.InsertColumn(nItem,&lvc);
  251. }
  253. //在列表框中加一条
  254. BOOL CPortScanDlg::AddItem(int nItem,int nSubItem,LPCTSTR strItem ,int nImageIndex)
  255. {
  256. LV_ITEM lvItem;
  257. lvItem.mask = LVIF_TEXT;
  258. lvItem.iItem = nItem;
  259. lvItem.iSubItem = nSubItem;
  260. lvItem.pszText = (LPTSTR)strItem;
  262. if(nImageIndex != -1)
  263. {
  264. lvItem.mask |= LVIF_IMAGE;
  265. lvItem.iImage |= LVIF_IMAGE;
  266. }
  268. if(nSubItem == 0)
  269. return m_cResult.InsertItem(&lvItem);
  271. return m_cResult.SetItem(&lvItem);
  272. }
  274. //加一列标题栏字符
  275. void CPortScanDlg::AddHeader(LPTSTR hdr)
  276. {
  277. if (m_pColumns)
  278. m_pColumns->AddTail(hdr);
  279. }
  281. //显示列表框标题栏
  282. void CPortScanDlg::ShowHeaders()
  283. {
  284. int nIndex = 0;
  285. POSITION pos = m_pColumns->GetHeadPosition();
  286. while(pos)
  287. {
  288. CString hdr = (CString)m_pColumns->GetNext(pos);
  289. AddColumn(hdr,nIndex++);
  290. }
  291. }
  293. //单击“扫描单个端口”单选按钮
  294. void CPortScanDlg::OnRadioSingle() 
  295. {
  296. m_bSinglePort = TRUE;
  297. m_cSinglePort.EnableWindow();
  298. m_cPortFrom.EnableWindow(FALSE);
  299. m_cPortTo.EnableWindow(FALSE);
  300. }
  302. //单击“扫描多个端口”单选按钮
  303. void CPortScanDlg::OnRadioRange() 
  304. {
  305. m_bSinglePort = FALSE;
  306. m_cSinglePort.EnableWindow(FALSE);
  307. m_cPortFrom.EnableWindow();
  308. m_cPortTo.EnableWindow();
  309. m_cBtnStop.EnableWindow(FALSE);
  310. }
  312. //开始扫描
  313. void CPortScanDlg::OnButtonScan() 
  314. {
  315. CString btnTxt,IP;
  316. UINT nSinglePort;
  317. BYTE f1,f2,f3,f4;
  318. TCHAR temp[10] = "";
  320. //设置进度条初始位置
  321. m_cProgress.SetPos(0);
  323. //清空列表框
  324. m_cResult.DeleteAllItems();
  326. //清空扫描结果链表
  327. POSITION p = m_pStatusList->GetHeadPosition();
  328. while (p)
  329. {
  330. POSITION temp = p;
  331. DATA* pNode = (DATA*)m_pStatusList->GetNext(p);
  332. m_pStatusList->RemoveAt(temp);
  333. if (pNode)
  334. delete pNode;
  335. }
  337. //如果输入没有主机地址
  338. if (m_cIP.IsBlank())
  339. {
  340. MessageBox(_T("Please specify the remote machine's IP address."),
  341. _T("Error"),
  342. MB_OK | MB_ICONEXCLAMATION);
  343. return;
  344. }
  346. //不合理的IP
  347. if (m_cIP.GetAddress(f1,f2,f3,f4) < 4)
  348. {
  349. MessageBox(_T("Please specify the IP address again."),
  350. _T("Invalid IP address"),MB_OK | MB_ICONEXCLAMATION);
  351. return;
  352. }
  353. //获得IP字符串
  354. IP = _itoa(f1,temp,10);
  355. IP += _T('.');
  356. IP += _itoa(f2,temp,10);
  357. IP += _T('.');
  358. IP += _itoa(f3,temp,10);
  359. IP += _T('.');
  360. IP += _itoa(f4,temp,10);
  362. //设置按钮状态
  363. m_cBtnStop.EnableWindow();
  364. m_cBtnScan.EnableWindow(FALSE);
  366. //扫描单个端口
  367. if (m_bSinglePort)
  368. {
  369. CString port;
  370. m_cSinglePort.GetWindowText(port);
  371. m_minPort = m_maxPort = nSinglePort = atoi(port);
  372. }
  373. else//扫描多个端口
  374. {
  375. CString port1,port2;
  376. m_cPortFrom.GetWindowText(port1);
  377. m_cPortTo.GetWindowText(port2);
  378. m_minPort = atoi(port1);
  379. m_maxPort = atoi(port2);
  380. m_cProgress.SetRange32(0,m_maxPort - m_minPort+1);
  381. m_cProgress.SetStep(1);
  382. }
  384. //不合理的端口范围
  385. if (!m_bSinglePort && m_maxPort < m_minPort)
  386. {
  387. MessageBox(_T("The maximum range cannot be less than the minimum one."),
  388. _T("Caution"),
  389. MB_OK | MB_ICONINFORMATION);
  390. return;
  391. }
  393. UINT m_nMaxAttempts = GetDlgItemInt(IDC_EDIT_ATTEMPTS);
  395. for (m_nCounter = m_minPort; m_nCounter <= m_maxPort; m_nCounter++)
  396. {
  397. BOOL bIsOpen = FALSE;
  398. UINT nAttempt = 1;
  400. while(nAttempt <= m_nMaxAttempts && !bIsOpen)
  401. {
  402. //显示扫描状态
  403. TCHAR temp[10]="";
  404. CString str = _T("Trying port# ");
  406. str += itoa(m_nCounter,temp,10);
  407. str += _T(", IP Address=");
  408. str += IP;
  409. str += _T(", Attempt=");
  410. str += itoa(nAttempt,temp,10);
  412. CStatic* pStatic = (CStatic*)GetDlgItem(IDC_STATIC_STATUS);
  413. pStatic->SetWindowText(str);
  414. str.Empty();
  416. //连接主机，测试端口是否打开
  417. bIsOpen = TestConnection(IP,m_nCounter);
  419. //如果是打开的，将结果保存到链表中
  420. if (bIsOpen)
  421. {
  422. DATA* pNode = new DATA;
  423. ASSERT(pNode);
  424. strcpy(pNode->IPAddress,IP.GetBuffer(IP.GetLength()));
  425. strcpy(pNode->port,_itoa(m_nCounter,temp,10));
  426. pNode->bStatus = 1; //open
  427. pNode->nAttempts = nAttempt;
  428. m_pStatusList->AddTail(pNode);
  429. }
  430. nAttempt++;
  431. }
  433. //如果没有被打开，也将结果保存到链表中
  434. if (!bIsOpen)
  435. {
  436. DATA* pNode = new DATA;
  437. ASSERT(pNode);
  438. strcpy(pNode->IPAddress,IP.GetBuffer(IP.GetLength()));
  439. strcpy(pNode->port,_itoa(m_nCounter,temp,10));
  440. pNode->bStatus = 0; //close
  441. pNode->nAttempts = nAttempt-1;
  442. m_pStatusList->AddTail(pNode);
  443. }
  445. MSG message;
  446. if (::PeekMessage(&message,NULL,0,0,PM_REMOVE))
  447. {
  448. ::TranslateMessage(&message);
  449. ::DispatchMessage(&message);
  450. }
  451. m_cProgress.StepIt();
  452. }
  454. //设置扫描完毕状态
  455. CStatic* pStatic = (CStatic*)GetDlgItem(IDC_STATIC_STATUS);
  456. pStatic->SetWindowText((CString)_T("Ready"));
  457. m_cBtnScan.EnableWindow();
  458. m_cBtnStop.EnableWindow(FALSE);
  460. //将扫描结果填充到列表框中
  461. UINT nIndex = 0;
  462. POSITION pos = m_pStatusList->GetHeadPosition();
  463. while (pos)
  464. {
  465. DATA* pNode = (DATA*)m_pStatusList->GetNext(pos);
  466. AddItem(nIndex,0,pNode->IPAddress);
  467. AddItem(nIndex,1,pNode->port);
  468. if (pNode->bStatus)
  469. {
  470. AddItem(nIndex,2,_T("Open"));
  471. AddItem(nIndex,4,_T("*"));
  472. }
  473. else
  474. {
  475. AddItem(nIndex,2,_T("Close"));
  476. AddItem(nIndex,4,_T(" "));
  477. }
  478. AddItem(nIndex++,3,_itoa(pNode->nAttempts,temp,10));
  479. }
  480. }
  482. //测试主机的某个端口是否打开
  483. BOOL CPortScanDlg::TestConnection(CString IP, UINT nPort)
  484. {
  485. //创建套节字
  486. CSocket* pSocket;
  487. pSocket = new CSocket;
  488. ASSERT(pSocket);
  490. if (!pSocket->Create())
  491. {
  492. delete pSocket;
  493. pSocket = NULL;
  494. return FALSE;
  495. }
  497. //连接主机
  498. while (!pSocket->Connect(IP , nPort))
  499. {
  500. delete pSocket;
  501. pSocket = NULL;
  502. return FALSE;
  503. }
  505. //清除套节字
  506. pSocket->Close();
  507. delete pSocket;
  508. return TRUE;
  509. }
  511. //停止扫描
  512. void CPortScanDlg::OnButtonStop() 
  513. {
  514. //设置控件状态
  515. m_nCounter = m_maxPort+1;
  516. m_cBtnStop.EnableWindow(FALSE);
  517. m_cBtnScan.EnableWindow();
  518. CStatic* pStatic = (CStatic*)GetDlgItem(IDC_STATIC_STATUS);
  519. pStatic->SetWindowText((CString)_T("Ready"));
  520. }
  522. //保存扫描结果到文本文件中
  523. void CPortScanDlg::OnButtonSave() 
  524. {
  525. //构造文件对话框
  526. CFileDialog* pDlg = new CFileDialog(FALSE,
  527. _T("txt"),NULL,
  528. OFN_OVERWRITEPROMPT | 
  529. OFN_EXPLORER | OFN_LONGNAMES,
  530. _T("Scanned ports files (*.txt)"),this);
  531. ASSERT(pDlg);
  532. if (pDlg->DoModal() == IDOK)
  533. {
  534. int nHandle,retVal;
  535. //打开文件
  536. nHandle = _open(pDlg->GetPathName(),_O_BINARY | _O_CREAT | _O_TRUNC | _O_RDWR);
  537. if (nHandle == -1)
  538. {
  539. MessageBox(_T("Unable to open output file to write."),
  540. _T("Error"),
  541. MB_OK | MB_ICONEXCLAMATION);
  542. delete pDlg;
  543. return;
  544. }
  546. //保存结果
  547. POSITION pos = m_pStatusList->GetHeadPosition();
  548. while (pos)
  549. {
  550. CString buffer;
  551. DATA* pNode = (DATA*)m_pStatusList->GetNext(pos);
  552. buffer = pNode->IPAddress;
  553. buffer += _T(',');
  554. buffer += pNode->port;
  555. buffer += _T(',');
  557. if (pNode->bStatus)
  558. buffer += _T("Open");
  559. else
  560. buffer += _T("Close");
  562. buffer += _T("rn");
  563. retVal = _write(nHandle,
  564. (void*)buffer.GetBuffer(buffer.GetLength()), 
  565. buffer.GetLength());
  567. if (retVal != buffer.GetLength())
  568. {
  569. MessageBox(_T("An error occured while writing records."),
  570. _T("Error"),MB_OK | MB_ICONEXCLAMATION);
  571. delete pDlg;
  572. return;
  573. }
  574. buffer.Empty();
  575. }
  576. //关闭文件
  577. _close(nHandle);
  578. }
  579. delete pDlg;
  580. }