6 files changed, 11 insertions, 0 deletions
diff --git a/source4/torture/smb2/config.mk b/source4/torture/smb2/config.mk
index 90e57bd6d8..8c95208151 100644
--- a/source4/torture/smb2/config.mk
+++ b/source4/torture/smb2/config.mk
@@ -2,6 +2,8 @@
 #################################
 # Start SUBSYSTEM TORTURE_SMB2
 [SUBSYSTEM::TORTURE_SMB2]
+PRIVATE_PROTO_HEADER = \
+		proto.h
 OBJ_FILES = \
 		connect.o \
 		scan.o \
diff --git a/source4/torture/smb2/connect.c b/source4/torture/smb2/connect.c
index d4016b8770..7472493894 100644
--- a/source4/torture/smb2/connect.c
+++ b/source4/torture/smb2/connect.c
@@ -25,6 +25,8 @@
 #include "libcli/smb2/smb2.h"
 #include "libcli/smb2/smb2_calls.h"
 
+#include "torture/smb2/proto.h"
+
 /*
   send a close
 */
diff --git a/source4/torture/smb2/find.c b/source4/torture/smb2/find.c
index aae8f8042f..07c3f865e2 100644
--- a/source4/torture/smb2/find.c
+++ b/source4/torture/smb2/find.c
@@ -24,6 +24,8 @@
 #include "libcli/smb2/smb2.h"
 #include "libcli/smb2/smb2_calls.h"
 
+#include "torture/smb2/proto.h"
+
 static struct {
 	const char *name;
 	uint16_t level;
diff --git a/source4/torture/smb2/getinfo.c b/source4/torture/smb2/getinfo.c
index faf889f8ff..2e3d2f2fac 100644
--- a/source4/torture/smb2/getinfo.c
+++ b/source4/torture/smb2/getinfo.c
@@ -24,6 +24,8 @@
 #include "libcli/smb2/smb2.h"
 #include "libcli/smb2/smb2_calls.h"
 
+#include "torture/smb2/proto.h"
+
 static struct {
 	const char *name;
 	uint16_t level;
diff --git a/source4/torture/smb2/scan.c b/source4/torture/smb2/scan.c
index 646f790e18..44dced9427 100644
--- a/source4/torture/smb2/scan.c
+++ b/source4/torture/smb2/scan.c
@@ -26,6 +26,7 @@
 #include "lib/cmdline/popt_common.h"
 #include "lib/events/events.h"
 
+#include "torture/smb2/proto.h"
 
 #define FNAME "scan-getinfo.dat"
 #define DNAME "scan-getinfo.dir"
diff --git a/source4/torture/smb2/setinfo.c b/source4/torture/smb2/setinfo.c
index bf63d26dbb..562493b047 100644
--- a/source4/torture/smb2/setinfo.c
+++ b/source4/torture/smb2/setinfo.c
@@ -25,6 +25,8 @@
 #include "libcli/smb2/smb2.h"
 #include "libcli/smb2/smb2_calls.h"
 
+#include "torture/smb2/proto.h"
+
 #define BASEDIR ""
 
 /* basic testing of all SMB2 setinfo calls