r5866: Add InitShutdown IDL and torture test.

Implement push side of NDR_LEN4|NDR_NOTERM strings (pull side was already present)
(This used to be commit ea61ec1122841716ed5d90085ba79e7bf691bd6a)

2 files changed, 160 insertions, 6 deletions

diff --git a/source4/torture/rpc/initshutdown.c b/source4/torture/rpc/initshutdown.c
new file mode 100644
index 0000000000..0aa64e3eb2
--- /dev/null
+++ b/source4/torture/rpc/initshutdown.c
@@ -0,0 +1,146 @@
+/* 
+   Unix SMB/CIFS implementation.
+   test suite for initshutdown operations
+
+   Copyright (C) Tim Potter 2003
+   Copyright (C) Jelmer Vernooij 2004-2005
+   
+   This program is free software; you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation; either version 2 of the License, or
+   (at your option) any later version.
+   
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+   
+   You should have received a copy of the GNU General Public License
+   along with this program; if not, write to the Free Software
+   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+*/
+
+#include "includes.h"
+#include "librpc/gen_ndr/ndr_initshutdown.h"
+
+static void init_initshutdown_String(TALLOC_CTX *mem_ctx, struct initshutdown_String *name, const char *s)
+{
+	name->name = talloc(mem_ctx, struct initshutdown_String_sub);
+	name->name->name = s;
+}
+
+static BOOL test_Init(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
+			const char *msg, uint32_t timeout)
+{
+	struct initshutdown_Init r;
+	NTSTATUS status;
+	uint16_t hostname = 0x0;
+	
+	r.in.hostname = &hostname;
+	r.in.message = talloc(mem_ctx, struct initshutdown_String);
+	init_initshutdown_String(mem_ctx, r.in.message, msg);
+	r.in.force_apps = 1;
+	r.in.timeout = timeout;
+	r.in.reboot = 1;
+
+	status = dcerpc_initshutdown_Init(p, mem_ctx, &r);
+
+	if (!NT_STATUS_IS_OK(status)) {
+		printf("initshutdown_Init failed - %s\n", nt_errstr(status));
+		return False;
+	}
+
+	if (!W_ERROR_IS_OK(r.out.result)) {
+		printf("initshutdown_Init failed - %s\n", win_errstr(r.out.result));
+		return False;
+	}
+
+	return True;
+}
+
+static BOOL test_InitEx(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
+			const char *msg, uint32_t timeout)
+{
+	struct initshutdown_InitEx r;
+	NTSTATUS status;
+	uint16_t hostname = 0x0;
+	
+	r.in.hostname = &hostname;
+	r.in.message = talloc(mem_ctx, struct initshutdown_String);
+	init_initshutdown_String(mem_ctx, r.in.message, msg);
+	r.in.force_apps = 1;
+	r.in.timeout = timeout;
+	r.in.reboot = 1;
+	r.in.reason = 0;
+
+	status = dcerpc_initshutdown_InitEx(p, mem_ctx, &r);
+
+	if (!NT_STATUS_IS_OK(status)) {
+		printf("initshutdown_InitEx failed - %s\n", nt_errstr(status));
+		return False;
+	}
+
+	if (!W_ERROR_IS_OK(r.out.result)) {
+		printf("initshutdown_InitEx failed - %s\n", win_errstr(r.out.result));
+		return False;
+	}
+
+	return True;
+}
+
+static BOOL test_Abort(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
+{
+	struct initshutdown_Abort r;
+	NTSTATUS status;
+	uint16_t server = 0x0;
+
+	r.in.server = &server;
+	
+	status = dcerpc_initshutdown_Abort(p, mem_ctx, &r);
+
+	if (!NT_STATUS_IS_OK(status)) {
+		printf("initshutdown_Abort failed - %s\n", nt_errstr(status));
+		return False;
+	}
+
+	if (!W_ERROR_IS_OK(r.out.result)) {
+		printf("initshutdown_Abort failed - %s\n", win_errstr(r.out.result));
+		return False;
+	}
+
+	return True;
+}
+
+BOOL torture_rpc_initshutdown(void)
+{
+    NTSTATUS status;
+    struct dcerpc_pipe *p;
+	TALLOC_CTX *mem_ctx;
+	BOOL ret = True;
+
+	mem_ctx = talloc_init("torture_rpc_initshutdown");
+
+	status = torture_rpc_connection(&p, 
+					DCERPC_INITSHUTDOWN_NAME, 
+					DCERPC_INITSHUTDOWN_UUID, 
+					DCERPC_INITSHUTDOWN_VERSION);
+
+	if (!NT_STATUS_IS_OK(status)) {
+		return False;
+	}
+
+	if (!lp_parm_bool(-1, "torture", "dangerous", False)) {
+		printf("initshutdown tests disabled - enable dangerous tests to use\n");
+	} else {
+		ret &= test_Init(p, mem_ctx, "spottyfood", 30);
+		ret &= test_Abort(p, mem_ctx);
+		ret &= test_InitEx(p, mem_ctx, "spottyfood", 30);
+		ret &= test_Abort(p, mem_ctx);
+	}
+
+	talloc_free(mem_ctx);
+
+    torture_rpc_close(p);
+
+	return ret;
+}
diff --git a/source4/torture/rpc/winreg.c b/source4/torture/rpc/winreg.c
index 9d259ae47a..df0114b669 100644
--- a/source4/torture/rpc/winreg.c
+++ b/source4/torture/rpc/winreg.c
@@ -23,6 +23,12 @@
 #include "includes.h"
 #include "librpc/gen_ndr/ndr_winreg.h"
 
+static void init_initshutdown_String(TALLOC_CTX *mem_ctx, struct initshutdown_String *name, const char *s)
+{
+	name->name = talloc(mem_ctx, struct initshutdown_String_sub);
+	name->name->name = s;
+}
+
 static void init_winreg_String(struct winreg_String *name, const char *s)
 {
 	name->name = s;
@@ -555,10 +561,11 @@ static BOOL test_InitiateSystemShutdown(struct dcerpc_pipe *p, TALLOC_CTX *mem_c
 {
 	struct winreg_InitiateSystemShutdown r;
 	NTSTATUS status;
+	uint16_t hostname = 0x0;
 	
-	r.in.hostname = NULL;
-	r.in.message = talloc(mem_ctx, struct winreg_String);
-	init_winreg_String(r.in.message, msg);
+	r.in.hostname = &hostname;
+	r.in.message = talloc(mem_ctx, struct initshutdown_String);
+	init_initshutdown_String(mem_ctx, r.in.message, msg);
 	r.in.force_apps = 1;
 	r.in.timeout = timeout;
 	r.in.reboot = 1;
@@ -583,10 +590,11 @@ static BOOL test_InitiateSystemShutdownEx(struct dcerpc_pipe *p, TALLOC_CTX *mem
 {
 	struct winreg_InitiateSystemShutdownEx r;
 	NTSTATUS status;
+	uint16_t hostname = 0x0;
 	
-	r.in.hostname = NULL;
-	r.in.message = talloc(mem_ctx, struct winreg_String);
-	init_winreg_String(r.in.message, msg);
+	r.in.hostname = &hostname;
+	r.in.message = talloc(mem_ctx, struct initshutdown_String);
+	init_initshutdown_String(mem_ctx, r.in.message, msg);
 	r.in.force_apps = 1;
 	r.in.timeout = timeout;
 	r.in.reboot = 1;