mainfrm.cpp

这是一个嗅探器

// MainFrm.cpp : implementation of the CMainFrame class
//

#include "stdafx.h"
#include "SnifferPro.h"
#include "MainFrm.h"
#include "StcView.h"
#include "DlgAdp.h"
#include "DlgFlt.h"
#include "DlgRea.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CMainFrame

IMPLEMENT_DYNCREATE(CMainFrame, CFrameWnd)

BEGIN_MESSAGE_MAP(CMainFrame, CFrameWnd)
	//{{AFX_MSG_MAP(CMainFrame)
	ON_WM_CREATE()
	ON_WM_TIMER()
	ON_COMMAND(ID_OPT_ADPATER, OnOptAdpater)
	ON_COMMAND(ID_FILE_START, OnFileStart)
	ON_COMMAND(ID_FILE_STOP, OnFileStop)
	ON_UPDATE_COMMAND_UI(ID_FILE_START, OnUpdateFileStart)
	ON_UPDATE_COMMAND_UI(ID_FILE_STOP, OnUpdateFileStop)
	ON_COMMAND(ID_OPT_FILTER, OnOptFilter)
	ON_WM_CLOSE()
	ON_COMMAND(ID_FILE_REASSEMBLY, OnFileReassembly)
	ON_UPDATE_COMMAND_UI(ID_FILE_REASSEMBLY, OnUpdateFileReassembly)
	ON_UPDATE_COMMAND_UI(ID_OPT_ADPATER, OnUpdateOptAdpater)
	ON_UPDATE_COMMAND_UI(ID_OPT_FILTER, OnUpdateOptFilter)
	ON_COMMAND(ID_FILE_ADVANCE, OnFileAdvance)
	ON_UPDATE_COMMAND_UI(ID_FILE_ADVANCE, OnUpdateFileAdvance)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

static UINT indicators[] =
{
	ID_SEPARATOR,          // status line indicator
	ID_INDICATOR_CAPS,
	ID_INDICATOR_NUM,
	ID_INDICATOR_SCRL,
};

/////////////////////////////////////////////////////////////////////////////
// CMainFrame construction/destruction

CMainFrame::CMainFrame()
{
	// TODO: add member initialization code here
	alldevs=NULL;
	dev=NULL;
	isStart=false;
	isStop=true;
	isAddrA=false;
	isAddrB=false;
	//addrA=0xc0a80003;//192.168.0.3
	//addrB=0xca7802b4;//202.120.2.180
	//addrB=0xca780201;//202.120.2.1
	dirmode=0;
	filtertype[0]=TRUE;//tcp
	filtertype[1]=TRUE;//udp
	filtertype[2]=TRUE;//icmp
	filtertype[3]=TRUE;//ospf
	filtertype[4]=TRUE;//arp
	filtertype[5]=TRUE;//rarp
	//filtertype[6]=TRUE;
	//isDown=FALSE;
	//fp=NULL;
	//isDown=TRUE;
	//fp=fopen("down.txt","w+");
}

CMainFrame::~CMainFrame()
{
}

int CMainFrame::OnCreate(LPCREATESTRUCT lpCreateStruct)
{
	if (CFrameWnd::OnCreate(lpCreateStruct) == -1)
		return -1;
	
	if (!m_wndToolBar.CreateEx(this, TBSTYLE_FLAT, WS_CHILD | WS_VISIBLE | CBRS_TOP
		| CBRS_GRIPPER | CBRS_TOOLTIPS | CBRS_FLYBY | CBRS_SIZE_DYNAMIC) ||
		!m_wndToolBar.LoadToolBar(IDR_MAINFRAME))
	{
		TRACE0("Failed to create toolbar\n");
		return -1;      // fail to create
	}

	if (!m_wndStatusBar.Create(this) ||
		!m_wndStatusBar.SetIndicators(indicators,
		  sizeof(indicators)/sizeof(UINT)))
	{
		TRACE0("Failed to create status bar\n");
		return -1;      // fail to create
	}
	
	// TODO: Delete these three lines if you don't want the toolbar to
	//  be dockable
	m_wndToolBar.EnableDocking(CBRS_ALIGN_ANY);
	EnableDocking(CBRS_ALIGN_ANY);
	DockControlBar(&m_wndToolBar);

	m_progStatusBar.Create(this);//m_progStatusBar.m_wndProgBar.SetRange(0,100);
	return 0;
}

