mirror of
https://github.com/samba-team/samba.git
synced 2024-12-25 23:21:54 +03:00
r20685: apply records using dsdb_extended_replicated_objects_commit()
metze
This commit is contained in:
Stefan Metzmacher
Gerald (Jerry) Carter
parent
492c79de13
commit
8f1fd303a7
@ -106,6 +106,7 @@ struct test_become_dc_state {
|
||||
struct ldb_context *ldb;
|
||||
|
||||
struct {
|
||||
uint32_t object_count;
|
||||
struct drsuapi_DsReplicaObjectListItemEx *first_object;
|
||||
struct drsuapi_DsReplicaObjectListItemEx *last_object;
|
||||
} schema_part;
|
||||
@ -232,256 +233,51 @@ static NTSTATUS test_become_dc_prepare_db(void *private_data,
|
||||
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;
|
||||
}
|
||||
|
||||
static WERROR test_object_to_ldb(struct test_become_dc_state *s,
|
||||
const struct libnet_BecomeDC_StoreChunk *c,
|
||||
struct drsuapi_DsReplicaObjectListItemEx *obj,
|
||||
TALLOC_CTX *mem_ctx,
|
||||
struct ldb_message **_msg)
|
||||
{
|
||||
NTSTATUS nt_status;
|
||||
WERROR status;
|
||||
uint32_t i;
|
||||
struct ldb_message *msg;
|
||||
struct replPropertyMetaDataBlob md;
|
||||
struct ldb_val md_value;
|
||||
struct drsuapi_DsReplicaObjMetaDataCtr mdc;
|
||||
struct ldb_val guid_value;
|
||||
NTTIME whenChanged = 0;
|
||||
time_t whenChanged_t;
|
||||
const char *whenChanged_s;
|
||||
const char *rdn_name;
|
||||
const struct ldb_val *rdn_value;
|
||||
const struct dsdb_attribute *rdn_attr;
|
||||
uint32_t rdn_attid;
|
||||
struct drsuapi_DsReplicaAttribute *name_a;
|
||||
struct drsuapi_DsReplicaMetaData *name_d;
|
||||
struct replPropertyMetaData1 *rdn_m;
|
||||
struct drsuapi_DsReplicaObjMetaData *rdn_mc;
|
||||
struct ldb_request *req;
|
||||
struct ldb_control **ctrls;
|
||||
struct dsdb_control_replicated_object *ctrl;
|
||||
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);
|
||||
|
||||
msg->dn = ldb_dn_new(msg, s->ldb, obj->object.identifier->dn);
|
||||
W_ERROR_HAVE_NO_MEMORY(msg->dn);
|
||||
|
||||
rdn_name = ldb_dn_get_rdn_name(msg->dn);
|
||||
rdn_attr = dsdb_attribute_by_lDAPDisplayName(s->schema, rdn_name);
|
||||
if (!rdn_attr) {
|
||||
return WERR_FOOBAR;
|
||||
}
|
||||
rdn_attid = rdn_attr->attributeID_id;
|
||||
rdn_value = ldb_dn_get_rdn_val(msg->dn);
|
||||
|
||||
msg->num_elements = obj->object.attribute_ctr.num_attributes;
|
||||
msg->elements = talloc_array(msg, struct ldb_message_element,
|
||||
msg->num_elements);
|
||||
W_ERROR_HAVE_NO_MEMORY(msg->elements);
|
||||
|
||||
for (i=0; i < msg->num_elements; i++) {
|
||||
status = dsdb_attribute_drsuapi_to_ldb(s->schema,
|
||||
&obj->object.attribute_ctr.attributes[i],
|
||||
msg->elements, &msg->elements[i]);
|
||||
W_ERROR_NOT_OK_RETURN(status);
|
||||
}
|
||||
|
||||
if (obj->object.attribute_ctr.num_attributes != 0 && !obj->meta_data_ctr) {
|
||||
return WERR_FOOBAR;
|
||||
}
|
||||
|
||||
if (obj->object.attribute_ctr.num_attributes != obj->meta_data_ctr->count) {
|
||||
return WERR_FOOBAR;
|
||||
}
|
||||
|
||||
md.version = 1;
|
||||
md.reserved = 0;
|
||||
md.ctr.ctr1.count = obj->meta_data_ctr->count;
|
||||
md.ctr.ctr1.reserved = 0;
|
||||
md.ctr.ctr1.array = talloc_array(mem_ctx,
|
||||
struct replPropertyMetaData1,
|
||||
md.ctr.ctr1.count + 1);
|
||||
W_ERROR_HAVE_NO_MEMORY(md.ctr.ctr1.array);
|
||||
|
||||
mdc.count = obj->meta_data_ctr->count;
|
||||
mdc.reserved = 0;
|
||||
mdc.array = talloc_array(mem_ctx,
|
||||
struct drsuapi_DsReplicaObjMetaData,
|
||||
mdc.count + 1);
|
||||
W_ERROR_HAVE_NO_MEMORY(mdc.array);
|
||||
|
||||
for (i=0; i < obj->meta_data_ctr->count; i++) {
|
||||
struct drsuapi_DsReplicaAttribute *a;
|
||||
struct drsuapi_DsReplicaMetaData *d;
|
||||
struct replPropertyMetaData1 *m;
|
||||
struct drsuapi_DsReplicaObjMetaData *mc;
|
||||
|
||||
a = &obj->object.attribute_ctr.attributes[i];
|
||||
d = &obj->meta_data_ctr->meta_data[i];
|
||||
m = &md.ctr.ctr1.array[i];
|
||||
mc = &mdc.array[i];
|
||||
|
||||
m->attid = a->attid;
|
||||
m->version = d->version;
|
||||
m->orginating_time = d->orginating_time;
|
||||
m->orginating_invocation_id = d->orginating_invocation_id;
|
||||
m->orginating_usn = d->orginating_usn;
|
||||
m->local_usn = 0;
|
||||
|
||||
mc->attribute_name = dsdb_lDAPDisplayName_by_id(s->schema, a->attid);
|
||||
mc->version = d->version;
|
||||
mc->originating_last_changed = d->orginating_time;
|
||||
mc->originating_dsa_invocation_id= d->orginating_invocation_id;
|
||||
mc->originating_usn = d->orginating_usn;
|
||||
mc->local_usn = 0;
|
||||
|
||||
if (d->orginating_time > whenChanged) {
|
||||
whenChanged = d->orginating_time;
|
||||
}
|
||||
|
||||
if (a->attid == DRSUAPI_ATTRIBUTE_name) {
|
||||
name_a = a;
|
||||
name_d = d;
|
||||
rdn_m = &md.ctr.ctr1.array[md.ctr.ctr1.count];
|
||||
rdn_mc = &mdc.array[mdc.count];
|
||||
}
|
||||
}
|
||||
|
||||
if (!name_d) {
|
||||
return WERR_FOOBAR;
|
||||
}
|
||||
|
||||
ret = ldb_msg_add_value(msg, rdn_attr->lDAPDisplayName, rdn_value, NULL);
|
||||
if (ret != LDB_SUCCESS) {
|
||||
return WERR_FOOBAR;
|
||||
}
|
||||
|
||||
nt_status = ndr_push_struct_blob(&guid_value, msg, &obj->object.identifier->guid,
|
||||
(ndr_push_flags_fn_t)ndr_push_GUID);
|
||||
if (!NT_STATUS_IS_OK(nt_status)) {
|
||||
return ntstatus_to_werror(nt_status);
|
||||
}
|
||||
ret = ldb_msg_add_value(msg, "objectGUID", &guid_value, NULL);
|
||||
if (ret != LDB_SUCCESS) {
|
||||
return WERR_FOOBAR;
|
||||
}
|
||||
|
||||
whenChanged_t = nt_time_to_unix(whenChanged);
|
||||
whenChanged_s = ldb_timestring(msg, whenChanged_t);
|
||||
W_ERROR_HAVE_NO_MEMORY(whenChanged_s);
|
||||
ret = ldb_msg_add_string(msg, "whenChanged", whenChanged_s);
|
||||
if (ret != LDB_SUCCESS) {
|
||||
return WERR_FOOBAR;
|
||||
}
|
||||
|
||||
rdn_m->attid = rdn_attid;
|
||||
rdn_m->version = name_d->version;
|
||||
rdn_m->orginating_time = name_d->orginating_time;
|
||||
rdn_m->orginating_invocation_id = name_d->orginating_invocation_id;
|
||||
rdn_m->orginating_usn = name_d->orginating_usn;
|
||||
rdn_m->local_usn = 0;
|
||||
md.ctr.ctr1.count++;
|
||||
|
||||
rdn_mc->attribute_name = rdn_attr->lDAPDisplayName;
|
||||
rdn_mc->version = name_d->version;
|
||||
rdn_mc->originating_last_changed = name_d->orginating_time;
|
||||
rdn_mc->originating_dsa_invocation_id = name_d->orginating_invocation_id;
|
||||
rdn_mc->originating_usn = name_d->orginating_usn;
|
||||
rdn_mc->local_usn = 0;
|
||||
mdc.count++;
|
||||
|
||||
nt_status = ndr_push_struct_blob(&md_value, msg, &md,
|
||||
(ndr_push_flags_fn_t)ndr_push_replPropertyMetaDataBlob);
|
||||
if (!NT_STATUS_IS_OK(nt_status)) {
|
||||
return ntstatus_to_werror(nt_status);
|
||||
}
|
||||
ret = ldb_msg_add_value(msg, "replPropertyMetaData", &md_value, NULL);
|
||||
if (ret != LDB_SUCCESS) {
|
||||
return WERR_FOOBAR;
|
||||
}
|
||||
|
||||
if (lp_parm_bool(-1, "become dc", "dump objects", False)) {
|
||||
struct ldb_ldif ldif;
|
||||
fprintf(stdout, "#\n");
|
||||
ldif.changetype = LDB_CHANGETYPE_NONE;
|
||||
ldif.msg = msg;
|
||||
ldb_ldif_write_file(s->ldb, stdout, &ldif);
|
||||
NDR_PRINT_DEBUG(drsuapi_DsReplicaObjMetaDataCtr, &mdc);
|
||||
}
|
||||
|
||||
/*
|
||||
* apply the record to the ldb
|
||||
* using an ldb_control so indicate
|
||||
* that it's a replicated change
|
||||
*/
|
||||
ret = ldb_msg_sanity_check(s->ldb, msg);
|
||||
if (ret != LDB_SUCCESS) {
|
||||
return WERR_FOOBAR;
|
||||
}
|
||||
ctrls = talloc_array(msg, struct ldb_control *, 2);
|
||||
W_ERROR_HAVE_NO_MEMORY(ctrls);
|
||||
ctrls[0] = talloc(ctrls, struct ldb_control);
|
||||
W_ERROR_HAVE_NO_MEMORY(ctrls[0]);
|
||||
ctrls[1] = NULL;
|
||||
|
||||
ctrl = talloc(ctrls, struct dsdb_control_replicated_object);
|
||||
W_ERROR_HAVE_NO_MEMORY(ctrl);
|
||||
ctrls[0]->oid = DSDB_CONTROL_REPLICATED_OBJECT_OID;
|
||||
ctrls[0]->critical = True;
|
||||
ctrls[0]->data = ctrl;
|
||||
|
||||
ret = ldb_build_add_req(&req, s->ldb, msg, msg, ctrls, NULL, NULL);
|
||||
if (ret != LDB_SUCCESS) {
|
||||
return WERR_FOOBAR;
|
||||
}
|
||||
ldb_set_timeout(s->ldb, req, 0); /* use default timeout */
|
||||
ret = ldb_request(s->ldb, req);
|
||||
if (ret == LDB_SUCCESS) {
|
||||
ret = ldb_wait(req->handle, LDB_WAIT_ALL);
|
||||
}
|
||||
talloc_free(req);
|
||||
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;
|
||||
}
|
||||
|
||||
static NTSTATUS test_apply_schema(struct test_become_dc_state *s,
|
||||
const struct libnet_BecomeDC_StoreChunk *c)
|
||||
{
|
||||
WERROR status;
|
||||
const struct drsuapi_DsReplicaOIDMapping_Ctr *mapping_ctr;
|
||||
uint32_t total_object_count;
|
||||
uint32_t object_count;
|
||||
struct drsuapi_DsReplicaObjectListItemEx *first_object;
|
||||
struct drsuapi_DsReplicaObjectListItemEx *cur;
|
||||
int ret;
|
||||
|
||||
for (cur = s->schema_part.first_object; cur; cur = cur->next_object) {
|
||||
uint32_t linked_attributes_count;
|
||||
struct drsuapi_DsReplicaLinkedAttribute *linked_attributes;
|
||||
const struct drsuapi_DsReplicaHighWaterMark *new_highwatermark;
|
||||
const struct drsuapi_DsReplicaCursor2CtrEx *uptodateness_vector;
|
||||
struct dsdb_extended_replicated_objects *objs;
|
||||
uint32_t i;
|
||||
|
||||
switch (c->ctr_level) {
|
||||
case 1:
|
||||
mapping_ctr = &c->ctr1->mapping_ctr;
|
||||
total_object_count = c->ctr1->total_object_count;
|
||||
object_count = s->schema_part.object_count;
|
||||
first_object = s->schema_part.first_object;
|
||||
linked_attributes_count = 0;
|
||||
linked_attributes = NULL;
|
||||
new_highwatermark = &c->ctr1->new_highwatermark;
|
||||
uptodateness_vector = NULL; /* TODO: map it */
|
||||
break;
|
||||
case 6:
|
||||
mapping_ctr = &c->ctr6->mapping_ctr;
|
||||
total_object_count = c->ctr6->total_object_count;
|
||||
object_count = s->schema_part.object_count;
|
||||
first_object = s->schema_part.first_object;
|
||||
linked_attributes_count = 0; /* TODO: ! */
|
||||
linked_attributes = NULL; /* TODO: ! */;
|
||||
new_highwatermark = &c->ctr6->new_highwatermark;
|
||||
uptodateness_vector = c->ctr6->uptodateness_vector;
|
||||
break;
|
||||
default:
|
||||
return NT_STATUS_INVALID_PARAMETER;
|
||||
}
|
||||
|
||||
for (cur = first_object; cur; cur = cur->next_object) {
|
||||
bool is_attr = false;
|
||||
bool is_class = false;
|
||||
|
||||
@ -549,25 +345,34 @@ static NTSTATUS test_apply_schema(struct test_become_dc_state *s,
|
||||
}
|
||||
}
|
||||
|
||||
for (cur = s->schema_part.first_object; cur; cur = cur->next_object) {
|
||||
struct ldb_message *msg;
|
||||
status = test_object_to_ldb(s, c, cur, s, &msg);
|
||||
status = dsdb_extended_replicated_objects_commit(s->ldb,
|
||||
c->partition->nc.dn,
|
||||
s->schema,
|
||||
mapping_ctr,
|
||||
object_count,
|
||||
first_object,
|
||||
linked_attributes_count,
|
||||
linked_attributes,
|
||||
new_highwatermark,
|
||||
uptodateness_vector,
|
||||
s, &objs);
|
||||
if (!W_ERROR_IS_OK(status)) {
|
||||
DEBUG(0,("Failed to commit objects: %s\n", win_errstr(status)));
|
||||
return werror_to_ntstatus(status);
|
||||
}
|
||||
|
||||
if (lp_parm_bool(-1, "become dc", "dump objects", False)) {
|
||||
for (i=0; i < objs->num_objects; i++) {
|
||||
struct ldb_ldif ldif;
|
||||
fprintf(stdout, "#\n");
|
||||
ldif.changetype = LDB_CHANGETYPE_NONE;
|
||||
ldif.msg = objs->objects[i].msg;
|
||||
ldb_ldif_write_file(s->ldb, stdout, &ldif);
|
||||
NDR_PRINT_DEBUG(replPropertyMetaDataBlob, objs->objects[i].meta_data);
|
||||
}
|
||||
}
|
||||
|
||||
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;
|
||||
}
|
||||
|
||||
talloc_free(objs);
|
||||
return NT_STATUS_OK;
|
||||
}
|
||||
|
||||
@ -623,8 +428,10 @@ static NTSTATUS test_become_dc_schema_chunk(void *private_data,
|
||||
}
|
||||
|
||||
if (!s->schema_part.first_object) {
|
||||
s->schema_part.object_count = object_count;
|
||||
s->schema_part.first_object = talloc_steal(s, first_object);
|
||||
} else {
|
||||
s->schema_part.object_count += object_count;
|
||||
s->schema_part.last_object->next_object = talloc_steal(s->schema_part.last_object,
|
||||
first_object);
|
||||
}
|
||||
@ -647,11 +454,12 @@ static NTSTATUS test_become_dc_store_chunk(void *private_data,
|
||||
uint32_t total_object_count;
|
||||
uint32_t object_count;
|
||||
struct drsuapi_DsReplicaObjectListItemEx *first_object;
|
||||
struct drsuapi_DsReplicaObjectListItemEx *cur;
|
||||
uint32_t linked_attributes_count;
|
||||
struct drsuapi_DsReplicaLinkedAttribute *linked_attributes;
|
||||
const struct drsuapi_DsReplicaHighWaterMark *new_highwatermark;
|
||||
const struct drsuapi_DsReplicaCursor2CtrEx *uptodateness_vector;
|
||||
struct dsdb_extended_replicated_objects *objs;
|
||||
uint32_t i;
|
||||
int ret;
|
||||
|
||||
switch (c->ctr_level) {
|
||||
case 1:
|
||||
@ -661,6 +469,8 @@ static NTSTATUS test_become_dc_store_chunk(void *private_data,
|
||||
first_object = c->ctr1->first_object;
|
||||
linked_attributes_count = 0;
|
||||
linked_attributes = NULL;
|
||||
new_highwatermark = &c->ctr1->new_highwatermark;
|
||||
uptodateness_vector = NULL; /* TODO: map it */
|
||||
break;
|
||||
case 6:
|
||||
mapping_ctr = &c->ctr6->mapping_ctr;
|
||||
@ -669,6 +479,8 @@ static NTSTATUS test_become_dc_store_chunk(void *private_data,
|
||||
first_object = c->ctr6->first_object;
|
||||
linked_attributes_count = c->ctr6->linked_attributes_count;
|
||||
linked_attributes = c->ctr6->linked_attributes;
|
||||
new_highwatermark = &c->ctr6->new_highwatermark;
|
||||
uptodateness_vector = c->ctr6->uptodateness_vector;
|
||||
break;
|
||||
default:
|
||||
return NT_STATUS_INVALID_PARAMETER;
|
||||
@ -682,18 +494,33 @@ static NTSTATUS test_become_dc_store_chunk(void *private_data,
|
||||
c->partition->nc.dn, object_count));
|
||||
}
|
||||
|
||||
status = dsdb_verify_oid_mappings(s->schema, mapping_ctr);
|
||||
status = dsdb_extended_replicated_objects_commit(s->ldb,
|
||||
c->partition->nc.dn,
|
||||
s->schema,
|
||||
mapping_ctr,
|
||||
object_count,
|
||||
first_object,
|
||||
linked_attributes_count,
|
||||
linked_attributes,
|
||||
new_highwatermark,
|
||||
uptodateness_vector,
|
||||
s, &objs);
|
||||
if (!W_ERROR_IS_OK(status)) {
|
||||
DEBUG(0,("Failed to commit objects: %s\n", win_errstr(status)));
|
||||
return werror_to_ntstatus(status);
|
||||
}
|
||||
|
||||
for (cur = first_object; cur; cur = cur->next_object) {
|
||||
struct ldb_message *msg;
|
||||
status = test_object_to_ldb(s, c, cur, s, &msg);
|
||||
if (!W_ERROR_IS_OK(status)) {
|
||||
return werror_to_ntstatus(status);
|
||||
if (lp_parm_bool(-1, "become dc", "dump objects", False)) {
|
||||
for (i=0; i < objs->num_objects; i++) {
|
||||
struct ldb_ldif ldif;
|
||||
fprintf(stdout, "#\n");
|
||||
ldif.changetype = LDB_CHANGETYPE_NONE;
|
||||
ldif.msg = objs->objects[i].msg;
|
||||
ldb_ldif_write_file(s->ldb, stdout, &ldif);
|
||||
NDR_PRINT_DEBUG(replPropertyMetaDataBlob, objs->objects[i].meta_data);
|
||||
}
|
||||
}
|
||||
talloc_free(objs);
|
||||
|
||||
for (i=0; i < linked_attributes_count; i++) {
|
||||
const struct dsdb_attribute *sa;
|
||||
@ -721,17 +548,6 @@ 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;
|
||||
}
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user