Mercurial > zfc9000 / file diff

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/sources/network/udpsocket.cpp	Thu Dec 11 05:59:43 2014 +0200
@@ -0,0 +1,98 @@
+#include <sys/socket.h>
+#include <sys/types.h>
+#include <sys/time.h>
+#include <netinet/in.h>
+#include <string.h>
+#include <fcntl.h>
+#include "udpsocket.h"
+
+// -----------------------------------------------------------------------------
+//
+UDPSocket::UDPSocket() :
+	m_socket (socket (AF_INET, SOCK_DGRAM, 0)) {}
+
+// -----------------------------------------------------------------------------
+//
+UDPSocket::~UDPSocket() {}
+
+// -------------------------------------------------------------------------------------------------
+//
+METHOD
+UDPSocket::set_blocking (bool a) -> bool
+{
+	int flags = fcntl (m_socket, F_GETFL, 0);
+	int newflags = (a ? (flags | O_NONBLOCK) : (flags & ~O_NONBLOCK));
+
+	if (flags < 0 || fcntl (m_socket, F_SETFL, newflags) != 0)
+	{
+		m_error = "Unable to set socket as non-blocking";
+		return false;
+	}
+
+	return true;
+}
+
+// -------------------------------------------------------------------------------------------------
+//
+METHOD
+UDPSocket::bind (unsigned short port) -> bool
+{
+	struct sockaddr_in svaddr;
+	memset (&svaddr, 0, sizeof svaddr);
+	svaddr.sin_family = AF_INET;
+	svaddr.sin_port = htons (port);
+	svaddr.sin_addr.s_addr = htonl (INADDR_ANY);
+
+	if (::bind (m_socket, reinterpret_cast<struct sockaddr*> (&svaddr), sizeof svaddr) == -1)
+	{
+		m_error = String ("Couldn't bind to port ") + String::from_number (port);
+		return false;
+	}
+
+	return true;
+}
+
+// -------------------------------------------------------------------------------------------------
+//
+METHOD
+UDPSocket::read (Datagram& datagram) -> bool
+{
+	sockaddr_in claddr;
+	socklen_t socklen = sizeof claddr;
+	static unsigned char packet[MAX_DATAGRAM_LENGTH];
+	int length = ::recvfrom (m_socket, packet, sizeof packet, 0,
+		reinterpret_cast<struct sockaddr*> (&claddr), &socklen);
+
+	if (length == -1)
+	{
+		// We got an error, though EWOULDBLOCK is silent as it means no packets recieved.
+		if (errno != EWOULDBLOCK)
+			m_error.sprintf ("recvfrom error: %s", strerror (errno));;
+
+		return false;
+	}
+
+	datagram.from.host = ntohl (claddr.sin_addr.s_addr);
+	datagram.from.port = ntohs (claddr.sin_port);
+	datagram.data = Bytestream (packet, length);
+	return true;
+}
+
+// -------------------------------------------------------------------------------------------------
+//
+METHOD
+UDPSocket::send (const Bytestream& data, const IPAddress& addr) -> bool
+{
+	struct sockaddr_in claddr = addr.to_sockaddr_in();
+
+	int res = ::sendto (m_socket, data.data(), data.written_length(), 0,
+		reinterpret_cast<struct sockaddr*> (&claddr), sizeof claddr);
+
+	if (res == -1)
+	{
+		m_error = String ("Unable to launch packet: ") + strerror (errno);
+		return false;
+	}
+
+	return true;
+}