BOOL CMainFrame::PreCreateWindow(CREATESTRUCT& cs)
{
	if( !CFrameWnd::PreCreateWindow(cs) )
		return FALSE;
	// TODO: Modify the Window class or styles here by modifying
	//  the CREATESTRUCT cs

	return TRUE;
}

/////////////////////////////////////////////////////////////////////////////
// CMainFrame diagnostics

#ifdef _DEBUG
void CMainFrame::AssertValid() const
{
	CFrameWnd::AssertValid();
}

void CMainFrame::Dump(CDumpContext& dc) const
{
	CFrameWnd::Dump(dc);
}

#endif //_DEBUG

/////////////////////////////////////////////////////////////////////////////
// CMainFrame message handlers


BOOL CMainFrame::OnCreateClient(LPCREATESTRUCT lpcs, CCreateContext* pContext) 
{
	// TODO: Add your specialized code here and/or call the base class
	//将用户区分成3各视图
	if	(m_wndSplitter1.CreateStatic(this,1,2)==NULL)
		return FALSE;
	if	(m_wndSplitter2.CreateStatic(&m_wndSplitter1,2,1,WS_CHILD | WS_VISIBLE,m_wndSplitter1.IdFromRowCol(0,0))==NULL)
		return FALSE;
	m_wndSplitter1.CreateView(0,1,RUNTIME_CLASS(CSinPackView),CSize(100,100),pContext);
	m_wndSplitter2.CreateView(0,0,RUNTIME_CLASS(CMulPackView),CSize(100,100),pContext);
	m_wndSplitter2.CreateView(1,0,RUNTIME_CLASS(CStcView),CSize(100,100),pContext);
	SetTimer(1,10,NULL);
	return TRUE;
}

void CMainFrame::OnTimer(UINT nIDEvent) 
{
	// TODO: Add your message handler code here and/or call default
	CRect rect;
	switch(nIDEvent){
	case 1:
		//设置视图高和宽
		GetClientRect(rect);
		m_wndSplitter1.SetColumnInfo(0,rect.Width()/16*11,0);
		m_wndSplitter1.SetColumnInfo(1,rect.Width()/16*5,0);
		m_wndSplitter2.SetRowInfo(0,rect.Height()/16*10-20,0);
		m_wndSplitter2.SetRowInfo(1,rect.Height()/16*6,0);
		m_wndSplitter1.RecalcLayout();
		m_wndSplitter2.RecalcLayout();
		KillTimer(1);
		break;
	case 2:
		::PostMessage(this->mulPackView->m_hWnd,WM_MESSAGE_ONTIMER,0,0);
		break;
	}
	CFrameWnd::OnTimer(nIDEvent);
}

void CMainFrame::OnOptAdpater() 
{
	// TODO: Add your command handler code here
	CDlgAdp dlgAdp;//为Adpater对话框初始化值
	dlgAdp.mainFrm=this;
	dlgAdp.alldevs=this->alldevs;
	dlgAdp.dev=this->dev;
	if(dlgAdp.DoModal()==IDOK){//返回Adpater对话框用户输入信息
		this->alldevs=dlgAdp.alldevs;
		this->dev=dlgAdp.dev;
	}
		return ;
}

