add us_socket_remote_port(int ssl, struct us_socket_t *s) (#210)

uNetworking · Oct 13, 2023 · bb86c9c · bb86c9c
1 parent ac4e90e
commit bb86c9c
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 2 deletions.
diff --git a/src/io_uring/io_socket.c b/src/io_uring/io_socket.c
@@ -33,6 +33,10 @@ int us_socket_local_port(int ssl, struct us_socket_t *s) {
     return 0;
 }
 
+int us_socket_remote_port(int ssl, struct us_socket_t *s) {
+    return 0;
+}
+
 void us_socket_shutdown_read(int ssl, struct us_socket_t *s) {
 
 }
@@ -132,4 +136,4 @@ void us_socket_shutdown(int ssl, struct us_socket_t *s) {
 
 }
 
-#endif
+#endif
diff --git a/src/libusockets.h b/src/libusockets.h
@@ -320,6 +320,9 @@ struct us_socket_t *us_socket_close(int ssl, struct us_socket_t *s, int code, vo
 /* Returns local port or -1 on failure. */
 int us_socket_local_port(int ssl, struct us_socket_t *s);
 
+/* Returns remote ephemeral port or -1 on failure. */
+int us_socket_remote_port(int ssl, struct us_socket_t *s);
+
 /* Copy remote (IP) address of socket, or fail with zero length. */
 void us_socket_remote_address(int ssl, struct us_socket_t *s, char *buf, int *length);
 

diff --git a/src/socket.c b/src/socket.c
@@ -34,6 +34,15 @@ int us_socket_local_port(int ssl, struct us_socket_t *s) {
     }
 }
 
+int us_socket_remote_port(int ssl, struct us_socket_t *s) {
+    struct bsd_addr_t addr;
+    if (bsd_remote_addr(us_poll_fd(&s->p), &addr)) {
+        return -1;
+    } else {
+        return bsd_addr_get_port(&addr);
+    }
+}
+
 void us_socket_shutdown_read(int ssl, struct us_socket_t *s) {
     /* This syscall is idempotent so no extra check is needed */
     bsd_shutdown_socket_read(us_poll_fd((struct us_poll_t *) s));
@@ -219,4 +228,4 @@ void us_socket_shutdown(int ssl, struct us_socket_t *s) {
     }
 }
 
-#endif
+#endif