netinfodlg.cpp

查询主机的相关信息。。如主机名。。主机IP。。网卡信息。。DNS信息等

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

#include "stdafx.h"
#include "NetInfo.h"
#include "NetInfoDlg.h"
#include "ScanDlg.h"
#include <Iphlpapi.h>				//用于支持获取网卡信息的函数
#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// CNetInfoDlg 对话框




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

void CNetInfoDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
DDX_Control(pDX, IDC_TREE, m_treeCtrl);
DDX_Control(pDX, IDC_LIST, m_listCtrl);
	}

BEGIN_MESSAGE_MAP(CNetInfoDlg, CDialog)
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	//}}AFX_MSG_MAP
//	ON_NOTIFY(NM_CLICK, IDC_TREE, &CNetInfoDlg::OnNMClickTree)
	ON_NOTIFY(TVN_SELCHANGED, IDC_TREE, &CNetInfoDlg::OnTvnSelchangedTree)
END_MESSAGE_MAP()


// CNetInfoDlg 消息处理程序

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

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

	// TODO: 在此添加额外的初始化代码
//-------------------------------启动Winsock----------------------------------------
	WORD wVersion=MAKEWORD(2,0);			//winsock版本号
	WSADATA wsData;							//版本信息
	
	//对Winsock DLL进行初始化，协商Winsock的版本支持并分配必要的资源
	int StartResult = WSAStartup(wVersion,&wsData);
	
	if(StartResult != 0)
	{
		//错误处理
		MessageBox("WSAStartup error");
	  
	}
	
	InitTree();				//初始化树
	GetHostInfo();			//获取主机信息
	GetAdapterInform();		//获取适配器信息
	GetDnsInfo();			//获取域名服务器相关信息
	return TRUE;  // 除非将焦点设置到控件，否则返回 TRUE
}

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

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


void CNetInfoDlg::InitTree(void)
{ 
      tRoot.hParent = TVI_ROOT;		//增加根项
      tRoot.hInsertAfter = TVI_LAST;	//在最后项之后
      tRoot.item.mask = TVIF_TEXT|TVIF_PARAM|TVIF_IMAGE|TVIF_SELECTEDIMAGE;//设屏蔽
      tRoot.item.pszText = "基本网络应用汇总";
      tRoot.item.lParam = 0;			//序号
      tRoot.item.iImage = 0;			//正常图标
      tRoot.item.iSelectedImage = 1;	//选中时图标
      hRoot = m_treeCtrl.InsertItem(&tRoot);//返回根项句柄
      
      tOne.hParent = hRoot;			//增加子项1
      tOne.hInsertAfter = TVI_LAST;	//在最后项之后
      tOne.item.lParam = 1;			//序号
      tOne.item.mask = TVIF_TEXT|TVIF_PARAM|TVIF_IMAGE|TVIF_SELECTEDIMAGE;//设屏蔽
      tOne.item.pszText = "获取主机IP地址、子网掩码和计算机名";			//选项值
      m_treeCtrl.InsertItem(&tOne);//插入项
      
      tTwo.hParent = hRoot;			//增加子项2
      tTwo.hInsertAfter = TVI_LAST;	//在最后项之后
      tTwo.item.lParam = 2;			//序号
      tTwo.item.mask = TVIF_TEXT|TVIF_PARAM|TVIF_IMAGE|TVIF_SELECTEDIMAGE;//设屏蔽
      tTwo.item.pszText = "获取域名服务器、子网掩码";
      m_treeCtrl.InsertItem(&tTwo);
      
      tThree.hParent = hRoot;			//增加子项3
      tThree.hInsertAfter = TVI_LAST;	//在最后项之后
      tThree.item.lParam = 3;			//序号
      tThree.item.mask = TVIF_TEXT|TVIF_PARAM|TVIF_IMAGE|TVIF_SELECTEDIMAGE;//设屏蔽
      tThree.item.pszText = "获取网卡类型、网卡mac地址";
      m_treeCtrl.InsertItem(&tThree);
      
      tFour.hParent = hRoot;			//增加子项4
      tFour.hInsertAfter = TVI_LAST;	//在最后项之后
      tFour.item.lParam = 4;			//序号
      tFour.item.mask = TVIF_TEXT|TVIF_PARAM|TVIF_IMAGE|TVIF_SELECTEDIMAGE;//设屏蔽
      tFour.item.pszText = "端口扫描";
      m_treeCtrl.InsertItem(&tFour);

}

