smartkiddlg.cpp

smartkid网络安全检测工具

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

#include "stdafx.h"
#include "smartkid.h"
#include "smartkidDlg.h"
#include "AboutCtrl.h"
#include "Login.h"
#include "Splash.h"
#include ".\smartkiddlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif

#define WM_SHOWTASK WM_USER+10

// 用于应用程序“关于”菜单项的 CAboutDlg 对话框

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

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

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

// 实现
protected:
	DECLARE_MESSAGE_MAP()
public:
	CAboutCtrl	m_AboutCtrl;
	afx_msg void OnLButtonDown(UINT nFlags, CPoint point);
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
	CString strCredits = "\t智慧小子 SmartKid\n\n"
		"\rProgrammed by:\n"
		"菠菜 [e-mail]\n\n"
		"\rSpecial thanks to:\n冰血封情\nFr.Qaker\nZV\nGxter\nzzzevazzz\nharambo\nand RedSword\n\n"
		"\rCopyright (c)2006-2007 \n\r梦软科技有限公司\n"
		"\rAll right reserved.\n\n"
		"Solutions at http://www.dream2fly.net\n";

	m_AboutCtrl.SetCredits(strCredits);
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_ABOUTCTRL, m_AboutCtrl);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
	ON_WM_LBUTTONDOWN()
END_MESSAGE_MAP()



void CAboutDlg::OnLButtonDown(UINT nFlags, CPoint point)
{
	SendMessage( WM_CLOSE );

	CDialog::OnLButtonDown(nFlags, point);
}

bool g_stop=true;
// CsmartkidDlg 对话框

CsmartkidDlg::CsmartkidDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CsmartkidDlg::IDD, pParent)
	, m_domain(_T(""))
	, m_isspecial(false)
	, m_startport(21)
	, m_endport(80)
	, m_drdosport(80)
	, m_ddosport(80)
	, m_reflectport(_T("179"))
	, m_scantype(_CONNECT)
	, m_ddostype(_SYN)
	, m_drdostype(_SYN)
	, m_threadnum(200)
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}
CsmartkidDlg::~CsmartkidDlg()
{
}
void CsmartkidDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_URL, m_url);
	DDX_Control(pDX, IDC_EMAIL, m_email);
	DDX_Control(pDX, IDC_REFLECTLIST, m_reflectlist);
	DDX_IPAddress(pDX, IDC_REFLECTIP, m_reflectip);
	DDX_Text(pDX, IDC_TARGETDOMAIN, m_domain);
	DDX_Text(pDX, IDC_STARTPORT, m_startport);
	DDX_Text(pDX, IDC_ENDPORT, m_endport);
	DDX_Text(pDX, IDC_DRDOSPORT, m_drdosport);
	DDX_Text(pDX, IDC_DDOSPORT, m_ddosport);
	DDX_Text(pDX, IDC_REFLECTPORT, m_reflectport);
	DDX_Text(pDX, IDC_THREADNUM, m_threadnum);
	DDX_Control(pDX, IDC_IPPORT, m_listInfo);
	DDX_Control(pDX, IDC_ATTACKIPADDRESS, m_ctrtargetip);
	DDX_Control(pDX, IDC_LOCALIP, m_ctrlocalip);
	DDX_Control(pDX, IDC_REFLECTIP, m_ctrreflectip);
}

BEGIN_MESSAGE_MAP(CsmartkidDlg, CDialog)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	//}}AFX_MSG_MAP
	ON_BN_CLICKED(IDC_ADDIP, OnBnClickedAddip)
	ON_BN_CLICKED(IDC_LOAD, OnBnClickedLoad)
	ON_BN_CLICKED(IDC_SAVE, OnBnClickedSave)
	ON_LBN_DBLCLK(IDC_REFLECTLIST, OnLbnDblclkReflectlist)
	ON_BN_CLICKED(IDC_SCANPORT, OnBnClickedScanport)
	ON_BN_CLICKED(IDC_DOMAINTOIP, OnBnClickedDomaintoip)
	ON_BN_CLICKED(IDC_CHECKALL, OnBnClickedCheckall)
	ON_BN_CLICKED(IDC_DDOSATTACK, OnBnClickedDdosattack)
	ON_BN_CLICKED(IDC_DRDOSATTACK, OnBnClickedDrdosattack)
	ON_BN_CLICKED(IDC_SYN, OnBnClickedSyn)
	ON_BN_CLICKED(IDC_DDOSSYN, OnBnClickedDdossyn)
	ON_BN_CLICKED(IDC_DDOSICMP, OnBnClickedDdosicmp)
	ON_BN_CLICKED(IDC_DRDOSSYN, OnBnClickedDrdossyn)
	ON_BN_CLICKED(IDC_DRDOSICMP, OnBnClickedDrdosicmp)
	ON_BN_CLICKED(IDC_PAUSE, OnBnClickedPause)
	ON_WM_LBUTTONDOWN()
	ON_COMMAND(ID_ABOUT, OnAbout)
	ON_COMMAND(ID_FAQ, OnFaq)
	ON_COMMAND(ID_UPDATE, OnUpdate)
	ON_COMMAND(ID_OPENSOURCE, OnOpensource)
	ON_BN_CLICKED(IDC_TRAY, OnBnClickedTray)
	ON_MESSAGE(WM_SHOWTASK,OnShowTask)
	ON_WM_CLOSE()
