qqsnifferdlg.cpp

主要是关于贪吃蛇算法在ＱＱ通信中应用．使用ＶＣ＋＋６.０.

// QQSnifferDlg.cpp : 实现文件
//

#include "stdafx.h"
#include "QQSniffer.h"
#include "QQSnifferDlg.h"
#include "md5_en.h"
#include ".\qqsnifferdlg.h"
#include "sniffer.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif



DWORD ThreadStartCapture(void *psnif){
	CSniffer *snif;
	snif=(CSniffer*)psnif;
	snif->Start();
	while(1);
	return 0;
}
// 用于应用程序“关于”菜单项的 CAboutDlg 对话框

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// 对话框数据
	enum { IDD = IDD_ABOUTBOX };

	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持

// 实现
protected:
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
END_MESSAGE_MAP()


// CQQSnifferDlg 对话框



CQQSnifferDlg::CQQSnifferDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CQQSnifferDlg::IDD, pParent)
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CQQSnifferDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_LIST2, m_ListCtrl);
	DDX_Control(pDX, IDC_EDIT1, m_edit);
	DDX_Control(pDX, IDC_EDIT2, m_uid);
	DDX_Control(pDX, IDC_EDIT3, m_pwd);
}

BEGIN_MESSAGE_MAP(CQQSnifferDlg, CDialog)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_MESSAGE(WM_DATAREACH,OnCapture)
	//}}AFX_MSG_MAP
	ON_BN_CLICKED(IDC_BUTTON1, OnBnClickedButton1)
	ON_NOTIFY(LVN_COLUMNCLICK, IDC_LIST2, OnLvnColumnclickList2)
	ON_NOTIFY(LVN_ODSTATECHANGED, IDC_LIST2, OnLvnOdstatechangedList2)
	ON_NOTIFY(NM_CLICK, IDC_LIST2, OnNMClickList2)
	ON_BN_CLICKED(IDC_BUTTON2, OnBnClickedButton2)
	ON_NOTIFY(LVN_KEYDOWN, IDC_LIST2, OnLvnKeydownList2)
END_MESSAGE_MAP()


// CQQSnifferDlg 消息处理程序

BOOL CQQSnifferDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// 将\“关于...\”菜单项添加到系统菜单中。

	// IDM_ABOUTBOX 必须在系统命令范围内。
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		CString strAboutMenu;
		strAboutMenu.LoadString(IDS_ABOUTBOX);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// 设置此对话框的图标。当应用程序主窗口不是对话框时，框架将自动
	//  执行此操作
	SetIcon(m_hIcon, TRUE);			// 设置大图标
	SetIcon(m_hIcon, FALSE);		// 设置小图标
	
	

	m_ListCtrl.InsertColumn(0,"源IP",LVCFMT_LEFT,120);
	m_ListCtrl.InsertColumn(1,"目标",LVCFMT_LEFT,120);
	m_ListCtrl.InsertColumn(2,"源端口",LVCFMT_LEFT,60);
	m_ListCtrl.InsertColumn(3,"目标端口",LVCFMT_LEFT,60);
	m_ListCtrl.InsertColumn(4,"UDP长度",LVCFMT_LEFT,60);
	m_ListCtrl.InsertColumn(5,"QQ包类型",LVCFMT_LEFT,117);

	
	m_itemcount=0;
	m_tid=0;
	m_hThread=0;
	m_uid.SetWindowText("");
	m_pwd.SetWindowText("");

	return TRUE;  // 除非设置了控件的焦点，否则返回 TRUE
}

void CQQSnifferDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。

void CQQSnifferDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // 用于绘制的设备上下文

		SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

		// 使图标在工作矩形中居中
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// 绘制图标
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

//当用户拖动最小化窗口时系统调用此函数取得光标显示。
HCURSOR CQQSnifferDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}

