beastscanerdlg.cpp

综合扫描程序 提供各种扫描编程

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

void CBeastScanerDlg::OnBnClickedPause()
{
	// TODO: 在此添加控件通知处理程序代码
	if(this->enCurrentScanType!=NO_SCAN)
	{
		this->cTabCtrl.EnableWindow();
		if(bPause==false)
		{
			this->cButtonPause.SetWindowText("Resume");
			bPause=true;
			this->stCurrentState="Scan is Paused.";
			this->UpdateData(FALSE);
		}
		else
		{
			this->cButtonPause.SetWindowText("Pause");
			bPause=false;
			this->stCurrentState="Resume Scan";
			this->UpdateData(FALSE);
		}
	}
}

void CBeastScanerDlg::OnBnClickedStop()
{
	// TODO: 在此添加控件通知处理程序代码
	if(this->enCurrentScanType!=NO_SCAN)
	{
		this->enCurrentScanType=NO_SCAN;
		this->cTabCtrl.EnableWindow();
	
		this->stCurrentState="Scan is Stoped.";
		this->UpdateData(FALSE);
		bStop=true;
	}
	this->cButtonPause.SetWindowText("Pause");
	bPause=false;
}

LRESULT CBeastScanerDlg::OnBeginScan(WPARAM enCurrentScanType,LPARAM pThreadParament)
{
	if(this->enCurrentScanType==NO_SCAN)	
	{	
		if(((ThreadParament*)pThreadParament)->dwOriginalIP>((ThreadParament*)pThreadParament)->dwLastIP||((ThreadParament*)pThreadParament)->dwOriginalPort>((ThreadParament*)pThreadParament)->dwLastPort)
		{
			this->MessageBox("Set Parament Error ,Check what you set!","Error",MB_OK|MB_ICONERROR);
			return 0;
		}
		bStop=false;
		dwCurrentHostNo=0;
		dwCurrentPortInfoNo=0;
		this->enCurrentScanType=(SCAN_TYPE)enCurrentScanType;
		if(!(this->enCurrentScanType!=ARP_SCAN &&this->enCurrentScanType!=ARP_FAST_SCAN &&this->enCurrentScanType!=ICMP_WRONG_PORT_SCAN&&this->enCurrentScanType!=ICMP_IP_REORGANIZATION_SCAN&&this->enCurrentScanType!=ICMP_WRONG_PROTOCOL_SCAN&&this->enCurrentScanType!=ICMP_SCAN))
		{
			this->cListHostInfo.DeleteAllItems();
		}
		this->cListPortInfo.DeleteAllItems();
		this->cTabCtrl.EnableWindow(FALSE);
		
		this->stCurrentState="Waiting for Scan Finish…………";
		this->UpdateData(TRUE);
		((ThreadParament*)pThreadParament)->SelectDev=this->SelectDev;
		((ThreadParament*)pThreadParament)->dwHostIP=this->dwLocalIP;
		((ThreadParament*)pThreadParament)->dwDefaultGateway=this->dwDefaultGateway;
		((ThreadParament*)pThreadParament)->dwNetMAsk=this->dwNetMask;
		memcpy(((ThreadParament*)pThreadParament)->HostMac,this->bLocalMac,6);
		ScanHandler(this->enCurrentScanType,(ThreadParament*)pThreadParament);
	}
	return 0;
}
LRESULT CBeastScanerDlg::OnHandleUdpScan(WPARAM wparam,LPARAM lparam)
{
	if(this->dwCurrentPortInfoNo==0xfffffffe)
	{
		this->MessageBox("Too many PortInfo!");
		return 0;
	}
	this->cListPortInfo.InsertItem(dwCurrentPortInfoNo,"Udp Scan");
	CString Tmp;
	Tmp.Format("%u",0xffff&lparam);
	in_addr inHostIp;
	inHostIp.S_un.S_addr=htonl((DWORD)wparam);
	this->cListPortInfo.SetItemText(this->dwCurrentPortInfoNo,1,inet_ntoa(inHostIp));
	this->cListPortInfo.SetItemText(this->dwCurrentPortInfoNo,2,Tmp.GetBuffer());
	if((0xffff0000&lparam)!=0)
	{
		this->cListPortInfo.SetItemText(this->dwCurrentPortInfoNo,3,"Open/Host not exitst");
	}
	else
	{
		this->cListPortInfo.SetItemText(this->dwCurrentPortInfoNo,3,"Closed");
	}
	this->dwCurrentPortInfoNo++;
	return 0;   
}
LRESULT CBeastScanerDlg::OnHandleSynScan(WPARAM StateAndPort,LPARAM dwHostIP)
{
	if(this->dwCurrentPortInfoNo==0xfffffffe)
	{
		this->MessageBox("Too many PortInfo!");
		return 0;
	}
	this->cListPortInfo.InsertItem(dwCurrentPortInfoNo,"Tcp Syn Scan");
	in_addr inHostIp;
	inHostIp.S_un.S_addr=htonl((DWORD)dwHostIP);
	CString Tmp;
	Tmp.Format("%u",0xffff&StateAndPort);
	this->cListPortInfo.SetItemText(this->dwCurrentPortInfoNo,1,inet_ntoa(inHostIp));
	this->cListPortInfo.SetItemText(this->dwCurrentPortInfoNo,2,Tmp.GetBuffer());
	if((0xffff0000&StateAndPort)!=0)
	{
		this->cListPortInfo.SetItemText(this->dwCurrentPortInfoNo,3,"Open");
	}
	else
	{
		this->cListPortInfo.SetItemText(this->dwCurrentPortInfoNo,3,"Closed");
	}
	this->dwCurrentPortInfoNo++;
	return 0;   
}
LRESULT CBeastScanerDlg::OnHandleNullScan(WPARAM StateAndPort,LPARAM dwHostIP)
{
	if(this->dwCurrentPortInfoNo==0xfffffffe)
	{
		this->MessageBox("Too many PortInfo!");
		return 0;
	}
	this->cListPortInfo.InsertItem(dwCurrentPortInfoNo,"Tcp Null Scan");
	in_addr inHostIp;
	inHostIp.S_un.S_addr=htonl((DWORD)dwHostIP);
	CString Tmp;
	Tmp.Format("%u",0xffff&StateAndPort);
	this->cListPortInfo.SetItemText(this->dwCurrentPortInfoNo,1,inet_ntoa(inHostIp));
	this->cListPortInfo.SetItemText(this->dwCurrentPortInfoNo,2,Tmp.GetBuffer());
	this->cListPortInfo.SetItemText(this->dwCurrentPortInfoNo,3,"Closed");
	this->dwCurrentPortInfoNo++;
	return 0;      
}
LRESULT CBeastScanerDlg::OnHandleAckScan(WPARAM StateAndPort,LPARAM dwHostIP)
{
	if(this->dwCurrentPortInfoNo==0xfffffffe)
	{
		this->MessageBox("Too many PortInfo!");
		return 0;
	}
	this->cListPortInfo.InsertItem(dwCurrentPortInfoNo,"Tcp Ack Scan");
	in_addr inHostIp;
	inHostIp.S_un.S_addr=htonl((DWORD)dwHostIP);
	CString Tmp;
	Tmp.Format("%u",0xffff&StateAndPort);
	this->cListPortInfo.SetItemText(this->dwCurrentPortInfoNo,1,inet_ntoa(inHostIp));
	this->cListPortInfo.SetItemText(this->dwCurrentPortInfoNo,2,Tmp.GetBuffer());
	this->cListPortInfo.SetItemText(this->dwCurrentPortInfoNo,3,"Open");
	this->dwCurrentPortInfoNo++;
	return 0;      
}
LRESULT CBeastScanerDlg::OnHandleFinScan(WPARAM StateAndPort,LPARAM dwHostIP)
{
	if(this->dwCurrentPortInfoNo==0xfffffffe)
	{
		this->MessageBox("Too many PortInfo!");
		return 0;
	}
	this->cListPortInfo.InsertItem(dwCurrentPortInfoNo,"Tcp Fin Scan");
	in_addr inHostIp;
	inHostIp.S_un.S_addr=htonl((DWORD)dwHostIP);
	CString Tmp;
	Tmp.Format("%u",0xffff&StateAndPort);
	this->cListPortInfo.SetItemText(this->dwCurrentPortInfoNo,1,inet_ntoa(inHostIp));
	this->cListPortInfo.SetItemText(this->dwCurrentPortInfoNo,2,Tmp.GetBuffer());
	this->cListPortInfo.SetItemText(this->dwCurrentPortInfoNo,3,"Closed");
	this->dwCurrentPortInfoNo++;
	return 0;      
}
LRESULT CBeastScanerDlg::OnFinishScan(WPARAM enCurrentScanType,LPARAM pThreadParament)
{
	if(this->enCurrentScanType!=NO_SCAN)	
	{	
		if(bStop==true)
		{
			bStop=false;
			this->stCurrentState="Scan Stoped.";
		}
		else
		{
			this->stCurrentState="Scan Finished.";
		}
		
		this->cTabCtrl.EnableWindow();
		this->enCurrentScanType=NO_SCAN;
		this->UpdateData(FALSE);
	}
	this->cButtonPause.SetWindowText("Pause");
	bPause=false;
	return 0;
}
LRESULT CBeastScanerDlg::OnAddHostInfo(WPARAM ScanType/*0: ARP ;1 Icmp*/, LPARAM dwHostIP)
{
	if(dwCurrentHostNo==0xFFFFFFFe)
	{
		this->MessageBox("Too many Host!");
		return 0;
	}
	if(ScanType==0)
	{
		this->cListHostInfo.InsertItem(dwCurrentHostNo,"ARP");
	}
	else
	{
		this->cListHostInfo.InsertItem(dwCurrentHostNo,"ICMP");
	}
	in_addr inHostIp;
	inHostIp.S_un.S_addr=htonl((DWORD)dwHostIP);
	this->cListHostInfo.SetItemText(dwCurrentHostNo,1,inet_ntoa(inHostIp));
	dwCurrentHostNo++;
	return 0;
}
LRESULT CBeastScanerDlg::OnUpdataLog(WPARAM wparam,LPARAM lparam)
{
	this->stCurrentState=(char *)wparam;
	this->UpdateData(FALSE);
	return 0;
}

