1

lib/socket_compat.c

@@ -230,3 +230,18 @@ ssize_t socket_recvfrom(socket_t sock, void* buf, size_t len,
     return recvfrom(sock, buf, len, 0, src, src_len);
 #endif
 }
+
+uint16_t ss_get_port(const struct sockaddr_storage *addr) {
+    if (!addr) return 0;
+
+    if (addr->ss_family == AF_INET) {
+        const struct sockaddr_in *ipv4 = (const struct sockaddr_in *)addr;
+        return ntohs(ipv4->sin_port);
+    }
+    else if (addr->ss_family == AF_INET6) {
+        const struct sockaddr_in6 *ipv6 = (const struct sockaddr_in6 *)addr;
+        return ntohs(ipv6->sin6_port);
+    }
+
+    return 0;
+}