www.pudn.com > PortScan_sjtu.rar > PortScanDlg.cpp

// PortScanDlg.cpp : 实现文件 
// 
 
#include "stdafx.h" 
#include "PortScan.h" 
#include "PortScanDlg.h" 
#include ".\portscandlg.h" 
#include "TCPConnect.h" 
#include "SYNScan.h" 
#include "FinScan.h" 
#include "ICMP_ECHO.h" 
 
#ifdef _DEBUG 
#define new DEBUG_NEW 
#endif 
 
 
// CPortScanDlg 对话框 
 
 
 
CPortScanDlg::CPortScanDlg(CWnd* pParent /*=NULL*/) 
	: CDialog(CPortScanDlg::IDD, pParent) 
{ 
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME); 
} 
 
void CPortScanDlg::DoDataExchange(CDataExchange* pDX) 
{ 
	CDialog::DoDataExchange(pDX); 
	DDX_Control(pDX, IDC_IPADDRESS1, IPADDRESS1); 
	DDX_Control(pDX, CEDIT_StartPort, IDC_StartPort); 
	DDX_Control(pDX, CEDIT_EndPort, IDC_EndPort); 
	DDX_Control(pDX, CEDIT_Result, IDC_Result); 
	DDX_Control(pDX, IDOK, ButtonOK); 
	DDX_Control(pDX, IDC_IPADDRESS2, IPADDRESS2); 
} 
 
BEGIN_MESSAGE_MAP(CPortScanDlg, CDialog) 
	ON_WM_PAINT() 
	ON_WM_QUERYDRAGICON() 
	//}}AFX_MSG_MAP 
	ON_BN_CLICKED(IDOK, OnBnClickedOk) 
	ON_BN_CLICKED(IDC_ScanByConnect, OnBnClickedScanbyconnect) 
	ON_BN_CLICKED(IDC_ScanBySyn, OnBnClickedScanbysyn) 
	ON_BN_CLICKED(IDC_ScanByFin, OnBnClickedScanbyfin) 
	ON_BN_CLICKED(IDC_ScanByICMP, OnBnClickedScanbyicmp) 
END_MESSAGE_MAP() 
 
 
// CPortScanDlg 消息处理程序 
 
BOOL CPortScanDlg::OnInitDialog() 
{ 
	CDialog::OnInitDialog(); 
 
	// 设置此对话框的图标。当应用程序主窗口不是对话框时，框架将自动 
	//  执行此操作 
	SetIcon(m_hIcon, TRUE);			// 设置大图标 
	SetIcon(m_hIcon, FALSE);		// 设置小图标 
 
	// TODO: 在此添加额外的初始化代码 
 
	IPADDRESS1.SetWindowText("192.168.1.5"); 
	IPADDRESS2.SetWindowText("192.168.1.10"); 
 
	IDC_StartPort.SetWindowText("20"); 
	IDC_EndPort.SetWindowText("100"); 
 
	ScanByConnect = false; 
	ScanBySyn = false; 
	ScanByFin = false; 
	ScanByICMP = false; 
	 
	return TRUE;  // 除非设置了控件的焦点，否则返回 TRUE 
} 
 
// 如果向对话框添加最小化按钮，则需要下面的代码 
//  来绘制该图标。对于使用文档/视图模型的 MFC 应用程序， 
//  这将由框架自动完成。 
 
