Refactored socket code into UDPSocket class.

2015-04-10 15:58:25 -07:00
parent aa74bc57f5
commit 2736cd86de
9 changed files with 283 additions and 31 deletions
--- a/xpcPlugin/UDPSocket.cpp
+++ b/xpcPlugin/UDPSocket.cpp
@@ -0,0 +1,168 @@
+//Copyright (c) 2013-2015 United States Government as represented by the Administrator of the
+//National Aeronautics and Space Administration. All Rights Reserved.
+#include "Log.h"
+#include "UDPSocket.h"
+
+namespace XPC
+{
+	UDPSocket::UDPSocket(std::uint16_t recvPort)
+	{
+#if LOG_VERBOSITY > 1
+		Log::FormatLine("[UDPSocket] Opening socket (port:%d)",	recvPort);
+#endif
+		// Setup Port
+		struct sockaddr_in localAddr;
+		localAddr.sin_family = AF_INET;
+		localAddr.sin_addr.s_addr = INADDR_ANY;
+		localAddr.sin_port = htons(recvPort);
+		
+		//Create and bind the socket
+#ifdef _WIN32
+		WSADATA wsa;
+        int startResult = WSAStartup(MAKEWORD(2, 2), &wsa);
+        if (startResult != 0)
+		{
+#if LOG_VERBOSITY > 0
+            Log::FormatLine("[UDPSocket] ERROR: WSAStartup failed with error code %i.", startResult);
+#endif
+			return;
+		}
+        if ((this->sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) == INVALID_SOCKET)
+        {
+#if LOG_VERBOSITY > 0
+            int err = WSAGetLastError();
+            Log::FormatLine("[UDPSocket] ERROR: Failed to open socket. (Error code %i)", err);
+#endif
+            return;
+        }
+#elif (__APPLE__ || __linux)
+        if ((this->sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) == -1)
+        {
+            Log::WriteLine("[UDPSocket] ERROR: Failed to open socket");
+            return;
+        }
+        int optval = 1;
+        setsockopt(theSocket.sock, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval));
+        setsockopt(theSocket.sock, SOL_SOCKET, SO_REUSEPORT, &optval, sizeof(optval));
+#endif
+        if (bind(this->sock, (struct sockaddr*)&localAddr, sizeof(localAddr)) != 0)
+        {
+#if LOG_VERBOSITY > 0
+            int err = WSAGetLastError();
+            Log::FormatLine("[UDPSocket] ERROR: Failed to bind socket. (Error code %i)", err);
+#endif
+            return;
+        }
+
+        //Set Timout
+        int usTimeOut = 500;
+
+#ifdef _WIN32
+        DWORD msTimeOutWin = 1; // Minimum socket timeout in Windows is 1ms
+        if(setsockopt(this->sock, SOL_SOCKET, SO_RCVTIMEO, (char *)&msTimeOutWin, sizeof(msTimeOutWin)) != 0)
+        {
+#if LOG_VERBOSITY > 1
+            int err = WSAGetLastError();
+            Log::FormatLine("[UDPSocket] ERROR: Failed to set timeout. (Error code %i)", err);
+#endif
+        }
+#else
+		struct timeval tv;
+		tv.tv_sec = 0;  /* Sec Timeout */
+		tv.tv_usec = usTimeOut;  // Microsec Timeout
+		setsockopt(this->sock, SOL_SOCKET, SO_RCVTIMEO, (char *)&tv, sizeof(struct timeval));
+#endif
+	}
+
+	UDPSocket::~UDPSocket()
+	{
+#if LOG_VERBOSITY > 1
+		Log::FormatLine("[UDPSocket] Closing socket (%d)", this->sock);
+#endif
+#ifdef _WIN32
+		closesocket(this->sock);
+#elif (__APPLE__ || __linux)
+		close(this->sock);
+#endif
+	}
+
+	int UDPSocket::Read(std::uint8_t* dst, int maxLen, struct sockaddr* recvAddr)
+	{
+		socklen_t recvaddrlen = sizeof(*recvAddr);
+		int status = 0;
+
+#ifdef _WIN32
+		// Windows readUDP needs the select command- minimum timeout is 1ms.
+		// Without this playback becomes choppy
+
+		// Definitions
+		FD_SET stReadFDS;
+		FD_SET stExceptFDS;
+		struct timeval tv;
+		struct sockaddr_in sourceAddr;
+
+		// Setup for Select
+		FD_ZERO(&stReadFDS);
+		FD_SET(sock, &stReadFDS);
+		FD_ZERO(&stExceptFDS);
+		FD_SET(sock, &stExceptFDS);
+		tv.tv_sec = 0;  /* Sec Timeout */
+		tv.tv_usec = 250;  // Microsec Timeout
+
+		// Select Command
+		int result = select(-1, &stReadFDS, (FD_SET *)0, &stExceptFDS, &tv);
+#if LOG_VERBOSITY > 1
+        if (result == SOCKET_ERROR)
+        {
+            int err = WSAGetLastError();
+            Log::FormatLine("[UDPSocket] ERROR: Select failed. (Error code %i)", err);
+        }
+#endif
+        if (result <= 0) // No Data or error
+		{
+			return -1;
+		}
+
+		// If no error: Read Data
+		recvaddrlen = sizeof(sourceAddr);
+        status = recvfrom(sock, (char*)dst, maxLen, 0, (SOCKADDR *)&sourceAddr, &recvaddrlen);
+#else
+		// For apple or linux-just read - will timeout in 0.5 ms
+		status = (int)recvfrom(recfd.sock, dataRef, 5000, 0, recvaddr, &recvaddrlen);
+#endif
+		return status;
+	}
+
+	void UDPSocket::SendTo(std::uint8_t* buffer, std::size_t len, std::string remoteHost, std::uint16_t remotePort)
+	{
+		if (remoteHost == "localhost")
+		{
+			remoteHost = "127.0.0.1";
+		}
+		std::uint32_t ip = inet_addr(remoteHost.c_str());
+		SendTo(buffer, len, ip, remotePort);
+	}
+
+	void UDPSocket::SendTo(std::uint8_t* buffer, std::size_t len, std::uint32_t remoteIP, std::uint16_t remotePort)
+	{
+		struct sockaddr_in remoteAddr;
+		remoteAddr.sin_family = AF_INET;
+		remoteAddr.sin_port = htons(remotePort);
+		remoteAddr.sin_addr.s_addr = remoteIP;
+
+#ifdef _WIN32
+		const char on = 1;
+#else
+		int on = 1;
+#endif
+		setsockopt(sock, SOL_SOCKET, SO_BROADCAST, &on, sizeof(on));
+
+		buffer[4] = (std::uint8_t)len;
+		if (sendto(sock, (char*)buffer, len, 0, (const struct sockaddr *) &remoteAddr, sizeof(remoteAddr)) < 0)
+		{
+#if LOG_VERBOSITY > 0
+            Log::FormatLine("[UDPSocket] Send failed. (remote: %s:%d)", remoteAddr, remotePort);
+#endif
+		}
+	}
+}