END_MESSAGE_MAP()


// CsmartkidDlg 消息处理程序

BOOL CsmartkidDlg::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);		// 设置小图标

#ifndef _DEBUG
	CLogin loginDlg;
	loginDlg.DoModal();

	OS_LEVEL os_level=get_win32_type();
	if(os_level==WIN_XP_SP2)
	{
		AfxMessageBox("系统检测到您的操作系统是WIN_XP_SP2，\n\n此系统目前只能使用connect扫描功能！");
	}
#endif

	CSplashWnd::ShowSplashScreen(this);	
	Sleep(1000);//同步splash延迟

	m_prog=(CProgressCtrl*)GetDlgItem(IDC_SCANPROGRESS);
	((CButton *)GetDlgItem(IDC_CHECKALL))->SetCheck(BST_CHECKED);
	m_isspecial=true;
	GetDlgItem(IDC_STARTPORT)->EnableWindow(FALSE);
	GetDlgItem(IDC_ENDPORT)->EnableWindow(FALSE);

	((CButton *)GetDlgItem(IDC_DDOSSYN))->SetCheck(BST_CHECKED);
	((CButton *)GetDlgItem(IDC_DRDOSSYN))->SetCheck(BST_CHECKED);

	//初始化信息列表框
	m_listInfo.InsertColumn(0,"IP",LVCFMT_LEFT,0,0);
	m_listInfo.InsertColumn(1,"PORT",LVCFMT_LEFT,0,0);
	m_listInfo.SetColumnWidth(0,100);
	m_listInfo.SetColumnWidth(1,60);


	///<获取本机IP地址
	m_localip=GetLocalIP();
	m_ctrlocalip.SetWindowText("您的本机IP:"+m_localip);	
	//初始化窗体IP

	//m_ctrtargetip.SetAddress(192,168,1,139);
	m_ctrreflectip.ClearAddress();

	m_url.SetURL(_T("http://www.dream2fly.net"));
	m_url.SetUnderline(FALSE);

	m_email.SetURL(_T("mailto:[e-mail]"));
	m_email.SetUnderline(FALSE);

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

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

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

void CsmartkidDlg::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 CsmartkidDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}

void CsmartkidDlg::OnBnClickedAbout()
{
	CAboutDlg dlgAbout;
	dlgAbout.DoModal();
}

void CsmartkidDlg::OnBnClickedAddip()
{
	UpdateData(true);
	if(m_reflectip>0)
	{
		char* reflectip=new char[256];
		u_long reflect_ip=htonl(m_reflectip);
		TransformIp(reflectip,reflect_ip,true);
		strcat(reflectip,":");
		m_reflectlist.AddString(reflectip+m_reflectport);
		delete 	[]reflectip;
	}	
}

void CsmartkidDlg::OnBnClickedLoad()
{
	CFileDialog LoadDlg(true);
	LoadDlg.m_ofn.lpstrTitle = "导入反射源地址";
	if (LoadDlg.DoModal()==IDOK)
	{
		CString strFileName=LoadDlg.GetPathName();
		CFile file(strFileName,CFile::modeRead);
		UINT len=static_cast<UINT>(file.GetLength());
		char *buf = new char[len];
		file.Read(buf,len);
		UINT i=0,j=0;
		while(i<len)
		{
			if ((buf[i]=='\r')&&(buf[i+1]=='\n'))
			{
				buf[i] = '\0';
				m_reflectlist.AddString(buf + j);
				j = i + 2;
				i+=2;
			}
			else if ((buf[i]!='\r')&&(buf[i]=='\n'))
			{
				buf[i] = '\0';
				m_reflectlist.AddString(buf + j);
				j = i + 1;
				++i;
			}
			else
			{
				++i;
			}
		}
		delete []buf;
		file.Close();
	}
}

