source3/libnet/libnet_dssync.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    Copyright (C) Stefan (metze) Metzmacher 2005
   5    Copyright (C) Guenther Deschner 2008
   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 "libnet/libnet.h"
  24
  25 /****************************************************************
  26 ****************************************************************/
  27
  28 static int libnet_dssync_free_context(struct dssync_context *ctx)
  29 {
  30         if (!ctx) {
  31                 return 0;
  32         }
  33
  34         if (is_valid_policy_hnd(&ctx->bind_handle) && ctx->cli) {
  35                 rpccli_drsuapi_DsUnbind(ctx->cli, ctx, &ctx->bind_handle, NULL);
  36         }
  37
  38         return 0;
  39 }
  40
  41 /****************************************************************
  42 ****************************************************************/
  43
  44 NTSTATUS libnet_dssync_init_context(TALLOC_CTX *mem_ctx,
  45                                     struct dssync_context **ctx_p)
  46 {
  47         struct dssync_context *ctx;
  48
  49         ctx = TALLOC_ZERO_P(mem_ctx, struct dssync_context);
  50         NT_STATUS_HAVE_NO_MEMORY(ctx);
  51
  52         talloc_set_destructor(ctx, libnet_dssync_free_context);
  53
  54         *ctx_p = ctx;
  55
  56         return NT_STATUS_OK;
  57 }
  58
  59 /****************************************************************
  60 ****************************************************************/
  61
  62 static DATA_BLOB *decrypt_attr_val(TALLOC_CTX *mem_ctx,
  63                                    DATA_BLOB *session_key,
  64                                    uint32_t rid,
  65                                    enum drsuapi_DsAttributeId id,
  66                                    DATA_BLOB *raw_data)
  67 {
  68         bool rcrypt = false;
  69         DATA_BLOB out_data;
  70
  71         ZERO_STRUCT(out_data);
  72
  73         switch (id) {
  74                 case DRSUAPI_ATTRIBUTE_dBCSPwd:
  75                 case DRSUAPI_ATTRIBUTE_unicodePwd:
  76                 case DRSUAPI_ATTRIBUTE_ntPwdHistory:
  77                 case DRSUAPI_ATTRIBUTE_lmPwdHistory:
  78                         rcrypt  = true;
  79                         break;
  80                 case DRSUAPI_ATTRIBUTE_supplementalCredentials:
  81                 case DRSUAPI_ATTRIBUTE_priorValue:
  82                 case DRSUAPI_ATTRIBUTE_currentValue:
  83                 case DRSUAPI_ATTRIBUTE_trustAuthOutgoing:
  84                 case DRSUAPI_ATTRIBUTE_trustAuthIncoming:
  85                 case DRSUAPI_ATTRIBUTE_initialAuthOutgoing:
  86                 case DRSUAPI_ATTRIBUTE_initialAuthIncoming:
  87                         break;
  88                 default:
  89                         return raw_data;
  90         }
  91
  92         out_data = decrypt_drsuapi_blob(mem_ctx, session_key, rcrypt,
  93                                           rid, raw_data);
  94
  95         if (out_data.length) {
  96                 return (DATA_BLOB *)talloc_memdup(mem_ctx, &out_data, sizeof(DATA_BLOB));
  97         }
  98
  99         return raw_data;
 100 }
 101
 102 /****************************************************************
 103 ****************************************************************/
 104
 105 static void parse_obj_identifier(struct drsuapi_DsReplicaObjectIdentifier *id,
 106                                  uint32_t *rid)
 107 {
 108         if (!id || !rid) {
 109                 return;
 110         }
 111
 112         *rid = 0;
 113
 114         if (id->sid.num_auths > 0) {
 115                 *rid = id->sid.sub_auths[id->sid.num_auths - 1];
 116         }
 117 }
 118
 119 /****************************************************************
 120 ****************************************************************/
 121
 122 static void parse_obj_attribute(TALLOC_CTX *mem_ctx,
 123                                 DATA_BLOB *session_key,
 124                                 uint32_t rid,
 125                                 struct drsuapi_DsReplicaAttribute *attr)
 126 {
 127         int i = 0;
 128
 129         for (i=0; i<attr->value_ctr.num_values; i++) {
 130
 131                 DATA_BLOB *plain_data = NULL;
 132
 133                 plain_data = decrypt_attr_val(mem_ctx,
 134                                               session_key,
 135                                               rid,
 136                                               attr->attid,
 137                                               attr->value_ctr.values[i].blob);
 138
 139                 attr->value_ctr.values[i].blob = plain_data;
 140         }
 141 }
 142
 143 /****************************************************************
 144 ****************************************************************/
 145
 146 static void libnet_dssync_decrypt_attributes(TALLOC_CTX *mem_ctx,
 147                                              DATA_BLOB *session_key,
 148                                              struct drsuapi_DsReplicaObjectListItemEx *cur)
 149 {
 150         for (; cur; cur = cur->next_object) {
 151
 152                 uint32_t i;
 153                 uint32_t rid = 0;
 154
 155                 parse_obj_identifier(cur->object.identifier, &rid);
 156
 157                 for (i=0; i < cur->object.attribute_ctr.num_attributes; i++) {
 158
 159                         struct drsuapi_DsReplicaAttribute *attr;
 160
 161                         attr = &cur->object.attribute_ctr.attributes[i];
 162
 163                         if (attr->value_ctr.num_values < 1) {
 164                                 continue;
 165                         }
 166
 167                         if (!attr->value_ctr.values[0].blob) {
 168                                 continue;
 169                         }
 170
 171                         parse_obj_attribute(mem_ctx,
 172                                             session_key,
 173                                             rid,
 174                                             attr);
 175                 }
 176         }
 177 }
 178 /****************************************************************
 179 ****************************************************************/
 180
 181 static NTSTATUS libnet_dssync_bind(TALLOC_CTX *mem_ctx,
 182                                    struct dssync_context *ctx)
 183 {
 184         NTSTATUS status;
 185         WERROR werr;
 186
 187         struct GUID bind_guid;
 188         struct drsuapi_DsBindInfoCtr bind_info;
 189         struct drsuapi_DsBindInfo28 info28;
 190
 191         ZERO_STRUCT(info28);
 192
 193         GUID_from_string(DRSUAPI_DS_BIND_GUID, &bind_guid);
 194
 195         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_BASE;
 196         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_ASYNC_REPLICATION;
 197         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_REMOVEAPI;
 198         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_MOVEREQ_V2;
 199         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_GETCHG_COMPRESS;
 200         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_DCINFO_V1;
 201         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_RESTORE_USN_OPTIMIZATION;
 202         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_KCC_EXECUTE;
 203         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_ADDENTRY_V2;
 204         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_LINKED_VALUE_REPLICATION;
 205         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_DCINFO_V2;
 206         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_INSTANCE_TYPE_NOT_REQ_ON_MOD;
 207         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_CRYPTO_BIND;
 208         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_GET_REPL_INFO;
 209         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_STRONG_ENCRYPTION;
 210         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_DCINFO_V01;
 211         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_TRANSITIVE_MEMBERSHIP;
 212         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_ADD_SID_HISTORY;
 213         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_POST_BETA3;
 214         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_GET_MEMBERSHIPS2;
 215         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREQ_V6;
 216         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_NONDOMAIN_NCS;
 217         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREQ_V8;
 218         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREPLY_V5;
 219         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREPLY_V6;
 220         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_ADDENTRYREPLY_V3;
 221         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREPLY_V7;
 222         info28.supported_extensions     |= DRSUAPI_SUPPORTED_EXTENSION_VERIFY_OBJECT;
 223         info28.site_guid                = GUID_zero();
 224         info28.pid                      = 508;
 225         info28.repl_epoch               = 0;
 226
 227         bind_info.length = 28;
 228         bind_info.info.info28 = info28;
 229
 230         status = rpccli_drsuapi_DsBind(ctx->cli, mem_ctx,
 231                                        &bind_guid,
 232                                        &bind_info,
 233                                        &ctx->bind_handle,
 234                                        &werr);
 235
 236         if (!NT_STATUS_IS_OK(status)) {
 237                 return status;
 238         }
 239
 240         if (!W_ERROR_IS_OK(werr)) {
 241                 return werror_to_ntstatus(werr);
 242         }
 243
 244         ZERO_STRUCT(ctx->remote_info28);
 245         switch (bind_info.length) {
 246         case 24: {
 247                 struct drsuapi_DsBindInfo24 *info24;
 248                 info24 = &bind_info.info.info24;
 249                 ctx->remote_info28.site_guid            = info24->site_guid;
 250                 ctx->remote_info28.supported_extensions = info24->supported_extensions;
 251                 ctx->remote_info28.pid                  = info24->pid;
 252                 ctx->remote_info28.repl_epoch           = 0;
 253                 break;
 254         }
 255         case 28:
 256                 ctx->remote_info28 = bind_info.info.info28;
 257                 break;
 258         case 48: {
 259                 struct drsuapi_DsBindInfo48 *info48;
 260                 info48 = &bind_info.info.info48;
 261                 ctx->remote_info28.site_guid            = info48->site_guid;
 262                 ctx->remote_info28.supported_extensions = info48->supported_extensions;
 263                 ctx->remote_info28.pid                  = info48->pid;
 264                 ctx->remote_info28.repl_epoch           = info48->repl_epoch;
 265                 break;
 266         }
 267         default:
 268                 DEBUG(1, ("Warning: invalid info length in bind info: %d\n",
 269                           bind_info.length));
 270                 break;
 271         }
 272
 273         return status;
 274 }
 275
 276 /****************************************************************
 277 ****************************************************************/
 278
 279 static NTSTATUS libnet_dssync_lookup_nc(TALLOC_CTX *mem_ctx,
 280                                         struct dssync_context *ctx)
 281 {
 282         NTSTATUS status;
 283         WERROR werr;
 284         int32_t level = 1;
 285         union drsuapi_DsNameRequest req;
 286         int32_t level_out;
 287         struct drsuapi_DsNameString names[1];
 288         union drsuapi_DsNameCtr ctr;
 289
 290         names[0].str = talloc_asprintf(mem_ctx, "%s\\", ctx->domain_name);
 291         NT_STATUS_HAVE_NO_MEMORY(names[0].str);
 292
 293         req.req1.codepage       = 1252; /* german */
 294         req.req1.language       = 0x00000407; /* german */
 295         req.req1.count          = 1;
 296         req.req1.names          = names;
 297         req.req1.format_flags   = DRSUAPI_DS_NAME_FLAG_NO_FLAGS;
 298         req.req1.format_offered = DRSUAPI_DS_NAME_FORMAT_UKNOWN;
 299         req.req1.format_desired = DRSUAPI_DS_NAME_FORMAT_FQDN_1779;
 300
 301         status = rpccli_drsuapi_DsCrackNames(ctx->cli, mem_ctx,
 302                                              &ctx->bind_handle,
 303                                              level,
 304                                              &req,
 305                                              &level_out,
 306                                              &ctr,
 307                                              &werr);
 308         if (!NT_STATUS_IS_OK(status)) {
 309                 ctx->error_message = talloc_asprintf(mem_ctx,
 310                         "Failed to lookup DN for domain name: %s",
 311                         get_friendly_werror_msg(werr));
 312                 return status;
 313         }
 314
 315         if (!W_ERROR_IS_OK(werr)) {
 316                 return werror_to_ntstatus(werr);
 317         }
 318
 319         if (ctr.ctr1->count != 1) {
 320                 return NT_STATUS_UNSUCCESSFUL;
 321         }
 322
 323         if (ctr.ctr1->array[0].status != DRSUAPI_DS_NAME_STATUS_OK) {
 324                 return NT_STATUS_UNSUCCESSFUL;
 325         }
 326
 327         ctx->nc_dn = talloc_strdup(mem_ctx, ctr.ctr1->array[0].result_name);
 328         NT_STATUS_HAVE_NO_MEMORY(ctx->nc_dn);
 329
 330         if (!ctx->dns_domain_name) {
 331                 ctx->dns_domain_name = talloc_strdup_upper(mem_ctx,
 332                         ctr.ctr1->array[0].dns_domain_name);
 333                 NT_STATUS_HAVE_NO_MEMORY(ctx->dns_domain_name);
 334         }
 335
 336         return NT_STATUS_OK;
 337 }
 338
 339 /****************************************************************
 340 ****************************************************************/
 341
 342 static NTSTATUS libnet_dssync_init(TALLOC_CTX *mem_ctx,
 343                                    struct dssync_context *ctx)
 344 {
 345         NTSTATUS status;
 346
 347         status = libnet_dssync_bind(mem_ctx, ctx);
 348         if (!NT_STATUS_IS_OK(status)) {
 349                 return status;
 350         }
 351
 352         if (!ctx->nc_dn) {
 353                 status = libnet_dssync_lookup_nc(mem_ctx, ctx);
 354         }
 355
 356         return status;
 357 }
 358
 359 /****************************************************************
 360 ****************************************************************/
 361
 362 static NTSTATUS libnet_dssync_build_request(TALLOC_CTX *mem_ctx,
 363                                             struct dssync_context *ctx,
 364                                             const char *dn,
 365                                             struct replUpToDateVectorBlob *utdv,
 366                                             int32_t level,
 367                                             union drsuapi_DsGetNCChangesRequest *preq)
 368 {
 369         NTSTATUS status;
 370         uint32_t count;
 371         union drsuapi_DsGetNCChangesRequest req;
 372         struct dom_sid null_sid;
 373         enum drsuapi_DsExtendedOperation extended_op;
 374         struct drsuapi_DsReplicaObjectIdentifier *nc = NULL;
 375         struct drsuapi_DsReplicaCursorCtrEx *cursors = NULL;
 376
 377         uint32_t replica_flags  = DRSUAPI_DS_REPLICA_NEIGHBOUR_WRITEABLE |
 378                                   DRSUAPI_DS_REPLICA_NEIGHBOUR_SYNC_ON_STARTUP |
 379                                   DRSUAPI_DS_REPLICA_NEIGHBOUR_DO_SCHEDULED_SYNCS |
 380                                   DRSUAPI_DS_REPLICA_NEIGHBOUR_RETURN_OBJECT_PARENTS |
 381                                   DRSUAPI_DS_REPLICA_NEIGHBOUR_NEVER_SYNCED;
 382
 383         ZERO_STRUCT(null_sid);
 384         ZERO_STRUCT(req);
 385
 386         nc = TALLOC_ZERO_P(mem_ctx, struct drsuapi_DsReplicaObjectIdentifier);
 387         if (!nc) {
 388                 status = NT_STATUS_NO_MEMORY;
 389                 goto fail;
 390         }
 391         nc->dn = dn;
 392         nc->guid = GUID_zero();
 393         nc->sid = null_sid;
 394
 395         if (!ctx->single && !ctx->repl_nodiff && utdv) {
 396                 cursors = TALLOC_ZERO_P(mem_ctx,
 397                                          struct drsuapi_DsReplicaCursorCtrEx);
 398                 if (!cursors) {
 399                         status = NT_STATUS_NO_MEMORY;
 400                         goto fail;
 401                 }
 402
 403                 switch (utdv->version) {
 404                 case 1:
 405                         cursors->count = utdv->ctr.ctr1.count;
 406                         cursors->cursors = utdv->ctr.ctr1.cursors;
 407                         break;
 408                 case 2:
 409                         cursors->count = utdv->ctr.ctr2.count;
 410                         cursors->cursors = talloc_array(cursors,
 411                                                 struct drsuapi_DsReplicaCursor,
 412                                                 cursors->count);
 413                         if (!cursors->cursors) {
 414                                 status = NT_STATUS_NO_MEMORY;
 415                                 goto fail;
 416                         }
 417                         for (count = 0; count < cursors->count; count++) {
 418                                 cursors->cursors[count].source_dsa_invocation_id =
 419                                         utdv->ctr.ctr2.cursors[count].source_dsa_invocation_id;
 420                                 cursors->cursors[count].highest_usn =
 421                                         utdv->ctr.ctr2.cursors[count].highest_usn;
 422                         }
 423                         break;
 424                 }
 425         }
 426
 427         if (ctx->single) {
 428                 extended_op = DRSUAPI_EXOP_REPL_OBJ;
 429         } else {
 430                 extended_op = DRSUAPI_EXOP_NONE;
 431         }
 432
 433         if (level == 8) {
 434                 req.req8.naming_context         = nc;
 435                 req.req8.replica_flags          = replica_flags;
 436                 req.req8.max_object_count       = 402;
 437                 req.req8.max_ndr_size           = 402116;
 438                 req.req8.uptodateness_vector    = cursors;
 439                 req.req8.extended_op            = extended_op;
 440         } else if (level == 5) {
 441                 req.req5.naming_context         = nc;
 442                 req.req5.replica_flags          = replica_flags;
 443                 req.req5.max_object_count       = 402;
 444                 req.req5.max_ndr_size           = 402116;
 445                 req.req5.uptodateness_vector    = cursors;
 446                 req.req5.extended_op            = extended_op;
 447         } else {
 448                 status = NT_STATUS_INVALID_PARAMETER;
 449                 goto fail;
 450         }
 451
 452         if (preq) {
 453                 *preq = req;
 454         }
 455
 456         return NT_STATUS_OK;
 457
 458 fail:
 459         TALLOC_FREE(nc);
 460         TALLOC_FREE(cursors);
 461         return status;
 462 }
 463
 464 static NTSTATUS libnet_dssync_getncchanges(TALLOC_CTX *mem_ctx,
 465                                            struct dssync_context *ctx,
 466                                            int32_t level,
 467                                            union drsuapi_DsGetNCChangesRequest *req,
 468                                            struct replUpToDateVectorBlob **pnew_utdv)
 469 {
 470         NTSTATUS status;
 471         WERROR werr;
 472         union drsuapi_DsGetNCChangesCtr ctr;
 473         struct drsuapi_DsGetNCChangesCtr1 *ctr1 = NULL;
 474         struct drsuapi_DsGetNCChangesCtr6 *ctr6 = NULL;
 475         struct replUpToDateVectorBlob *new_utdv = NULL;
 476         int32_t level_out = 0;
 477         int32_t out_level = 0;
 478         int y;
 479         bool last_query;
 480
 481         if (!ctx->single) {
 482                 new_utdv = TALLOC_ZERO_P(mem_ctx, struct replUpToDateVectorBlob);
 483                 if (!new_utdv) {
 484                         status = NT_STATUS_NO_MEMORY;
 485                         goto out;
 486                 }
 487         }
 488
 489         for (y=0, last_query = false; !last_query; y++) {
 490                 struct drsuapi_DsReplicaObjectListItemEx *first_object;
 491                 struct drsuapi_DsReplicaOIDMapping_Ctr *mapping_ctr;
 492
 493                 if (level == 8) {
 494                         DEBUG(1,("start[%d] tmp_higest_usn: %llu , highest_usn: %llu\n",y,
 495                                 (long long)req->req8.highwatermark.tmp_highest_usn,
 496                                 (long long)req->req8.highwatermark.highest_usn));
 497                 } else if (level == 5) {
 498                         DEBUG(1,("start[%d] tmp_higest_usn: %llu , highest_usn: %llu\n",y,
 499                                 (long long)req->req5.highwatermark.tmp_highest_usn,
 500                                 (long long)req->req5.highwatermark.highest_usn));
 501                 }
 502
 503                 status = rpccli_drsuapi_DsGetNCChanges(ctx->cli, mem_ctx,
 504                                                        &ctx->bind_handle,
 505                                                        level,
 506                                                        req,
 507                                                        &level_out,
 508                                                        &ctr,
 509                                                        &werr);
 510                 if (!NT_STATUS_IS_OK(status)) {
 511                         ctx->error_message = talloc_asprintf(mem_ctx,
 512                                 "Failed to get NC Changes: %s",
 513                                 get_friendly_werror_msg(werr));
 514                         goto out;
 515                 }
 516
 517                 if (!W_ERROR_IS_OK(werr)) {
 518                         status = werror_to_ntstatus(werr);
 519                         goto out;
 520                 }
 521
 522                 if (level_out == 1) {
 523                         out_level = 1;
 524                         ctr1 = &ctr.ctr1;
 525                 } else if (level_out == 2) {
 526                         out_level = 1;
 527                         ctr1 = ctr.ctr2.ctr.mszip1.ctr1;
 528                 } else if (level_out == 6) {
 529                         out_level = 6;
 530                         ctr6 = &ctr.ctr6;
 531                 } else if (level_out == 7
 532                            && ctr.ctr7.level == 6
 533                            && ctr.ctr7.type == DRSUAPI_COMPRESSION_TYPE_MSZIP) {
 534                         out_level = 6;
 535                         ctr6 = ctr.ctr7.ctr.mszip6.ctr6;
 536                 }
 537
 538                 if (out_level == 1) {
 539                         DEBUG(1,("end[%d] tmp_highest_usn: %llu , highest_usn: %llu\n",y,
 540                                 (long long)ctr1->new_highwatermark.tmp_highest_usn,
 541                                 (long long)ctr1->new_highwatermark.highest_usn));
 542
 543                         first_object = ctr1->first_object;
 544                         mapping_ctr = &ctr1->mapping_ctr;
 545
 546                         if (ctr1->more_data) {
 547                                 req->req5.highwatermark = ctr1->new_highwatermark;
 548                         } else {
 549                                 last_query = true;
 550                                 if (ctr1->uptodateness_vector && !ctx->single) {
 551                                         new_utdv->version = 1;
 552                                         new_utdv->ctr.ctr1.count =
 553                                                 ctr1->uptodateness_vector->count;
 554                                         new_utdv->ctr.ctr1.cursors =
 555                                                 ctr1->uptodateness_vector->cursors;
 556                                 }
 557                         }
 558                 } else if (out_level == 6) {
 559                         DEBUG(1,("end[%d] tmp_highest_usn: %llu , highest_usn: %llu\n",y,
 560                                 (long long)ctr6->new_highwatermark.tmp_highest_usn,
 561                                 (long long)ctr6->new_highwatermark.highest_usn));
 562
 563                         first_object = ctr6->first_object;
 564                         mapping_ctr = &ctr6->mapping_ctr;
 565
 566                         if (ctr6->more_data) {
 567                                 req->req8.highwatermark = ctr6->new_highwatermark;
 568                         } else {
 569                                 last_query = true;
 570                                 if (ctr6->uptodateness_vector && !ctx->single) {
 571                                         new_utdv->version = 2;
 572                                         new_utdv->ctr.ctr2.count =
 573                                                 ctr6->uptodateness_vector->count;
 574                                         new_utdv->ctr.ctr2.cursors =
 575                                                 ctr6->uptodateness_vector->cursors;
 576                                 }
 577                         }
 578                 }
 579
 580                 status = cli_get_session_key(mem_ctx, ctx->cli, &ctx->session_key);
 581                 if (!NT_STATUS_IS_OK(status)) {
 582                         ctx->error_message = talloc_asprintf(mem_ctx,
 583                                 "Failed to get Session Key: %s",
 584                                 nt_errstr(status));
 585                         goto out;
 586                 }
 587
 588                 libnet_dssync_decrypt_attributes(mem_ctx,
 589                                                  &ctx->session_key,
 590                                                  first_object);
 591
 592                 if (ctx->ops->process_objects) {
 593                         status = ctx->ops->process_objects(ctx, mem_ctx,
 594                                                            first_object,
 595                                                            mapping_ctr);
 596                         if (!NT_STATUS_IS_OK(status)) {
 597                                 ctx->error_message = talloc_asprintf(mem_ctx,
 598                                         "Failed to call processing function: %s",
 599                                         nt_errstr(status));
 600                                 goto out;
 601                         }
 602                 }
 603         }
 604
 605         *pnew_utdv = new_utdv;
 606
 607 out:
 608         return status;
 609 }
 610
 611 static NTSTATUS libnet_dssync_process(TALLOC_CTX *mem_ctx,
 612                                       struct dssync_context *ctx)
 613 {
 614         NTSTATUS status;
 615
 616         int32_t level;
 617         union drsuapi_DsGetNCChangesRequest req;
 618         struct replUpToDateVectorBlob *old_utdv = NULL;
 619         struct replUpToDateVectorBlob *pnew_utdv = NULL;
 620         const char *dn;
 621
 622         status = ctx->ops->startup(ctx, mem_ctx, &old_utdv);
 623         if (!NT_STATUS_IS_OK(status)) {
 624                 ctx->error_message = talloc_asprintf(mem_ctx,
 625                         "Failed to call startup operation: %s",
 626                         nt_errstr(status));
 627                 goto out;
 628         }
 629
 630         if (ctx->remote_info28.supported_extensions
 631             & DRSUAPI_SUPPORTED_EXTENSION_GETCHGREQ_V8)
 632         {
 633                 level = 8;
 634         } else {
 635                 level = 5;
 636         }
 637
 638         if (ctx->single && ctx->object_dn) {
 639                 dn = ctx->object_dn;
 640         } else {
 641                 dn = ctx->nc_dn;
 642         }
 643
 644         status = libnet_dssync_build_request(mem_ctx, ctx, dn, old_utdv, level,
 645                                              &req);
 646         if (!NT_STATUS_IS_OK(status)) {
 647                 goto out;
 648         }
 649
 650         status = libnet_dssync_getncchanges(mem_ctx, ctx, level, &req,
 651                                             &pnew_utdv);
 652         if (!NT_STATUS_IS_OK(status)) {
 653                 ctx->error_message = talloc_asprintf(mem_ctx,
 654                         "Failed to call DsGetNCCHanges: %s",
 655                         nt_errstr(status));
 656                 goto out;
 657         }
 658
 659         status = ctx->ops->finish(ctx, mem_ctx, pnew_utdv);
 660         if (!NT_STATUS_IS_OK(status)) {
 661                 ctx->error_message = talloc_asprintf(mem_ctx,
 662                         "Failed to call finishing operation: %s",
 663                         nt_errstr(status));
 664                 goto out;
 665         }
 666
 667  out:
 668         return status;
 669 }
 670
 671 /****************************************************************
 672 ****************************************************************/
 673
 674 NTSTATUS libnet_dssync(TALLOC_CTX *mem_ctx,
 675                        struct dssync_context *ctx)
 676 {
 677         NTSTATUS status;
 678
 679         status = libnet_dssync_init(mem_ctx, ctx);
 680         if (!NT_STATUS_IS_OK(status)) {
 681                 goto out;
 682         }
 683
 684         status = libnet_dssync_process(mem_ctx, ctx);
 685         if (!NT_STATUS_IS_OK(status)) {
 686                 goto out;
 687         }
 688
 689  out:
 690         return status;
 691 }
 692