torture: complete dfs referral tests

Autobuild-User: Matthieu Patou <mat@samba.org>
Autobuild-Date: Tue Oct  5 08:01:39 UTC 2010 on sn-devel-104

diff --git a/source4/torture/dfs/domaindfs.c b/source4/torture/dfs/domaindfs.c
index 18c84396d389a2dcefd15acd015e70f7d0dda4a5..022aafff99b967a01a4f9732b4b6aff56a9e0c48 100644 (file)
--- a/source4/torture/dfs/domaindfs.c
+++ b/source4/torture/dfs/domaindfs.c
@@ -47,6 +47,10 @@ static bool test_getdomainreferral(struct torture_context *tctx,
                                 "0 domains referrals returned");
        torture_assert_int_equal(tctx, resp.header_flags, 0,
                                 "Header flag different it's not a referral server");
+       torture_assert_int_equal(tctx, resp.referral_entries[1].version, 3,
+                                talloc_asprintf(tctx,
+                                       "Not expected version for referral entry 1 got %d expected 3",
+                                       resp.referral_entries[1].version));
        torture_assert_int_equal(tctx, resp.referral_entries[0].version, 3,
                                 talloc_asprintf(tctx,
                                        "Not expected version for referral entry 0 got %d expected 3",
@@ -74,6 +78,7 @@ static bool test_getdcreferral(struct torture_context *tctx,
        struct dfs_GetDFSReferral r, r2, r3;
        struct dfs_referral_resp resp, resp2, resp3;
        const char* str;
+       const char* str2;
 
        r.in.req.max_referral_level = 3;
        r.in.req.servername = "";
@@ -125,6 +130,16 @@ static bool test_getdcreferral(struct torture_context *tctx,
                                 resp2.referral_entries[0].referral.v3.data.referrals.r2.expanded_names[0]) > 0,
                                 1,
                                 "Length of first dc is less than 0");
+       str = strchr(resp2.referral_entries[0].referral.v3.data.referrals.r2.expanded_names[0], '.');
+       str2 = resp2.referral_entries[0].referral.v3.data.referrals.r2.special_name;
+       if (str2[0] == '\\') {
+               str2++;
+       }
+       torture_assert_int_equal(tctx, strlen(str) >0, 1 ,"Length of domain too short");
+       str++;
+       torture_assert_int_equal(tctx, strcmp(str,str2), 0,
+                                       talloc_asprintf(tctx, "Pb domain of the dc is not"\
+                                                               "the same as the requested: domain = %s got =%s",str2 ,str));
 
        r3.in.req.max_referral_level = 3;
        /*

diff --git a/source4/torture/ndr/dfsblob.c b/source4/torture/ndr/dfsblob.c
index 96d3407815de6491506c0acadec7958e8fb3e163..23a32e163f64761ddc02b9be1cac3e689015f179 100644 (file)
--- a/source4/torture/ndr/dfsblob.c
+++ b/source4/torture/ndr/dfsblob.c
@@ -3,7 +3,7 @@
 
    Test DFS blobs.
 
-   Copyright (C) Matthieu Patou <mat@matws.net> 2009
+   Copyright (C) Matthieu Patou <mat@matws.net> 2009-2010
 
    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
@@ -22,29 +22,63 @@
 #include "includes.h"
 #include "torture/ndr/ndr.h"
 #include "librpc/gen_ndr/ndr_dfsblobs.h"
+#include "librpc/gen_ndr/dfsblobs.h"
 
+DATA_BLOB blob;
 static const uint8_t dfs_get_ref_in[] = {
        0x03, 0x00, 0x5c, 0x00, 0x57, 0x00, 0x32, 0x00,
        0x4b, 0x00, 0x33, 0x00, 0x00, 0x00 };
 
 static const uint8_t dfs_get_ref_out[] = {
-       0x00, 0x00, 0x01, 0x00, 0x00, 0x00,
-       0x00, 0x00, 0x03, 0x00, 0x22, 0x00, 0x00, 0x00,
-       0x02, 0x00, 0x58, 0x02, 0x00, 0x00, 0x22, 0x00,
-       0x01, 0x00, 0x2e, 0x00, 0x00, 0x00, 0x00, 0x00,
+       0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00,
+       0x03, 0x00, 0x22, 0x00, 0x00, 0x00, 0x02, 0x00,
+       0x58, 0x02, 0x00, 0x00, 0x22, 0x00, 0x01, 0x00,
+       0x3a, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+       0x00, 0x00, 0x5c, 0x00, 0x6d, 0x00, 0x73, 0x00,
+       0x77, 0x00, 0x32, 0x00, 0x6b, 0x00, 0x33, 0x00,
+       0x2e, 0x00, 0x74, 0x00, 0x73, 0x00, 0x74, 0x00,
+       0x00, 0x00, 0x5c, 0x00, 0x77, 0x00, 0x32, 0x00,
+       0x6b, 0x00, 0x33, 0x00, 0x61, 0x00, 0x64, 0x00,
+       0x76, 0x00, 0x7a, 0x00, 0x30, 0x00, 0x31, 0x00,
+       0x2e, 0x00, 0x6d, 0x00, 0x73, 0x00, 0x77, 0x00,
+       0x32, 0x00, 0x6b, 0x00, 0x33, 0x00, 0x2e, 0x00,
+       0x74, 0x00, 0x73, 0x00, 0x74, 0x00, 0x00, 0x00};
+
+static const uint8_t dfs_get_ref_out2[] = {
+       0x00, 0x00, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00,
+       0x03, 0x00, 0x12, 0x00, 0x00, 0x00, 0x02, 0x00,
+       0x58, 0x02, 0x00, 0x00, 0x24, 0x00, 0x00, 0x00,
+       0x00, 0x00, 0x03, 0x00, 0x12, 0x00, 0x00, 0x00,
+       0x02, 0x00, 0x58, 0x02, 0x00, 0x00, 0x22, 0x00,
        0x00, 0x00, 0x00, 0x00, 0x5c, 0x00, 0x57, 0x00,
-       0x32, 0x00, 0x4b, 0x00, 0x33, 0x00, 0x00, 0x00,
-       0x5c, 0x00, 0x57, 0x00, 0x32, 0x00, 0x4b, 0x00,
-       0x33, 0x00, 0x2d, 0x00, 0x31, 0x00, 0x30, 0x00,
-       0x31, 0x00, 0x00, 0x00 };
+       0x32, 0x00, 0x4b, 0x00, 0x38, 0x00, 0x52, 0x00,
+       0x32, 0x00, 0x00, 0x00, 0x5c, 0x00, 0x77, 0x00,
+       0x32, 0x00, 0x6b, 0x00, 0x38, 0x00, 0x72, 0x00,
+       0x32, 0x00, 0x2e, 0x00, 0x6d, 0x00, 0x61, 0x00,
+       0x74, 0x00, 0x77, 0x00, 0x73, 0x00, 0x2e, 0x00,
+       0x6e, 0x00, 0x65, 0x00, 0x74, 0x00, 0x00, 0x00
+};
+static bool dfs_referral_out_check(struct torture_context *tctx, struct dfs_referral_resp *r)
+{
+       torture_assert_str_equal(tctx,
+               r->referral_entries[0].referral.v3.data.referrals.r2.special_name,
+               "\\msw2k3.tst", "Special name");
+       ndr_push_struct_blob(&blob, tctx, r, (ndr_push_flags_fn_t)ndr_push_dfs_referral_resp);
+       torture_assert_int_equal(tctx, blob.data[blob.length-2], 0, "expanded names not null terminated");
+       torture_assert_int_equal(tctx, blob.data[blob.length-1], 0, "expanded names not null terminated");
+       return true;
+}
 
 struct torture_suite *ndr_dfsblob_suite(TALLOC_CTX *ctx)
 {
        struct torture_suite *suite = torture_suite_create(ctx, "dfsblob");
 
        torture_suite_add_ndr_pull_fn_test(suite, dfs_GetDFSReferral_in, dfs_get_ref_in, NDR_IN, NULL);
-       torture_suite_add_ndr_pull_fn_test(suite, dfs_referral_resp, dfs_get_ref_out, NDR_BUFFERS|NDR_SCALARS, NULL);
+
+       torture_suite_add_ndr_pull_fn_test(suite, dfs_referral_resp, dfs_get_ref_out2, NDR_BUFFERS|NDR_SCALARS, NULL);
+
+       torture_suite_add_ndr_pull_fn_test(suite, dfs_referral_resp, dfs_get_ref_out, NDR_BUFFERS|NDR_SCALARS,dfs_referral_out_check);
 
        return suite;
 }