udp_socket.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 "udp_socket.h"
#include "event.h"
#include "exception.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_UDPSocket Construction:

CL_UDPSocket::CL_UDPSocket()
: handle(-1), socket_event_handler(0), read_event(0), write_event(0), exception_event(0)
{
        handle = socket(PF_INET, SOCK_DGRAM, 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);
}

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

// CL_UDPSocket Attributes:

CL_Event *CL_UDPSocket::get_read_event()
{
        return read_event;
}

CL_Event *CL_UDPSocket::get_write_event()
{
        return write_event;
}

CL_Event *CL_UDPSocket::get_exception_event()
{
        return exception_event;
}

// CL_UDPSocket Operations:

int CL_UDPSocket::send(const void *data, int len, const CL_IPAddress &to)
{
        const sockaddr_in &name = to.get_sockaddr_v4();
        int res = ::sendto(handle, (const char *) data, len, 0, (const sockaddr *) &name, sizeof(sockaddr_in));
        if (res == -1)
                throw CL_Exception(TEXT("Write failed to socket!"));
        return res;
}

int CL_UDPSocket::receive(void *data, int len, CL_IPAddress &from)
{
        sockaddr_in &name = from.get_sockaddr_v4();
        memset(&name, 0, sizeof(sockaddr_in));
        name.sin_family = AF_INET;

        socklen_t addr_len = sizeof(sockaddr_in);
        int res = recvfrom(handle, (char *) data, len, 0, (sockaddr *) &name, &addr_len);
        if (res == -1)
                throw CL_Exception(TEXT("Read failed on socket!"));
        return res;
}

int CL_UDPSocket::peek(void *data, int len, CL_IPAddress &from, int timeout)
{
        if (get_read_event()->wait(timeout) == false)
                return -1;

        sockaddr_in &name = from.get_sockaddr_v4();
        memset(&name, 0, sizeof(sockaddr_in));
        name.sin_family = AF_INET;

        socklen_t addr_len = sizeof(sockaddr_in);
        int res = recvfrom(handle, (char *) data, len, MSG_PEEK, (sockaddr *) &name, &addr_len);
        if (res == -1)
                throw CL_Exception(TEXT("Read failed on socket!"));
        return res;
}

// CL_UDPSocket Implementation:

---