You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
211 lines
5.1 KiB
211 lines
5.1 KiB
14 years ago
|
/*==LICENSE==*
|
||
|
|
||
|
CyanWorlds.com Engine - MMOG client, server and tools
|
||
13 years ago
|
Copyright (C) 2011 Cyan Worlds, Inc.
|
||
14 years ago
|
|
||
|
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
|
||
13 years ago
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
14 years ago
|
GNU General Public License for more details.
|
||
|
|
||
|
You should have received a copy of the GNU General Public License
|
||
13 years ago
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
||
|
Additional permissions under GNU GPL version 3 section 7
|
||
|
|
||
|
If you modify this Program, or any covered work, by linking or
|
||
|
combining it with any of RAD Game Tools Bink SDK, Autodesk 3ds Max SDK,
|
||
|
NVIDIA PhysX SDK, Microsoft DirectX SDK, OpenSSL library, Independent
|
||
|
JPEG Group JPEG library, Microsoft Windows Media SDK, or Apple QuickTime SDK
|
||
|
(or a modified version of those libraries),
|
||
|
containing parts covered by the terms of the Bink SDK EULA, 3ds Max EULA,
|
||
|
PhysX SDK EULA, DirectX SDK EULA, OpenSSL and SSLeay licenses, IJG
|
||
|
JPEG Library README, Windows Media SDK EULA, or QuickTime SDK EULA, the
|
||
|
licensors of this Program grant you additional
|
||
|
permission to convey the resulting work. Corresponding Source for a
|
||
|
non-source form of such a combination shall include the source code for
|
||
|
the parts of OpenSSL and IJG JPEG Library used as well as that of the covered
|
||
|
work.
|
||
14 years ago
|
|
||
|
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;
|
||
|
}
|