DEBUG s4:torture/dssync: ...
[metze/samba/wip.git] / source4 / torture / drs / rpc / dssync.c
1 /*
2    Unix SMB/CIFS implementation.
3
4    DsGetNCChanges replication test
5
6    Copyright (C) Stefan (metze) Metzmacher 2005
7    Copyright (C) Brad Henry 2005
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 "lib/cmdline/popt_common.h"
25 #include "librpc/gen_ndr/ndr_drsuapi_c.h"
26 #include "librpc/gen_ndr/ndr_drsblobs.h"
27 #include "libcli/cldap/cldap.h"
28 #include "torture/torture.h"
29 #include "../libcli/drsuapi/drsuapi.h"
30 #include "auth/gensec/gensec.h"
31 #include "param/param.h"
32 #include "dsdb/samdb/samdb.h"
33 #include "torture/rpc/torture_rpc.h"
34 #include "torture/drs/proto.h"
35 #include "lib/tsocket/tsocket.h"
36 #include "libcli/resolve/resolve.h"
37
38 struct DsSyncBindInfo {
39         struct dcerpc_pipe *drs_pipe;
40         struct dcerpc_binding_handle *drs_handle;
41         struct drsuapi_DsBind req;
42         struct GUID bind_guid;
43         struct drsuapi_DsBindInfoCtr our_bind_info_ctr;
44         struct drsuapi_DsBindInfo28 our_bind_info28;
45         struct drsuapi_DsBindInfo28 peer_bind_info28;
46         struct policy_handle bind_handle;
47 };
48
49 struct DsSyncLDAPInfo {
50         struct ldb_context *ldb;
51 };
52
53 struct DsSyncTest {
54         struct dcerpc_binding *drsuapi_binding;
55
56         const char *ldap_url;
57         const char *dest_address;
58         const char *domain_dn;
59         const char *config_dn;
60         const char *schema_dn;
61
62         /* what we need to do as 'Administrator' */
63         struct {
64                 struct cli_credentials *credentials;
65                 struct DsSyncBindInfo drsuapi;
66                 struct DsSyncLDAPInfo ldap;
67         } admin;
68
69         /* what we need to do as the new dc machine account */
70         struct {
71                 struct cli_credentials *credentials;
72                 struct DsSyncBindInfo drsuapi;
73                 struct drsuapi_DsGetDCInfo2 dc_info2;
74                 struct GUID invocation_id;
75                 struct GUID object_guid;
76         } new_dc;
77
78         /* info about the old dc */
79         struct {
80                 struct drsuapi_DsGetDomainControllerInfo dc_info;
81         } old_dc;
82 };
83
84 static struct DsSyncTest *test_create_context(struct torture_context *tctx)
85 {
86         NTSTATUS status;
87         struct DsSyncTest *ctx;
88         struct drsuapi_DsBindInfo28 *our_bind_info28;
89         struct drsuapi_DsBindInfoCtr *our_bind_info_ctr;
90         const char *binding = torture_setting_string(tctx, "binding", NULL);
91         struct nbt_name name;
92
93         ctx = talloc_zero(tctx, struct DsSyncTest);
94         if (!ctx) return NULL;
95
96         status = dcerpc_parse_binding(ctx, binding, &ctx->drsuapi_binding);
97         if (!NT_STATUS_IS_OK(status)) {
98                 printf("Bad binding string %s\n", binding);
99                 return NULL;
100         }
101         ctx->drsuapi_binding->flags |= DCERPC_SIGN | DCERPC_SEAL;
102
103         ctx->ldap_url = talloc_asprintf(ctx, "ldap://%s", ctx->drsuapi_binding->host);
104
105         make_nbt_name_server(&name, ctx->drsuapi_binding->host);
106
107         /* do an initial name resolution to find its IP */
108         status = resolve_name_ex(lpcfg_resolve_context(tctx->lp_ctx),
109                                  0, 0, &name, tctx,
110                                  &ctx->dest_address, tctx->ev);
111         if (!NT_STATUS_IS_OK(status)) {
112                 printf("Failed to resolve %s - %s\n",
113                        name.name, nt_errstr(status));
114                 return NULL;
115         }
116
117         /* ctx->admin ...*/
118         ctx->admin.credentials                          = cmdline_credentials;
119
120         our_bind_info28                         = &ctx->admin.drsuapi.our_bind_info28;
121         our_bind_info28->supported_extensions   = 0xFFFFFFFF;
122         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_ADDENTRYREPLY_V3;
123         our_bind_info28->site_guid              = GUID_zero();
124         our_bind_info28->pid                    = 0;
125         our_bind_info28->repl_epoch             = 1;
126
127         our_bind_info_ctr                       = &ctx->admin.drsuapi.our_bind_info_ctr;
128         our_bind_info_ctr->length               = 28;
129         our_bind_info_ctr->info.info28          = *our_bind_info28;
130
131         GUID_from_string(DRSUAPI_DS_BIND_GUID, &ctx->admin.drsuapi.bind_guid);
132
133         ctx->admin.drsuapi.req.in.bind_guid             = &ctx->admin.drsuapi.bind_guid;
134         ctx->admin.drsuapi.req.in.bind_info             = our_bind_info_ctr;
135         ctx->admin.drsuapi.req.out.bind_handle          = &ctx->admin.drsuapi.bind_handle;
136
137         /* ctx->new_dc ...*/
138         ctx->new_dc.credentials                 = cmdline_credentials;
139
140         our_bind_info28                         = &ctx->new_dc.drsuapi.our_bind_info28;
141         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_BASE;
142         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_ASYNC_REPLICATION;
143         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_REMOVEAPI;
144         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_MOVEREQ_V2;
145         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_GETCHG_COMPRESS;
146         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_DCINFO_V1;
147         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_RESTORE_USN_OPTIMIZATION;
148         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_KCC_EXECUTE;
149         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_ADDENTRY_V2;
150         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_LINKED_VALUE_REPLICATION;
151         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_DCINFO_V2;
152         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_INSTANCE_TYPE_NOT_REQ_ON_MOD;
153         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_CRYPTO_BIND;
154         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_GET_REPL_INFO;
155         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_STRONG_ENCRYPTION;
156         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_DCINFO_V01;
157         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_TRANSITIVE_MEMBERSHIP;
158         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_ADD_SID_HISTORY;
159         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_POST_BETA3;
160         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_GET_MEMBERSHIPS2;
161         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREQ_V6;
162         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_NONDOMAIN_NCS;
163         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREQ_V8;
164         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREPLY_V5;
165         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREPLY_V6;
166         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_ADDENTRYREPLY_V3;
167         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREPLY_V7;
168         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_VERIFY_OBJECT;
169         if (lpcfg_parm_bool(tctx->lp_ctx, NULL, "dssync", "xpress", false)) {
170                 our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_XPRESS_COMPRESS;
171         }
172         our_bind_info28->site_guid              = GUID_zero();
173         our_bind_info28->pid                    = 0;
174         our_bind_info28->repl_epoch             = 0;
175
176         our_bind_info_ctr                       = &ctx->new_dc.drsuapi.our_bind_info_ctr;
177         our_bind_info_ctr->length               = 28;
178         our_bind_info_ctr->info.info28          = *our_bind_info28;
179
180         GUID_from_string(DRSUAPI_DS_BIND_GUID_W2K3, &ctx->new_dc.drsuapi.bind_guid);
181
182         ctx->new_dc.drsuapi.req.in.bind_guid            = &ctx->new_dc.drsuapi.bind_guid;
183         ctx->new_dc.drsuapi.req.in.bind_info            = our_bind_info_ctr;
184         ctx->new_dc.drsuapi.req.out.bind_handle         = &ctx->new_dc.drsuapi.bind_handle;
185
186         ctx->new_dc.invocation_id                       = ctx->new_dc.drsuapi.bind_guid;
187
188         /* ctx->old_dc ...*/
189
190         return ctx;
191 }
192
193 static bool _test_DsBind(struct torture_context *tctx,
194                          struct DsSyncTest *ctx, struct cli_credentials *credentials, struct DsSyncBindInfo *b)
195 {
196         NTSTATUS status;
197         bool ret = true;
198
199         status = dcerpc_pipe_connect_b(ctx,
200                                        &b->drs_pipe, ctx->drsuapi_binding,
201                                        &ndr_table_drsuapi,
202                                        credentials, tctx->ev, tctx->lp_ctx);
203
204         if (!NT_STATUS_IS_OK(status)) {
205                 printf("Failed to connect to server as a BDC: %s\n", nt_errstr(status));
206                 return false;
207         }
208         b->drs_handle = b->drs_pipe->binding_handle;
209
210         status = dcerpc_drsuapi_DsBind_r(b->drs_handle, ctx, &b->req);
211         if (!NT_STATUS_IS_OK(status)) {
212                 const char *errstr = nt_errstr(status);
213                 printf("dcerpc_drsuapi_DsBind failed - %s\n", errstr);
214                 ret = false;
215         } else if (!W_ERROR_IS_OK(b->req.out.result)) {
216                 printf("DsBind failed - %s\n", win_errstr(b->req.out.result));
217                 ret = false;
218         }
219
220         ZERO_STRUCT(b->peer_bind_info28);
221         if (b->req.out.bind_info) {
222                 switch (b->req.out.bind_info->length) {
223                 case 24: {
224                         struct drsuapi_DsBindInfo24 *info24;
225                         info24 = &b->req.out.bind_info->info.info24;
226                         b->peer_bind_info28.supported_extensions= info24->supported_extensions;
227                         b->peer_bind_info28.site_guid           = info24->site_guid;
228                         b->peer_bind_info28.pid                 = info24->pid;
229                         b->peer_bind_info28.repl_epoch          = 0;
230                         break;
231                 }
232                 case 48: {
233                         struct drsuapi_DsBindInfo48 *info48;
234                         info48 = &b->req.out.bind_info->info.info48;
235                         b->peer_bind_info28.supported_extensions= info48->supported_extensions;
236                         b->peer_bind_info28.site_guid           = info48->site_guid;
237                         b->peer_bind_info28.pid                 = info48->pid;
238                         b->peer_bind_info28.repl_epoch          = info48->repl_epoch;
239                         break;
240                 }
241                 case 28:
242                         b->peer_bind_info28 = b->req.out.bind_info->info.info28;
243                         break;
244                 default:
245                         printf("DsBind - warning: unknown BindInfo length: %u\n",
246                                b->req.out.bind_info->length);
247                 }
248         }
249
250         return ret;
251 }
252
253 static bool test_LDAPBind(struct torture_context *tctx, struct DsSyncTest *ctx,
254                           struct cli_credentials *credentials, struct DsSyncLDAPInfo *l)
255 {
256         bool ret = true;
257
258         struct ldb_context *ldb;
259
260         const char *modules_option[] = { "modules:paged_searches", NULL };
261         ctx->admin.ldap.ldb = ldb = ldb_init(ctx, tctx->ev);
262         if (ldb == NULL) {
263                 return false;
264         }
265
266         /* Despite us loading the schema from the AD server, we need
267          * the samba handlers to get the extended DN syntax stuff */
268         ret = ldb_register_samba_handlers(ldb);
269         if (ret != LDB_SUCCESS) {
270                 talloc_free(ldb);
271                 return NULL;
272         }
273
274         ldb_set_modules_dir(ldb, modules_path(ldb, "ldb"));
275
276         if (ldb_set_opaque(ldb, "credentials", credentials)) {
277                 talloc_free(ldb);
278                 return NULL;
279         }
280
281         if (ldb_set_opaque(ldb, "loadparm", tctx->lp_ctx)) {
282                 talloc_free(ldb);
283                 return NULL;
284         }
285
286         ret = ldb_connect(ldb, ctx->ldap_url, 0, modules_option);
287         if (ret != LDB_SUCCESS) {
288                 talloc_free(ldb);
289                 torture_assert_int_equal(tctx, ret, LDB_SUCCESS, "Failed to make LDB connection to target");
290         }
291
292         printf("connected to LDAP: %s\n", ctx->ldap_url);
293
294         return true;
295 }
296
297 static bool test_GetInfo(struct torture_context *tctx, struct DsSyncTest *ctx)
298 {
299         struct ldb_context *ldb = ctx->admin.ldap.ldb;
300
301         /* We must have LDB connection ready by this time */
302         SMB_ASSERT(ldb != NULL);
303
304         ctx->domain_dn = ldb_dn_get_linearized(ldb_get_default_basedn(ldb));
305         torture_assert(tctx, ctx->domain_dn != NULL, "Failed to get Domain DN");
306
307         ctx->config_dn = ldb_dn_get_linearized(ldb_get_config_basedn(ldb));
308         torture_assert(tctx, ctx->config_dn != NULL, "Failed to get Domain DN");
309
310         ctx->schema_dn = ldb_dn_get_linearized(ldb_get_schema_basedn(ldb));
311         torture_assert(tctx, ctx->schema_dn != NULL, "Failed to get Domain DN");
312
313         return true;
314 }
315
316 static bool test_analyse_objects(struct torture_context *tctx,
317                                  struct DsSyncTest *ctx,
318                                  const char *partition,
319                                  const struct drsuapi_DsReplicaOIDMapping_Ctr *mapping_ctr,
320                                  uint32_t object_count,
321                                  const struct drsuapi_DsReplicaObjectListItemEx *first_object,
322                                  uint32_t linked_attributes_count,
323                                  const struct drsuapi_DsReplicaLinkedAttribute *linked_attributes,
324                                  const DATA_BLOB *gensec_skey)
325 {
326         static uint32_t object_id;
327         const char *save_values_dir;
328         const struct drsuapi_DsReplicaObjectListItemEx *cur;
329         struct ldb_context *ldb = ctx->admin.ldap.ldb;
330         struct ldb_dn *deleted_dn;
331         WERROR status;
332         int i, j, ret;
333         struct dsdb_extended_replicated_objects *objs;
334         struct ldb_extended_dn_control *extended_dn_ctrl;
335         struct dsdb_schema *ldap_schema;
336
337         /* load dsdb_schema using remote prefixMap */
338         torture_assert(tctx,
339                        drs_util_dsdb_schema_load_ldb(tctx, ldb, mapping_ctr, false),
340                        "drs_util_dsdb_schema_load_ldb() failed");
341         ldap_schema = dsdb_get_schema(ldb, NULL);
342
343         status = dsdb_replicated_objects_convert(ldb,
344                                                  ldap_schema,
345                                                  partition,
346                                                  mapping_ctr,
347                                                  object_count,
348                                                  first_object,
349                                                  linked_attributes_count,
350                                                  linked_attributes,
351                                                  NULL, NULL,
352                                                  gensec_skey,
353                                                  0,
354                                                  ctx, &objs);
355         torture_assert_werr_ok(tctx, status, "dsdb_extended_replicated_objects_convert() failed!");
356
357         extended_dn_ctrl = talloc(objs, struct ldb_extended_dn_control);
358         extended_dn_ctrl->type = 1;
359
360         deleted_dn = ldb_dn_new(objs, ldb, partition);
361         ldb_dn_add_child_fmt(deleted_dn, "CN=Deleted Objects");
362
363         for (i=0; i < object_count; i++) {
364                 struct ldb_request *search_req;
365                 struct ldb_result *res;
366                 struct ldb_message *new_msg, *drs_msg, *ldap_msg;
367                 const char **attrs = talloc_array(objs, const char *, objs->objects[i].msg->num_elements+1+1);
368                 for (j=0; j < objs->objects[i].msg->num_elements; j++) {
369                         attrs[j] = objs->objects[i].msg->elements[j].name;
370                 }
371                 attrs[j] = "uSNChanged";
372                 j++;
373                 attrs[j] = NULL;
374                 res = talloc_zero(objs, struct ldb_result);
375                 if (!res) {
376                         return LDB_ERR_OPERATIONS_ERROR;
377                 }
378                 ret = ldb_build_search_req(&search_req, ldb, objs,
379                                            objs->objects[i].msg->dn,
380                                            LDB_SCOPE_BASE,
381                                            NULL,
382                                            attrs,
383                                            NULL,
384                                            res,
385                                            ldb_search_default_callback,
386                                            NULL);
387                 if (ret != LDB_SUCCESS) {
388                         return false;
389                 }
390                 talloc_steal(search_req, res);
391                 ret = ldb_request_add_control(search_req, LDB_CONTROL_SHOW_DELETED_OID, true, NULL);
392                 if (ret != LDB_SUCCESS) {
393                         return false;
394                 }
395
396                 ret = ldb_request_add_control(search_req, LDB_CONTROL_SHOW_RECYCLED_OID, false, NULL);
397                 if (ret != LDB_SUCCESS) {
398                         return false;
399                 }
400
401                 ret = ldb_request_add_control(search_req, LDB_CONTROL_EXTENDED_DN_OID, true, extended_dn_ctrl);
402                 if (ret != LDB_SUCCESS) {
403                         return false;
404                 }
405
406                 ret = ldb_request(ldb, search_req);
407                 if (ret == LDB_SUCCESS) {
408                         ret = ldb_wait(search_req->handle, LDB_WAIT_ALL);
409                 }
410
411                 torture_assert_int_equal(tctx, ret, LDB_SUCCESS,
412                                          talloc_asprintf(tctx,
413                                                          "Could not re-fetch object just delivered over DRS: %s",
414                                                          ldb_errstring(ldb)));
415                 torture_assert_int_equal(tctx, res->count, 1, "Could not re-fetch object just delivered over DRS");
416                 ldap_msg = res->msgs[0];
417
418 {
419 uint64_t usn = ldb_msg_find_attr_as_int64(ldap_msg, "uSNChanged", 0);
420 struct GUID g;
421 GUID_from_ndr_blob(&objs->objects[i].guid_value, &g);
422 torture_comment(tctx, "o[%d] usn_changed[%llu]: %s - %s\n", i, (unsigned long long)usn,
423         GUID_string(objs, &g),
424                 ldb_dn_get_linearized(objs->objects[i].msg->dn));
425 continue;
426 }
427                 for (j=0; j < ldap_msg->num_elements; j++) {
428                         ldap_msg->elements[j].flags = LDB_FLAG_MOD_ADD;
429                         /* For unknown reasons, there is no nTSecurityDescriptor on cn=deleted objects over LDAP, but there is over DRS!  Skip it on both transports for now here so */
430                         if ((ldb_attr_cmp(ldap_msg->elements[j].name, "nTSecurityDescriptor") == 0) &&
431                             (ldb_dn_compare(ldap_msg->dn, deleted_dn) == 0)) {
432                                 ldb_msg_remove_element(ldap_msg, &ldap_msg->elements[j]);
433                                 /* Don't skip one */
434                                 j--;
435                         }
436                 }
437
438                 ret = ldb_msg_normalize(ldb, search_req,
439                                         objs->objects[i].msg, &drs_msg);
440                 torture_assert(tctx, ret == LDB_SUCCESS,
441                                "ldb_msg_normalize() has failed");
442
443                 for (j=0; j < drs_msg->num_elements; j++) {
444                         if (drs_msg->elements[j].num_values == 0) {
445                                 ldb_msg_remove_element(drs_msg, &drs_msg->elements[j]);
446                                 /* Don't skip one */
447                                 j--;
448
449                                 /* For unknown reasons, there is no nTSecurityDescriptor on cn=deleted objects over LDAP, but there is over DRS! */
450                         } else if ((ldb_attr_cmp(drs_msg->elements[j].name, "nTSecurityDescriptor") == 0) &&
451                                    (ldb_dn_compare(drs_msg->dn, deleted_dn) == 0)) {
452                                 ldb_msg_remove_element(drs_msg, &drs_msg->elements[j]);
453                                 /* Don't skip one */
454                                 j--;
455                         } else if (ldb_attr_cmp(drs_msg->elements[j].name, "unicodePwd") == 0 ||
456                                    ldb_attr_cmp(drs_msg->elements[j].name, "dBCSPwd") == 0 ||
457                                    ldb_attr_cmp(drs_msg->elements[j].name, "ntPwdHistory") == 0 ||
458                                    ldb_attr_cmp(drs_msg->elements[j].name, "lmPwdHistory") == 0 ||
459                                    ldb_attr_cmp(drs_msg->elements[j].name, "supplementalCredentials") == 0 ||
460                                    ldb_attr_cmp(drs_msg->elements[j].name, "priorValue") == 0 ||
461                                    ldb_attr_cmp(drs_msg->elements[j].name, "currentValue") == 0 ||
462                                    ldb_attr_cmp(drs_msg->elements[j].name, "trustAuthOutgoing") == 0 ||
463                                    ldb_attr_cmp(drs_msg->elements[j].name, "trustAuthIncoming") == 0 ||
464                                    ldb_attr_cmp(drs_msg->elements[j].name, "initialAuthOutgoing") == 0 ||
465                                    ldb_attr_cmp(drs_msg->elements[j].name, "initialAuthIncoming") == 0) {
466
467                                 /* These are not shown over LDAP, so we need to skip them for the comparison */
468                                 ldb_msg_remove_element(drs_msg, &drs_msg->elements[j]);
469                                 /* Don't skip one */
470                                 j--;
471                         } else {
472                                 drs_msg->elements[j].flags = LDB_FLAG_MOD_ADD;
473                         }
474                 }
475
476
477                 ret = ldb_msg_difference(ldb, search_req,
478                                          drs_msg, ldap_msg, &new_msg);
479                 torture_assert(tctx, ret == LDB_SUCCESS, "ldb_msg_difference() has failed");
480                 if (new_msg->num_elements != 0) {
481                         char *s;
482                         bool is_warning = true;
483                         unsigned int idx;
484                         struct ldb_message_element *el;
485                         const struct dsdb_attribute * a;
486                         struct ldb_ldif ldif;
487                         ldif.changetype = LDB_CHANGETYPE_MODIFY;
488                         ldif.msg = new_msg;
489                         s = ldb_ldif_write_string(ldb, new_msg, &ldif);
490                         s = talloc_asprintf(tctx, "\n# Difference in between DRS and LDAP objects: \n%s", s);
491
492                         ret = ldb_msg_difference(ldb, search_req,
493                                                  ldap_msg, drs_msg, &ldif.msg);
494                         torture_assert(tctx, ret == LDB_SUCCESS, "ldb_msg_difference() has failed");
495                         s = talloc_asprintf_append(s,
496                                                    "\n# Difference in between LDAP and DRS objects: \n%s",
497                                                    ldb_ldif_write_string(ldb, new_msg, &ldif));
498
499                         s = talloc_asprintf_append(s,
500                                                    "# Should have no objects in 'difference' message. Diff elements: %d",
501                                                    new_msg->num_elements);
502
503                         /*
504                          * In case differences in messages are:
505                          * 1. Attributes with different values, i.e. 'replace'
506                          * 2. Those attributes are forward-link attributes
507                          * then we just warn about those differences.
508                          * It turns out windows doesn't send all of those values
509                          * in replicated_object but in linked_attributes.
510                          */
511                         for (idx = 0; idx < new_msg->num_elements && is_warning; idx++) {
512                                 el = &new_msg->elements[idx];
513                                 a = dsdb_attribute_by_lDAPDisplayName(ldap_schema,
514                                                                       el->name);
515                                 if (!(el->flags & (LDB_FLAG_MOD_ADD|LDB_FLAG_MOD_REPLACE))) {
516                                         /* DRS only value */
517                                         is_warning = false;
518                                 } else if (a->linkID & 1) {
519                                         is_warning = false;
520                                 }
521                         }
522                         if (is_warning) {
523                                 torture_warning(tctx, "%s", s);
524                         } else {
525                                 torture_fail(tctx, s);
526                         }
527                 }
528
529                 /* search_req is used as a tmp talloc context in the above */
530                 talloc_free(search_req);
531         }
532
533         for (i=0; i < objs->linked_attributes_count; i++) {
534 torture_comment(tctx, "l[%d] usn_changed[%llu]: attid[%u] %s\n", i,
535         (unsigned long long)objs->linked_attributes[i].meta_data.originating_usn,
536         objs->linked_attributes[i].attid,
537         GUID_string(objs, &objs->linked_attributes[i].identifier->guid));
538         }
539         if (!lpcfg_parm_bool(tctx->lp_ctx, NULL, "dssync", "print_pwd_blobs", false)) {
540                 talloc_free(objs);
541                 return true;
542         }
543
544         save_values_dir = lpcfg_parm_string(tctx->lp_ctx, NULL, "dssync", "save_pwd_blobs_dir");
545
546         for (cur = first_object; cur; cur = cur->next_object) {
547                 const char *dn;
548                 struct dom_sid *sid = NULL;
549                 uint32_t rid = 0;
550                 bool dn_printed = false;
551
552                 if (!cur->object.identifier) continue;
553
554                 dn = cur->object.identifier->dn;
555                 if (cur->object.identifier->sid.num_auths > 0) {
556                         sid = &cur->object.identifier->sid;
557                         rid = sid->sub_auths[sid->num_auths - 1];
558                 }
559
560                 for (i=0; i < cur->object.attribute_ctr.num_attributes; i++) {
561                         const char *name = NULL;
562                         DATA_BLOB plain_data;
563                         struct drsuapi_DsReplicaAttribute *attr;
564                         ndr_pull_flags_fn_t pull_fn = NULL;
565                         ndr_print_fn_t print_fn = NULL;
566                         void *ptr = NULL;
567                         attr = &cur->object.attribute_ctr.attributes[i];
568
569                         switch (attr->attid) {
570                         case DRSUAPI_ATTID_dBCSPwd:
571                                 name    = "dBCSPwd";
572                                 break;
573                         case DRSUAPI_ATTID_unicodePwd:
574                                 name    = "unicodePwd";
575                                 break;
576                         case DRSUAPI_ATTID_ntPwdHistory:
577                                 name    = "ntPwdHistory";
578                                 break;
579                         case DRSUAPI_ATTID_lmPwdHistory:
580                                 name    = "lmPwdHistory";
581                                 break;
582                         case DRSUAPI_ATTID_supplementalCredentials:
583                                 name    = "supplementalCredentials";
584                                 pull_fn = (ndr_pull_flags_fn_t)ndr_pull_supplementalCredentialsBlob;
585                                 print_fn = (ndr_print_fn_t)ndr_print_supplementalCredentialsBlob;
586                                 ptr = talloc(ctx, struct supplementalCredentialsBlob);
587                                 break;
588                         case DRSUAPI_ATTID_priorValue:
589                                 name    = "priorValue";
590                                 break;
591                         case DRSUAPI_ATTID_currentValue:
592                                 name    = "currentValue";
593                                 break;
594                         case DRSUAPI_ATTID_trustAuthOutgoing:
595                                 name    = "trustAuthOutgoing";
596                                 pull_fn = (ndr_pull_flags_fn_t)ndr_pull_trustAuthInOutBlob;
597                                 print_fn = (ndr_print_fn_t)ndr_print_trustAuthInOutBlob;
598                                 ptr = talloc(ctx, struct trustAuthInOutBlob);
599                                 break;
600                         case DRSUAPI_ATTID_trustAuthIncoming:
601                                 name    = "trustAuthIncoming";
602                                 pull_fn = (ndr_pull_flags_fn_t)ndr_pull_trustAuthInOutBlob;
603                                 print_fn = (ndr_print_fn_t)ndr_print_trustAuthInOutBlob;
604                                 ptr = talloc(ctx, struct trustAuthInOutBlob);
605                                 break;
606                         case DRSUAPI_ATTID_initialAuthOutgoing:
607                                 name    = "initialAuthOutgoing";
608                                 break;
609                         case DRSUAPI_ATTID_initialAuthIncoming:
610                                 name    = "initialAuthIncoming";
611                                 break;
612                         default:
613                                 continue;
614                         }
615
616                         if (attr->value_ctr.num_values != 1) continue;
617
618                         if (!attr->value_ctr.values[0].blob) continue;
619
620                         plain_data = *attr->value_ctr.values[0].blob;
621
622                         if (!dn_printed) {
623                                 object_id++;
624                                 DEBUG(0,("DN[%u] %s\n", object_id, dn));
625                                 dn_printed = true;
626                         }
627                         DEBUGADD(0,("ATTR: %s plain.length=%lu\n",
628                                     name, (long)plain_data.length));
629                         if (plain_data.length) {
630                                 enum ndr_err_code ndr_err;
631                                 dump_data(0, plain_data.data, plain_data.length);
632                                 if (save_values_dir) {
633                                         char *fname;
634                                         fname = talloc_asprintf(ctx, "%s/%s%02d",
635                                                                 save_values_dir,
636                                                                 name, object_id);
637                                         if (fname) {
638                                                 bool ok;
639                                                 ok = file_save(fname, plain_data.data, plain_data.length);
640                                                 if (!ok) {
641                                                         DEBUGADD(0,("Failed to save '%s'\n", fname));
642                                                 }
643                                         }
644                                         talloc_free(fname);
645                                 }
646
647                                 if (pull_fn) {
648                                         /* Can't use '_all' because of PIDL bugs with relative pointers */
649                                         ndr_err = ndr_pull_struct_blob(&plain_data, ptr,
650                                                                        ptr, pull_fn);
651                                         if (NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
652                                                 ndr_print_debug(print_fn, name, ptr);
653                                         } else {
654                                                 DEBUG(0, ("Failed to decode %s\n", name));
655                                         }
656                                 }
657                         }
658                         talloc_free(ptr);
659                 }
660         }
661         talloc_free(objs);
662         return true;
663 }
664
665 static bool test_GetNCChanges(struct torture_context *tctx,
666                               struct DsSyncTest *ctx,
667                               const char *nc_dn_str)
668 {
669         NTSTATUS status;
670         bool ret = true;
671         int i, y = 0;
672         uint64_t highest_usn = 0;
673         struct drsuapi_DsGetNCChanges r;
674         union drsuapi_DsGetNCChangesRequest req;
675         struct drsuapi_DsReplicaObjectIdentifier nc;
676         struct drsuapi_DsGetNCChangesCtr1 *ctr1 = NULL;
677         struct drsuapi_DsGetNCChangesCtr6 *ctr6 = NULL;
678         uint32_t out_level = 0;
679         struct GUID null_guid;
680         struct dom_sid null_sid;
681         DATA_BLOB gensec_skey;
682         struct {
683                 uint32_t level;
684         } array[] = {
685 /*              {
686                         5
687                 },
688 */              {
689                         8
690                 }
691         };
692
693         struct drsuapi_DsReplicaCursorCtrEx utdv;
694         struct drsuapi_DsReplicaCursor cursors[1];
695
696         ZERO_STRUCT(utdv);
697         utdv.version = 1;
698         utdv.count = ARRAY_SIZE(cursors);
699         utdv.cursors = cursors;
700         ZERO_STRUCT(cursors);
701         GUID_from_string("0d36ca05-5507-4e62-aca3-354bab0d39e1",
702                         &cursors[0].source_dsa_invocation_id);
703         cursors[0].highest_usn = 12755;
704 /*
705                     uptodateness_vector      : *
706                         uptodateness_vector: struct drsuapi_DsReplicaCursorCtrEx
707                             version                  : 0x00000001 (1)
708                             reserved1                : 0x00000000 (0)
709                             count                    : 0x00000001 (1)
710                             reserved2                : 0x00000000 (0)
711                             cursors: ARRAY(1)
712                                 cursors: struct drsuapi_DsReplicaCursor
713                                     source_dsa_invocation_id : 0d36ca05-5507-4e62-aca3-354bab0d39e1
714                                     highest_usn              : 0x00000000000031d3 (12755)
715 */
716         ZERO_STRUCT(null_guid);
717         ZERO_STRUCT(null_sid);
718
719         highest_usn = lpcfg_parm_int(tctx->lp_ctx, NULL, "dssync", "highest_usn", 0);
720
721         array[0].level = lpcfg_parm_int(tctx->lp_ctx, NULL, "dssync", "get_nc_changes_level", array[0].level);
722
723         if (lpcfg_parm_bool(tctx->lp_ctx, NULL, "dssync", "print_pwd_blobs", false)) {
724                 const struct samr_Password *nthash;
725                 nthash = cli_credentials_get_nt_hash(ctx->new_dc.credentials, ctx);
726                 if (nthash) {
727                         dump_data_pw("CREDENTIALS nthash:", nthash->hash, sizeof(nthash->hash));
728                 }
729         }
730         status = gensec_session_key(ctx->new_dc.drsuapi.drs_pipe->conn->security_state.generic_state,
731                                     ctx, &gensec_skey);
732         if (!NT_STATUS_IS_OK(status)) {
733                 printf("failed to get gensec session key: %s\n", nt_errstr(status));
734                 return false;
735         }
736
737         for (i=0; i < ARRAY_SIZE(array); i++) {
738                 printf("Testing DsGetNCChanges level %d\n",
739                         array[i].level);
740
741                 r.in.bind_handle        = &ctx->new_dc.drsuapi.bind_handle;
742                 r.in.level              = array[i].level;
743
744                 switch (r.in.level) {
745                 case 5:
746                         nc.guid = null_guid;
747                         nc.sid  = null_sid;
748                         nc.dn   = nc_dn_str;
749
750                         r.in.req                                        = &req;
751                         r.in.req->req5.destination_dsa_guid             = ctx->new_dc.invocation_id;
752                         r.in.req->req5.source_dsa_invocation_id         = null_guid;
753                         r.in.req->req5.naming_context                   = &nc;
754                         r.in.req->req5.highwatermark.tmp_highest_usn    = highest_usn;
755                         r.in.req->req5.highwatermark.reserved_usn       = 0;
756                         r.in.req->req5.highwatermark.highest_usn        = highest_usn;
757                         r.in.req->req5.uptodateness_vector              = NULL;
758                         r.in.req->req5.replica_flags                    = 0;
759                         if (lpcfg_parm_bool(tctx->lp_ctx, NULL, "dssync", "compression", false)) {
760                                 r.in.req->req5.replica_flags            |= DRSUAPI_DRS_USE_COMPRESSION;
761                         }
762                         if (lpcfg_parm_bool(tctx->lp_ctx, NULL, "dssync", "neighbour_writeable", true)) {
763                                 r.in.req->req5.replica_flags            |= DRSUAPI_DRS_WRIT_REP;
764                         }
765                         r.in.req->req5.replica_flags                    |= DRSUAPI_DRS_INIT_SYNC
766                                                                         | DRSUAPI_DRS_PER_SYNC
767                                                                         | DRSUAPI_DRS_GET_ANC
768                                                                         | DRSUAPI_DRS_NEVER_SYNCED
769                                                                         ;
770                         r.in.req->req5.max_object_count                 = 133;
771                         r.in.req->req5.max_ndr_size                     = 1336770;
772                         r.in.req->req5.extended_op                      = DRSUAPI_EXOP_NONE;
773                         r.in.req->req5.fsmo_info                        = 0;
774
775                         break;
776                 case 8:
777                         nc.guid = null_guid;
778                         nc.sid  = null_sid;
779                         nc.dn   = nc_dn_str;
780                         /* nc.dn can be set to any other ad partition */
781
782                         r.in.req                                        = &req;
783                         r.in.req->req8.destination_dsa_guid             = ctx->new_dc.invocation_id;
784                         r.in.req->req8.source_dsa_invocation_id         = null_guid;
785                         r.in.req->req8.naming_context                   = &nc;
786                         r.in.req->req8.highwatermark.tmp_highest_usn    = highest_usn;
787                         r.in.req->req8.highwatermark.reserved_usn       = 0;
788                         r.in.req->req8.highwatermark.highest_usn        = highest_usn;
789                         r.in.req->req8.uptodateness_vector              = NULL;//&utdv;
790                         r.in.req->req8.replica_flags                    = 0;
791                         if (lpcfg_parm_bool(tctx->lp_ctx, NULL, "dssync", "compression", false)) {
792                                 r.in.req->req8.replica_flags            |= DRSUAPI_DRS_USE_COMPRESSION;
793                         }
794                         if (lpcfg_parm_bool(tctx->lp_ctx, NULL, "dssync", "neighbour_writeable", true)) {
795                                 r.in.req->req8.replica_flags            |= DRSUAPI_DRS_WRIT_REP;
796                         }
797                         r.in.req->req8.replica_flags                    |= DRSUAPI_DRS_INIT_SYNC
798                                                                         | DRSUAPI_DRS_PER_SYNC
799                                                                         | DRSUAPI_DRS_GET_ANC
800                                                                         | DRSUAPI_DRS_NEVER_SYNCED
801                                                                         ;
802                         r.in.req->req8.replica_flags = 0x00201074;
803                         //r.in.req->req8.replica_flags |= DRSUAPI_DRS_GET_ANC;
804                         r.in.req->req8.max_object_count                 = 402;
805                         r.in.req->req8.max_ndr_size                     = 402116;
806
807                         r.in.req->req8.extended_op                      = DRSUAPI_EXOP_NONE;
808                         r.in.req->req8.fsmo_info                        = 0;
809                         r.in.req->req8.partial_attribute_set            = NULL;
810                         r.in.req->req8.partial_attribute_set_ex         = NULL;
811                         r.in.req->req8.mapping_ctr.num_mappings         = 0;
812                         r.in.req->req8.mapping_ctr.mappings             = NULL;
813
814                         break;
815                 }
816
817                 for (y=0; ;y++) {
818                         uint32_t _level = 0;
819                         union drsuapi_DsGetNCChangesCtr ctr;
820
821                         ZERO_STRUCT(r.out);
822
823                         r.out.level_out = &_level;
824                         r.out.ctr       = &ctr;
825
826                         if (r.in.level == 5) {
827                                 torture_comment(tctx,
828                                                 "start[%d] tmp_higest_usn: %llu , highest_usn: %llu\n",
829                                                 y,
830                                                 (unsigned long long) r.in.req->req5.highwatermark.tmp_highest_usn,
831                                                 (unsigned long long) r.in.req->req5.highwatermark.highest_usn);
832                         }
833
834                         if (r.in.level == 8) {
835                                 torture_comment(tctx,
836                                                 "start[%d] tmp_higest_usn: %llu , highest_usn: %llu\n",
837                                                 y,
838                                                 (unsigned long long) r.in.req->req8.highwatermark.tmp_highest_usn,
839                                                 (unsigned long long) r.in.req->req8.highwatermark.highest_usn);
840                         }
841
842                         status = dcerpc_drsuapi_DsGetNCChanges_r(ctx->new_dc.drsuapi.drs_handle, ctx, &r);
843                         torture_drsuapi_assert_call(tctx, ctx->new_dc.drsuapi.drs_pipe, status,
844                                                     &r, "dcerpc_drsuapi_DsGetNCChanges");
845
846                         if (ret == true && *r.out.level_out == 1) {
847                                 out_level = 1;
848                                 ctr1 = &r.out.ctr->ctr1;
849                         } else if (ret == true && *r.out.level_out == 2 &&
850                                    r.out.ctr->ctr2.mszip1.ts) {
851                                 out_level = 1;
852                                 ctr1 = &r.out.ctr->ctr2.mszip1.ts->ctr1;
853                         }
854
855                         if (out_level == 1) {
856                                 torture_comment(tctx,
857                                                 "end[%d] tmp_highest_usn: %llu , highest_usn: %llu\n",
858                                                 y,
859                                                 (unsigned long long) ctr1->new_highwatermark.tmp_highest_usn,
860                                                 (unsigned long long) ctr1->new_highwatermark.highest_usn);
861
862                                 if (!test_analyse_objects(tctx, ctx, nc_dn_str, &ctr1->mapping_ctr,  ctr1->object_count,
863                                                           ctr1->first_object, 0, NULL, &gensec_skey)) {
864                                         return false;
865                                 }
866
867                                 if (ctr1->more_data) {
868                                         r.in.req->req5.highwatermark = ctr1->new_highwatermark;
869                                         continue;
870                                 }
871                         }
872
873                         if (ret == true && *r.out.level_out == 6) {
874                                 out_level = 6;
875                                 ctr6 = &r.out.ctr->ctr6;
876                         } else if (ret == true && *r.out.level_out == 7
877                                    && r.out.ctr->ctr7.level == 6
878                                    && r.out.ctr->ctr7.type == DRSUAPI_COMPRESSION_TYPE_MSZIP
879                                    && r.out.ctr->ctr7.ctr.mszip6.ts) {
880                                 out_level = 6;
881                                 ctr6 = &r.out.ctr->ctr7.ctr.mszip6.ts->ctr6;
882                         } else if (ret == true && *r.out.level_out == 7
883                                    && r.out.ctr->ctr7.level == 6
884                                    && r.out.ctr->ctr7.type == DRSUAPI_COMPRESSION_TYPE_XPRESS
885                                    && r.out.ctr->ctr7.ctr.xpress6.ts) {
886                                 out_level = 6;
887                                 ctr6 = &r.out.ctr->ctr7.ctr.xpress6.ts->ctr6;
888                         }
889
890                         if (out_level == 6) {
891                                 torture_comment(tctx,
892                                                 "end[%d] tmp_highest_usn: %llu , highest_usn: %llu\n",
893                                                 y,
894                                                 (unsigned long long) ctr6->new_highwatermark.tmp_highest_usn,
895                                                 (unsigned long long) ctr6->new_highwatermark.highest_usn);
896
897                                 if (!test_analyse_objects(tctx, ctx, nc_dn_str, &ctr6->mapping_ctr,  ctr6->object_count,
898                                                           ctr6->first_object,
899                                                           ctr6->linked_attributes_count,
900                                                           ctr6->linked_attributes,
901                                                           &gensec_skey)) {
902                                         return false;
903                                 }
904
905                                 if (ctr6->more_data) {
906                                         r.in.req->req8.highwatermark = ctr6->new_highwatermark;
907                                         continue;
908                                 }
909                         }
910
911                         break;
912                 }
913         }
914
915         return ret;
916 }
917
918 /**
919  * Test DsGetNCChanges() DRSUAPI call against one
920  * or more Naming Contexts.
921  * Specific NC to test with may be supplied
922  * in lp_ctx configuration. If no NC is specified,
923  * it will test DsGetNCChanges() on all NCs on remote DC
924  */
925 static bool test_FetchData(struct torture_context *tctx, struct DsSyncTest *ctx)
926 {
927         bool ret = true;
928         size_t i, count;
929         const char *nc_dn_str;
930         const char **nc_list;
931
932         nc_list = const_str_list(str_list_make_empty(ctx));
933         torture_assert(tctx, nc_list, "Not enough memory!");
934
935         /* make a list of partitions to test with */
936         nc_dn_str = lpcfg_parm_string(tctx->lp_ctx, NULL, "dssync", "partition");
937         if (nc_dn_str == NULL) {
938                 nc_list = str_list_add_const(nc_list, ctx->domain_dn);
939                 nc_list = str_list_add_const(nc_list, ctx->config_dn);
940                 nc_list = str_list_add_const(nc_list, ctx->schema_dn);
941         } else {
942                 nc_list = str_list_add_const(nc_list, nc_dn_str);
943         }
944
945         count = str_list_length(nc_list);
946         for (i = 0; i < count && ret; i++) {
947                 torture_comment(tctx, "\nNaming Context: %s\n", nc_list[i]);
948                 ret = test_GetNCChanges(tctx, ctx, nc_list[i]);
949         }
950
951         talloc_free(nc_list);
952         return ret;
953 }
954
955
956 static bool test_FetchNT4Data(struct torture_context *tctx,
957                               struct DsSyncTest *ctx)
958 {
959         NTSTATUS status;
960         struct drsuapi_DsGetNT4ChangeLog r;
961         union drsuapi_DsGetNT4ChangeLogRequest req;
962         union drsuapi_DsGetNT4ChangeLogInfo info;
963         uint32_t level_out = 0;
964         struct GUID null_guid;
965         struct dom_sid null_sid;
966         DATA_BLOB cookie;
967
968         ZERO_STRUCT(null_guid);
969         ZERO_STRUCT(null_sid);
970         ZERO_STRUCT(cookie);
971
972         ZERO_STRUCT(r);
973         r.in.bind_handle        = &ctx->new_dc.drsuapi.bind_handle;
974         r.in.level              = 1;
975         r.out.info              = &info;
976         r.out.level_out         = &level_out;
977
978         req.req1.flags = lpcfg_parm_int(tctx->lp_ctx, NULL,
979                                      "dssync", "nt4changelog_flags",
980                                      DRSUAPI_NT4_CHANGELOG_GET_CHANGELOG |
981                                      DRSUAPI_NT4_CHANGELOG_GET_SERIAL_NUMBERS);
982         req.req1.preferred_maximum_length = lpcfg_parm_int(tctx->lp_ctx, NULL,
983                                         "dssync", "nt4changelog_preferred_len",
984                                         0x00004000);
985
986         while (1) {
987                 req.req1.restart_length = cookie.length;
988                 req.req1.restart_data = cookie.data;
989
990                 r.in.req = &req;
991
992                 status = dcerpc_drsuapi_DsGetNT4ChangeLog_r(ctx->new_dc.drsuapi.drs_handle, ctx, &r);
993                 if (NT_STATUS_EQUAL(status, NT_STATUS_NOT_IMPLEMENTED)) {
994                         torture_skip(tctx,
995                                      "DsGetNT4ChangeLog not supported: NT_STATUS_NOT_IMPLEMENTED");
996                 } else if (!NT_STATUS_IS_OK(status)) {
997                         const char *errstr = nt_errstr(status);
998                         if (NT_STATUS_EQUAL(status, NT_STATUS_RPC_PROCNUM_OUT_OF_RANGE)) {
999                                 torture_skip(tctx,
1000                                              "DsGetNT4ChangeLog not supported: NT_STATUS_RPC_PROCNUM_OUT_OF_RANGE");
1001                         }
1002                         torture_fail(tctx,
1003                                      talloc_asprintf(tctx, "dcerpc_drsuapi_DsGetNT4ChangeLog failed - %s\n",
1004                                                      errstr));
1005                 } else if (W_ERROR_EQUAL(r.out.result, WERR_INVALID_DOMAIN_ROLE)) {
1006                         torture_skip(tctx,
1007                                      "DsGetNT4ChangeLog not supported: WERR_INVALID_DOMAIN_ROLE");
1008                 } else if (!W_ERROR_IS_OK(r.out.result)) {
1009                         torture_fail(tctx,
1010                                      talloc_asprintf(tctx, "DsGetNT4ChangeLog failed - %s\n",
1011                                                      win_errstr(r.out.result)));
1012                 } else if (*r.out.level_out != 1) {
1013                         torture_fail(tctx,
1014                                      talloc_asprintf(tctx, "DsGetNT4ChangeLog unknown level - %u\n",
1015                                                      *r.out.level_out));
1016                 } else if (NT_STATUS_IS_OK(r.out.info->info1.status)) {
1017                 } else if (NT_STATUS_EQUAL(r.out.info->info1.status, STATUS_MORE_ENTRIES)) {
1018                         cookie.length   = r.out.info->info1.restart_length;
1019                         cookie.data     = r.out.info->info1.restart_data;
1020                         continue;
1021                 } else {
1022                         torture_fail(tctx,
1023                                      talloc_asprintf(tctx, "DsGetNT4ChangeLog failed - %s\n",
1024                                                      nt_errstr(r.out.info->info1.status)));
1025                 }
1026
1027                 break;
1028         }
1029
1030         return true;
1031 }
1032
1033 /**
1034  * DSSYNC test case setup
1035  */
1036 static bool torture_dssync_tcase_setup(struct torture_context *tctx, void **data)
1037 {
1038         bool bret;
1039         struct DsSyncTest *ctx;
1040
1041         *data = ctx = test_create_context(tctx);
1042         torture_assert(tctx, ctx, "test_create_context() failed");
1043
1044         bret = _test_DsBind(tctx, ctx, ctx->admin.credentials, &ctx->admin.drsuapi);
1045         torture_assert(tctx, bret, "_test_DsBind() failed");
1046
1047         bret = test_LDAPBind(tctx, ctx, ctx->admin.credentials, &ctx->admin.ldap);
1048         torture_assert(tctx, bret, "test_LDAPBind() failed");
1049
1050         bret = test_GetInfo(tctx, ctx);
1051         torture_assert(tctx, bret, "test_GetInfo() failed");
1052
1053         bret = _test_DsBind(tctx, ctx, ctx->new_dc.credentials, &ctx->new_dc.drsuapi);
1054         torture_assert(tctx, bret, "_test_DsBind() failed");
1055
1056         return true;
1057 }
1058
1059 /**
1060  * DSSYNC test case cleanup
1061  */
1062 static bool torture_dssync_tcase_teardown(struct torture_context *tctx, void *data)
1063 {
1064         struct DsSyncTest *ctx;
1065         struct drsuapi_DsUnbind r;
1066         struct policy_handle bind_handle;
1067
1068         ctx = talloc_get_type(data, struct DsSyncTest);
1069
1070         ZERO_STRUCT(r);
1071         r.out.bind_handle = &bind_handle;
1072
1073         /* Unbing admin handle */
1074         r.in.bind_handle = &ctx->admin.drsuapi.bind_handle;
1075         dcerpc_drsuapi_DsUnbind_r(ctx->admin.drsuapi.drs_handle, ctx, &r);
1076
1077         /* Unbing new_dc handle */
1078         r.in.bind_handle = &ctx->new_dc.drsuapi.bind_handle;
1079         dcerpc_drsuapi_DsUnbind_r(ctx->new_dc.drsuapi.drs_handle, ctx, &r);
1080
1081         talloc_free(ctx);
1082
1083         return true;
1084 }
1085
1086 /**
1087  * DSSYNC test case implementation
1088  */
1089 void torture_drs_rpc_dssync_tcase(struct torture_suite *suite)
1090 {
1091         typedef bool (*run_func) (struct torture_context *test, void *tcase_data);
1092
1093         struct torture_test *test;
1094         struct torture_tcase *tcase = torture_suite_add_tcase(suite, "dssync");
1095
1096         torture_tcase_set_fixture(tcase,
1097                                   torture_dssync_tcase_setup,
1098                                   torture_dssync_tcase_teardown);
1099
1100         test = torture_tcase_add_simple_test(tcase, "DC_FetchData", (run_func)test_FetchData);
1101         test = torture_tcase_add_simple_test(tcase, "FetchNT4Data", (run_func)test_FetchNT4Data);
1102 }
1103