NatHelper Class Reference

#include <nat-helper.hpp>

Collaboration diagram for NatHelper:

Public Member Functions
	NatHelper ()
	NatHelper (int port)
	~NatHelper ()=default
void	run (bool async=true)
void	stop ()

Private Member Functions
void	cleanEmptyRooms ()
void	onClientDesc (Client &client, CompletionStatus status)
void	onConnectionAccepted (const std::error_code &ec)
void	worker ()

Private Attributes
asio::ip::tcp::acceptor	acceptor
std::map< Client *, std::shared_ptr< Client > >	incomingClients
asio::io_service	io_service
asio::ip::tcp::socket	pendingSocket
int	port
std::map< std::string, std::shared_ptr< Room > >	rooms
std::thread	worker_thread

Detailed Description

Definition at line 47 of file nat-helper.hpp.

Constructor & Destructor Documentation

NatHelper() [1/2]

NatHelper::NatHelper

(

)

Definition at line 50 of file nat-helper.cpp.

                     : NatHelper(12558)
{
 
}

NatHelper() [2/2]

NatHelper::NatHelper

(

int

port

)

Definition at line 41 of file nat-helper.cpp.

                             : port(port),
        io_service(),
        //work_guard(io_service.get_executor()),
        acceptor(io_service),
        pendingSocket(io_service)
{
 
}

~NatHelper()

NatHelper::~NatHelper ( )

default

Member Function Documentation

cleanEmptyRooms()

void NatHelper::cleanEmptyRooms ( )

private

Definition at line 104 of file nat-helper.cpp.

                               {
        for(auto it = rooms.begin(); it != rooms.end(); ){
                if(it->second->isEmpty()){
                        std::cout << "Removing empty room " << it->first << "\n";
                        it = rooms.erase(it);
                } else {
                        it++;
                }
        }
}

References rooms.

Referenced by onClientDesc().

onClientDesc()

void NatHelper::onClientDesc ( Client & client, CompletionStatus status )

private

Definition at line 115 of file nat-helper.cpp.

                                                                   {
        if(status != CompletionStatus::Success){
                std::cerr << "Error reading client description" << std::endl;
                incomingClients.erase(&client);
                return;
        }
 
        const auto& roomName = client.getRoomName();
        std::cout << "Moving client " << client.getClientName()
                << " to room " << roomName
                << std::endl;
 
        auto roomIt = rooms.find(roomName);
        if(roomIt == rooms.end()){
                cleanEmptyRooms();
                std::cout << "Creating room " << roomName << std::endl;
                roomIt = rooms.insert({roomName, std::make_shared<Room>(roomName)}).first;
        }
 
        auto clientIt = incomingClients.find(&client);
        auto& room = roomIt->second;
        if(room->isFull()){
                std::cerr << "Room " << roomName << " is full, dropping client" << std::endl;
        } else {
                roomIt->second->addClient(std::move(clientIt->second));
        }
        incomingClients.erase(clientIt);
}

References cleanEmptyRooms(), first, incomingClients, rooms, and Success.

Referenced by onConnectionAccepted().

Here is the call graph for this function:

onConnectionAccepted()

void NatHelper::onConnectionAccepted ( const std::error_code & ec )

private

Definition at line 87 of file nat-helper.cpp.

                                                           {
        if(ec){
                std::cerr << "Error accepting client connection: " << ec.message() << std::endl;
                return;
        }
 
        using namespace std::placeholders;
 
        auto client = std::make_shared<Client>(std::move(pendingSocket));
        client->readDescription(
                        std::bind(&NatHelper::onClientDesc, this, _1, _2));
        incomingClients.insert({client.get(), std::move(client)});
 
        acceptor.async_accept(pendingSocket,
                        std::bind(&NatHelper::onConnectionAccepted, this, _1));
}

References acceptor, incomingClients, onClientDesc(), onConnectionAccepted(), and pendingSocket.

Referenced by onConnectionAccepted(), and run().

Here is the call graph for this function:

run()

void NatHelper::run

(

bool

async = true

)

Definition at line 62 of file nat-helper.cpp.

                             {
        using namespace std::placeholders;
 
        asio::ip::tcp::endpoint endpoint(asio::ip::tcp::v4(), port);
        acceptor.open(endpoint.protocol());
        acceptor.bind(endpoint);
        acceptor.listen(asio::socket_base::max_connections);
        acceptor.set_option(asio::ip::tcp::acceptor::reuse_address(false));
        acceptor.async_accept(pendingSocket,
                        std::bind(&NatHelper::onConnectionAccepted, this, _1));
 
        if(async){
                std::cout << "Starting thread" << std::endl;
                worker_thread = std::thread(&NatHelper::worker, this);
        } else {
                worker();
        }
}

References acceptor, onConnectionAccepted(), pendingSocket, port, worker(), and worker_thread.

Referenced by main(), and worker().

Here is the call graph for this function:

stop()

void NatHelper::stop

(

)

Definition at line 81 of file nat-helper.cpp.

                    {
        std::cout << "Stopping..." << std::endl;
        io_service.stop();
        worker_thread.join();
}

References io_service, and worker_thread.

worker()

void NatHelper::worker ( )

private

Definition at line 56 of file nat-helper.cpp.

                      {
        std::cout << "Running" << std::endl;
        io_service.run();
        std::cout << "End" << std::endl;
}

References io_service, and run().

Referenced by run().

Here is the call graph for this function:

Member Data Documentation

acceptor

asio::ip::tcp::acceptor NatHelper::acceptor

private

Definition at line 70 of file nat-helper.hpp.

Referenced by onConnectionAccepted(), and run().

incomingClients

std::map<Client *, std::shared_ptr<Client> > NatHelper::incomingClients

private

Definition at line 73 of file nat-helper.hpp.

Referenced by onClientDesc(), and onConnectionAccepted().

io_service

asio::io_service NatHelper::io_service

private

Definition at line 66 of file nat-helper.hpp.

Referenced by stop(), and worker().

pendingSocket

asio::ip::tcp::socket NatHelper::pendingSocket

private

Definition at line 71 of file nat-helper.hpp.

Referenced by onConnectionAccepted(), and run().

port

int NatHelper::port

private

Definition at line 64 of file nat-helper.hpp.

Referenced by run().

rooms

std::map<std::string, std::shared_ptr<Room> > NatHelper::rooms

private

Definition at line 74 of file nat-helper.hpp.

Referenced by cleanEmptyRooms(), and onClientDesc().

worker_thread

std::thread NatHelper::worker_thread

private

Definition at line 68 of file nat-helper.hpp.

Referenced by run(), and stop().

---

The documentation for this class was generated from the following files:

• nat-helper/nat-helper.hpp
• nat-helper/nat-helper.cpp