Changeset 740 for vendor/current/source4/torture/rpc/lsa_lookup.c

Timestamp:

Nov 14, 2012, 12:59:34 PM (12 years ago)

Author:

Silvan Scherrer

Message:

Samba Server: update vendor to 3.6.0

File:

• vendor/current/source4/torture/rpc/lsa_lookup.c (modified) (26 diffs)

vendor/current/source4/torture/rpc/lsa_lookup.c ¶

@@ -20 +20 @@
 
 #include "includes.h"
-#include "torture/torture.h"
-#include "lib/events/events.h"
-#include "libnet/libnet_join.h"
-#include "torture/rpc/rpc.h"
+#include "torture/rpc/torture_rpc.h"
 #include "librpc/gen_ndr/ndr_lsa_c.h"
 #include "libcli/security/security.h"
 
-static bool open_policy(TALLOC_CTX *mem_ctx, struct dcerpc_pipe *p,
+static bool open_policy(struct torture_context *tctx,
+                        struct dcerpc_binding_handle *b,
                         struct policy_handle **handle)
 {
@@ -33 +31 @@
         struct lsa_QosInfo qos;
         struct lsa_OpenPolicy2 r;
-        NTSTATUS status;
-
-        *handle = talloc(mem_ctx, struct policy_handle);
+
+        *handle = talloc(tctx, struct policy_handle);
         if (!*handle) {
                 return false;
@@ -57 +54 @@
         r.out.handle = *handle;
 
-        status = dcerpc_lsa_OpenPolicy2(p, mem_ctx, &r);
-
-        return NT_STATUS_IS_OK(status);
-}
-
-static bool get_domainsid(TALLOC_CTX *mem_ctx, struct dcerpc_pipe *p,
+        torture_assert_ntstatus_ok(tctx, dcerpc_lsa_OpenPolicy2_r(b, tctx, &r),
+                "OpenPolicy2 failed");
+
+        return NT_STATUS_IS_OK(r.out.result);
+}
+
+static bool get_domainsid(struct torture_context *tctx,
+                          struct dcerpc_binding_handle *b,
                           struct policy_handle *handle,
                           struct dom_sid **sid)
@@ -68 +67 @@
         struct lsa_QueryInfoPolicy r;
         union lsa_PolicyInformation *info = NULL;
-        NTSTATUS status;
 
         r.in.level = LSA_POLICY_INFO_DOMAIN;
@@ -74 +72 @@
         r.out.info = &info;
 
-        status = dcerpc_lsa_QueryInfoPolicy(p, mem_ctx, &r);
-        if (!NT_STATUS_IS_OK(status)) return false;
+        torture_assert_ntstatus_ok(tctx, dcerpc_lsa_QueryInfoPolicy_r(b, tctx, &r),
+                "QueryInfoPolicy failed");
+        torture_assert_ntstatus_ok(tctx, r.out.result, "QueryInfoPolicy failed");
 
         *sid = info->domain.sid;
@@ -81 +80 @@
 }
 
-static NTSTATUS lookup_sids(TALLOC_CTX *mem_ctx, uint16_t level,
-                            struct dcerpc_pipe *p,
+static NTSTATUS lookup_sids(struct torture_context *tctx,
+                            uint16_t level,
+                            struct dcerpc_binding_handle *b,
                             struct policy_handle *handle,
                             struct dom_sid **sids, uint32_t num_sids,
@@ -92 +92 @@
         uint32_t count = 0;
         uint32_t i;
+        NTSTATUS status;
 
         names->count = 0;
@@ -97 +98 @@
 
         sidarray.num_sids = num_sids;
-        sidarray.sids = talloc_array(mem_ctx, struct lsa_SidPtr, num_sids);
+        sidarray.sids = talloc_array(tctx, struct lsa_SidPtr, num_sids);
 
         for (i=0; i<num_sids; i++) {
@@ -112 +113 @@
         r.out.domains = &domains;
 
-        return dcerpc_lsa_LookupSids(p, mem_ctx, &r);
-}
-
-static const char *sid_type_lookup(enum lsa_SidType r)
-{
-        switch (r) {
-                case SID_NAME_USE_NONE: return "SID_NAME_USE_NONE"; break;
-                case SID_NAME_USER: return "SID_NAME_USER"; break;
-                case SID_NAME_DOM_GRP: return "SID_NAME_DOM_GRP"; break;
-                case SID_NAME_DOMAIN: return "SID_NAME_DOMAIN"; break;
-                case SID_NAME_ALIAS: return "SID_NAME_ALIAS"; break;
-                case SID_NAME_WKN_GRP: return "SID_NAME_WKN_GRP"; break;
-                case SID_NAME_DELETED: return "SID_NAME_DELETED"; break;
-                case SID_NAME_INVALID: return "SID_NAME_INVALID"; break;
-                case SID_NAME_UNKNOWN: return "SID_NAME_UNKNOWN"; break;
-                case SID_NAME_COMPUTER: return "SID_NAME_COMPUTER"; break;
-        }
-        return "Invalid sid type\n";
-}
-
-static bool test_lookupsids(TALLOC_CTX *mem_ctx, struct dcerpc_pipe *p,
+        status = dcerpc_lsa_LookupSids_r(b, tctx, &r);
+        if (!NT_STATUS_IS_OK(status)) {
+                return status;
+        }
+        return r.out.result;
+}
+
+static bool test_lookupsids(struct torture_context *tctx,
+                            struct dcerpc_binding_handle *b,
                             struct policy_handle *handle,
                             struct dom_sid **sids, uint32_t num_sids,
@@ -143 +132 @@
         bool ret = true;
 
-        status = lookup_sids(mem_ctx, level, p, handle, sids, num_sids,
+        status = lookup_sids(tctx, level, b, handle, sids, num_sids,
                              &names);
         if (!NT_STATUS_EQUAL(status, expected_result)) {
@@ -161 +150 @@
                         printf("In level %d, for sid %s expected %s, "
                                "got %s\n", level,
-                               dom_sid_string(mem_ctx, sids[i]),
+                               dom_sid_string(tctx, sids[i]),
                                sid_type_lookup(types[i]),
                                sid_type_lookup(names.names[i].sid_type));
@@ -170 +159 @@
 }
 
-static bool get_downleveltrust(struct torture_context *tctx, struct dcerpc_pipe *p,
+static bool get_downleveltrust(struct torture_context *tctx, struct dcerpc_binding_handle *b,
                                struct policy_handle *handle,
                                struct dom_sid **sid)
@@ -177 +166 @@
         uint32_t resume_handle = 0;
         struct lsa_DomainList domains;
-        NTSTATUS status;
         int i;
 
@@ -186 +174 @@
         r.out.resume_handle = &resume_handle;
 
-        status = dcerpc_lsa_EnumTrustDom(p, tctx, &r);
-
-        if (NT_STATUS_EQUAL(status, NT_STATUS_NO_MORE_ENTRIES))
+        torture_assert_ntstatus_ok(tctx, dcerpc_lsa_EnumTrustDom_r(b, tctx, &r),
+                "EnumTrustDom failed");
+
+        if (NT_STATUS_EQUAL(r.out.result, NT_STATUS_NO_MORE_ENTRIES))
                 torture_fail(tctx, "no trusts");
 
@@ -207 +196 @@
                 q.out.info = &info;
 
-                status = dcerpc_lsa_QueryTrustedDomainInfoBySid(p, tctx, &q);
-                if (!NT_STATUS_IS_OK(status)) continue;
+                torture_assert_ntstatus_ok(tctx, dcerpc_lsa_QueryTrustedDomainInfoBySid_r(b, tctx, &q),
+                        "QueryTrustedDomainInfoBySid failed");
+                if (!NT_STATUS_IS_OK(q.out.result)) continue;
 
                 if ((info->info_ex.trust_direction & 2) &&
@@ -228 +218 @@
         bool ret = true;
         struct policy_handle *handle;
-        struct dom_sid *dom_sid;
-        struct dom_sid *trusted_sid;
+        struct dom_sid *dom_sid = NULL;
+        struct dom_sid *trusted_sid = NULL;
         struct dom_sid *sids[NUM_SIDS];
+        struct dcerpc_binding_handle *b;
 
         status = torture_rpc_connection(torture, &p, &ndr_table_lsarpc);
@@ -236 +227 @@
                 torture_fail(torture, "unable to connect to table");
         }
-
-        ret &= open_policy(torture, p, &handle);
+        b = p->binding_handle;
+
+        ret &= open_policy(torture, b, &handle);
         if (!ret) return false;
 
-        ret &= get_domainsid(torture, p, handle, &dom_sid);
+        ret &= get_domainsid(torture, b, handle, &dom_sid);
         if (!ret) return false;
 
-        ret &= get_downleveltrust(torture, p, handle, &trusted_sid);
+        ret &= get_downleveltrust(torture, b, handle, &trusted_sid);
         if (!ret) return false;
 
@@ -258 +250 @@
         sids[7] = dom_sid_add_rid(torture, trusted_sid, 512);
 
-        ret &= test_lookupsids(torture, p, handle, sids, NUM_SIDS, 0,
+        ret &= test_lookupsids(torture, b, handle, sids, NUM_SIDS, 0,
                                NT_STATUS_INVALID_PARAMETER, NULL);
 
@@ -267 +259 @@
                           SID_NAME_DOMAIN, SID_NAME_DOM_GRP };
 
-                ret &= test_lookupsids(torture, p, handle, sids, NUM_SIDS, 1,
+                ret &= test_lookupsids(torture, b, handle, sids, NUM_SIDS, 1,
                                        NT_STATUS_OK, types);
         }
@@ -277 +269 @@
                           SID_NAME_DOMAIN, SID_NAME_DOM_GRP,
                           SID_NAME_DOMAIN, SID_NAME_DOM_GRP };
-                ret &= test_lookupsids(torture, p, handle, sids, NUM_SIDS, 2,
+                ret &= test_lookupsids(torture, b, handle, sids, NUM_SIDS, 2,
                                        STATUS_SOME_UNMAPPED, types);
         }
@@ -287 +279 @@
                           SID_NAME_DOMAIN, SID_NAME_DOM_GRP,
                           SID_NAME_UNKNOWN, SID_NAME_UNKNOWN };
-                ret &= test_lookupsids(torture, p, handle, sids, NUM_SIDS, 3,
+                ret &= test_lookupsids(torture, b, handle, sids, NUM_SIDS, 3,
                                        STATUS_SOME_UNMAPPED, types);
         }
@@ -297 +289 @@
                           SID_NAME_DOMAIN, SID_NAME_DOM_GRP,
                           SID_NAME_UNKNOWN, SID_NAME_UNKNOWN };
-                ret &= test_lookupsids(torture, p, handle, sids, NUM_SIDS, 4,
+                ret &= test_lookupsids(torture, b, handle, sids, NUM_SIDS, 4,
                                        STATUS_SOME_UNMAPPED, types);
         }
 
-        ret &= test_lookupsids(torture, p, handle, sids, NUM_SIDS, 5,
+        ret &= test_lookupsids(torture, b, handle, sids, NUM_SIDS, 5,
                                NT_STATUS_NONE_MAPPED, NULL);
 
@@ -310 +302 @@
                           SID_NAME_DOMAIN, SID_NAME_DOM_GRP,
                           SID_NAME_UNKNOWN, SID_NAME_UNKNOWN };
-                ret &= test_lookupsids(torture, p, handle, sids, NUM_SIDS, 6,
+                ret &= test_lookupsids(torture, b, handle, sids, NUM_SIDS, 6,
                                        STATUS_SOME_UNMAPPED, types);
         }
 
-        ret &= test_lookupsids(torture, p, handle, sids, NUM_SIDS, 7,
+        ret &= test_lookupsids(torture, b, handle, sids, NUM_SIDS, 7,
                                NT_STATUS_INVALID_PARAMETER, NULL);
-        ret &= test_lookupsids(torture, p, handle, sids, NUM_SIDS, 8,
+        ret &= test_lookupsids(torture, b, handle, sids, NUM_SIDS, 8,
                                NT_STATUS_INVALID_PARAMETER, NULL);
-        ret &= test_lookupsids(torture, p, handle, sids, NUM_SIDS, 9,
+        ret &= test_lookupsids(torture, b, handle, sids, NUM_SIDS, 9,
                                NT_STATUS_INVALID_PARAMETER, NULL);
-        ret &= test_lookupsids(torture, p, handle, sids, NUM_SIDS, 10,
+        ret &= test_lookupsids(torture, b, handle, sids, NUM_SIDS, 10,
                                NT_STATUS_INVALID_PARAMETER, NULL);
 
@@ -341 +333 @@
 
         uint32_t i;
-        NTSTATUS status;
         const char *dom_sid = "S-1-5-21-1111111111-2222222222-3333333333";
         const char *dom_admin_sid;
-
-        if (!open_policy(tctx, p, &handle)) {
+        struct dcerpc_binding_handle *b = p->binding_handle;
+
+        if (!open_policy(tctx, b, &handle)) {
                 return false;
         }
@@ -374 +366 @@
         r.out.domains   = &domains;
 
-        status = dcerpc_lsa_LookupSids(p, tctx, &r);
-
-        torture_assert_ntstatus_equal(tctx, status, NT_STATUS_NONE_MAPPED,
+        torture_assert_ntstatus_ok(tctx, dcerpc_lsa_LookupSids_r(b, tctx, &r),
+                "LookupSids failed");
+
+        torture_assert_ntstatus_equal(tctx, r.out.result, NT_STATUS_NONE_MAPPED,
                 "unexpected error code");
 
@@ -405 +398 @@
         struct torture_rpc_tcase *tcase;
 
-        suite = torture_suite_create(mem_ctx, "LSA-LOOKUPSIDS");
+        suite = torture_suite_create(mem_ctx, "lsa.lookupsids");
         tcase = torture_suite_add_rpc_iface_tcase(suite, "lsa",
                                                   &ndr_table_lsarpc);