From 0697e76440c548910ebab955676114e1e26d7ee3 Mon Sep 17 00:00:00 2001
From: Stefan Metzmacher <metze@samba.org>
Date: Fri, 31 Mar 2006 13:41:49 +0000
Subject: r14843: the results of the srvsrv_* calls depends on the user, try
 some tests anonymous...

metze
(This used to be commit eb63d83f007e374f77ff63b3d7b81f187260c8fa)
---
 source4/torture/rpc/srvsvc.c | 217 ++++++++++++++++++++++++++-----------------
 1 file changed, 132 insertions(+), 85 deletions(-)

(limited to 'source4/torture/rpc/srvsvc.c')

diff --git a/source4/torture/rpc/srvsvc.c b/source4/torture/rpc/srvsvc.c
index 41b27df205..dc48516b91 100644
--- a/source4/torture/rpc/srvsvc.c
+++ b/source4/torture/rpc/srvsvc.c
@@ -415,12 +415,49 @@ static BOOL test_NetSessEnum(struct dcerpc_pipe *p,
 /**************************/
 /* srvsvc_NetShare        */
 /**************************/
+static BOOL test_NetShareCheck(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
+			       const char *device_name)
+{
+	NTSTATUS status;
+	struct srvsvc_NetShareCheck r;
+	BOOL ret = True;
+
+	r.in.server_unc = talloc_asprintf(mem_ctx, "\\\\%s", dcerpc_server_name(p));
+	r.in.device_name = device_name;
+
+	printf("testing NetShareCheck on device '%s'\n", r.in.device_name);
+
+	status = dcerpc_srvsvc_NetShareCheck(p, mem_ctx, &r);
+	if (!NT_STATUS_IS_OK(status)) {
+		printf("dcerpc_srvsvc_NetShareCheck on device '%s' failed - %s\n",
+			r.in.device_name, nt_errstr(status));
+		ret = False;
+	} else if (!W_ERROR_IS_OK(r.out.result)) {
+		printf("NetShareCheck on device '%s' failed - %s\n",
+			r.in.device_name, win_errstr(r.out.result));
+		ret = False;
+	}
+
+	return ret;
+}
+
 static BOOL test_NetShareGetInfo(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
-				 const char *sharename)
+				 const char *sharename, BOOL admin)
 {
 	NTSTATUS status;
 	struct srvsvc_NetShareGetInfo r;
-	uint32_t levels[] = {0, 1, 2, 501, 502, 1005};
+	struct {
+		uint32_t level;
+		WERROR anon_status;
+		WERROR admin_status;
+	} levels[] = {
+		 { 0,		WERR_OK,		WERR_OK },
+		 { 1,		WERR_OK,		WERR_OK },
+		 { 2,		WERR_ACCESS_DENIED,	WERR_OK },
+		 { 501,		WERR_OK,		WERR_OK },
+		 { 502,		WERR_ACCESS_DENIED,	WERR_OK },
+		 { 1005,	WERR_OK,		WERR_OK },
+	};
 	int i;
 	BOOL ret = True;
 
@@ -428,8 +465,12 @@ static BOOL test_NetShareGetInfo(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
 	r.in.share_name = sharename;
 
 	for (i=0;i<ARRAY_SIZE(levels);i++) {
+		WERROR expected;
+
+		r.in.level = levels[i].level;
+		expected = levels[i].anon_status;
+		if (admin) expected = levels[i].admin_status;
 		ZERO_STRUCT(r.out);
-		r.in.level = levels[i];
 
 		printf("testing NetShareGetInfo level %u on share '%s'\n", 
 		       r.in.level, r.in.share_name);
@@ -441,37 +482,19 @@ static BOOL test_NetShareGetInfo(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
 			ret = False;
 			continue;
 		}
-		if (!W_ERROR_IS_OK(r.out.result)) {
-			printf("NetShareGetInfo level %u on share '%s' failed - %s\n",
-				r.in.level, r.in.share_name, win_errstr(r.out.result));
+		if (!W_ERROR_EQUAL(r.out.result, expected)) {
+			printf("NetShareGetInfo level %u on share '%s' failed - %s (expected %s)\n",
+				r.in.level, r.in.share_name, win_errstr(r.out.result),
+				win_errstr(expected));
+			ret = False;
 			continue;
 		}
-	}
 
-	return ret;
-}
-
-static BOOL test_NetShareCheck(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
-			       const char *device_name)
-{
-	NTSTATUS status;
-	struct srvsvc_NetShareCheck r;
-	BOOL ret = True;
-
-	r.in.server_unc = talloc_asprintf(mem_ctx, "\\\\%s", dcerpc_server_name(p));
-	r.in.device_name = device_name;
-
-	printf("testing NetShareCheck on device '%s'\n", r.in.device_name);
-
-	status = dcerpc_srvsvc_NetShareCheck(p, mem_ctx, &r);
-	if (!NT_STATUS_IS_OK(status)) {
-		printf("dcerpc_srvsvc_NetShareCheck on device '%s' failed - %s\n",
-			r.in.device_name, nt_errstr(status));
-		ret = False;
-	} else if (!W_ERROR_IS_OK(r.out.result)) {
-		printf("NetShareCheck on device '%s' failed - %s\n",
-			r.in.device_name, win_errstr(r.out.result));
-		ret = False;
+		if (r.in.level != 2) continue;
+		if (!r.out.info.info2 || !r.out.info.info2->path) continue;
+		if (!test_NetShareCheck(p, mem_ctx, r.out.info.info2->path)) {
+			ret = False;
+		}
 	}
 
 	return ret;
@@ -481,12 +504,22 @@ static BOOL test_NetShareCheck(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
 /* srvsvc_NetShare        */
 /**************************/
 static BOOL test_NetShareEnumAll(struct dcerpc_pipe *p, 
-				 TALLOC_CTX *mem_ctx)
+				 TALLOC_CTX *mem_ctx, BOOL admin)
 {
 	NTSTATUS status;
 	struct srvsvc_NetShareEnumAll r;
 	struct srvsvc_NetShareCtr0 c0;
-	uint32_t levels[] = {0, 1, 2, 501, 502};
+	struct {
+		uint32_t level;
+		WERROR anon_status;
+		WERROR admin_status;
+	} levels[] = {
+		 { 0,	WERR_OK,		WERR_OK },
+		 { 1,	WERR_OK,		WERR_OK },
+		 { 2,	WERR_ACCESS_DENIED,	WERR_OK },
+		 { 501,	WERR_ACCESS_DENIED,	WERR_OK },
+		 { 502,	WERR_ACCESS_DENIED,	WERR_OK },
+	};
 	int i;
 	BOOL ret = True;
 	uint32_t resume_handle;
@@ -501,10 +534,15 @@ static BOOL test_NetShareEnumAll(struct dcerpc_pipe *p,
 
 	for (i=0;i<ARRAY_SIZE(levels);i++) {
 		int j;
+		WERROR expected;
+
+		r.in.level = levels[i].level;
+		expected = levels[i].anon_status;
+		if (admin) expected = levels[i].admin_status;
 
 		ZERO_STRUCT(r.out);
 		resume_handle = 0;
-		r.in.level = levels[i];
+
 		printf("testing NetShareEnumAll level %u\n", r.in.level);
 		status = dcerpc_srvsvc_NetShareEnumAll(p, mem_ctx, &r);
 		if (!NT_STATUS_IS_OK(status)) {
@@ -512,22 +550,19 @@ static BOOL test_NetShareEnumAll(struct dcerpc_pipe *p,
 			ret = False;
 			continue;
 		}
-		if (!W_ERROR_IS_OK(r.out.result)) {
-			printf("NetShareEnumAll level %u failed - %s\n", r.in.level, win_errstr(r.out.result));
+		if (!W_ERROR_EQUAL(r.out.result, expected)) {
+			printf("NetShareEnumAll level %u failed - %s (expected %s)\n",
+				r.in.level, win_errstr(r.out.result),
+				win_errstr(expected));
 			continue;
 		}
 
 		/* call srvsvc_NetShareGetInfo for each returned share */
-		if (r.in.level == 2) {
+		if (r.in.level == 2 && r.out.ctr.ctr2) {
 			for (j=0;j<r.out.ctr.ctr2->count;j++) {
 				const char *name;
-				const char *device;
 				name = r.out.ctr.ctr2->array[j].name;
-				if (!test_NetShareGetInfo(p, mem_ctx, name)) {
-					ret = False;
-				}
-				device = r.out.ctr.ctr2->array[j].path;
-				if (!test_NetShareCheck(p, mem_ctx, device)) {
+				if (!test_NetShareGetInfo(p, mem_ctx, name, admin)) {
 					ret = False;
 				}
 			}
@@ -538,12 +573,22 @@ static BOOL test_NetShareEnumAll(struct dcerpc_pipe *p,
 }
 
 static BOOL test_NetShareEnum(struct dcerpc_pipe *p, 
-			   TALLOC_CTX *mem_ctx)
+			      TALLOC_CTX *mem_ctx, BOOL admin)
 {
 	NTSTATUS status;
 	struct srvsvc_NetShareEnum r;
 	struct srvsvc_NetShareCtr0 c0;
-	uint32_t levels[] = {0, 1, 2, 502};
+	struct {
+		uint32_t level;
+		WERROR anon_status;
+		WERROR admin_status;
+	} levels[] = {
+		 { 0,	WERR_OK,		WERR_OK },
+		 { 1,	WERR_OK,		WERR_OK },
+		 { 2,	WERR_ACCESS_DENIED,	WERR_OK },
+		 { 501,	WERR_UNKNOWN_LEVEL,	WERR_UNKNOWN_LEVEL },
+		 { 502,	WERR_ACCESS_DENIED,	WERR_OK },
+	};
 	int i;
 	BOOL ret = True;
 
@@ -555,8 +600,14 @@ static BOOL test_NetShareEnum(struct dcerpc_pipe *p,
 	r.in.resume_handle = NULL;
 
 	for (i=0;i<ARRAY_SIZE(levels);i++) {
+		WERROR expected;
+
+		r.in.level = levels[i].level;
+		expected = levels[i].anon_status;
+		if (admin) expected = levels[i].admin_status;
+
 		ZERO_STRUCT(r.out);
-		r.in.level = levels[i];
+
 		printf("testing NetShareEnum level %u\n", r.in.level);
 		status = dcerpc_srvsvc_NetShareEnum(p, mem_ctx, &r);
 		if (!NT_STATUS_IS_OK(status)) {
@@ -564,8 +615,10 @@ static BOOL test_NetShareEnum(struct dcerpc_pipe *p,
 			ret = False;
 			continue;
 		}
-		if (!W_ERROR_IS_OK(r.out.result)) {
-			printf("NetShareEnum level %u failed - %s\n", r.in.level, win_errstr(r.out.result));
+		if (!W_ERROR_EQUAL(r.out.result, expected)) {
+			printf("NetShareEnum level %u failed - %s (expected %s)\n",
+				r.in.level, win_errstr(r.out.result),
+				win_errstr(expected));
 			continue;
 		}
 	}
@@ -624,9 +677,9 @@ static BOOL test_NetDiskEnum(struct dcerpc_pipe *p,
 	BOOL ret = True;
 	uint32_t resume_handle=0;
 
+	ZERO_STRUCT(r.in);
 	r.in.server_unc = NULL;
 	r.in.resume_handle = &resume_handle;
-	r.in.disks.discs = NULL;
 
 	for (i=0;i<ARRAY_SIZE(levels);i++) {
 		ZERO_STRUCT(r.out);
@@ -719,6 +772,8 @@ BOOL torture_rpc_srvsvc(struct torture_context *torture)
         struct dcerpc_pipe *p;
 	TALLOC_CTX *mem_ctx;
 	BOOL ret = True;
+	const char *binding = lp_parm_string(-1, "torture", "binding");
+	struct cli_credentials *anon_credentials;
 
 	mem_ctx = talloc_init("torture_rpc_srvsvc");
 
@@ -728,49 +783,41 @@ BOOL torture_rpc_srvsvc(struct torture_context *torture)
 		return False;
 	}
 
-	if (!test_NetCharDevEnum(p, mem_ctx)) {
-		ret = False;
-	}
-
-	if (!test_NetCharDevQEnum(p, mem_ctx)) {
-		ret = False;
-	}
+	ret &= test_NetCharDevEnum(p, mem_ctx);
+	ret &= test_NetCharDevQEnum(p, mem_ctx);
+	ret &= test_NetConnEnum(p, mem_ctx);
+	ret &= test_NetFileEnum(p, mem_ctx);
+	ret &= test_NetSessEnum(p, mem_ctx);
+	ret &= test_NetShareEnumAll(p, mem_ctx, True);
+	ret &= test_NetSrvGetInfo(p, mem_ctx);
+	ret &= test_NetDiskEnum(p, mem_ctx);
+	ret &= test_NetTransportEnum(p, mem_ctx);
+	ret &= test_NetRemoteTOD(p, mem_ctx);
+	ret &= test_NetShareEnum(p, mem_ctx, True);
+	ret &= test_NetShareGetInfo(p, mem_ctx, "ADMIN$", True);
 
-	if (!test_NetConnEnum(p, mem_ctx)) {
-		ret = False;
-	}
-
-	if (!test_NetFileEnum(p, mem_ctx)) {
-		ret = False;
-	}
-
-	if (!test_NetSessEnum(p, mem_ctx)) {
-		ret = False;
-	}
-
-	if (!test_NetShareEnumAll(p, mem_ctx)) {
-		ret = False;
-	}
-
-	if (!test_NetSrvGetInfo(p, mem_ctx)) {
-		ret = False;
-	}
+	status = torture_rpc_connection(mem_ctx, &p, &dcerpc_table_srvsvc);
 
-	if (!test_NetDiskEnum(p, mem_ctx)) {
-		ret = False;
+	if (!binding) {
+		printf("You must specify a ncacn binding string\n");
+		return False;
 	}
 
-	if (!test_NetTransportEnum(p, mem_ctx)) {
-		ret = False;
-	}
+	anon_credentials = cli_credentials_init(mem_ctx);
+	cli_credentials_set_conf(anon_credentials);
+	cli_credentials_set_anonymous(anon_credentials);
 
-	if (!test_NetRemoteTOD(p, mem_ctx)) {
-		ret = False;
+	status = dcerpc_pipe_connect(mem_ctx, 
+				     &p, binding, &dcerpc_table_srvsvc,
+				     anon_credentials, NULL);
+	if (!NT_STATUS_IS_OK(status)) {
+		talloc_free(mem_ctx);
+		return False;
 	}
 
-	if (!test_NetShareEnum(p, mem_ctx)) {
-		ret = False;
-	}
+	ret &= test_NetShareEnumAll(p, mem_ctx, False);
+	ret &= test_NetShareEnum(p, mem_ctx, False);
+	ret &= test_NetShareGetInfo(p, mem_ctx, "ADMIN$", False);
 
 	talloc_free(mem_ctx);
 
-- 
cgit