From c33a7ca84cc313299af746b60a7cf94c4589c44b Mon Sep 17 00:00:00 2001
From: Günther Deschner <gd@samba.org>
Date: Wed, 17 Mar 2010 00:53:21 +0100
Subject: s4-smbtorture: add test_CreateKey_opts() to RPC-WINREG test.

Guenther
---
 source4/torture/rpc/winreg.c | 44 +++++++++++++++++++++++++++++++++++---------
 1 file changed, 35 insertions(+), 9 deletions(-)

(limited to 'source4/torture')

diff --git a/source4/torture/rpc/winreg.c b/source4/torture/rpc/winreg.c
index b0ed61261a..f218258ad5 100644
--- a/source4/torture/rpc/winreg.c
+++ b/source4/torture/rpc/winreg.c
@@ -104,10 +104,16 @@ static bool test_NotifyChangeKeyValue(struct dcerpc_binding_handle *b,
 	return true;
 }
 
-static bool test_CreateKey(struct dcerpc_binding_handle *b,
-			   struct torture_context *tctx,
-			   struct policy_handle *handle, const char *name,
-			   const char *kclass)
+static bool test_CreateKey_opts(struct torture_context *tctx,
+				struct dcerpc_binding_handle *b,
+				struct policy_handle *handle,
+				const char *name,
+				const char *kclass,
+				enum winreg_KeyType options,
+				uint32_t access_mask,
+				struct winreg_SecBuf *secdesc,
+				enum winreg_CreateAction *action_taken_p,
+				struct policy_handle *new_handle_p)
 {
 	struct winreg_CreateKey r;
 	struct policy_handle newhandle;
@@ -115,22 +121,42 @@ static bool test_CreateKey(struct dcerpc_binding_handle *b,
 
 	ZERO_STRUCT(r);
 	r.in.handle = handle;
-	r.out.new_handle = &newhandle;
 	init_winreg_String(&r.in.name, name);
 	init_winreg_String(&r.in.keyclass, kclass);
-	r.in.options = 0x0;
-	r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
-	r.in.action_taken = r.out.action_taken = &action_taken;
-	r.in.secdesc = NULL;
+	r.in.options = options;
+	r.in.access_mask = access_mask;
+	r.in.action_taken = &action_taken;
+	r.in.secdesc = secdesc;
+	r.out.new_handle = &newhandle;
+	r.out.action_taken = &action_taken;
 
 	torture_assert_ntstatus_ok(tctx, dcerpc_winreg_CreateKey_r(b, tctx, &r),
 				   "CreateKey failed");
 
 	torture_assert_werr_ok(tctx,  r.out.result, "CreateKey failed");
 
+	if (new_handle_p) {
+		*new_handle_p = newhandle;
+	}
+	if (action_taken_p) {
+		*action_taken_p = action_taken;
+	}
+
 	return true;
 }
 
+static bool test_CreateKey(struct dcerpc_binding_handle *b,
+			   struct torture_context *tctx,
+			   struct policy_handle *handle, const char *name,
+			   const char *kclass)
+{
+	return test_CreateKey_opts(tctx, b, handle, name, kclass,
+				   REG_KEYTYPE_NON_VOLATILE,
+				   SEC_FLAG_MAXIMUM_ALLOWED,
+				   NULL, /* secdesc */
+				   NULL, /* action_taken */
+				   NULL /* new_handle */);
+}
 
 /*
   createkey testing with a SD
-- 
cgit