NetServerTCP3Recv.cpp

Go to the documentation of this file.

////////////////////////////////////////////////////////////////////////////////
//    Scorched3D (c) 2000-2009
//
//    This file is part of Scorched3D.
//
//    Scorched3D 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 2 of the License, or
//    (at your option) any later version.
//
//    Scorched3D 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 Scorched3D; if not, write to the Free Software
//    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
////////////////////////////////////////////////////////////////////////////////

#include <net/NetServerTCP3Recv.h>
#include <net/NetServerTCP3Coms.h>
#include <net/NetMessagePool.h>
#include <common/Logger.h>

NetServerTCP3Recv::NetServerTCP3Recv(
        TCPsocket socket, 
        unsigned int destinationId, unsigned int ipAddress,
        NetMessageHandler *recieveMessageHandler) :
        socket_(socket), 
        destinationId_(destinationId), ipAddress_(ipAddress),
        recieveMessageHandler_(recieveMessageHandler),
        messagesRecieved_(0), bytesIn_(0),
        stopped_(false), running_(true)
{
        socketSet_ = SDLNet_AllocSocketSet(1);
        SDLNet_TCP_AddSocket(socketSet_, socket_);
        recvThread_ = SDL_CreateThread(
                NetServerTCP3Recv::recvThreadFunc, (void *) this);
        if (recvThread_ == 0)
        {
                Logger::log(
                        "NetServerTCP3Recv: Failed to create recv thread");
        }
}

NetServerTCP3Recv::~NetServerTCP3Recv()
{
        SDLNet_FreeSocketSet(socketSet_);
        socketSet_ = 0;
}

int NetServerTCP3Recv::recvThreadFunc(void *c)
{
        // Call a non-static class thread to do the processing in (just for convienience)
        NetServerTCP3Recv *th = (NetServerTCP3Recv*) c;
        while (th->running_)
        {
                if (!th->actualRecvFunc()) break;
                SDL_Delay(100);
        }
        th->stopped_ = true;
        return 0;
}

bool NetServerTCP3Recv::actualRecvFunc()
{
        // Check if there is anything to recieve
        int numready = SDLNet_CheckSockets(socketSet_, 100);
        if (numready == -1) return false;
        if (numready == 0) return true;

        // Receive the length of the string message
        char lenbuf[4];
        if (!NetServerTCP3Coms::SDLNet_TCP_Recv_Full(socket_, lenbuf, 4))
        {
                Logger::log(S3D::formatStringBuffer(
                        "NetServerTCP3Recv: Read failed for length"));
                return false;
        }
        Uint32 len = SDLNet_Read32(lenbuf);
        
        // Cannot recieve a message large than .5 MB
        if (len > 5000000 || len == 0)
        {
                Logger::log(S3D::formatStringBuffer(
                        "NetServerTCP3Recv: Buffer was too large to recieve.  Size %i.",
                        len));
                return false;
        }

        // allocate the buffer memory
        NetMessage *buffer = NetMessagePool::instance()->
                getFromPool(NetMessage::BufferMessage, 
                                destinationId_, ipAddress_);
        buffer->getBuffer().allocate(len);
        buffer->getBuffer().setBufferUsed(len);

        // get the string buffer over the socket
        if (!NetServerTCP3Coms::SDLNet_TCP_Recv_Full(socket_, 
                buffer->getBuffer().getBuffer(),
                len))
        {
                Logger::log(S3D::formatStringBuffer(
                        "NetServerTCP3Recv: Read failed for buffer"));
                NetMessagePool::instance()->addToPool(buffer);
                return false;
        }

        // Notify that this message has been recieved
        NetInterface::getBytesIn() += len;
        bytesIn_ += len;
        messagesRecieved_ ++;
        recieveMessageHandler_->addMessage(buffer);

        return true;
}

void NetServerTCP3Recv::wait()
{
        int status;
        SDL_WaitThread(recvThread_, &status);
}

---