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.
195 lines
4.2 KiB
195 lines
4.2 KiB
/*==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; |
|
} |