fix LSA-PRIVILEGES
[ira/wip.git] / source4 / torture / rpc / lsa.c
index 34e3c1b78d26c0b8ad5c79def335571c39faec2e..b8b9ced660ab618246e964022597a83d075b7f3f 100644 (file)
@@ -1,20 +1,20 @@
-/* 
+/*
    Unix SMB/CIFS implementation.
    test suite for lsa rpc operations
 
    Copyright (C) Andrew Tridgell 2003
    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2004-2005
-   
+
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.
-   
+
    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.
-   
+
    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
@@ -37,7 +37,8 @@ static void init_lsa_String(struct lsa_String *name, const char *s)
        name->string = s;
 }
 
-static bool test_OpenPolicy(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
+static bool test_OpenPolicy(struct dcerpc_pipe *p,
+                           struct torture_context *tctx)
 {
        struct lsa_ObjectAttribute attr;
        struct policy_handle handle;
@@ -65,7 +66,7 @@ static bool test_OpenPolicy(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
        r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
        r.out.handle = &handle;
 
-       status = dcerpc_lsa_OpenPolicy(p, mem_ctx, &r);
+       status = dcerpc_lsa_OpenPolicy(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
                if (NT_STATUS_EQUAL(status, NT_STATUS_ACCESS_DENIED) ||
                    NT_STATUS_EQUAL(status, NT_STATUS_RPC_PROTSEQ_NOT_SUPPORTED)) {
@@ -80,7 +81,8 @@ static bool test_OpenPolicy(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
 }
 
 
-bool test_lsa_OpenPolicy2(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx, 
+bool test_lsa_OpenPolicy2(struct dcerpc_pipe *p,
+                         struct torture_context *tctx,
                          struct policy_handle **handle)
 {
        struct lsa_ObjectAttribute attr;
@@ -90,7 +92,7 @@ bool test_lsa_OpenPolicy2(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
 
        printf("\nTesting OpenPolicy2\n");
 
-       *handle = talloc(mem_ctx, struct policy_handle);
+       *handle = talloc(tctx, struct policy_handle);
        if (!*handle) {
                return false;
        }
@@ -112,7 +114,7 @@ bool test_lsa_OpenPolicy2(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
        r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
        r.out.handle = *handle;
 
-       status = dcerpc_lsa_OpenPolicy2(p, mem_ctx, &r);
+       status = dcerpc_lsa_OpenPolicy2(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
                if (NT_STATUS_EQUAL(status, NT_STATUS_ACCESS_DENIED) ||
                    NT_STATUS_EQUAL(status, NT_STATUS_RPC_PROTSEQ_NOT_SUPPORTED)) {
@@ -146,13 +148,14 @@ static const char *sid_type_lookup(enum lsa_SidType r)
        return "Invalid sid type\n";
 }
 
-static bool test_LookupNames(struct dcerpc_pipe *p, 
-                            TALLOC_CTX *mem_ctx, 
+static bool test_LookupNames(struct dcerpc_pipe *p,
+                            struct torture_context *tctx,
                             struct policy_handle *handle,
                             struct lsa_TransNameArray *tnames)
 {
        struct lsa_LookupNames r;
        struct lsa_TransSidArray sids;
+       struct lsa_RefDomainList *domains = NULL;
        struct lsa_String *names;
        uint32_t count = 0;
        NTSTATUS status;
@@ -163,7 +166,7 @@ static bool test_LookupNames(struct dcerpc_pipe *p,
        sids.count = 0;
        sids.sids = NULL;
 
-       names = talloc_array(mem_ctx, struct lsa_String, tnames->count);
+       names = talloc_array(tctx, struct lsa_String, tnames->count);
        for (i=0;i<tnames->count;i++) {
                init_lsa_String(&names[i], tnames->names[i].name.string);
        }
@@ -176,15 +179,16 @@ static bool test_LookupNames(struct dcerpc_pipe *p,
        r.in.count = &count;
        r.out.count = &count;
        r.out.sids = &sids;
+       r.out.domains = &domains;
 
-       status = dcerpc_lsa_LookupNames(p, mem_ctx, &r);
+       status = dcerpc_lsa_LookupNames(p, tctx, &r);
 
-       if (NT_STATUS_EQUAL(status, STATUS_SOME_UNMAPPED) || 
+       if (NT_STATUS_EQUAL(status, STATUS_SOME_UNMAPPED) ||
            NT_STATUS_EQUAL(status, NT_STATUS_NONE_MAPPED)) {
                for (i=0;i< tnames->count;i++) {
                        if (i < count && sids.sids[i].sid_type == SID_NAME_UNKNOWN) {
-                               printf("LookupName of %s was unmapped\n", 
-                                      tnames->names[i].name.string);   
+                               printf("LookupName of %s was unmapped\n",
+                                      tnames->names[i].name.string);
                        } else if (i >=count) {
                                printf("LookupName of %s failed to return a result\n",
                                       tnames->names[i].name.string);
@@ -196,10 +200,10 @@ static bool test_LookupNames(struct dcerpc_pipe *p,
                printf("LookupNames failed - %s\n", nt_errstr(status));
                return false;
        }
-       
+
        for (i=0;i< tnames->count;i++) {
                if (i < count && sids.sids[i].sid_type != tnames->names[i].sid_type) {
-                       printf("LookupName of %s got unexpected name type: %s\n", 
+                       printf("LookupName of %s got unexpected name type: %s\n",
                               tnames->names[i].name.string, sid_type_lookup(sids.sids[i].sid_type));
                } else if (i >=count) {
                        printf("LookupName of %s failed to return a result\n",
@@ -211,12 +215,13 @@ static bool test_LookupNames(struct dcerpc_pipe *p,
        return true;
 }
 
-static bool test_LookupNames_bogus(struct dcerpc_pipe *p, 
-                           TALLOC_CTX *mem_ctx, 
-                           struct policy_handle *handle)
+static bool test_LookupNames_bogus(struct dcerpc_pipe *p,
+                                  struct torture_context *tctx,
+                                  struct policy_handle *handle)
 {
        struct lsa_LookupNames r;
        struct lsa_TransSidArray sids;
+       struct lsa_RefDomainList *domains = NULL;
        struct lsa_String *names;
        uint32_t count = 0;
        NTSTATUS status;
@@ -235,7 +240,7 @@ static bool test_LookupNames_bogus(struct dcerpc_pipe *p,
        sids.count = 0;
        sids.sids = NULL;
 
-       names = talloc_array(mem_ctx, struct lsa_String, tnames.count);
+       names = talloc_array(tctx, struct lsa_String, tnames.count);
        for (i=0;i<tnames.count;i++) {
                init_lsa_String(&names[i], tnames.names[i].name.string);
        }
@@ -248,8 +253,9 @@ static bool test_LookupNames_bogus(struct dcerpc_pipe *p,
        r.in.count = &count;
        r.out.count = &count;
        r.out.sids = &sids;
+       r.out.domains = &domains;
 
-       status = dcerpc_lsa_LookupNames(p, mem_ctx, &r);
+       status = dcerpc_lsa_LookupNames(p, tctx, &r);
        if (!NT_STATUS_EQUAL(status, NT_STATUS_NONE_MAPPED)) {
                printf("LookupNames failed - %s\n", nt_errstr(status));
                return false;
@@ -260,8 +266,8 @@ static bool test_LookupNames_bogus(struct dcerpc_pipe *p,
        return true;
 }
 
-static bool test_LookupNames_wellknown(struct dcerpc_pipe *p, 
-                                      TALLOC_CTX *mem_ctx, 
+static bool test_LookupNames_wellknown(struct dcerpc_pipe *p,
+                                      struct torture_context *tctx,
                                       struct policy_handle *handle)
 {
        struct lsa_TranslatedName name;
@@ -274,49 +280,51 @@ static bool test_LookupNames_wellknown(struct dcerpc_pipe *p,
        tnames.count = 1;
        name.name.string = "NT AUTHORITY\\SYSTEM";
        name.sid_type = SID_NAME_WKN_GRP;
-       ret &= test_LookupNames(p, mem_ctx, handle, &tnames);
+       ret &= test_LookupNames(p, tctx, handle, &tnames);
 
        name.name.string = "NT AUTHORITY\\ANONYMOUS LOGON";
        name.sid_type = SID_NAME_WKN_GRP;
-       ret &= test_LookupNames(p, mem_ctx, handle, &tnames);
+       ret &= test_LookupNames(p, tctx, handle, &tnames);
 
        name.name.string = "NT AUTHORITY\\Authenticated Users";
        name.sid_type = SID_NAME_WKN_GRP;
-       ret &= test_LookupNames(p, mem_ctx, handle, &tnames);
+       ret &= test_LookupNames(p, tctx, handle, &tnames);
 
 #if 0
        name.name.string = "NT AUTHORITY";
-       ret &= test_LookupNames(p, mem_ctx, handle, &tnames);
+       ret &= test_LookupNames(p, tctx, handle, &tnames);
 
        name.name.string = "NT AUTHORITY\\";
-       ret &= test_LookupNames(p, mem_ctx, handle, &tnames);
+       ret &= test_LookupNames(p, tctx, handle, &tnames);
 #endif
 
        name.name.string = "BUILTIN\\";
        name.sid_type = SID_NAME_DOMAIN;
-       ret &= test_LookupNames(p, mem_ctx, handle, &tnames);
+       ret &= test_LookupNames(p, tctx, handle, &tnames);
 
        name.name.string = "BUILTIN\\Administrators";
        name.sid_type = SID_NAME_ALIAS;
-       ret &= test_LookupNames(p, mem_ctx, handle, &tnames);
+       ret &= test_LookupNames(p, tctx, handle, &tnames);
 
        name.name.string = "SYSTEM";
        name.sid_type = SID_NAME_WKN_GRP;
-       ret &= test_LookupNames(p, mem_ctx, handle, &tnames);
+       ret &= test_LookupNames(p, tctx, handle, &tnames);
 
        name.name.string = "Everyone";
        name.sid_type = SID_NAME_WKN_GRP;
-       ret &= test_LookupNames(p, mem_ctx, handle, &tnames);
+       ret &= test_LookupNames(p, tctx, handle, &tnames);
        return ret;
 }
 
-static bool test_LookupNames2(struct dcerpc_pipe *p, 
-                             TALLOC_CTX *mem_ctx, 
+static bool test_LookupNames2(struct dcerpc_pipe *p,
+                             struct torture_context *tctx,
                              struct policy_handle *handle,
-                             struct lsa_TransNameArray2 *tnames)
+                             struct lsa_TransNameArray2 *tnames,
+                             bool check_result)
 {
        struct lsa_LookupNames2 r;
        struct lsa_TransSidArray2 sids;
+       struct lsa_RefDomainList *domains = NULL;
        struct lsa_String *names;
        uint32_t count = 0;
        NTSTATUS status;
@@ -327,7 +335,7 @@ static bool test_LookupNames2(struct dcerpc_pipe *p,
        sids.count = 0;
        sids.sids = NULL;
 
-       names = talloc_array(mem_ctx, struct lsa_String, tnames->count);
+       names = talloc_array(tctx, struct lsa_String, tnames->count);
        for (i=0;i<tnames->count;i++) {
                init_lsa_String(&names[i], tnames->names[i].name.string);
        }
@@ -342,26 +350,37 @@ static bool test_LookupNames2(struct dcerpc_pipe *p,
        r.in.client_revision = 0;
        r.out.count = &count;
        r.out.sids = &sids;
+       r.out.domains = &domains;
 
-       status = dcerpc_lsa_LookupNames2(p, mem_ctx, &r);
+       status = dcerpc_lsa_LookupNames2(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
                printf("LookupNames2 failed - %s\n", nt_errstr(status));
                return false;
        }
 
+       if (check_result) {
+               torture_assert_int_equal(tctx, count, sids.count,
+                       "unexpected number of results returned");
+               if (sids.count > 0) {
+                       torture_assert(tctx, sids.sids, "invalid sid buffer");
+               }
+       }
+
        printf("\n");
 
        return true;
 }
 
 
-static bool test_LookupNames3(struct dcerpc_pipe *p, 
-                             TALLOC_CTX *mem_ctx, 
+static bool test_LookupNames3(struct dcerpc_pipe *p,
+                             struct torture_context *tctx,
                              struct policy_handle *handle,
-                             struct lsa_TransNameArray2 *tnames)
+                             struct lsa_TransNameArray2 *tnames,
+                             bool check_result)
 {
        struct lsa_LookupNames3 r;
        struct lsa_TransSidArray3 sids;
+       struct lsa_RefDomainList *domains = NULL;
        struct lsa_String *names;
        uint32_t count = 0;
        NTSTATUS status;
@@ -372,7 +391,7 @@ static bool test_LookupNames3(struct dcerpc_pipe *p,
        sids.count = 0;
        sids.sids = NULL;
 
-       names = talloc_array(mem_ctx, struct lsa_String, tnames->count);
+       names = talloc_array(tctx, struct lsa_String, tnames->count);
        for (i=0;i<tnames->count;i++) {
                init_lsa_String(&names[i], tnames->names[i].name.string);
        }
@@ -387,24 +406,35 @@ static bool test_LookupNames3(struct dcerpc_pipe *p,
        r.in.client_revision = 0;
        r.out.count = &count;
        r.out.sids = &sids;
+       r.out.domains = &domains;
 
-       status = dcerpc_lsa_LookupNames3(p, mem_ctx, &r);
+       status = dcerpc_lsa_LookupNames3(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
                printf("LookupNames3 failed - %s\n", nt_errstr(status));
                return false;
        }
 
+       if (check_result) {
+               torture_assert_int_equal(tctx, count, sids.count,
+                       "unexpected number of results returned");
+               if (sids.count > 0) {
+                       torture_assert(tctx, sids.sids, "invalid sid buffer");
+               }
+       }
+
        printf("\n");
 
        return true;
 }
 
-static bool test_LookupNames4(struct dcerpc_pipe *p, 
-                             TALLOC_CTX *mem_ctx, 
-                             struct lsa_TransNameArray2 *tnames)
+static bool test_LookupNames4(struct dcerpc_pipe *p,
+                             struct torture_context *tctx,
+                             struct lsa_TransNameArray2 *tnames,
+                             bool check_result)
 {
        struct lsa_LookupNames4 r;
        struct lsa_TransSidArray3 sids;
+       struct lsa_RefDomainList *domains = NULL;
        struct lsa_String *names;
        uint32_t count = 0;
        NTSTATUS status;
@@ -415,7 +445,7 @@ static bool test_LookupNames4(struct dcerpc_pipe *p,
        sids.count = 0;
        sids.sids = NULL;
 
-       names = talloc_array(mem_ctx, struct lsa_String, tnames->count);
+       names = talloc_array(tctx, struct lsa_String, tnames->count);
        for (i=0;i<tnames->count;i++) {
                init_lsa_String(&names[i], tnames->names[i].name.string);
        }
@@ -429,26 +459,36 @@ static bool test_LookupNames4(struct dcerpc_pipe *p,
        r.in.client_revision = 0;
        r.out.count = &count;
        r.out.sids = &sids;
+       r.out.domains = &domains;
 
-       status = dcerpc_lsa_LookupNames4(p, mem_ctx, &r);
+       status = dcerpc_lsa_LookupNames4(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
                printf("LookupNames4 failed - %s\n", nt_errstr(status));
                return false;
        }
 
+       if (check_result) {
+               torture_assert_int_equal(tctx, count, sids.count,
+                       "unexpected number of results returned");
+               if (sids.count > 0) {
+                       torture_assert(tctx, sids.sids, "invalid sid buffer");
+               }
+       }
+
        printf("\n");
 
        return true;
 }
 
 
-static bool test_LookupSids(struct dcerpc_pipe *p, 
-                           TALLOC_CTX *mem_ctx, 
+static bool test_LookupSids(struct dcerpc_pipe *p,
+                           struct torture_context *tctx,
                            struct policy_handle *handle,
                            struct lsa_SidArray *sids)
 {
        struct lsa_LookupSids r;
        struct lsa_TransNameArray names;
+       struct lsa_RefDomainList *domains = NULL;
        uint32_t count = sids->num_sids;
        NTSTATUS status;
 
@@ -464,8 +504,9 @@ static bool test_LookupSids(struct dcerpc_pipe *p,
        r.in.count = &count;
        r.out.count = &count;
        r.out.names = &names;
+       r.out.domains = &domains;
 
-       status = dcerpc_lsa_LookupSids(p, mem_ctx, &r);
+       status = dcerpc_lsa_LookupSids(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
                printf("LookupSids failed - %s\n", nt_errstr(status));
                return false;
@@ -473,7 +514,7 @@ static bool test_LookupSids(struct dcerpc_pipe *p,
 
        printf("\n");
 
-       if (!test_LookupNames(p, mem_ctx, handle, &names)) {
+       if (!test_LookupNames(p, tctx, handle, &names)) {
                return false;
        }
 
@@ -481,13 +522,14 @@ static bool test_LookupSids(struct dcerpc_pipe *p,
 }
 
 
-static bool test_LookupSids2(struct dcerpc_pipe *p, 
-                           TALLOC_CTX *mem_ctx, 
+static bool test_LookupSids2(struct dcerpc_pipe *p,
+                           struct torture_context *tctx,
                            struct policy_handle *handle,
                            struct lsa_SidArray *sids)
 {
        struct lsa_LookupSids2 r;
        struct lsa_TransNameArray2 names;
+       struct lsa_RefDomainList *domains = NULL;
        uint32_t count = sids->num_sids;
        NTSTATUS status;
 
@@ -505,8 +547,9 @@ static bool test_LookupSids2(struct dcerpc_pipe *p,
        r.in.unknown2 = 0;
        r.out.count = &count;
        r.out.names = &names;
+       r.out.domains = &domains;
 
-       status = dcerpc_lsa_LookupSids2(p, mem_ctx, &r);
+       status = dcerpc_lsa_LookupSids2(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
                printf("LookupSids2 failed - %s\n", nt_errstr(status));
                return false;
@@ -514,23 +557,24 @@ static bool test_LookupSids2(struct dcerpc_pipe *p,
 
        printf("\n");
 
-       if (!test_LookupNames2(p, mem_ctx, handle, &names)) {
+       if (!test_LookupNames2(p, tctx, handle, &names, false)) {
                return false;
        }
 
-       if (!test_LookupNames3(p, mem_ctx, handle, &names)) {
+       if (!test_LookupNames3(p, tctx, handle, &names, false)) {
                return false;
        }
 
        return true;
 }
 
-static bool test_LookupSids3(struct dcerpc_pipe *p, 
-                           TALLOC_CTX *mem_ctx, 
+static bool test_LookupSids3(struct dcerpc_pipe *p,
+                           struct torture_context *tctx,
                            struct lsa_SidArray *sids)
 {
        struct lsa_LookupSids3 r;
        struct lsa_TransNameArray2 names;
+       struct lsa_RefDomainList *domains = NULL;
        uint32_t count = sids->num_sids;
        NTSTATUS status;
 
@@ -545,32 +589,33 @@ static bool test_LookupSids3(struct dcerpc_pipe *p,
        r.in.count = &count;
        r.in.unknown1 = 0;
        r.in.unknown2 = 0;
+       r.out.domains = &domains;
        r.out.count = &count;
        r.out.names = &names;
 
-       status = dcerpc_lsa_LookupSids3(p, mem_ctx, &r);
+       status = dcerpc_lsa_LookupSids3(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
                if (NT_STATUS_EQUAL(status, NT_STATUS_ACCESS_DENIED) ||
                    NT_STATUS_EQUAL(status, NT_STATUS_RPC_PROTSEQ_NOT_SUPPORTED)) {
                        printf("not considering %s to be an error\n", nt_errstr(status));
                        return true;
                }
-               printf("LookupSids3 failed - %s - not considered an error\n", 
+               printf("LookupSids3 failed - %s - not considered an error\n",
                       nt_errstr(status));
                return false;
        }
 
        printf("\n");
 
-       if (!test_LookupNames4(p, mem_ctx, &names)) {
+       if (!test_LookupNames4(p, tctx, &names, false)) {
                return false;
        }
 
        return true;
 }
 
-bool test_many_LookupSids(struct dcerpc_pipe *p, 
-                         TALLOC_CTX *mem_ctx, 
+bool test_many_LookupSids(struct dcerpc_pipe *p,
+                         struct torture_context *tctx,
                          struct policy_handle *handle)
 {
        uint32_t count;
@@ -582,11 +627,11 @@ bool test_many_LookupSids(struct dcerpc_pipe *p,
 
        sids.num_sids = 100;
 
-       sids.sids = talloc_array(mem_ctx, struct lsa_SidPtr, sids.num_sids);
+       sids.sids = talloc_array(tctx, struct lsa_SidPtr, sids.num_sids);
 
        for (i=0; i<sids.num_sids; i++) {
                const char *sidstr = "S-1-5-32-545";
-               sids.sids[i].sid = dom_sid_parse_talloc(mem_ctx, sidstr);
+               sids.sids[i].sid = dom_sid_parse_talloc(tctx, sidstr);
        }
 
        count = sids.num_sids;
@@ -594,6 +639,7 @@ bool test_many_LookupSids(struct dcerpc_pipe *p,
        if (handle) {
                struct lsa_LookupSids r;
                struct lsa_TransNameArray names;
+               struct lsa_RefDomainList *domains = NULL;
                names.count = 0;
                names.names = NULL;
 
@@ -604,28 +650,30 @@ bool test_many_LookupSids(struct dcerpc_pipe *p,
                r.in.count = &names.count;
                r.out.count = &count;
                r.out.names = &names;
-               
-               status = dcerpc_lsa_LookupSids(p, mem_ctx, &r);
+               r.out.domains = &domains;
+
+               status = dcerpc_lsa_LookupSids(p, tctx, &r);
                if (!NT_STATUS_IS_OK(status)) {
                        printf("LookupSids failed - %s\n", nt_errstr(status));
                        return false;
                }
-               
+
                printf("\n");
-               
-               if (!test_LookupNames(p, mem_ctx, handle, &names)) {
+
+               if (!test_LookupNames(p, tctx, handle, &names)) {
                        return false;
                }
        } else if (p->conn->security_state.auth_info->auth_type == DCERPC_AUTH_TYPE_SCHANNEL &&
                   p->conn->security_state.auth_info->auth_level >= DCERPC_AUTH_LEVEL_INTEGRITY) {
                struct lsa_LookupSids3 r;
+               struct lsa_RefDomainList *domains = NULL;
                struct lsa_TransNameArray2 names;
 
                names.count = 0;
                names.names = NULL;
 
                printf("\nTesting LookupSids3\n");
-               
+
                r.in.sids = &sids;
                r.in.names = &names;
                r.in.level = 1;
@@ -634,19 +682,20 @@ bool test_many_LookupSids(struct dcerpc_pipe *p,
                r.in.unknown2 = 0;
                r.out.count = &count;
                r.out.names = &names;
-               
-               status = dcerpc_lsa_LookupSids3(p, mem_ctx, &r);
+               r.out.domains = &domains;
+
+               status = dcerpc_lsa_LookupSids3(p, tctx, &r);
                if (!NT_STATUS_IS_OK(status)) {
                        if (NT_STATUS_EQUAL(status, NT_STATUS_ACCESS_DENIED) ||
                            NT_STATUS_EQUAL(status, NT_STATUS_RPC_PROTSEQ_NOT_SUPPORTED)) {
                                printf("not considering %s to be an error\n", nt_errstr(status));
                                return true;
                        }
-                       printf("LookupSids3 failed - %s\n", 
+                       printf("LookupSids3 failed - %s\n",
                               nt_errstr(status));
                        return false;
                }
-               if (!test_LookupNames4(p, mem_ctx, &names)) {
+               if (!test_LookupNames4(p, tctx, &names, false)) {
                        return false;
                }
        }
@@ -674,8 +723,8 @@ static void lookupsids_cb(struct rpc_request *req)
        }
 }
 
-static bool test_LookupSids_async(struct dcerpc_pipe *p, 
-                                 TALLOC_CTX *mem_ctx, 
+static bool test_LookupSids_async(struct dcerpc_pipe *p,
+                                 struct torture_context *tctx,
                                  struct policy_handle *handle)
 {
        struct lsa_SidArray sids;
@@ -683,22 +732,23 @@ static bool test_LookupSids_async(struct dcerpc_pipe *p,
        uint32_t *count;
        struct lsa_TransNameArray *names;
        struct lsa_LookupSids *r;
+       struct lsa_RefDomainList *domains = NULL;
        struct rpc_request **req;
        int i, replies;
        bool ret = true;
        const int num_async_requests = 50;
 
-       count = talloc_array(mem_ctx, uint32_t, num_async_requests);
-       names = talloc_array(mem_ctx, struct lsa_TransNameArray, num_async_requests);
-       r = talloc_array(mem_ctx, struct lsa_LookupSids, num_async_requests);
+       count = talloc_array(tctx, uint32_t, num_async_requests);
+       names = talloc_array(tctx, struct lsa_TransNameArray, num_async_requests);
+       r = talloc_array(tctx, struct lsa_LookupSids, num_async_requests);
 
        printf("\nTesting %d async lookupsids request\n", num_async_requests);
 
-       req = talloc_array(mem_ctx, struct rpc_request *, num_async_requests);
+       req = talloc_array(tctx, struct rpc_request *, num_async_requests);
 
        sids.num_sids = 1;
        sids.sids = &sidptr;
-       sidptr.sid = dom_sid_parse_talloc(mem_ctx, "S-1-5-32-545");
+       sidptr.sid = dom_sid_parse_talloc(tctx, "S-1-5-32-545");
 
        replies = 0;
 
@@ -714,7 +764,8 @@ static bool test_LookupSids_async(struct dcerpc_pipe *p,
                r[i].in.count = &names[i].count;
                r[i].out.count = &count[i];
                r[i].out.names = &names[i];
-               
+               r[i].out.domains = &domains;
+
                req[i] = dcerpc_lsa_LookupSids_send(p, req, &r[i]);
                if (req[i] == NULL) {
                        ret = false;
@@ -738,8 +789,8 @@ static bool test_LookupSids_async(struct dcerpc_pipe *p,
        return ret;
 }
 
-static bool test_LookupPrivValue(struct dcerpc_pipe *p, 
-                                TALLOC_CTX *mem_ctx, 
+static bool test_LookupPrivValue(struct dcerpc_pipe *p,
+                                struct torture_context *tctx,
                                 struct policy_handle *handle,
                                 struct lsa_String *name)
 {
@@ -751,7 +802,7 @@ static bool test_LookupPrivValue(struct dcerpc_pipe *p,
        r.in.name = name;
        r.out.luid = &luid;
 
-       status = dcerpc_lsa_LookupPrivValue(p, mem_ctx, &r);
+       status = dcerpc_lsa_LookupPrivValue(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
                printf("\nLookupPrivValue failed - %s\n", nt_errstr(status));
                return false;
@@ -760,8 +811,8 @@ static bool test_LookupPrivValue(struct dcerpc_pipe *p,
        return true;
 }
 
-static bool test_LookupPrivName(struct dcerpc_pipe *p, 
-                               TALLOC_CTX *mem_ctx, 
+static bool test_LookupPrivName(struct dcerpc_pipe *p,
+                               struct torture_context *tctx,
                                struct policy_handle *handle,
                                struct lsa_LUID *luid)
 {
@@ -773,7 +824,7 @@ static bool test_LookupPrivName(struct dcerpc_pipe *p,
        r.in.luid = luid;
        r.out.name = &name;
 
-       status = dcerpc_lsa_LookupPrivName(p, mem_ctx, &r);
+       status = dcerpc_lsa_LookupPrivName(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
                printf("\nLookupPrivName failed - %s\n", nt_errstr(status));
                return false;
@@ -782,8 +833,8 @@ static bool test_LookupPrivName(struct dcerpc_pipe *p,
        return true;
 }
 
-static bool test_RemovePrivilegesFromAccount(struct dcerpc_pipe *p, 
-                                            TALLOC_CTX *mem_ctx,
+static bool test_RemovePrivilegesFromAccount(struct dcerpc_pipe *p,
+                                            struct torture_context *tctx,
                                             struct policy_handle *handle,
                                             struct policy_handle *acct_handle,
                                             struct lsa_LUID *luid)
@@ -801,21 +852,21 @@ static bool test_RemovePrivilegesFromAccount(struct dcerpc_pipe *p,
 
        privs.count = 1;
        privs.unknown = 0;
-       privs.set = talloc_array(mem_ctx, struct lsa_LUIDAttribute, 1);
+       privs.set = talloc_array(tctx, struct lsa_LUIDAttribute, 1);
        privs.set[0].luid = *luid;
        privs.set[0].attribute = 0;
 
-       status = dcerpc_lsa_RemovePrivilegesFromAccount(p, mem_ctx, &r);
+       status = dcerpc_lsa_RemovePrivilegesFromAccount(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
-               
+
                struct lsa_LookupPrivName r_name;
                struct lsa_StringLarge *name = NULL;
-               
+
                r_name.in.handle = handle;
                r_name.in.luid = luid;
                r_name.out.name = &name;
 
-               status = dcerpc_lsa_LookupPrivName(p, mem_ctx, &r_name);
+               status = dcerpc_lsa_LookupPrivName(p, tctx, &r_name);
                if (!NT_STATUS_IS_OK(status)) {
                        printf("\nLookupPrivName failed - %s\n", nt_errstr(status));
                        return false;
@@ -825,7 +876,7 @@ static bool test_RemovePrivilegesFromAccount(struct dcerpc_pipe *p,
                        return ret;
                }
 
-               printf("RemovePrivilegesFromAccount failed to remove %s - %s\n", 
+               printf("RemovePrivilegesFromAccount failed to remove %s - %s\n",
                       name->string,
                       nt_errstr(status));
                return false;
@@ -834,8 +885,8 @@ static bool test_RemovePrivilegesFromAccount(struct dcerpc_pipe *p,
        return ret;
 }
 
-static bool test_AddPrivilegesToAccount(struct dcerpc_pipe *p, 
-                                       TALLOC_CTX *mem_ctx,
+static bool test_AddPrivilegesToAccount(struct dcerpc_pipe *p,
+                                       struct torture_context *tctx,
                                        struct policy_handle *acct_handle,
                                        struct lsa_LUID *luid)
 {
@@ -851,11 +902,11 @@ static bool test_AddPrivilegesToAccount(struct dcerpc_pipe *p,
 
        privs.count = 1;
        privs.unknown = 0;
-       privs.set = talloc_array(mem_ctx, struct lsa_LUIDAttribute, 1);
+       privs.set = talloc_array(tctx, struct lsa_LUIDAttribute, 1);
        privs.set[0].luid = *luid;
        privs.set[0].attribute = 0;
 
-       status = dcerpc_lsa_AddPrivilegesToAccount(p, mem_ctx, &r);
+       status = dcerpc_lsa_AddPrivilegesToAccount(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
                printf("AddPrivilegesToAccount failed - %s\n", nt_errstr(status));
                return false;
@@ -864,8 +915,8 @@ static bool test_AddPrivilegesToAccount(struct dcerpc_pipe *p,
        return ret;
 }
 
-static bool test_EnumPrivsAccount(struct dcerpc_pipe *p, 
-                                 TALLOC_CTX *mem_ctx,
+static bool test_EnumPrivsAccount(struct dcerpc_pipe *p,
+                                 struct torture_context *tctx,
                                  struct policy_handle *handle,
                                  struct policy_handle *acct_handle)
 {
@@ -879,7 +930,7 @@ static bool test_EnumPrivsAccount(struct dcerpc_pipe *p,
        r.in.handle = acct_handle;
        r.out.privs = &privs;
 
-       status = dcerpc_lsa_EnumPrivsAccount(p, mem_ctx, &r);
+       status = dcerpc_lsa_EnumPrivsAccount(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
                printf("EnumPrivsAccount failed - %s\n", nt_errstr(status));
                return false;
@@ -888,13 +939,13 @@ static bool test_EnumPrivsAccount(struct dcerpc_pipe *p,
        if (privs && privs->count > 0) {
                int i;
                for (i=0;i<privs->count;i++) {
-                       test_LookupPrivName(p, mem_ctx, handle, 
+                       test_LookupPrivName(p, tctx, handle,
                                            &privs->set[i].luid);
                }
 
-               ret &= test_RemovePrivilegesFromAccount(p, mem_ctx, handle, acct_handle, 
+               ret &= test_RemovePrivilegesFromAccount(p, tctx, handle, acct_handle,
                                                        &privs->set[0].luid);
-               ret &= test_AddPrivilegesToAccount(p, mem_ctx, acct_handle, 
+               ret &= test_AddPrivilegesToAccount(p, tctx, acct_handle,
                                                   &privs->set[0].luid);
        }
 
@@ -902,7 +953,7 @@ static bool test_EnumPrivsAccount(struct dcerpc_pipe *p,
 }
 
 static bool test_GetSystemAccessAccount(struct dcerpc_pipe *p,
-                                       TALLOC_CTX *mem_ctx,
+                                       struct torture_context *tctx,
                                        struct policy_handle *handle,
                                        struct policy_handle *acct_handle)
 {
@@ -915,7 +966,7 @@ static bool test_GetSystemAccessAccount(struct dcerpc_pipe *p,
        r.in.handle = acct_handle;
        r.out.access_mask = &access_mask;
 
-       status = dcerpc_lsa_GetSystemAccessAccount(p, mem_ctx, &r);
+       status = dcerpc_lsa_GetSystemAccessAccount(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
                printf("GetSystemAccessAccount failed - %s\n", nt_errstr(status));
                return false;
@@ -955,9 +1006,9 @@ static bool test_GetSystemAccessAccount(struct dcerpc_pipe *p,
        return true;
 }
 
-static bool test_Delete(struct dcerpc_pipe *p, 
-                      TALLOC_CTX *mem_ctx, 
-                      struct policy_handle *handle)
+static bool test_Delete(struct dcerpc_pipe *p,
+                       struct torture_context *tctx,
+                       struct policy_handle *handle)
 {
        NTSTATUS status;
        struct lsa_Delete r;
@@ -965,7 +1016,7 @@ static bool test_Delete(struct dcerpc_pipe *p,
        printf("\nTesting Delete\n");
 
        r.in.handle = handle;
-       status = dcerpc_lsa_Delete(p, mem_ctx, &r);
+       status = dcerpc_lsa_Delete(p, tctx, &r);
        if (!NT_STATUS_EQUAL(status, NT_STATUS_NOT_SUPPORTED)) {
                printf("Delete should have failed NT_STATUS_NOT_SUPPORTED - %s\n", nt_errstr(status));
                return false;
@@ -974,8 +1025,8 @@ static bool test_Delete(struct dcerpc_pipe *p,
        return true;
 }
 
-static bool test_DeleteObject(struct dcerpc_pipe *p, 
-                             TALLOC_CTX *mem_ctx, 
+static bool test_DeleteObject(struct dcerpc_pipe *p,
+                             struct torture_context *tctx,
                              struct policy_handle *handle)
 {
        NTSTATUS status;
@@ -985,7 +1036,7 @@ static bool test_DeleteObject(struct dcerpc_pipe *p,
 
        r.in.handle = handle;
        r.out.handle = handle;
-       status = dcerpc_lsa_DeleteObject(p, mem_ctx, &r);
+       status = dcerpc_lsa_DeleteObject(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
                printf("DeleteObject failed - %s\n", nt_errstr(status));
                return false;
@@ -995,8 +1046,8 @@ static bool test_DeleteObject(struct dcerpc_pipe *p,
 }
 
 
-static bool test_CreateAccount(struct dcerpc_pipe *p, 
-                              TALLOC_CTX *mem_ctx, 
+static bool test_CreateAccount(struct dcerpc_pipe *p,
+                              struct torture_context *tctx,
                               struct policy_handle *handle)
 {
        NTSTATUS status;
@@ -1004,7 +1055,7 @@ static bool test_CreateAccount(struct dcerpc_pipe *p,
        struct dom_sid2 *newsid;
        struct policy_handle acct_handle;
 
-       newsid = dom_sid_parse_talloc(mem_ctx, "S-1-5-12349876-4321-2854");
+       newsid = dom_sid_parse_talloc(tctx, "S-1-5-12349876-4321-2854");
 
        printf("\nTesting CreateAccount\n");
 
@@ -1013,15 +1064,15 @@ static bool test_CreateAccount(struct dcerpc_pipe *p,
        r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
        r.out.acct_handle = &acct_handle;
 
-       status = dcerpc_lsa_CreateAccount(p, mem_ctx, &r);
+       status = dcerpc_lsa_CreateAccount(p, tctx, &r);
        if (NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_NAME_COLLISION)) {
                struct lsa_OpenAccount r_o;
                r_o.in.handle = handle;
                r_o.in.sid = newsid;
                r_o.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
                r_o.out.acct_handle = &acct_handle;
-               
-               status = dcerpc_lsa_OpenAccount(p, mem_ctx, &r_o);
+
+               status = dcerpc_lsa_OpenAccount(p, tctx, &r_o);
                if (!NT_STATUS_IS_OK(status)) {
                        printf("OpenAccount failed - %s\n", nt_errstr(status));
                        return false;
@@ -1031,19 +1082,19 @@ static bool test_CreateAccount(struct dcerpc_pipe *p,
                return false;
        }
 
-       if (!test_Delete(p, mem_ctx, &acct_handle)) {
+       if (!test_Delete(p, tctx, &acct_handle)) {
                return false;
        }
 
-       if (!test_DeleteObject(p, mem_ctx, &acct_handle)) {
+       if (!test_DeleteObject(p, tctx, &acct_handle)) {
                return false;
        }
 
        return true;
 }
 
-static bool test_DeleteTrustedDomain(struct dcerpc_pipe *p, 
-                                    TALLOC_CTX *mem_ctx, 
+static bool test_DeleteTrustedDomain(struct dcerpc_pipe *p,
+                                    struct torture_context *tctx,
                                     struct policy_handle *handle,
                                     struct lsa_StringLarge name)
 {
@@ -1056,25 +1107,25 @@ static bool test_DeleteTrustedDomain(struct dcerpc_pipe *p,
        r.in.access_mask = SEC_STD_DELETE;
        r.out.trustdom_handle = &trustdom_handle;
 
-       status = dcerpc_lsa_OpenTrustedDomainByName(p, mem_ctx, &r);
+       status = dcerpc_lsa_OpenTrustedDomainByName(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
                printf("OpenTrustedDomainByName failed - %s\n", nt_errstr(status));
                return false;
        }
 
-       if (!test_Delete(p, mem_ctx, &trustdom_handle)) {
+       if (!test_Delete(p, tctx, &trustdom_handle)) {
                return false;
        }
 
-       if (!test_DeleteObject(p, mem_ctx, &trustdom_handle)) {
+       if (!test_DeleteObject(p, tctx, &trustdom_handle)) {
                return false;
        }
 
        return true;
 }
 
-static bool test_DeleteTrustedDomainBySid(struct dcerpc_pipe *p, 
-                                         TALLOC_CTX *mem_ctx, 
+static bool test_DeleteTrustedDomainBySid(struct dcerpc_pipe *p,
+                                         struct torture_context *tctx,
                                          struct policy_handle *handle,
                                          struct dom_sid *sid)
 {
@@ -1084,7 +1135,7 @@ static bool test_DeleteTrustedDomainBySid(struct dcerpc_pipe *p,
        r.in.handle = handle;
        r.in.dom_sid = sid;
 
-       status = dcerpc_lsa_DeleteTrustedDomain(p, mem_ctx, &r);
+       status = dcerpc_lsa_DeleteTrustedDomain(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
                printf("DeleteTrustedDomain failed - %s\n", nt_errstr(status));
                return false;
@@ -1094,8 +1145,8 @@ static bool test_DeleteTrustedDomainBySid(struct dcerpc_pipe *p,
 }
 
 
-static bool test_CreateSecret(struct dcerpc_pipe *p, 
-                             TALLOC_CTX *mem_ctx, 
+static bool test_CreateSecret(struct dcerpc_pipe *p,
+                             struct torture_context *tctx,
                              struct policy_handle *handle)
 {
        NTSTATUS status;
@@ -1128,104 +1179,104 @@ static bool test_CreateSecret(struct dcerpc_pipe *p,
        const int LOCAL = 0;
        const int GLOBAL = 1;
 
-       secname[LOCAL] = talloc_asprintf(mem_ctx, "torturesecret-%u", (uint_t)random());
-       secname[GLOBAL] = talloc_asprintf(mem_ctx, "G$torturesecret-%u", (uint_t)random());
+       secname[LOCAL] = talloc_asprintf(tctx, "torturesecret-%u", (uint_t)random());
+       secname[GLOBAL] = talloc_asprintf(tctx, "G$torturesecret-%u", (uint_t)random());
 
        for (i=0; i< 2; i++) {
                printf("\nTesting CreateSecret of %s\n", secname[i]);
-               
+
                init_lsa_String(&r.in.name, secname[i]);
-               
+
                r.in.handle = handle;
                r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
                r.out.sec_handle = &sec_handle;
-               
-               status = dcerpc_lsa_CreateSecret(p, mem_ctx, &r);
+
+               status = dcerpc_lsa_CreateSecret(p, tctx, &r);
                if (!NT_STATUS_IS_OK(status)) {
                        printf("CreateSecret failed - %s\n", nt_errstr(status));
                        return false;
                }
-               
+
                r.in.handle = handle;
                r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
                r.out.sec_handle = &sec_handle3;
-               
-               status = dcerpc_lsa_CreateSecret(p, mem_ctx, &r);
+
+               status = dcerpc_lsa_CreateSecret(p, tctx, &r);
                if (!NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_NAME_COLLISION)) {
                        printf("CreateSecret should have failed OBJECT_NAME_COLLISION - %s\n", nt_errstr(status));
                        return false;
                }
-               
+
                r2.in.handle = handle;
                r2.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
                r2.in.name = r.in.name;
                r2.out.sec_handle = &sec_handle2;
-               
+
                printf("Testing OpenSecret\n");
-               
-               status = dcerpc_lsa_OpenSecret(p, mem_ctx, &r2);
+
+               status = dcerpc_lsa_OpenSecret(p, tctx, &r2);
                if (!NT_STATUS_IS_OK(status)) {
                        printf("OpenSecret failed - %s\n", nt_errstr(status));
                        return false;
                }
-               
+
                status = dcerpc_fetch_session_key(p, &session_key);
                if (!NT_STATUS_IS_OK(status)) {
                        printf("dcerpc_fetch_session_key failed - %s\n", nt_errstr(status));
                        return false;
                }
-               
+
                enc_key = sess_encrypt_string(secret1, &session_key);
-               
+
                r3.in.sec_handle = &sec_handle;
                r3.in.new_val = &buf1;
                r3.in.old_val = NULL;
                r3.in.new_val->data = enc_key.data;
                r3.in.new_val->length = enc_key.length;
                r3.in.new_val->size = enc_key.length;
-               
+
                printf("Testing SetSecret\n");
-               
-               status = dcerpc_lsa_SetSecret(p, mem_ctx, &r3);
+
+               status = dcerpc_lsa_SetSecret(p, tctx, &r3);
                if (!NT_STATUS_IS_OK(status)) {
                        printf("SetSecret failed - %s\n", nt_errstr(status));
                        return false;
                }
-               
+
                r3.in.sec_handle = &sec_handle;
                r3.in.new_val = &buf1;
                r3.in.old_val = NULL;
                r3.in.new_val->data = enc_key.data;
                r3.in.new_val->length = enc_key.length;
                r3.in.new_val->size = enc_key.length;
-               
+
                /* break the encrypted data */
                enc_key.data[0]++;
 
                printf("Testing SetSecret with broken key\n");
