net_SetPort() sets a port within a socket address

[vlc] / include / vlc_network.h

diff --git a/include/vlc_network.h b/include/vlc_network.h
index bece596243b2b556c8eaca4f17c5eb9fc71e80ca..e2495ad5bb20de4f16832976333caf6a6f0e9af3 100644 (file)
--- a/include/vlc_network.h
+++ b/include/vlc_network.h
@@ -331,6 +331,20 @@ static inline uint16_t net_GetPort (const struct sockaddr *addr)
     return 0;
 }
 
+static inline void net_SetPort (struct sockaddr *addr, uint16_t port)
+{
+    switch (addr->sa_family)
+    {
+#ifdef AF_INET6
+        case AF_INET6:
+            ((struct sockaddr_in6 *)addr)->sin6_port = port;
+           break;
+#endif
+        case AF_INET:
+            ((struct sockaddr_in *)addr)->sin_port = port;
+           break;
+    }
+}
 # ifdef __cplusplus
 }
 # endif