source4/dsdb/repl/replicated_objects.c

   1 /*
   2    Unix SMB/CIFS mplementation.
   3    Helper functions for applying replicated objects
   4
   5    Copyright (C) Stefan Metzmacher <metze@samba.org> 2007
   6
   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.
  11
  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.
  16
  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/>.
  19
  20 */
  21
  22 #include "includes.h"
  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"
  33
  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,
  38                                      TALLOC_CTX *mem_ctx,
  39                                      struct dsdb_extended_replicated_object *out)
  40 {
  41         NTSTATUS nt_status;
  42         enum ndr_err_code ndr_err;
  43         WERROR status;
  44         uint32_t i;
  45         struct ldb_message *msg;
  46         struct replPropertyMetaDataBlob *md;
  47         struct ldb_val guid_value;
  48         NTTIME whenChanged = 0;
  49         time_t whenChanged_t;
  50         const char *whenChanged_s;
  51         const char *rdn_name = NULL;
  52         const struct ldb_val *rdn_value = NULL;
  53         const struct dsdb_attribute *rdn_attr = NULL;
  54         uint32_t rdn_attid;
  55         struct drsuapi_DsReplicaAttribute *name_a = NULL;
  56         struct drsuapi_DsReplicaMetaData *name_d = NULL;
  57         struct replPropertyMetaData1 *rdn_m = NULL;
  58         struct dom_sid *sid = NULL;
  59         uint32_t rid = 0;
  60         int ret;
  61
  62         if (!in->object.identifier) {
  63                 return WERR_FOOBAR;
  64         }
  65
  66         if (!in->object.identifier->dn || !in->object.identifier->dn[0]) {
  67                 return WERR_FOOBAR;
  68         }
  69
  70         if (in->object.attribute_ctr.num_attributes != 0 && !in->meta_data_ctr) {
  71                 return WERR_FOOBAR;
  72         }
  73
  74         if (in->object.attribute_ctr.num_attributes != in->meta_data_ctr->count) {
  75                 return WERR_FOOBAR;
  76         }
  77
  78         sid = &in->object.identifier->sid;
  79         if (sid->num_auths > 0) {
  80                 rid = sid->sub_auths[sid->num_auths - 1];
  81         }
  82
  83         msg = ldb_msg_new(mem_ctx);
  84         W_ERROR_HAVE_NO_MEMORY(msg);
  85
  86         msg->dn                 = ldb_dn_new(msg, ldb, in->object.identifier->dn);
  87         W_ERROR_HAVE_NO_MEMORY(msg->dn);
  88
  89         rdn_name        = ldb_dn_get_rdn_name(msg->dn);
  90         rdn_attr        = dsdb_attribute_by_lDAPDisplayName(schema, rdn_name);
  91         if (!rdn_attr) {
  92                 return WERR_FOOBAR;
  93         }
  94         rdn_attid       = rdn_attr->attributeID_id;
  95         rdn_value       = ldb_dn_get_rdn_val(msg->dn);
  96
  97         msg->num_elements       = in->object.attribute_ctr.num_attributes;
  98         msg->elements           = talloc_array(msg, struct ldb_message_element,
  99                                                msg->num_elements);
 100         W_ERROR_HAVE_NO_MEMORY(msg->elements);
 101
 102         md = talloc(mem_ctx, struct replPropertyMetaDataBlob);
 103         W_ERROR_HAVE_NO_MEMORY(md);
 104
 105         md->version             = 1;
 106         md->reserved            = 0;
 107         md->ctr.ctr1.count      = in->meta_data_ctr->count;
 108         md->ctr.ctr1.reserved   = 0;
 109         md->ctr.ctr1.array      = talloc_array(mem_ctx,
 110                                                struct replPropertyMetaData1,
 111                                                md->ctr.ctr1.count + 1); /* +1 because of the RDN attribute */
 112         W_ERROR_HAVE_NO_MEMORY(md->ctr.ctr1.array);
 113
 114         for (i=0; i < in->meta_data_ctr->count; i++) {
 115                 struct drsuapi_DsReplicaAttribute *a;
 116                 struct drsuapi_DsReplicaMetaData *d;
 117                 struct replPropertyMetaData1 *m;
 118                 struct ldb_message_element *e;
 119                 int j;
 120
 121                 a = &in->object.attribute_ctr.attributes[i];
 122                 d = &in->meta_data_ctr->meta_data[i];
 123                 m = &md->ctr.ctr1.array[i];
 124                 e = &msg->elements[i];
 125
 126                 for (j=0; j<a->value_ctr.num_values; j++) {
 127                         status = drsuapi_decrypt_attribute(a->value_ctr.values[j].blob, gensec_skey, rid, a);
 128                         W_ERROR_NOT_OK_RETURN(status);
 129                 }
 130
 131                 status = dsdb_attribute_drsuapi_to_ldb(ldb, schema, a, msg->elements, e);
 132                 W_ERROR_NOT_OK_RETURN(status);
 133
 134                 m->attid                        = a->attid;
 135                 m->version                      = d->version;
 136                 m->originating_change_time      = d->originating_change_time;
 137                 m->originating_invocation_id    = d->originating_invocation_id;
 138                 m->originating_usn              = d->originating_usn;
 139                 m->local_usn                    = 0;
 140
 141                 if (d->originating_change_time > whenChanged) {
 142                         whenChanged = d->originating_change_time;
 143                 }
 144
 145                 if (a->attid == DRSUAPI_ATTRIBUTE_name) {
 146                         name_a = a;
 147                         name_d = d;
 148                         rdn_m = &md->ctr.ctr1.array[md->ctr.ctr1.count];
 149                 }
 150         }
 151
 152         if (rdn_m) {
 153                 struct ldb_message_element *el;
 154                 el = ldb_msg_find_element(msg, rdn_attr->lDAPDisplayName);
 155                 if (!el) {
 156                         ret = ldb_msg_add_value(msg, rdn_attr->lDAPDisplayName, rdn_value, NULL);
 157                         if (ret != LDB_SUCCESS) {
 158                                 return WERR_FOOBAR;
 159                         }
 160                 } else {
 161                         if (el->num_values != 1) {
 162                                 DEBUG(0,(__location__ ": Unexpected num_values=%u\n",
 163                                          el->num_values));
 164                                 return WERR_FOOBAR;
 165                         }
 166                         if (!ldb_val_equal_exact(&el->values[0], rdn_value)) {
 167                                 DEBUG(0,(__location__ ": RDN value changed? '%*.*s' '%*.*s'\n",
 168                                          (int)el->values[0].length, (int)el->values[0].length, el->values[0].data,
 169                                          (int)rdn_value->length, (int)rdn_value->length, rdn_value->data));
 170                                 return WERR_FOOBAR;
 171                         }
 172                 }
 173
 174                 rdn_m->attid                            = rdn_attid;
 175                 rdn_m->version                          = name_d->version;
 176                 rdn_m->originating_change_time          = name_d->originating_change_time;
 177                 rdn_m->originating_invocation_id        = name_d->originating_invocation_id;
 178                 rdn_m->originating_usn                  = name_d->originating_usn;
 179                 rdn_m->local_usn                        = 0;
 180                 md->ctr.ctr1.count++;
 181
 182         }
 183
 184         whenChanged_t = nt_time_to_unix(whenChanged);
 185         whenChanged_s = ldb_timestring(msg, whenChanged_t);
 186         W_ERROR_HAVE_NO_MEMORY(whenChanged_s);
 187
 188         ndr_err = ndr_push_struct_blob(&guid_value, msg,
 189                                        lp_iconv_convenience(ldb_get_opaque(ldb, "loadparm")),
 190                                        &in->object.identifier->guid,
 191                                          (ndr_push_flags_fn_t)ndr_push_GUID);
 192         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
 193                 nt_status = ndr_map_error2ntstatus(ndr_err);
 194                 return ntstatus_to_werror(nt_status);
 195         }
 196
 197         out->msg                = msg;
 198         out->guid_value         = guid_value;
 199         out->when_changed       = whenChanged_s;
 200         out->meta_data          = md;
 201         return WERR_OK;
 202 }
 203
 204 WERROR dsdb_extended_replicated_objects_commit(struct ldb_context *ldb,
 205                                                const char *partition_dn,
 206                                                const struct drsuapi_DsReplicaOIDMapping_Ctr *mapping_ctr,
 207                                                uint32_t object_count,
 208                                                const struct drsuapi_DsReplicaObjectListItemEx *first_object,
 209                                                uint32_t linked_attributes_count,
 210                                                const struct drsuapi_DsReplicaLinkedAttribute *linked_attributes,
 211                                                const struct repsFromTo1 *source_dsa,
 212                                                const struct drsuapi_DsReplicaCursor2CtrEx *uptodateness_vector,
 213                                                const DATA_BLOB *gensec_skey,
 214                                                TALLOC_CTX *mem_ctx,
 215                                                struct dsdb_extended_replicated_objects **_out)
 216 {
 217         WERROR status;
 218         const struct dsdb_schema *schema;
 219         struct dsdb_extended_replicated_objects *out;
 220         struct ldb_result *ext_res;
 221         const struct drsuapi_DsReplicaObjectListItemEx *cur;
 222         uint32_t i;
 223         int ret;
 224
 225         schema = dsdb_get_schema(ldb);
 226         if (!schema) {
 227                 return WERR_DS_SCHEMA_NOT_LOADED;
 228         }
 229
 230         status = dsdb_verify_oid_mappings_drsuapi(schema, mapping_ctr);
 231         W_ERROR_NOT_OK_RETURN(status);
 232
 233         out = talloc_zero(mem_ctx, struct dsdb_extended_replicated_objects);
 234         W_ERROR_HAVE_NO_MEMORY(out);
 235         out->version            = DSDB_EXTENDED_REPLICATED_OBJECTS_VERSION;
 236
 237         out->partition_dn       = ldb_dn_new(out, ldb, partition_dn);
 238         W_ERROR_HAVE_NO_MEMORY(out->partition_dn);
 239
 240         out->source_dsa         = source_dsa;
 241         out->uptodateness_vector= uptodateness_vector;
 242
 243         out->num_objects        = object_count;
 244         out->objects            = talloc_array(out,
 245                                                struct dsdb_extended_replicated_object,
 246                                                out->num_objects);
 247         W_ERROR_HAVE_NO_MEMORY(out->objects);
 248
 249         /* pass the linked attributes down to the repl_meta_data
 250            module */
 251         out->linked_attributes_count = linked_attributes_count;
 252         out->linked_attributes       = linked_attributes;
 253
 254         for (i=0, cur = first_object; cur; cur = cur->next_object, i++) {
 255                 if (i == out->num_objects) {
 256                         return WERR_FOOBAR;
 257                 }
 258
 259                 status = dsdb_convert_object_ex(ldb, schema,
 260                                                 cur, gensec_skey,
 261                                                 out->objects, &out->objects[i]);
 262                 if (!W_ERROR_IS_OK(status)) {
 263                         DEBUG(0,("Failed to convert object %s\n", cur->object.identifier->dn));
 264                         return status;
 265                 }
 266         }
 267         if (i != out->num_objects) {
 268                 return WERR_FOOBAR;
 269         }
 270
 271         /* TODO: handle linked attributes */
 272
 273         /* wrap the extended operation in a transaction
 274            See [MS-DRSR] 3.3.2 Transactions
 275          */
 276         ret = ldb_transaction_start(ldb);
 277         if (ret != LDB_SUCCESS) {
 278                 DEBUG(0,(__location__ " Failed to start transaction\n"));
 279                 talloc_free(out);
 280                 return WERR_FOOBAR;
 281         }
 282
 283         ret = ldb_extended(ldb, DSDB_EXTENDED_REPLICATED_OBJECTS_OID, out, &ext_res);
 284         if (ret != LDB_SUCCESS) {
 285                 DEBUG(0,("Failed to apply records: %s: %s\n",
 286                          ldb_errstring(ldb), ldb_strerror(ret)));
 287                 talloc_free(out);
 288                 ldb_transaction_cancel(ldb);
 289                 return WERR_FOOBAR;
 290         }
 291         talloc_free(ext_res);
 292
 293         ret = ldb_transaction_commit(ldb);
 294         if (ret != LDB_SUCCESS) {
 295                 DEBUG(0,(__location__ " Failed to commit transaction\n"));
 296                 talloc_free(out);
 297                 return WERR_FOOBAR;
 298         }
 299
 300         DEBUG(2,("Replicated %u objects (%u linked attributes) for %s\n",
 301                  out->num_objects, out->linked_attributes_count,
 302                  ldb_dn_get_linearized(out->partition_dn)));
 303
 304
 305         if (_out) {
 306                 *_out = out;
 307         } else {
 308                 talloc_free(out);
 309         }
 310
 311         return WERR_OK;
 312 }
 313
 314 static WERROR dsdb_convert_object(struct ldb_context *ldb,
 315                                   const struct dsdb_schema *schema,
 316                                   const struct drsuapi_DsReplicaObjectListItem *in,
 317                                   TALLOC_CTX *mem_ctx,
 318                                   struct ldb_message **_msg)
 319 {
 320         WERROR status;
 321         uint32_t i;
 322         struct ldb_message *msg;
 323
 324         if (!in->object.identifier) {
 325                 return WERR_FOOBAR;
 326         }
 327
 328         if (!in->object.identifier->dn || !in->object.identifier->dn[0]) {
 329                 return WERR_FOOBAR;
 330         }
 331
 332         msg = ldb_msg_new(mem_ctx);
 333         W_ERROR_HAVE_NO_MEMORY(msg);
 334
 335         msg->dn = ldb_dn_new(msg, ldb, in->object.identifier->dn);
 336         W_ERROR_HAVE_NO_MEMORY(msg->dn);
 337
 338         msg->num_elements       = in->object.attribute_ctr.num_attributes;
 339         msg->elements           = talloc_array(msg, struct ldb_message_element,
 340                                                msg->num_elements);
 341         W_ERROR_HAVE_NO_MEMORY(msg->elements);
 342
 343         for (i=0; i < msg->num_elements; i++) {
 344                 struct drsuapi_DsReplicaAttribute *a;
 345                 struct ldb_message_element *e;
 346
 347                 a = &in->object.attribute_ctr.attributes[i];
 348                 e = &msg->elements[i];
 349
 350                 status = dsdb_attribute_drsuapi_to_ldb(ldb, schema, a, msg->elements, e);
 351                 W_ERROR_NOT_OK_RETURN(status);
 352         }
 353
 354
 355         *_msg = msg;
 356
 357         return WERR_OK;
 358 }
 359
 360 WERROR dsdb_origin_objects_commit(struct ldb_context *ldb,
 361                                   TALLOC_CTX *mem_ctx,
 362                                   const struct drsuapi_DsReplicaObjectListItem *first_object,
 363                                   uint32_t *_num,
 364                                   struct drsuapi_DsReplicaObjectIdentifier2 **_ids)
 365 {
 366         WERROR status;
 367         const struct dsdb_schema *schema;
 368         const struct drsuapi_DsReplicaObjectListItem *cur;
 369         struct ldb_message **objects;
 370         struct drsuapi_DsReplicaObjectIdentifier2 *ids;
 371         uint32_t i;
 372         uint32_t num_objects = 0;
 373         const char * const attrs[] = {
 374                 "objectGUID",
 375                 "objectSid",
 376                 NULL
 377         };
 378         struct ldb_result *res;
 379         int ret;
 380
 381         schema = dsdb_get_schema(ldb);
 382         if (!schema) {
 383                 return WERR_DS_SCHEMA_NOT_LOADED;
 384         }
 385
 386         for (cur = first_object; cur; cur = cur->next_object) {
 387                 num_objects++;
 388         }
 389
 390         if (num_objects == 0) {
 391                 return WERR_OK;
 392         }
 393
 394         objects = talloc_array(mem_ctx, struct ldb_message *,
 395                                num_objects);
 396         W_ERROR_HAVE_NO_MEMORY(objects);
 397
 398         for (i=0, cur = first_object; cur; cur = cur->next_object, i++) {
 399                 status = dsdb_convert_object(ldb, schema,
 400                                              cur, objects, &objects[i]);
 401                 W_ERROR_NOT_OK_RETURN(status);
 402         }
 403
 404         ids = talloc_array(mem_ctx,
 405                            struct drsuapi_DsReplicaObjectIdentifier2,
 406                            num_objects);
 407         W_ERROR_HAVE_NO_MEMORY(objects);
 408
 409         for (i=0; i < num_objects; i++) {
 410                 struct dom_sid *sid = NULL;
 411                 ret = ldb_add(ldb, objects[i]);
 412                 if (ret != 0) {
 413                         goto cancel;
 414                 }
 415                 ret = ldb_search(ldb, objects, &res, objects[i]->dn,
 416                                  LDB_SCOPE_BASE, attrs,
 417                                  "(objectClass=*)");
 418                 if (ret != 0) {
 419                         goto cancel;
 420                 }
 421                 ids[i].guid = samdb_result_guid(res->msgs[0], "objectGUID");
 422                 sid = samdb_result_dom_sid(objects, res->msgs[0], "objectSid");
 423                 if (sid) {
 424                         ids[i].sid = *sid;
 425                 } else {
 426                         ZERO_STRUCT(ids[i].sid);
 427                 }
 428         }
 429
 430         talloc_free(objects);
 431
 432         *_num = num_objects;
 433         *_ids = ids;
 434         return WERR_OK;
 435 cancel:
 436         talloc_free(objects);
 437         ldb_transaction_cancel(ldb);
 438         return WERR_FOOBAR;
 439 }