void CMainFrame::OnFileStart() 
{
	// TODO: Add your command handler code here
	
	if(dev==NULL){
		MessageBox("请选择网卡!");
		return ;
	}
	isStart=true;//设置状态
	isStop=false;
	//以下为清空上次抓取内容,进行初始化
	for(int i=0;i<this->mulPackView->pkt_datas.GetSize();i++){
		delete[] (unsigned char *)this->mulPackView->pkt_datas.GetAt(i);//释放所捕获得包
		delete (struct pcap_pkthdr *)this->mulPackView->pkt_headers.GetAt(i);//释放每个包的pcap头信息
	}
	this->founders.RemoveAll();//清空定位器
	//以下为清空上次抓取内容
	this->mulPackView->pkt_headers.RemoveAll();
	this->mulPackView->pkt_datas.RemoveAll();
	this->mulPackView->time=0;
	this->mulPackView->totalmem=0;
	this->mulPackView->delta=0;
	this->mulPackView->GetListCtrl().DeleteAllItems();
	this->sinPackView->GetTreeCtrl().DeleteAllItems();
	this->stcView->m_list.DeleteAllItems();
	this->stcView->isReassembly=FALSE;
	//if(this->stcView->buf!=NULL)
		//delete this->stcView->buf;
	//this->conView->GetListCtrl().DeleteAllItems();

	SetTimer(2,1000,NULL);//设置时间,没过一秒统计包数量和总字节
	
	AfxBeginThread(AFX_THREADPROC(ReceivePacket),(LPVOID)this);//启动收包线程
}

void CMainFrame::OnFileStop() 
{
	// TODO: Add your command handler code here
	isStop=true;
	isStart=false;
	KillTimer(2);
	this->m_progStatusBar.OnProgress(-1);
	/*if(isDown){
		if(fclose(fp)==0){
			fp=NULL;
		}
		else
			MessageBox("File Close Failed!");
	}*/
}

void CMainFrame::OnUpdateFileStart(CCmdUI* pCmdUI) 
{
	// TODO: Add your command update UI handler code here
	if(isStart)
		pCmdUI->Enable(false);
}

void CMainFrame::OnUpdateFileStop(CCmdUI* pCmdUI) 
{
	// TODO: Add your command update UI handler code here
	if(isStop)
		pCmdUI->Enable(false);
}

void pcap_handle(u_char *user, const struct pcap_pkthdr *pkt_header, const u_char *pkt_data)
{
	if(((CMainFrame *)AfxGetApp()->GetMainWnd())->isStart==true){
		if(((CMainFrame *)AfxGetApp()->GetMainWnd())->Filter(pkt_data)==TRUE)//若过滤成功
			::PostMessage(*(((CMainFrame *)AfxGetApp()->GetMainWnd())->mulPackView),WM_MESSAGE_PACKET_RECEIVE,(WPARAM)pkt_header,(LPARAM)pkt_data);//通知CMulView处理收到的包
		((CMainFrame *)AfxGetApp()->GetMainWnd())->mulPackView->delta+=pkt_header->caplen;//统计当前网络状况,不考虑过滤
	}
}

UINT ReceivePacket(LPVOID param)
{
	CMainFrame *mainFrm=(CMainFrame*)param;
	pcap_t *p;
	char	errbuf[PCAP_ERRBUF_SIZE];

	if((p=pcap_open_live(mainFrm->dev->name,65536,1,1000,errbuf))==NULL){//若网卡无法绑定
		::MessageBox(*mainFrm,"网卡打开失败!",NULL,0);
		return -1;
	}
	
	do{//进行抓包,每抓一个,调用pcap_handle函数
		pcap_loop(p,1,pcap_handle,NULL);
	}while(mainFrm->isStop==false);

	pcap_close(p);//关闭网卡
	return 0;
} 

void CMainFrame::OnOptFilter() 
{
	// TODO: Add your command handler code here
	CDlgFlt dlgFlt;
	dlgFlt.mainFrm=this;//初始化Filter的筛选信息
	dlgFlt.m_checkA=this->isAddrA;
	dlgFlt.m_checkB=this->isAddrB;
	dlgFlt.ipA=this->addrA;
	dlgFlt.ipB=this->addrB;
	dlgFlt.m_dirmode=this->dirmode;
	for(int i=0;i<TYPE_COUNT;i++){
		dlgFlt.type[i]=this->filtertype[i];
	}

	if(dlgFlt.DoModal()==IDOK){//从Filter对话框中获取输入筛选信息