Merge commit 'release-4-0-0alpha1' into v4-0-test
[samba.git] / source4 / torture / 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 "libcli/cldap/cldap.h"
27 #include "libcli/ldap/ldap_client.h"
28 #include "torture/torture.h"
29 #include "torture/ldap/proto.h"
30 #include "libcli/auth/libcli_auth.h"
31 #include "lib/crypto/crypto.h"
32 #include "auth/credentials/credentials.h"
33 #include "libcli/auth/libcli_auth.h"
34 #include "auth/gensec/gensec.h"
35 #include "param/param.h"
36
37 struct DsSyncBindInfo {
38         struct dcerpc_pipe *pipe;
39         struct drsuapi_DsBind req;
40         struct GUID bind_guid;
41         struct drsuapi_DsBindInfoCtr our_bind_info_ctr;
42         struct drsuapi_DsBindInfo28 our_bind_info28;
43         struct drsuapi_DsBindInfo28 peer_bind_info28;
44         struct policy_handle bind_handle;
45 };
46
47 struct DsSyncLDAPInfo {
48         struct ldap_connection *conn;
49 };
50
51 struct DsSyncTest {
52         struct dcerpc_binding *drsuapi_binding;
53         
54         const char *ldap_url;
55         const char *site_name;
56         
57         const char *domain_dn;
58
59         /* what we need to do as 'Administrator' */
60         struct {
61                 struct cli_credentials *credentials;
62                 struct DsSyncBindInfo drsuapi;
63                 struct DsSyncLDAPInfo ldap;
64         } admin;
65
66         /* what we need to do as the new dc machine account */
67         struct {
68                 struct cli_credentials *credentials;
69                 struct DsSyncBindInfo drsuapi;
70                 struct drsuapi_DsGetDCInfo2 dc_info2;
71                 struct GUID invocation_id;
72                 struct GUID object_guid;
73         } new_dc;
74
75         /* info about the old dc */
76         struct {
77                 struct drsuapi_DsGetDomainControllerInfo dc_info;
78         } old_dc;
79 };
80
81 static struct DsSyncTest *test_create_context(TALLOC_CTX *mem_ctx)
82 {
83         NTSTATUS status;
84         struct DsSyncTest *ctx;
85         struct drsuapi_DsBindInfo28 *our_bind_info28;
86         struct drsuapi_DsBindInfoCtr *our_bind_info_ctr;
87         const char *binding = lp_parm_string(global_loadparm, NULL, "torture", "binding");
88         ctx = talloc_zero(mem_ctx, struct DsSyncTest);
89         if (!ctx) return NULL;
90
91         status = dcerpc_parse_binding(ctx, binding, &ctx->drsuapi_binding);
92         if (!NT_STATUS_IS_OK(status)) {
93                 printf("Bad binding string %s\n", binding);
94                 return NULL;
95         }
96         ctx->drsuapi_binding->flags |= DCERPC_SIGN | DCERPC_SEAL;
97
98         ctx->ldap_url = talloc_asprintf(ctx, "ldap://%s/", ctx->drsuapi_binding->host);
99
100         /* ctx->admin ...*/
101         ctx->admin.credentials                          = cmdline_credentials;
102
103         our_bind_info28                         = &ctx->admin.drsuapi.our_bind_info28;
104         our_bind_info28->supported_extensions   = 0xFFFFFFFF;
105         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_ADDENTRYREPLY_V3;
106         our_bind_info28->site_guid              = GUID_zero();
107         our_bind_info28->u1                     = 0;
108         our_bind_info28->repl_epoch             = 1;
109
110         our_bind_info_ctr                       = &ctx->admin.drsuapi.our_bind_info_ctr;
111         our_bind_info_ctr->length               = 28;
112         our_bind_info_ctr->info.info28          = *our_bind_info28;
113
114         GUID_from_string(DRSUAPI_DS_BIND_GUID, &ctx->admin.drsuapi.bind_guid);
115
116         ctx->admin.drsuapi.req.in.bind_guid             = &ctx->admin.drsuapi.bind_guid;
117         ctx->admin.drsuapi.req.in.bind_info             = our_bind_info_ctr;
118         ctx->admin.drsuapi.req.out.bind_handle          = &ctx->admin.drsuapi.bind_handle;
119
120         /* ctx->new_dc ...*/
121         ctx->new_dc.credentials                 = cmdline_credentials;
122
123         our_bind_info28                         = &ctx->new_dc.drsuapi.our_bind_info28;
124         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_BASE;
125         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_ASYNC_REPLICATION;
126         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_REMOVEAPI;
127         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_MOVEREQ_V2;
128         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_GETCHG_COMPRESS;
129         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_DCINFO_V1;
130         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_RESTORE_USN_OPTIMIZATION;
131         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_KCC_EXECUTE;
132         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_ADDENTRY_V2;
133         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_LINKED_VALUE_REPLICATION;
134         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_DCINFO_V2;
135         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_INSTANCE_TYPE_NOT_REQ_ON_MOD;
136         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_CRYPTO_BIND;
137         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_GET_REPL_INFO;
138         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_STRONG_ENCRYPTION;
139         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_DCINFO_V01;
140         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_TRANSITIVE_MEMBERSHIP;
141         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_ADD_SID_HISTORY;
142         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_POST_BETA3;
143         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_GET_MEMBERSHIPS2;
144         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREQ_V6;
145         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_NONDOMAIN_NCS;
146         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREQ_V8;
147         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREPLY_V5;
148         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREPLY_V6;
149         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_ADDENTRYREPLY_V3;
150         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_GETCHGREPLY_V7;
151         our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_VERIFY_OBJECT;
152         if (lp_parm_bool(global_loadparm, NULL, "dssync", "xpress", false)) {
153                 our_bind_info28->supported_extensions   |= DRSUAPI_SUPPORTED_EXTENSION_XPRESS_COMPRESS;
154         }
155         our_bind_info28->site_guid              = GUID_zero();
156         our_bind_info28->u1                     = 508;
157         our_bind_info28->repl_epoch             = 0;
158
159         our_bind_info_ctr                       = &ctx->new_dc.drsuapi.our_bind_info_ctr;
160         our_bind_info_ctr->length               = 28;
161         our_bind_info_ctr->info.info28          = *our_bind_info28;
162
163         GUID_from_string(DRSUAPI_DS_BIND_GUID_W2K3, &ctx->new_dc.drsuapi.bind_guid);
164
165         ctx->new_dc.drsuapi.req.in.bind_guid            = &ctx->new_dc.drsuapi.bind_guid;
166         ctx->new_dc.drsuapi.req.in.bind_info            = our_bind_info_ctr;
167         ctx->new_dc.drsuapi.req.out.bind_handle         = &ctx->new_dc.drsuapi.bind_handle;
168
169         ctx->new_dc.invocation_id                       = ctx->new_dc.drsuapi.bind_guid;
170
171         /* ctx->old_dc ...*/
172
173         return ctx;
174 }
175
176 static bool _test_DsBind(struct DsSyncTest *ctx, struct cli_credentials *credentials, struct DsSyncBindInfo *b)
177 {
178         NTSTATUS status;
179         bool ret = true;
180         struct event_context *event = NULL;
181
182         status = dcerpc_pipe_connect_b(ctx,
183                                        &b->pipe, ctx->drsuapi_binding, 
184                                        &ndr_table_drsuapi,
185                                        credentials, event);
186         
187         if (!NT_STATUS_IS_OK(status)) {
188                 printf("Failed to connect to server as a BDC: %s\n", nt_errstr(status));
189                 return false;
190         }
191
192         status = dcerpc_drsuapi_DsBind(b->pipe, ctx, &b->req);
193         if (!NT_STATUS_IS_OK(status)) {
194                 const char *errstr = nt_errstr(status);
195                 if (NT_STATUS_EQUAL(status, NT_STATUS_NET_WRITE_FAULT)) {
196                         errstr = dcerpc_errstr(ctx, b->pipe->last_fault_code);
197                 }
198                 printf("dcerpc_drsuapi_DsBind failed - %s\n", errstr);
199                 ret = false;
200         } else if (!W_ERROR_IS_OK(b->req.out.result)) {
201                 printf("DsBind failed - %s\n", win_errstr(b->req.out.result));
202                 ret = false;
203         }
204
205         ZERO_STRUCT(b->peer_bind_info28);
206         if (b->req.out.bind_info) {
207                 switch (b->req.out.bind_info->length) {
208                 case 24: {
209                         struct drsuapi_DsBindInfo24 *info24;
210                         info24 = &b->req.out.bind_info->info.info24;
211                         b->peer_bind_info28.supported_extensions= info24->supported_extensions;
212                         b->peer_bind_info28.site_guid           = info24->site_guid;
213                         b->peer_bind_info28.u1                  = info24->u1;
214                         b->peer_bind_info28.repl_epoch          = 0;
215                         break;
216                 }
217                 case 28:
218                         b->peer_bind_info28 = b->req.out.bind_info->info.info28;
219                         break;
220                 }
221         }
222
223         return ret;
224 }
225
226 static bool test_LDAPBind(struct DsSyncTest *ctx, struct cli_credentials *credentials, struct DsSyncLDAPInfo *l)
227 {
228         NTSTATUS status;
229         bool ret = true;
230
231         status = torture_ldap_connection(ctx, &l->conn, ctx->ldap_url);
232         if (!NT_STATUS_IS_OK(status)) {
233                 printf("failed to connect to LDAP: %s\n", ctx->ldap_url);
234                 return false;
235         }
236
237         printf("connected to LDAP: %s\n", ctx->ldap_url);
238
239         status = torture_ldap_bind_sasl(l->conn, credentials);
240         if (!NT_STATUS_IS_OK(status)) {
241                 printf("failed to bind to LDAP:\n");
242                 return false;
243         }
244         printf("bound to LDAP.\n");
245
246         return ret;
247 }
248
249 static bool test_GetInfo(struct DsSyncTest *ctx)
250 {
251         NTSTATUS status;
252         struct drsuapi_DsCrackNames r;
253         struct drsuapi_DsNameString names[1];
254         bool ret = true;
255
256         struct cldap_socket *cldap = cldap_socket_init(ctx, NULL);
257         struct cldap_netlogon search;
258         
259         r.in.bind_handle                = &ctx->admin.drsuapi.bind_handle;
260         r.in.level                      = 1;
261         r.in.req.req1.codepage          = 1252; /* western european */
262         r.in.req.req1.language          = 0x00000407; /* german */
263         r.in.req.req1.count             = 1;
264         r.in.req.req1.names             = names;
265         r.in.req.req1.format_flags      = DRSUAPI_DS_NAME_FLAG_NO_FLAGS;                
266         r.in.req.req1.format_offered    = DRSUAPI_DS_NAME_FORMAT_NT4_ACCOUNT;
267         r.in.req.req1.format_desired    = DRSUAPI_DS_NAME_FORMAT_FQDN_1779;
268         names[0].str = talloc_asprintf(ctx, "%s\\", lp_workgroup(global_loadparm));
269
270         status = dcerpc_drsuapi_DsCrackNames(ctx->admin.drsuapi.pipe, ctx, &r);
271         if (!NT_STATUS_IS_OK(status)) {
272                 const char *errstr = nt_errstr(status);
273                 if (NT_STATUS_EQUAL(status, NT_STATUS_NET_WRITE_FAULT)) {
274                         errstr = dcerpc_errstr(ctx, ctx->admin.drsuapi.pipe->last_fault_code);
275                 }
276                 printf("dcerpc_drsuapi_DsCrackNames failed - %s\n", errstr);
277                 return false;
278         } else if (!W_ERROR_IS_OK(r.out.result)) {
279                 printf("DsCrackNames failed - %s\n", win_errstr(r.out.result));
280                 return false;
281         }
282
283         ctx->domain_dn = r.out.ctr.ctr1->array[0].result_name;
284         
285         ZERO_STRUCT(search);
286         search.in.dest_address = ctx->drsuapi_binding->host;
287         search.in.acct_control = -1;
288         search.in.version = 6;
289         status = cldap_netlogon(cldap, ctx, &search);
290         if (!NT_STATUS_IS_OK(status)) {
291                 const char *errstr = nt_errstr(status);
292                 ctx->site_name = talloc_asprintf(ctx, "%s", "Default-First-Site-Name");
293                 printf("cldap_netlogon() returned %s. Defaulting to Site-Name: %s\n", errstr, ctx->site_name);          
294         } else {
295                 ctx->site_name = talloc_steal(ctx, search.out.netlogon.logon5.client_site);
296                 printf("cldap_netlogon() returned Client Site-Name: %s.\n",ctx->site_name);
297                 printf("cldap_netlogon() returned Server Site-Name: %s.\n",search.out.netlogon.logon5.server_site);
298         }
299
300         return ret;
301 }
302
303 static DATA_BLOB decrypt_blob(TALLOC_CTX *mem_ctx,
304                               const DATA_BLOB *gensec_skey,
305                               bool rcrypt,
306                               struct drsuapi_DsReplicaObjectIdentifier *id,
307                               uint32_t rid,
308                               const DATA_BLOB *buffer)
309 {
310         DATA_BLOB confounder;
311         DATA_BLOB enc_buffer;
312
313         struct MD5Context md5;
314         uint8_t _enc_key[16];
315         DATA_BLOB enc_key;
316
317         DATA_BLOB dec_buffer;
318
319         uint32_t crc32_given;
320         uint32_t crc32_calc;
321         DATA_BLOB checked_buffer;
322
323         DATA_BLOB plain_buffer;
324
325         /*
326          * the combination "c[3] s[1] e[1] d[0]..."
327          * was successful!!!!!!!!!!!!!!!!!!!!!!!!!!
328          */
329
330         /* 
331          * the first 16 bytes at the beginning are the confounder
332          * followed by the 4 byte crc32 checksum
333          */
334         if (buffer->length < 20) {
335                 return data_blob_const(NULL, 0);
336         }
337         confounder = data_blob_const(buffer->data, 16);
338         enc_buffer = data_blob_const(buffer->data + 16, buffer->length - 16);
339
340         /* 
341          * build the encryption key md5 over the session key followed
342          * by the confounder
343          * 
344          * here the gensec session key is used and
345          * not the dcerpc ncacn_ip_tcp "SystemLibraryDTC" key!
346          */
347         enc_key = data_blob_const(_enc_key, sizeof(_enc_key));
348         MD5Init(&md5);
349         MD5Update(&md5, gensec_skey->data, gensec_skey->length);
350         MD5Update(&md5, confounder.data, confounder.length);
351         MD5Final(enc_key.data, &md5);
352
353         /*
354          * copy the encrypted buffer part and 
355          * decrypt it using the created encryption key using arcfour
356          */
357         dec_buffer = data_blob_talloc(mem_ctx, enc_buffer.data, enc_buffer.length);
358         if (!dec_buffer.data) {
359                 return data_blob_const(NULL, 0);
360         }
361         arcfour_crypt_blob(dec_buffer.data, dec_buffer.length, &enc_key);
362
363         /* 
364          * the first 4 byte are the crc32 checksum
365          * of the remaining bytes
366          */
367         crc32_given = IVAL(dec_buffer.data, 0);
368         crc32_calc = crc32_calc_buffer(dec_buffer.data + 4 , dec_buffer.length - 4);
369         if (crc32_given != crc32_calc) {
370                 DEBUG(0,("CRC32: given[0x%08X] calc[0x%08X]\n",
371                       crc32_given, crc32_calc));
372                 return data_blob_const(NULL, 0);
373         }
374         checked_buffer = data_blob_talloc(mem_ctx, dec_buffer.data + 4, dec_buffer.length - 4);
375         if (!checked_buffer.data) {
376                 return data_blob_const(NULL, 0);
377         }
378
379         /*
380          * some attributes seem to be in a usable form after this decryption
381          * (supplementalCredentials, priorValue, currentValue, trustAuthOutgoing,
382          *  trustAuthIncoming, initialAuthOutgoing, initialAuthIncoming)
383          * At least supplementalCredentials contains plaintext
384          * like "Primary:Kerberos" (in unicode form)
385          *
386          * some attributes seem to have some additional encryption
387          * dBCSPwd, unicodePwd, ntPwdHistory, lmPwdHistory
388          *
389          * it's the sam_rid_crypt() function, as the value is constant,
390          * so it doesn't depend on sessionkeys.
391          */
392         if (rcrypt) {
393                 uint32_t i, num_hashes;
394
395                 if ((checked_buffer.length % 16) != 0) {
396                         return data_blob_const(NULL, 0);
397                 }
398
399                 plain_buffer = data_blob_talloc(mem_ctx, checked_buffer.data, checked_buffer.length);
400                 if (!plain_buffer.data) {
401                         return data_blob_const(NULL, 0);
402                 }
403                         
404                 num_hashes = plain_buffer.length / 16;
405                 for (i = 0; i < num_hashes; i++) {
406                         uint32_t offset = i * 16;
407                         sam_rid_crypt(rid, checked_buffer.data + offset, plain_buffer.data + offset, 0);
408                 }
409         } else {
410                 plain_buffer = checked_buffer;
411         }
412
413         return plain_buffer;
414 }
415
416 static void test_analyse_objects(struct DsSyncTest *ctx,
417                                  const DATA_BLOB *gensec_skey,
418                                  struct drsuapi_DsReplicaObjectListItemEx *cur)
419 {
420         static uint32_t object_id;
421         const char *save_values_dir;
422
423         if (!lp_parm_bool(global_loadparm, NULL,"dssync","print_pwd_blobs", false)) {
424                 return; 
425         }
426
427         save_values_dir = lp_parm_string(global_loadparm, NULL, "dssync", "save_pwd_blobs_dir");
428
429         for (; cur; cur = cur->next_object) {
430                 const char *dn;
431                 struct dom_sid *sid = NULL;
432                 uint32_t rid = 0;
433                 bool dn_printed = false;
434                 uint32_t i;
435
436                 if (!cur->object.identifier) continue;
437
438                 dn = cur->object.identifier->dn;
439                 if (cur->object.identifier->sid.num_auths > 0) {
440                         sid = &cur->object.identifier->sid;
441                         rid = sid->sub_auths[sid->num_auths - 1];
442                 }
443
444                 for (i=0; i < cur->object.attribute_ctr.num_attributes; i++) {
445                         const char *name = NULL;
446                         bool rcrypt = false;
447                         DATA_BLOB *enc_data = NULL;
448                         DATA_BLOB plain_data;
449                         struct drsuapi_DsReplicaAttribute *attr;
450                         attr = &cur->object.attribute_ctr.attributes[i];
451
452                         switch (attr->attid) {
453                         case DRSUAPI_ATTRIBUTE_dBCSPwd:
454                                 name    = "dBCSPwd";
455                                 rcrypt  = true;
456                                 break;
457                         case DRSUAPI_ATTRIBUTE_unicodePwd:
458                                 name    = "unicodePwd";
459                                 rcrypt  = true;
460                                 break;
461                         case DRSUAPI_ATTRIBUTE_ntPwdHistory:
462                                 name    = "ntPwdHistory";
463                                 rcrypt  = true;
464                                 break;
465                         case DRSUAPI_ATTRIBUTE_lmPwdHistory:
466                                 name    = "lmPwdHistory";
467                                 rcrypt  = true;
468                                 break;
469                         case DRSUAPI_ATTRIBUTE_supplementalCredentials:
470                                 name    = "supplementalCredentials";
471                                 break;
472                         case DRSUAPI_ATTRIBUTE_priorValue:
473                                 name    = "priorValue";
474                                 break;
475                         case DRSUAPI_ATTRIBUTE_currentValue:
476                                 name    = "currentValue";
477                                 break;
478                         case DRSUAPI_ATTRIBUTE_trustAuthOutgoing:
479                                 name    = "trustAuthOutgoing";
480                                 break;
481                         case DRSUAPI_ATTRIBUTE_trustAuthIncoming:
482                                 name    = "trustAuthIncoming";
483                                 break;
484                         case DRSUAPI_ATTRIBUTE_initialAuthOutgoing:
485                                 name    = "initialAuthOutgoing";
486                                 break;
487                         case DRSUAPI_ATTRIBUTE_initialAuthIncoming:
488                                 name    = "initialAuthIncoming";
489                                 break;
490                         default:
491                                 continue;
492                         }
493
494                         if (attr->value_ctr.num_values != 1) continue;
495
496                         if (!attr->value_ctr.values[0].blob) continue;
497
498                         enc_data = attr->value_ctr.values[0].blob;
499                         ZERO_STRUCT(plain_data);
500
501                         plain_data = decrypt_blob(ctx, gensec_skey, rcrypt,
502                                                   cur->object.identifier, rid,
503                                                   enc_data);
504                         if (!dn_printed) {
505                                 object_id++;
506                                 DEBUG(0,("DN[%u] %s\n", object_id, dn));
507                                 dn_printed = true;
508                         }
509                         DEBUGADD(0,("ATTR: %s enc.length=%lu plain.length=%lu\n",
510                                     name, (long)enc_data->length, (long)plain_data.length));
511                         if (plain_data.length) {
512                                 dump_data(0, plain_data.data, plain_data.length);
513                                 if (save_values_dir) {
514                                         char *fname;
515                                         fname = talloc_asprintf(ctx, "%s/%s%02d",
516                                                                 save_values_dir,
517                                                                 name, object_id);
518                                         if (fname) {
519                                                 bool ok;
520                                                 ok = file_save(fname, plain_data.data, plain_data.length);
521                                                 if (!ok) {
522                                                         DEBUGADD(0,("Failed to save '%s'\n", fname));
523                                                 }
524                                         }
525                                         talloc_free(fname);
526                                 }
527                         } else {
528                                 dump_data(0, enc_data->data, enc_data->length);
529                         }
530                 }
531         }
532 }
533
534 static bool test_FetchData(struct DsSyncTest *ctx)
535 {
536         NTSTATUS status;
537         bool ret = true;
538         int i, y = 0;
539         uint64_t highest_usn = 0;
540         const char *partition = NULL;
541         struct drsuapi_DsGetNCChanges r;
542         struct drsuapi_DsReplicaObjectIdentifier nc;
543         struct drsuapi_DsGetNCChangesCtr1 *ctr1 = NULL;
544         struct drsuapi_DsGetNCChangesCtr6 *ctr6 = NULL;
545         int32_t out_level = 0;
546         struct GUID null_guid;
547         struct dom_sid null_sid;
548         DATA_BLOB gensec_skey;
549         struct {
550                 int32_t level;
551         } array[] = {
552 /*              {
553                         5
554                 },
555 */              {
556                         8
557                 }
558         };
559
560         ZERO_STRUCT(null_guid);
561         ZERO_STRUCT(null_sid);
562
563         partition = lp_parm_string(global_loadparm, NULL, "dssync", "partition");
564         if (partition == NULL) {
565                 partition = ctx->domain_dn;
566                 printf("dssync:partition not specified, defaulting to %s.\n", ctx->domain_dn);
567         }
568
569         highest_usn = lp_parm_int(global_loadparm, NULL, "dssync", "highest_usn", 0);
570
571         array[0].level = lp_parm_int(global_loadparm, NULL, "dssync", "get_nc_changes_level", array[0].level);
572
573         if (lp_parm_bool(global_loadparm, NULL, "dssync", "print_pwd_blobs", false)) {
574                 const struct samr_Password *nthash;
575                 nthash = cli_credentials_get_nt_hash(ctx->new_dc.credentials, ctx);
576                 if (nthash) {
577                         DEBUG(0,("CREDENTIALS nthash:\n"));
578                         dump_data(0, nthash->hash, sizeof(nthash->hash));
579                 }
580         }
581         status = gensec_session_key(ctx->new_dc.drsuapi.pipe->conn->security_state.generic_state,
582                                     &gensec_skey);
583         if (!NT_STATUS_IS_OK(status)) {
584                 printf("failed to get gensec session key: %s\n", nt_errstr(status));
585                 return false;
586         }
587
588         for (i=0; i < ARRAY_SIZE(array); i++) {
589                 printf("testing DsGetNCChanges level %d\n",
590                         array[i].level);
591
592                 r.in.bind_handle        = &ctx->new_dc.drsuapi.bind_handle;
593                 r.in.level              = &array[i].level;
594
595                 switch (*r.in.level) {
596                 case 5:
597                         nc.guid = null_guid;
598                         nc.sid  = null_sid;
599                         nc.dn   = partition; 
600
601                         r.in.req.req5.destination_dsa_guid              = ctx->new_dc.invocation_id;
602                         r.in.req.req5.source_dsa_invocation_id          = null_guid;
603                         r.in.req.req5.naming_context                    = &nc;
604                         r.in.req.req5.highwatermark.tmp_highest_usn     = highest_usn;
605                         r.in.req.req5.highwatermark.reserved_usn        = 0;
606                         r.in.req.req5.highwatermark.highest_usn         = highest_usn;
607                         r.in.req.req5.uptodateness_vector               = NULL;
608                         r.in.req.req5.replica_flags                     = 0;
609                         if (lp_parm_bool(global_loadparm, NULL, "dssync", "compression", false)) {
610                                 r.in.req.req5.replica_flags             |= DRSUAPI_DS_REPLICA_NEIGHBOUR_COMPRESS_CHANGES;
611                         }
612                         if (lp_parm_bool(global_loadparm, NULL, "dssync", "neighbour_writeable", true)) {
613                                 r.in.req.req5.replica_flags             |= DRSUAPI_DS_REPLICA_NEIGHBOUR_WRITEABLE;
614                         }
615                         r.in.req.req5.replica_flags                     |= DRSUAPI_DS_REPLICA_NEIGHBOUR_SYNC_ON_STARTUP
616                                                                         | DRSUAPI_DS_REPLICA_NEIGHBOUR_DO_SCHEDULED_SYNCS
617                                                                         | DRSUAPI_DS_REPLICA_NEIGHBOUR_RETURN_OBJECT_PARENTS
618                                                                         | DRSUAPI_DS_REPLICA_NEIGHBOUR_NEVER_SYNCED
619                                                                         ;
620                         r.in.req.req5.max_object_count                  = 133;
621                         r.in.req.req5.max_ndr_size                      = 1336770;
622                         r.in.req.req5.unknown4                          = 0;
623                         r.in.req.req5.h1                                = 0;
624
625                         break;
626                 case 8:
627                         nc.guid = null_guid;
628                         nc.sid  = null_sid;
629                         nc.dn   = partition; 
630                         /* nc.dn can be set to any other ad partition */
631                         
632                         r.in.req.req8.destination_dsa_guid              = ctx->new_dc.invocation_id;
633                         r.in.req.req8.source_dsa_invocation_id          = null_guid;
634                         r.in.req.req8.naming_context                    = &nc;
635                         r.in.req.req8.highwatermark.tmp_highest_usn     = highest_usn;
636                         r.in.req.req8.highwatermark.reserved_usn        = 0;
637                         r.in.req.req8.highwatermark.highest_usn         = highest_usn;
638                         r.in.req.req8.uptodateness_vector               = NULL;
639                         r.in.req.req8.replica_flags                     = 0;
640                         if (lp_parm_bool(global_loadparm, NULL, "dssync", "compression", false)) {
641                                 r.in.req.req8.replica_flags             |= DRSUAPI_DS_REPLICA_NEIGHBOUR_COMPRESS_CHANGES;
642                         }
643                         if (lp_parm_bool(global_loadparm, NULL, "dssync", "neighbour_writeable", true)) {
644                                 r.in.req.req8.replica_flags             |= DRSUAPI_DS_REPLICA_NEIGHBOUR_WRITEABLE;
645                         }
646                         r.in.req.req8.replica_flags                     |= DRSUAPI_DS_REPLICA_NEIGHBOUR_SYNC_ON_STARTUP
647                                                                         | DRSUAPI_DS_REPLICA_NEIGHBOUR_DO_SCHEDULED_SYNCS
648                                                                         | DRSUAPI_DS_REPLICA_NEIGHBOUR_RETURN_OBJECT_PARENTS
649                                                                         | DRSUAPI_DS_REPLICA_NEIGHBOUR_NEVER_SYNCED
650                                                                         ;
651                         r.in.req.req8.max_object_count                  = 402;
652                         r.in.req.req8.max_ndr_size                      = 402116;
653
654                         r.in.req.req8.unknown4                          = 0;
655                         r.in.req.req8.h1                                = 0;
656                         r.in.req.req8.unique_ptr1                       = 0;
657                         r.in.req.req8.unique_ptr2                       = 0;
658                         r.in.req.req8.mapping_ctr.num_mappings          = 0;
659                         r.in.req.req8.mapping_ctr.mappings              = NULL;
660
661                         break;
662                 }
663                 
664                 printf("Dumping AD partition: %s\n", nc.dn);
665                 for (y=0; ;y++) {
666                         int32_t _level = 0;
667                         ZERO_STRUCT(r.out);
668                         r.out.level = &_level;
669
670                         if (*r.in.level == 5) {
671                                 DEBUG(0,("start[%d] tmp_higest_usn: %llu , highest_usn: %llu\n",y,
672                                         (long long)r.in.req.req5.highwatermark.tmp_highest_usn,
673                                         (long long)r.in.req.req5.highwatermark.highest_usn));
674                         }
675
676                         if (*r.in.level == 8) {
677                                 DEBUG(0,("start[%d] tmp_higest_usn: %llu , highest_usn: %llu\n",y,
678                                         (long long)r.in.req.req8.highwatermark.tmp_highest_usn,
679                                         (long long)r.in.req.req8.highwatermark.highest_usn));
680                         }
681
682                         status = dcerpc_drsuapi_DsGetNCChanges(ctx->new_dc.drsuapi.pipe, ctx, &r);
683                         if (!NT_STATUS_IS_OK(status)) {
684                                 const char *errstr = nt_errstr(status);
685                                 if (NT_STATUS_EQUAL(status, NT_STATUS_NET_WRITE_FAULT)) {
686                                         errstr = dcerpc_errstr(ctx, ctx->new_dc.drsuapi.pipe->last_fault_code);
687                                 }
688                                 printf("dcerpc_drsuapi_DsGetNCChanges failed - %s\n", errstr);
689                                 ret = false;
690                         } else if (!W_ERROR_IS_OK(r.out.result)) {
691                                 printf("DsGetNCChanges failed - %s\n", win_errstr(r.out.result));
692                                 ret = false;
693                         }
694
695                         if (ret == true && *r.out.level == 1) {
696                                 out_level = 1;
697                                 ctr1 = &r.out.ctr.ctr1;
698                         } else if (ret == true && *r.out.level == 2) {
699                                 out_level = 1;
700                                 ctr1 = r.out.ctr.ctr2.ctr.mszip1.ctr1;
701                         }
702
703                         if (out_level == 1) {
704                                 DEBUG(0,("end[%d] tmp_highest_usn: %llu , highest_usn: %llu\n",y,
705                                         (long long)ctr1->new_highwatermark.tmp_highest_usn,
706                                         (long long)ctr1->new_highwatermark.highest_usn));
707
708                                 test_analyse_objects(ctx, &gensec_skey, ctr1->first_object);
709
710                                 if (ctr1->new_highwatermark.tmp_highest_usn > ctr1->new_highwatermark.highest_usn) {
711                                         r.in.req.req5.highwatermark = ctr1->new_highwatermark;
712                                         continue;
713                                 }
714                         }
715
716                         if (ret == true && *r.out.level == 6) {
717                                 out_level = 6;
718                                 ctr6 = &r.out.ctr.ctr6;
719                         } else if (ret == true && *r.out.level == 7
720                                    && r.out.ctr.ctr7.level == 6
721                                    && r.out.ctr.ctr7.type == DRSUAPI_COMPRESSION_TYPE_MSZIP) {
722                                 out_level = 6;
723                                 ctr6 = r.out.ctr.ctr7.ctr.mszip6.ctr6;
724                         }
725
726                         if (out_level == 6) {
727                                 DEBUG(0,("end[%d] tmp_highest_usn: %llu , highest_usn: %llu\n",y,
728                                         (long long)ctr6->new_highwatermark.tmp_highest_usn,
729                                         (long long)ctr6->new_highwatermark.highest_usn));
730
731                                 test_analyse_objects(ctx, &gensec_skey, ctr6->first_object);
732
733                                 if (ctr6->new_highwatermark.tmp_highest_usn > ctr6->new_highwatermark.highest_usn) {
734                                         r.in.req.req8.highwatermark = ctr6->new_highwatermark;
735                                         continue;
736                                 }
737                         }
738
739                         break;
740                 }
741         }
742
743         return ret;
744 }
745
746 static bool test_FetchNT4Data(struct DsSyncTest *ctx)
747 {
748         NTSTATUS status;
749         bool ret = true;
750         struct drsuapi_DsGetNT4ChangeLog r;
751         struct GUID null_guid;
752         struct dom_sid null_sid;
753         DATA_BLOB cookie;
754
755         ZERO_STRUCT(null_guid);
756         ZERO_STRUCT(null_sid);
757         ZERO_STRUCT(cookie);
758
759         ZERO_STRUCT(r);
760         r.in.bind_handle        = &ctx->new_dc.drsuapi.bind_handle;
761         r.in.level              = 1;
762
763         r.in.req.req1.unknown1  = lp_parm_int(global_loadparm, NULL, "dssync", "nt4-1", 3);
764         r.in.req.req1.unknown2  = lp_parm_int(global_loadparm, NULL, "dssync", "nt4-2", 0x00004000);
765
766         while (1) {
767                 r.in.req.req1.length    = cookie.length;
768                 r.in.req.req1.data      = cookie.data;
769
770                 status = dcerpc_drsuapi_DsGetNT4ChangeLog(ctx->new_dc.drsuapi.pipe, ctx, &r);
771                 if (!NT_STATUS_IS_OK(status)) {
772                         const char *errstr = nt_errstr(status);
773                         if (NT_STATUS_EQUAL(status, NT_STATUS_NET_WRITE_FAULT)) {
774                                 errstr = dcerpc_errstr(ctx, ctx->new_dc.drsuapi.pipe->last_fault_code);
775                         }
776                         printf("dcerpc_drsuapi_DsGetNT4ChangeLog failed - %s\n", errstr);
777                         ret = false;
778                 } else if (W_ERROR_EQUAL(r.out.result, WERR_INVALID_DOMAIN_ROLE)) {
779                         printf("DsGetNT4ChangeLog not supported by target server\n");
780                         break;
781                 } else if (!W_ERROR_IS_OK(r.out.result)) {
782                         printf("DsGetNT4ChangeLog failed - %s\n", win_errstr(r.out.result));
783                         ret = false;
784                 } else if (r.out.level != 1) {
785                         printf("DsGetNT4ChangeLog unknown level - %u\n", r.out.level);
786                         ret = false;
787                 } else if (NT_STATUS_IS_OK(r.out.info.info1.status)) {
788                 } else if (NT_STATUS_EQUAL(r.out.info.info1.status, STATUS_MORE_ENTRIES)) {
789                         cookie.length   = r.out.info.info1.length1;
790                         cookie.data     = r.out.info.info1.data1;
791                         continue;
792                 } else {
793                         printf("DsGetNT4ChangeLog failed - %s\n", nt_errstr(r.out.info.info1.status));
794                         ret = false;
795                 }
796
797                 break;
798         }
799
800         return ret;
801 }
802
803 bool torture_rpc_dssync(struct torture_context *torture)
804 {
805         bool ret = true;
806         TALLOC_CTX *mem_ctx;
807         struct DsSyncTest *ctx;
808         
809         mem_ctx = talloc_init("torture_rpc_dssync");
810         ctx = test_create_context(mem_ctx);
811         
812         ret &= _test_DsBind(ctx, ctx->admin.credentials, &ctx->admin.drsuapi);
813         if (!ret) {
814                 return ret;
815         }
816         ret &= test_LDAPBind(ctx, ctx->admin.credentials, &ctx->admin.ldap);
817         if (!ret) {
818                 return ret;
819         }
820         ret &= test_GetInfo(ctx);
821         ret &= _test_DsBind(ctx, ctx->new_dc.credentials, &ctx->new_dc.drsuapi);
822         if (!ret) {
823                 return ret;
824         }
825         ret &= test_FetchData(ctx);
826         ret &= test_FetchNT4Data(ctx);
827
828         return ret;
829 }