svn commit: samba r10166 - in branches/tmp/samba4-winsrepl: . source/torture/nbt

[metze at samba.org]

Author: metze
Date: 2005-09-12 10:40:36 +0000 (Mon, 12 Sep 2005)
New Revision: 10166

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

Log:
 r12169 at SERNOX (orig r10164):  metze | 2005-09-12 12:06:22 +0200
 - add first assoc_ctx test
 - handle the case where we're no valid pull partner of the tested server
 
 metze

Modified:
   branches/tmp/samba4-winsrepl/
   branches/tmp/samba4-winsrepl/source/torture/nbt/winsreplication.c


Changeset:

Property changes on: branches/tmp/samba4-winsrepl
___________________________________________________________________
Name: svk:merge
   - 0c0555d6-39d7-0310-84fc-f1cc0bd64818:/branches/SAMBA_4_0:10161
3a72dc49-98ff-0310-ab52-9b7ed7945d91:/local/samba4:9495
a953eb74-4aff-0310-a63c-855d20285ebb:/local/samba4:11627
   + 0c0555d6-39d7-0310-84fc-f1cc0bd64818:/branches/SAMBA_4_0:10164
3a72dc49-98ff-0310-ab52-9b7ed7945d91:/local/samba4:9495
a953eb74-4aff-0310-a63c-855d20285ebb:/local/samba4:11627

Modified: branches/tmp/samba4-winsrepl/source/torture/nbt/winsreplication.c
===================================================================
--- branches/tmp/samba4-winsrepl/source/torture/nbt/winsreplication.c	2005-09-12 10:23:14 UTC (rev 10165)
+++ branches/tmp/samba4-winsrepl/source/torture/nbt/winsreplication.c	2005-09-12 10:40:36 UTC (rev 10166)
@@ -41,6 +41,61 @@
 	}} while (0)
 
 /*
+  test how assoc_ctx's are only usable on the connection
+  they are created on.
+*/
+static BOOL test_assoc_ctx1(TALLOC_CTX *mem_ctx, const char *address)
+{
+	BOOL ret = True;
+	struct wrepl_request *req;
+	struct wrepl_socket *wrepl_socket1;
+	struct wrepl_associate associate1;
+	struct wrepl_socket *wrepl_socket2;
+	struct wrepl_associate associate2;
+	struct wrepl_pull_table pull_table;
+	NTSTATUS status;
+
+	printf("Test if assoc_ctx is only valid on the conection it was created on\n");
+
+	wrepl_socket1 = wrepl_socket_init(mem_ctx, NULL);
+	wrepl_socket2 = wrepl_socket_init(mem_ctx, NULL);
+	
+	printf("Setup 2 wrepl connections\n");
+	status = wrepl_connect(wrepl_socket1, address);
+	CHECK_STATUS(status, NT_STATUS_OK);
+
+	status = wrepl_connect(wrepl_socket2, address);
+	CHECK_STATUS(status, NT_STATUS_OK);
+
+	printf("Send a start association request (conn1)\n");
+	status = wrepl_associate(wrepl_socket1, &associate1);
+	CHECK_STATUS(status, NT_STATUS_OK);
+
+	printf("association context (conn1): 0x%x\n", associate1.out.assoc_ctx);
+
+	printf("Send a start association request (conn2)\n");
+	status = wrepl_associate(wrepl_socket2, &associate2);
+	CHECK_STATUS(status, NT_STATUS_OK);
+
+	printf("association context (conn2): 0x%x\n", associate2.out.assoc_ctx);
+
+	printf("Send a replication table query, with assoc 1 (conn2), should be ignored\n");
+	pull_table.in.assoc_ctx = associate1.out.assoc_ctx;
+	req = wrepl_pull_table_send(wrepl_socket2, &pull_table);
+	talloc_free(req);
+
+	printf("Send a association request (conn2), to make sure the last request was ignored\n");
+	status = wrepl_associate(wrepl_socket2, &associate2);
+	CHECK_STATUS(status, NT_STATUS_OK);
+
+done:
+	printf("Close 2 wrepl connections\n");
+	talloc_free(wrepl_socket1);
+	talloc_free(wrepl_socket2);
+	return ret;
+}
+
+/*
   display a replication entry
 */
 static void display_entry(TALLOC_CTX *mem_ctx, struct wrepl_name *name)
@@ -57,7 +112,7 @@
 /*
   test a full replication dump from a WINS server
 */
-static BOOL nbt_test_wins_replication(TALLOC_CTX *mem_ctx, const char *address)
+static BOOL test_wins_replication(TALLOC_CTX *mem_ctx, const char *address)
 {
 	BOOL ret = True;
 	struct wrepl_socket *wrepl_socket;
@@ -67,8 +122,11 @@
 	struct wrepl_pull_table pull_table;
 	struct wrepl_pull_names pull_names;
 
+	printf("Test one pull replication cycle\n");
+
 	wrepl_socket = wrepl_socket_init(mem_ctx, NULL);
 	
+	printf("Setup wrepl connections\n");
 	status = wrepl_connect(wrepl_socket, address);
 	CHECK_STATUS(status, NT_STATUS_OK);
 
@@ -83,6 +141,23 @@
 	pull_table.in.assoc_ctx = associate.out.assoc_ctx;
 
 	status = wrepl_pull_table(wrepl_socket, mem_ctx, &pull_table);
+	if (NT_STATUS_EQUAL(NT_STATUS_NETWORK_ACCESS_DENIED,status)) {
+		struct wrepl_packet packet;
+		struct wrepl_request *req;
+
+		printf("We are not a valid pull partner for the server\n");
+
+		ZERO_STRUCT(packet);
+		packet.opcode                      = WREPL_OPCODE_BITS;
+		packet.assoc_ctx                   = associate.out.assoc_ctx;
+		packet.mess_type                   = WREPL_STOP_ASSOCIATION;
+		packet.message.stop.reason         = 0;
+
+		req = wrepl_request_send(wrepl_socket, &packet);
+		talloc_free(req);
+		ret = False;
+		goto done;
+	}
 	CHECK_STATUS(status, NT_STATUS_OK);
 
 	printf("Found %d replication partners\n", pull_table.out.num_partners);
@@ -109,6 +184,7 @@
 	}
 
 done:
+	printf("Close wrepl connections\n");
 	talloc_free(wrepl_socket);
 	return ret;
 }
@@ -135,8 +211,10 @@
 		return False;
 	}
 
-	ret &= nbt_test_wins_replication(mem_ctx, address);
+	ret &= test_assoc_ctx1(mem_ctx, address);
 
+	ret &= test_wins_replication(mem_ctx, address);
+
 	talloc_free(mem_ctx);
 
 	return ret;