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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 系统/网络安全 > 扫描程序 > 查看源码
PortScanDlg.cpp
资源名称：PortScan0623.rar　[点击查看]
上传用户：zhyl198712
上传日期：2013-02-03
资源大小：35k
文件大小：11k
源码类别：

扫描程序

开发平台：

Visual C++

PortScanDlg.cpp：源码内容
  1. // PortScanDlg.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "PortScan.h"
  6. #include "PortScanDlg.h"
  8. #ifdef _DEBUG
  9. #define new DEBUG_NEW
  10. #undef THIS_FILE
  11. static char THIS_FILE[] = __FILE__;
  12. #endif
  14. //#include <Winsock2.h>
  15. /////////////////////////////////////////////////////////////////////////////
  16. // CAboutDlg dialog used for App About
  17. int v; //
  18. UINT m_nCounter; //循环中的端口值
  19. BOOL m_bIsSinglePort; //是不是单端口
  20. UINT m_nPortFrom,m_nPortTo; //端口范围
  21. CString m_sIP;      //ip地址
  23. int listCounter; //列表框的纪录行数
  24. int m_nThreadNumber; //开启线程数
  25. struct thread m_thread;
  27. class CAboutDlg : public CDialog
  28. {
  29. public:
  30. CAboutDlg();
  32. // Dialog Data
  33. //{{AFX_DATA(CAboutDlg)
  34. enum { IDD = IDD_ABOUTBOX };
  35. //}}AFX_DATA
  37. // ClassWizard generated virtual function overrides
  38. //{{AFX_VIRTUAL(CAboutDlg)
  39. protected:
  40. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  41. //}}AFX_VIRTUAL
  43. // Implementation
  44. protected:
  45. //{{AFX_MSG(CAboutDlg)
  46. //}}AFX_MSG
  47. DECLARE_MESSAGE_MAP()
  48. };
  50. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  51. {
  52. //{{AFX_DATA_INIT(CAboutDlg)
  53. //}}AFX_DATA_INIT
  54. }
  57. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  58. {
  59. CDialog::DoDataExchange(pDX);
  60. //{{AFX_DATA_MAP(CAboutDlg)
  61. //}}AFX_DATA_MAP
  62. }
  64. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  65. //{{AFX_MSG_MAP(CAboutDlg)
  66. // No message handlers
  67. //}}AFX_MSG_MAP
  68. END_MESSAGE_MAP()
  71. /////////////////////////////////////////////////////////////////////////////
  72. // CPortScanDlg dialog
  74. CPortScanDlg::CPortScanDlg(CWnd* pParent /*=NULL*/)
  75. : CDialog(CPortScanDlg::IDD, pParent)
  76. {
  77. //{{AFX_DATA_INIT(CPortScanDlg)
  78. //}}AFX_DATA_INIT
  79. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  80. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  81. m_bIsSinglePort=TRUE;
  82. }
  84. CPortScanDlg::~CPortScanDlg()
  85. {
  86. WSACleanup();
  87. }
  88. void CPortScanDlg::DoDataExchange(CDataExchange* pDX)
  89. {
  90. CDialog::DoDataExchange(pDX);
  91. //{{AFX_DATA_MAP(CPortScanDlg)
  92. DDX_Control(pDX, IDC_THREADNUMBER, m_threadNumber);
  93. DDX_Control(pDX, IDC_BUTTONSTOP, m_buttonStop);
  94. DDX_Control(pDX, IDC_BUTTONSCAN, m_buttonStart);
  95. DDX_Control(pDX, IDC_STATIC_PROMPT, m_prompt);
  96. DDX_Control(pDX, IDC_PROGRESS_SCAN, m_progressScan);
  97. DDX_Control(pDX, IDC_EDIT_PORT_FROM, m_portFrom);
  98. DDX_Control(pDX, IDC_EDIT_PORT_TO, m_portTo);
  99. DDX_Control(pDX, IDC_EDIT_SINGLE_PORT, m_singlePort);
  100. DDX_Control(pDX, IDC_IP_ADDRESS, m_IP);
  101. DDX_Control(pDX, IDC_LISTINFO, m_listInfo);
  102. //}}AFX_DATA_MAP
  103. }
  107. BEGIN_MESSAGE_MAP(CPortScanDlg, CDialog)
  108. ON_MESSAGE(WM_USER_PRINT_START,OnPrintStart)
  109. //{{AFX_MSG_MAP(CPortScanDlg)
  110. ON_WM_SYSCOMMAND()
  111. ON_WM_PAINT()
  112. ON_WM_QUERYDRAGICON()
  113. ON_BN_CLICKED(IDC_RADIO_SINGLE, OnRadioSingle)
  114. ON_BN_CLICKED(IDC_RADIO_MULTI, OnRadioMulti)
  115. ON_BN_CLICKED(IDC_BUTTONSCAN, OnButtonscan)
  116. ON_BN_CLICKED(IDC_BUTTONSTOP, OnButtonstop)
  117. ON_BN_CLICKED(IDC_BUTTONABOUT, OnButtonabout)
  118. //}}AFX_MSG_MAP
  119. END_MESSAGE_MAP()
  120. /////////////////////////////////////////////////////////////////////////////
  121. // CPortScanDlg message handlers
  123. BOOL CPortScanDlg::OnInitDialog()
  124. {
  125. CDialog::OnInitDialog();
  127. // Add "About..." menu item to system menu.
  129. // IDM_ABOUTBOX must be in the system command range.
  130. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  131. ASSERT(IDM_ABOUTBOX < 0xF000);
  133. CMenu* pSysMenu = GetSystemMenu(FALSE);
  134. if (pSysMenu != NULL)
  135. {
  136. CString strAboutMenu;
  137. strAboutMenu.LoadString(IDS_ABOUTBOX);
  138. if (!strAboutMenu.IsEmpty())
  139. {
  140. pSysMenu->AppendMenu(MF_SEPARATOR);
  141. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  142. }
  143. }
  145. // Set the icon for this dialog.  The framework does this automatically
  146. //  when the application's main window is not a dialog
  147. SetIcon(m_hIcon, TRUE); // Set big icon
  148. SetIcon(m_hIcon, FALSE); // Set small icon
  149. //单选按钮
  150. CheckRadioButton(IDC_RADIO_SINGLE,IDC_RADIO_MULTI,IDC_RADIO_SINGLE);
  151. m_portFrom.EnableWindow(FALSE);
  152. m_portTo.EnableWindow(FALSE);
  153. m_buttonStop.EnableWindow(FALSE);
  154. //初始化信息列表框
  155. m_listInfo.InsertColumn(0,"IP地址",LVCFMT_LEFT,0,0);
  156. m_listInfo.InsertColumn(1,"开放的端口",LVCFMT_LEFT,0,0);
  157. m_listInfo.SetColumnWidth(0,150);
  158. m_listInfo.SetColumnWidth(1,150);
  161. //默认的IP地址
  162. m_IP.SetAddress(172,21,41,200);
  163. //默认的线程个数
  164. m_threadNumber.SetWindowText("200");
  165. //开始winsock
  166. WSADATA wsaData;  
  167.     if(WSAStartup(0x0002, &wsaData ))
  168.     {
  169.         AfxMessageBox("Init TCP/IP stack error!");
  170.         return 1;
  171.     }
  172.     if(wsaData.wVersion != 0x0002)
  173.     {
  174.         printf("Winsock version is incorrect!");
  175.         WSACleanup();
  176.         return 1;
  177.     }
  179. return TRUE;  // return TRUE  unless you set the focus to a control
  180. }
  182. void CPortScanDlg::OnSysCommand(UINT nID, LPARAM lParam)
  183. {
  184. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  185. {
  186. CAboutDlg dlgAbout;
  187. dlgAbout.DoModal();
  188. }
  189. else
  190. {
  191. CDialog::OnSysCommand(nID, lParam);
  192. }
  193. }
  195. // If you add a minimize button to your dialog, you will need the code below
  196. //  to draw the icon.  For MFC applications using the document/view model,
  197. //  this is automatically done for you by the framework.
  199. void CPortScanDlg::OnPaint() 
  200. {
  201. if (IsIconic())
  202. {
  203. CPaintDC dc(this); // device context for painting
  205. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  207. // Center icon in client rectangle
  208. int cxIcon = GetSystemMetrics(SM_CXICON);
  209. int cyIcon = GetSystemMetrics(SM_CYICON);
  210. CRect rect;
  211. GetClientRect(&rect);
  212. int x = (rect.Width() - cxIcon + 1) / 2;
  213. int y = (rect.Height() - cyIcon + 1) / 2;
  215. // Draw the icon
  216. dc.DrawIcon(x, y, m_hIcon);
  217. }
  218. else
  219. {
  220. CDialog::OnPaint();
  221. }
  222. }
  224. // The system calls this to obtain the cursor to display while the user drags
  225. //  the minimized window.
  226. HCURSOR CPortScanDlg::OnQueryDragIcon()
  227. {
  228. return (HCURSOR) m_hIcon;
  229. }
  231. void CPortScanDlg::OnRadioSingle() 
  232. {
  233. m_bIsSinglePort=TRUE;
  234. m_singlePort.EnableWindow(TRUE);
  235. m_portFrom.EnableWindow(FALSE);
  236. m_portTo.EnableWindow(FALSE);
  238. }
  240. void CPortScanDlg::OnRadioMulti() 
  241. {
  242. m_bIsSinglePort=FALSE;
  243. m_singlePort.EnableWindow(FALSE);
  244. m_portFrom.EnableWindow(TRUE);
  245. m_portTo.EnableWindow(TRUE);
  246. }
  248. void CPortScanDlg::OnButtonscan() 
  249. {
  251. CString ip;
  253. BYTE f1,f2,f3,f4;
  254. char temp[10];
  256. m_listInfo.DeleteAllItems();
  259. listCounter=0;
  260. /****获取IP地址****/
  261. if(m_IP.IsBlank())
  262. {
  263. MessageBox("请输入IP地址",_T("错误"),MB_ICONWARNING|MB_OK);
  264. return;
  265. }
  267. if(m_IP.GetAddress(f1,f2,f3,f4)<4)
  268. {
  269. MessageBox("请输入完整的IP地址",_T("错误"),MB_ICONWARNING|MB_OK);
  270. return;
  271. }
  272. ip=itoa(f1,temp,10);
  273. ip+=_T(".");
  274. ip+=itoa(f2,temp,10);
  275. ip+=_T(".");
  276. ip+=itoa(f3,temp,10);
  277. ip+=_T(".");
  278. ip+=itoa(f4,temp,10);
  279. m_sIP=ip;
  280. /**********************/
  282. /***获取端口地址****/
  283. if(m_bIsSinglePort==TRUE)
  284. {
  285. CString strPort;
  286. m_singlePort.GetWindowText(strPort);
  287. m_nPortFrom=m_nPortTo=atoi(strPort);
  288. }
  289. else
  290. {
  291. CString strPort;
  292. m_portFrom.GetWindowText(strPort);
  293. m_nPortFrom=atoi(strPort);
  294. m_portTo.GetWindowText(strPort);
  295. m_nPortTo=atoi(strPort);
  296. }
  297. if(m_nPortFrom>m_nPortTo)
  298. {
  299. MessageBox("开始端口要小于结束端口",_T("错误"),MB_ICONWARNING|MB_OK);
  300. return;
  301. }
  302. /**********************/
  305. /*********取得开启线程数*****/
  306. char strThreadNumber[10];
  307. m_threadNumber.GetWindowText(strThreadNumber,10);
  308. m_nThreadNumber=atoi(strThreadNumber);
  309. if(m_nThreadNumber>1000)
  310. {
  311. AfxMessageBox("最大线程数不能超过1000");
  312. return;
  313. }
  315. m_buttonStart.EnableWindow(FALSE);
  316. m_buttonStop.EnableWindow(TRUE);
  318. /*********设置进度条**************/
  319. m_progressScan.SetRange32(0,m_nPortTo-m_nPortFrom+1);
  320. m_progressScan.SetStep(1);
  322. /********** v是批量创建线程的次数*************/
  324. if(m_nPortFrom==m_nPortTo)
  325. {
  326. v=1;
  327. }
  328. else
  329. {
  330. v=(m_nPortTo-m_nPortFrom+1)/m_nThreadNumber;
  331. if((m_nPortTo-m_nPortFrom+1)%m_nThreadNumber > 0)
  332. {
  333. v++;
  334. }
  335. }
  336. m_thread.m_hwnd=this->m_hWnd;
  337. m_thread.ip=m_sIP;
  339. m_nCounter=m_nPortFrom;
  341. AfxBeginThread((AFX_THREADPROC)threadM,NULL);
  343. }
  345. //一个独立线程，主要是处理循环，与窗体脱离，提高窗体的反应能力
  346. UINT threadM(LPVOID pParam)
  347. {
  349. CPortScanDlg *dlg=(CPortScanDlg*)AfxGetApp()->GetMainWnd();
  350. /*********创建互斥量************/
  351. HANDLE hCounter=NULL;
  352. if( (hCounter=OpenMutex(MUTEX_ALL_ACCESS,FALSE,"sam sp 44"))==NULL)
  353. {
  354. //如果没有其他进程创建这个互斥量，则重新创建
  355. hCounter = CreateMutex(NULL,FALSE,"sam sp 44");
  356. }
  358. CWinThread *pT[1001];
  359. HANDLE hThread[1001];
  361. for(int i=0;i<v;i++)
  362. {
  363. int nThreadCounter=0;//每次批量创建的线程实际个数,最后一次是一个余数值
  364. int k;
  365. for(k=0;k<m_nThreadNumber;k++)
  366. {
  368. if(m_nCounter>m_nPortTo)
  369. {
  370. break;
  371. }
  372. //设置扫描进度提示
  373. char temp[10];
  374. memset(temp,0,10);
  375. itoa(m_nCounter,temp,10);
  376. CString strTemp=(CString)temp;
  377. dlg->m_prompt.SetWindowText("端口:"+strTemp);
  379. //内循环计数
  380. nThreadCounter++;
  381. dlg->m_progressScan.StepIt();
  383. pT[k]=AfxBeginThread((AFX_THREADPROC)threadA,&m_thread);
  384. hThread[k]=pT[k]->m_hThread;
  386. Sleep(5);
  387. }
  389. hThread[k]=NULL;//非常重要,如果不加这一句的话,会经常出错,有点奇怪
  390. //如果k=0，表示没有开启线程
  391. if(k!=0)
  392. {
  393. //WaitForMultipleObjects(nThreadCounter,hThread,TRUE,INFINITE);
  394. WaitForMultipleObjects(nThreadCounter,hThread,TRUE,500);
  395. }
  397. }
  398. CloseHandle(hCounter);
  399. //设置提示
  400. dlg->m_buttonStart.EnableWindow(TRUE);
  401. dlg->m_buttonStop.EnableWindow(FALSE);
  402. dlg->m_prompt.SetWindowText(_T(""));
  403. dlg->m_progressScan.SetPos(0);
  404. AfxEndThread(0);
  406. return 0;
  407. }
  409. //核心线程体
  410. UINT threadA(LPVOID pParam)
  411. {
  412. struct thread *threada=(struct thread*)pParam;
  413. //在内部重新打开互斥量
  414. HANDLE hCounterIn=OpenMutex(MUTEX_ALL_ACCESS,FALSE,"sam sp 44");
  415. WaitForSingleObject(hCounterIn,INFINITE);
  416. UINT v=m_nCounter;
  417. m_nCounter++;
  418. ReleaseMutex(hCounterIn);
  419. CloseHandle(hCounterIn);
  420.  
  421. struct sockaddr_in sin;
  422.   
  423. SOCKET sd;
  424.     int IpPort;
  425. char IpAddr[16];
  426. strcpy(IpAddr,threada->ip);
  427. IpPort=v;
  429.     // Create the local socket
  430.     if ((sd = socket (AF_INET, SOCK_STREAM, IPPROTO_IP)) == INVALID_SOCKET) {
  431.         printf("Create socket error!");
  432.         return 1;
  433.     }
  435.     // Connect to the victim IP Address
  436.     sin.sin_family=AF_INET;
  437.     sin.sin_addr.s_addr=inet_addr(IpAddr);
  438.     sin.sin_port=htons((short)IpPort);
  439.     if (connect (sd, (struct sockaddr *)&sin, sizeof (sin)) == SOCKET_ERROR) {
  440.         printf("Connect the remote IP error!");
  441.         closesocket (sd);
  442.         return 1;
  443.     }
  444. else
  445. {
  446. ::PostMessage(threada->m_hwnd,WM_USER_PRINT_START,v,0);
  447. }
  448.     closesocket (sd);
  449. return 0;
  450. }
  452. LRESULT CPortScanDlg::OnPrintStart(WPARAM wParam,LPARAM lParam)
  453. {
  455. int port=(int)wParam;
  456. char temp[10];
  457. memset(temp,0,10);
  458. itoa(port,temp,10);
  459. m_listInfo.InsertItem(listCounter,m_sIP,0);
  460. m_listInfo.SetItemText(listCounter,1,temp);
  461. listCounter++;
  462. return 0;
  463. }
  467. void CPortScanDlg::OnButtonstop() 
  468. {
  469. m_nCounter=m_nPortTo+1;
  470. m_buttonStart.EnableWindow(TRUE);
  471. m_buttonStop.EnableWindow(FALSE);
  473. }
  475. void CPortScanDlg::OnButtonabout() 
  476. {
  477. CAboutDlg dlg;
  478. dlg.DoModal();
  479. }