2 Unix SMB/CIFS mplementation.
3 Helper functions for applying replicated objects
5 Copyright (C) Stefan Metzmacher <metze@samba.org> 2007
7 This program is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 3 of the License, or
10 (at your option) any later version.
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
17 You should have received a copy of the GNU General Public License
18 along with this program. If not, see <http://www.gnu.org/licenses/>.
23 #include "dsdb/samdb/samdb.h"
24 #include "lib/ldb/include/ldb_errors.h"
25 #include "../lib/util/dlinklist.h"
26 #include "librpc/gen_ndr/ndr_misc.h"
27 #include "librpc/gen_ndr/ndr_drsuapi.h"
28 #include "librpc/gen_ndr/ndr_drsblobs.h"
29 #include "../lib/crypto/crypto.h"
30 #include "../libcli/drsuapi/drsuapi.h"
31 #include "libcli/auth/libcli_auth.h"
32 #include "param/param.h"
34 static WERROR dsdb_convert_object_ex(struct ldb_context *ldb,
35 const struct dsdb_schema *schema,
36 const struct drsuapi_DsReplicaObjectListItemEx *in,
37 const DATA_BLOB *gensec_skey,
39 struct dsdb_extended_replicated_object *out)
44 struct ldb_message *msg;
45 struct replPropertyMetaDataBlob *md;
46 struct ldb_val guid_value;
47 NTTIME whenChanged = 0;
49 const char *whenChanged_s;
50 const char *rdn_name = NULL;
51 const struct ldb_val *rdn_value = NULL;
52 const struct dsdb_attribute *rdn_attr = NULL;
54 struct drsuapi_DsReplicaAttribute *name_a = NULL;
55 struct drsuapi_DsReplicaMetaData *name_d = NULL;
56 struct replPropertyMetaData1 *rdn_m = NULL;
57 struct dom_sid *sid = NULL;
61 if (!in->object.identifier) {
65 if (!in->object.identifier->dn || !in->object.identifier->dn[0]) {
69 if (in->object.attribute_ctr.num_attributes != 0 && !in->meta_data_ctr) {
73 if (in->object.attribute_ctr.num_attributes != in->meta_data_ctr->count) {
77 sid = &in->object.identifier->sid;
78 if (sid->num_auths > 0) {
79 rid = sid->sub_auths[sid->num_auths - 1];
82 msg = ldb_msg_new(mem_ctx);
83 W_ERROR_HAVE_NO_MEMORY(msg);
85 msg->dn = ldb_dn_new(msg, ldb, in->object.identifier->dn);
86 W_ERROR_HAVE_NO_MEMORY(msg->dn);
88 rdn_name = ldb_dn_get_rdn_name(msg->dn);
89 rdn_attr = dsdb_attribute_by_lDAPDisplayName(schema, rdn_name);
93 rdn_attid = rdn_attr->attributeID_id;
94 rdn_value = ldb_dn_get_rdn_val(msg->dn);
96 msg->num_elements = in->object.attribute_ctr.num_attributes;
97 msg->elements = talloc_array(msg, struct ldb_message_element,
99 W_ERROR_HAVE_NO_MEMORY(msg->elements);
101 md = talloc(mem_ctx, struct replPropertyMetaDataBlob);
102 W_ERROR_HAVE_NO_MEMORY(md);
106 md->ctr.ctr1.count = in->meta_data_ctr->count;
107 md->ctr.ctr1.reserved = 0;
108 md->ctr.ctr1.array = talloc_array(mem_ctx,
109 struct replPropertyMetaData1,
110 md->ctr.ctr1.count + 1); /* +1 because of the RDN attribute */
111 W_ERROR_HAVE_NO_MEMORY(md->ctr.ctr1.array);
113 for (i=0; i < in->meta_data_ctr->count; i++) {
114 struct drsuapi_DsReplicaAttribute *a;
115 struct drsuapi_DsReplicaMetaData *d;
116 struct replPropertyMetaData1 *m;
117 struct ldb_message_element *e;
120 a = &in->object.attribute_ctr.attributes[i];
121 d = &in->meta_data_ctr->meta_data[i];
122 m = &md->ctr.ctr1.array[i];
123 e = &msg->elements[i];
125 for (j=0; j<a->value_ctr.num_values; j++) {
126 status = drsuapi_decrypt_attribute(a->value_ctr.values[j].blob, gensec_skey, rid, a);
127 W_ERROR_NOT_OK_RETURN(status);
130 status = dsdb_attribute_drsuapi_to_ldb(ldb, schema, a, msg->elements, e);
131 if (!NT_STATUS_IS_OK(status) && a->value_ctr.num_values == 0) {
132 /* w2k8-r2 occasionally sends bogus empty
133 attributes with rubbish attribute IDs. The
134 only think we can do is discard these */
135 DEBUG(0,(__location__ ": Discarding bogus empty DsReplicaAttribute with attid 0x%x\n",
140 W_ERROR_NOT_OK_RETURN(status);
143 m->version = d->version;
144 m->originating_change_time = d->originating_change_time;
145 m->originating_invocation_id = d->originating_invocation_id;
146 m->originating_usn = d->originating_usn;
149 if (d->originating_change_time > whenChanged) {
150 whenChanged = d->originating_change_time;
153 if (a->attid == DRSUAPI_ATTRIBUTE_name) {
156 rdn_m = &md->ctr.ctr1.array[md->ctr.ctr1.count];
160 /* delete any empty elements */
161 for (i=0; i < msg->num_elements; i++) {
162 if (msg->elements[i].name == NULL) {
163 ldb_msg_remove_element(msg, &msg->elements[i]);
169 struct ldb_message_element *el;
170 el = ldb_msg_find_element(msg, rdn_attr->lDAPDisplayName);
172 ret = ldb_msg_add_value(msg, rdn_attr->lDAPDisplayName, rdn_value, NULL);
173 if (ret != LDB_SUCCESS) {
177 if (el->num_values != 1) {
178 DEBUG(0,(__location__ ": Unexpected num_values=%u\n",
182 if (!ldb_val_equal_exact(&el->values[0], rdn_value)) {
183 DEBUG(0,(__location__ ": RDN value changed? '%*.*s' '%*.*s'\n",
184 (int)el->values[0].length, (int)el->values[0].length, el->values[0].data,
185 (int)rdn_value->length, (int)rdn_value->length, rdn_value->data));
190 rdn_m->attid = rdn_attid;
191 rdn_m->version = name_d->version;
192 rdn_m->originating_change_time = name_d->originating_change_time;
193 rdn_m->originating_invocation_id = name_d->originating_invocation_id;
194 rdn_m->originating_usn = name_d->originating_usn;
195 rdn_m->local_usn = 0;
196 md->ctr.ctr1.count++;
200 whenChanged_t = nt_time_to_unix(whenChanged);
201 whenChanged_s = ldb_timestring(msg, whenChanged_t);
202 W_ERROR_HAVE_NO_MEMORY(whenChanged_s);
204 nt_status = GUID_to_ndr_blob(&in->object.identifier->guid, msg, &guid_value);
205 if (!NT_STATUS_IS_OK(nt_status)) {
206 return ntstatus_to_werror(nt_status);
210 out->guid_value = guid_value;
211 out->when_changed = whenChanged_s;
216 WERROR dsdb_extended_replicated_objects_convert(struct ldb_context *ldb,
217 const char *partition_dn,
218 const struct drsuapi_DsReplicaOIDMapping_Ctr *mapping_ctr,
219 uint32_t object_count,
220 const struct drsuapi_DsReplicaObjectListItemEx *first_object,
221 uint32_t linked_attributes_count,
222 const struct drsuapi_DsReplicaLinkedAttribute *linked_attributes,
223 const struct repsFromTo1 *source_dsa,
224 const struct drsuapi_DsReplicaCursor2CtrEx *uptodateness_vector,
225 const DATA_BLOB *gensec_skey,
227 struct dsdb_extended_replicated_objects **objects)
230 const struct dsdb_schema *schema;
231 struct dsdb_extended_replicated_objects *out;
232 const struct drsuapi_DsReplicaObjectListItemEx *cur;
235 schema = dsdb_get_schema(ldb);
237 return WERR_DS_SCHEMA_NOT_LOADED;
240 status = dsdb_schema_pfm_contains_drsuapi_pfm(schema->prefixmap, mapping_ctr);
241 W_ERROR_NOT_OK_RETURN(status);
243 out = talloc_zero(mem_ctx, struct dsdb_extended_replicated_objects);
244 W_ERROR_HAVE_NO_MEMORY(out);
245 out->version = DSDB_EXTENDED_REPLICATED_OBJECTS_VERSION;
247 out->partition_dn = ldb_dn_new(out, ldb, partition_dn);
248 W_ERROR_HAVE_NO_MEMORY(out->partition_dn);
250 out->source_dsa = source_dsa;
251 out->uptodateness_vector= uptodateness_vector;
253 out->num_objects = object_count;
254 out->objects = talloc_array(out,
255 struct dsdb_extended_replicated_object,
257 W_ERROR_HAVE_NO_MEMORY(out->objects);
259 /* pass the linked attributes down to the repl_meta_data
261 out->linked_attributes_count = linked_attributes_count;
262 out->linked_attributes = linked_attributes;
264 for (i=0, cur = first_object; cur; cur = cur->next_object, i++) {
265 if (i == out->num_objects) {
269 status = dsdb_convert_object_ex(ldb, schema,
271 out->objects, &out->objects[i]);
272 if (!W_ERROR_IS_OK(status)) {
273 DEBUG(0,("Failed to convert object %s\n", cur->object.identifier->dn));
277 if (i != out->num_objects) {
285 WERROR dsdb_extended_replicated_objects_commit(struct ldb_context *ldb,
286 struct dsdb_extended_replicated_objects *objects,
287 uint64_t *notify_uSN)
289 struct ldb_result *ext_res;
291 uint64_t seq_num1, seq_num2;
293 /* TODO: handle linked attributes */
295 /* wrap the extended operation in a transaction
296 See [MS-DRSR] 3.3.2 Transactions
298 ret = ldb_transaction_start(ldb);
299 if (ret != LDB_SUCCESS) {
300 DEBUG(0,(__location__ " Failed to start transaction\n"));
304 ret = dsdb_load_partition_usn(ldb, objects->partition_dn, &seq_num1);
305 if (ret != LDB_SUCCESS) {
306 DEBUG(0,(__location__ " Failed to load partition uSN\n"));
307 ldb_transaction_cancel(ldb);
311 ret = ldb_extended(ldb, DSDB_EXTENDED_REPLICATED_OBJECTS_OID, objects, &ext_res);
312 if (ret != LDB_SUCCESS) {
313 DEBUG(0,("Failed to apply records: %s: %s\n",
314 ldb_errstring(ldb), ldb_strerror(ret)));
315 ldb_transaction_cancel(ldb);
318 talloc_free(ext_res);
320 ret = ldb_transaction_prepare_commit(ldb);
321 if (ret != LDB_SUCCESS) {
322 DEBUG(0,(__location__ " Failed to prepare commit of transaction\n"));
326 ret = dsdb_load_partition_usn(ldb, objects->partition_dn, &seq_num2);
327 if (ret != LDB_SUCCESS) {
328 DEBUG(0,(__location__ " Failed to load partition uSN\n"));
329 ldb_transaction_cancel(ldb);
333 /* if this replication partner didn't need to be notified
334 before this transaction then it still doesn't need to be
335 notified, as the changes came from this server */
336 if (seq_num2 > seq_num1 && seq_num1 <= *notify_uSN) {
337 *notify_uSN = seq_num2;
340 ret = ldb_transaction_commit(ldb);
341 if (ret != LDB_SUCCESS) {
342 DEBUG(0,(__location__ " Failed to commit transaction\n"));
347 DEBUG(2,("Replicated %u objects (%u linked attributes) for %s\n",
348 objects->num_objects, objects->linked_attributes_count,
349 ldb_dn_get_linearized(objects->partition_dn)));
354 static WERROR dsdb_convert_object(struct ldb_context *ldb,
355 const struct dsdb_schema *schema,
356 const struct drsuapi_DsReplicaObjectListItem *in,
358 struct ldb_message **_msg)
362 struct ldb_message *msg;
364 if (!in->object.identifier) {
368 if (!in->object.identifier->dn || !in->object.identifier->dn[0]) {
372 msg = ldb_msg_new(mem_ctx);
373 W_ERROR_HAVE_NO_MEMORY(msg);
375 msg->dn = ldb_dn_new(msg, ldb, in->object.identifier->dn);
376 W_ERROR_HAVE_NO_MEMORY(msg->dn);
378 msg->num_elements = in->object.attribute_ctr.num_attributes;
379 msg->elements = talloc_array(msg, struct ldb_message_element,
381 W_ERROR_HAVE_NO_MEMORY(msg->elements);
383 for (i=0; i < msg->num_elements; i++) {
384 struct drsuapi_DsReplicaAttribute *a;
385 struct ldb_message_element *e;
387 a = &in->object.attribute_ctr.attributes[i];
388 e = &msg->elements[i];
390 status = dsdb_attribute_drsuapi_to_ldb(ldb, schema, a, msg->elements, e);
391 W_ERROR_NOT_OK_RETURN(status);
400 WERROR dsdb_origin_objects_commit(struct ldb_context *ldb,
402 const struct drsuapi_DsReplicaObjectListItem *first_object,
404 struct drsuapi_DsReplicaObjectIdentifier2 **_ids)
407 const struct dsdb_schema *schema;
408 const struct drsuapi_DsReplicaObjectListItem *cur;
409 struct ldb_message **objects;
410 struct drsuapi_DsReplicaObjectIdentifier2 *ids;
412 uint32_t num_objects = 0;
413 const char * const attrs[] = {
418 struct ldb_result *res;
421 schema = dsdb_get_schema(ldb);
423 return WERR_DS_SCHEMA_NOT_LOADED;
426 for (cur = first_object; cur; cur = cur->next_object) {
430 if (num_objects == 0) {
434 ret = ldb_transaction_start(ldb);
435 if (ret != LDB_SUCCESS) {
436 return WERR_DS_INTERNAL_FAILURE;
439 objects = talloc_array(mem_ctx, struct ldb_message *,
441 if (objects == NULL) {
446 for (i=0, cur = first_object; cur; cur = cur->next_object, i++) {
447 status = dsdb_convert_object(ldb, schema,
448 cur, objects, &objects[i]);
449 if (!W_ERROR_IS_OK(status)) {
454 ids = talloc_array(mem_ctx,
455 struct drsuapi_DsReplicaObjectIdentifier2,
462 for (i=0; i < num_objects; i++) {
463 struct dom_sid *sid = NULL;
464 struct ldb_request *add_req;
466 DEBUG(6,(__location__ ": adding %s\n",
467 ldb_dn_get_linearized(objects[i]->dn)));
469 ret = ldb_build_add_req(&add_req,
475 ldb_op_default_callback,
477 if (ret != LDB_SUCCESS) {
478 status = WERR_DS_INTERNAL_FAILURE;
482 ret = ldb_request_add_control(add_req, LDB_CONTROL_RELAX_OID, true, NULL);
483 if (ret != LDB_SUCCESS) {
484 status = WERR_DS_INTERNAL_FAILURE;
488 ret = ldb_request(ldb, add_req);
489 if (ret == LDB_SUCCESS) {
490 ret = ldb_wait(add_req->handle, LDB_WAIT_ALL);
492 if (ret != LDB_SUCCESS) {
493 DEBUG(0,(__location__ ": Failed add of %s - %s\n",
494 ldb_dn_get_linearized(objects[i]->dn), ldb_errstring(ldb)));
495 status = WERR_DS_INTERNAL_FAILURE;
499 talloc_free(add_req);
501 ret = ldb_search(ldb, objects, &res, objects[i]->dn,
502 LDB_SCOPE_BASE, attrs,
504 if (ret != LDB_SUCCESS) {
505 status = WERR_DS_INTERNAL_FAILURE;
508 ids[i].guid = samdb_result_guid(res->msgs[0], "objectGUID");
509 sid = samdb_result_dom_sid(objects, res->msgs[0], "objectSid");
513 ZERO_STRUCT(ids[i].sid);
517 ret = ldb_transaction_commit(ldb);
518 if (ret != LDB_SUCCESS) {
519 return WERR_DS_INTERNAL_FAILURE;
522 talloc_free(objects);
529 talloc_free(objects);
530 ldb_transaction_cancel(ldb);