src/net/server_client.cpp

Go to the documentation of this file.

#include <netinet/in.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include "server_client.h"
#include "tools/conversion.h"
#include "engine/mutex.h"

Mutex inet_ntoa_mutex;

ServerClient::ServerClient(std::string _name, int _fd)
        : ClientCommon(_fd), FastLoopThreadObj(_name)
{
        is_connected = true;
        client_host = "?.?.?.?";
        client_port = -1;

        struct sockaddr_in sock;
        socklen_t sock_len = sizeof(sock);
        if(getsockname(_fd, (sockaddr*)&sock, &sock_len) != 0)
                log(LogError, "Unable to get peer address");
        else
        {
                inet_ntoa_mutex.Lock();
                client_host = inet_ntoa(sock.sin_addr);
                inet_ntoa_mutex.Unlock();
                client_port = sock.sin_port;
        }
}

void ServerClient::Receive()
{
        Lock();
        ClientCommon::Receive();
        Unlock();
}

void ServerClient::Loop()
{
        Lock();

        if(!IsConnected())
        {
                Unlock();
                return;
        }

        fd_set sock_set;
        memset(&sock_set, 0, sizeof(sock_set));
        FD_ZERO(&sock_set);
        FD_SET(fd, &sock_set);

        struct timeval timeout;
        memset(&timeout, 0, sizeof(struct timeval));

        timeout.tv_sec = 0;
        timeout.tv_usec = 100000;

        if(select(fd + 1, &sock_set, NULL, NULL, &timeout) < 0)
        {
                if(timeout.tv_sec != 0 && timeout.tv_usec != 0)
                        log(LogPanic, "Error during select");
        }

        if(timeout.tv_sec == 0 && timeout.tv_usec == 0)
        {
                Unlock();
                return;
        }

        if( FD_ISSET( fd, &sock_set) )
        {
                Receive();
                if(IsConnected())
                        while(ParseBuffer())
                                ;
        }
        else
        {
                log(LogInfo, "%s dirty disconnected", GetAddress().c_str());
                Disconnect();
        }

        Unlock();
}

const std::string ServerClient::GetAddress()
{
        return client_host + std::string(":") + int2str(client_port);
}

void ServerClient::Disconnect()
{
        log(LogInfo, "%s disconnected", GetAddress().c_str());
        info_mutex.Lock();
        is_connected = false;
        info_mutex.Unlock();
        ClientCommon::Disconnect();
}

bool ServerClient::IsConnected()
{
        info_mutex.Lock();
        bool ret = is_connected;
        info_mutex.Unlock();
        return ret;
}

---