1 files changed, 4 insertions, 0 deletions
diff --git a/util.h b/util.h
index 7d79856..936c2de 100644
--- a/util.h
+++ b/util.h
@@ -1,6 +1,10 @@
 #include <sys/socket.h>
+#include <netdb.h>
 
 char *stringcopy(const char *s, int len);
 char *addr2string(struct sockaddr *addr, socklen_t len);
+uint16_t port_get(struct sockaddr *sa);
+void port_set(struct sockaddr *sa, uint16_t port);
+
 void printfchars(char *prefixfmt, char *prefix, char *charfmt, char *chars, int len);
 int connectport(int type, char *host, char *port);