Test for netinet/tcp.h and udp.h and use those
diff --git a/net.c b/net.c
index 17ace69..066965a 100644
--- a/net.c
+++ b/net.c
@@ -36,9 +36,12 @@
 #include <sys/socket.h>
 #include <sys/un.h>
 #include <netinet/in.h>
-#ifdef FREEBSD
+#ifdef HAVE_NETINET_TCP_H
 #include <netinet/tcp.h>
 #endif
+#ifdef HAVE_NETINET_UDP_H
+#include <netinet/udp.h>
+#endif
 #include <arpa/inet.h>
 #if defined(LINUX)
 #include <asm/types.h>