Merge pull request #55 from cookiengineer/feature/port

(int) APE_socket_port(socket) implementation
nidium · Apr 24, 2017 · 2a8998d · 2a8998d
2 parents 611d48d + cd8125d
commit 2a8998d
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/src/ape_socket.c b/src/ape_socket.c
@@ -906,6 +906,17 @@ int APE_socket_write(ape_socket *socket, void *data, size_t len,
     return 0;
 }
 
+uint16_t APE_socket_port(ape_socket *socket)
+{
+    if (!socket) {
+        return 0;
+    }
+
+
+    return ntohs(socket->sockaddr.sin_port);
+
+}
+
 char *APE_socket_ipv4(ape_socket *socket)
 {
     if (!socket) {

diff --git a/src/ape_socket.h b/src/ape_socket.h
@@ -257,6 +257,7 @@ int APE_socket_writev(ape_socket *socket, const struct iovec *iov, int iovcnt);
 int APE_sendfile(ape_socket *socket, const char *file);
 int APE_socket_is_online(ape_socket *socket);
 
+uint16_t APE_socket_port(ape_socket *socket);
 char *APE_socket_ipv4(ape_socket *socket);
 
 int ape_socket_destroy(ape_socket *socket);

diff --git a/tests/unittest_socket.cpp b/tests/unittest_socket.cpp
@@ -23,6 +23,7 @@
 //@TODO: void APE_socket_shutdown(ape_socket *socket);
 //@TODO: void APE_socket_shutdown_now(ape_socket *socket);
 //@TODO: int APE_sendfile(ape_socket *socket, const char *file);
+//@TODO: int APE_socket_port(ape_socket *socket);
 //@TODO: char *APE_socket_ipv4(ape_socket *socket);
 //@TODO: int APE_socket_is_online(ape_socket *socket)
 //@TODO: int ape_socket_do_jobs(ape_socket *socket);