1 files changed, 11 insertions, 0 deletions
diff --git a/general.h b/general.h
index 52b2ed1..397264f 100644
--- a/general.h
+++ b/general.h
@@ -43,6 +43,16 @@
 #  include <limits.h>
 #endif
 
+#if defined(HAVE_UNISTD_H)
+#  include <unistd.h>
+#endif
+
+#include <pwd.h>
+
+#if defined(HAVE_GRP_H)
+#  include <grp.h>
+#endif
+
 #include "xmalloc.h"
 
 /* NULL pointer type. */
@@ -317,6 +327,7 @@ extern int group_member __P((gid_t));
 extern char **get_group_list __P((int *));
 extern int *get_group_array __P((int *));
 
+extern void set_vyatta_restricted_mode __P((void));
 enum vyatta_restricted_type { OUTPUT, FULL };
 extern int in_vyatta_restricted_mode __P((enum vyatta_restricted_type));