CString CQQSnifferDlg::BufToHex(char *buf,int len)
{
	int i;
	CString ret;
	CString tmp;
	for(i=0;i<len;i++){
		tmp.Format("%2.2X ",(unsigned char)buf[i]);
		ret=ret+tmp;
	}
	ret=ret+"\n";
	return ret;
}
void CQQSnifferDlg::OnBnClickedButton1()
{
	CSniffer *snif=new CSniffer();
	HANDLE hThread;
	DWORD uid;
	char pwd[20];
	char szid[20];
	char szmd5[16];
	if(m_hThread!=0){
		TerminateThread(m_hThread,0);
	}
	m_uid.GetWindowText(szid,19);
	m_pwd.GetWindowText(pwd,19);
	uid=atol(szid);
	md5_en(pwd,szmd5);

	CWnd *tt;
	CString md5_pwd;
	md5_pwd=BufToHex(szmd5,16);
	tt=this->GetDlgItem(IDC_STATIC_MD5);
	tt->SetWindowText(md5_pwd);

	snif->Init(WM_DATAREACH,this->m_hWnd,uid,szmd5);
	
	hThread=CreateThread(0,0,(LPTHREAD_START_ROUTINE)ThreadStartCapture,snif,0,&m_tid);
	if(hThread==NULL){
		m_hThread=0;
		MessageBox("启动监听线程失败!");
		return;
	}
	m_hThread=hThread;
}

LRESULT CQQSnifferDlg::OnCapture(WPARAM wParam,LPARAM lParam)
{
	PACK_HEADER *phdr;
	char *pdata;
	phdr=(PACK_HEADER*)wParam;
	pdata=(char*)lParam;
	char szdPort[10],szsPort[10],szLen[10];
	sprintf(szdPort,"%d",phdr->dPORT);
	sprintf(szsPort,"%d",phdr->sPORT);
	sprintf(szLen,"%d",phdr->Len);
	m_ListCtrl.InsertItem(m_itemcount,phdr->sIP);
	m_ListCtrl.SetItemText(m_itemcount,1,phdr->dIP);
	m_ListCtrl.SetItemText(m_itemcount,2,szsPort);
	m_ListCtrl.SetItemText(m_itemcount,3,szdPort);
	m_ListCtrl.SetItemText(m_itemcount,4,szLen);
	m_ListCtrl.SetItemData(m_itemcount,(DWORD_PTR)pdata);
	m_itemcount++;
	return 0;
}
void CQQSnifferDlg::OnLvnColumnclickList2(NMHDR *pNMHDR, LRESULT *pResult)
{
	LPNMLISTVIEW pNMLV = reinterpret_cast<LPNMLISTVIEW>(pNMHDR);
	*pResult = 0;
}

void CQQSnifferDlg::WriteLog(CString str){
	str.Replace("\n","\r\n");
	m_edit.SetWindowText(str);
}

void CQQSnifferDlg::OnLvnOdstatechangedList2(NMHDR *pNMHDR, LRESULT *pResult)
{
	LPNMLVODSTATECHANGE pStateChanged = reinterpret_cast<LPNMLVODSTATECHANGE>(pNMHDR);
	*pResult = 0;
}

void CQQSnifferDlg::OnNMClickList2(NMHDR *pNMHDR, LRESULT *pResult)
{
	POSITION pos;
	int i;
	char *pstr;
	CString str;
	pos=m_ListCtrl.GetFirstSelectedItemPosition();
	if(pos!=NULL){
		i=m_ListCtrl.GetNextSelectedItem(pos);
		pstr=(char*)m_ListCtrl.GetItemData(i);
		str=CString(pstr);
		WriteLog(str);
	}

	*pResult = 0;
}

void CQQSnifferDlg::OnBnClickedButton2()
{
	if(m_hThread!=0){
		TerminateThread(m_hThread,0);
		MessageBox("已经停止！");
	}
}

void CQQSnifferDlg::OnLvnKeydownList2(NMHDR *pNMHDR, LRESULT *pResult)
{
	LPNMLVKEYDOWN pLVKeyDow = reinterpret_cast<LPNMLVKEYDOWN>(pNMHDR);
	POSITION pos;
	int i;
	char *pstr;
	CString str;
	pos=m_ListCtrl.GetFirstSelectedItemPosition();
	if(pos!=NULL){
		i=m_ListCtrl.GetNextSelectedItem(pos);
		pstr=(char*)m_ListCtrl.GetItemData(i);
		str=CString(pstr);
		WriteLog(str);
	}
	*pResult = 0;
}