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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 网络截获/分析 > 查看源码
GetPacketListView.cpp
资源名称：ntpacket.zip　[点击查看]
上传用户：qxsjcl
上传日期：2007-01-08
资源大小：263k
文件大小：11k
源码类别：

网络截获/分析

开发平台：

Visual C++

GetPacketListView.cpp：源码内容
  1. // GetPacketListView.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "GetPacket.h"
  6. #include "GetPacketListView.h"
  7. #include "mainfrm.h"
  8. #ifdef _DEBUG
  9. #define new DEBUG_NEW
  10. #undef THIS_FILE
  11. static char THIS_FILE[] = __FILE__;
  12. #endif
  14. //global variables
  15. CONTROL_BLOCK cbAdapter;
  16. PVOID    pPacket;
  17. BOOL     bSniff=FALSE;
  19. GetPacketListView* pThreadParam;
  20. DWORD WINAPI GetPacketThread(LPVOID param)
  21. {
  22. //wsastartup , I don't know why
  23. WSADATA wData;
  24. WORD   wVersionRequested = MAKEWORD(2, 0);
  25.     if (WSAStartup(wVersionRequested, &wData) != 0)
  26. {
  27. MessageBox(NULL,"WSAStartup","Error",MB_OK);
  28. return -1;
  29. }
  31. //get adapter's name
  32. ULONG NameLength=64;
  33. PacketGetAdapterNames(cbAdapter.AdapterName,&NameLength);
  35. cbAdapter.BufferSize=1514;
  36. cbAdapter.hMem=GlobalAlloc(GMEM_MOVEABLE,1514);
  37. if(cbAdapter.hMem==NULL)
  38. {
  39. MessageBox(NULL,"GlobalAlloc","Error",MB_OK);
  40. return -1;
  41. }
  42. cbAdapter.lpMem=(LPBYTE)GlobalLock(cbAdapter.hMem);
  44.     //open adapter
  45. cbAdapter.hFile=PacketOpenAdapter(cbAdapter.AdapterName);
  46.     if (cbAdapter.hFile == NULL)
  47. {
  48. MessageBox(NULL,"PacketOpenAdapter","Error",MB_OK);
  49. return -1;
  50. }
  52. //set NIC to PROMISCUOUS status
  53. PacketSetFilter((LPADAPTER)cbAdapter.hFile,NDIS_PACKET_TYPE_PROMISCUOUS);
  55. //allocate packet
  56.     pPacket=PacketAllocatePacket((LPADAPTER)cbAdapter.hFile);
  57. if(!pPacket)
  58. {
  59. MessageBox(NULL,"PacketAllocatePacket","Error",MB_OK);
  60. return -1;
  61. }
  63. PacketInitPacket((LPPACKET)pPacket,cbAdapter.lpMem,1514);
  65. bSniff=TRUE;
  66. while(bSniff)
  67. {
  68. PacketReceivePacket((LPADAPTER)cbAdapter.hFile,(LPPACKET)pPacket,TRUE,&cbAdapter.PacketLength);
  70. //the packet is valid***total frame size should exceeds 14+20+20=54
  71. if (cbAdapter.PacketLength > (sizeof(ETH_HEADER) + sizeof(IP_HEADER) + sizeof(TCP_HEADER)))
  72. {
  73. ETHER_PACKET   *pEthPkt = (ETHER_PACKET *) cbAdapter.lpMem;
  75. ETH_HEADER     *pEthHdr = (ETH_HEADER *) cbAdapter.lpMem;
  76. IP_HEADER      *pIpHdr = (IP_HEADER *) ((char *) pEthPkt + sizeof(ETH_HEADER));
  77. TCP_HEADER     *pTcpHdr = (TCP_HEADER *) ((char *) pIpHdr + sizeof(IP_HEADER));
  78. int             i_IPH_Size = pIpHdr->ihl * 4;
  79. int             i_TCPH_Size = pTcpHdr->doff * 4;
  81. //determine if this is a valid IP packet
  82. if(pEthPkt->ETH.h_proto!=8)
  83. continue;
  84. if (pIpHdr->protocol != 6)
  85. continue;
  87. int  iDataSize = ((int) ntohs(pIpHdr->tot_len) -  i_IPH_Size - i_TCPH_Size);
  88. if(!iDataSize)
  89. continue;
  90. BYTE *pPktData = (BYTE *) pTcpHdr + i_TCPH_Size;
  91. *(pPktData+iDataSize)='';
  92. pThreadParam->SetData(pIpHdr->saddr,pTcpHdr->source,pIpHdr->daddr,pTcpHdr->dest,pPktData,iDataSize);
  93. }
  94. }//end while
  96. PacketFreePacket((LPPACKET)pPacket);
  97. PacketCloseAdapter((LPADAPTER)cbAdapter.hFile);
  98. GlobalUnlock(cbAdapter.hMem);
  99. GlobalFree(cbAdapter.hMem);
  100. WSACleanup();
  101. return 0;
  102. }
  103. /////////////////////////////////////////////////////////////////////////////
  104. // GetPacketListView
  106. IMPLEMENT_DYNCREATE(GetPacketListView, CListView)
  108. GetPacketListView::GetPacketListView()
  109. {
  110. pThreadParam=this;
  111. //bIsSniffing=FALSE;
  112. }
  114. GetPacketListView::~GetPacketListView()
  115. {
  116. }
  119. BEGIN_MESSAGE_MAP(GetPacketListView, CListView)
  120. //{{AFX_MSG_MAP(GetPacketListView)
  121. ON_COMMAND(ID_FILE_START, OnFileStart)
  122. ON_COMMAND(ID_FILE_STOP, OnFileStop)
  123. ON_COMMAND(ID_EDIT_FILTER, OnEditFilter)
  124. ON_UPDATE_COMMAND_UI(ID_FILE_START, OnUpdateFileStart)
  125. ON_UPDATE_COMMAND_UI(ID_FILE_STOP, OnUpdateFileStop)
  126. ON_COMMAND(ID_EDIT_CLEARALL, OnEditClearall)
  127. //}}AFX_MSG_MAP
  128. END_MESSAGE_MAP()
  130. /////////////////////////////////////////////////////////////////////////////
  131. // GetPacketListView drawing
  133. void GetPacketListView::OnDraw(CDC* pDC)
  134. {
  135. CDocument* pDoc = GetDocument();
  136. // TODO: add draw code here
  137. }
  139. /////////////////////////////////////////////////////////////////////////////
  140. // GetPacketListView diagnostics
  142. #ifdef _DEBUG
  143. void GetPacketListView::AssertValid() const
  144. {
  145. CListView::AssertValid();
  146. }
  148. void GetPacketListView::Dump(CDumpContext& dc) const
  149. {
  150. CListView::Dump(dc);
  151. }
  152. #endif //_DEBUG
  154. /////////////////////////////////////////////////////////////////////////////
  155. // GetPacketListView message handlers
  157. void GetPacketListView::OnInitialUpdate() 
  158. {
  159. CListView::OnInitialUpdate();
  160. /*CWnd *pmain;
  161. pmain=AfxGetMainWnd();
  162. pmain->SetWindowText("GetPacket1.0");*/
  163. //set list control's style
  164. LONG dwLong = GetWindowLong(GetListCtrl().m_hWnd, GWL_STYLE);
  165. dwLong ^= LVS_REPORT|LVS_SINGLESEL ;
  166. LONG err=SetWindowLong(GetListCtrl().m_hWnd, GWL_STYLE, dwLong);
  167. ListView_SetExtendedListViewStyle(GetListCtrl(), LVS_EX_FULLROWSELECT); 
  168. //insert 5 columns
  169. LV_COLUMN lv_column;
  170. lv_column.mask=LVCF_TEXT|LVCF_WIDTH|LVCF_SUBITEM ;
  171. lv_column.cx=110;
  172. lv_column.pszText="  Source IP";
  173. GetListCtrl().InsertColumn( 0, &lv_column);
  175. lv_column.cx=80;
  176. lv_column.pszText="Source Port";
  177. GetListCtrl().InsertColumn( 1, &lv_column);
  179. lv_column.cx=110;
  180. lv_column.pszText="    Dest IP";
  181. GetListCtrl().InsertColumn( 2, &lv_column);
  183. lv_column.cx=80;
  184. lv_column.pszText="  Dest Port";
  185. GetListCtrl().InsertColumn( 3, &lv_column);
  187. lv_column.cx=370;
  188. lv_column.pszText="            packet content";
  189. GetListCtrl().InsertColumn( 4, &lv_column);
  190. }
  192. void GetPacketListView::OnFileStart() 
  193. {
  195. DWORD dwThreadID;
  196. hThread=CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)GetPacketThread,
  197.                 (LPVOID)NULL,(DWORD)0,&dwThreadID);
  198. ((CMainFrame*)AfxGetMainWnd())->m_wndStatusBar.SetPaneText( 0, "START RETRIEVING PACKETS...", TRUE );
  200. /*//if(bIsSniffing)
  201. // return;
  202. //bIsSniffing=TRUE;
  204. //wsastartup , I don't know why
  205. WORD   wVersionRequested = MAKEWORD(2, 0);
  206.     if (WSAStartup(wVersionRequested, &wsaData) != 0)
  207. {
  208. MessageBox("WSAStartup error","",MB_OK);
  209. return;
  210. }
  212. //get adapter's name
  213. ULONG NameLength=64;
  214. PacketGetAdapterNames(Adapter.AdapterName,&NameLength);
  216. Adapter.BufferSize=1514;
  217. Adapter.hMem=GlobalAlloc(GMEM_MOVEABLE,1514);
  218. Adapter.lpMem=(LPBYTE)GlobalLock(Adapter.hMem);
  220.     //open adapter
  221. Adapter.hFile=PacketOpenAdapter(Adapter.AdapterName);
  222.     if (Adapter.hFile == NULL)
  223. {
  224. MessageBox("PacketOpenAdapter error",Adapter.AdapterName,MB_OK);
  225. return;
  226. }
  228. //set NIC to PROMISCUOUS status
  229. PacketSetFilter((LPADAPTER)Adapter.hFile,NDIS_PACKET_TYPE_PROMISCUOUS);
  231. //allocate packet
  232.     Packet=PacketAllocatePacket((LPADAPTER)Adapter.hFile);
  233.     if (Packet != NULL) 
  234. {
  235. PacketInitPacket((LPPACKET)Packet,Adapter.lpMem,1514);
  237. int i=0;
  238. while(bIsSniffing)
  239. {
  240. PacketReceivePacket((LPADAPTER)Adapter.hFile,(LPPACKET)Packet,TRUE,&Adapter.PacketLength);
  242. //the packet is valid***total frame size should exceeds 14+20+20=54
  243. if (Adapter.PacketLength > (sizeof(ETH_HEADER) + sizeof(IP_HEADER) + sizeof(TCP_HEADER)))
  244. {
  245. ETHER_PACKET   *pEthPkt = (ETHER_PACKET *) Adapter.lpMem;
  246.             
  247. ETH_HEADER     *pEthHdr = (ETH_HEADER *) Adapter.lpMem;
  248.     IP_HEADER      *pIpHdr = (IP_HEADER *) ((char *) pEthPkt + sizeof(ETH_HEADER));
  249. TCP_HEADER     *pTcpHdr = (TCP_HEADER *) ((char *) pIpHdr + sizeof(IP_HEADER));
  250. int             i_IPH_Size = pIpHdr->ihl * 4;
  251. int             i_TCPH_Size = pTcpHdr->doff * 4;
  253. //determine if this is a valid IP packet
  254. if(pEthPkt->ETH.h_proto!=8)
  255. continue;
  256. if (pIpHdr->protocol != 6)
  257.         continue;
  258.         {
  259.         int  iDataSize = ((int) ntohs(pIpHdr->tot_len) -  i_IPH_Size - i_TCPH_Size);
  260. if(!iDataSize)
  261. continue;
  262.     BYTE *pPktData = (BYTE *) pTcpHdr + i_TCPH_Size;
  263. *(pPktData+iDataSize)='';
  264. SetData(pIpHdr->saddr,pTcpHdr->source,pIpHdr->daddr,pTcpHdr->dest,pPktData,iDataSize);
  265. }
  266. i++;
  267. if(i==1000)
  268. return;
  269. }
  270. }//end while
  271. }*/
  273. }
  275. void GetPacketListView::OnFileStop() 
  276. {
  277. bSniff=FALSE;
  279. //TerminateThread(hThread,0);
  280. /*if(!bIsSniffing)
  281. return;
  282. bIsSniffing=FALSE;
  283. WSACleanup();
  284.     PacketCloseAdapter((LPADAPTER)Adapter.hFile);
  285. if(Packet!=NULL)
  286. PacketFreePacket((LPPACKET)Packet);
  287. PacketSetFilter((LPADAPTER)Adapter.hFile,NDIS_PACKET_TYPE_DIRECTED);*/
  288. }
  291. void GetPacketListView::SetData(__u32 saddr, __u16 source, __u32 daddr, __u16 dest, BYTE * pPktData, int iDataSize)
  292. {
  293. static char     SourceAddress[512] = "";
  294.   static char     DestAddress[512] = "";
  295. char buffer[1024],data[2048],temp[512],*dot;
  296.     struct in_addr  iaddr;
  297. int iSourcePort,iDestPort,i;
  298. CGetPacketApp *pApp;
  299. pApp=(CGetPacketApp*)AfxGetApp();
  301. //save packet data
  302. strcpy(data,(char*)pPktData);
  304. memset(&iaddr,0,sizeof(struct in_addr));
  305.     iaddr.s_addr = saddr;
  306.     strcpy(SourceAddress, inet_ntoa(iaddr));//source IP address
  308. if(pApp->SourceIP.GetSize()>0)
  309. {
  310. for(i=0;i<pApp->SourceIP.GetSize();i++)
  311. if(!strcmp(SourceAddress,pApp->SourceIP.GetAt(i)))
  312. break;
  313. if(i==pApp->SourceIP.GetSize())
  314. return;
  315. }
  317. memset(&iaddr,0,sizeof(struct in_addr));
  318.     iaddr.s_addr = daddr;
  319. strcpy(DestAddress,inet_ntoa(iaddr));   //dest IP address
  320. if(pApp->DestIP.GetSize()>0)
  321. {
  322. for(i=0;i<pApp->DestIP.GetSize();i++)
  323. if(!strcmp(DestAddress,pApp->DestIP.GetAt(i)))
  324. break;
  325. if(i==pApp->DestIP.GetSize())
  326. return;
  327. }
  329. iSourcePort=(int) ntohs(source);   //source port no
  330. if(pApp->SourcePort.GetSize()>0)
  331. {
  332. for(i=0;i<pApp->SourcePort.GetSize();i++)
  333. if((unsigned int)iSourcePort==pApp->SourcePort.GetAt(i))
  334. break;
  335. if(i==pApp->SourcePort.GetSize())
  336. return;
  337. }
  339. iDestPort  =(int) ntohs(dest  );   //destination port no
  340. if(pApp->DestPort.GetSize()>0)
  341. {
  342. for(i=0;i<pApp->DestPort.GetSize();i++)
  343. if((unsigned int)iDestPort==pApp->DestPort.GetAt(i))
  344. break;
  345. if(i==pApp->DestPort.GetSize())
  346. return;
  347. }
  348. #ifndef _DEBUG //filter "172.28.9.*"
  349. strcpy(temp,SourceAddress);
  350. dot=strrchr( temp, '.' );
  351. *dot='';
  352. if(!strcmp(temp,"172.28.9"))
  353. return;
  354. strcpy(temp,DestAddress);
  355. dot=strrchr( temp, '.' );
  356. *dot='';
  357. if(!strcmp(temp,"172.28.9"))
  358. return;
  359. #endif
  360. //have been filtered, can be insert into listview.
  361. int index;
  362. index=GetListCtrl().InsertItem(0,SourceAddress);
  363. if(index!=-1)
  364. {
  365. GetListCtrl().SetItem( index, 1, LVIF_TEXT , _itoa( iSourcePort, buffer, 10 ), 0, 0, 0,0);
  366. GetListCtrl().SetItem( index, 2, LVIF_TEXT , DestAddress     , 0, 0, 0,0);
  367. GetListCtrl().SetItem( index, 3, LVIF_TEXT , _itoa( iDestPort, buffer, 10 ), 0, 0, 0,0);
  368. GetListCtrl().SetItem( index, 4, LVIF_TEXT , data, 0, 0, 0,0);
  369. }
  371.     return ;
  372. }
  374. void GetPacketListView::OnEditFilter() 
  375. {
  376. CFileterDlg FilterDlg;
  377. int ret=FilterDlg.DoModal();
  378. if(ret==IDOK)
  379. {
  381. }
  382. }
  384. void GetPacketListView::OnUpdateFileStart(CCmdUI* pCmdUI) 
  385. {
  386. if(bSniff)
  387. pCmdUI->Enable(FALSE);
  388. else
  389. pCmdUI->Enable(TRUE);
  390. }
  392. void GetPacketListView::OnUpdateFileStop(CCmdUI* pCmdUI) 
  393. {
  394. if(!bSniff)
  395. pCmdUI->Enable(FALSE);
  396. else
  397. pCmdUI->Enable(TRUE);
  398. }
  400. void GetPacketListView::OnEditClearall() 
  401. {
  402. GetListCtrl().DeleteAllItems();
  403. }