svn commit: samba r6637 - in branches/SAMBA_4_0/source/torture/libnet: .

[mimir at samba.org]

Author: mimir
Date: 2005-05-06 19:36:49 +0000 (Fri, 06 May 2005)
New Revision: 6637

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

Log:
Test for asynchronous function and monitor messages.


rafal


Modified:
   branches/SAMBA_4_0/source/torture/libnet/userinfo.c


Changeset:
Modified: branches/SAMBA_4_0/source/torture/libnet/userinfo.c
===================================================================
--- branches/SAMBA_4_0/source/torture/libnet/userinfo.c	2005-05-06 18:35:41 UTC (rev 6636)
+++ branches/SAMBA_4_0/source/torture/libnet/userinfo.c	2005-05-06 19:36:49 UTC (rev 6637)
@@ -22,6 +22,7 @@
 #include "includes.h"
 #include "librpc/gen_ndr/ndr_samr.h"
 #include "libnet/composite.h"
+#include "libcli/composite/monitor.h"
 
 #define TEST_USERNAME  "libnetuserinfotest"
 
@@ -185,9 +186,9 @@
 	NTSTATUS status;
 	struct rpc_composite_userinfo user;
 	struct dom_sid *user_sid;
-
+	
 	user_sid = dom_sid_add_rid(mem_ctx, domain_sid, *rid);
-
+	
 	user.in.domain_handle = *domain_handle;
 	user.in.sid           = dom_sid_string(mem_ctx, user_sid);
 	user.in.level         = 5;       /* this should be extended */
@@ -203,6 +204,57 @@
 }
 
 
+void msg_handler(struct monitor_msg *m)
+{
+	switch (m->type) {
+	case rpc_open_user:
+		printf("monitor_msg: user opened (rid=%d, access_mask=0x%08x)\n",
+		       m->data.rpc_open_user.rid, m->data.rpc_open_user.access_mask);
+		break;
+	case rpc_query_user:
+		printf("monitor_msg: user queried (level=%d)\n", m->data.rpc_query_user.level);
+		break;
+	case rpc_close_user:
+		printf("monitor_msg: user closed (rid=%d)\n", m->data.rpc_close_user.rid);
+		break;
+	}
+}
+
+
+static BOOL test_userinfo_async(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
+				struct policy_handle *domain_handle,
+				struct dom_sid2 *domain_sid, const char* user_name,
+				uint32_t *rid)
+{
+	NTSTATUS status;
+	struct composite_context *c;
+	struct rpc_composite_userinfo user;
+	struct dom_sid *user_sid;
+
+	user_sid = dom_sid_add_rid(mem_ctx, domain_sid, *rid);
+
+	user.in.domain_handle = *domain_handle;
+	user.in.sid           = dom_sid_string(mem_ctx, user_sid);
+	user.in.level         = 10;       /* this should be extended */
+
+	printf("Testing async rpc_composite_userinfo\n");
+
+	c = rpc_composite_userinfo_send(p, &user, msg_handler);
+	if (!c) {
+		printf("Failed to call sync rpc_composite_userinfo_send\n");
+		return False;
+	}
+
+	status = rpc_composite_userinfo_recv(c, mem_ctx, &user);
+	if (!NT_STATUS_IS_OK(status)) {
+		printf("Calling async rpc_composite_userinfo failed - %s\n", nt_errstr(status));
+		return False;
+	}
+
+	return True;
+}
+
+
 BOOL torture_userinfo(void)
 {
 	NTSTATUS status;
@@ -231,6 +283,9 @@
 
 	name.string = lp_workgroup();
 
+	/*
+	 * Testing synchronous version
+	 */
 	if (!test_opendomain(p, mem_ctx, &h, &name, &sid)) {
 		ret = False;
 		goto done;
@@ -250,6 +305,30 @@
 		ret = False;
 		goto done;
 	}
+
+	/*
+	 * Testing asynchronous version and monitor messages
+	 */
+	if (!test_opendomain(p, mem_ctx, &h, &name, &sid)) {
+		ret = False;
+		goto done;
+	}
+
+	if (!test_create(p, mem_ctx, &h, TEST_USERNAME, &rid)) {
+		ret = False;
+		goto done;
+	}
+
+	if (!test_userinfo_async(p, mem_ctx, &h, &sid, TEST_USERNAME, &rid)) {
+		ret = False;
+		goto done;
+	}
+
+	if (!test_cleanup(p, mem_ctx, &h, TEST_USERNAME)) {
+		ret = False;
+		goto done;
+	}
+
 done:
 	talloc_free(mem_ctx);