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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 网络截获/分析 > 查看源码
zxDlg.cpp
资源名称：zx.rar　[点击查看]
上传用户：aaagroup07
上传日期：2022-03-30
资源大小：1526k
文件大小：16k
源码类别：

网络截获/分析

开发平台：

C/C++

zxDlg.cpp：源码内容
  1. // zxDlg.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "zx.h"
  6. #include "zxDlg.h"
  7. #include "pcap.h"
  9. #ifdef _DEBUG
  10. #define new DEBUG_NEW
  11. #undef THIS_FILE
  12. static char THIS_FILE[] = __FILE__;
  13. #endif
  15. #pragma pack(1) //进入字节对齐方式
  16. typedef struct FrameHeader_t  { //帧首部
  17.     BYTE DesMAC[6]; // 目的地址
  18.     BYTE  SrcMAC[6]; // 源地址
  19.     WORD FrameType; // 帧类型
  20. } FrameHeader_t;
  21. typedef struct IPHeader_t { //IP首部
  22. BYTE Ver_HLen;
  23. BYTE TOS;
  24. WORD TotalLen;
  25. WORD ID;
  26. WORD Flag_Segment;
  27. BYTE TTL;
  28. BYTE Protocol;
  29. WORD Checksum;
  30. ULONG SrcIP;
  31. ULONG DstIP;
  32. } IPHeader_t;
  33. typedef struct Data_t { //包含帧首部和IP首部的数据包
  34. FrameHeader_t FrameHeader;
  35. IPHeader_t IPHeader;
  36. } Data_t;
  37. #pragma pack() //恢复缺省对齐方式
  39. pcap_t* thread_pNet;
  40. CListBox* thread_list_out;
  41. bool thread_can;
  42. CSliderCtrl *thread_pSlidCtrl;
  44. int TIME_ERR;
  45. int TIME_CANNOT;
  46. int PCAP_OPEN_FLAGS;
  47. int REC_MAX;
  48. int TIME_OUT;
  49. pcap_if_t* now_dev;
  50. CRITICAL_SECTION can_mutex; //串口读取 临界区变量
  51. /////////////////////////////////////////////////////////////////////////////
  52. // CAboutDlg dialog used for App About
  54. class CAboutDlg : public CDialog
  55. {
  56. public:
  57. CAboutDlg();
  59. // Dialog Data
  60. //{{AFX_DATA(CAboutDlg)
  61. enum { IDD = IDD_ABOUTBOX };
  62. CEdit m_edit_out;
  63. //}}AFX_DATA
  65. // ClassWizard generated virtual function overrides
  66. //{{AFX_VIRTUAL(CAboutDlg)
  67. protected:
  68. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  69. //}}AFX_VIRTUAL
  71. // Implementation
  72. protected:
  73. //{{AFX_MSG(CAboutDlg)
  74. virtual BOOL OnInitDialog();
  75. //}}AFX_MSG
  76. DECLARE_MESSAGE_MAP()
  77. };
  79. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  80. {
  81. //{{AFX_DATA_INIT(CAboutDlg)
  82. //}}AFX_DATA_INIT
  83. }
  85. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  86. {
  87. CDialog::DoDataExchange(pDX);
  88. //{{AFX_DATA_MAP(CAboutDlg)
  89. DDX_Control(pDX, IDC_EDIT_OUT, m_edit_out);
  90. //}}AFX_DATA_MAP
  91. }
  93. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  94. //{{AFX_MSG_MAP(CAboutDlg)
  95. //}}AFX_MSG_MAP
  96. END_MESSAGE_MAP()
  98. /////////////////////////////////////////////////////////////////////////////
  99. // CZxDlg dialog
  101. CZxDlg::CZxDlg(CWnd* pParent /*=NULL*/)
  102. : CDialog(CZxDlg::IDD, pParent)
  103. {
  104. //{{AFX_DATA_INIT(CZxDlg)
  105. //}}AFX_DATA_INIT
  106. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  107. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  108. }
  110. void CZxDlg::DoDataExchange(CDataExchange* pDX)
  111. {
  112. CDialog::DoDataExchange(pDX);
  113. //{{AFX_DATA_MAP(CZxDlg)
  114. DDX_Control(pDX, IDC_LIST_OUT, m_list_out);
  115. DDX_Control(pDX, IDC_LIST_DEV, m_list_dev);
  116. DDX_Control(pDX, IDC_COMBO_DEV, m_com_dev);
  117. //}}AFX_DATA_MAP
  118. }
  120. BEGIN_MESSAGE_MAP(CZxDlg, CDialog)
  121. //{{AFX_MSG_MAP(CZxDlg)
  122. ON_WM_SYSCOMMAND()
  123. ON_WM_PAINT()
  124. ON_WM_QUERYDRAGICON()
  125. ON_CBN_SELCHANGE(IDC_COMBO_DEV, OnSelchangeComboDev)
  126. ON_WM_DESTROY()
  127. ON_BN_CLICKED(IDCLEAR, OnClear)
  128. ON_BN_CLICKED(IDC_BUTTON_SET, OnButtonSet)
  129. ON_BN_CLICKED(IDC_BUTTON_HELP, OnButtonHelp)
  130. //}}AFX_MSG_MAP
  131. END_MESSAGE_MAP()
  133. /////////////////////////////////////////////////////////////////////////////
  134. // CZxDlg message handlers
  136. BOOL CZxDlg::OnInitDialog()
  137. {
  138. CDialog::OnInitDialog();
  140. // Add "About..." menu item to system menu.
  142. // IDM_ABOUTBOX must be in the system command range.
  143. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  144. ASSERT(IDM_ABOUTBOX < 0xF000);
  146. CMenu* pSysMenu = GetSystemMenu(FALSE);
  147. if (pSysMenu != NULL)
  148. {
  149. CString strAboutMenu;
  150. strAboutMenu.LoadString(IDS_ABOUTBOX);
  151. if (!strAboutMenu.IsEmpty())
  152. {
  153. pSysMenu->AppendMenu(MF_SEPARATOR);
  154. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  155. }
  156. }
  158. // Set the icon for this dialog.  The framework does this automatically
  159. //  when the application's main window is not a dialog
  160. SetIcon(m_hIcon, TRUE); // Set big icon
  161. SetIcon(m_hIcon, FALSE); // Set small icon
  163. // TODO: Add extra initialization here
  165. InitializeCriticalSection(&can_mutex);
  167. pcap_if_t *d; 
  169. //获得本机的设备列表
  170. if (pcap_findalldevs_ex(PCAP_SRC_IF_STRING,  //获取本机的接口设备
  171. NULL, //无需认证
  172. &alldevs, //指向设备列表首部
  173. errbuf //出错信息保存缓存区
  174. ) == -1)
  175. {
  176. AfxMessageBox("获得本机的设备列表失败!");
  177. return FALSE; //错误处理
  178. }
  179. m_list_out.AddString("获得本机的设备列表成功");
  180. //显示接口列表
  181. for(d= alldevs; d != NULL; d= d->next)
  182. {
  183. m_com_dev.AddString(d->name); //利用d->name获取该网络接口设备的名字
  184. }
  186. pNet=NULL;
  187. thread_list_out=&m_list_out;
  188. thread_can=false;
  189. now_dev=NULL;
  191. m_list_out.AddString("读取配置文件");
  192. ReadProfile();
  194. CSliderCtrl *pSlidCtrl=(CSliderCtrl*)GetDlgItem(IDC_SLIDER);
  195. pSlidCtrl->SetRange(0,10,TRUE);//设置滑动条范围
  196. pSlidCtrl->SetPos(5);//设置滑动条位置
  198. thread_pSlidCtrl=pSlidCtrl;
  200. HANDLE Capturer_Thread=CreateThread(NULL,0,Capturer,0,CREATE_SUSPENDED,NULL);
  201. if( Capturer_Thread==NULL )
  202. {
  203. AfxMessageBox("Capturer线程创建线程失败!");
  204. return FALSE;
  205. }
  206. else
  207. {
  208. ResumeThread(Capturer_Thread); // 恢复线程运行
  209. m_list_out.AddString("创建捕获线程成功");
  210. }
  212. return TRUE;  // return TRUE  unless you set the focus to a control
  213. }
  215. void CZxDlg::OnSysCommand(UINT nID, LPARAM lParam)
  216. {
  217. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  218. {
  219. CAboutDlg dlgAbout;
  220. dlgAbout.DoModal();
  221. }
  222. else
  223. {
  224. CDialog::OnSysCommand(nID, lParam);
  225. }
  226. }
  228. // If you add a minimize button to your dialog, you will need the code below
  229. //  to draw the icon.  For MFC applications using the document/view model,
  230. //  this is automatically done for you by the framework.
  232. void CZxDlg::OnPaint() 
  233. {
  234. if (IsIconic())
  235. {
  236. CPaintDC dc(this); // device context for painting
  238. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  240. // Center icon in client rectangle
  241. int cxIcon = GetSystemMetrics(SM_CXICON);
  242. int cyIcon = GetSystemMetrics(SM_CYICON);
  243. CRect rect;
  244. GetClientRect(&rect);
  245. int x = (rect.Width() - cxIcon + 1) / 2;
  246. int y = (rect.Height() - cyIcon + 1) / 2;
  248. // Draw the icon
  249. dc.DrawIcon(x, y, m_hIcon);
  250. }
  251. else
  252. {
  253. CDialog::OnPaint();
  254. }
  255. }
  257. // The system calls this to obtain the cursor to display while the user drags
  258. //  the minimized window.
  259. HCURSOR CZxDlg::OnQueryDragIcon()
  260. {
  261. return (HCURSOR) m_hIcon;
  262. }
  264. void CZxDlg::OnSelchangeComboDev() 
  265. {
  266. // TODO: Add your control notification handler code here
  268. while(m_list_dev.GetCount()>0)
  269. {
  270. m_list_dev.DeleteString(0);
  271. }
  273. pcap_addr_t *a; 
  274. pcap_if_t *d;
  275. d=alldevs;
  276. int com_num=m_com_dev.GetCurSel();
  277. int i=0;
  278. while(i<com_num) 
  279. {
  280. d=d->next;
  281. i++;
  282. }
  284. now_dev=d;
  286. //获取该网络接口设备的描述字符串
  287. m_list_dev.AddString(d->description);
  290. //打开该网络接口
  291. pNet=pcap_open(d->name,REC_MAX,PCAP_OPEN_FLAGS,TIME_OUT,NULL,errbuf);//PCAP_OPENFLAG_PROMISCUOUS
  292. if(pNet==NULL)
  293. {
  294. AfxMessageBox("打开该网络接口失败!");
  295. }
  296. thread_pNet=pNet;
  298. ////////////////
  300. //获取该网络接口设备的IP地址信息
  301. a=d->addresses;
  302. if(a!=NULL)
  303. {
  304. for(; a!=NULL; a=a->next)
  305. {
  306. if (a->addr->sa_family==AF_INET) //判断该地址是否IP地址
  307. {
  308. CString out;
  309. //利用a->addr获取IP地址
  310. if(a->addr!=NULL) 
  311. {
  312. out.Format("IP地址:   %d.%d.%d.%d ",(int)a->addr->sa_data[2]&255,
  313. (int)a->addr->sa_data[3]&255,(int)a->addr->sa_data[4]&255,
  314. (int)a->addr->sa_data[5]&255);
  315. }
  316. else
  317. {
  318. out="IP地址:   空";
  319. }
  320. m_list_dev.AddString(out);
  321. //利用a->netmask获取网络掩码
  322. if(a->netmask!=NULL) 
  323. {
  324. out.Format("网络掩码:   %d.%d.%d.%d ",(int)a->netmask->sa_data[2]&255,
  325. (int)a->netmask->sa_data[3]&255,(int)a->netmask->sa_data[4]&255,
  326. (int)a->netmask->sa_data[5]&255);
  327. }
  328. else
  329. {
  330. out="网络掩码:   空";
  331. }
  332. m_list_dev.AddString(out);
  333. //利用a->broadaddr获取广播地址
  334. if(a->broadaddr!=NULL) 
  335. {
  336. out.Format("广播地址:   %d.%d.%d.%d ",(int)a->broadaddr->sa_data[2]&255,
  337. (int)a->broadaddr->sa_data[3]&255,(int)a->broadaddr->sa_data[4]&255,
  338. (int)a->broadaddr->sa_data[5]&255);
  339. }
  340. else
  341. {
  342. out="广播地址:   空";
  343. }
  344. m_list_dev.AddString(out);
  345. //利用a->dstaddr)获取目的地址
  346. if(a->dstaddr!=NULL) 
  347. {
  348. out.Format("目的地址:   %d.%d.%d.%d ",(int)a->dstaddr->sa_data[2]&255,
  349. (int)a->dstaddr->sa_data[3]&255,(int)a->dstaddr->sa_data[4]&255,
  350. (int)a->dstaddr->sa_data[5]&255);
  351. }
  352. else
  353. {
  354. out="目的地址:   空";
  355. }
  356. m_list_dev.AddString(out);
  357. }
  358. }
  359. }
  360. else
  361. {
  362. m_list_dev.AddString("该网卡无地址");
  363. }
  365. }
  367. void CZxDlg::OnDestroy() 
  368. {
  369. CDialog::OnDestroy();
  371. // TODO: Add your message handler code here
  372. //释放设备列表
  373. pcap_freealldevs(alldevs);
  374. }
  377. void CZxDlg::OnOK() 
  378. {
  379. // TODO: Add extra validation here
  380. if(thread_can==true)
  381. {
  382. thread_can=false;
  383. }
  384. else
  385. {
  386. thread_can=true;
  387. }
  388. }
  390. void CZxDlg::OnClear()
  391. {
  392. // TODO: Add your control notification handler code here
  393. EnterCriticalSection(&can_mutex); //进入临界区
  394. bool last_can=thread_can;
  395. thread_can=false;
  396. while(m_list_out.GetCount()>0)
  397. {
  398. m_list_out.DeleteString(0);
  399. }
  400. thread_can=last_can;
  401. LeaveCriticalSection(&can_mutex); //离开临界区
  402. }
  404. DWORD WINAPI CZxDlg::Capturer(PVOID pvoid)
  405. {
  406. while(true)
  407. {
  408. if(thread_pNet!=NULL && thread_can==true)
  409. {
  410. pcap_pkthdr* pkt_header,kt_header;
  411. pkt_header=&kt_header;
  412. const u_char* pkt_data;
  414. EnterCriticalSection(&can_mutex); //进入临界区
  416. if(pcap_next_ex(thread_pNet,&pkt_header,&pkt_data)==1)
  417. {
  418. LeaveCriticalSection(&can_mutex); //离开临界区
  419. /////////////////////////
  420. thread_list_out->AddString("------------------------------------------------------------");
  421. //////////////////////////
  422. CString cout_time,cout_len;
  423. struct timeval ts; /* time stamp */
  424. bpf_u_int32 len; /* length this packet (off wire) */
  425. len=pkt_header->len;
  426. ts=pkt_header->ts;
  427. cout_len.Format("数据包长度:   %d",len);
  428. COleDateTime   DataTime   =   ts.tv_sec;  
  429. int   nYear   =   DataTime.GetYear();  
  430. int   nMonth   =   DataTime.GetMonth();  
  431. int   nDay   =   DataTime.GetDay();  
  432. int   nHour   =   DataTime.GetHour();  
  433. int   nMin   =   DataTime.GetMinute();  
  434. int   nSec   =   DataTime.GetSecond();
  435. cout_time.Format("时间:   %d年%d月%d号%d时%d分%d秒",nYear,nMonth,nDay,nHour,nMin,nSec);
  436. //////////////////////////
  437. Data_t * IPPacket;
  438. ULONG SourceIP,DestinationIP;
  439. IPPacket = (Data_t *) pkt_data;
  440. SourceIP = (IPPacket->IPHeader.SrcIP);
  441. DestinationIP = (IPPacket->IPHeader.DstIP);
  442. BYTE DesMAC[6]; // 目的地址
  443. BYTE  SrcMAC[6]; // 源地址
  444. WORD FrameType; // 帧类型
  445. FrameType=ntohs(IPPacket->FrameHeader.FrameType);
  446. memcpy(DesMAC, IPPacket->FrameHeader.DesMAC, 6);
  447. memcpy(SrcMAC, IPPacket->FrameHeader.SrcMAC, 6);
  448. CString cout_DesMAC,cout_SrcMAC,cout_FrameType;
  449. cout_DesMAC.Format("目的地址:   %d-%d-%d-%d-%d-%d ",(int)DesMAC[0]&255,
  450. (int)DesMAC[1]&255,(int)DesMAC[2]&255,(int)DesMAC[3]&255,
  451. (int)DesMAC[4]&255,(int)DesMAC[5]&255);
  452. cout_SrcMAC.Format("源地址:   %d-%d-%d-%d-%d-%d ",(int)SrcMAC[0]&255,
  453. (int)SrcMAC[1]&255,(int)SrcMAC[2]&255,(int)SrcMAC[3]&255,
  454. (int)SrcMAC[4]&255,(int)SrcMAC[5]&255);
  455. if(FrameType>=2048)
  456. {
  457. cout_FrameType.Format("帧类型:   %d ",FrameType);
  458. }
  459. else
  460. {
  461. cout_FrameType.Format("帧长度:   %d ",FrameType);
  462. }
  464. /////////////////////////
  466. thread_list_out->AddString(cout_len+"       "+cout_time);
  467. thread_list_out->AddString(cout_DesMAC+"  "+cout_SrcMAC+"  "+cout_FrameType);
  468. /////////////////////////
  469. if(FrameType==0x0800)
  470. {
  471. in_addr SourceIP_addr,DestinationIP_addr;
  472. CString cout_SourceIP,cout_DestinationIP;
  473. memcpy(&SourceIP_addr, &SourceIP, 4);
  474. memcpy(&DestinationIP_addr, &DestinationIP, 4);
  475. cout_SourceIP.Format("源IP地址:   %s",inet_ntoa(SourceIP_addr));
  476. cout_DestinationIP.Format("目的IP地址:   %s",inet_ntoa(DestinationIP_addr));
  478. thread_list_out->AddString(cout_DestinationIP+"  "+cout_SourceIP);
  479. }
  480. /////////////////////////
  481. int sleep_time=thread_pSlidCtrl->GetPos()*100;
  483. Sleep(sleep_time);
  484. } 
  485. else
  486. {
  487. LeaveCriticalSection(&can_mutex); //离开临界区
  489. thread_list_out->AddString("************************************************************");
  490. thread_list_out->AddString("捕获数据超时或失败!");
  491. Sleep(TIME_ERR);
  492. }
  493. }
  494. else
  495. {
  496. Sleep(TIME_CANNOT);
  497. }
  498. }
  499. return 0;
  500. }
  502. void CZxDlg::ReadProfile()
  503. {
  504. if((TIME_ERR=(GetPrivateProfileInt("捕获数据包参数", "捕获出错后睡眠时间", -1, "MyWinPcap.ini")))==-1) 
  505.     {
  506. WritePrivateProfileString ("捕获数据包参数", "捕获出错后睡眠时间", "1000", "MyWinPcap.ini");  
  507. TIME_ERR=1000;                                        
  508.     }
  509. if((TIME_CANNOT=(GetPrivateProfileInt("捕获数据包参数", "不能捕获时睡眠时间", -1, "MyWinPcap.ini")))==-1) 
  510.     {
  511. WritePrivateProfileString ("捕获数据包参数", "不能捕获时睡眠时间", "1000", "MyWinPcap.ini");  
  512. TIME_CANNOT=1000;                                        
  513.     }
  514. if((PCAP_OPEN_FLAGS=(GetPrivateProfileInt("捕获数据包参数", "捕获方式", -1, "MyWinPcap.ini")))==-1) 
  515.     {
  516. WritePrivateProfileString ("捕获数据包参数", "捕获方式", "1", "MyWinPcap.ini");  
  517. PCAP_OPEN_FLAGS=1;                                        
  518.     }
  519. if((REC_MAX=(GetPrivateProfileInt("捕获数据包参数", "获取数据包最大长度", 0, "MyWinPcap.ini")))==0) 
  520.     {
  521. WritePrivateProfileString ("捕获数据包参数", "获取数据包最大长度", "1024", "MyWinPcap.ini");  
  522. REC_MAX=1024;                                        
  523.     }
  524. if((TIME_OUT=(GetPrivateProfileInt("捕获数据包参数", "获取数据包超时时间", 0, "MyWinPcap.ini")))==0) 
  525.     {
  526. WritePrivateProfileString ("捕获数据包参数", "获取数据包超时时间", "1000", "MyWinPcap.ini");  
  527. TIME_OUT=1000;                                        
  528.     }
  529. CString out;
  530. out.Format("捕获出错后睡眠时间:   %d",TIME_ERR);
  531. m_list_out.AddString(out);
  532. out.Format("不能捕获时睡眠时间:   %d",TIME_CANNOT);
  533. m_list_out.AddString(out);
  534. out.Format("捕获方式:   %d",PCAP_OPEN_FLAGS);
  535. m_list_out.AddString(out);
  536. out.Format("获取数据包最大长度:   %d",REC_MAX);
  537. m_list_out.AddString(out);
  538. out.Format("获取数据包超时时间:   %d",TIME_OUT);
  539. m_list_out.AddString(out);
  540. }
  542. void CZxDlg::OnButtonSet() 
  543. {
  544. // TODO: Add your control notification handler code here
  546. EnterCriticalSection(&can_mutex); //进入临界区
  548. bool last_can=thread_can;
  549. thread_can=false;
  551. //////////////////////////////// 
  552. if(now_dev!=NULL)
  553. {
  554. //char filter[100] = "arp"; 
  555. char filter[100];
  556. CEdit *pedit_in=(CEdit*)GetDlgItem(IDC_EDIT_IN);
  557. pedit_in->GetLine(0,filter,100);
  559. struct bpf_program fcode;
  561. bpf_u_int32 localnet;
  562. bpf_u_int32 netmask;
  563. char thread_errbuf[PCAP_ERRBUF_SIZE];
  564. if(pcap_lookupnet(now_dev->name,&localnet,&netmask,thread_errbuf)  < 0)
  565. {
  566. AfxMessageBox("查找子网掩码失败!");
  567. return;
  568. }
  569. //compile the filter
  570. if(pcap_compile(thread_pNet, &fcode, filter, 0, netmask)  < 0)
  571. {
  572. AfxMessageBox("编译过滤字符串失败!");
  573. return;
  574. }
  576. //set the filter
  577. if(pcap_setfilter(thread_pNet, &fcode) <0)
  578. {
  579. AfxMessageBox("设置过滤字符串失败!");
  580. return;
  581. }
  582. ///////////////
  583. }
  584. else
  585. {
  586. thread_list_out->AddString("设备为空，无法过滤");
  587. }
  589. ///////////////////////////////////
  590. thread_can=last_can;
  592. LeaveCriticalSection(&can_mutex); //离开临界区
  593. }
  595. void CZxDlg::OnButtonHelp() 
  596. {
  597. // TODO: Add your control notification handler code here
  598. CAboutDlg help_dlg;
  599. help_dlg.DoModal();
  600. }
  602. BOOL CAboutDlg::OnInitDialog() 
  603. {
  604. CDialog::OnInitDialog();
  606. // TODO: Add extra initialization here
  608. CString out;
  609. out.LoadString(IDS_STRING_HELP);
  610. m_edit_out.SetWindowText(out);
  611. return TRUE;  // return TRUE unless you set the focus to a control
  612.               // EXCEPTION: OCX Property Pages should return FALSE
  613. }