X-Git-Url: http://git.samba.org/?a=blobdiff_plain;ds=sidebyside;f=source%2Ftorture%2Frpc%2Falter_context.c;h=ad00a349137841aed95b8c359bdab1b8860874ca;hb=7b6b8d6b34de01d2585b23ddb994d8e7403a21f6;hp=79135ba01837d021e9372af99111cae3172a7f40;hpb=ec32b22ed5ec224f6324f5e069d15e92e38e15c0;p=samba.git

diff --git a/source/torture/rpc/alter_context.c b/source/torture/rpc/alter_context.c
index 79135ba0183..ad00a349137 100644
--- a/source/torture/rpc/alter_context.c
+++ b/source/torture/rpc/alter_context.c
@@ -38,11 +38,12 @@ BOOL torture_rpc_alter_context(void)
 	mem_ctx = talloc_init("torture_rpc_alter_context");
 
 	printf("opening LSA connection\n");
-	status = torture_rpc_connection(&p, 
+	status = torture_rpc_connection(mem_ctx, &p, 
 					DCERPC_LSARPC_NAME, 
 					DCERPC_LSARPC_UUID, 
 					DCERPC_LSARPC_VERSION);
 	if (!NT_STATUS_IS_OK(status)) {
+		talloc_free(mem_ctx);
 		return False;
 	}
 
@@ -53,6 +54,7 @@ BOOL torture_rpc_alter_context(void)
 	printf("Opening secondary DSSETUP context\n");
 	status = dcerpc_secondary_context(p, &p2, DCERPC_DSSETUP_UUID, DCERPC_DSSETUP_VERSION);
 	if (!NT_STATUS_IS_OK(status)) {
+		talloc_free(mem_ctx);
 		printf("dcerpc_alter_context failed - %s\n", nt_errstr(status));
 		return False;
 	}
@@ -60,6 +62,7 @@ BOOL torture_rpc_alter_context(void)
 	printf("Opening bad secondary connection\n");
 	status = dcerpc_secondary_context(p, &p2, DCERPC_DSSETUP_UUID, DCERPC_DSSETUP_VERSION+100);
 	if (NT_STATUS_IS_OK(status)) {
+		talloc_free(mem_ctx);
 		printf("dcerpc_alter_context with wrong version should fail\n");
 		return False;
 	}
@@ -77,6 +80,7 @@ BOOL torture_rpc_alter_context(void)
 	printf("Testing change of primary context\n");
 	status = dcerpc_alter_context(p, mem_ctx, &p2->syntax, &p2->transfer_syntax);
 	if (!NT_STATUS_IS_OK(status)) {
+		talloc_free(mem_ctx);
 		printf("dcerpc_alter_context failed - %s\n", nt_errstr(status));
 		return False;
 	}
@@ -99,7 +103,5 @@ BOOL torture_rpc_alter_context(void)
 
 	talloc_free(mem_ctx);
 
-        torture_rpc_close(p);
-
 	return ret;
 }