void CNetInfoDlg::GetHostInfo(void)
{
	char * strHost = NULL;
	::gethostname(strHost,256);						//获取主机名
	hostent *pHost = ::gethostbyname(strHost);		//通过主机名字取得主机信息
	m_hostname = pHost->h_name;				//取得主机名
	
	in_addr addr;							//定义一个IP地址
	for(int i = 0; ;i++) 
	{
		char *p = pHost->h_addr_list[i];	//p指向一个32位IP地址
		if(p == NULL) 
			{
				break;
			}
		memcpy(&addr.S_un.S_addr, p, pHost->h_length);
		m_hostip = ::inet_ntoa(addr);		//取得主机IP
	}	
}

void CNetInfoDlg::ShowHostInfo(void)
{
	//清空原有的list中的信息，因为有3列所有删除3次
	m_listCtrl.DeleteAllItems();
	m_listCtrl.DeleteColumn(0);
	m_listCtrl.DeleteColumn(0);
	m_listCtrl.DeleteColumn(0);		
	
	//插入信息
	m_listCtrl.InsertColumn(0, "主机IP地址", 0, 110, 0);
	m_listCtrl.InsertColumn(1, "主机子网掩码", 0, 100, 0);
	m_listCtrl.InsertColumn(2, "计算机名", 0, 100 ,0);
	m_listCtrl.InsertItem(0, m_hostip);
	m_listCtrl.SetItemText(0, 1, m_hostsub);
	m_listCtrl.SetItemText(0, 2, m_hostname);
	
}

void CNetInfoDlg::GetAdapterInform(void)
{
	//定义适配器
	PIP_ADAPTER_INFO pAdapterInfo;
	PIP_ADAPTER_INFO pAdapter = NULL;
	DWORD dwRetVal;							//用于接收返回值
	u_char mac[6];								//存储mac地址	
	pAdapterInfo = (IP_ADAPTER_INFO*) malloc ( sizeof(IP_ADAPTER_INFO) );
	ULONG ulOutBufLen = sizeof(IP_ADAPTER_INFO);
	
	//分配存取适配器信息空间
	if (GetAdaptersInfo( pAdapterInfo, &ulOutBufLen) == ERROR_BUFFER_OVERFLOW)
		{
			free(pAdapterInfo);
			pAdapterInfo = (IP_ADAPTER_INFO*) malloc (ulOutBufLen);
		}
		
	//获取适配器信息	
	if ((dwRetVal = GetAdaptersInfo( pAdapterInfo, &ulOutBufLen)) == NO_ERROR)
		{
			pAdapter = pAdapterInfo;			//赋值给临时变量
			int i = 0;
			while(pAdapter && i < 5)						//如果还有其他适配器的话循环
				{
					m_hostsub = pAdapter->IpAddressList.IpMask.String;		//取得主机子网掩码
		
					//mac地址相关
					memcpy(mac, pAdapter->Address, 6);
					u_char *a = mac;
					CString b,c,d,e,f,g;
					b.Format("%02X-",a[0]);
					c.Format("%02X-",a[1]);
					d.Format("%02X-",a[2]);
					e.Format("%02X-",a[3]);
					f.Format("%02X-",a[4]);
					g.Format("%02X", a[5]);	
					CString tmp = b + c + d + e;
					m_LocalMac[i] = b + c + d + e + f + g;		//获取mac地址
					m_MacType[i] = pAdapter->Description;		//网卡类型
					i++;
					
					pAdapter = pAdapter->Next;
				}
		}
}

