r6961: Add RPC-WINREG to list of win2k3 tests

[samba.git] / source / torture / rpc / alter_context.c

diff --git a/source/torture/rpc/alter_context.c b/source/torture/rpc/alter_context.c
index 79135ba01837d021e9372af99111cae3172a7f40..ad00a349137841aed95b8c359bdab1b8860874ca 100644 (file)
--- 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;
 }