tcp_connection.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_connection.h"
#include "exception.h"
#include "string_help.h"
#include "event.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

// CL_TCPConnection Construction:

CL_TCPConnection::CL_TCPConnection()
: handle(-1), socket_event_handler(0), read_event(0), write_event(0), exception_event(0)
{
        handle = socket(PF_INET, SOCK_STREAM, 0);
        if (handle == -1)
                throw CL_Exception(TEXT("Socket create failed"));
                
        socket_event_handler = new CL_SocketEventHandler(handle);
        read_event = new CL_Event(socket_event_handler, CL_Event::type_socket_read);
        write_event = new CL_Event(socket_event_handler, CL_Event::type_socket_write);
        exception_event = new CL_Event(socket_event_handler, CL_Event::type_socket_exception);
}

#ifdef WIN32
CL_TCPConnection::CL_TCPConnection(SOCKET socket_handle)
: handle(socket_handle), socket_event_handler(0), read_event(0), write_event(0), exception_event(0)
{
        if (handle == -1)
                throw CL_Exception(TEXT("Invalid socket value"));

        socket_event_handler = new CL_SocketEventHandler(handle);
        read_event = new CL_Event(socket_event_handler, CL_Event::type_socket_read);
        write_event = new CL_Event(socket_event_handler, CL_Event::type_socket_write);
        exception_event = new CL_Event(socket_event_handler, CL_Event::type_socket_exception);
}
#else
CL_TCPConnection::CL_TCPConnection(int socket_handle)
: handle(socket_handle), socket_event_handler(0), read_event(0), write_event(0), exception_event(0)
{
        if (handle == -1)
                throw CL_Exception(TEXT("Invalid socket value"));

        socket_event_handler = new CL_SocketEventHandler(handle);
        read_event = new CL_Event(socket_event_handler, CL_Event::type_socket_read);
        write_event = new CL_Event(socket_event_handler, CL_Event::type_socket_write);
        exception_event = new CL_Event(socket_event_handler, CL_Event::type_socket_exception);
}
#endif

CL_TCPConnection::~CL_TCPConnection()
{
        delete exception_event;
        delete write_event;
        delete read_event;
        delete socket_event_handler;
#ifdef WIN32
        closesocket(handle);
#else
        close(handle);
#endif
}

// CL_TCPConnection Attributes:

CL_Event *CL_TCPConnection::get_read_event()
{
        return read_event;
}
        
CL_Event *CL_TCPConnection::get_write_event()
{
        return write_event;
}
        
CL_Event *CL_TCPConnection::get_exception_event()
{
        return exception_event;
}

// CL_TCPConnection Operations:

void CL_TCPConnection::connect(const CL_String &server, const CL_String &port)
{
        CL_StringA server_a = CL_StringHelp::text_to_local8(server);
        int port_i = CL_StringHelp::text_to_int(port);

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

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

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

        int res = ::connect(handle, (sockaddr *) &name, sizeof(sockaddr_in));
        if (res == -1)
                throw CL_Exception(TEXT("Could not connect to server."));
}

int CL_TCPConnection::send(const void *_data, int length, bool all)
{
        const char *data = (const char *) _data;
        int pos = 0;
        while (pos < length)
        {
                int res = ::send(handle, data+pos, length-pos, 0);
                if (res == -1)
                        throw CL_Exception(TEXT("Write failed to socket!"));

                pos += res;
                if (!all)
                        return pos;
        }
        return pos;
}

int CL_TCPConnection::receive(void *_data, int len, bool all)
{
        char *data = (char *) _data;
        int received = 0;
        while (received < len)
        {
                int res = ::recv(handle, data+received, len-received, 0);
                if (res == -1)
                        throw CL_Exception(TEXT("Read failed on socket!"));

                received += res;
                if (!all)
                        return received;
        }
        return received;
}

int CL_TCPConnection::peek(void *_data, int len)
{
        char *data = (char *) _data;
        int received = 0;
        int res = ::recv(handle, data+received, len-received, MSG_PEEK);
        if (res == -1)
                throw CL_Exception(TEXT("Read peek failed on socket!"));
        received += res;
        return received;
}

void CL_TCPConnection::send_line(const CL_StringA &line)
{
        send(line.data(), (socklen_t) line.length());
        send("\r\n", 2);
}

CL_StringA CL_TCPConnection::receive_line()
{
        char buffer[16*1024];
        CL_StringA line;
        while (line.length() < 128*1024)
        {
                int received = peek(buffer, 16*1024);
                for (int i=0; i<received; i++)
                {
                        if (buffer[i] == '\n')
                        {
                                receive(buffer, i+1);
                                line.append(buffer, i+1);
                                return line;
                        }
                }

                receive(buffer, received);
                line.append(buffer, received);
        }

        throw CL_Exception(TEXT("Received over 128 kb of data from socket. Expected a single line reply!"));
        return CL_StringA();
}

// CL_TCPConnection Implementation:

---