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

[abartlet at samba.org]

Author: abartlet
Date: 2005-01-11 03:49:29 +0000 (Tue, 11 Jan 2005)
New Revision: 4671

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

Log:
Expand the RPC-LSA test to set secret values twice.  

Andrew Bartlett

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


Changeset:
Modified: branches/SAMBA_4_0/source/torture/rpc/lsa.c
===================================================================
--- branches/SAMBA_4_0/source/torture/rpc/lsa.c	2005-01-11 03:26:26 UTC (rev 4670)
+++ branches/SAMBA_4_0/source/torture/rpc/lsa.c	2005-01-11 03:49:29 UTC (rev 4671)
@@ -678,6 +678,8 @@
 	struct lsa_OpenSecret r2;
 	struct lsa_SetSecret r3;
 	struct lsa_QuerySecret r4;
+	struct lsa_SetSecret r5;
+	struct lsa_QuerySecret r6;
 	struct policy_handle sec_handle, sec_handle2;
 	struct lsa_Delete d;
 	struct lsa_DATA_BUF buf1;
@@ -689,6 +691,8 @@
 	DATA_BLOB blob1, blob2;
 	const char *secret1 = "abcdef12345699qwerty";
 	char *secret2;
+ 	const char *secret3 = "ABCDEF12345699QWERTY";
+	char *secret4;
 	char *secname[2];
 	int i;
 
@@ -786,6 +790,62 @@
 			}
 		}
 		
+		enc_key = sess_encrypt_string(secret3, &session_key);
+		
+		r5.in.handle = &sec_handle;
+		r5.in.new_val = &buf1;
+		r5.in.old_val = NULL;
+		r5.in.new_val->data = enc_key.data;
+		r5.in.new_val->length = enc_key.length;
+		r5.in.new_val->size = enc_key.length;
+		
+		printf("Testing SetSecret\n");
+		
+		status = dcerpc_lsa_SetSecret(p, mem_ctx, &r5);
+		if (!NT_STATUS_IS_OK(status)) {
+			printf("SetSecret failed - %s\n", nt_errstr(status));
+			ret = False;
+		}
+		
+		data_blob_free(&enc_key);
+		
+		ZERO_STRUCT(new_mtime);
+		ZERO_STRUCT(old_mtime);
+		
+		/* fetch the secret back again */
+		r6.in.handle = &sec_handle;
+		r6.in.new_val = &bufp1;
+		r6.in.new_mtime = &new_mtime;
+		r6.in.old_val = NULL;
+		r6.in.old_mtime = NULL;
+		
+		bufp1.buf = NULL;
+		
+		status = dcerpc_lsa_QuerySecret(p, mem_ctx, &r6);
+		if (!NT_STATUS_IS_OK(status)) {
+			printf("QuerySecret failed - %s\n", nt_errstr(status));
+			ret = False;
+		}
+		
+		if (r6.out.new_val->buf == NULL) {
+			printf("No secret buffer returned\n");
+			ret = False;
+		} else {
+			blob1.data = r6.out.new_val->buf->data;
+			blob1.length = r6.out.new_val->buf->length;
+			
+			blob2 = data_blob(NULL, blob1.length);
+			
+			secret4 = sess_decrypt_string(&blob1, &session_key);
+			
+			printf("returned secret '%s'\n", secret3);
+			
+			if (strcmp(secret3, secret4) != 0) {
+				printf("Returned secret %s doesn't match %s\n", secret4, secret3);
+				ret = False;
+			}
+		}
+		
 		if (!test_Delete(p, mem_ctx, &sec_handle)) {
 			ret = False;
 		}