1 files changed, 6 insertions, 4 deletions
diff --git a/src/socket.h b/src/socket.h
index 1b1a762..289c4f1 100644
--- a/src/socket.h
+++ b/src/socket.h
@@ -1,19 +1,21 @@
 #ifndef POTD_SOCKET_H
 #define POTD_SOCKET_H 1
 
-#include <netinet/in.h>
+#include <netdb.h>
 
 #define POTD_BACKLOG 3
 
 typedef struct psocket {
     int fd;
-    struct sockaddr_in sock;
+    struct addrinfo addr;
 } psocket;
 
 
 int socket_init_in(psocket *psocket, const char *listen_addr,
-                   unsigned int listen_port);
+                   const char *listen_port, struct addrinfo **results);
 
-int socket_bind_listen(psocket *psocket);
+int socket_bind_in(psocket *psocket, struct addrinfo *results);
+
+int socket_listen_in(psocket *psocket);
 
 #endif