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.

158 lines
3.5 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 "plFdSet.h"
#include "plSocket.h"
#if HS_BUILD_FOR_UNIX
#include <sys/time.h>
#endif
plFdSet::plFdSet(void)
{
ZeroFds();
}
void plFdSet::SetForSocket(plSocket & in)
{
SOCKET sck = in.GetSocket();
hsAssert(sck>=0, "plFdSet::SetForSocket: socket<0");
if ( sck<0 )
return;
FD_SET(sck, &fFds);
FD_SET(sck, &fErrFds);
if(fMaxFd < sck)
fMaxFd = sck;
}
void plFdSet::ClearForSocket(plSocket & in)
{
SOCKET sck = in.GetSocket();
hsAssert(sck>=0, "plFdSet::ClearForSocket: socket<0");
if ( sck<0 )
return;
FD_CLR(sck, &fFds);
FD_CLR(sck, &fErrFds);
}
bool plFdSet::IsSetFor(plSocket & in)
{
SOCKET sck = in.GetSocket();
hsAssert(sck>=0, "plFdSet::IsSetFor: socket<0");
if ( sck<0 )
return false;
return (FD_ISSET(sck, &fFds) != 0);
}
bool plFdSet::IsErrFor(plSocket & in)
{
SOCKET sck = in.GetSocket();
hsAssert(sck>=0, "plFdSet::IsErrFor: socket<0");
if ( sck<0 )
return false;
return (FD_ISSET(sck, &fErrFds) != 0);
}
int plFdSet::WaitForRead(bool shouldZeroFds, unsigned long timeoutMillis)
{
int ret_val = 0;
if(timeoutMillis == kInfinite)
{
ret_val = select(fMaxFd+1,&fFds,NULL,&fErrFds,NULL);
}
else
{
struct timeval tv;
tv.tv_sec = timeoutMillis / 1000;
tv.tv_usec = (timeoutMillis % 1000) * 1000;
ret_val = select(fMaxFd+1,&fFds,NULL,&fErrFds,&tv);
}
if (shouldZeroFds)
ZeroFds();
return ret_val;
}
int plFdSet::WaitForWrite(bool shouldZeroFds, unsigned long timeoutMillis)
{
int ret_val = 0;
if(timeoutMillis == kInfinite)
{
ret_val = select(fMaxFd+1,NULL,&fFds,&fErrFds,NULL);
}
else
{
timeval tv;
tv.tv_sec = timeoutMillis / 1000;
tv.tv_usec = (timeoutMillis % 1000) * 1000;
ret_val = select(fMaxFd+1,NULL,&fFds,&fErrFds,&tv);
}
if (shouldZeroFds)
ZeroFds();
return ret_val;
}
int plFdSet::WaitForError(bool shouldZeroFds, unsigned long timeoutMillis)
{
int ret_val = 0;
if(timeoutMillis == kInfinite)
{
ret_val = select(fMaxFd+1,NULL,NULL,&fErrFds,NULL);
}
else
{
timeval tv;
tv.tv_sec = timeoutMillis / 1000;
tv.tv_usec = (timeoutMillis % 1000) * 1000;
ret_val = select(fMaxFd+1,NULL,NULL,&fErrFds,&tv);
}
if (shouldZeroFds)
ZeroFds();
return ret_val;
}
void plFdSet::ZeroFds(void)
{
fMaxFd = 0;
FD_ZERO(&fFds);
FD_ZERO(&fErrFds);
}