void CsmartkidDlg::OnBnClickedSave()
{
	CFileDialog SaveDlg(false);
	SaveDlg.m_ofn.lpstrTitle = "导出反射源地址";
	if (SaveDlg.DoModal()==IDOK)
	{
		CString strFileName=SaveDlg.GetPathName ();
		CFile file(strFileName,CFile::modeWrite|CFile::modeCreate);
		CString strCurSel;
		CString strEnd="\r\n";
		for(int i=0;i<m_reflectlist.GetCount();i++)
		{
			m_reflectlist.GetText(i,strCurSel);
			strCurSel.Append(strEnd);
			file.Write (strCurSel,strCurSel.GetLength());
		}
		file.Close () ;
		MessageBox("成功导出反射源地址列表!");
	}
}

void CsmartkidDlg::OnLbnDblclkReflectlist()
{
	m_reflectlist.DeleteString (m_reflectlist.GetCurSel());
}

BOOL CsmartkidDlg::check_target_ip(void)
{
	if(m_ctrtargetip.IsBlank())
	{
		MessageBox("请输入IP地址",_T("错误"),MB_ICONWARNING|MB_OK);
		return -1;
	}
	DWORD ip;

	if(m_ctrtargetip.GetAddress(ip)<4)
	{
		MessageBox("请输入完整的IP地址",_T("错误"),MB_ICONWARNING|MB_OK);
		return -1;
	}

	m_ctrtargetip.GetWindowText(m_targetip);
	if(0==m_targetip.Compare("208.113.145.214"))
	{
		AfxMessageBox("用偶的工具攻击俺的站，不厚道啊!");
		return -1;
	}

	return 0;
}
void CsmartkidDlg::OnBnClickedScanport()
{
	g_stop=false;
	if(-1==check_target_ip())
		return;	
	
	if(m_startport>m_endport)
	{
		MessageBox(_T("开始端口必须小于等于结束端口！"));
		return;
	}

	m_listInfo.DeleteAllItems();
	UpdateData(true);

	m_portscan.start_scan();
}

void CsmartkidDlg::OnBnClickedDdosattack()
{
	g_stop=false;
	if(-1==check_target_ip())
		return;
	UpdateData(true);
	m_ddos.start_ddos();
}

void CsmartkidDlg::OnBnClickedDrdosattack()
{
	g_stop=false;
	UpdateData(true);
	if(-1==check_target_ip())
		return;

	if(m_reflectlist.GetCount()==0)
	{
		AfxMessageBox("反射源列表不能为空!");
		return;
	}

	std::vector<CString> reflectlist;
	CString strCurSel;
	for(int i=0;i<m_reflectlist.GetCount();i++)
	{
		m_reflectlist.GetText(i,strCurSel);
		reflectlist.push_back(strCurSel);		
	}

	m_drdos.start_drdos(reflectlist);
}

void CsmartkidDlg::OnBnClickedDomaintoip()
{
	UpdateData(true);

	if (m_domain.Find ("dream2fly.net",0)>=0) 
	{
		AfxMessageBox("用偶的工具攻击俺的站，不厚道啊!");
		return;
	}
	if (m_domain.Find ("eviloctal.com",0)>=0) 
	{
		AfxMessageBox("曾经沧海,八进制门前止步!");
		return;
	}

	hostent* pHostent = gethostbyname(m_domain.GetBuffer());
	if (pHostent == 0) 
	{
		AfxMessageBox("输入的域名无效!");
		return;
	}			
	for (int nAdapter=0; pHostent->h_addr_list[nAdapter]; nAdapter++) 
	{
		char* ipaddr = NULL; 
		in_addr inaddr; 
		inaddr.s_addr=*(DWORD*)pHostent->h_addr_list[nAdapter]; 
		ipaddr= inet_ntoa(inaddr); 
		m_ctrtargetip.SetWindowText(ipaddr);
		break;
	}

	UpdateData(false);
}

void CsmartkidDlg::OnBnClickedCheckall()
{
	m_isspecial=m_isspecial?false:true;
	if(m_isspecial)
	{	
		GetDlgItem(IDC_STARTPORT)->EnableWindow(FALSE);
		GetDlgItem(IDC_ENDPORT)->EnableWindow(FALSE);
	}
	else
	{
		GetDlgItem(IDC_STARTPORT)->EnableWindow(TRUE);
		GetDlgItem(IDC_ENDPORT)->EnableWindow(TRUE);
	}
}
void CsmartkidDlg::OnBnClickedSyn()
{
	if(m_scantype==_CONNECT)
	{
		m_scantype=_SYN;
	}
	else
	{
		m_scantype=_CONNECT;
	}
}

