1 files changed, 3 insertions, 2 deletions
diff --git a/src/net.h b/src/net.h
index 125d4f3..5afe4b7 100644
--- a/src/net.h
+++ b/src/net.h
@@ -1,6 +1,7 @@
 #ifndef __NET_H__
 #define __NET_H__
 
+#include <stdint.h>
 #include <stdlib.h>
 #include <sys/types.h>
 
@@ -9,10 +10,10 @@ int net_accept(int fd);
 int net_connect(const char *host, const char *port);
 
 int net_send_all(int fd, const void *, size_t);
-int net_send_buf(int fd, const void *, size_t);
+int net_send_buf(int fd, const void *, uint32_t);
 
 ssize_t net_recv_all(int fd, void *, size_t);
-int net_recv_buf(int fd, void **, size_t *);
+int net_recv_buf(int fd, void **, uint32_t *);
 int net_recv_static(int fd, void *, size_t);
 
 #endif