source4/torture: Fix prototypes for all functions.
[sfrench/samba-autobuild/.git] / source4 / torture / drs / unit / schemainfo_tests.c
index f197b479e981e567a10bba0db288793162ae45b5..2fb5308f741c1644cac3ecada90f3bcdefb4350c 100644 (file)
 #include "torture/smbtorture.h"
 #include "dsdb/samdb/samdb.h"
 #include "dsdb/samdb/ldb_modules/util.h"
-#include "lib/ldb_wrap.h"
-#include "lib/ldb/include/ldb_module.h"
+#include "ldb_wrap.h"
+#include <ldb_module.h>
 #include "torture/rpc/drsuapi.h"
 #include "librpc/ndr/libndr.h"
 #include "param/param.h"
+#include "torture/drs/proto.h"
+#include "torture/drs/proto.h"
 
 
 /**
@@ -297,6 +299,82 @@ static bool test_dsdb_blob_from_schema_info(struct torture_context *tctx,
        return true;
 }
 
+static bool test_dsdb_schema_info_cmp(struct torture_context *tctx,
+                                     struct drsut_schemainfo_data *priv)
+{
+       DATA_BLOB blob;
+       struct drsuapi_DsReplicaOIDMapping_Ctr *ctr;
+
+       ctr = talloc_zero(priv, struct drsuapi_DsReplicaOIDMapping_Ctr);
+       torture_assert(tctx, ctr, "Not enough memory!");
+
+       /* not enough elements */
+       torture_assert_werr_equal(tctx,
+                                 dsdb_schema_info_cmp(priv->schema, ctr),
+                                 WERR_INVALID_PARAMETER,
+                                 "dsdb_schema_info_cmp(): unexpected result");
+
+       /* an empty element for schemaInfo */
+       ctr->num_mappings = 1;
+       ctr->mappings = talloc_zero_array(ctr, struct drsuapi_DsReplicaOIDMapping, 1);
+       torture_assert(tctx, ctr->mappings, "Not enough memory!");
+       torture_assert_werr_equal(tctx,
+                                 dsdb_schema_info_cmp(priv->schema, ctr),
+                                 WERR_INVALID_PARAMETER,
+                                 "dsdb_schema_info_cmp(): unexpected result");
+
+       /* test with invalid schemaInfo - length != 21 */
+       blob = strhex_to_data_blob(ctr, "FF00000001FD821C07C7455143A3DB51F75A630A7F00");
+       torture_assert(tctx, blob.data, "Not enough memory!");
+       ctr->mappings[0].oid.length     = blob.length;
+       ctr->mappings[0].oid.binary_oid = blob.data;
+       torture_assert_werr_equal(tctx,
+                                 dsdb_schema_info_cmp(priv->schema, ctr),
+                                 WERR_INVALID_PARAMETER,
+                                 "dsdb_schema_info_cmp(): unexpected result");
+
+       /* test with invalid schemaInfo - marker != 0xFF */
+       blob = strhex_to_data_blob(ctr, "AA00000001FD821C07C7455143A3DB51F75A630A7F");
+       torture_assert(tctx, blob.data, "Not enough memory!");
+       ctr->mappings[0].oid.length     = blob.length;
+       ctr->mappings[0].oid.binary_oid = blob.data;
+       torture_assert_werr_equal(tctx,
+                                 dsdb_schema_info_cmp(priv->schema, ctr),
+                                 WERR_INVALID_PARAMETER,
+                                 "dsdb_schema_info_cmp(): unexpected result");
+
+       /* test with valid schemaInfo, but not correct one */
+       blob = strhex_to_data_blob(ctr, "FF0000000000000000000000000000000000000000");
+       torture_assert(tctx, blob.data, "Not enough memory!");
+       ctr->mappings[0].oid.length     = blob.length;
+       ctr->mappings[0].oid.binary_oid = blob.data;
+       torture_assert_werr_equal(tctx,
+                                 dsdb_schema_info_cmp(priv->schema, ctr),
+                                 WERR_DS_DRA_SCHEMA_MISMATCH,
+                                 "dsdb_schema_info_cmp(): unexpected result");
+
+       /* test with correct schemaInfo, but invalid ATTID */
+       blob = strhex_to_data_blob(ctr, priv->schema->schema_info);
+       torture_assert(tctx, blob.data, "Not enough memory!");
+       ctr->mappings[0].id_prefix      = 1;
+       ctr->mappings[0].oid.length     = blob.length;
+       ctr->mappings[0].oid.binary_oid = blob.data;
+       torture_assert_werr_equal(tctx,
+                                 dsdb_schema_info_cmp(priv->schema, ctr),
+                                 WERR_INVALID_PARAMETER,
+                                 "dsdb_schema_info_cmp(): unexpected result");
+
+       /* test with valid schemaInfo */
+       blob = strhex_to_data_blob(ctr, priv->schema->schema_info);
+       ctr->mappings[0].id_prefix      = 0;
+       torture_assert_werr_ok(tctx,
+                              dsdb_schema_info_cmp(priv->schema, ctr),
+                              "dsdb_schema_info_cmp(): unexpected result");
+
+       talloc_free(ctr);
+       return true;
+}
+
 /*
  * Tests dsdb_module_schema_info_blob_read()
  *   and dsdb_module_schema_info_blob_write()
@@ -304,7 +382,7 @@ static bool test_dsdb_blob_from_schema_info(struct torture_context *tctx,
 static bool test_dsdb_module_schema_info_blob_rw(struct torture_context *tctx,
                                                struct drsut_schemainfo_data *priv)
 {
-       WERROR werr;
+       int ldb_err;
        DATA_BLOB blob_write;
        DATA_BLOB blob_read;
 
@@ -317,14 +395,14 @@ static bool test_dsdb_module_schema_info_blob_rw(struct torture_context *tctx,
        blob_write = strhex_to_data_blob(priv, SCHEMA_INFO_DEFAULT_STR);
        torture_assert(tctx, blob_write.data, "Not enough memory!");
 
-       werr = dsdb_module_schema_info_blob_write(priv->ldb_module,
-                                                 DSDB_FLAG_TOP_MODULE,
-                                                 &blob_write);
-       torture_assert_werr_ok(tctx, werr, "dsdb_module_schema_info_blob_write() failed");
+       ldb_err = dsdb_module_schema_info_blob_write(priv->ldb_module,
+                                                    DSDB_FLAG_TOP_MODULE,
+                                                    &blob_write, NULL);
+       torture_assert_int_equal(tctx, ldb_err, LDB_SUCCESS, "dsdb_module_schema_info_blob_write() failed");
 
-       werr = dsdb_module_schema_info_blob_read(priv->ldb_module, DSDB_FLAG_TOP_MODULE,
-                                                priv, &blob_read);
-       torture_assert_werr_ok(tctx, werr, "dsdb_module_schema_info_blob_read() failed");
+       ldb_err = dsdb_module_schema_info_blob_read(priv->ldb_module, DSDB_FLAG_TOP_MODULE,
+                                                   priv, &blob_read, NULL);
+       torture_assert_int_equal(tctx, ldb_err, LDB_SUCCESS, "dsdb_module_schema_info_blob_read() failed");
 
        /* check if we get what we wrote */
        torture_assert_data_blob_equal(tctx, blob_read, blob_write,
@@ -339,6 +417,7 @@ static bool test_dsdb_module_schema_info_blob_rw(struct torture_context *tctx,
 static bool test_dsdb_module_schema_info_update(struct torture_context *tctx,
                                                struct drsut_schemainfo_data *priv)
 {
+       int ldb_err;
        WERROR werr;
        DATA_BLOB blob;
        struct dsdb_schema_info *schema_info;
@@ -348,15 +427,15 @@ static bool test_dsdb_module_schema_info_update(struct torture_context *tctx,
                       _drsut_ldb_schema_info_reset(tctx, priv->ldb, SCHEMA_INFO_INIT_STR, false),
                       "_drsut_ldb_schema_info_reset() failed");
 
-       werr = dsdb_module_schema_info_update(priv->ldb_module,
-                                             priv->schema,
-                                             DSDB_FLAG_TOP_MODULE | DSDB_FLAG_AS_SYSTEM);
-       torture_assert_werr_ok(tctx, werr, "dsdb_module_schema_info_update() failed");
+       ldb_err = dsdb_module_schema_info_update(priv->ldb_module,
+                                                priv->schema,
+                                                DSDB_FLAG_TOP_MODULE | DSDB_FLAG_AS_SYSTEM, NULL);
+       torture_assert_int_equal(tctx, ldb_err, LDB_SUCCESS, "dsdb_module_schema_info_update() failed");
 
        /* get updated schemaInfo */
-       werr = dsdb_module_schema_info_blob_read(priv->ldb_module, DSDB_FLAG_TOP_MODULE,
-                                                priv, &blob);
-       torture_assert_werr_ok(tctx, werr, "dsdb_module_schema_info_blob_read() failed");
+       ldb_err = dsdb_module_schema_info_blob_read(priv->ldb_module, DSDB_FLAG_TOP_MODULE,
+                                                   priv, &blob, NULL);
+       torture_assert_int_equal(tctx, ldb_err, LDB_SUCCESS, "dsdb_module_schema_info_blob_read() failed");
 
        werr = dsdb_schema_info_from_blob(&blob, priv, &schema_info);
        torture_assert_werr_ok(tctx, werr, "dsdb_schema_info_from_blob() failed");
@@ -513,7 +592,10 @@ static bool torture_drs_unit_schemainfo_setup(struct torture_context *tctx,
        torture_assert(tctx, priv->ldb_module, "Not enough memory!");
 
        /* create schema mockup object */
-       priv->schema = dsdb_new_schema(priv, lp_iconv_convenience(tctx->lp_ctx));
+       priv->schema = dsdb_new_schema(priv);
+
+       /* set schema_info in dsdb_schema for testing */
+       priv->schema->schema_info = talloc_strdup(priv->schema, SCHEMA_INFO_DEFAULT_STR);
 
        /* pre-cache invocationId for samdb_ntds_invocation_id()
         * to work with our mock ldb */
@@ -553,7 +635,7 @@ static bool torture_drs_unit_schemainfo_teardown(struct torture_context *tctx,
 
 /**
  * Test case initialization for
- * DRS-UNIT.schemaInfo
+ * drs.unit.schemaInfo
  */
 struct torture_tcase * torture_drs_unit_schemainfo(struct torture_suite *suite)
 {
@@ -575,6 +657,8 @@ struct torture_tcase * torture_drs_unit_schemainfo(struct torture_suite *suite)
                                      (pfn_run)test_dsdb_schema_info_from_blob);
        torture_tcase_add_simple_test(tc, "dsdb_blob_from_schema_info",
                                      (pfn_run)test_dsdb_blob_from_schema_info);
+       torture_tcase_add_simple_test(tc, "dsdb_schema_info_cmp",
+                                     (pfn_run)test_dsdb_schema_info_cmp);
        torture_tcase_add_simple_test(tc, "dsdb_module_schema_info_blob read|write",
                                      (pfn_run)test_dsdb_module_schema_info_blob_rw);
        torture_tcase_add_simple_test(tc, "dsdb_module_schema_info_update",