initial commit

third_party/gamepad/src/udp_receiver.cpp

@@ -0,0 +1,60 @@
+#include "udp_receiver.h"
+#include <iostream>
+#include <cstring>
+#include <unistd.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+
+/**
+ * @brief Constructor for UdpReceiver.
+ * @param port The port to listen for UDP data.
+ */
+UdpReceiver::UdpReceiver(int port) : socket_fd_(-1) {
+    socket_fd_ = socket(AF_INET, SOCK_DGRAM, 0);
+    if (socket_fd_ == -1) {
+        std::cerr << "Failed to create socket." << std::endl;
+        return;
+    }
+
+    sockaddr_in serverAddress;
+    serverAddress.sin_family = AF_INET;
+    serverAddress.sin_addr.s_addr = INADDR_ANY;
+    serverAddress.sin_port = htons(port);
+
+    if (bind(socket_fd_, (struct sockaddr*)&serverAddress, sizeof(serverAddress)) < 0) {
+        std::cerr << "Failed to bind socket." << std::endl;
+        close(socket_fd_);
+        return;
+    }
+}
+
+/**
+ * @brief Receive UDP data.
+ * @return A vector containing received data as uint8_t.
+ */
+std::vector<uint8_t> UdpReceiver::ReceiveData() {
+    std::vector<uint8_t> buffer(1024);
+    sockaddr_in clientAddress;
+    socklen_t clientAddressLen = sizeof(clientAddress);
+
+    int bytesRead = recvfrom(socket_fd_, buffer.data(), buffer.size(), 0, (struct sockaddr*)&clientAddress, &clientAddressLen);
+    if (bytesRead == -1) {
+        std::cerr << "Failed to receive data." << std::endl;
+        return std::vector<uint8_t>();
+    }
+
+    buffer.resize(bytesRead);
+
+    return buffer;
+}
+
+/**
+ * @brief Destructor for UdpReceiver.
+ */
+UdpReceiver::~UdpReceiver() {
+    if (socket_fd_ != -1) {
+        close(socket_fd_);
+    }
+}