void CsmartkidDlg::OnBnClickedDdossyn()
{
	m_ddostype=_SYN;
}

void CsmartkidDlg::OnBnClickedDdosicmp()
{
	m_ddostype=_ICMP;
}

void CsmartkidDlg::OnBnClickedDrdossyn()
{
	m_drdostype=_SYN;
}

void CsmartkidDlg::OnBnClickedDrdosicmp()
{
	m_drdostype=_ICMP;
}

void CsmartkidDlg::OnBnClickedPause()
{
	g_stop=true;
}


void CsmartkidDlg::OnLButtonDown(UINT nFlags, CPoint point)
{
	//鼠标拖动窗口
	SendMessage(WM_SYSCOMMAND,0xF012,0);

	CDialog::OnLButtonDown(nFlags, point);
}

void CsmartkidDlg::OnAbout()
{
	CAboutDlg dlgAbout;
	dlgAbout.DoModal();
}

void CsmartkidDlg::OnFaq()
{
	WinExec("notepad.exe fag.txt",SW_SHOW); 
}

void CsmartkidDlg::OnUpdate()
{
	UINT ret=WinExec("update.exe smartkid 0.90 http://www.dream2fly.net/projects/smartkid_update.ini" ,SW_SHOW); 
	if(ERROR_FILE_NOT_FOUND==ret)
	{
		AfxMessageBox("梦软升级软件丢失！请确认update.exe是否存在！");
	}
}
void CsmartkidDlg::OnOpensource()
{
	CString url="http://www.dream2fly.net/";
	ShellExecute(NULL, _T("open"), url, NULL,NULL, SW_NORMAL);
}
void CsmartkidDlg::OnBnClickedTray()
{
	NOTIFYICONDATA nid;

	nid.cbSize=(DWORD)sizeof(NOTIFYICONDATA);

	nid.hWnd=this->m_hWnd;

	nid.uID=IDR_MAINFRAME;

	nid.uFlags=NIF_ICON|NIF_MESSAGE|NIF_TIP ;

	nid.uCallbackMessage=WM_SHOWTASK;//自定义的消息名称

	nid.hIcon=LoadIcon(AfxGetInstanceHandle(),MAKEINTRESOURCE(IDR_MAINFRAME));

	strcpy(nid.szTip,"智慧小子SmartKid");//信息提示条

	Shell_NotifyIcon(NIM_ADD,&nid);//在托盘区添加图标

	ShowWindow(SW_HIDE);//隐藏主窗口
}
LRESULT CsmartkidDlg::OnShowTask(WPARAM wParam,LPARAM lParam)
//wParam接收的是图标的ID，而lParam接收的是鼠标的行为
{
	if(wParam!=IDR_MAINFRAME)
	{
		return 1;
	}

	switch(lParam)
	{
		case WM_RBUTTONUP://右键起来时弹出快捷菜单，这里只有一个“关闭”
		{
			{
				POINT point;
				::GetCursorPos(&point);//得到鼠标位置

				CMenu menu;
				menu.CreatePopupMenu();//声明一个弹出式菜单

				//增加菜单项“关闭”，点击则发送消息WM_DESTROY给主窗口（已
				//隐藏），将程序结束。
				menu.AppendMenu(MF_STRING,WM_DESTROY,"关闭"); 

				//确定弹出式菜单的位置
				menu.TrackPopupMenu(TPM_LEFTALIGN,point.x,point.y,this);

				//资源回收
				HMENU hmenu=menu.Detach();
				menu.DestroyMenu();
			}
			break;
		}
	case WM_LBUTTONDBLCLK://双击左键的处理
		{
			static bool IsTray=true;
			if(IsTray)
			{		
				IsTray=false;
				this->ShowWindow(SW_SHOW);//简单的显示主窗口完事儿
			}
			else
			{
				IsTray=true;
				this->ShowWindow(SW_HIDE);//简单的隐藏主窗口完事儿
			}

		}
		break;
	}

	return 0;

}
void CsmartkidDlg::OnClose()
{
	if(g_stop==false)
	{
		AfxMessageBox("请先停止攻击或扫描线程！");
		return;
	}
	
	CDialog::OnClose();
}