diff options
Diffstat (limited to 'source4/torture/libnet/libnet_BecomeDC.c')
| -rw-r--r-- | source4/torture/libnet/libnet_BecomeDC.c | 87 |
1 files changed, 83 insertions, 4 deletions
diff --git a/source4/torture/libnet/libnet_BecomeDC.c b/source4/torture/libnet/libnet_BecomeDC.c index 2fcec59fc8..0d4114c170 100644 --- a/source4/torture/libnet/libnet_BecomeDC.c +++ b/source4/torture/libnet/libnet_BecomeDC.c @@ -35,7 +35,8 @@ #include "librpc/gen_ndr/ndr_drsblobs.h" #include "librpc/gen_ndr/ndr_misc.h" #include "system/time.h" - +#include "auth/auth.h" +#include "lib/db_wrap.h" #include "lib/appweb/ejs/ejs.h" #include "lib/appweb/ejs/ejsInternal.h" #include "scripting/ejs/smbcalls.h" @@ -94,6 +95,7 @@ failed: } #define TORTURE_NETBIOS_NAME "smbtorturedc" +#define TORTURE_SAMDB_LDB "test_samdb.ldb" struct test_become_dc_state { struct libnet_context *ctx; @@ -174,8 +176,25 @@ static NTSTATUS test_become_dc_prepare_db(void *private_data, "subobj.DNSNAME = \"%s\";\n" "subobj.DEFAULTSITE = \"%s\";\n" "\n" + "modules_list = new Array(\"rootdse\",\n" + " \"kludge_acl\",\n" + " \"paged_results\",\n" + " \"server_sort\",\n" + " \"extended_dn\",\n" + " \"asq\",\n" + " //\"samldb\",should only handle originating changes...\n" + " \"password_hash\",\n" + " \"operational\",\n" + " \"objectclass\",\n" + " \"rdn_name\",\n" + " \"partition\");\n" + "subobj.MODULES_LIST = join(\",\", modules_list);\n" + "subobj.DOMAINDN_MOD = \"objectguid\";\n" + "subobj.CONFIGDN_MOD = \"objectguid\";\n" + "subobj.SCHEMADN_MOD = \"objectguid\";\n" + "\n" "var paths = provision_default_paths(subobj);\n" - "paths.samdb = \"test_samdb.ldb\";\n" + "paths.samdb = \"%s\";\n" "\n" "var system_session = system_session();\n" "\n" @@ -189,7 +208,8 @@ static NTSTATUS test_become_dc_prepare_db(void *private_data, p->forest->schema_dn_str, p->dest_dsa->netbios_name, p->dest_dsa->dns_name, - p->dest_dsa->site_name); + p->dest_dsa->site_name, + TORTURE_SAMDB_LDB); NT_STATUS_HAVE_NO_MEMORY(ejs); ret = test_run_ejs(ejs); @@ -199,9 +219,24 @@ static NTSTATUS test_become_dc_prepare_db(void *private_data, talloc_free(ejs); return NT_STATUS_FOOBAR; } - talloc_free(ejs); + talloc_free(s->ldb); + + s->ldb = ldb_wrap_connect(s, TORTURE_SAMDB_LDB, + system_session(s), + NULL, 0, NULL); + if (!s->ldb) { + DEBUG(0,("Failed to open '%s'\n", + TORTURE_SAMDB_LDB)); + return NT_STATUS_INTERNAL_DB_ERROR; + } + + ret = ldb_transaction_start(s->ldb); + if (ret != LDB_SUCCESS) { + return NT_STATUS_INTERNAL_DB_CORRUPTION; + } + return NT_STATUS_OK; } @@ -232,6 +267,14 @@ static WERROR test_object_to_ldb(struct test_become_dc_state *s, struct drsuapi_DsReplicaObjMetaData *rdn_mc; int ret; + if (!obj->object.identifier) { + return WERR_FOOBAR; + } + + if (!obj->object.identifier->dn || !obj->object.identifier->dn[0]) { + return WERR_FOOBAR; + } + msg = ldb_msg_new(mem_ctx); W_ERROR_HAVE_NO_MEMORY(msg); @@ -381,6 +424,18 @@ static WERROR test_object_to_ldb(struct test_become_dc_state *s, NDR_PRINT_DEBUG(drsuapi_DsReplicaObjMetaDataCtr, &mdc); } + ret = ldb_add(s->ldb, msg); + if (ret != LDB_SUCCESS) { + if (ret == LDB_ERR_ENTRY_ALREADY_EXISTS) { + DEBUG(0,("record exists (ignored): %s: %d\n", + obj->object.identifier->dn, ret)); + } else { + DEBUG(0,("Failed to add record: %s: %d\n", + obj->object.identifier->dn, ret)); + return WERR_FOOBAR; + } + } + *_msg = msg; return WERR_OK; } @@ -390,6 +445,7 @@ static NTSTATUS test_apply_schema(struct test_become_dc_state *s, { WERROR status; struct drsuapi_DsReplicaObjectListItemEx *cur; + int ret; for (cur = s->schema_part.first_object; cur; cur = cur->next_object) { uint32_t i; @@ -468,6 +524,17 @@ static NTSTATUS test_apply_schema(struct test_become_dc_state *s, } } + ret = ldb_transaction_commit(s->ldb); + if (ret != LDB_SUCCESS) { + DEBUG(0,("Failed to commit the schema changes: %d\n", ret)); + return NT_STATUS_INTERNAL_DB_CORRUPTION; + } + + ret = ldb_transaction_start(s->ldb); + if (ret != LDB_SUCCESS) { + return NT_STATUS_INTERNAL_DB_CORRUPTION; + } + return NT_STATUS_OK; } @@ -551,6 +618,7 @@ static NTSTATUS test_become_dc_store_chunk(void *private_data, uint32_t linked_attributes_count; struct drsuapi_DsReplicaLinkedAttribute *linked_attributes; uint32_t i; + int ret; switch (c->ctr_level) { case 1: @@ -620,6 +688,17 @@ static NTSTATUS test_become_dc_store_chunk(void *private_data, } } + ret = ldb_transaction_commit(s->ldb); + if (ret != LDB_SUCCESS) { + DEBUG(0,("Failed to commit the changes: %d\n", ret)); + return NT_STATUS_INTERNAL_DB_CORRUPTION; + } + + ret = ldb_transaction_start(s->ldb); + if (ret != LDB_SUCCESS) { + return NT_STATUS_INTERNAL_DB_CORRUPTION; + } + return NT_STATUS_OK; } |