tcpserver.hpp

#ifndef TCPSERVER_H
#define TCPSERVER_H
 
#include <memory>
#include <vector>
 
#include "socketdevice.h"
 
class TcpServer
{
public:
    TcpServer(int listen_port, const std::string& listen_on )
        : _socket( std::make_unique<SocketDevice>( SocketDevice::tcpip, listen_on, listen_port, "", 0 ) )
    {
        // Setting SO_REUSEADDR ensure that upon termination, the socket can be reopened without the "bind" error
        _socket->setReuseAddr();
    }
    virtual ~TcpServer() {}
 
    bool initialize() {
        if ( _socket->initialize() )
        {
            if ( _socket->bind() )
            {
                if ( _socket->listen() )
                {
                    return true;
                }
            }
        }
        return false;
    }
 
    void shutdown() {
        _socket->shutdown();
    }
 
    bool isReady() {
        return _socket->isInitialized();
    }
 
    std::vector<std::unique_ptr<SocketDevice>> checkNewClients() {
        std::vector<std::unique_ptr<SocketDevice>> new_clients;
        // Accept all incoming sockets:
        while ( _socket->hasIncomingData() )
        {
            std::unique_ptr<SocketDevice> new_client = _socket->accept();
            if ( new_client != NULL )
                new_clients.push_back( std::move(new_client) );
        }
        return new_clients;
    }
 
private:
    std::unique_ptr<SocketDevice> _socket;
};
 
#endif // TCPSERVER_H