tcp_listen.cpp

Go to the documentation of this file.

/*
**  ClanLib SDK
**  Copyright (c) 1997-2005 The ClanLib Team
**
**  This software is provided 'as-is', without any express or implied
**  warranty.  In no event will the authors be held liable for any damages
**  arising from the use of this software.
**
**  Permission is granted to anyone to use this software for any purpose,
**  including commercial applications, and to alter it and redistribute it
**  freely, subject to the following restrictions:
**
**  1. The origin of this software must not be misrepresented; you must not
**     claim that you wrote the original software. If you use this software
**     in a product, an acknowledgment in the product documentation would be
**     appreciated but is not required.
**  2. Altered source versions must be plainly marked as such, and must not be
**     misrepresented as being the original software.
**  3. This notice may not be removed or altered from any source distribution.
**
**  Note: Some of the libraries ClanLib link to may have additional
**  requirements or restrictions.
**
**  File Author(s):
**
**    Magnus Norddahl
*/

#include "precomp.h"
#include "tcp_listen.h"
#include "tcp_connection.h"
#include "exception.h"
#include "event.h"
#include "string_help.h"
#include "socket_event_handler.h"
#ifdef WIN32
#include <winsock.h>
#define socklen_t int
#else
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <sys/types.h>
#include <unistd.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#endif
#include <iostream>

// CL_TCPListen Construction:

CL_TCPListen::CL_TCPListen(const CL_String &port, int queue_size, bool force_bind)
: handle(-1), socket_event_handler(0), accept_event(0)
{
        handle = socket(PF_INET, SOCK_STREAM, 0);
        if (handle == -1)
                throw CL_Exception(TEXT("Socket create failed"));

        // force_bind: SO_REUSEADDR

        int portI = CL_StringHelp::text_to_int(port);

        sockaddr_in name;
        memset(&name, 0, sizeof(sockaddr_in));
        name.sin_family = AF_INET;
        name.sin_addr.s_addr = INADDR_ANY;
        name.sin_port = htons(portI);

        int res = ::bind(handle, (sockaddr *) &name, sizeof(sockaddr_in));
        if (res != -1)
                res = ::listen(handle, queue_size);
        if (res == -1)
        {
                #ifdef WIN32
                closesocket(handle);
                #else
                close(handle);
                #endif
                throw CL_Exception(TEXT("Could not bind socket port ") + port);
        }
        
        socket_event_handler = new CL_SocketEventHandler(handle);
        accept_event = new CL_Event(socket_event_handler, CL_Event::type_socket_read);
}
        
CL_TCPListen::CL_TCPListen(const CL_String &address, const CL_String &port, int queue_size, bool force_bind)
: handle(-1), socket_event_handler(0), accept_event(0)
{
        std::string addressA = CL_StringHelp::text_to_local8(address);
        int portI = CL_StringHelp::text_to_int(port);

        int ip = inet_addr(addressA.c_str());
        if (ip == INADDR_NONE)
        {
                hostent *host_info = gethostbyname(addressA.c_str());
                if (host_info == NULL)
                        throw CL_Exception(TEXT("Could not lookup ") + address);

                ip = *((int *) host_info->h_addr_list[0]);
        }

        handle = socket(PF_INET, SOCK_STREAM, 0);
        if (handle == -1)
                throw CL_Exception(TEXT("Socket create failed"));

        sockaddr_in name;
        memset(&name, 0, sizeof(sockaddr_in));
        name.sin_family = AF_INET;
        name.sin_addr.s_addr = ip;
        name.sin_port = htons(portI);

        int res = ::bind(handle, (sockaddr *) &name, sizeof(sockaddr_in));
        if (res != -1)
                res = ::listen(handle, queue_size);
        if (res == -1)
        {
                #ifdef WIN32
                closesocket(handle);
                #else
                close(handle);
                #endif
                throw CL_Exception(TEXT("Could not bind socket on ") + address + TEXT(" port ") + port);
        }

        socket_event_handler = new CL_SocketEventHandler(handle);
        accept_event = new CL_Event(socket_event_handler, CL_Event::type_socket_read);
}
        
CL_TCPListen::~CL_TCPListen()
{
        delete accept_event;
        delete socket_event_handler;
#ifdef WIN32
        closesocket(handle);
#else
        close(handle);
#endif
}

// CL_TCPListen Attributes:

CL_Event *CL_TCPListen::get_accept_event()
{
        return accept_event;
}

// CL_TCPListen Operations:

CL_TCPConnection *CL_TCPListen::accept(int timeout)
{
        // EAGAIN
        // O_NONBLOCK

        sockaddr_in name;
        memset(&name, 0, sizeof(sockaddr_in));
        name.sin_family = AF_INET;
        
        socklen_t size = sizeof(sockaddr_in);

#ifdef WIN32
        SOCKET sock = ::accept(handle, (sockaddr *) &name, &size);
        if (sock == -1)
                return 0;
#else
        int sock = ::accept(handle, (sockaddr *) &name, &size);
        if (sock == -1)
                return 0;
#endif
        return new CL_TCPConnection(sock);
}

// CL_TCPListen Implementation:

---