void CPortScanDlg::OnPaint()  
{ 
	if (IsIconic()) 
	{ 
		CPaintDC dc(this); // 用于绘制的设备上下文 
 
		SendMessage(WM_ICONERASEBKGND, reinterpret_cast(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 CPortScanDlg::OnQueryDragIcon() 
{ 
	return static_cast(m_hIcon); 
} 
 
void CPortScanDlg::OnBnClickedOk() 
{ 
	// TODO: 在此添加控件通知处理程序代码 
	//OnOK(); 
 
	CString IP_Dest_buf,StartPort_buf,EndPort_buf; 
	char ip_dest[20]; 
	int Port_Start, Port_End; 
	 
	IPADDRESS1.GetWindowText(IP_Dest_buf); 
	sprintf(ip_dest,"%s",IP_Dest_buf); 
 
	IDC_StartPort.GetWindowText(StartPort_buf); 
	Port_Start = atoi(StartPort_buf); 
 
	IDC_EndPort.GetWindowText(EndPort_buf); 
	Port_End = atoi(EndPort_buf); 
 
	char scan_result[4096],result_buf[4096]; 
	scan_result[0] = '\0'; 
	result_buf[0] = '\0'; 
 
	if (ScanByConnect)       //connect扫描 
	{ 
		if (Port_Start>Port_End) 
			sprintf(scan_result,"扫描起始端口不应该大于结束端口"); 
		else 
		{ 
			TCPConnect* ConnectScan = new TCPConnect(); 
 
			ConnectScan->Scan(ip_dest,Port_Start,Port_End,500,result_buf); 
 
			strcat(scan_result,result_buf); 
 
			delete ConnectScan; 
		} 
	} 
 
	if (ScanBySyn)           //syn扫描 
	{	 
		result_buf[0] = '\0'; 
 
		if (Port_Start>Port_End) 
			sprintf(scan_result,"扫描起始端口不应该大于结束端口"); 
		else 
		{			 
			SYNScan* Synscan = new SYNScan(); 
			for (int i=Port_Start;i < (Port_End+1);i++) 
			{ 
				Synscan->Scan(ip_dest,i,1000,result_buf); 
				strcat(scan_result,result_buf); 
				result_buf[0] = '\0'; 
			} 
			 
			delete Synscan; 
		} 
	} 
 
	if (ScanByFin)           //Fin扫描 
	{ 
		if (Port_Start>Port_End) 
			sprintf(scan_result,"扫描起始端口不应该大于结束端口"); 
		else 
		{		 
			result_buf[0] = '\0'; 
 
			FinScan* FINscan = new FinScan(); 
 
			for (int i=Port_Start;i < (Port_End+1);i++) 
			{ 
				FINscan->Scan(ip_dest,i,1000,result_buf); 
 
				strcat(scan_result,result_buf); 
 
				result_buf[0] = '\0'; 
 
			} 
 
			delete FINscan; 
		} 
 
	} 
     
	if (ScanByICMP)          //主机探测 
	{ 
		char result[100]; 
		result[0] = '\0'; 
		char ping_ip[20]; 
 
		byte startIP_1,startIP_2,startIP_3,startIP_4;           //取得主机地址范围 
		byte EndIP_1,EndIP_2,EndIP_3,EndIP_4; 
		byte currentIP_1,currentIP_2,currentIP_3,currentIP_4; 
 
 
		IPADDRESS1.GetAddress(startIP_1,startIP_2,startIP_3,startIP_4); 
		IPADDRESS2.GetAddress(EndIP_1,EndIP_2,EndIP_3,EndIP_4); 
 
		currentIP_1 = startIP_1; 
		currentIP_2 = startIP_2; 
		currentIP_3 = startIP_3; 
		currentIP_4 = startIP_4; 
 
		ULONG IP_End = ((EndIP_1*256+EndIP_2)*256+EndIP_3)*256+EndIP_4; 
		ULONG IP_Current = ((currentIP_1*256+currentIP_2)*256+currentIP_3)*256+currentIP_4; 
 
		if (IP_Current>IP_End) 
			sprintf(scan_result,"网段起始地址不应该大于结束地址"); 
		else 
		{						 
			ICMP_ECHO* PingScan = new ICMP_ECHO(); 
 
			while (IP_Current <= IP_End)                        //循环，探测网段内主机 
			{ 
				sprintf(ping_ip,"%d.%d.%d.%d",currentIP_1,currentIP_2,currentIP_3,currentIP_4); 
				PingScan->Ping(ping_ip,1000,result_buf); 
				strcat(scan_result,result_buf); 
				result_buf[0] = '\0'; 
 
				currentIP_4++; 
 
				if (currentIP_4 == 0) 
				{ 
					currentIP_3++; 
					currentIP_4 = 0; 
					if (currentIP_3 == 0) 
					{ 
						currentIP_2++; 
						currentIP_3 = 0; 
						if (currentIP_2 == 0) 
						{ 
							currentIP_1++; 
							currentIP_2 = 0; 
						} 
					} 
				} 
				 
				IP_Current = ((currentIP_1*256+currentIP_2)*256+currentIP_3)*256+currentIP_4; 
			} 
 
			delete PingScan; 
		} 
	} 
	IDC_Result.SetWindowText(scan_result); 
} 
 
void CPortScanDlg::OnBnClickedScanbyconnect() 
{ 
	ScanByConnect = true; 
	ScanBySyn = false; 
	ScanByFin = false; 
	ScanByICMP = false; 
} 
 
void CPortScanDlg::OnBnClickedScanbysyn() 
{ 
	ScanByConnect = false; 
	ScanBySyn = true; 
	ScanByFin = false; 
	ScanByICMP = false; 
} 
 
void CPortScanDlg::OnBnClickedScanbyfin() 
{ 
	ScanByConnect = false; 
	ScanBySyn = false; 
	ScanByFin = true; 
	ScanByICMP = false; 
} 
 
void CPortScanDlg::OnBnClickedScanbyicmp() 
{ 
	ScanByConnect = false; 
	ScanBySyn = false; 
	ScanByFin = false; 
	ScanByICMP = true; 
}