svn commit: samba r18764 - in branches/SAMBA_4_0/source/torture/rpc: .

gd at samba.org gd at samba.org
Thu Sep 21 02:45:03 GMT 2006


Author: gd
Date: 2006-09-21 02:45:02 +0000 (Thu, 21 Sep 2006)
New Revision: 18764

WebSVN: http://websvn.samba.org/cgi-bin/viewcvs.cgi?view=rev&root=samba&rev=18764

Log:
Test all standalone DFS root functions in the torture test.

Guenther

Modified:
   branches/SAMBA_4_0/source/torture/rpc/dfs.c


Changeset:
Modified: branches/SAMBA_4_0/source/torture/rpc/dfs.c
===================================================================
--- branches/SAMBA_4_0/source/torture/rpc/dfs.c	2006-09-21 02:41:52 UTC (rev 18763)
+++ branches/SAMBA_4_0/source/torture/rpc/dfs.c	2006-09-21 02:45:02 UTC (rev 18764)
@@ -23,7 +23,122 @@
 #include "torture/torture.h"
 #include "torture/rpc/rpc.h"
 #include "librpc/gen_ndr/ndr_dfs_c.h"
+#include "librpc/gen_ndr/ndr_srvsvc_c.h"
+#include "libnet/libnet.h"
+#include "libcli/raw/libcliraw.h"
+#include "torture/util.h"
+#include "libcli/libcli.h"
+#include "lib/cmdline/popt_common.h"
 
+#define SMBTORTURE_DFS_SHARENAME "smbtorture_dfs_share"
+#define SMBTORTURE_DFS_DIRNAME "\\smbtorture_dfs_dir"
+#define SMBTORTURE_DFS_PATHNAME "C:"SMBTORTURE_DFS_DIRNAME
+
+#define IS_DFS_VERSION_UNSUPPORTED_CALL_W2K3(x,y)\
+	if (x == DFS_MANAGER_VERSION_W2K3) {\
+		if (!W_ERROR_EQUAL(y,WERR_NOT_SUPPORTED)) {\
+			printf("expected WERR_NOT_SUPPORTED\n");\
+			return False;\
+		}\
+		return True;\
+	}\
+
+static BOOL test_NetShareAdd(TALLOC_CTX *mem_ctx, const char *host, const char *sharename, const char *dir)
+{
+	NTSTATUS status;
+	struct srvsvc_NetShareInfo2 i;
+	struct libnet_context* libnetctx;
+	struct libnet_AddShare r;
+
+	printf("Creating share %s\n", sharename);
+
+	if (!(libnetctx = libnet_context_init(NULL))) {
+		return False;
+	}
+
+	libnetctx->cred = cmdline_credentials;
+
+	i.name			= sharename;
+	i.type			= STYPE_DISKTREE;
+	i.path			= dir;
+	i.max_users		= (uint32_t) -1;
+	i.comment		= "created by smbtorture";
+	i.password		= NULL;
+	i.permissions		= 0x0;
+	i.current_users		= 0x0;
+
+	r.level 		= 2;
+	r.in.server_name	= host;
+	r.in.share 		= i;
+
+	status = libnet_AddShare(libnetctx, mem_ctx, &r);
+	if (!NT_STATUS_IS_OK(status)) {
+		d_printf("Failed to add new share: %s (%s)\n", 
+			nt_errstr(status), r.out.error_string);
+		return False;
+	}
+
+	return True;
+}
+
+static BOOL test_NetShareDel(TALLOC_CTX *mem_ctx, const char *host, const char *sharename)
+{
+	NTSTATUS status;
+	struct libnet_context* libnetctx;
+	struct libnet_DelShare r;
+
+	printf("Deleting share %s\n", sharename);
+
+	if (!(libnetctx = libnet_context_init(NULL))) {
+		return False;
+	}
+
+	libnetctx->cred = cmdline_credentials;
+
+	r.in.share_name		= sharename;
+	r.in.server_name	= host;
+
+	status = libnet_DelShare(libnetctx, mem_ctx, &r);
+	if (!NT_STATUS_IS_OK(status)) {
+		d_printf("Failed to delete share: %s (%s)\n", 
+			nt_errstr(status), r.out.error_string);
+		return False;
+	}
+
+	return True;
+}
+
+static BOOL test_CreateDir(TALLOC_CTX *mem_ctx, 
+			   struct smbcli_state **cli, 
+			   const char *host, 
+			   const char *share, 
+			   const char *dir)
+{
+	printf("Creating directory %s\n", dir);
+
+	if (!torture_open_connection_share(mem_ctx, cli, host, share, NULL)) {
+		return False;
+	}
+
+	if (!torture_setup_dir(*cli, dir)) {
+		return False;
+	}
+
+	return True;
+}
+
+static BOOL test_DeleteDir(struct smbcli_state *cli, const char *dir)
+{
+	printf("Deleting directory %s\n", dir);
+
+	if (smbcli_deltree(cli->tree, dir) == -1) {
+		printf("Unable to delete dir %s - %s\n", dir, smbcli_errstr(cli->tree));
+		return False;
+	}
+
+	return True;
+}
+
 static BOOL test_GetManagerVersion(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, enum dfs_ManagerVersion *version)
 {
 	NTSTATUS status;
@@ -218,6 +333,127 @@
 }
 
 