LRESULT CBeastScanerDlg::OnHandleConnectScan(WPARAM wparam,LPARAM lparam)
{

	if(bPause||bStop)
	{
		return 0;
	}
	WORD wEvent = WSAGETSELECTEVENT (lparam) ;   // ie, LOWORD
    WORD wError = WSAGETSELECTERROR (lparam) ;   // ie, HIWORD
	SOCKET theSocket=(SOCKET)wparam;
	if(wEvent==FD_CONNECT)
	{
		if(!wError)
		{
			if(this->dwCurrentPortInfoNo==0xfffffffe)
			{
				this->MessageBox("Too many PortInfo!");
				return 0;
			}
			SOCKADDR thePeerName;
			int size=sizeof(sockaddr);
			getpeername(theSocket,&thePeerName,&size);
			if(closesocket(theSocket)==0)
			{
					
				this->cListPortInfo.InsertItem(dwCurrentPortInfoNo,"Tcp Connect Scan");
				CString Tmp;
				Tmp.Format("%u",htons(((sockaddr_in*)(&thePeerName))->sin_port));
				this->cListPortInfo.SetItemText(this->dwCurrentPortInfoNo,1,inet_ntoa(((sockaddr_in*)(&thePeerName))->sin_addr));
				this->cListPortInfo.SetItemText(this->dwCurrentPortInfoNo,2,Tmp.GetBuffer());
				this->cListPortInfo.SetItemText(this->dwCurrentPortInfoNo,3,"Open");
				this->dwCurrentPortInfoNo++;
				return 1;
			}
		}
		
	}
	closesocket(theSocket);
	return 0;
}

void CBeastScanerDlg::OnDestroy()
{
	CDialog::OnDestroy();

	// TODO: 在此处添加消息处理程序代码
	pcap_freealldevs(alldevs);
}

CBeastScanerDlg::~CBeastScanerDlg(void)
{
	bStop=true;
	while(dwCurrentThreadNo!=0)
	{
		Sleep(100);
	}
}