1 files changed, 12 insertions, 0 deletions
diff --git a/utils.c b/utils.c
index 91e44b3..87ed0fe 100644
--- a/utils.c
+++ b/utils.c
@@ -7,6 +7,18 @@
 #include "main.h"
 #include "utils.h"
 
+uint16_t sockaddr_port(struct sockaddr_in46 *addr)
+{
+	switch (addr->storage.ss_family) {
+	case AF_INET:
+		return ntohs(addr->in4.sin_port);
+	case AF_INET6:
+		return ntohs(addr->in6.sin6_port);
+	default:
+		return 0;
+	}
+}
+
 char *
 sockaddr_to_str(struct sockaddr_in46 *addr, char *buf, size_t buflen)
 {