cube-game-plus-plus/client/network/Client.cpp

#include <iostream>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <cstring>
#include <stdexcept>
#include <unistd.h>
#include <poll.h>

#include "client/network/Client.h"
#include "common/stream/Stream.h"

Client::Client()
{
}

Client::~Client()
{
    if(connectionSocket != -1)
    {
        close(connectionSocket);
    }
}

bool Client::start(const string& ip, unsigned short port, IClientListener* listener)
{
    connectionSocket = socket(AF_INET, SOCK_STREAM, 0);
    if(connectionSocket == -1)
    {
        throw runtime_error(string("cannot create socket: ") + strerror(errno));
    }

    struct sockaddr_in socketData;
    memset(&socketData, 0, sizeof(struct sockaddr_in));
    socketData.sin_family = AF_INET;
    socketData.sin_port = htons(port);
    if(inet_aton(ip.data(), &socketData.sin_addr) == 0)
    {
        throw runtime_error(ip + " is not a valid ip");
    }

    if(connect(connectionSocket, (struct sockaddr*) &socketData, sizeof(struct sockaddr_in)) == 0)
    {
        clientListener = listener;
        shouldRun = true;
        serverListenThread = thread(&Client::listenOnServer, this);
        return true;
    }
    return false;
}

void Client::stop()
{
    shouldRun = false;
    serverListenThread.join();
}

void Client::listenOnServer()
{
    struct pollfd fds;
    fds.fd = connectionSocket;
    fds.events = POLLIN;
    fds.revents = 0;
    
    clientListener->onServerJoin(connectionSocket);
    
    Stream st;
    
    while(shouldRun)
    {
        int pollData = poll(&fds, 1, 100);
        if(pollData > 0)
        {
            st.readSocket(connectionSocket);
            if(st.hasData())
            {
                clientListener->onServerPackage(connectionSocket, st);
            }
            else
            {
                // server closed connection
                clientListener->onServerClose(connectionSocket);
                break;
            }
        }
        else if(pollData == -1)
        {
            cout << "poll error: " << strerror(errno) << endl;
        }
    }
}