IpMonitorView.cpp
资源名称:IPbao.rar [点击查看]
上传用户:ykzxjx
上传日期:2022-04-03
资源大小:1175k
文件大小:8k
源码类别:
防火墙与安全工具
开发平台:
Visual C++
- ///////////////////////////////////////////////////////////////////////////////
- //
- // (C) Copyright 1999 - 2000 Mark Roddy
- // All Rights Reserved
- //
- // Hollis Technology Solutions
- // 94 Dow Road
- // Hollis, NH 03049
- // info@hollistech.com
- //
- // Synopsis:
- //
- //
- // Version Information:
- //
- // $Header: /iphook/usr/IpMonitor/IpMonitorView.cpp 3 1/27/00 10:35p Markr $
- //
- ///////////////////////////////////////////////////////////////////////////////
- // IpMonitorView.cpp : implementation of the CIpMonitorView class
- //
- #include "stdafx.h"
- #include "IpMonitor.h"
- #include "IpMonitorDoc.h"
- #include "IpMonitorView.h"
- #include "Mainfrm.h"
- #include "winsock2.h"
- #ifdef _DEBUG
- #define new DEBUG_NEW
- #undef THIS_FILE
- static char THIS_FILE[] = __FILE__;
- #endif
- /////////////////////////////////////////////////////////////////////////////
- // CIpMonitorView
- IMPLEMENT_DYNCREATE(CIpMonitorView, CListView)
- BEGIN_MESSAGE_MAP(CIpMonitorView, CListView)
- //{{AFX_MSG_MAP(CIpMonitorView)
- ON_WM_CREATE()
- ON_MESSAGE(BUFFER_AVAILABLE_MSG, OnBufferMessage)
- ON_COMMAND(ID_DOC_CLEAR, OnDocClear)
- //}}AFX_MSG_MAP
- // Standard printing commands
- ON_COMMAND(ID_FILE_PRINT, CView::OnFilePrint)
- ON_COMMAND(ID_FILE_PRINT_DIRECT, CView::OnFilePrint)
- ON_COMMAND(ID_FILE_PRINT_PREVIEW, CView::OnFilePrintPreview)
- END_MESSAGE_MAP()
- /////////////////////////////////////////////////////////////////////////////
- // CIpMonitorView construction/destruction
- CIpMonitorView::CIpMonitorView()
- {
- m_dwDefaultStyle |= LVS_REPORT;
- theApp.m_theView = this;
- }
- CIpMonitorView::~CIpMonitorView()
- {
- }
- BOOL CIpMonitorView::PreCreateWindow(CREATESTRUCT& cs)
- {
- // TODO: Modify the Window class or styles here by modifying
- // the CREATESTRUCT cs
- cs.style |= LVS_REPORT;
- return CListView::PreCreateWindow(cs);
- }
- /////////////////////////////////////////////////////////////////////////////
- // CIpMonitorView drawing
- void CIpMonitorView::OnDraw(CDC* pDC)
- {
- CIpMonitorDoc* pDoc = GetDocument();
- ASSERT_VALID(pDoc);
- // TODO: add draw code for native data here
- }
- /////////////////////////////////////////////////////////////////////////////
- // CIpMonitorView printing
- BOOL CIpMonitorView::OnPreparePrinting(CPrintInfo* pInfo)
- {
- // default preparation
- return DoPreparePrinting(pInfo);
- }
- void CIpMonitorView::OnBeginPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
- {
- // TODO: add extra initialization before printing
- }
- void CIpMonitorView::OnEndPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
- {
- // TODO: add cleanup after printing
- }
- /////////////////////////////////////////////////////////////////////////////
- // CIpMonitorView diagnostics
- #ifdef _DEBUG
- void CIpMonitorView::AssertValid() const
- {
- CListView::AssertValid();
- }
- void CIpMonitorView::Dump(CDumpContext& dc) const
- {
- CListView::Dump(dc);
- }
- CIpMonitorDoc* CIpMonitorView::GetDocument() // non-debug version is inline
- {
- ASSERT(m_pDocument->IsKindOf(RUNTIME_CLASS(CIpMonitorDoc)));
- return (CIpMonitorDoc*)m_pDocument;
- }
- #endif //_DEBUG
- /////////////////////////////////////////////////////////////////////////////
- // CIpMonitorView message handlers
- void CIpMonitorView::OnInitialUpdate()
- {
- CListView::OnInitialUpdate();
- // TODO: Add your specialized code here and/or call the base class
- // this code only works for a report-mode list view
- ASSERT(GetStyle() & LVS_REPORT);
- // Gain a reference to the list control itself
- CListCtrl& theCtrl = GetListCtrl();
- // Insert a column. This override is the most convenient.
- theCtrl.InsertColumn(0, _T("TIME "), LVCFMT_LEFT);
- // The other InsertColumn() override requires an initialized
- // LVCOLUMN structure.
- LVCOLUMN col;
- col.mask = LVCF_FMT | LVCF_TEXT;
- col.pszText = _T("SEQ ");
- col.fmt = LVCFMT_LEFT;
- theCtrl.InsertColumn(1, &col);
- col.mask = LVCF_FMT | LVCF_TEXT;
- col.pszText = _T("S/R");
- col.fmt = LVCFMT_LEFT;
- theCtrl.InsertColumn(2, &col);
- col.mask = LVCF_FMT | LVCF_TEXT;
- col.pszText = _T("IP HEADER ");
- col.fmt = LVCFMT_LEFT;
- theCtrl.InsertColumn(3, &col);
- //
- // this last column is just and 'end of row marker'.
- //
- col.mask = LVCF_FMT | LVCF_TEXT;
- col.pszText = _T("");
- col.fmt = LVCFMT_LEFT;
- theCtrl.InsertColumn(4, &col);
- // Set reasonable widths for our columns
- theCtrl.SetColumnWidth(0, LVSCW_AUTOSIZE_USEHEADER);
- theCtrl.SetColumnWidth(1, LVSCW_AUTOSIZE_USEHEADER);
- theCtrl.SetColumnWidth(2, LVSCW_AUTOSIZE_USEHEADER);
- theCtrl.SetColumnWidth(3, LVSCW_AUTOSIZE_USEHEADER);
- theCtrl.SetColumnWidth(4, 0);
- }
- void CIpMonitorView::displayContents(IPHOOK_DATA * data)
- {
- if (!validIpHookData(data)) {
- return;
- }
- LVITEM item;
- CString msg;
- CString docItem;
- //
- // time
- //
- msg.Format(_T("%016.16I64x"), data->timestamp);
- LPTSTR tbuf = msg.LockBuffer();
- item.mask = LVIF_TEXT | LVIF_PARAM;
- item.iItem = 0x7FFFFFFF;
- item.iSubItem = 0;
- item.pszText = tbuf;
- item.cchTextMax = _tcslen( item.pszText ) + 1;
- item.lParam = 0;
- int row = GetListCtrl().InsertItem(&item);
- if (row == -1) {
- AfxErrorMessageBox(CString(_T("InsertItem failed")), 0);
- return;
- }
- msg.UnlockBuffer();
- docItem = msg;
- msg.Format("%08.8x", data->sequence);
- tbuf = msg.LockBuffer();
- GetListCtrl().SetItemText(row, 1, tbuf);
- msg.UnlockBuffer();
- docItem += CString(" ") + msg;
- //
- // S/R
- //
- if (data->direction) {
- msg = _T(" R ");
- } else {
- msg = _T(" S ");
- }
- tbuf = msg.LockBuffer();
- GetListCtrl().SetItemText(row, 2, tbuf);
- msg.UnlockBuffer();
- docItem += CString(" ") + msg;
- //
- // IP Header
- //
- msg.Format(
- _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"),
- data->header.iph_verlen,
- data->header.iph_tos,
- ntohs(data->header.iph_length),
- ntohs(data->header.iph_id),
- ntohs(data->header.iph_offset),
- data->header.iph_ttl,
- data->header.iph_protocol,
- ntohl(data->header.iph_src),
- ntohl(data->header.iph_dest));
- tbuf = msg.LockBuffer();
- GetListCtrl().SetItemText(row, 3, tbuf);
- msg.UnlockBuffer();
- docItem += CString(" ") + msg;
- if (theApp.m_theDoc) {
- theApp.m_theDoc->addItem(docItem);
- }
- }
- int CIpMonitorView::OnCreate(LPCREATESTRUCT lpCreateStruct)
- {
- lpCreateStruct->style |= LVS_REPORT;
- if (CListView::OnCreate(lpCreateStruct) == -1)
- return -1;
- // TODO: Add your specialized creation code here
- return 0;
- }
- afx_msg LRESULT CIpMonitorView::OnBufferMessage(WPARAM wParam, LPARAM lParam)
- {
- IPHOOK_BUFFER * buffer = (IPHOOK_BUFFER * )wParam;
- UCHAR * rawBuffer = (UCHAR *) buffer;
- if (!validIpHookBuffer(buffer)) {
- delete[] rawBuffer;
- rawBuffer = NULL;
- return 0;
- }
- //
- // lets just assume that it is valid? not!
- //
- if (buffer->valid > buffer->entries) {
- delete[] rawBuffer;
- return 0;
- }
- //
- // display the contents
- //
- for (UINT index = 0; index < buffer->valid; index++) {
- displayContents(&buffer->buffer[index]);
- }
- //
- // delete the buffer
- //
- delete[] rawBuffer;
- //
- // if scrolling is enabled, make the last item visible
- //
- // Ensure that the last item is visible.
- //
- int nCount = GetListCtrl().GetItemCount();
- if (nCount > 0) {
- GetListCtrl().EnsureVisible( nCount -1, FALSE );
- }
- return 0;
- }
- void CIpMonitorView::reInitView()
- {
- GetListCtrl().DeleteAllItems();
- }
- void CIpMonitorView::OnDocClear()
- {
- reInitView();
- }
- ///////////////////////////////////////////////////////////////////////////////
- //
- // Change History Log
- //
- // $Log: /iphook/usr/IpMonitor/IpMonitorView.cpp $
- //
- // 3 1/27/00 10:35p Markr
- // Prepare to release!
- //
- ///////////////////////////////////////////////////////////////////////////////