Play with includes so that libalias can be compiled both as userland

library and kernel module.

1 files changed, 13 insertions, 2 deletions

diff --git a/sys/netinet/libalias/alias_nbt.c b/sys/netinet/libalias/alias_nbt.c
index b51abd1..dc25a35 100644
--- a/sys/netinet/libalias/alias_nbt.c
+++ b/sys/netinet/libalias/alias_nbt.c
@@ -41,18 +41,29 @@ __FBSDID("$FreeBSD$");
 */
 
 /* Includes */
+#ifdef _KERNEL
+#include <sys/param.h>
+#include <sys/ctype.h>
+#include <sys/libkern.h>
+#else
+#include <sys/types.h>
 #include <ctype.h>
 #include <stdio.h>
 #include <string.h>
-#include <sys/types.h>
+#include <arpa/inet.h>
+#endif
+
 #include <netinet/in_systm.h>
 #include <netinet/in.h>
-#include <arpa/inet.h>
 #include <netinet/ip.h>
 #include <netinet/udp.h>
 #include <netinet/tcp.h>
 
+#ifdef _KERNEL
+#include <netinet/libalias/alias_local.h>
+#else
 #include "alias_local.h"
+#endif
 
 typedef struct {
 	struct in_addr	oldaddr;