hostinfo

/****************************************************************************@File Name: hostinfo.h@Author: wangzhicheng@mail: 2363702560@QQ.com@Created Time: Sun 05 Mar 2017 08:20:39 AM CST****************************************************************************/#ifndef HOST_INFO_H#define HOST_INFO_H#include <stdio.h>#include <unistd.h>#include <stdlib.h>#include <string.h>#include <netdb.h>#include <sys/socket.h>#include <arpa/inet.h>#include <sys/ioctl.h>#include <net/if.h>#include <netinet/in.h>#include <string>#include <vector>namespace hostinfo{static const int MAX_INTERFACE = 16;using namespace std;class HostInfo{public:	HostInfo()	{		m_hptr = 0;	}	/*	 * @brief init by host domain	 *        first find in DNS then find in /etc/hosts	 * */	inline bool InitByDomain(const string &domain)	{		if(!(m_hptr = gethostbyname(domain.c_str()))) return false;		return true;	}	/*	 * @brief init by host ip addr	 * */	inline bool InitByIpAddr(const string &ipaddr)	{		bool success = true;		struct in_addr *hipaddr = (struct in_addr *)malloc(sizeof(struct in_addr));		if(!hipaddr) return false;		if(!inet_aton(ipaddr.c_str(), hipaddr)) 		{			success = false;		}		else		if(!(m_hptr = gethostbyaddr(hipaddr, 4, AF_INET))) 		{			success = false;		}		free(hipaddr);		return success;	}	/*	 * @brief get the official host name 	 * */	inline void GetHostName_official(string &hostname)	{		if(m_hptr) hostname = m_hptr->h_name;	}	/*	 * @brief get the alias host names 	 * */	void GetHostName_alias(vector<string>&hostnames);	/*	 * @brief get the ip addr 	 * */	bool GetIpaddrs(vector<string>&ipaddrs);public:	/*	 * @brief the ip and mac address from NIC	 * */	static bool GetNICIp(vector<string>&ipaddrs, vector<string>&macaddrs);PRivate:	struct hostent *m_hptr;};}#endif/****************************************************************************@File Name: hostinfo.cpp@Author: wangzhicheng@mail: 2363702560@qq.com@Created Time: Sun 05 Mar 2017 08:20:39 AM CST****************************************************************************/#include "hostinfo.h"namespace hostinfo{/* * @brief get the alias host names  * */void HostInfo::GetHostName_alias(vector<string>&hostnames){	char **PPTr;	hostnames.clear();	if(!m_hptr) return;	for(pptr = m_hptr->h_aliases;*pptr;pptr++) 	{		hostnames.push_back(*pptr);	}}/* * @brief get the ip addr  * */bool HostInfo::GetIpaddrs(vector<string>&ipaddrs){	char **pptr;	char buf[64] = {0};	ipaddrs.clear();	if(!m_hptr) return false;	switch(m_hptr->h_addrtype)	{	case AF_INET:	case AF_INET6:		for(pptr = m_hptr->h_addr_list;*pptr;pptr++)		{			inet_ntop(m_hptr->h_addrtype, *pptr, buf, sizeof(buf));			ipaddrs.push_back(buf);		}		return true;	default:		return false;	}}/* * @brief the ip and mac address from NIC * */bool HostInfo::GetNICIp(vector<string>&ipaddrs, vector<string>&macaddrs){	int fd;	int if_num;			// NIC num	char tmp[64];	struct ifreq buf[MAX_INTERFACE];	struct ifconf ifc;	ipaddrs.clear();	macaddrs.clear();	if((fd = socket(AF_INET, SOCK_DGRAM, 0)) < 0) return false;	ifc.ifc_len = sizeof(buf);	ifc.ifc_buf = (caddr_t)buf;	if(ioctl(fd, SIOCGIFCONF, (char *)&ifc)) return false;	if_num = ifc.ifc_len / sizeof(struct ifreq);	while(if_num-- > 0)	{		if(ioctl(fd, SIOCGIFADDR, (char *)&buf[if_num])) continue;		const char *ptr = inet_ntoa(((struct sockaddr_in *)(&buf[if_num].ifr_addr))->sin_addr);		ipaddrs.push_back(ptr);		if(ioctl(fd, SIOCGIFHWADDR, (char*)&buf[if_num])) continue;		snprintf(tmp, sizeof tmp, "%02x:%02x:%02x:%02x:%02x:%02x", 		        (unsigned char)buf[if_num].ifr_hwaddr.sa_data[0],  				(unsigned char)buf[if_num].ifr_hwaddr.sa_data[1],  				(unsigned char)buf[if_num].ifr_hwaddr.sa_data[2],  				(unsigned char)buf[if_num].ifr_hwaddr.sa_data[3],  				(unsigned char)buf[if_num].ifr_hwaddr.sa_data[4],  				(unsigned char)buf[if_num].ifr_hwaddr.sa_data[5]);		macaddrs.push_back(tmp);	}	close(fd);	return true;}}/****************************************************************************@File Name: test.cpp@Author: wangzhicheng@mail: 2363702560@qq.com@Created Time: Sun 05 Mar 2017 08:46:02 AM CST****************************************************************************/#include "hostinfo.h"#include <iostream>using namespace hostinfo;int main(){	string domain;	string ipaddr;	string str;	vector<string>strs;	vector<string>str1s;	vector<string>::iterator it;	cout << "domain = ";	cin >> domain;	cout << "ip addr = ";	cin >> ipaddr;	HostInfo hostinfo;	cout << "init by domain===============================================" << endl;	// init by domain	if(hostinfo.InitByDomain(domain))	{		hostinfo.GetHostName_official(str);		cout << "offical host name = " << str << endl;		hostinfo.GetHostName_alias(strs);		for(it = strs.begin();it != strs.end();++it)		{			cout << "alias host name = " << *it << endl;		}		hostinfo.GetIpaddrs(strs);		for(it = strs.begin();it != strs.end();++it)		{			cout << "ip addr = " << *it << endl;		}			}	cout << "init by ip addr===============================================" << endl;	// init by ip addr	if(hostinfo.InitByDomain(ipaddr))	{		hostinfo.GetHostName_official(str);		cout << "offical host name = " << str << endl;		hostinfo.GetHostName_alias(strs);		for(it = strs.begin();it != strs.end();++it)		{			cout << "alias host name = " << *it << endl;		}		hostinfo.GetIpaddrs(strs);		for(it = strs.begin();it != strs.end();++it)		{			cout << "ip addr = " << *it << endl;		}	}	cout << "get NIC ip and mac===============================================" << endl;	// get NIC IP	if(HostInfo::GetNICIp(strs, str1s))	{		for(it = strs.begin();it != strs.end();++it)		{			cout << "NIC ip addr = " << *it << endl;		}		for(it = str1s.begin();it != str1s.end();++it)		{			cout << "NIC mac addr = " << *it << endl;		}	}   return 0;}CC=g++all:	$(CC) -g -o hostinfoTest test.cpp hostinfo.cpp hostinfo.h