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

[vlendec at samba.org]

Author: vlendec
Date: 2005-11-21 21:13:01 +0000 (Mon, 21 Nov 2005)
New Revision: 11837

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

Log:
Add a timeout before starting a torture test. It's really fun to watch 50
simulated logins doing stuff simultaneously. This is where the async
architecture *really* pays off.

Volker

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


Changeset:
Modified: branches/SAMBA_4_0/source/torture/rpc/xplogin.c
===================================================================
--- branches/SAMBA_4_0/source/torture/rpc/xplogin.c	2005-11-21 20:42:18 UTC (rev 11836)
+++ branches/SAMBA_4_0/source/torture/rpc/xplogin.c	2005-11-21 21:13:01 UTC (rev 11837)
@@ -1373,6 +1373,8 @@
 }
 
 struct xp_login_state {
+	struct timeval timeout;
+
 	const char *dc_name;
 	const char *dc_ip;
 	const char *wks_domain;
@@ -1404,6 +1406,8 @@
 };
 
 static void xp_login_recv_conn(struct composite_context *ctx);
+static void xp_login_start(struct event_context *ev, struct timed_event *te,
+			   struct timeval tv, void *p);
 static void xp_login_recv_auth2(struct composite_context *ctx);
 static void xp_login_recv_trusts(struct composite_context *creq);
 static void xp_login_recv_schannel(struct composite_context *creq);
@@ -1413,6 +1417,7 @@
 static void xp_login_recv_memberships(struct composite_context *creq);
 
 static struct composite_context *xp_login_send(TALLOC_CTX *mem_ctx,
+					       struct timeval timeout,
 					       struct event_context *event_ctx,
 					       const char *dc_name,
 					       const char *dc_ip,
@@ -1423,7 +1428,7 @@
 					       const char *user_name,
 					       const char *user_pwd)
 {
-	struct composite_context *c, *creq;
+	struct composite_context *c;
 	struct xp_login_state *state;
 
 	c = talloc_zero(mem_ctx, struct composite_context);
@@ -1439,6 +1444,7 @@
 	c->private_data = state;
 	c->event_ctx = event_ctx;
 
+	state->timeout = timeout;
 	state->dc_name = dc_name;
 	state->dc_ip = dc_ip;
 	state->wks_domain = wks_domain;
@@ -1474,8 +1480,10 @@
 	state->conn.in.fallback_to_anonymous = False;
 	state->conn.in.workgroup = wks_domain;
 
-	creq = smb_composite_connect_send(&state->conn, state, event_ctx);
-	composite_continue(c, creq, xp_login_recv_conn, c);
+	event_add_timed(c->event_ctx, state,
+			timeval_current_ofs(state->timeout.tv_sec,
+					    state->timeout.tv_usec),
+			xp_login_start, c);
 	return c;
 
  failed:
@@ -1483,6 +1491,19 @@
 	return c;
 }
 
+static void xp_login_start(struct event_context *ev, struct timed_event *te,
+			   struct timeval tv, void *p)
+{
+	struct composite_context *c =
+		talloc_get_type(p, struct composite_context);
+	struct xp_login_state *state =
+		talloc_get_type(c->private_data, struct xp_login_state);
+	struct composite_context *creq;
+
+	creq = smb_composite_connect_send(&state->conn, state, c->event_ctx);
+	composite_continue(c, creq, xp_login_recv_conn, c);
+}
+
 static void xp_login_recv_conn(struct composite_context *creq)
 {
 	struct composite_context *c =
@@ -1768,12 +1789,15 @@
 	}
 
 	for (i=0; i<torture_numops; i++) {
-		ctx[i] = xp_login_send(mem_ctx, event_ctx,
-				       lp_parm_string(-1, "torture", "host"),
-				       lp_parm_string(-1, "torture", "host"),
-				       lp_workgroup(),
-				       lp_netbios_name(), "5,eEp_D2",
-				       lp_workgroup(), "vl", "asdf");
+		ctx[i] = xp_login_send(
+			mem_ctx, timeval_set(0, i*lp_parm_int(-1, "torture",
+							      "timeout", 0)),
+			event_ctx,
+			lp_parm_string(-1, "torture", "host"),
+			lp_parm_string(-1, "torture", "host"),
+			lp_workgroup(),
+			lp_netbios_name(), "5,eEp_D2",
+			lp_workgroup(), "vl", "asdf");
 		if (ctx[i] == NULL) {
 			DEBUG(0, ("xp_login_send failed\n"));
 			goto done;