Real Name

ixwebsocket/IXUdpSocket.cpp

@@ -0,0 +1,125 @@
+/*
+ *  IXUdpSocket.cpp
+ *  Author: Benjamin Sergeant
+ *  Copyright (c) 2020 Machine Zone, Inc. All rights reserved.
+ */
+
+#include "IXUdpSocket.h"
+
+#include "IXNetSystem.h"
+#include <cstring>
+#include <sstream>
+
+namespace ix
+{
+UdpSocket::UdpSocket(int fd) : _sockfd(fd) { ; }
+
+UdpSocket::~UdpSocket() { close(); }
+
+void UdpSocket::close()
+{
+    if (_sockfd == -1)
+        return;
+
+    closeSocket(_sockfd);
+    _sockfd = -1;
+}
+
+int UdpSocket::getErrno()
+{
+    int err;
+
+#ifdef _WIN32
+    err = WSAGetLastError();
+#else
+    err = errno;
+#endif
+
+    return err;
+}
+
+bool UdpSocket::isWaitNeeded()
+{
+    int err = getErrno();
+
+    if (err == EWOULDBLOCK || err == EAGAIN || err == EINPROGRESS) {
+        return true;
+    }
+
+    return false;
+}
+
+void UdpSocket::closeSocket(int fd)
+{
+#ifdef _WIN32
+    closesocket(fd);
+#else
+    ::close(fd);
+#endif
+}
+
+bool UdpSocket::init(const std::string &host, int port, std::string &errMsg)
+{
+    _sockfd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
+    if (_sockfd < 0) {
+        errMsg = "Could not create socket";
+        return false;
+    }
+
+#ifdef _WIN32
+    unsigned long nonblocking = 1;
+    ioctlsocket(_sockfd, FIONBIO, &nonblocking);
+#else
+    fcntl(_sockfd, F_SETFL, O_NONBLOCK); // make socket non blocking
+#endif
+
+    memset(&_server, 0, sizeof(_server));
+    _server.sin_family = AF_INET;
+    _server.sin_port = htons(port);
+
+    // DNS resolution.
+    struct addrinfo hints, *result = nullptr;
+    memset(&hints, 0, sizeof(hints));
+    hints.ai_family = AF_INET;
+    hints.ai_socktype = SOCK_DGRAM;
+
+    int ret = getaddrinfo(host.c_str(), nullptr, &hints, &result);
+    if (ret != 0) {
+        errMsg = strerror(UdpSocket::getErrno());
+        freeaddrinfo(result);
+        close();
+        return false;
+    }
+
+    struct sockaddr_in *host_addr = (struct sockaddr_in *)result->ai_addr;
+    memcpy(&_server.sin_addr, &host_addr->sin_addr, sizeof(struct in_addr));
+    freeaddrinfo(result);
+
+    return true;
+}
+
+ssize_t UdpSocket::sendto(const std::string &buffer)
+{
+    return (ssize_t)::sendto(_sockfd,
+                             buffer.data(),
+                             buffer.size(),
+                             0,
+                             (struct sockaddr *)&_server,
+                             sizeof(_server));
+}
+
+ssize_t UdpSocket::recvfrom(char *buffer, size_t length)
+{
+#ifdef _WIN32
+    int addressLen = (int)sizeof(_server);
+#else
+    socklen_t addressLen = (socklen_t)sizeof(_server);
+#endif
+    return (ssize_t)::recvfrom(_sockfd,
+                               buffer,
+                               length,
+                               0,
+                               (struct sockaddr *)&_server,
+                               &addressLen);
+}
+} // namespace ix