udp: Added option to query local address of socket

2 files changed, 25 insertions, 1 deletions

diff --git a/host/lib/transport/udp_wsa_zero_copy.cpp b/host/lib/transport/udp_wsa_zero_copy.cpp
index 66af394a9..68e94d319 100644
--- a/host/lib/transport/udp_wsa_zero_copy.cpp
+++ b/host/lib/transport/udp_wsa_zero_copy.cpp
@@ -286,6 +286,21 @@ public:
         return local_port;
     }
 
+    std::string get_local_addr(void) const {
+        // Behold the beauty of winsock
+        struct sockaddr_in addr_info;
+        int addr_len = sizeof(addr_info);
+        std::string local_addr;
+        if (getsockname(_sock_fd, (SOCKADDR*) &addr_info, &addr_len) == 0) {
+            // inet_ntoa() guarantees either NULL or null-terminated array
+            char *local_ip = inet_ntoa(addr_info.sin_addr);
+            if (local_ip) {
+                local_addr = std::string(local_ip);
+            }
+        }
+        return local_addr;
+    }
+
     //! Read back the socket's buffer space reserved for receives
     size_t get_recv_buff_size(void) {
         int recv_buff_size = 0;
diff --git a/host/lib/transport/udp_zero_copy.cpp b/host/lib/transport/udp_zero_copy.cpp
index 814f41500..cd271eccf 100644
--- a/host/lib/transport/udp_zero_copy.cpp
+++ b/host/lib/transport/udp_zero_copy.cpp
@@ -244,7 +244,16 @@ public:
 
     size_t get_num_send_frames(void) const {return _num_send_frames;}
     size_t get_send_frame_size(void) const {return _send_frame_size;}
-    uint16_t get_local_port(void) const {return _socket->local_endpoint().port();}
+
+    uint16_t get_local_port(void) const
+    {
+        return _socket->local_endpoint().port();
+    }
+
+    std::string get_local_addr(void) const
+    {
+        return _socket->local_endpoint().address().to_string();
+    }
 
 private:
     //memory management -> buffers and fifos