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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 系统/网络安全 > 查看源码
MySnifferDlg.cpp
资源名称：little sniffer.rar　[点击查看]
上传用户：bjaaa555
上传日期：2009-10-18
资源大小：33k
文件大小：15k
源码类别：

系统/网络安全

开发平台：

Visual C++

MySnifferDlg.cpp：源码内容
  1. // MySnifferDlg.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "MySniffer.h"
  6. #include "MySnifferDlg.h"
  8. #ifdef _DEBUG
  9. #define new DEBUG_NEW
  10. #undef THIS_FILE
  11. static char THIS_FILE[] = __FILE__;
  12. #endif
  14. /////////////////////////////////////////////////////////////////////////////
  15. // CAboutDlg dialog used for App About
  17. PROTN2T aOfProto[ PROTO_NUM + 1] = 
  18. {  
  19. { IPPROTO_IP   , "IP" },
  20. { IPPROTO_ICMP , "ICMP" },  
  21. { IPPROTO_IGMP , "IGMP" }, 
  22. { IPPROTO_GGP  , "GGP" },  
  23. { IPPROTO_TCP  , "TCP" },  
  24. { IPPROTO_PUP  , "PUP" },  
  25. { IPPROTO_UDP  , "UDP" },  
  26. { IPPROTO_IDP  , "IDP" },  
  27. { IPPROTO_ND   , "NP"  },  
  28. { IPPROTO_RAW  , "RAW" },  
  29. { IPPROTO_MAX  , "MAX" },
  30. { NULL , "" } 
  31. };  
  33. class CAboutDlg : public CDialog
  34. {
  35. public:
  36. CAboutDlg();
  38. // Dialog Data
  39. //{{AFX_DATA(CAboutDlg)
  40. enum { IDD = IDD_ABOUTBOX };
  41. //}}AFX_DATA
  43. // ClassWizard generated virtual function overrides
  44. //{{AFX_VIRTUAL(CAboutDlg)
  45. protected:
  46. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  47. //}}AFX_VIRTUAL
  49. // Implementation
  50. protected:
  51. //{{AFX_MSG(CAboutDlg)
  52. //}}AFX_MSG
  53. DECLARE_MESSAGE_MAP()
  54. };
  56. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  57. {
  58. //{{AFX_DATA_INIT(CAboutDlg)
  59. //}}AFX_DATA_INIT
  60. }
  62. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  63. {
  64. CDialog::DoDataExchange(pDX);
  65. //{{AFX_DATA_MAP(CAboutDlg)
  66. //}}AFX_DATA_MAP
  67. }
  69. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  70. //{{AFX_MSG_MAP(CAboutDlg)
  71. // No message handlers
  72. //}}AFX_MSG_MAP
  73. END_MESSAGE_MAP()
  75. /////////////////////////////////////////////////////////////////////////////
  76. // CMySnifferDlg dialog
  78. CMySnifferDlg::CMySnifferDlg(CWnd* pParent /*=NULL*/)
  79. : CDialog(CMySnifferDlg::IDD, pParent)
  80. {
  81. //{{AFX_DATA_INIT(CMySnifferDlg)
  82. // NOTE: the ClassWizard will add member initialization here
  83. //}}AFX_DATA_INIT
  84. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  85. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  87. m_threadID = 0;
  88. }
  90. void CMySnifferDlg::DoDataExchange(CDataExchange* pDX)
  91. {
  92. CDialog::DoDataExchange(pDX);
  93. //{{AFX_DATA_MAP(CMySnifferDlg)
  94. DDX_Control(pDX, IDC_BUTTON1, m_Start);
  95. DDX_Control(pDX, IDC_LIST2, m_ListCtrl);
  96. //}}AFX_DATA_MAP
  97. }
  99. BEGIN_MESSAGE_MAP(CMySnifferDlg, CDialog)
  100. //{{AFX_MSG_MAP(CMySnifferDlg)
  101. ON_WM_SYSCOMMAND()
  102. ON_WM_PAINT()
  103. ON_WM_QUERYDRAGICON()
  104. ON_BN_CLICKED(IDC_BUTTON1, OnButton1)
  105. //}}AFX_MSG_MAP
  106. END_MESSAGE_MAP()
  108. /////////////////////////////////////////////////////////////////////////////
  109. // CMySnifferDlg message handlers
  111. BOOL CMySnifferDlg::OnInitDialog()
  112. {
  113. CDialog::OnInitDialog();
  115. // Add "About..." menu item to system menu.
  117. // IDM_ABOUTBOX must be in the system command range.
  118. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  119. ASSERT(IDM_ABOUTBOX < 0xF000);
  121. CMenu* pSysMenu = GetSystemMenu(FALSE);
  122. if (pSysMenu != NULL)
  123. {
  124. CString strAboutMenu;
  125. strAboutMenu.LoadString(IDS_ABOUTBOX);
  126. if (!strAboutMenu.IsEmpty())
  127. {
  128. pSysMenu->AppendMenu(MF_SEPARATOR);
  129. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  130. }
  131. }
  133. // Set the icon for this dialog.  The framework does this automatically
  134. //  when the application's main window is not a dialog
  135. SetIcon(m_hIcon, TRUE); // Set big icon
  136. SetIcon(m_hIcon, FALSE); // Set small icon
  138. // TODO: Add extra initialization here
  139. WORD wVersionRequested;
  140. WSADATA wsaData;
  141. int err;
  143. CHAR szHostName[128] = {0};
  144. HOSTENT* pHost = NULL;
  145. CHAR* pszIp = NULL;
  146. int iNum = 0;
  148. wVersionRequested = MAKEWORD(2, 0);
  149. err = WSAStartup(wVersionRequested, &wsaData);
  150. if(err != 0)
  151. {
  152. AfxMessageBox("Load WinSock DLL failed!");
  153. exit(1);
  154. }
  156. if(gethostname(szHostName, 128) == 0)
  157. {
  158. pHost = gethostbyname(szHostName);
  159. if(pHost != NULL)
  160. {
  161. pszIp = inet_ntoa(*(in_addr*)pHost->h_addr_list[iNum]);
  162. m_ipsource = inet_addr(pszIp);
  163. }
  164. else AfxMessageBox("pszIp = NULL");
  165. }
  166. else AfxMessageBox("can't find host name!");
  168. DWORD dwStyle = GetWindowLong(m_ListCtrl.GetSafeHwnd(), GWL_STYLE);
  169. dwStyle &= ~LVS_TYPEMASK;
  170. dwStyle |= LVS_REPORT;
  171. SetWindowLong(m_ListCtrl.GetSafeHwnd(), GWL_STYLE, dwStyle);
  173. m_ListCtrl.InsertColumn(0, "数据", LVCFMT_LEFT, 525);
  174. m_ListCtrl.InsertColumn(0, "大小", LVCFMT_LEFT, 80);
  175. m_ListCtrl.InsertColumn(0, "目的端口", LVCFMT_LEFT, 60);
  176. m_ListCtrl.InsertColumn(0, "目的地址", LVCFMT_LEFT, 100);
  177. m_ListCtrl.InsertColumn(0, "源端口", LVCFMT_LEFT, 60);
  178. m_ListCtrl.InsertColumn(0, "源地址", LVCFMT_LEFT, 100);
  179. m_ListCtrl.InsertColumn(0, "协议", LVCFMT_LEFT, 40);
  181. ::SendMessage(m_ListCtrl.m_hWnd, LVM_SETEXTENDEDLISTVIEWSTYLE,
  182. LVS_EX_FULLROWSELECT, LVS_EX_FULLROWSELECT);
  184. SetWindowText("My Sample Sniffer");
  186. return TRUE;  // return TRUE  unless you set the focus to a control
  187. }
  189. char *get_proto_name(unsigned char proto)
  190. {
  191. BOOL bFound = FALSE;
  192. for(int i=0; i < PROTO_NUM; i++)
  193. {
  194. if(aOfProto[i].proto == proto)
  195. {
  196. bFound = TRUE;
  197. break;
  198. }
  199. }
  200. if(bFound)
  201. return aOfProto[i].pprototext;
  202. return aOfProto[PROTO_NUM].pprototext;
  203. }
  205. UINT threadFunc ( LPVOID p )
  206. {
  207.   CMySnifferDlg *pDlg = static_cast<CMySnifferDlg *>(p) ;
  208. char  buf [1000] , *bufwork ;
  209. MSG   msg ;
  210. int   iRet ;
  211. DWORD dwErr ;
  212.     char  *pSource , *pDest ;
  213. IPHEADER *pIpHeader ;
  214. in_addr ina ;
  215. char   szSource [16] , szDest[16] , szErr [ 50 ];
  216. char *pLastBuf = NULL ;
  217. /*-------------------zhuwei add(2002.11.9)-----------------------------*/
  218. int HdrLen, totallen;
  219. WORD sourport, destport;
  221. //struct IPPacketHead *pIPHead;
  222. struct TCPPacketHead *pTCPHead;
  223. //struct ARPPacket *pARPHead;
  224. struct ICMPPacketHead *pICMPHead;
  225. struct UDPPacketHead *pUDPHead;
  226. BYTE *pdata = NULL;
  228. /*---------------------------------------------------------------------*/
  230.     PeekMessage(&msg, NULL, WM_USER, WM_USER, PM_NOREMOVE) ; // Force to make the queue
  231.   pDlg->m_threadID = GetCurrentThreadId() ;
  233. while( TRUE )
  234. {
  235.         if( PeekMessage( &msg , 0 , WM_CLOSE,WM_CLOSE,PM_NOREMOVE ) )
  236. {
  237.           closesocket( pDlg->m_s ) ;
  238.           pDlg->m_threadID = 0 ;
  239. // Only after you see Next message you can press on Start button 
  240. pDlg->m_Start.EnableWindow(TRUE) ; 
  241. break ;
  242. }
  243. memset( buf , 0 , sizeof(buf) ) ;
  244. iRet =  recv( pDlg->m_s , buf , sizeof( buf ) , 0 ) ;
  245. if( iRet == SOCKET_ERROR )
  246. {
  247. dwErr = WSAGetLastError() ;
  248. sprintf( szErr , "Error recv() = %ld " , dwErr ) ;
  249. continue ;
  250. }
  251. else
  252. if( *buf )
  253. {
  254. // check IP here
  255. //  sprintf( szErr , "recv() = %ld " , iRet ) ;
  256. // AfxMessageBox( szErr ) ;
  257. bufwork   = buf ;
  258.          pIpHeader = (IPHEADER *)bufwork ;
  259.             WORD iLen = ntohs(pIpHeader->total_len) ;
  260. while( TRUE )
  261. {
  262. if( iLen <= iRet )
  263. {
  264.     //if( pDlg->IsIpofHost( pIpHeader ) )
  265. { // Form the line for m_list
  266.   ina.S_un.S_addr = pIpHeader->sourceIP ;
  267. pSource = inet_ntoa( ina ) ;
  268. strcpy( szSource , pSource ) ;
  269. ina.S_un.S_addr = pIpHeader->destIP ;
  270. pDest = inet_ntoa( ina ) ;
  271. strcpy( szDest , pDest ) ;
  272. CString str, strProto, strSourPort, strDestPort, strData, strSize;
  274. /*
  275. if( pIpHeader->sourceIP == ( pDlg->m_ipcheckedhost ? 
  276.            pDlg->m_ipcheckedhost : pDlg->m_iphostsource ) )
  277. strIP.Format("%s -> %s len = %d ttl=%d proto=%s" , szSource , szDest ,
  278.   iLen , pIpHeader->ttl , get_proto_name( pIpHeader->proto ) ) ;
  279. else
  280. strIP.Format("%s -> %s len = %d ttl=%d proto=%s" , szSource , szDest ,
  281. iLen , pIpHeader->ttl , get_proto_name( pIpHeader->proto ) ) ;
  282. */
  284. strProto = get_proto_name( pIpHeader->proto );
  286. /*-------------------zhuwei add(2002.11.9)-----------------------------*/
  287. HdrLen = pIpHeader->header_len&0xf;
  288. HdrLen *= 4;
  289. totallen = ntohs(pIpHeader->total_len);
  290. totallen-=HdrLen;
  291. switch(pIpHeader->proto)
  292. {
  293. case IPPROTO_ICMP:
  294. {
  295. pICMPHead=(struct ICMPPacketHead *)(buf+HdrLen);
  296. //strL4.Format(" type:%d code:%dn",pICMPHead->Type,pICMPHead->Code);
  297. strSourPort = "-";
  298. strDestPort = "-";
  299. pdata=((BYTE *)pICMPHead)+ICMP_HEAD_LEN;
  300. totallen -= ICMP_HEAD_LEN;
  301. break;
  302. }
  303. case IPPROTO_TCP:
  304. {
  305. pTCPHead=(struct TCPPacketHead *)(buf+HdrLen);
  306. sourport = ntohs(pTCPHead->SourPort);
  307. destport = ntohs(pTCPHead->DestPort);
  308. //strL4.Format(" sour port:%d,dest port:%d",sourport,destport);
  309. strSourPort.Format("%d",sourport);
  310. strDestPort.Format("%d",destport);
  311. HdrLen = (pTCPHead->HLen)>>4;  //in fact only 4 bits
  312. HdrLen *= 4;
  313. pdata=((BYTE *)pTCPHead)+HdrLen;
  314. totallen -= HdrLen;
  315. break;
  316. }
  317. case IPPROTO_UDP:
  318. {
  319. pUDPHead=(struct UDPPacketHead *)(buf+HdrLen);
  320. sourport = ntohs(pUDPHead->SourPort);
  321. destport = ntohs(pUDPHead->DestPort);
  322. //strL4.Format(" sour port:%d,dest port:%d",sourport,destport);
  323. strSourPort.Format("%d",sourport);
  324. strDestPort.Format("%d",destport);
  325. pdata=((BYTE *)pUDPHead)+UDP_HEAD_LEN;
  326. totallen -= UDP_HEAD_LEN;
  327. break;
  328. }
  329. }
  331. if(pIpHeader->proto == IPPROTO_ICMP)
  332. strData.Format("type:%d code:%d data:%s",pICMPHead->Type,pICMPHead->Code,pdata);
  333. else strData.Format("  %s",pdata);
  335. strSize.Format("%d",totallen);
  337. //str = strIP+strL4+strData;
  338. pDlg->AddData(strProto,szSource,strSourPort,szDest,strDestPort,strSize,strData);
  339. /*---------------------------------------------------------------------*/
  340.    //pDlg->m_list.AddString( str ) ;  /* zhuwei add(2002.11.9) comment */
  341. //pDlg->m_list.InsertString(0,str);  /* zhuwei add(2002.11.9) */
  343. // Sleep stabilize work of list , otherwise  sometimes pressing on scroll
  344. // cased close of program
  346. //Sleep(50) ; /* zhuwei add(2002.11.9) comment */
  347. }
  348. if( iLen < iRet )
  349. {
  350. iRet -= iLen ;
  351. bufwork  += iLen ;
  352.          pIpHeader = (IPHEADER *)bufwork ;
  353. }
  354. else
  355. break ; // pIpHeader->total_len == iRet and go out
  356. }
  357. else
  358. { // read last part of buf. I wrote it , but always recv() read exactly 
  359.                       // the lenght of the packet
  360. int iLast = iLen - iRet ;
  361. pLastBuf = new char [ iLen ] ;
  362. int iReaden = iRet ;
  363. memcpy( pLastBuf , bufwork , iReaden ) ;
  364. iRet =  recv( pDlg->m_s , pLastBuf + iReaden , iLast , 0 ) ;
  365. if( iRet == SOCKET_ERROR )
  366. {
  367. dwErr = WSAGetLastError() ;
  368. sprintf( szErr , "Error recv() = %ld " , dwErr ) ;
  369. break ;
  370. }
  371. else
  372. {
  373. bufwork = pLastBuf ;
  374.         pIpHeader = (IPHEADER *)bufwork ;
  375. if( iRet == iLast )
  376. iRet = iLen ;
  377. else
  378. { // read all last data
  379. iReaden += iRet ;
  380. iLast -= iRet ;
  381. while( TRUE )
  382. {
  383. iRet = recv( pDlg->m_s , pLastBuf +iReaden , iLast , 0 ) ;
  384. if( iRet == SOCKET_ERROR )
  385. {
  386. dwErr = WSAGetLastError() ;
  387. sprintf( szErr , "Error recv() = %ld " , dwErr ) ;
  388. break ;
  389. }
  390. else
  391. {
  392.         iReaden += iRet ;
  393.         iLast -= iRet ;
  394.     if( iLast <= 0 ) 
  395.     break ;
  396. }
  397. } // while
  398. }
  399. }
  400. }
  401. }   // while
  402. if( pLastBuf )
  403. delete [ ] pLastBuf ;
  404. }
  405. else
  406. {
  407. AfxMessageBox( "No data on network" ) ;
  408. continue ;
  409. }
  410. //Sleep( 100 ) ; //  Polling each 100 millisecond /* zhuwei add(2002.11.9) comment */
  411. }
  413. return TRUE ;
  414. }
  416. void CMySnifferDlg::OnSysCommand(UINT nID, LPARAM lParam)
  417. {
  418. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  419. {
  420. CAboutDlg dlgAbout;
  421. dlgAbout.DoModal();
  422. }
  423. else
  424. {
  425. CDialog::OnSysCommand(nID, lParam);
  426. }
  427. }
  429. // If you add a minimize button to your dialog, you will need the code below
  430. //  to draw the icon.  For MFC applications using the document/view model,
  431. //  this is automatically done for you by the framework.
  433. void CMySnifferDlg::OnPaint() 
  434. {
  435. if (IsIconic())
  436. {
  437. CPaintDC dc(this); // device context for painting
  439. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  441. // Center icon in client rectangle
  442. int cxIcon = GetSystemMetrics(SM_CXICON);
  443. int cyIcon = GetSystemMetrics(SM_CYICON);
  444. CRect rect;
  445. GetClientRect(&rect);
  446. int x = (rect.Width() - cxIcon + 1) / 2;
  447. int y = (rect.Height() - cyIcon + 1) / 2;
  449. // Draw the icon
  450. dc.DrawIcon(x, y, m_hIcon);
  451. }
  452. else
  453. {
  454. CDialog::OnPaint();
  455. }
  456. }
  458. // The system calls this to obtain the cursor to display while the user drags
  459. //  the minimized window.
  460. HCURSOR CMySnifferDlg::OnQueryDragIcon()
  461. {
  462. return (HCURSOR) m_hIcon;
  463. }
  465. void CMySnifferDlg::OnButton1() 
  466. {
  467. // TODO: Add your control notification handler code here
  468. char szErr[50], szHostName[MAX_PATH];
  469. DWORD dwErr;
  470. SOCKADDR_IN sa;
  472. gethostname(szHostName, sizeof(szHostName));
  474. m_iphostsource = m_ipsource;
  476. //m_ipcheckedhost = ntohl(m_iphost);
  478. CString str;
  479. if(m_threadID == 0)
  480. {
  481. str.LoadString(IDS_STOP);
  482. m_Start.SetWindowText(str);
  483. }
  484. else
  485. {
  486. if(m_threadID)
  487. {
  488. PostThreadMessage(m_threadID, WM_CLOSE, 0, 0);
  489. str.LoadString(IDS_START);
  490. m_Start.SetWindowText(str);
  491. m_Start.EnableWindow(FALSE);
  492. }
  493. return;
  494. }
  495. DWORD dwBufferLen[10];
  496. DWORD dwBufferInLen = 1;
  497. DWORD dwBytesReturned = 0;
  499. m_s = socket(AF_INET, SOCK_RAW, IPPROTO_IP);
  500. if(m_s == INVALID_SOCKET)
  501. {
  502. dwErr = WSAGetLastError();
  503. sprintf(szErr, "Error Socket() = %d", dwErr);
  504. AfxMessageBox(szErr);
  505. closesocket(m_s);
  506. return;
  507. }
  508. sa.sin_family = AF_INET;
  509. sa.sin_port = htons(7000);
  510. sa.sin_addr.s_addr = m_iphostsource;
  512. if(bind(m_s, (PSOCKADDR)&sa, sizeof(sa)) == SOCKET_ERROR)
  513. {
  514. dwErr = WSAGetLastError();
  515. sprintf(szErr, "Error bind() = %d", dwErr);
  516. AfxMessageBox(szErr);
  517. closesocket(m_s);
  518. return;
  519. }
  520. if(SOCKET_ERROR != WSAIoctl(m_s, SIO_RCVALL, &dwBufferInLen, sizeof(dwBufferInLen),
  521. &dwBufferLen, sizeof(dwBufferLen),
  522. &dwBytesReturned, NULL, NULL))
  523. AfxBeginThread(threadFunc, (LPVOID)this);
  524. else
  525. {
  526. dwErr = WSAGetLastError();
  527. sprintf(szErr, "Error WSAIoctl = %ld", dwErr);
  528. AfxMessageBox(szErr);
  529. closesocket(m_s);
  530. return;
  531. }
  532. }
  534. void CMySnifferDlg::AddData(CString s0, CString s1, CString s2, CString s3, CString s4, CString s5, CString s6)
  535. {
  536. int index;
  537. //if(s0 == "TCP")
  538. if((s0 == "TCP") && (s4 == "80"))
  539. {
  540. index = m_ListCtrl.InsertItem(0,s0);
  541. m_ListCtrl.SetItem(index,1,LVIF_TEXT,s1, 0, 0, 0,0);
  542. m_ListCtrl.SetItem(index,2,LVIF_TEXT,s2, 0, 0, 0,0);
  543. m_ListCtrl.SetItem(index,3,LVIF_TEXT,s3, 0, 0, 0,0);
  544. m_ListCtrl.SetItem(index,4,LVIF_TEXT,s4, 0, 0, 0,0);
  545. m_ListCtrl.SetItem(index,5,LVIF_TEXT,s5, 0, 0, 0,0);
  546. m_ListCtrl.SetItem(index,6,LVIF_TEXT,s6, 0, 0, 0,0);
  547. }
  548. //m_ctrList.SetItem(index,1,LVIF_TEXT,s2, 0, 0, 0,0);
  550. }