CWE-ou-minkata/MOULOpenSourceClientPlugin/Plasma20/Sources/Plasma/PubUtilLib/plSockets/plSocket.cpp

/*==LICENSE==*

CyanWorlds.com Engine - MMOG client, server and tools
Copyright (C) 2011  Cyan Worlds, Inc.

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program 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 General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.

You can contact Cyan Worlds, Inc. by email legal@cyan.com
 or by snail mail at:
      Cyan Worlds, Inc.
      14617 N Newport Hwy
      Mead, WA   99021

*==LICENSE==*/
#include "plSocket.h"

#if HS_BUILD_FOR_UNIX
#include <sys/ioctl.h>
#endif

/////////////////////////////////////////////////

plSocket::plSocket()
{
	fCloseOnDestroy	= false;
	fCloseBeforeSet	= false;
	fTimeoutsSet = 0;
	fSocket = kBadSocket;
}


plSocket::plSocket(SOCKET sck)
{
	fCloseOnDestroy	= false;
	fCloseBeforeSet	= false;
	fTimeoutsSet = 0;
	SetSocket(sck);
}

plSocket::~plSocket()
{
	if (fCloseOnDestroy)
		Close();
}

bool plSocket::operator==(const plSocket & rhs)
{
	return fSocket == rhs.fSocket;
}

bool plSocket::ErrorClose()
{
	if(Active())
		plNet::Close(fSocket);
	fSocket = kBadSocket;
	return false;
}


bool plSocket::Active() 
{ 
	return (fSocket != kBadSocket); 
}


void plSocket::Close()
{
	if(Active())
		plNet::Close(fSocket);
	fSocket = kBadSocket;
}


int plSocket::GetLastError()
{
	return plNet::GetError();
}


void plSocket::SetSocket(SOCKET sck)
{
	if (fSocket!=sck)
	{
		if (fCloseBeforeSet)
			Close();
		fSocket = sck;
		if (fTimeoutsSet & kRecvTimeoutSet)
			setsockopt(fSocket, SOL_SOCKET, (int)SO_RCVTIMEO,(const char*)&fRecvTimeOut,sizeof(fRecvTimeOut));

		if (fTimeoutsSet & kSendTimeoutSet)
			setsockopt(fSocket, SOL_SOCKET, (int)SO_SNDTIMEO,(const char*)&fSendTimeOut,sizeof(fSendTimeOut));

		IGetTimeOutsFromSocket();
	}
}


int plSocket::SetRecvBufferSize(int insize)
{
	if (setsockopt(fSocket, (int)SOL_SOCKET, (int)SO_RCVBUF, (char *)&insize, sizeof(int)))
		return 0;
	
	return -1;
}


void plSocket::SetBlocking(bool value)
{
	unsigned long val = value?0:1;
#ifdef BSDBLOCK
// NOTE: Bsd does this a little differently, using fcntl()
#error "BSDBLOCK: This code needs to be verified right now. Don't assume it will work at all."
	int flags = fcntl(fSocket,F_GETFL,val);
	flags = flags |O_NONBLOCK;
	fcntl(fSocket,F_SETFL,flags);        
#else
	plNet::Ioctl(fSocket,FIONBIO,&val);
#endif	
}


bool plSocket::IsBlocking()
{
#ifdef BSDBLOCK
#error "BSDBLOCK: TODO: IsBlocking() for bsd."
#else
	int ans = plNet::Ioctl(fSocket,FIONBIO,NULL);
	return (ans!=0);
#endif
}


void plSocket::SetReuseAddress()
{
	int opt = 1; 
	setsockopt(fSocket, SOL_SOCKET, SO_REUSEADDR,  (const char *)&opt, sizeof(opt));
}

int plSocket::SetSendTimeOut(unsigned int milliSecs/* =0 */)
{
	fTimeoutsSet |= kSendTimeoutSet;
	fSendTimeOut = milliSecs;

	if (fSocket != kBadSocket)
	{
		if (setsockopt(fSocket, SOL_SOCKET, (int)SO_SNDTIMEO,(const char*)&fSendTimeOut,sizeof(fSendTimeOut)) != 0)
			return -1;
	}

	return 0;
}

int plSocket::SetRecvTimeOut(unsigned int milliSecs/* =0 */)
{
	fTimeoutsSet |= kRecvTimeoutSet;
	fRecvTimeOut = milliSecs;

	if (fSocket != kBadSocket)
	{
		if (setsockopt(fSocket, SOL_SOCKET, (int)SO_RCVTIMEO,(const char*)&fRecvTimeOut,sizeof(fRecvTimeOut)) != 0)
			return -1;
	}

	return 0;
}

int plSocket::IGetTimeOutsFromSocket()
{
	unsigned int rtimeoutval, stimeoutval;

	if (fSocket != kBadSocket)
	{
		socklen_t sizeval;
		sizeval = sizeof(rtimeoutval);
		if (getsockopt(fSocket, SOL_SOCKET, (int)SO_RCVTIMEO,(char*)&rtimeoutval,&sizeval) != 0)
			return -1;
		sizeval = sizeof(stimeoutval);
		if (getsockopt(fSocket, SOL_SOCKET, (int)SO_SNDTIMEO,(char*)&stimeoutval,&sizeval) != 0)
			return -1;
		fRecvTimeOut = rtimeoutval;
		fSendTimeOut = stimeoutval;
	}
	return 0;
}