SingularityViewer/indra/llmessage/llhost.cpp

/** 
 * @file llhost.cpp
 * @brief Encapsulates an IP address and a port.
 *
 * $LicenseInfo:firstyear=2000&license=viewerlgpl$
 * Second Life Viewer Source Code
 * Copyright (C) 2010, Linden Research, Inc.
 * 
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation;
 * version 2.1 of the License only.
 * 
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 * 
 * Linden Research, Inc., 945 Battery Street, San Francisco, CA  94111  USA
 * $/LicenseInfo$
 */

#include "linden_common.h"

#include "llhost.h"

#include "llerror.h"

#if LL_WINDOWS
	#define WIN32_LEAN_AND_MEAN
	#include <winsock2.h>
#else
	#include <netdb.h>
	#include <netinet/in.h>	// ntonl()
	#include <sys/types.h>
	#include <sys/socket.h>
	#include <arpa/inet.h>
#endif

LLHost LLHost::invalid(INVALID_PORT,INVALID_HOST_IP_ADDRESS);

LLHost::LLHost(const std::string& ip_and_port)
{
	std::string::size_type colon_index = ip_and_port.find(":");
	if (colon_index == std::string::npos)
	{
		mIP = ip_string_to_u32(ip_and_port.c_str());
		mPort = 0;
	}
	else
	{
		std::string ip_str(ip_and_port, 0, colon_index);
		std::string port_str(ip_and_port, colon_index+1);

		mIP = ip_string_to_u32(ip_str.c_str());
		mPort = atol(port_str.c_str());
	}
	mHostNotFound = 0;
}

std::string LLHost::getString() const
{
	return llformat("%s:%hu", u32_to_ip_string(mIP), mPort);
}


std::string LLHost::getIPandPort() const
{
	return getString();
}


std::string LLHost::getIPString() const
{
	return std::string( u32_to_ip_string( mIP ) );
}


std::string LLHost::getHostName() const
{
	hostent* he;
	if (INVALID_HOST_IP_ADDRESS == mIP)
	{
		LL_WARNS() << "LLHost::getHostName() : Invalid IP address" << LL_ENDL;
		return std::string();
	}
	if (mHostNotFound)
	{
		// We already checked this... avoid freezing the viewer 5 seconds again and again.
		LL_WARNS() << "LLHost::getHostName() : Returning cached HOST_NOT_FOUND." << LL_ENDL;
		return std::string();
	}
	he = gethostbyaddr((char *)&mIP, sizeof(mIP), AF_INET);
	if (!he)
	{
#if LL_WINDOWS
		int err = WSAGetLastError();
		int err_host_not_found = WSAHOST_NOT_FOUND;
#else
		int err = h_errno;
		int err_host_not_found = HOST_NOT_FOUND;
#endif
		LL_WARNS() << "LLHost::getHostName() : Couldn't find host name for address " << mIP << ", Error: " << err << LL_ENDL;
		if (err == err_host_not_found)
		{
			mHostNotFound = 1;
		}
		return std::string();
	}
	else
	{
		return ll_safe_string(he->h_name);
	}
}

BOOL LLHost::setHostByName(const std::string& hostname)
{
	hostent *he;
	std::string local_name(hostname);

#if LL_WINDOWS
	// We may need an equivalent for Linux, but not sure - djs
	LLStringUtil::toUpper(local_name);
#endif

	he = gethostbyname(local_name.c_str());	
	if(!he) 
	{
		U32 ip_address = ip_string_to_u32(hostname.c_str());
		he = gethostbyaddr((char *)&ip_address, sizeof(ip_address), AF_INET);
	}

	if (he)
	{
		mIP = *(U32 *)he->h_addr_list[0];
		mHostNotFound = 0;
		return TRUE;
	}
	else 
	{
		setAddress(local_name);

		// In windows, h_errno is a macro for WSAGetLastError(), so store value here
		S32 error_number = h_errno;
		switch(error_number) 
		{
			case TRY_AGAIN:	// XXX how to handle this case? 
				LL_WARNS() << "LLHost::setAddress(): try again" << LL_ENDL;
				break;
			case HOST_NOT_FOUND:
			case NO_ADDRESS:	// NO_DATA
				LL_WARNS() << "LLHost::setAddress(): host not found" << LL_ENDL;
				break;
			case NO_RECOVERY:
				LL_WARNS() << "LLHost::setAddress(): unrecoverable error" << LL_ENDL;
				break;
			default:
				LL_WARNS() << "LLHost::setAddress(): unknown error - " << error_number << LL_ENDL;
				break;
		}
		return FALSE;
	}
}

LLHost&	LLHost::operator=(const LLHost &rhs)
{ 
	if (this != &rhs)
	{
		set(rhs.getAddress(), rhs.getPort());
	}
	return *this;		
}


std::ostream& operator<< (std::ostream& os, const LLHost &hh)
{
	os << u32_to_ip_string(hh.mIP) << ":" << hh.mPort ;
	return os;
}


//std::istream& operator>> (std::istream& is, LLHost &rh)
//{
//	is >> rh.mIP;
//    is >> rh.mPort;
//    return is;
//}