source: trunk/ETSI-Testsuites/ETSI_auto_IOT/adapter/src/Capture/TcpipServer.cpp @ 35

/**
 * @file TcpipServer.cpp
  * @author Tomas Urban
 * @version 0.3
 * @date 23/07/2009
 */
#include "TcpipServer.h"
#include <iostream>
#include <boost/thread/thread.hpp>
#include <signal.h>
#include "Logger/Logger.h"

using namespace std;

TcpipServer::TcpipServer(void) :
        m_hListeningSocket (SOCKET_CALL_ERROR)
{
}

TcpipServer::~TcpipServer(void)
{
}

void TcpipServer::Run(int nPort)
{
        signal(SIGINT, OnExit);

        Logger::Instance().LogDebug("Initializing TCP/IP connection...");
        
        // create socket and start listening
        struct sockaddr_in srvAddr;

        memset(&srvAddr, 0, sizeof srvAddr);
        srvAddr.sin_family = AF_INET;
        srvAddr.sin_addr.s_addr = INADDR_ANY;
        srvAddr.sin_port = htons(nPort);

        if ( (m_hListeningSocket = socket(AF_INET, SOCK_STREAM, 0)) == SOCKET_CALL_ERROR )
        {
                Logger::Instance().LogError("Error creating listening socket");
                return;
        }
        Logger::Instance().LogDebug("Listening socket successfully created");

        if (bind(m_hListeningSocket, reinterpret_cast<sockaddr*> (&srvAddr), sizeof(srvAddr)) != 0)
        {
                Logger::Instance().LogError("Error binding listening socket; the port is probably used by other process");
                return;
        }

        Logger::Instance().LogDebug("Listening socket successfully bound");

        if (listen(m_hListeningSocket, 0) != 0 )
        {
                Logger::Instance().LogError("Listening operation failed");
                return;         
        }

        std::string s = "Waiting for connection request on port ";
        s += boost::lexical_cast<std::string>(nPort);
        s += "...";
        Logger::Instance().LogInfo(s);

        struct sockaddr_in clientAddr;
        while (-1)
        {
                int nLen = sizeof(clientAddr);
                SOCKET_TYPE sock = accept(
                        m_hListeningSocket, reinterpret_cast<sockaddr*>(&clientAddr), &nLen);
                if (sock == SOCKET_CALL_ERROR)
                        break;
                Logger::Instance().LogInfo("Connection accepted");
                Logger::Instance().LogDebug("Starting socket thread...");
                ConnectionHandler handler(sock);
                boost::thread thread(handler);
        }
        OnExit();
}

TcpipServer::ConnectionHandler::ConnectionHandler(SOCKET_TYPE hSocket) :
        m_hSocket(hSocket)
{
}

void TcpipServer::ConnectionHandler::operator()()
{
        Logger::Instance().LogDebug("Socket thread started successfully");

        ConnectionController cc(m_hSocket);
        {
                boost::mutex::scoped_lock lock(TcpipServer::Instance().m_mutex);
                TcpipServer::Instance().m_lActiveControllers.push_back(&cc);
        }
        cc.Run();
        {
                boost::mutex::scoped_lock lock(TcpipServer::Instance().m_mutex);
                TcpipServer::Instance().m_lActiveControllers.remove(&cc);
        
                if (m_hSocket != SOCKET_CALL_ERROR)
                {
                        CLOSE_SOCKET(m_hSocket);
                        m_hSocket = SOCKET_CALL_ERROR;
                }
                Logger::Instance().LogDebug("Socket thread terminated");
                TcpipServer::Instance().m_operationEnded.notify_one();
        }
}

void TcpipServer::OnExit()
{
        if (m_hListeningSocket != SOCKET_CALL_ERROR)
        {
                {
                        boost::mutex::scoped_lock lock(m_mutex);
                        for(std::list<ConnectionController *>::iterator it = m_lActiveControllers.begin(); 
                                it != m_lActiveControllers.end(); it++)
                                (*it)->Stop();
                }
                // wait for threads to quit
                while (-1)
                {
                        boost::mutex::scoped_lock cond(m_mutex);
                        if (m_lActiveControllers.begin() == m_lActiveControllers.end())
                                break;
                        m_operationEnded.wait(cond);
                }
                CLOSE_SOCKET(m_hListeningSocket);
                m_hListeningSocket = SOCKET_CALL_ERROR;
                Logger::Instance().LogDebug("Listening socket closed");
        }
}

void TcpipServer::OnExit(int nSig)
{
        Instance().OnExit();
}