+static BOOL test_StdRootForced(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, const char *host, const char *sharename)
+{
+	NTSTATUS status;
+	struct dfs_AddStdRootForced r;
+	enum dfs_ManagerVersion version;
+
+	printf("Testing StdRootForced\n");
+
+	if (!test_GetManagerVersion(p, mem_ctx, &version)) {
+		return False;
+	}
+
+	r.in.servername	= host;
+	r.in.rootshare	= sharename;
+	r.in.comment	= "standard dfs forced standalone DFS root created by smbtorture (dfs_AddStdRootForced)";
+	r.in.store	= SMBTORTURE_DFS_PATHNAME;
+
+	status = dcerpc_dfs_AddStdRootForced(p, mem_ctx, &r);
+	if (!NT_STATUS_IS_OK(status)) {
+		printf("AddStdRootForced failed - %s\n", nt_errstr(status));
+		return False;
+	} else if (!W_ERROR_IS_OK(r.out.result)) {
+		printf("AddStdRootForced failed - %s\n", win_errstr(r.out.result));
+		IS_DFS_VERSION_UNSUPPORTED_CALL_W2K3(version, r.out.result);
+		return False;
+	}
+
+	return True;
+}
+
+static BOOL test_RemoveStdRoot(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, const char *host, const char *sharename)
+{
+	struct dfs_RemoveStdRoot r;
+	NTSTATUS status;
+
+	printf("Testing RemoveStdRoot\n");
+
+	r.in.servername	= host;
+	r.in.rootshare	= sharename;
+	r.in.flags	= 0;
+
+	status = dcerpc_dfs_RemoveStdRoot(p, mem_ctx, &r);
+	if (!NT_STATUS_IS_OK(status)) {
+		printf("RemoveStdRoot failed - %s\n", nt_errstr(status));
+		return False;
+	} else if (!W_ERROR_IS_OK(r.out.result)) {
+		printf("dfs_RemoveStdRoot failed - %s\n", win_errstr(r.out.result));
+		return False;
+	}
+
+	return True;
+}
+
+static BOOL test_AddStdRoot(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, const char *host, const char *sharename)
+{
+	NTSTATUS status;
+	struct dfs_AddStdRoot r;
+
+	printf("Testing AddStdRoot\n");
+
+	r.in.servername	= host;
+	r.in.rootshare	= sharename;
+	r.in.comment	= "standard dfs standalone DFS root created by smbtorture (dfs_AddStdRoot)";
+	r.in.flags	= 0;
+
+	status = dcerpc_dfs_AddStdRoot(p, mem_ctx, &r);
+	if (!NT_STATUS_IS_OK(status)) {
+		printf("AddStdRoot failed - %s\n", nt_errstr(status));
+		return False;
+	} else if (!W_ERROR_IS_OK(r.out.result)) {
+		printf("dfs_AddStdRoot failed - %s\n", win_errstr(r.out.result));
+		return False;
+	}
+
+	return True;
+}
+
+static void test_cleanup_stdroot(struct dcerpc_pipe *p, 
+				 TALLOC_CTX *mem_ctx, 
+				 const char *host, 
+				 const char *sharename, 
+				 const char *dir)
+{
+	struct smbcli_state *cli;
+
+	printf("Cleaning up StdRoot\n");
+
+	test_RemoveStdRoot(p, mem_ctx, host, sharename);
+	test_NetShareDel(mem_ctx, host, sharename);
+	torture_open_connection_share(mem_ctx, &cli, host, "C$", NULL);
+	test_DeleteDir(cli, dir);
+	torture_close_connection(cli);
+}
+
+static BOOL test_StdRoot(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, const char *host)
+{
+	enum dfs_ManagerVersion version;
+	const char *sharename = SMBTORTURE_DFS_SHARENAME;
+	const char *dir = SMBTORTURE_DFS_DIRNAME;
+	const char *path = SMBTORTURE_DFS_PATHNAME;
+	struct smbcli_state *cli;
+	BOOL ret;
+
+	printf("Testing StdRoot\n");
+
+	test_cleanup_stdroot(p, mem_ctx, host, sharename, dir);
+
+	ret &= test_CreateDir(mem_ctx, &cli, host, "C$", dir);
+	ret &= test_NetShareAdd(mem_ctx, host, sharename, path);
+	ret &= test_GetManagerVersion(p, mem_ctx, &version);
+	ret &= test_AddStdRoot(p, mem_ctx, host, sharename);
+	ret &= test_RemoveStdRoot(p, mem_ctx, host, sharename);
+	ret &= test_StdRootForced(p, mem_ctx, host, sharename);
+	ret &= test_NetShareDel(mem_ctx, host, sharename);
+	ret &= test_DeleteDir(cli, dir);
+
+	torture_close_connection(cli);
+
+	return ret;
+}
+
 BOOL torture_rpc_dfs(struct torture_context *torture)
 {
 	NTSTATUS status;
@@ -239,6 +475,7 @@
 	ret &= test_GetManagerVersion(p, mem_ctx, &version);
 	ret &= test_Enum(p, mem_ctx);
 	ret &= test_EnumEx(p, mem_ctx, host);
+	ret &= test_StdRoot(p, mem_ctx, host);
 
 	talloc_free(mem_ctx);
 



More information about the samba-cvs mailing list