source4/rpc_server/drsuapi/getncchanges.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    implement the DRSUpdateRefs call
   5
   6    Copyright (C) Anatoliy Atanasov 2009
   7    Copyright (C) Andrew Tridgell 2009
   8
   9    This program is free software; you can redistribute it and/or modify
  10    it under the terms of the GNU General Public License as published by
  11    the Free Software Foundation; either version 3 of the License, or
  12    (at your option) any later version.
  13
  14    This program is distributed in the hope that it will be useful,
  15    but WITHOUT ANY WARRANTY; without even the implied warranty of
  16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17    GNU General Public License for more details.
  18
  19    You should have received a copy of the GNU General Public License
  20    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21 */
  22
  23 #include "includes.h"
  24 #include "rpc_server/dcerpc_server.h"
  25 #include "dsdb/samdb/samdb.h"
  26 #include "param/param.h"
  27 #include "librpc/gen_ndr/ndr_drsblobs.h"
  28 #include "librpc/gen_ndr/ndr_drsuapi.h"
  29 #include "rpc_server/drsuapi/dcesrv_drsuapi.h"
  30 #include "rpc_server/dcerpc_server_proto.h"
  31 #include "../libcli/drsuapi/drsuapi.h"
  32 #include "libcli/security/security.h"
  33
  34 /*
  35   build a DsReplicaObjectIdentifier from a ldb msg
  36  */
  37 static struct drsuapi_DsReplicaObjectIdentifier *get_object_identifier(TALLOC_CTX *mem_ctx,
  38                                                                        struct ldb_message *msg)
  39 {
  40         struct drsuapi_DsReplicaObjectIdentifier *identifier;
  41         struct dom_sid *sid;
  42
  43         identifier = talloc(mem_ctx, struct drsuapi_DsReplicaObjectIdentifier);
  44         if (identifier == NULL) {
  45                 return NULL;
  46         }
  47
  48         identifier->dn = ldb_dn_alloc_linearized(identifier, msg->dn);
  49         identifier->guid = samdb_result_guid(msg, "objectGUID");
  50
  51         sid = samdb_result_dom_sid(identifier, msg, "objectSid");
  52         if (sid) {
  53                 identifier->sid = *sid;
  54         } else {
  55                 ZERO_STRUCT(identifier->sid);
  56         }
  57         return identifier;
  58 }
  59
  60 /*
  61   drsuapi_DsGetNCChanges for one object
  62 */
  63 static WERROR get_nc_changes_build_object(struct drsuapi_DsReplicaObjectListItemEx *obj,
  64                                           struct ldb_message *msg,
  65                                           struct ldb_context *sam_ctx,
  66                                           struct ldb_dn *ncRoot_dn,
  67                                           struct dsdb_schema *schema,
  68                                           DATA_BLOB *session_key,
  69                                           uint64_t highest_usn,
  70                                           uint32_t replica_flags)
  71 {
  72         const struct ldb_val *md_value;
  73         int i, n;
  74         struct replPropertyMetaDataBlob md;
  75         uint32_t rid = 0;
  76         enum ndr_err_code ndr_err;
  77         uint32_t *attids;
  78         const char *rdn;
  79         const struct dsdb_attribute *rdn_sa;
  80
  81         if (ldb_dn_compare(ncRoot_dn, msg->dn) == 0) {
  82                 obj->is_nc_prefix = true;
  83                 obj->parent_object_guid = NULL;
  84         } else {
  85                 obj->is_nc_prefix = false;
  86                 obj->parent_object_guid = talloc(obj, struct GUID);
  87                 if (obj->parent_object_guid == NULL) {
  88                         return WERR_DS_DRA_INTERNAL_ERROR;
  89                 }
  90                 *obj->parent_object_guid = samdb_result_guid(msg, "parentGUID");
  91                 if (GUID_all_zero(obj->parent_object_guid)) {
  92                         DEBUG(0,(__location__ ": missing parentGUID for %s\n",
  93                                  ldb_dn_get_linearized(msg->dn)));
  94                         return WERR_DS_DRA_INTERNAL_ERROR;
  95                 }
  96         }
  97         obj->next_object = NULL;
  98
  99         md_value = ldb_msg_find_ldb_val(msg, "replPropertyMetaData");
 100         if (!md_value) {
 101                 /* nothing to send */
 102                 return WERR_OK;
 103         }
 104
 105         ndr_err = ndr_pull_struct_blob(md_value, obj,
 106                                        lp_iconv_convenience(ldb_get_opaque(sam_ctx, "loadparm")), &md,
 107                                        (ndr_pull_flags_fn_t)ndr_pull_replPropertyMetaDataBlob);
 108         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
 109                 return WERR_DS_DRA_INTERNAL_ERROR;
 110         }
 111
 112         if (md.version != 1) {
 113                 return WERR_DS_DRA_INTERNAL_ERROR;
 114         }
 115
 116         rdn = ldb_dn_get_rdn_name(msg->dn);
 117         if (rdn == NULL) {
 118                 DEBUG(0,(__location__ ": No rDN for %s\n", ldb_dn_get_linearized(msg->dn)));
 119                 return WERR_DS_DRA_INTERNAL_ERROR;
 120         }
 121
 122         rdn_sa = dsdb_attribute_by_lDAPDisplayName(schema, rdn);
 123         if (rdn_sa == NULL) {
 124                 DEBUG(0,(__location__ ": Can't find dsds_attribute for rDN %s in %s\n",
 125                          rdn, ldb_dn_get_linearized(msg->dn)));
 126                 return WERR_DS_DRA_INTERNAL_ERROR;
 127         }
 128
 129         obj->meta_data_ctr = talloc(obj, struct drsuapi_DsReplicaMetaDataCtr);
 130         attids = talloc_array(obj, uint32_t, md.ctr.ctr1.count);
 131
 132         obj->object.identifier = get_object_identifier(obj, msg);
 133         if (obj->object.identifier == NULL) {
 134                 return WERR_NOMEM;
 135         }
 136         dom_sid_split_rid(NULL, &obj->object.identifier->sid, NULL, &rid);
 137
 138         obj->meta_data_ctr->meta_data = talloc_array(obj, struct drsuapi_DsReplicaMetaData, md.ctr.ctr1.count);
 139         for (n=i=0; i<md.ctr.ctr1.count; i++) {
 140                 const struct dsdb_attribute *sa;
 141                 /* if the attribute has not changed, and it is not the
 142                    instanceType then don't include it */
 143                 if (md.ctr.ctr1.array[i].local_usn < highest_usn &&
 144                     md.ctr.ctr1.array[i].attid != DRSUAPI_ATTRIBUTE_instanceType) continue;
 145
 146                 /* don't include the rDN */
 147                 if (md.ctr.ctr1.array[i].attid == rdn_sa->attributeID_id) continue;
 148
 149                 sa = dsdb_attribute_by_attributeID_id(schema, md.ctr.ctr1.array[i].attid);
 150                 if (sa->linkID) {
 151                         struct ldb_message_element *el;
 152                         el = ldb_msg_find_element(msg, sa->lDAPDisplayName);
 153                         if (el && el->num_values && dsdb_dn_is_upgraded_link_val(&el->values[0])) {
 154                                 /* don't send upgraded links inline */
 155                                 continue;
 156                         }
 157                 }
 158
 159                 obj->meta_data_ctr->meta_data[n].originating_change_time = md.ctr.ctr1.array[i].originating_change_time;
 160                 obj->meta_data_ctr->meta_data[n].version = md.ctr.ctr1.array[i].version;
 161                 obj->meta_data_ctr->meta_data[n].originating_invocation_id = md.ctr.ctr1.array[i].originating_invocation_id;
 162                 obj->meta_data_ctr->meta_data[n].originating_usn = md.ctr.ctr1.array[i].originating_usn;
 163                 attids[n] = md.ctr.ctr1.array[i].attid;
 164                 n++;
 165         }
 166
 167         /*
 168           note that if n==0 we still need to send the change, as it
 169           could be a rename, which changes the uSNChanged, but not any
 170           of the replicated attributes
 171          */
 172
 173         obj->meta_data_ctr->count = n;
 174
 175         obj->object.flags = DRSUAPI_DS_REPLICA_OBJECT_FROM_MASTER;
 176         obj->object.attribute_ctr.num_attributes = obj->meta_data_ctr->count;
 177         obj->object.attribute_ctr.attributes = talloc_array(obj, struct drsuapi_DsReplicaAttribute,
 178                                                             obj->object.attribute_ctr.num_attributes);
 179
 180         /*
 181          * Note that the meta_data array and the attributes array must
 182          * be the same size and in the same order
 183          */
 184         for (i=0; i<obj->object.attribute_ctr.num_attributes; i++) {
 185                 struct ldb_message_element *el;
 186                 WERROR werr;
 187                 const struct dsdb_attribute *sa;
 188
 189                 sa = dsdb_attribute_by_attributeID_id(schema, attids[i]);
 190                 if (!sa) {
 191                         DEBUG(0,("Unable to find attributeID %u in schema\n", attids[i]));
 192                         return WERR_DS_DRA_INTERNAL_ERROR;
 193                 }
 194
 195                 el = ldb_msg_find_element(msg, sa->lDAPDisplayName);
 196                 if (el == NULL) {
 197                         /* this happens for attributes that have been removed */
 198                         DEBUG(5,("No element '%s' for attributeID %u in message\n",
 199                                  sa->lDAPDisplayName, attids[i]));
 200                         ZERO_STRUCT(obj->object.attribute_ctr.attributes[i]);
 201                         obj->object.attribute_ctr.attributes[i].attid = attids[i];
 202                 } else {
 203                         werr = dsdb_attribute_ldb_to_drsuapi(sam_ctx, schema, el, obj,
 204                                                              &obj->object.attribute_ctr.attributes[i]);
 205                         if (!W_ERROR_IS_OK(werr)) {
 206                                 DEBUG(0,("Unable to convert %s to DRS object - %s\n",
 207                                          sa->lDAPDisplayName, win_errstr(werr)));
 208                                 return werr;
 209                         }
 210                         /* if DRSUAPI_DRS_SPECIAL_SECRET_PROCESSING is set
 211                          * check if attribute is secret and send a null value
 212                          */
 213                         if (replica_flags & DRSUAPI_DRS_SPECIAL_SECRET_PROCESSING) {
 214                                 drsuapi_process_secret_attribute(&obj->object.attribute_ctr.attributes[i],
 215                                                                  &obj->meta_data_ctr->meta_data[i]);
 216                         }
 217                         /* some attributes needs to be encrypted
 218                            before being sent */
 219                         werr = drsuapi_encrypt_attribute(obj, session_key, rid,
 220                                                          &obj->object.attribute_ctr.attributes[i]);
 221                         if (!W_ERROR_IS_OK(werr)) {
 222                                 DEBUG(0,("Unable to encrypt %s in DRS object - %s\n",
 223                                          sa->lDAPDisplayName, win_errstr(werr)));
 224                                 return werr;
 225                         }
 226                 }
 227         }
 228
 229         return WERR_OK;
 230 }
 231
 232
 233 /*
 234   add one linked attribute from an object to the list of linked
 235   attributes in a getncchanges request
 236  */
 237 static WERROR get_nc_changes_add_la(TALLOC_CTX *mem_ctx,
 238                                     struct ldb_context *sam_ctx,
 239                                     const struct dsdb_schema *schema,
 240                                     const struct dsdb_attribute *sa,
 241                                     struct ldb_message *msg,
 242                                     struct dsdb_dn *dsdb_dn,
 243                                     struct drsuapi_DsReplicaLinkedAttribute **la_list,
 244                                     uint32_t *la_count)
 245 {
 246         struct drsuapi_DsReplicaLinkedAttribute *la;
 247         bool active;
 248         NTSTATUS status;
 249         WERROR werr;
 250
 251         (*la_list) = talloc_realloc(mem_ctx, *la_list, struct drsuapi_DsReplicaLinkedAttribute, (*la_count)+1);
 252         W_ERROR_HAVE_NO_MEMORY(*la_list);
 253
 254         la = &(*la_list)[*la_count];
 255
 256         la->identifier = get_object_identifier(*la_list, msg);
 257         W_ERROR_HAVE_NO_MEMORY(la->identifier);
 258
 259         active = (dsdb_dn_rmd_flags(dsdb_dn->dn) & DSDB_RMD_FLAG_DELETED) == 0;
 260
 261         la->attid = sa->attributeID_id;
 262         la->flags = active?DRSUAPI_DS_LINKED_ATTRIBUTE_FLAG_ACTIVE:0;
 263
 264         status = dsdb_get_extended_dn_nttime(dsdb_dn->dn, &la->originating_add_time, "RMD_ADDTIME");
 265         if (!NT_STATUS_IS_OK(status)) {
 266                 return ntstatus_to_werror(status);
 267         }
 268         status = dsdb_get_extended_dn_uint32(dsdb_dn->dn, &la->meta_data.version, "RMD_VERSION");
 269         if (!NT_STATUS_IS_OK(status)) {
 270                 return ntstatus_to_werror(status);
 271         }
 272         status = dsdb_get_extended_dn_nttime(dsdb_dn->dn, &la->meta_data.originating_change_time, "RMD_CHANGETIME");
 273         if (!NT_STATUS_IS_OK(status)) {
 274                 return ntstatus_to_werror(status);
 275         }
 276         status = dsdb_get_extended_dn_guid(dsdb_dn->dn, &la->meta_data.originating_invocation_id, "RMD_INVOCID");
 277         if (!NT_STATUS_IS_OK(status)) {
 278                 return ntstatus_to_werror(status);
 279         }
 280         status = dsdb_get_extended_dn_uint64(dsdb_dn->dn, &la->meta_data.originating_usn, "RMD_ORIGINATING_USN");
 281         if (!NT_STATUS_IS_OK(status)) {
 282                 return ntstatus_to_werror(status);
 283         }
 284
 285         werr = dsdb_dn_la_to_blob(sam_ctx, sa, schema, *la_list, dsdb_dn, &la->value.blob);
 286         W_ERROR_NOT_OK_RETURN(werr);
 287
 288         (*la_count)++;
 289         return WERR_OK;
 290 }
 291
 292
 293 /*
 294   add linked attributes from an object to the list of linked
 295   attributes in a getncchanges request
 296  */
 297 static WERROR get_nc_changes_add_links(struct ldb_context *sam_ctx,
 298                                        TALLOC_CTX *mem_ctx,
 299                                        struct ldb_dn *ncRoot_dn,
 300                                        struct dsdb_schema *schema,
 301                                        uint64_t highest_usn,
 302                                        uint32_t replica_flags,
 303                                        struct ldb_message *msg,
 304                                        struct drsuapi_DsReplicaLinkedAttribute **la_list,
 305                                        uint32_t *la_count)
 306 {
 307         int i;
 308         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
 309         uint64_t uSNChanged = ldb_msg_find_attr_as_int(msg, "uSNChanged", -1);
 310
 311         for (i=0; i<msg->num_elements; i++) {
 312                 struct ldb_message_element *el = &msg->elements[i];
 313                 const struct dsdb_attribute *sa;
 314                 int j;
 315
 316                 sa = dsdb_attribute_by_lDAPDisplayName(schema, el->name);
 317
 318                 if (!sa || sa->linkID == 0 || (sa->linkID & 1)) {
 319                         /* we only want forward links */
 320                         continue;
 321                 }
 322
 323                 if (el->num_values && !dsdb_dn_is_upgraded_link_val(&el->values[0])) {
 324                         /* its an old style link, it will have been
 325                          * sent in the main replication data */
 326                         continue;
 327                 }
 328
 329                 for (j=0; j<el->num_values; j++) {
 330                         struct dsdb_dn *dsdb_dn;
 331                         uint64_t local_usn;
 332                         NTSTATUS status;
 333                         WERROR werr;
 334
 335                         dsdb_dn = dsdb_dn_parse(tmp_ctx, sam_ctx, &el->values[j], sa->syntax->ldap_oid);
 336                         if (dsdb_dn == NULL) {
 337                                 DEBUG(1,(__location__ ": Failed to parse DN for %s in %s\n",
 338                                          el->name, ldb_dn_get_linearized(msg->dn)));
 339                                 talloc_free(tmp_ctx);
 340                                 return WERR_DS_DRA_INTERNAL_ERROR;
 341                         }
 342
 343                         status = dsdb_get_extended_dn_uint64(dsdb_dn->dn, &local_usn, "RMD_LOCAL_USN");
 344                         if (!NT_STATUS_IS_OK(status)) {
 345                                 /* this can happen for attributes
 346                                    given to us with old style meta
 347                                    data */
 348                                 continue;
 349                         }
 350
 351                         if (local_usn > uSNChanged) {
 352                                 DEBUG(1,(__location__ ": uSNChanged less than RMD_LOCAL_USN for %s on %s\n",
 353                                          el->name, ldb_dn_get_linearized(msg->dn)));
 354                                 talloc_free(tmp_ctx);
 355                                 return WERR_DS_DRA_INTERNAL_ERROR;
 356                         }
 357
 358                         if (local_usn < highest_usn) {
 359                                 continue;
 360                         }
 361
 362                         werr = get_nc_changes_add_la(mem_ctx, sam_ctx, schema, sa, msg,
 363                                                      dsdb_dn, la_list, la_count);
 364                         if (!W_ERROR_IS_OK(werr)) {
 365                                 talloc_free(tmp_ctx);
 366                                 return werr;
 367                         }
 368                 }
 369         }
 370
 371         talloc_free(tmp_ctx);
 372         return WERR_OK;
 373 }
 374
 375 /*
 376   load replUpToDateVector from a DN
 377  */
 378 static WERROR load_udv(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx,
 379                        struct ldb_dn *dn, struct replUpToDateVectorBlob *ouv)
 380 {
 381         const char *attrs[] = { "replUpToDateVector", NULL };
 382         struct ldb_result *res = NULL;
 383         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
 384         struct ldb_message_element *el;
 385         enum ndr_err_code ndr_err;
 386
 387         ZERO_STRUCTP(ouv);
 388
 389         if (ldb_search(sam_ctx, tmp_ctx, &res, dn, LDB_SCOPE_BASE, attrs, NULL) != LDB_SUCCESS ||
 390             res->count < 1) {
 391                 DEBUG(0,("load_udv: failed to read partition object\n"));
 392                 talloc_free(tmp_ctx);
 393                 return WERR_DS_DRA_INTERNAL_ERROR;
 394         }
 395
 396         el = ldb_msg_find_element(res->msgs[0], "replUpToDateVector");
 397         if (el == NULL || el->num_values < 1) {
 398                 talloc_free(tmp_ctx);
 399                 ouv->version = 2;
 400                 return WERR_OK;
 401         }
 402
 403         ndr_err = ndr_pull_struct_blob(&el->values[0],
 404                                        mem_ctx, lp_iconv_convenience(ldb_get_opaque(sam_ctx, "loadparm")),
 405                                        ouv,
 406                                        (ndr_pull_flags_fn_t)ndr_pull_replUpToDateVectorBlob);
 407         talloc_free(tmp_ctx);
 408         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
 409                 DEBUG(0,(__location__ ": Failed to parse replUpToDateVector for %s\n",
 410                          ldb_dn_get_linearized(dn)));
 411                 return WERR_DS_DRA_INTERNAL_ERROR;
 412         }
 413
 414         return WERR_OK;
 415
 416 }
 417
 418 /*
 419   fill in the cursors return based on the replUpToDateVector for the ncRoot_dn
 420  */
 421 static WERROR get_nc_changes_udv(struct ldb_context *sam_ctx,
 422                                  struct ldb_dn *ncRoot_dn,
 423                                  struct drsuapi_DsReplicaCursor2CtrEx *udv,
 424                                  uint64_t highestUSN)
 425 {
 426         WERROR werr;
 427         struct drsuapi_DsReplicaCursor2 *tmp_cursor;
 428         NTTIME now;
 429         time_t t = time(NULL);
 430         struct replUpToDateVectorBlob ouv;
 431         int i;
 432
 433         werr = load_udv(sam_ctx, udv, ncRoot_dn, &ouv);
 434         if (!W_ERROR_IS_OK(werr)) {
 435                 return werr;
 436         }
 437
 438         tmp_cursor = talloc(udv, struct drsuapi_DsReplicaCursor2);
 439         tmp_cursor->source_dsa_invocation_id = *(samdb_ntds_invocation_id(sam_ctx));
 440         tmp_cursor->highest_usn = highestUSN;
 441         unix_to_nt_time(&now, t);
 442         tmp_cursor->last_sync_success = now;
 443
 444         udv->count = ouv.ctr.ctr2.count;
 445         udv->cursors = talloc_steal(udv, ouv.ctr.ctr2.cursors);
 446
 447         for (i=0; i<udv->count; i++) {
 448                 if (GUID_equal(&tmp_cursor->source_dsa_invocation_id,
 449                                &udv->cursors[i].source_dsa_invocation_id)) {
 450                         udv->cursors[i] = *tmp_cursor;
 451                         break;
 452                 }
 453         }
 454         if (i == udv->count) {
 455                 udv->cursors = talloc_realloc(udv, udv->cursors, struct drsuapi_DsReplicaCursor2, udv->count+1);
 456                 if (!udv->cursors) {
 457                         return WERR_DS_DRA_INTERNAL_ERROR;
 458                 }
 459                 udv->cursors[udv->count] = *tmp_cursor;
 460                 udv->count++;
 461         }
 462
 463         qsort(udv->cursors, udv->count,
 464               sizeof(struct drsuapi_DsReplicaCursor2),
 465               (comparison_fn_t)drsuapi_DsReplicaCursor2_compare);
 466
 467         return WERR_OK;
 468 }
 469
 470
 471 /* comparison function for linked attributes - see CompareLinks() in
 472  * MS-DRSR section 4.1.10.5.17 */
 473 static int linked_attribute_compare(const struct drsuapi_DsReplicaLinkedAttribute *la1,
 474                                     const struct drsuapi_DsReplicaLinkedAttribute *la2,
 475                                     struct ldb_context *sam_ctx)
 476 {
 477         int c;
 478         WERROR werr;
 479         TALLOC_CTX *tmp_ctx;
 480         const struct dsdb_schema *schema;
 481         const struct dsdb_attribute *schema_attrib;
 482         struct dsdb_dn *dn1, *dn2;
 483         struct GUID guid1, guid2;
 484         NTSTATUS status;
 485
 486         c = GUID_compare(&la1->identifier->guid,
 487                          &la2->identifier->guid);
 488         if (c != 0) return c;
 489
 490         if (la1->attid != la2->attid) {
 491                 return la1->attid < la2->attid? -1:1;
 492         }
 493
 494         if ((la1->flags & DRSUAPI_DS_LINKED_ATTRIBUTE_FLAG_ACTIVE) !=
 495             (la2->flags & DRSUAPI_DS_LINKED_ATTRIBUTE_FLAG_ACTIVE)) {
 496                 return (la1->flags & DRSUAPI_DS_LINKED_ATTRIBUTE_FLAG_ACTIVE)? 1:-1;
 497         }
 498
 499         /* we need to get the target GUIDs to compare */
 500         tmp_ctx = talloc_new(sam_ctx);
 501
 502         schema = dsdb_get_schema(sam_ctx);
 503         schema_attrib = dsdb_attribute_by_attributeID_id(schema, la1->attid);
 504
 505         werr = dsdb_dn_la_from_blob(sam_ctx, schema_attrib, schema, tmp_ctx, la1->value.blob, &dn1);
 506         if (!W_ERROR_IS_OK(werr)) {
 507                 DEBUG(0,(__location__ ": Bad la1 blob in sort\n"));
 508                 talloc_free(tmp_ctx);
 509                 return 0;
 510         }
 511
 512         werr = dsdb_dn_la_from_blob(sam_ctx, schema_attrib, schema, tmp_ctx, la2->value.blob, &dn2);
 513         if (!W_ERROR_IS_OK(werr)) {
 514                 DEBUG(0,(__location__ ": Bad la2 blob in sort\n"));
 515                 talloc_free(tmp_ctx);
 516                 return 0;
 517         }
 518
 519         status = dsdb_get_extended_dn_guid(dn1->dn, &guid1, "GUID");
 520         if (!NT_STATUS_IS_OK(status)) {
 521                 DEBUG(0,(__location__ ": Bad la1 guid in sort\n"));
 522                 talloc_free(tmp_ctx);
 523                 return 0;
 524         }
 525         status = dsdb_get_extended_dn_guid(dn2->dn, &guid2, "GUID");
 526         if (!NT_STATUS_IS_OK(status)) {
 527                 DEBUG(0,(__location__ ": Bad la2 guid in sort\n"));
 528                 talloc_free(tmp_ctx);
 529                 return 0;
 530         }
 531
 532         talloc_free(tmp_ctx);
 533
 534         return GUID_compare(&guid1, &guid2);
 535 }
 536
 537
 538 /*
 539   sort the objects we send by tree order
 540  */
 541 static int site_res_cmp_parent_order(const struct ldb_message **m1, const struct ldb_message **m2)
 542 {
 543         return ldb_dn_compare((*m2)->dn, (*m1)->dn);
 544 }
 545
 546 /*
 547   sort the objects we send first by uSNChanged
 548  */
 549 static int site_res_cmp_usn_order(const struct ldb_message **m1, const struct ldb_message **m2)
 550 {
 551         unsigned usnchanged1, usnchanged2;
 552         unsigned cn1, cn2;
 553         cn1 = ldb_dn_get_comp_num((*m1)->dn);
 554         cn2 = ldb_dn_get_comp_num((*m2)->dn);
 555         if (cn1 != cn2) {
 556                 return cn1 > cn2 ? 1 : -1;
 557         }
 558         usnchanged1 = ldb_msg_find_attr_as_uint(*m1, "uSNChanged", 0);
 559         usnchanged2 = ldb_msg_find_attr_as_uint(*m2, "uSNChanged", 0);
 560         if (usnchanged1 == usnchanged2) {
 561                 return 0;
 562         }
 563         return usnchanged1 > usnchanged2 ? 1 : -1;
 564 }
 565
 566
 567 /* state of a partially completed getncchanges call */
 568 struct drsuapi_getncchanges_state {
 569         struct ldb_result *site_res;
 570         uint32_t num_sent;
 571         struct ldb_dn *ncRoot_dn;
 572         uint64_t min_usn;
 573         uint64_t highest_usn;
 574         struct ldb_dn *last_dn;
 575         struct drsuapi_DsReplicaLinkedAttribute *la_list;
 576         uint32_t la_count;
 577 };
 578
 579 /*
 580   drsuapi_DsGetNCChanges
 581
 582   see MS-DRSR 4.1.10.5.2 for basic logic of this function
 583 */
 584 WERROR dcesrv_drsuapi_DsGetNCChanges(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
 585                                      struct drsuapi_DsGetNCChanges *r)
 586 {
 587         struct drsuapi_DsReplicaObjectIdentifier *ncRoot;
 588         int ret;
 589         int i;
 590         struct dsdb_schema *schema;
 591         struct drsuapi_DsReplicaOIDMapping_Ctr *ctr;
 592         struct drsuapi_DsReplicaObjectListItemEx **currentObject;
 593         NTSTATUS status;
 594         DATA_BLOB session_key;
 595         const char *attrs[] = { "*", "distinguishedName",
 596                                 "nTSecurityDescriptor",
 597                                 "parentGUID",
 598                                 "replPropertyMetaData",
 599                                 "unicodePwd",
 600                                 "dBCSPwd",
 601                                 "ntPwdHistory",
 602                                 "lmPwdHistory",
 603                                 "supplementalCredentials",
 604                                 NULL };
 605         WERROR werr;
 606         struct dcesrv_handle *h;
 607         struct drsuapi_bind_state *b_state;
 608         struct drsuapi_getncchanges_state *getnc_state;
 609         struct drsuapi_DsGetNCChangesRequest8 *req8;
 610         uint32_t options;
 611         uint32_t max_objects;
 612
 613         DCESRV_PULL_HANDLE_WERR(h, r->in.bind_handle, DRSUAPI_BIND_HANDLE);
 614         b_state = h->data;
 615
 616         *r->out.level_out = 6;
 617         /* TODO: linked attributes*/
 618         r->out.ctr->ctr6.linked_attributes_count = 0;
 619         r->out.ctr->ctr6.linked_attributes = NULL;
 620
 621         r->out.ctr->ctr6.object_count = 0;
 622         r->out.ctr->ctr6.nc_object_count = 0;
 623         r->out.ctr->ctr6.more_data = false;
 624         r->out.ctr->ctr6.uptodateness_vector = NULL;
 625
 626         /* a RODC doesn't allow for any replication */
 627         if (samdb_rodc(ldb_get_opaque(b_state->sam_ctx, "loadparm"))) {
 628                 DEBUG(0,(__location__ ": DsGetNCChanges attempt on RODC\n"));
 629                 return WERR_DS_DRA_SOURCE_DISABLED;
 630         }
 631
 632         /* Check request revision.
 633            TODO: Adding mappings to req8 from the other levels
 634          */
 635         if (r->in.level != 8) {
 636                 DEBUG(0,(__location__ ": Request for DsGetNCChanges with unsupported level %u\n",
 637                          r->in.level));
 638                 return WERR_REVISION_MISMATCH;
 639         }
 640
 641         req8 = &r->in.req->req8;
 642
 643         /* Perform access checks. */
 644         /* TODO: we need to support a sync on a specific non-root
 645          * DN. We'll need to find the real partition root here */
 646         ncRoot = req8->naming_context;
 647         if (ncRoot == NULL) {
 648                 DEBUG(0,(__location__ ": Request for DsGetNCChanges with no NC\n"));
 649                 return WERR_DS_DRA_INVALID_PARAMETER;
 650         }
 651
 652         if (samdb_ntds_options(b_state->sam_ctx, &options) != LDB_SUCCESS) {
 653                 return WERR_DS_DRA_INTERNAL_ERROR;
 654         }
 655
 656         if ((options & DS_NTDSDSA_OPT_DISABLE_OUTBOUND_REPL) &&
 657             !(req8->replica_flags & DRSUAPI_DRS_SYNC_FORCED)) {
 658                 return WERR_DS_DRA_SOURCE_DISABLED;
 659         }
 660
 661
 662         if (req8->replica_flags & DRSUAPI_DS_REPLICA_NEIGHBOUR_FULL_SYNC_PACKET) {
 663                 /* Ignore the _in_ uptpdateness vector*/
 664                 req8->uptodateness_vector = NULL;
 665         }
 666
 667         werr = drs_security_level_check(dce_call, "DsGetNCChanges");
 668         if (!W_ERROR_IS_OK(werr)) {
 669                 return werr;
 670         }
 671
 672         getnc_state = b_state->getncchanges_state;
 673
 674         /* see if a previous replication has been abandoned */
 675         if (getnc_state) {
 676                 struct ldb_dn *new_dn = ldb_dn_new(getnc_state, b_state->sam_ctx, ncRoot->dn);
 677                 if (ldb_dn_compare(new_dn, getnc_state->ncRoot_dn) != 0) {
 678                         DEBUG(0,(__location__ ": DsGetNCChanges 2nd replication on different DN %s %s (last_dn %s)\n",
 679                                  ldb_dn_get_linearized(new_dn),
 680                                  ldb_dn_get_linearized(getnc_state->ncRoot_dn),
 681                                  ldb_dn_get_linearized(getnc_state->last_dn)));
 682                         talloc_free(getnc_state);
 683                         getnc_state = NULL;
 684                 }
 685         }
 686
 687         if (getnc_state == NULL) {
 688                 getnc_state = talloc_zero(b_state, struct drsuapi_getncchanges_state);
 689                 if (getnc_state == NULL) {
 690                         return WERR_NOMEM;
 691                 }
 692                 b_state->getncchanges_state = getnc_state;
 693                 getnc_state->ncRoot_dn = ldb_dn_new(getnc_state, b_state->sam_ctx, ncRoot->dn);
 694         }
 695
 696         if (!ldb_dn_validate(getnc_state->ncRoot_dn) ||
 697             ldb_dn_is_null(getnc_state->ncRoot_dn)) {
 698                 DEBUG(0,(__location__ ": Bad DN '%s'\n", ncRoot->dn));
 699                 return WERR_DS_DRA_INVALID_PARAMETER;
 700         }
 701
 702         /* we need the session key for encrypting password attributes */
 703         status = dcesrv_inherited_session_key(dce_call->conn, &session_key);
 704         if (!NT_STATUS_IS_OK(status)) {
 705                 DEBUG(0,(__location__ ": Failed to get session key\n"));
 706                 return WERR_DS_DRA_INTERNAL_ERROR;
 707         }
 708
 709         /* we don't yet support extended operations */
 710         if (req8->extended_op != DRSUAPI_EXOP_NONE) {
 711                 DEBUG(0,(__location__ ": Request for DsGetNCChanges extended op 0x%x\n",
 712                          (unsigned)req8->extended_op));
 713                 return WERR_DS_DRA_NOT_SUPPORTED;
 714         }
 715
 716         /*
 717            TODO: MS-DRSR section 4.1.10.1.1
 718            Work out if this is the start of a new cycle */
 719
 720         if (getnc_state->site_res == NULL) {
 721                 char* search_filter;
 722                 enum ldb_scope scope = LDB_SCOPE_SUBTREE;
 723                 const char *extra_filter;
 724
 725                 extra_filter = lp_parm_string(dce_call->conn->dce_ctx->lp_ctx, NULL, "drs", "object filter");
 726
 727                 getnc_state->min_usn = req8->highwatermark.highest_usn;
 728
 729                 /* Construct response. */
 730                 search_filter = talloc_asprintf(mem_ctx,
 731                                                 "(uSNChanged>=%llu)",
 732                                                 (unsigned long long)(getnc_state->min_usn+1));
 733
 734                 if (extra_filter) {
 735                         search_filter = talloc_asprintf(mem_ctx, "(&%s(%s))", search_filter, extra_filter);
 736                 }
 737
 738                 if (req8->replica_flags & DRSUAPI_DS_REPLICA_NEIGHBOUR_CRITICAL_ONLY) {
 739                         search_filter = talloc_asprintf(mem_ctx,
 740                                                         "(&%s(isCriticalSystemObject=TRUE))",
 741                                                         search_filter);
 742                 }
 743
 744                 if (req8->replica_flags & DRSUAPI_DS_REPLICA_NEIGHBOUR_ASYNC_REP) {
 745                         scope = LDB_SCOPE_BASE;
 746                 }
 747
 748                 DEBUG(1,(__location__ ": getncchanges on %s using filter %s\n",
 749                          ldb_dn_get_linearized(getnc_state->ncRoot_dn), search_filter));
 750                 ret = drsuapi_search_with_extended_dn(b_state->sam_ctx, getnc_state, &getnc_state->site_res,
 751                                                       getnc_state->ncRoot_dn, scope, attrs,
 752                                                       search_filter);
 753                 if (ret != LDB_SUCCESS) {
 754                         return WERR_DS_DRA_INTERNAL_ERROR;
 755                 }
 756
 757                 if (req8->replica_flags & DRSUAPI_DS_REPLICA_NEIGHBOUR_RETURN_OBJECT_PARENTS) {
 758                         qsort(getnc_state->site_res->msgs,
 759                               getnc_state->site_res->count,
 760                               sizeof(getnc_state->site_res->msgs[0]),
 761                               (comparison_fn_t)site_res_cmp_parent_order);
 762                 } else {
 763                         qsort(getnc_state->site_res->msgs,
 764                               getnc_state->site_res->count,
 765                               sizeof(getnc_state->site_res->msgs[0]),
 766                               (comparison_fn_t)site_res_cmp_usn_order);
 767                 }
 768
 769         }
 770
 771         /* Prefix mapping */
 772         schema = dsdb_get_schema(b_state->sam_ctx);
 773         if (!schema) {
 774                 DEBUG(0,("No schema in sam_ctx\n"));
 775                 return WERR_DS_DRA_INTERNAL_ERROR;
 776         }
 777
 778         r->out.ctr->ctr6.naming_context = talloc(mem_ctx, struct drsuapi_DsReplicaObjectIdentifier);
 779         *r->out.ctr->ctr6.naming_context = *ncRoot;
 780
 781         if (dsdb_find_guid_by_dn(b_state->sam_ctx, getnc_state->ncRoot_dn,
 782                                  &r->out.ctr->ctr6.naming_context->guid) != LDB_SUCCESS) {
 783                 DEBUG(0,(__location__ ": Failed to find GUID of ncRoot_dn %s\n",
 784                          ldb_dn_get_linearized(getnc_state->ncRoot_dn)));
 785                 return WERR_DS_DRA_INTERNAL_ERROR;
 786         }
 787
 788         /* find the SID if there is one */
 789         dsdb_find_sid_by_dn(b_state->sam_ctx, getnc_state->ncRoot_dn, &r->out.ctr->ctr6.naming_context->sid);
 790
 791         dsdb_get_oid_mappings_drsuapi(schema, true, mem_ctx, &ctr);
 792         r->out.ctr->ctr6.mapping_ctr = *ctr;
 793
 794         r->out.ctr->ctr6.source_dsa_guid = *(samdb_ntds_objectGUID(b_state->sam_ctx));
 795         r->out.ctr->ctr6.source_dsa_invocation_id = *(samdb_ntds_invocation_id(b_state->sam_ctx));
 796
 797         r->out.ctr->ctr6.old_highwatermark = req8->highwatermark;
 798         r->out.ctr->ctr6.new_highwatermark = req8->highwatermark;
 799
 800         r->out.ctr->ctr6.first_object = NULL;
 801         currentObject = &r->out.ctr->ctr6.first_object;
 802
 803         /* use this to force single objects at a time, which is useful
 804          * for working out what object is giving problems
 805          */
 806         max_objects = lp_parm_int(dce_call->conn->dce_ctx->lp_ctx, NULL, "drs", "max object sync", 1000);
 807         if (req8->max_object_count < max_objects) {
 808                 max_objects = req8->max_object_count;
 809         }
 810
 811         for(i=getnc_state->num_sent;
 812             i<getnc_state->site_res->count &&
 813                     (r->out.ctr->ctr6.object_count < max_objects);
 814             i++) {
 815                 int uSN;
 816                 struct drsuapi_DsReplicaObjectListItemEx *obj;
 817                 struct ldb_message *msg = getnc_state->site_res->msgs[i];
 818
 819                 obj = talloc_zero(mem_ctx, struct drsuapi_DsReplicaObjectListItemEx);
 820
 821                 werr = get_nc_changes_build_object(obj, msg,
 822                                                    b_state->sam_ctx, getnc_state->ncRoot_dn,
 823                                                    schema, &session_key, getnc_state->min_usn,
 824                                                    req8->replica_flags);
 825                 if (!W_ERROR_IS_OK(werr)) {
 826                         return werr;
 827                 }
 828
 829                 werr = get_nc_changes_add_links(b_state->sam_ctx, getnc_state,
 830                                                 getnc_state->ncRoot_dn,
 831                                                 schema, getnc_state->min_usn,
 832                                                 req8->replica_flags,
 833                                                 msg,
 834                                                 &getnc_state->la_list,
 835                                                 &getnc_state->la_count);
 836                 if (!W_ERROR_IS_OK(werr)) {
 837                         return werr;
 838                 }
 839
 840                 uSN = ldb_msg_find_attr_as_int(msg, "uSNChanged", -1);
 841                 if (uSN > r->out.ctr->ctr6.new_highwatermark.tmp_highest_usn) {
 842                         r->out.ctr->ctr6.new_highwatermark.tmp_highest_usn = uSN;
 843                 }
 844                 if (uSN > getnc_state->highest_usn) {
 845                         getnc_state->highest_usn = uSN;
 846                 }
 847
 848                 if (obj->meta_data_ctr == NULL) {
 849                         DEBUG(0,(__location__ ": getncchanges skipping send of object %s\n",
 850                                  ldb_dn_get_linearized(msg->dn)));
 851                         /* no attributes to send */
 852                         talloc_free(obj);
 853                         continue;
 854                 }
 855
 856                 r->out.ctr->ctr6.object_count++;
 857
 858                 *currentObject = obj;
 859                 currentObject = &obj->next_object;
 860
 861                 talloc_free(getnc_state->last_dn);
 862                 getnc_state->last_dn = ldb_dn_copy(getnc_state, msg->dn);
 863
 864                 DEBUG(8,(__location__ ": replicating object %s\n", ldb_dn_get_linearized(msg->dn)));
 865         }
 866
 867         getnc_state->num_sent += r->out.ctr->ctr6.object_count;
 868
 869         r->out.ctr->ctr6.nc_object_count = getnc_state->site_res->count;
 870
 871         /* the client can us to call UpdateRefs on its behalf to
 872            re-establish monitoring of the NC */
 873         if ((req8->replica_flags & DRSUAPI_DRS_ADD_REF) &&
 874             !GUID_all_zero(&req8->destination_dsa_guid)) {
 875                 struct drsuapi_DsReplicaUpdateRefsRequest1 ureq;
 876                 ureq.naming_context = ncRoot;
 877                 ureq.dest_dsa_dns_name = talloc_asprintf(mem_ctx, "%s._msdcs.%s",
 878                                                          GUID_string(mem_ctx, &req8->destination_dsa_guid),
 879                                                          lp_realm(dce_call->conn->dce_ctx->lp_ctx));
 880                 if (!ureq.dest_dsa_dns_name) {
 881                         return WERR_NOMEM;
 882                 }
 883                 ureq.dest_dsa_guid = req8->destination_dsa_guid;
 884                 ureq.options = DRSUAPI_DS_REPLICA_UPDATE_ADD_REFERENCE |
 885                         DRSUAPI_DS_REPLICA_UPDATE_ASYNCHRONOUS_OPERATION |
 886                         DRSUAPI_DS_REPLICA_UPDATE_GETCHG_CHECK;
 887                 werr = drsuapi_UpdateRefs(b_state, mem_ctx, &ureq);
 888                 if (!W_ERROR_IS_OK(werr)) {
 889                         DEBUG(0,(__location__ ": Failed UpdateRefs in DsGetNCChanges - %s\n",
 890                                  win_errstr(werr)));
 891                 }
 892         }
 893
 894         if (i < getnc_state->site_res->count) {
 895                 r->out.ctr->ctr6.more_data = true;
 896         } else {
 897                 r->out.ctr->ctr6.linked_attributes_count = getnc_state->la_count;
 898                 r->out.ctr->ctr6.linked_attributes = talloc_steal(mem_ctx, getnc_state->la_list);
 899
 900                 ldb_qsort(r->out.ctr->ctr6.linked_attributes, r->out.ctr->ctr6.linked_attributes_count,
 901                           sizeof(r->out.ctr->ctr6.linked_attributes[0]),
 902                           b_state->sam_ctx, (ldb_qsort_cmp_fn_t)linked_attribute_compare);
 903
 904                 r->out.ctr->ctr6.uptodateness_vector = talloc(mem_ctx, struct drsuapi_DsReplicaCursor2CtrEx);
 905                 r->out.ctr->ctr6.uptodateness_vector->version = 2;
 906                 r->out.ctr->ctr6.uptodateness_vector->reserved1 = 0;
 907                 r->out.ctr->ctr6.uptodateness_vector->reserved2 = 0;
 908
 909                 r->out.ctr->ctr6.new_highwatermark.highest_usn = r->out.ctr->ctr6.new_highwatermark.tmp_highest_usn;
 910
 911                 werr = get_nc_changes_udv(b_state->sam_ctx, getnc_state->ncRoot_dn,
 912                                           r->out.ctr->ctr6.uptodateness_vector,
 913                                           getnc_state->highest_usn);
 914                 if (!W_ERROR_IS_OK(werr)) {
 915                         return werr;
 916                 }
 917
 918                 talloc_free(getnc_state);
 919                 b_state->getncchanges_state = NULL;
 920         }
 921
 922         DEBUG(r->out.ctr->ctr6.more_data?2:1,
 923               ("DsGetNCChanges with uSNChanged >= %llu flags 0x%08x on %s gave %u objects (done %d/%d la=%d)\n",
 924                (unsigned long long)(req8->highwatermark.highest_usn+1),
 925                req8->replica_flags,
 926                ncRoot->dn, r->out.ctr->ctr6.object_count,
 927                i, r->out.ctr->ctr6.more_data?getnc_state->site_res->count:i,
 928                r->out.ctr->ctr6.linked_attributes_count));
 929
 930 #if 0
 931         if (!r->out.ctr->ctr6.more_data) {
 932                 NDR_PRINT_FUNCTION_DEBUG(drsuapi_DsGetNCChanges, NDR_BOTH, r);
 933         }
 934 #endif
 935
 936         return WERR_OK;
 937 }