-               
-               status = dcerpc_lsa_SetSecret(p, mem_ctx, &r3);
+
+               status = dcerpc_lsa_SetSecret(p, tctx, &r3);
                if (!NT_STATUS_EQUAL(status, NT_STATUS_UNKNOWN_REVISION)) {
                        printf("SetSecret should have failed UNKNOWN_REVISION - %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 */
                r4.in.sec_handle = &sec_handle;
                r4.in.new_val = &bufp1;
                r4.in.new_mtime = &new_mtime;
                r4.in.old_val = NULL;
                r4.in.old_mtime = NULL;
-               
+
                bufp1.buf = NULL;
-               
+
                printf("Testing QuerySecret\n");
-               status = dcerpc_lsa_QuerySecret(p, mem_ctx, &r4);
+               status = dcerpc_lsa_QuerySecret(p, tctx, &r4);
                if (!NT_STATUS_IS_OK(status)) {
                        printf("QuerySecret failed - %s\n", nt_errstr(status));
                        ret = false;
@@ -1236,22 +1287,22 @@ static bool test_CreateSecret(struct dcerpc_pipe *p,
                        } else {
                                blob1.data = r4.out.new_val->buf->data;
                                blob1.length = r4.out.new_val->buf->size;
-                               
-                               blob2 = data_blob_talloc(mem_ctx, NULL, blob1.length);
-                               
-                               secret2 = sess_decrypt_string(mem_ctx, 
+
+                               blob2 = data_blob_talloc(tctx, NULL, blob1.length);
+
+                               secret2 = sess_decrypt_string(tctx,
                                                              &blob1, &session_key);
-                               
+
                                if (strcmp(secret1, secret2) != 0) {
-                                       printf("Returned secret (r4) '%s' doesn't match '%s'\n", 
+                                       printf("Returned secret (r4) '%s' doesn't match '%s'\n",
                                               secret2, secret1);
                                        ret = false;
                                }
                        }
                }
-               
+
                enc_key = sess_encrypt_string(secret3, &session_key);
-               
+
                r5.in.sec_handle = &sec_handle;
                r5.in.new_val = &buf1;
                r5.in.old_val = NULL;
@@ -1262,36 +1313,36 @@ static bool test_CreateSecret(struct dcerpc_pipe *p,
 
                msleep(200);
                printf("Testing SetSecret (existing value should move to old)\n");
-               
-               status = dcerpc_lsa_SetSecret(p, mem_ctx, &r5);
+
+               status = dcerpc_lsa_SetSecret(p, tctx, &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.sec_handle = &sec_handle;
                r6.in.new_val = &bufp1;
                r6.in.new_mtime = &new_mtime;
                r6.in.old_val = &bufp2;
                r6.in.old_mtime = &old_mtime;
-               
+
                bufp1.buf = NULL;
                bufp2.buf = NULL;
-               
-               status = dcerpc_lsa_QuerySecret(p, mem_ctx, &r6);
+
+               status = dcerpc_lsa_QuerySecret(p, tctx, &r6);
                if (!NT_STATUS_IS_OK(status)) {
                        printf("QuerySecret failed - %s\n", nt_errstr(status));
                        ret = false;
                        secret4 = NULL;
                } else {
 
-                       if (r6.out.new_val->buf == NULL || r6.out.old_val->buf == NULL 
+                       if (r6.out.new_val->buf == NULL || r6.out.old_val->buf == NULL
                                || r6.out.new_mtime == NULL || r6.out.old_mtime == NULL) {
                                printf("Both secret buffers and both times not returned\n");
                                ret = false;
@@ -1299,12 +1350,12 @@ static bool test_CreateSecret(struct dcerpc_pipe *p,
                        } else {
                                blob1.data = r6.out.new_val->buf->data;
                                blob1.length = r6.out.new_val->buf->size;
-                               
-                               blob2 = data_blob_talloc(mem_ctx, NULL, blob1.length);
-                               
-                               secret4 = sess_decrypt_string(mem_ctx, 
+
+                               blob2 = data_blob_talloc(tctx, NULL, blob1.length);
+
+                               secret4 = sess_decrypt_string(tctx,
                                                              &blob1, &session_key);
-                               
+
                                if (strcmp(secret3, secret4) != 0) {
                                        printf("Returned NEW secret %s doesn't match %s\n", secret4, secret3);
                                        ret = false;
@@ -1312,58 +1363,58 @@ static bool test_CreateSecret(struct dcerpc_pipe *p,
 
                                blob1.data = r6.out.old_val->buf->data;
                                blob1.length = r6.out.old_val->buf->length;
-                               
-                               blob2 = data_blob_talloc(mem_ctx, NULL, blob1.length);
-                               
-                               secret2 = sess_decrypt_string(mem_ctx, 
+
+                               blob2 = data_blob_talloc(tctx, NULL, blob1.length);
+
+                               secret2 = sess_decrypt_string(tctx,
                                                              &blob1, &session_key);
-                               
+
                                if (strcmp(secret1, secret2) != 0) {
                                        printf("Returned OLD secret %s doesn't match %s\n", secret2, secret1);
                                        ret = false;
                                }
-                               
+
                                if (*r6.out.new_mtime == *r6.out.old_mtime) {
-                                       printf("Returned secret (r6-%d) %s must not have same mtime for both secrets: %s != %s\n", 
+                                       printf("Returned secret (r6-%d) %s must not have same mtime for both secrets: %s != %s\n",
                                               i,
                                               secname[i],
-                                              nt_time_string(mem_ctx, *r6.out.old_mtime), 
-                                              nt_time_string(mem_ctx, *r6.out.new_mtime));
+                                              nt_time_string(tctx, *r6.out.old_mtime),
+                                              nt_time_string(tctx, *r6.out.new_mtime));
                                        ret = false;
                                }
                        }
                }
 
                enc_key = sess_encrypt_string(secret5, &session_key);
-               
+
                r7.in.sec_handle = &sec_handle;
                r7.in.old_val = &buf1;
                r7.in.old_val->data = enc_key.data;
                r7.in.old_val->length = enc_key.length;
                r7.in.old_val->size = enc_key.length;
                r7.in.new_val = NULL;
-               
+
                printf("Testing SetSecret of old Secret only\n");
-               
-               status = dcerpc_lsa_SetSecret(p, mem_ctx, &r7);
+
+               status = dcerpc_lsa_SetSecret(p, tctx, &r7);
                if (!NT_STATUS_IS_OK(status)) {
                        printf("SetSecret failed - %s\n", nt_errstr(status));
                        ret = false;
                }
-               
+
                data_blob_free(&enc_key);
-               
+
                /* fetch the secret back again */
                r8.in.sec_handle = &sec_handle;
                r8.in.new_val = &bufp1;
                r8.in.new_mtime = &new_mtime;
                r8.in.old_val = &bufp2;
                r8.in.old_mtime = &old_mtime;
-               
+
                bufp1.buf = NULL;
                bufp2.buf = NULL;
-               
-               status = dcerpc_lsa_QuerySecret(p, mem_ctx, &r8);
+
+               status = dcerpc_lsa_QuerySecret(p, tctx, &r8);
                if (!NT_STATUS_IS_OK(status)) {
                        printf("QuerySecret failed - %s\n", nt_errstr(status));
                        ret = false;
@@ -1383,60 +1434,60 @@ static bool test_CreateSecret(struct dcerpc_pipe *p,
                        } else {
                                blob1.data = r8.out.old_val->buf->data;
                                blob1.length = r8.out.old_val->buf->size;
-                               
-                               blob2 = data_blob_talloc(mem_ctx, NULL, blob1.length);
-                               
-                               secret6 = sess_decrypt_string(mem_ctx,
+
+                               blob2 = data_blob_talloc(tctx, NULL, blob1.length);
+
+                               secret6 = sess_decrypt_string(tctx,
                                                              &blob1, &session_key);
-                               
+
                                if (strcmp(secret5, secret6) != 0) {
                                        printf("Returned OLD secret %s doesn't match %s\n", secret5, secret6);
                                        ret = false;
                                }
-                               
+
                                if (*r8.out.new_mtime != *r8.out.old_mtime) {
-                                       printf("Returned secret (r8) %s did not had same mtime for both secrets: %s != %s\n", 
+                                       printf("Returned secret (r8) %s did not had same mtime for both secrets: %s != %s\n",
                                               secname[i],
-                                              nt_time_string(mem_ctx, *r8.out.old_mtime),
-                                              nt_time_string(mem_ctx, *r8.out.new_mtime));
+                                              nt_time_string(tctx, *r8.out.old_mtime),
+                                              nt_time_string(tctx, *r8.out.new_mtime));
                                        ret = false;
                                }
                        }
                }
 
-               if (!test_Delete(p, mem_ctx, &sec_handle)) {
+               if (!test_Delete(p, tctx, &sec_handle)) {
                        ret = false;
                }
-               
-               if (!test_DeleteObject(p, mem_ctx, &sec_handle)) {
+
+               if (!test_DeleteObject(p, tctx, &sec_handle)) {
                        return false;
                }
 
                d_o.in.handle = &sec_handle2;
                d_o.out.handle = &sec_handle2;
-               status = dcerpc_lsa_DeleteObject(p, mem_ctx, &d_o);
+               status = dcerpc_lsa_DeleteObject(p, tctx, &d_o);
                if (!NT_STATUS_EQUAL(status, NT_STATUS_INVALID_HANDLE)) {
                        printf("Second delete expected INVALID_HANDLE - %s\n", nt_errstr(status));
                        ret = false;
                } else {
 
                        printf("Testing OpenSecret of just-deleted secret\n");
-                       
-                       status = dcerpc_lsa_OpenSecret(p, mem_ctx, &r2);
+
+                       status = dcerpc_lsa_OpenSecret(p, tctx, &r2);
                        if (!NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_NAME_NOT_FOUND)) {
                                printf("OpenSecret expected OBJECT_NAME_NOT_FOUND - %s\n", nt_errstr(status));
                                ret = false;
                        }
                }
-               
+
        }
 
        return ret;
 }
 
 
-static bool test_EnumAccountRights(struct dcerpc_pipe *p, 
-                                  TALLOC_CTX *mem_ctx, 
+static bool test_EnumAccountRights(struct dcerpc_pipe *p,
+                                  struct torture_context *tctx,
                                   struct policy_handle *acct_handle,
                                   struct dom_sid *sid)
 {
@@ -1450,10 +1501,10 @@ static bool test_EnumAccountRights(struct dcerpc_pipe *p,
        r.in.sid = sid;
        r.out.rights = &rights;
 
-       status = dcerpc_lsa_EnumAccountRights(p, mem_ctx, &r);
+       status = dcerpc_lsa_EnumAccountRights(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
-               printf("EnumAccountRights of %s failed - %s\n", 
-                      dom_sid_string(mem_ctx, sid), nt_errstr(status));
+               printf("EnumAccountRights of %s failed - %s\n",
+                      dom_sid_string(tctx, sid), nt_errstr(status));
                return false;
        }
 
@@ -1461,8 +1512,8 @@ static bool test_EnumAccountRights(struct dcerpc_pipe *p,
 }
 
 
-static bool test_QuerySecurity(struct dcerpc_pipe *p, 
-                            struct torture_context *tctx, 
+static bool test_QuerySecurity(struct dcerpc_pipe *p,
+                            struct torture_context *tctx,
                             struct policy_handle *handle,
                             struct policy_handle *acct_handle)
 {
@@ -1490,8 +1541,8 @@ static bool test_QuerySecurity(struct dcerpc_pipe *p,
        return true;
 }
 
-static bool test_OpenAccount(struct dcerpc_pipe *p, 
-                            TALLOC_CTX *mem_ctx, 
+static bool test_OpenAccount(struct dcerpc_pipe *p,
+                            struct torture_context *tctx,
                             struct policy_handle *handle,
                             struct dom_sid *sid)
 {
@@ -1506,30 +1557,30 @@ static bool test_OpenAccount(struct dcerpc_pipe *p,
        r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
        r.out.acct_handle = &acct_handle;
 
-       status = dcerpc_lsa_OpenAccount(p, mem_ctx, &r);
+       status = dcerpc_lsa_OpenAccount(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
                printf("OpenAccount failed - %s\n", nt_errstr(status));
                return false;
        }
 
-       if (!test_EnumPrivsAccount(p, mem_ctx, handle, &acct_handle)) {
+       if (!test_EnumPrivsAccount(p, tctx, handle, &acct_handle)) {
                return false;
        }
 
-       if (!test_GetSystemAccessAccount(p, mem_ctx, handle, &acct_handle)) {
+       if (!test_GetSystemAccessAccount(p, tctx, handle, &acct_handle)) {
                return false;
        }
 
-       if (!test_QuerySecurity(p, mem_ctx, handle, &acct_handle)) {
+       if (!test_QuerySecurity(p, tctx, handle, &acct_handle)) {
                return false;
        }
 
        return true;
 }
 
-static bool test_EnumAccounts(struct dcerpc_pipe *p, 
-                         TALLOC_CTX *mem_ctx, 
-                         struct policy_handle *handle)
+static bool test_EnumAccounts(struct dcerpc_pipe *p,
+                             struct torture_context *tctx,
+                             struct policy_handle *handle)
 {
        NTSTATUS status;
        struct lsa_EnumAccounts r;
@@ -1548,7 +1599,7 @@ static bool test_EnumAccounts(struct dcerpc_pipe *p,
 
        resume_handle = 0;
        while (true) {
-               status = dcerpc_lsa_EnumAccounts(p, mem_ctx, &r);
+               status = dcerpc_lsa_EnumAccounts(p, tctx, &r);
                if (NT_STATUS_EQUAL(status, NT_STATUS_NO_MORE_ENTRIES)) {
                        break;
                }
@@ -1557,11 +1608,11 @@ static bool test_EnumAccounts(struct dcerpc_pipe *p,
                        return false;
                }
 
-               if (!test_LookupSids(p, mem_ctx, handle, &sids1)) {
+               if (!test_LookupSids(p, tctx, handle, &sids1)) {
                        return false;
                }
 
-               if (!test_LookupSids2(p, mem_ctx, handle, &sids1)) {
+               if (!test_LookupSids2(p, tctx, handle, &sids1)) {
                        return false;
                }
 
@@ -1571,8 +1622,8 @@ static bool test_EnumAccounts(struct dcerpc_pipe *p,
 
                printf("Testing all accounts\n");
                for (i=0;i<sids1.num_sids;i++) {
-                       ret &= test_OpenAccount(p, mem_ctx, handle, sids1.sids[i].sid);
-                       ret &= test_EnumAccountRights(p, mem_ctx, handle, sids1.sids[i].sid);
+                       ret &= test_OpenAccount(p, tctx, handle, sids1.sids[i].sid);
+                       ret &= test_EnumAccountRights(p, tctx, handle, sids1.sids[i].sid);
                }
                printf("\n");
        }
@@ -1580,13 +1631,13 @@ static bool test_EnumAccounts(struct dcerpc_pipe *p,
        if (sids1.num_sids < 3) {
                return ret;
        }
-       
+
        printf("Trying EnumAccounts partial listing (asking for 1 at 2)\n");
        resume_handle = 2;
        r.in.num_entries = 1;
        r.out.sids = &sids2;
 
-       status = dcerpc_lsa_EnumAccounts(p, mem_ctx, &r);
+       status = dcerpc_lsa_EnumAccounts(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
                printf("EnumAccounts failed - %s\n", nt_errstr(status));
                return false;
@@ -1601,9 +1652,9 @@ static bool test_EnumAccounts(struct dcerpc_pipe *p,
 }
 
 static bool test_LookupPrivDisplayName(struct dcerpc_pipe *p,
-                               TALLOC_CTX *mem_ctx,
-                               struct policy_handle *handle,
-                               struct lsa_String *priv_name)
+                                      struct torture_context *tctx,
+                                      struct policy_handle *handle,
+                                      struct lsa_String *priv_name)
 {
        struct lsa_LookupPrivDisplayName r;
        NTSTATUS status;
@@ -1614,7 +1665,7 @@ static bool test_LookupPrivDisplayName(struct dcerpc_pipe *p,
        struct lsa_StringLarge *disp_name = NULL;
 
        printf("\nTesting LookupPrivDisplayName(%s)\n", priv_name->string);
-       
+
        r.in.handle = handle;
        r.in.name = priv_name;
        r.in.language_id = language_id;
@@ -1622,36 +1673,36 @@ static bool test_LookupPrivDisplayName(struct dcerpc_pipe *p,
        r.out.returned_language_id = &returned_language_id;
        r.out.disp_name = &disp_name;
 
-       status = dcerpc_lsa_LookupPrivDisplayName(p, mem_ctx, &r);
+       status = dcerpc_lsa_LookupPrivDisplayName(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
                printf("LookupPrivDisplayName failed - %s\n", nt_errstr(status));
                return false;
        }
-       printf("%s -> \"%s\"  (language 0x%x/0x%x)\n", 
+       printf("%s -> \"%s\"  (language 0x%x/0x%x)\n",
               priv_name->string, disp_name->string,
               r.in.language_id, *r.out.returned_language_id);
 
        return true;
 }
 
-static bool test_EnumAccountsWithUserRight(struct dcerpc_pipe *p, 
-                               TALLOC_CTX *mem_ctx,
-                               struct policy_handle *handle,
-                               struct lsa_String *priv_name)
+static bool test_EnumAccountsWithUserRight(struct dcerpc_pipe *p,
+                                          struct torture_context *tctx,
+                                          struct policy_handle *handle,
+                                          struct lsa_String *priv_name)
 {
        struct lsa_EnumAccountsWithUserRight r;
        struct lsa_SidArray sids;
        NTSTATUS status;
 
        ZERO_STRUCT(sids);
-       
+
        printf("\nTesting EnumAccountsWithUserRight(%s)\n", priv_name->string);
-       
+
        r.in.handle = handle;
        r.in.name = priv_name;
        r.out.sids = &sids;
 
-       status = dcerpc_lsa_EnumAccountsWithUserRight(p, mem_ctx, &r);
+       status = dcerpc_lsa_EnumAccountsWithUserRight(p, tctx, &r);
 
        /* NT_STATUS_NO_MORE_ENTRIES means noone has this privilege */
        if (NT_STATUS_EQUAL(status, NT_STATUS_NO_MORE_ENTRIES)) {
@@ -1662,13 +1713,13 @@ static bool test_EnumAccountsWithUserRight(struct dcerpc_pipe *p,
                printf("EnumAccountsWithUserRight failed - %s\n", nt_errstr(status));
                return false;
        }
-       
+
        return true;
 }
 
 
-static bool test_EnumPrivs(struct dcerpc_pipe *p, 
-                          TALLOC_CTX *mem_ctx, 
+static bool test_EnumPrivs(struct dcerpc_pipe *p,
+                          struct torture_context *tctx,
                           struct policy_handle *handle)
 {
        NTSTATUS status;
@@ -1687,16 +1738,16 @@ static bool test_EnumPrivs(struct dcerpc_pipe *p,
        r.out.privs = &privs1;
 
        resume_handle = 0;
-       status = dcerpc_lsa_EnumPrivs(p, mem_ctx, &r);
+       status = dcerpc_lsa_EnumPrivs(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
                printf("EnumPrivs failed - %s\n", nt_errstr(status));
                return false;
        }
 
        for (i = 0; i< privs1.count; i++) {
-               test_LookupPrivDisplayName(p, mem_ctx, handle, (struct lsa_String *)&privs1.privs[i].name);
-               test_LookupPrivValue(p, mem_ctx, handle, (struct lsa_String *)&privs1.privs[i].name);
-               if (!test_EnumAccountsWithUserRight(p, mem_ctx, handle, (struct lsa_String *)&privs1.privs[i].name)) {
+               test_LookupPrivDisplayName(p, tctx, handle, (struct lsa_String *)&privs1.privs[i].name);
+               test_LookupPrivValue(p, tctx, handle, (struct lsa_String *)&privs1.privs[i].name);
+               if (!test_EnumAccountsWithUserRight(p, tctx, handle, (struct lsa_String *)&privs1.privs[i].name)) {
                        ret = false;
                }
        }
@@ -1704,8 +1755,8 @@ static bool test_EnumPrivs(struct dcerpc_pipe *p,
        return ret;
 }
 
-static bool test_QueryForestTrustInformation(struct dcerpc_pipe *p, 
-                                            struct torture_context *tctx, 
+static bool test_QueryForestTrustInformation(struct dcerpc_pipe *p,
+                                            struct torture_context *tctx,
                                             struct policy_handle *handle,
                                             const char *trusted_domain_name)
 {
@@ -1745,10 +1796,10 @@ static bool test_QueryForestTrustInformation(struct dcerpc_pipe *p,
        return ret;
 }
 
-static bool test_query_each_TrustDomEx(struct dcerpc_pipe *p, 
-                                      TALLOC_CTX *mem_ctx, 
-                                      struct policy_handle *handle, 
-                                      struct lsa_DomainListEx *domains) 
+static bool test_query_each_TrustDomEx(struct dcerpc_pipe *p,
+                                      struct torture_context *tctx,
+                                      struct policy_handle *handle,
+                                      struct lsa_DomainListEx *domains)
 {
        int i;
        bool ret = true;
@@ -1756,7 +1807,7 @@ static bool test_query_each_TrustDomEx(struct dcerpc_pipe *p,
        for (i=0; i< domains->count; i++) {
 
                if (domains->domains[i].trust_attributes & NETR_TRUST_ATTRIBUTE_FOREST_TRANSITIVE) {
-                       ret &= test_QueryForestTrustInformation(p, mem_ctx, handle, 
+                       ret &= test_QueryForestTrustInformation(p, tctx, handle,
                                                                domains->domains[i].domain_name.string);
                }
        }
@@ -1764,15 +1815,15 @@ static bool test_query_each_TrustDomEx(struct dcerpc_pipe *p,
        return ret;
 }
 
-static bool test_query_each_TrustDom(struct dcerpc_pipe *p, 
-                                    TALLOC_CTX *mem_ctx, 
-                                    struct policy_handle *handle, 
-                                    struct lsa_DomainList *domains) 
+static bool test_query_each_TrustDom(struct dcerpc_pipe *p,
+                                    struct torture_context *tctx,
+                                    struct policy_handle *handle,
+                                    struct lsa_DomainList *domains)
 {
        NTSTATUS status;
        int i,j;
        bool ret = true;
-               
+
        printf("\nTesting OpenTrustedDomain, OpenTrustedDomainByName and QueryInfoTrustedDomain\n");
        for (i=0; i< domains->count; i++) {
                struct lsa_OpenTrustedDomain trust;
@@ -1789,48 +1840,48 @@ static bool test_query_each_TrustDom(struct dcerpc_pipe *p,
                        trust.in.sid = domains->domains[i].sid;
                        trust.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
                        trust.out.trustdom_handle = &trustdom_handle;
-                       
-                       status = dcerpc_lsa_OpenTrustedDomain(p, mem_ctx, &trust);
-                       
+
+                       status = dcerpc_lsa_OpenTrustedDomain(p, tctx, &trust);
+
                        if (!NT_STATUS_IS_OK(status)) {
                                printf("OpenTrustedDomain failed - %s\n", nt_errstr(status));
                                return false;
                        }
-                       
+
                        c.in.handle = &trustdom_handle;
                        c.out.handle = &handle2;
-                       
+
                        c_trust.in.handle = &trustdom_handle;
                        c_trust.out.handle = &handle2;
-                       
+
                        for (j=0; j < ARRAY_SIZE(levels); j++) {
                                struct lsa_QueryTrustedDomainInfo q;
-                               union lsa_TrustedDomainInfo info;
+                               union lsa_TrustedDomainInfo *info = NULL;
                                q.in.trustdom_handle = &trustdom_handle;
                                q.in.level = levels[j];
                                q.out.info = &info;
-                               status = dcerpc_lsa_QueryTrustedDomainInfo(p, mem_ctx, &q);
+                               status = dcerpc_lsa_QueryTrustedDomainInfo(p, tctx, &q);
                                if (!NT_STATUS_IS_OK(status) && ok[j]) {
-                                       printf("QueryTrustedDomainInfo level %d failed - %s\n", 
+                                       printf("QueryTrustedDomainInfo level %d failed - %s\n",
                                               levels[j], nt_errstr(status));
                                        ret = false;
                                } else if (NT_STATUS_IS_OK(status) && !ok[j]) {
-                                       printf("QueryTrustedDomainInfo level %d unexpectedly succeeded - %s\n", 
+                                       printf("QueryTrustedDomainInfo level %d unexpectedly succeeded - %s\n",
                                               levels[j], nt_errstr(status));
                                        ret = false;
                                }
                        }
-                       
-                       status = dcerpc_lsa_CloseTrustedDomainEx(p, mem_ctx, &c_trust);
+
+                       status = dcerpc_lsa_CloseTrustedDomainEx(p, tctx, &c_trust);
                        if (!NT_STATUS_EQUAL(status, NT_STATUS_NOT_IMPLEMENTED)) {
                                printf("Expected CloseTrustedDomainEx to return NT_STATUS_NOT_IMPLEMENTED, instead - %s\n", nt_errstr(status));
                                return false;
                        }
-                       
+
                        c.in.handle = &trustdom_handle;
                        c.out.handle = &handle2;
-                       
-                       status = dcerpc_lsa_Close(p, mem_ctx, &c);
+
+                       status = dcerpc_lsa_Close(p, tctx, &c);
                        if (!NT_STATUS_IS_OK(status)) {
                                printf("Close of trusted domain failed - %s\n", nt_errstr(status));
                                return false;
@@ -1839,23 +1890,23 @@ static bool test_query_each_TrustDom(struct dcerpc_pipe *p,
                        for (j=0; j < ARRAY_SIZE(levels); j++) {
                                struct lsa_QueryTrustedDomainInfoBySid q;
                                union lsa_TrustedDomainInfo *info = NULL;
-                               
+
                                if (!domains->domains[i].sid) {
                                        continue;
                                }
-                               
+
                                q.in.handle  = handle;
                                q.in.dom_sid = domains->domains[i].sid;
                                q.in.level   = levels[j];
                                q.out.info   = &info;
 
-                               status = dcerpc_lsa_QueryTrustedDomainInfoBySid(p, mem_ctx, &q);
+                               status = dcerpc_lsa_QueryTrustedDomainInfoBySid(p, tctx, &q);
                                if (!NT_STATUS_IS_OK(status) && ok[j]) {
-                                       printf("QueryTrustedDomainInfoBySid level %d failed - %s\n", 
+                                       printf("QueryTrustedDomainInfoBySid level %d failed - %s\n",
                                               levels[j], nt_errstr(status));
                                        ret = false;
                                } else if (NT_STATUS_IS_OK(status) && !ok[j]) {
-                                       printf("QueryTrustedDomainInfoBySid level %d unexpectedly succeeded - %s\n", 
+                                       printf("QueryTrustedDomainInfoBySid level %d unexpectedly succeeded - %s\n",
                                               levels[j], nt_errstr(status));
                                        ret = false;
                                }
@@ -1866,9 +1917,9 @@ static bool test_query_each_TrustDom(struct dcerpc_pipe *p,
                trust_by_name.in.name.string = domains->domains[i].name.string;
                trust_by_name.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
                trust_by_name.out.trustdom_handle = &trustdom_handle;
-                       
-               status = dcerpc_lsa_OpenTrustedDomainByName(p, mem_ctx, &trust_by_name);
-                       
+
+               status = dcerpc_lsa_OpenTrustedDomainByName(p, tctx, &trust_by_name);
+
                if (!NT_STATUS_IS_OK(status)) {
                        printf("OpenTrustedDomainByName failed - %s\n", nt_errstr(status));
                        return false;
@@ -1876,26 +1927,26 @@ static bool test_query_each_TrustDom(struct dcerpc_pipe *p,
 
                for (j=0; j < ARRAY_SIZE(levels); j++) {
                        struct lsa_QueryTrustedDomainInfo q;
-                       union lsa_TrustedDomainInfo info;
+                       union lsa_TrustedDomainInfo *info = NULL;
                        q.in.trustdom_handle = &trustdom_handle;
                        q.in.level = levels[j];
                        q.out.info = &info;
-                       status = dcerpc_lsa_QueryTrustedDomainInfo(p, mem_ctx, &q);
+                       status = dcerpc_lsa_QueryTrustedDomainInfo(p, tctx, &q);
                        if (!NT_STATUS_IS_OK(status) && ok[j]) {
-                               printf("QueryTrustedDomainInfo level %d failed - %s\n", 
+                               printf("QueryTrustedDomainInfo level %d failed - %s\n",
                                       levels[j], nt_errstr(status));
                                ret = false;
                        } else if (NT_STATUS_IS_OK(status) && !ok[j]) {
-                               printf("QueryTrustedDomainInfo level %d unexpectedly succeeded - %s\n", 
+                               printf("QueryTrustedDomainInfo level %d unexpectedly succeeded - %s\n",
                                       levels[j], nt_errstr(status));
                                ret = false;
                        }
                }
-               
+
                c.in.handle = &trustdom_handle;
                c.out.handle = &handle2;
 
-               status = dcerpc_lsa_Close(p, mem_ctx, &c);
+               status = dcerpc_lsa_Close(p, tctx, &c);
                if (!NT_STATUS_IS_OK(status)) {
                        printf("Close of trusted domain failed - %s\n", nt_errstr(status));
                        return false;
@@ -1903,18 +1954,22 @@ static bool test_query_each_TrustDom(struct dcerpc_pipe *p,
 
                for (j=0; j < ARRAY_SIZE(levels); j++) {
                        struct lsa_QueryTrustedDomainInfoByName q;
-                       union lsa_TrustedDomainInfo info;
+                       union lsa_TrustedDomainInfo *info = NULL;
+                       struct lsa_String name;
+
+                       name.string = domains->domains[i].name.string;
+
                        q.in.handle         = handle;
-                       q.in.trusted_domain.string = domains->domains[i].name.string;
+                       q.in.trusted_domain = &name;
                        q.in.level          = levels[j];
                        q.out.info          = &info;
-                       status = dcerpc_lsa_QueryTrustedDomainInfoByName(p, mem_ctx, &q);
+                       status = dcerpc_lsa_QueryTrustedDomainInfoByName(p, tctx, &q);
                        if (!NT_STATUS_IS_OK(status) && ok[j]) {
-                               printf("QueryTrustedDomainInfoByName level %d failed - %s\n", 
+                               printf("QueryTrustedDomainInfoByName level %d failed - %s\n",
                                       levels[j], nt_errstr(status));
                                ret = false;
                        } else if (NT_STATUS_IS_OK(status) && !ok[j]) {
-                               printf("QueryTrustedDomainInfoByName level %d unexpectedly succeeded - %s\n", 
+                               printf("QueryTrustedDomainInfoByName level %d unexpectedly succeeded - %s\n",
                                       levels[j], nt_errstr(status));
                                ret = false;
                        }
@@ -1923,8 +1978,8 @@ static bool test_query_each_TrustDom(struct dcerpc_pipe *p,
        return ret;
 }
 
-static bool test_EnumTrustDom(struct dcerpc_pipe *p, 
-                             TALLOC_CTX *mem_ctx, 
+static bool test_EnumTrustDom(struct dcerpc_pipe *p,
+                             struct torture_context *tctx,
                              struct policy_handle *handle)
 {
        struct lsa_EnumTrustDom r;
@@ -1942,9 +1997,9 @@ static bool test_EnumTrustDom(struct dcerpc_pipe *p,
        r.in.max_size = 0;
        r.out.domains = &domains;
        r.out.resume_handle = &resume_handle;
-       
-       enum_status = dcerpc_lsa_EnumTrustDom(p, mem_ctx, &r);
-       
+
+       enum_status = dcerpc_lsa_EnumTrustDom(p, tctx, &r);
+
        if (NT_STATUS_IS_OK(enum_status)) {
                if (domains.count == 0) {
                        printf("EnumTrustDom failed - should have returned 'NT_STATUS_NO_MORE_ENTRIES' for 0 trusted domains\n");
@@ -1954,7 +2009,7 @@ static bool test_EnumTrustDom(struct dcerpc_pipe *p,
                printf("EnumTrustDom of zero size failed - %s\n", nt_errstr(enum_status));
                return false;
        }
-               
+
        /* Start from the bottom again */
        resume_handle = 0;
 
@@ -1964,9 +2019,9 @@ static bool test_EnumTrustDom(struct dcerpc_pipe *p,
                r.in.max_size = LSA_ENUM_TRUST_DOMAIN_MULTIPLIER * 3;
                r.out.domains = &domains;
                r.out.resume_handle = &resume_handle;
-               
-               enum_status = dcerpc_lsa_EnumTrustDom(p, mem_ctx, &r);
-               
+
+               enum_status = dcerpc_lsa_EnumTrustDom(p, tctx, &r);
+
                /* NO_MORE_ENTRIES is allowed */
                if (NT_STATUS_EQUAL(enum_status, NT_STATUS_NO_MORE_ENTRIES)) {
                        if (domains.count == 0) {
@@ -1979,7 +2034,7 @@ static bool test_EnumTrustDom(struct dcerpc_pipe *p,
                        if (r.out.domains->count < 3 || r.out.domains->count > 4) {
                                printf("EnumTrustDom didn't fill the buffer we "
                                       "asked it to (got %d, expected %d / %d == %d entries)\n",
-                                      r.out.domains->count, LSA_ENUM_TRUST_DOMAIN_MULTIPLIER * 3, 
+                                      r.out.domains->count, LSA_ENUM_TRUST_DOMAIN_MULTIPLIER * 3,
                                       LSA_ENUM_TRUST_DOMAIN_MULTIPLIER, r.in.max_size);
                                ret = false;
                        }
@@ -1987,14 +2042,14 @@ static bool test_EnumTrustDom(struct dcerpc_pipe *p,
                        printf("EnumTrustDom failed - %s\n", nt_errstr(enum_status));
                        return false;
                }
-               
+
                if (domains.count == 0) {
                        printf("EnumTrustDom failed - should have returned 'NT_STATUS_NO_MORE_ENTRIES' for 0 trusted domains\n");
                        return false;
                }
 
-               ret &= test_query_each_TrustDom(p, mem_ctx, handle, &domains);
-               
+               ret &= test_query_each_TrustDom(p, tctx, handle, &domains);
+
        } while ((NT_STATUS_EQUAL(enum_status, STATUS_MORE_ENTRIES)));
 
        printf("\nTesting EnumTrustedDomainsEx\n");
@@ -2004,14 +2059,14 @@ static bool test_EnumTrustDom(struct dcerpc_pipe *p,
        r_ex.in.max_size = LSA_ENUM_TRUST_DOMAIN_EX_MULTIPLIER * 3;
        r_ex.out.domains = &domains_ex;
        r_ex.out.resume_handle = &resume_handle;
-       
-       enum_status = dcerpc_lsa_EnumTrustedDomainsEx(p, mem_ctx, &r_ex);
-       
+
+       enum_status = dcerpc_lsa_EnumTrustedDomainsEx(p, tctx, &r_ex);
+
        if (!(NT_STATUS_EQUAL(enum_status, STATUS_MORE_ENTRIES) || NT_STATUS_EQUAL(enum_status, NT_STATUS_NO_MORE_ENTRIES))) {
                printf("EnumTrustedDomainEx of zero size failed - %s\n", nt_errstr(enum_status));
                return false;
        }
-               
+
        resume_handle = 0;
        do {
                r_ex.in.handle = handle;
@@ -2019,9 +2074,9 @@ static bool test_EnumTrustDom(struct dcerpc_pipe *p,
                r_ex.in.max_size = LSA_ENUM_TRUST_DOMAIN_EX_MULTIPLIER * 3;
                r_ex.out.domains = &domains_ex;
                r_ex.out.resume_handle = &resume_handle;
-               
-               enum_status = dcerpc_lsa_EnumTrustedDomainsEx(p, mem_ctx, &r_ex);
-               
+
+               enum_status = dcerpc_lsa_EnumTrustedDomainsEx(p, tctx, &r_ex);
+
                /* NO_MORE_ENTRIES is allowed */
                if (NT_STATUS_EQUAL(enum_status, NT_STATUS_NO_MORE_ENTRIES)) {
                        if (domains_ex.count == 0) {
@@ -2034,9 +2089,9 @@ static bool test_EnumTrustDom(struct dcerpc_pipe *p,
                        if (r_ex.out.domains->count < 3 || r_ex.out.domains->count > 4) {
                                printf("EnumTrustDom didn't fill the buffer we "
                                       "asked it to (got %d, expected %d / %d == %d entries)\n",
-                                      r_ex.out.domains->count, 
+                                      r_ex.out.domains->count,
                                       r_ex.in.max_size,
-                                      LSA_ENUM_TRUST_DOMAIN_EX_MULTIPLIER, 
+                                      LSA_ENUM_TRUST_DOMAIN_EX_MULTIPLIER,
                                       r_ex.in.max_size / LSA_ENUM_TRUST_DOMAIN_EX_MULTIPLIER);
                        }
                } else if (!NT_STATUS_IS_OK(enum_status)) {
@@ -2049,15 +2104,15 @@ static bool test_EnumTrustDom(struct dcerpc_pipe *p,
                        return false;
                }
 
-               ret &= test_query_each_TrustDomEx(p, mem_ctx, handle, &domains_ex);
-               
+               ret &= test_query_each_TrustDomEx(p, tctx, handle, &domains_ex);
+
        } while ((NT_STATUS_EQUAL(enum_status, STATUS_MORE_ENTRIES)));
 
        return ret;
 }
 
-static bool test_CreateTrustedDomain(struct dcerpc_pipe *p, 
-                                    TALLOC_CTX *mem_ctx, 
+static bool test_CreateTrustedDomain(struct dcerpc_pipe *p,
+                                    struct torture_context *tctx,
                                     struct policy_handle *handle)
 {
        NTSTATUS status;
@@ -2067,19 +2122,20 @@ static bool test_CreateTrustedDomain(struct dcerpc_pipe *p,
        struct dom_sid *domsid[12];
        struct policy_handle trustdom_handle[12];
        struct lsa_QueryTrustedDomainInfo q;
+       union lsa_TrustedDomainInfo *info = NULL;
        int i;
 
        printf("\nTesting CreateTrustedDomain for 12 domains\n");
 
-       if (!test_EnumTrustDom(p, mem_ctx, handle)) {
+       if (!test_EnumTrustDom(p, tctx, handle)) {
                ret = false;
        }
-       
+
        for (i=0; i< 12; i++) {
-               char *trust_name = talloc_asprintf(mem_ctx, "torturedom%02d", i);
-               char *trust_sid = talloc_asprintf(mem_ctx, "S-1-5-21-97398-379795-100%02d", i);
-               
-               domsid[i] = dom_sid_parse_talloc(mem_ctx, trust_sid);
+               char *trust_name = talloc_asprintf(tctx, "torturedom%02d", i);
+               char *trust_sid = talloc_asprintf(tctx, "S-1-5-21-97398-379795-100%02d", i);
+
+               domsid[i] = dom_sid_parse_talloc(tctx, trust_sid);
 
                trustinfo.sid = domsid[i];
                init_lsa_String((struct lsa_String *)&trustinfo.name, trust_name);
@@ -2088,44 +2144,45 @@ static bool test_CreateTrustedDomain(struct dcerpc_pipe *p,
                r.in.info = &trustinfo;
                r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
                r.out.trustdom_handle = &trustdom_handle[i];
-               
-               status = dcerpc_lsa_CreateTrustedDomain(p, mem_ctx, &r);
+
+               status = dcerpc_lsa_CreateTrustedDomain(p, tctx, &r);
                if (NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_NAME_COLLISION)) {
-                       test_DeleteTrustedDomain(p, mem_ctx, handle, trustinfo.name);
-                       status = dcerpc_lsa_CreateTrustedDomain(p, mem_ctx, &r);
+                       test_DeleteTrustedDomain(p, tctx, handle, trustinfo.name);
+                       status = dcerpc_lsa_CreateTrustedDomain(p, tctx, &r);
                }
                if (!NT_STATUS_IS_OK(status)) {
                        printf("CreateTrustedDomain failed - %s\n", nt_errstr(status));
                        ret = false;
                } else {
-               
+
                        q.in.trustdom_handle = &trustdom_handle[i];
                        q.in.level = LSA_TRUSTED_DOMAIN_INFO_INFO_EX;
-                       status = dcerpc_lsa_QueryTrustedDomainInfo(p, mem_ctx, &q);
+                       q.out.info = &info;
+                       status = dcerpc_lsa_QueryTrustedDomainInfo(p, tctx, &q);
                        if (!NT_STATUS_IS_OK(status)) {
                                printf("QueryTrustedDomainInfo level 1 failed - %s\n", nt_errstr(status));
                                ret = false;
                        } else if (!q.out.info) {
                                ret = false;
                        } else {
-                               if (strcmp(q.out.info->info_ex.netbios_name.string, trustinfo.name.string) != 0) {
+                               if (strcmp(info->info_ex.netbios_name.string, trustinfo.name.string) != 0) {
                                        printf("QueryTrustedDomainInfo returned inconsistant short name: %s != %s\n",
-                                              q.out.info->info_ex.netbios_name.string, trustinfo.name.string);
+                                              info->info_ex.netbios_name.string, trustinfo.name.string);
                                        ret = false;
                                }
-                               if (q.out.info->info_ex.trust_type != LSA_TRUST_TYPE_DOWNLEVEL) {
-                                       printf("QueryTrustedDomainInfo of %s returned incorrect trust type %d != %d\n", 
-                                              trust_name, q.out.info->info_ex.trust_type, LSA_TRUST_TYPE_DOWNLEVEL);
+                               if (info->info_ex.trust_type != LSA_TRUST_TYPE_DOWNLEVEL) {
+                                       printf("QueryTrustedDomainInfo of %s returned incorrect trust type %d != %d\n",
+                                              trust_name, info->info_ex.trust_type, LSA_TRUST_TYPE_DOWNLEVEL);
                                        ret = false;
                                }
-                               if (q.out.info->info_ex.trust_attributes != 0) {
-                                       printf("QueryTrustedDomainInfo of %s returned incorrect trust attributes %d != %d\n", 
-                                              trust_name, q.out.info->info_ex.trust_attributes, 0);
+                               if (info->info_ex.trust_attributes != 0) {
+                                       printf("QueryTrustedDomainInfo of %s returned incorrect trust attributes %d != %d\n",
+                                              trust_name, info->info_ex.trust_attributes, 0);
                                        ret = false;
                                }
-                               if (q.out.info->info_ex.trust_direction != LSA_TRUST_DIRECTION_OUTBOUND) {
-                                       printf("QueryTrustedDomainInfo of %s returned incorrect trust direction %d != %d\n", 
-                                              trust_name, q.out.info->info_ex.trust_direction, LSA_TRUST_DIRECTION_OUTBOUND);
+                               if (info->info_ex.trust_direction != LSA_TRUST_DIRECTION_OUTBOUND) {
+                                       printf("QueryTrustedDomainInfo of %s returned incorrect trust direction %d != %d\n",
+                                              trust_name, info->info_ex.trust_direction, LSA_TRUST_DIRECTION_OUTBOUND);
                                        ret = false;
                                }
                        }
@@ -2133,12 +2190,12 @@ static bool test_CreateTrustedDomain(struct dcerpc_pipe *p,
        }
 
        /* now that we have some domains to look over, we can test the enum calls */
-       if (!test_EnumTrustDom(p, mem_ctx, handle)) {
+       if (!test_EnumTrustDom(p, tctx, handle)) {
                ret = false;
        }
-       
+
        for (i=0; i<12; i++) {
-               if (!test_DeleteTrustedDomainBySid(p, mem_ctx, handle, domsid[i])) {
+               if (!test_DeleteTrustedDomainBySid(p, tctx, handle, domsid[i])) {
                        ret = false;
                }
        }
@@ -2146,9 +2203,8 @@ static bool test_CreateTrustedDomain(struct dcerpc_pipe *p,
        return ret;
 }
 
-static bool test_CreateTrustedDomainEx2(struct dcerpc_pipe *p, 
-                                       struct torture_context *tctx, 
-                                       TALLOC_CTX *mem_ctx, 
+static bool test_CreateTrustedDomainEx2(struct dcerpc_pipe *p,
+                                       struct torture_context *tctx,
                                        struct policy_handle *handle)
 {
        NTSTATUS status;
@@ -2161,6 +2217,7 @@ static bool test_CreateTrustedDomainEx2(struct dcerpc_pipe *p,
        struct dom_sid *domsid[12];
        struct policy_handle trustdom_handle[12];
        struct lsa_QueryTrustedDomainInfo q;
+       union lsa_TrustedDomainInfo *info = NULL;
        DATA_BLOB session_key;
        enum ndr_err_code ndr_err;
        int i;
@@ -2172,13 +2229,13 @@ static bool test_CreateTrustedDomainEx2(struct dcerpc_pipe *p,
                printf("dcerpc_fetch_session_key failed - %s\n", nt_errstr(status));
                return false;
        }
-               
+
        for (i=0; i< 12; i++) {
-               char *trust_name = talloc_asprintf(mem_ctx, "torturedom%02d", i);
-               char *trust_name_dns = talloc_asprintf(mem_ctx, "torturedom%02d.samba.example.com", i);
-               char *trust_sid = talloc_asprintf(mem_ctx, "S-1-5-21-97398-379795-100%02d", i);
-               
-               domsid[i] = dom_sid_parse_talloc(mem_ctx, trust_sid);
+               char *trust_name = talloc_asprintf(tctx, "torturedom%02d", i);
+               char *trust_name_dns = talloc_asprintf(tctx, "torturedom%02d.samba.example.com", i);
+               char *trust_sid = talloc_asprintf(tctx, "S-1-5-21-97398-379795-100%02d", i);
+
+               domsid[i] = dom_sid_parse_talloc(tctx, trust_sid);
 
                trustinfo.sid = domsid[i];
                trustinfo.netbios_name.string = trust_name;
@@ -2203,7 +2260,7 @@ static bool test_CreateTrustedDomainEx2(struct dcerpc_pipe *p,
                auth_struct.outgoing.count = 0;
                auth_struct.incoming.count = 0;
 
-               ndr_err = ndr_push_struct_blob(&auth_blob, mem_ctx, lp_iconv_convenience(tctx->lp_ctx), &auth_struct,
+               ndr_err = ndr_push_struct_blob(&auth_blob, tctx, lp_iconv_convenience(tctx->lp_ctx), &auth_struct,
                                               (ndr_push_flags_fn_t)ndr_push_trustDomainPasswords);
                if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
                        printf("ndr_push_struct_blob of trustDomainPasswords structure failed");
@@ -2220,20 +2277,21 @@ static bool test_CreateTrustedDomainEx2(struct dcerpc_pipe *p,
                r.in.auth_info = &authinfo;
                r.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
                r.out.trustdom_handle = &trustdom_handle[i];
-               
-               status = dcerpc_lsa_CreateTrustedDomainEx2(p, mem_ctx, &r);
+
+               status = dcerpc_lsa_CreateTrustedDomainEx2(p, tctx, &r);
                if (NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_NAME_COLLISION)) {
-                       test_DeleteTrustedDomain(p, mem_ctx, handle, trustinfo.netbios_name);
-                       status = dcerpc_lsa_CreateTrustedDomainEx2(p, mem_ctx, &r);
+                       test_DeleteTrustedDomain(p, tctx, handle, trustinfo.netbios_name);
+                       status = dcerpc_lsa_CreateTrustedDomainEx2(p, tctx, &r);
                }
                if (!NT_STATUS_IS_OK(status)) {
                        printf("CreateTrustedDomainEx failed2 - %s\n", nt_errstr(status));
                        ret = false;
                } else {
-               
+
                        q.in.trustdom_handle = &trustdom_handle[i];
                        q.in.level = LSA_TRUSTED_DOMAIN_INFO_INFO_EX;
-                       status = dcerpc_lsa_QueryTrustedDomainInfo(p, mem_ctx, &q);
+                       q.out.info = &info;
+                       status = dcerpc_lsa_QueryTrustedDomainInfo(p, tctx, &q);
                        if (!NT_STATUS_IS_OK(status)) {
                                printf("QueryTrustedDomainInfo level 1 failed - %s\n", nt_errstr(status));
                                ret = false;
@@ -2241,24 +2299,24 @@ static bool test_CreateTrustedDomainEx2(struct dcerpc_pipe *p,
                                printf("QueryTrustedDomainInfo level 1 failed to return an info pointer\n");
                                ret = false;
                        } else {
-                               if (strcmp(q.out.info->info_ex.netbios_name.string, trustinfo.netbios_name.string) != 0) {
+                               if (strcmp(info->info_ex.netbios_name.string, trustinfo.netbios_name.string) != 0) {
                                        printf("QueryTrustedDomainInfo returned inconsistant short name: %s != %s\n",
-                                              q.out.info->info_ex.netbios_name.string, trustinfo.netbios_name.string);
+                                              info->info_ex.netbios_name.string, trustinfo.netbios_name.string);
                                        ret = false;
                                }
-                               if (q.out.info->info_ex.trust_type != trustinfo.trust_type) {
-                                       printf("QueryTrustedDomainInfo of %s returned incorrect trust type %d != %d\n", 
-                                              trust_name, q.out.info->info_ex.trust_type, trustinfo.trust_type);
+                               if (info->info_ex.trust_type != trustinfo.trust_type) {
+                                       printf("QueryTrustedDomainInfo of %s returned incorrect trust type %d != %d\n",
+                                              trust_name, info->info_ex.trust_type, trustinfo.trust_type);
                                        ret = false;
                                }
-                               if (q.out.info->info_ex.trust_attributes != LSA_TRUST_ATTRIBUTE_USES_RC4_ENCRYPTION) {
-                                       printf("QueryTrustedDomainInfo of %s returned incorrect trust attributes %d != %d\n", 
-                                              trust_name, q.out.info->info_ex.trust_attributes, LSA_TRUST_ATTRIBUTE_USES_RC4_ENCRYPTION);
+                               if (info->info_ex.trust_attributes != LSA_TRUST_ATTRIBUTE_USES_RC4_ENCRYPTION) {
+                                       printf("QueryTrustedDomainInfo of %s returned incorrect trust attributes %d != %d\n",
+                                              trust_name, info->info_ex.trust_attributes, LSA_TRUST_ATTRIBUTE_USES_RC4_ENCRYPTION);
                                        ret = false;
                                }
-                               if (q.out.info->info_ex.trust_direction != trustinfo.trust_direction) {
-                                       printf("QueryTrustedDomainInfo of %s returned incorrect trust direction %d != %d\n", 
-                                              trust_name, q.out.info->info_ex.trust_direction, trustinfo.trust_direction);
+                               if (info->info_ex.trust_direction != trustinfo.trust_direction) {
+                                       printf("QueryTrustedDomainInfo of %s returned incorrect trust direction %d != %d\n",
+                                              trust_name, info->info_ex.trust_direction, trustinfo.trust_direction);
                                        ret = false;
                                }
                        }
@@ -2266,13 +2324,13 @@ static bool test_CreateTrustedDomainEx2(struct dcerpc_pipe *p,
        }
 
        /* now that we have some domains to look over, we can test the enum calls */
-       if (!test_EnumTrustDom(p, mem_ctx, handle)) {
+       if (!test_EnumTrustDom(p, tctx, handle)) {
                printf("test_EnumTrustDom failed\n");
                ret = false;
        }
-       
+
        for (i=0; i<12; i++) {
-               if (!test_DeleteTrustedDomainBySid(p, mem_ctx, handle, domsid[i])) {
+               if (!test_DeleteTrustedDomainBySid(p, tctx, handle, domsid[i])) {
                        printf("test_DeleteTrustedDomainBySid failed\n");
                        ret = false;
                }
@@ -2281,11 +2339,12 @@ static bool test_CreateTrustedDomainEx2(struct dcerpc_pipe *p,
        return ret;
 }
 
-static bool test_QueryDomainInfoPolicy(struct dcerpc_pipe *p, 
-                                struct torture_context *tctx, 
+static bool test_QueryDomainInfoPolicy(struct dcerpc_pipe *p,
+                                struct torture_context *tctx,
                                 struct policy_handle *handle)
 {
        struct lsa_QueryDomainInformationPolicy r;
+       union lsa_DomainInformationPolicy *info = NULL;
        NTSTATUS status;
        int i;
        bool ret = true;
@@ -2295,6 +2354,7 @@ static bool test_QueryDomainInfoPolicy(struct dcerpc_pipe *p,
        for (i=2;i<4;i++) {
                r.in.handle = handle;
                r.in.level = i;
+               r.out.info = &info;
 
                printf("\nTrying QueryDomainInformationPolicy level %d\n", i);
 
@@ -2320,6 +2380,7 @@ static bool test_QueryInfoPolicyCalls(    bool version2,
                                        struct policy_handle *handle)
 {
        struct lsa_QueryInfoPolicy r;
+       union lsa_PolicyInformation *info = NULL;
        NTSTATUS status;
        int i;
        bool ret = true;
@@ -2332,6 +2393,7 @@ static bool test_QueryInfoPolicyCalls(    bool version2,
        for (i=1;i<=14;i++) {
                r.in.handle = handle;
                r.in.level = i;
+               r.out.info = &info;
 
                if (version2)
                        printf("\nTrying QueryInfoPolicy2 level %d\n", i);
@@ -2401,33 +2463,33 @@ static bool test_QueryInfoPolicyCalls(  bool version2,
                        struct lsa_TransNameArray tnames;
                        tnames.count = 14;
                        tnames.names = talloc_zero_array(tctx, struct lsa_TranslatedName, tnames.count);
-                       tnames.names[0].name.string = r.out.info->dns.name.string;
+                       tnames.names[0].name.string = info->dns.name.string;
                        tnames.names[0].sid_type = SID_NAME_DOMAIN;
-                       tnames.names[1].name.string = r.out.info->dns.dns_domain.string;
+                       tnames.names[1].name.string = info->dns.dns_domain.string;
                        tnames.names[1].sid_type = SID_NAME_DOMAIN;
-                       tnames.names[2].name.string = talloc_asprintf(tctx, "%s\\", r.out.info->dns.name.string);
+                       tnames.names[2].name.string = talloc_asprintf(tctx, "%s\\", info->dns.name.string);
                        tnames.names[2].sid_type = SID_NAME_DOMAIN;
-                       tnames.names[3].name.string = talloc_asprintf(tctx, "%s\\", r.out.info->dns.dns_domain.string);
+                       tnames.names[3].name.string = talloc_asprintf(tctx, "%s\\", info->dns.dns_domain.string);
                        tnames.names[3].sid_type = SID_NAME_DOMAIN;
-                       tnames.names[4].name.string = talloc_asprintf(tctx, "%s\\guest", r.out.info->dns.name.string);
+                       tnames.names[4].name.string = talloc_asprintf(tctx, "%s\\guest", info->dns.name.string);
                        tnames.names[4].sid_type = SID_NAME_USER;
-                       tnames.names[5].name.string = talloc_asprintf(tctx, "%s\\krbtgt", r.out.info->dns.name.string);
+                       tnames.names[5].name.string = talloc_asprintf(tctx, "%s\\krbtgt", info->dns.name.string);
                        tnames.names[5].sid_type = SID_NAME_USER;
-                       tnames.names[6].name.string = talloc_asprintf(tctx, "%s\\guest", r.out.info->dns.dns_domain.string);
+                       tnames.names[6].name.string = talloc_asprintf(tctx, "%s\\guest", info->dns.dns_domain.string);
                        tnames.names[6].sid_type = SID_NAME_USER;
-                       tnames.names[7].name.string = talloc_asprintf(tctx, "%s\\krbtgt", r.out.info->dns.dns_domain.string);
+                       tnames.names[7].name.string = talloc_asprintf(tctx, "%s\\krbtgt", info->dns.dns_domain.string);
                        tnames.names[7].sid_type = SID_NAME_USER;
-                       tnames.names[8].name.string = talloc_asprintf(tctx, "krbtgt@%s", r.out.info->dns.name.string);
+                       tnames.names[8].name.string = talloc_asprintf(tctx, "krbtgt@%s", info->dns.name.string);
                        tnames.names[8].sid_type = SID_NAME_USER;
-                       tnames.names[9].name.string = talloc_asprintf(tctx, "krbtgt@%s", r.out.info->dns.dns_domain.string);
+                       tnames.names[9].name.string = talloc_asprintf(tctx, "krbtgt@%s", info->dns.dns_domain.string);
                        tnames.names[9].sid_type = SID_NAME_USER;
-                       tnames.names[10].name.string = talloc_asprintf(tctx, "%s\\"TEST_MACHINENAME "$", r.out.info->dns.name.string);
+                       tnames.names[10].name.string = talloc_asprintf(tctx, "%s\\"TEST_MACHINENAME "$", info->dns.name.string);
                        tnames.names[10].sid_type = SID_NAME_USER;
-                       tnames.names[11].name.string = talloc_asprintf(tctx, "%s\\"TEST_MACHINENAME "$", r.out.info->dns.dns_domain.string);
+                       tnames.names[11].name.string = talloc_asprintf(tctx, "%s\\"TEST_MACHINENAME "$", info->dns.dns_domain.string);
                        tnames.names[11].sid_type = SID_NAME_USER;
-                       tnames.names[12].name.string = talloc_asprintf(tctx, TEST_MACHINENAME "$@%s", r.out.info->dns.name.string);
+                       tnames.names[12].name.string = talloc_asprintf(tctx, TEST_MACHINENAME "$@%s", info->dns.name.string);
                        tnames.names[12].sid_type = SID_NAME_USER;
-                       tnames.names[13].name.string = talloc_asprintf(tctx, TEST_MACHINENAME "$@%s", r.out.info->dns.dns_domain.string);
+                       tnames.names[13].name.string = talloc_asprintf(tctx, TEST_MACHINENAME "$@%s", info->dns.dns_domain.string);
                        tnames.names[13].sid_type = SID_NAME_USER;
                        ret &= test_LookupNames(p, tctx, handle, &tnames);
 
@@ -2437,21 +2499,22 @@ static bool test_QueryInfoPolicyCalls(  bool version2,
        return ret;
 }
 
-static bool test_QueryInfoPolicy(struct dcerpc_pipe *p, 
-                                struct torture_context *tctx, 
+static bool test_QueryInfoPolicy(struct dcerpc_pipe *p,
+                                struct torture_context *tctx,
                                 struct policy_handle *handle)
 {
        return test_QueryInfoPolicyCalls(false, p, tctx, handle);
 }
 
-static bool test_QueryInfoPolicy2(struct dcerpc_pipe *p, 
-                                 struct torture_context *tctx, 
+static bool test_QueryInfoPolicy2(struct dcerpc_pipe *p,
+                                 struct torture_context *tctx,
                                  struct policy_handle *handle)
 {
        return test_QueryInfoPolicyCalls(true, p, tctx, handle);
 }
 
-static bool test_GetUserName(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
+static bool test_GetUserName(struct dcerpc_pipe *p,
+                            struct torture_context *tctx)
 {
        struct lsa_GetUserName r;
        NTSTATUS status;
@@ -2466,7 +2529,7 @@ static bool test_GetUserName(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
        r.in.authority_name     = NULL;
        r.out.account_name      = &account_name_p;
 
-       status = dcerpc_lsa_GetUserName(p, mem_ctx, &r);
+       status = dcerpc_lsa_GetUserName(p, tctx, &r);
 
        if (!NT_STATUS_IS_OK(status)) {
                printf("GetUserName failed - %s\n", nt_errstr(status));
@@ -2478,7 +2541,7 @@ static bool test_GetUserName(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
        r.in.authority_name     = &authority_name_p;
        r.out.account_name      = &account_name_p;
 
-       status = dcerpc_lsa_GetUserName(p, mem_ctx, &r);
+       status = dcerpc_lsa_GetUserName(p, tctx, &r);
 
        if (!NT_STATUS_IS_OK(status)) {
                printf("GetUserName failed - %s\n", nt_errstr(status));
@@ -2488,8 +2551,8 @@ static bool test_GetUserName(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx)
        return ret;
 }
 
-bool test_lsa_Close(struct dcerpc_pipe *p, 
-                   TALLOC_CTX *mem_ctx, 
+bool test_lsa_Close(struct dcerpc_pipe *p,
+                   struct torture_context *tctx,
                    struct policy_handle *handle)
 {
        NTSTATUS status;
@@ -2501,13 +2564,13 @@ bool test_lsa_Close(struct dcerpc_pipe *p,
        r.in.handle = handle;
        r.out.handle = &handle2;
 
-       status = dcerpc_lsa_Close(p, mem_ctx, &r);
+       status = dcerpc_lsa_Close(p, tctx, &r);
        if (!NT_STATUS_IS_OK(status)) {
                printf("Close failed - %s\n", nt_errstr(status));
                return false;
        }
 
-       status = dcerpc_lsa_Close(p, mem_ctx, &r);
+       status = dcerpc_lsa_Close(p, tctx, &r);
        /* its really a fault - we need a status code for rpc fault */
        if (!NT_STATUS_EQUAL(status, NT_STATUS_NET_WRITE_FAULT)) {
                printf("Close failed - %s\n", nt_errstr(status));
@@ -2548,11 +2611,11 @@ bool torture_rpc_lsa(struct torture_context *tctx)
                }
                if (!test_LookupNames_wellknown(p, tctx, handle)) {
                        ret = false;
-               }               
+               }
 
                if (!test_LookupNames_bogus(p, tctx, handle)) {
                        ret = false;
-               }               
+               }
 
                if (!test_LookupSids_async(p, tctx, handle)) {
                        ret = false;
@@ -2561,50 +2624,31 @@ bool torture_rpc_lsa(struct torture_context *tctx)
                if (!test_QueryDomainInfoPolicy(p, tctx, handle)) {
                        ret = false;
                }
-               
-               if (!test_CreateAccount(p, tctx, handle)) {
-                       ret = false;
-               }
-               
-               if (!test_CreateSecret(p, tctx, handle)) {
-                       ret = false;
-               }
-               if (!test_CreateTrustedDomain(p, tctx, handle)) {
-                       ret = false;
-               }
 
-               if (!test_CreateTrustedDomainEx2(p, tctx, tctx, handle)) {
+               if (!test_CreateSecret(p, tctx, handle)) {
                        ret = false;
                }
 
-               if (!test_EnumAccounts(p, tctx, handle)) {
-                       ret = false;
-               }
-               
-               if (!test_EnumPrivs(p, tctx, handle)) {
-                       ret = false;
-               }
-               
                if (!test_QueryInfoPolicy(p, tctx, handle)) {
                        ret = false;
                }
-               
+
                if (!test_QueryInfoPolicy2(p, tctx, handle)) {
                        ret = false;
                }
-               
+
                if (!test_Delete(p, tctx, handle)) {
                        ret = false;
                }
-               
+
                if (!test_many_LookupSids(p, tctx, handle)) {
                        ret = false;
                }
-               
+
                if (!test_lsa_Close(p, tctx, handle)) {
                        ret = false;
                }
-               
+
                torture_leave_domain(tctx, join);
 
        } else {
@@ -2620,26 +2664,189 @@ bool torture_rpc_lsa(struct torture_context *tctx)
        return ret;
 }
 
-bool torture_rpc_lsa_get_user(struct torture_context *torture)
+bool torture_rpc_lsa_get_user(struct torture_context *tctx)
 {
         NTSTATUS status;
         struct dcerpc_pipe *p;
-       TALLOC_CTX *mem_ctx;
        bool ret = true;
 
-       mem_ctx = talloc_init("torture_rpc_lsa_get_user");
-
-       status = torture_rpc_connection(torture, &p, &ndr_table_lsarpc);
+       status = torture_rpc_connection(tctx, &p, &ndr_table_lsarpc);
        if (!NT_STATUS_IS_OK(status)) {
-               talloc_free(mem_ctx);
                return false;
        }
 
-       if (!test_GetUserName(p, mem_ctx)) {
+       if (!test_GetUserName(p, tctx)) {
+               ret = false;
+       }
+
+       return ret;
+}
+
+static bool testcase_LookupNames(struct torture_context *tctx,
+                                struct dcerpc_pipe *p)
+{
+       bool ret = true;
+       struct policy_handle *handle;
+       struct lsa_TransNameArray tnames;
+       struct lsa_TransNameArray2 tnames2;
+
+       if (!test_OpenPolicy(p, tctx)) {
+               ret = false;
+       }
+
+       if (!test_lsa_OpenPolicy2(p, tctx, &handle)) {
+               ret = false;
+       }
+
+       if (!handle) {
+               ret = false;
+       }
+
+       tnames.count = 1;
+       tnames.names = talloc_array(tctx, struct lsa_TranslatedName, tnames.count);
+       ZERO_STRUCT(tnames.names[0]);
+       tnames.names[0].name.string = "BUILTIN";
+       tnames.names[0].sid_type = SID_NAME_DOMAIN;
+
+       if (!test_LookupNames(p, tctx, handle, &tnames)) {
+               ret = false;
+       }
+
+       tnames2.count = 1;
+       tnames2.names = talloc_array(tctx, struct lsa_TranslatedName2, tnames2.count);
+       ZERO_STRUCT(tnames2.names[0]);
+       tnames2.names[0].name.string = "BUILTIN";
+       tnames2.names[0].sid_type = SID_NAME_DOMAIN;
+
+       if (!test_LookupNames2(p, tctx, handle, &tnames2, true)) {
+               ret = false;
+       }
+
+       if (!test_LookupNames3(p, tctx, handle, &tnames2, true)) {
+               ret = false;
+       }
+
+       if (!test_lsa_Close(p, tctx, handle)) {
+               ret = false;
+       }
+
+       return ret;
+}
+
+struct torture_suite *torture_rpc_lsa_lookup_names(TALLOC_CTX *mem_ctx)
+{
+       struct torture_suite *suite;
+       struct torture_rpc_tcase *tcase;
+
+       suite = torture_suite_create(mem_ctx, "LSA-LOOKUPNAMES");
+
+       tcase = torture_suite_add_rpc_iface_tcase(suite, "lsa",
+                                                 &ndr_table_lsarpc);
+       torture_rpc_tcase_add_test(tcase, "LookupNames",
+                                  testcase_LookupNames);
+
+       return suite;
+}
+
+static bool testcase_TrustedDomains(struct torture_context *tctx,
+                                   struct dcerpc_pipe *p)
+{
+       bool ret = true;
+       struct policy_handle *handle;
+
+       if (!test_OpenPolicy(p, tctx)) {
+               ret = false;
+       }
+
+       if (!test_lsa_OpenPolicy2(p, tctx, &handle)) {
+               ret = false;
+       }
+
+       if (!handle) {
+               ret = false;
+       }
+
+       if (!test_CreateTrustedDomain(p, tctx, handle)) {
+               ret = false;
+       }
+
+       if (!test_CreateTrustedDomainEx2(p, tctx, handle)) {
+               ret = false;
+       }
+
+       if (!test_lsa_Close(p, tctx, handle)) {
+               ret = false;
+       }
+
+       return ret;
+}
+
+struct torture_suite *torture_rpc_lsa_trusted_domains(TALLOC_CTX *mem_ctx)
+{
+       struct torture_suite *suite;
+       struct torture_rpc_tcase *tcase;
+
+       suite = torture_suite_create(mem_ctx, "LSA-TRUSTED-DOMAINS");
+
+       tcase = torture_suite_add_machine_workstation_rpc_iface_tcase(suite, "lsa",
+                                                                     &ndr_table_lsarpc,
+                                                                     TEST_MACHINENAME);
+       torture_rpc_tcase_add_test(tcase, "TrustedDomains",
+                                  testcase_TrustedDomains);
+
+       return suite;
+}
+
+static bool testcase_Privileges(struct torture_context *tctx,
+                               struct dcerpc_pipe *p)
+{
+       bool ret = true;
+       struct policy_handle *handle;
+
+       if (!test_OpenPolicy(p, tctx)) {
+               ret = false;
+       }
+
+       if (!test_lsa_OpenPolicy2(p, tctx, &handle)) {
+               ret = false;
+       }
+
+       if (!handle) {
+               ret = false;
+       }
+
+       if (!test_CreateAccount(p, tctx, handle)) {
+               ret = false;
+       }
+
+       if (!test_EnumAccounts(p, tctx, handle)) {
+               ret = false;
+       }
+
+       if (!test_EnumPrivs(p, tctx, handle)) {
+               ret = false;
+       }
+
+       if (!test_lsa_Close(p, tctx, handle)) {
                ret = false;
        }
-               
-       talloc_free(mem_ctx);
 
        return ret;
 }
+
+
+struct torture_suite *torture_rpc_lsa_privileges(TALLOC_CTX *mem_ctx)
+{
+       struct torture_suite *suite;
+       struct torture_rpc_tcase *tcase;
+
+       suite = torture_suite_create(mem_ctx, "LSA-PRIVILEGES");
+
+       tcase = torture_suite_add_machine_workstation_rpc_iface_tcase(suite, "lsa",
+                                                                     &ndr_table_lsarpc,
+                                                                     TEST_MACHINENAME);
+       torture_rpc_tcase_add_test(tcase, "Privileges",
+                                  testcase_Privileges);
+
+       return suite;
+}