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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 防火墙与安全工具 > 查看源码
IpMonitorView.cpp
资源名称：ipfilter.zip　[点击查看]
上传用户：nnxzhh
上传日期：2007-01-11
资源大小：742k
文件大小：8k
源码类别：

防火墙与安全工具

开发平台：

WINDOWS

IpMonitorView.cpp：源码内容
  1. ///////////////////////////////////////////////////////////////////////////////
  2. //
  3. // (C) Copyright 1999 - 2000 Mark Roddy
  4. // All Rights Reserved
  5. //
  6. // Hollis Technology Solutions
  7. // 94 Dow Road
  8. // Hollis, NH 03049
  9. // info@hollistech.com
  10. //
  11. // Synopsis: 
  12. // 
  13. //
  14. // Version Information:
  15. //
  16. // $Header: /iphook/usr/IpMonitor/IpMonitorView.cpp 3     1/27/00 10:35p Markr $ 
  17. //
  18. ///////////////////////////////////////////////////////////////////////////////
  19. // IpMonitorView.cpp : implementation of the CIpMonitorView class
  20. //
  22. #include "stdafx.h"
  23. #include "IpMonitor.h"
  25. #include "IpMonitorDoc.h"
  26. #include "IpMonitorView.h"
  27. #include "Mainfrm.h"
  28. #include "winsock2.h"
  30. #ifdef _DEBUG
  31. #define new DEBUG_NEW
  32. #undef THIS_FILE
  33. static char THIS_FILE[] = __FILE__;
  34. #endif
  36. /////////////////////////////////////////////////////////////////////////////
  37. // CIpMonitorView
  39. IMPLEMENT_DYNCREATE(CIpMonitorView, CListView)
  41. BEGIN_MESSAGE_MAP(CIpMonitorView, CListView)
  42. //{{AFX_MSG_MAP(CIpMonitorView)
  43. ON_WM_CREATE()
  44. ON_MESSAGE(BUFFER_AVAILABLE_MSG, OnBufferMessage)
  45. ON_COMMAND(ID_DOC_CLEAR, OnDocClear)
  46. //}}AFX_MSG_MAP
  47. // Standard printing commands
  48. ON_COMMAND(ID_FILE_PRINT, CView::OnFilePrint)
  49. ON_COMMAND(ID_FILE_PRINT_DIRECT, CView::OnFilePrint)
  50. ON_COMMAND(ID_FILE_PRINT_PREVIEW, CView::OnFilePrintPreview)
  51. END_MESSAGE_MAP()
  53. /////////////////////////////////////////////////////////////////////////////
  54. // CIpMonitorView construction/destruction
  56. CIpMonitorView::CIpMonitorView()
  57. {
  58. m_dwDefaultStyle |= LVS_REPORT;
  59. theApp.m_theView = this;
  61. }
  63. CIpMonitorView::~CIpMonitorView()
  64. {
  65. }
  67. BOOL CIpMonitorView::PreCreateWindow(CREATESTRUCT& cs)
  68. {
  69. // TODO: Modify the Window class or styles here by modifying
  70. //  the CREATESTRUCT cs
  71. cs.style |= LVS_REPORT;
  74. return CListView::PreCreateWindow(cs);
  75. }
  77. /////////////////////////////////////////////////////////////////////////////
  78. // CIpMonitorView drawing
  80. void CIpMonitorView::OnDraw(CDC* pDC)
  81. {
  82. CIpMonitorDoc* pDoc = GetDocument();
  83. ASSERT_VALID(pDoc);
  84. // TODO: add draw code for native data here
  85. }
  87. /////////////////////////////////////////////////////////////////////////////
  88. // CIpMonitorView printing
  90. BOOL CIpMonitorView::OnPreparePrinting(CPrintInfo* pInfo)
  91. {
  92. // default preparation
  93. return DoPreparePrinting(pInfo);
  94. }
  96. void CIpMonitorView::OnBeginPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
  97. {
  98. // TODO: add extra initialization before printing
  99. }
  101. void CIpMonitorView::OnEndPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
  102. {
  103. // TODO: add cleanup after printing
  104. }
  106. /////////////////////////////////////////////////////////////////////////////
  107. // CIpMonitorView diagnostics
  109. #ifdef _DEBUG
  110. void CIpMonitorView::AssertValid() const
  111. {
  112. CListView::AssertValid();
  113. }
  115. void CIpMonitorView::Dump(CDumpContext& dc) const
  116. {
  117. CListView::Dump(dc);
  118. }
  120. CIpMonitorDoc* CIpMonitorView::GetDocument() // non-debug version is inline
  121. {
  122. ASSERT(m_pDocument->IsKindOf(RUNTIME_CLASS(CIpMonitorDoc)));
  123. return (CIpMonitorDoc*)m_pDocument;
  124. }
  125. #endif //_DEBUG
  127. /////////////////////////////////////////////////////////////////////////////
  128. // CIpMonitorView message handlers
  130. void CIpMonitorView::OnInitialUpdate() 
  131. {
  132. CListView::OnInitialUpdate();
  134. // TODO: Add your specialized code here and/or call the base class
  135. // this code only works for a report-mode list view
  136. ASSERT(GetStyle() & LVS_REPORT);
  138. // Gain a reference to the list control itself
  139. CListCtrl& theCtrl = GetListCtrl();
  141. // Insert a column. This override is the most convenient.
  142. theCtrl.InsertColumn(0, _T("TIME                        "), LVCFMT_LEFT);
  144. // The other InsertColumn() override requires an initialized
  145. // LVCOLUMN structure.
  146. LVCOLUMN col;
  148. col.mask = LVCF_FMT | LVCF_TEXT;
  149. col.pszText = _T("SEQ      ");
  150. col.fmt = LVCFMT_LEFT;
  151. theCtrl.InsertColumn(1, &col);
  153. col.mask = LVCF_FMT | LVCF_TEXT;
  154. col.pszText = _T("S/R");
  155. col.fmt = LVCFMT_LEFT;
  156. theCtrl.InsertColumn(2, &col);
  158. col.mask = LVCF_FMT | LVCF_TEXT;
  159. col.pszText = _T("IP HEADER                                            ");
  160. col.fmt = LVCFMT_LEFT;
  161. theCtrl.InsertColumn(3, &col);
  163. //
  164. // this last column is just and 'end of row marker'.
  165. //
  166. col.mask = LVCF_FMT | LVCF_TEXT;
  167. col.pszText = _T("");
  168. col.fmt = LVCFMT_LEFT;
  169. theCtrl.InsertColumn(4, &col);
  171. // Set reasonable widths for our columns
  172. theCtrl.SetColumnWidth(0, LVSCW_AUTOSIZE_USEHEADER);
  173. theCtrl.SetColumnWidth(1, LVSCW_AUTOSIZE_USEHEADER);
  174. theCtrl.SetColumnWidth(2, LVSCW_AUTOSIZE_USEHEADER);
  175. theCtrl.SetColumnWidth(3, LVSCW_AUTOSIZE_USEHEADER);
  176. theCtrl.SetColumnWidth(4, 0);
  177. }
  180. void CIpMonitorView::displayContents(IPHOOK_DATA * data)
  181. {
  182. if (!validIpHookData(data)) {
  184. return;
  185. }
  187. LVITEM item;
  188. CString msg;
  189. CString docItem;
  190. //
  191. // time 
  192. //
  193. msg.Format(_T("%016.16I64x"), data->timestamp);
  195. LPTSTR  tbuf = msg.LockBuffer();
  197. item.mask = LVIF_TEXT | LVIF_PARAM;
  198. item.iItem = 0x7FFFFFFF;
  199. item.iSubItem = 0;
  200. item.pszText = tbuf;
  201. item.cchTextMax = _tcslen( item.pszText ) + 1;
  202. item.lParam = 0;
  204. int row = GetListCtrl().InsertItem(&item);
  206. if (row == -1) {
  208. AfxErrorMessageBox(CString(_T("InsertItem failed")), 0);
  210. return;
  211. }
  213. msg.UnlockBuffer();
  215. docItem = msg;
  217. msg.Format("%08.8x", data->sequence);
  219. tbuf = msg.LockBuffer();
  221. GetListCtrl().SetItemText(row, 1, tbuf);
  223. msg.UnlockBuffer();
  225. docItem += CString(" ") + msg;
  227. //
  228. // S/R
  229. //
  230. if (data->direction) {
  232. msg = _T(" R ");
  234. } else {
  236. msg = _T(" S ");
  238. }
  240. tbuf = msg.LockBuffer();
  242. GetListCtrl().SetItemText(row, 2, tbuf);
  244. msg.UnlockBuffer();
  246. docItem += CString(" ") + msg;
  248. //
  249. // IP Header
  250. //
  251. msg.Format(
  252. _T("VL: %02.2x TOS: %02.2x LEN: %04.4x ID: %04.4x OFF: %04.4x TTL: %02.2x  PROT: %02.2x SRC: %08.8x DEST: %08.8x"),
  253. data->header.iph_verlen,
  254. data->header.iph_tos,
  255. ntohs(data->header.iph_length),
  256. ntohs(data->header.iph_id),
  257. ntohs(data->header.iph_offset),
  258. data->header.iph_ttl,
  259. data->header.iph_protocol,
  260. ntohl(data->header.iph_src),
  261. ntohl(data->header.iph_dest));
  263. tbuf = msg.LockBuffer();
  265. GetListCtrl().SetItemText(row, 3, tbuf);
  267. msg.UnlockBuffer();
  269. docItem += CString(" ") + msg;
  271. if (theApp.m_theDoc) {
  273. theApp.m_theDoc->addItem(docItem);
  274. }
  276. }
  279. int CIpMonitorView::OnCreate(LPCREATESTRUCT lpCreateStruct) 
  280. {
  281. lpCreateStruct->style |= LVS_REPORT;
  283. if (CListView::OnCreate(lpCreateStruct) == -1)
  284. return -1;
  286. // TODO: Add your specialized creation code here
  288. return 0;
  289. }
  291. afx_msg LRESULT CIpMonitorView::OnBufferMessage(WPARAM wParam, LPARAM lParam)
  292. {
  293. IPHOOK_BUFFER * buffer = (IPHOOK_BUFFER * )wParam;
  295. UCHAR * rawBuffer = (UCHAR *) buffer;
  297. if (!validIpHookBuffer(buffer)) {
  299. delete[] rawBuffer;
  301. rawBuffer = NULL;
  303. return 0;
  305. }
  306. //
  307. // lets just assume that it is valid? not!
  308. //
  309. if (buffer->valid > buffer->entries) {
  311. delete[] rawBuffer;
  313. return 0;
  314. }
  316. //
  317. // display the contents
  318. //
  319. for (UINT index = 0; index < buffer->valid; index++) {
  321. displayContents(&buffer->buffer[index]);
  323. }
  324. //
  325. // delete the buffer
  326. //
  328. delete[] rawBuffer;
  330. //
  331. // if scrolling is enabled, make the last item visible
  332. //
  333. // Ensure that the last item is visible.
  334. //
  335. int nCount = GetListCtrl().GetItemCount();
  336. if (nCount > 0) {
  337.    
  338. GetListCtrl().EnsureVisible( nCount -1, FALSE );
  339. }
  342. return 0;
  343. }
  347. void CIpMonitorView::reInitView()
  348. {
  350. GetListCtrl().DeleteAllItems();
  352. }
  354. void CIpMonitorView::OnDocClear() 
  355. {
  356. reInitView();
  358. }
  359. ///////////////////////////////////////////////////////////////////////////////
  360. // 
  361. // Change History Log
  362. //
  363. // $Log: /iphook/usr/IpMonitor/IpMonitorView.cpp $
  364. // 
  365. // 3     1/27/00 10:35p Markr
  366. // Prepare to release!
  367. //
  368. ///////////////////////////////////////////////////////////////////////////////