void CNetInfoDlg::GetDnsInfo()
{
   FIXED_INFO * FixedInfo;
   ULONG    ulOutBufLen;
   DWORD    dwRetVal;
   IP_ADDR_STRING * pIPAddr;				//存储域名服务器信息结构

   FixedInfo = (FIXED_INFO *) GlobalAlloc( GPTR, sizeof( FIXED_INFO ) );
   ulOutBufLen = sizeof( FIXED_INFO );
   
   if( ERROR_BUFFER_OVERFLOW == GetNetworkParams( FixedInfo, &ulOutBufLen ) ) {
      GlobalFree( FixedInfo );
      FixedInfo = (FIXED_INFO *) GlobalAlloc( GPTR, ulOutBufLen );
   }
	//判断是否获取成功
   if ( dwRetVal = GetNetworkParams( FixedInfo, &ulOutBufLen ) ) {
        MessageBox( "Call to GetNetworkParams failed.");
   }
   else {
	  int i = 0;
	  m_DnsIP[i] = FixedInfo ->DnsServerList.IpAddress.String;	//取得域名服务器IP
	  m_DnsSub[i] = m_hostsub;	//取得子网掩码
	  i++;
      pIPAddr = FixedInfo -> DnsServerList.Next;
      while ( pIPAddr && i < 10 )						//读取所有域名服务器
      {
         m_DnsIP[i] = pIPAddr->IpAddress.String;//IP
         m_DnsSub[i] = m_hostsub;				//子网掩码
         i++;
         pIPAddr = pIPAddr ->Next;
      }
   }
}

void CNetInfoDlg::ShowMacInfo()
{
    //清空原有的list中的信息，因为有3列所有删除3次
	m_listCtrl.DeleteAllItems();
	m_listCtrl.DeleteColumn(0);
	m_listCtrl.DeleteColumn(0);
	m_listCtrl.DeleteColumn(0);		
	int i = 0;
	
	//插入信息
	m_listCtrl.InsertColumn(0, "网卡类型", 0, 200, 0);
	m_listCtrl.InsertColumn(1, "网卡mac地址", 0, 120, 0);
	while(i < 5 && !m_LocalMac[i].IsEmpty())
	{
		m_listCtrl.InsertItem(0, m_MacType[i]);
		m_listCtrl.SetItemText(0, 1, m_LocalMac[i]);
		i++;
	}	
}

void CNetInfoDlg::ShowDnsInfo()
{
	//清空原有的list中的信息，因为有3列所有删除3次
	m_listCtrl.DeleteAllItems();
	m_listCtrl.DeleteColumn(0);
	m_listCtrl.DeleteColumn(0);
	m_listCtrl.DeleteColumn(0);		
	
	int i = 0;
	
	//插入信息
	m_listCtrl.InsertColumn(0, "域名服务器IP", 0, 200, 0);
	m_listCtrl.InsertColumn(1, "子网掩码", 0, 120, 0);
	while(i < 10 && !m_DnsIP[i].IsEmpty())
	{
		m_listCtrl.InsertItem(i, m_DnsIP[i]);
		m_listCtrl.SetItemText(i, 1, m_DnsSub[i]);
		i++;
	}
	}
void CNetInfoDlg::OnTvnSelchangedTree(NMHDR *pNMHDR, LRESULT *pResult)
{
	LPNMTREEVIEW pNMTreeView = reinterpret_cast<LPNMTREEVIEW>(pNMHDR);
	// TODO: 在此添加控件通知处理程序代码
	*pResult = 0;
	HTREEITEM hSel=m_treeCtrl.GetSelectedItem();//取得选项句柄;
	long id=m_treeCtrl.GetItemData(hSel);      //取得数据项序号
	
	switch(id)	
	{	
	default:
			m_listCtrl.DeleteAllItems();
			m_listCtrl.DeleteColumn(0);
			m_listCtrl.DeleteColumn(0);
			m_listCtrl.DeleteColumn(0);
			break;
	
	case 1:	
			ShowHostInfo();					//显示主机信息
			break;
			
	case 2:
			ShowDnsInfo();					//显示域名服务器信息
			break;
			
	case 3:
			ShowMacInfo();					//显示网卡信息	
			break;
			
	case 4:
			if(scanDlg.DoModal() == IDOK)	//弹出扫描对话框
			{
				;
			}
			break;	
	}
	
}