ctdb-scripts: further untangle logic for success of interface monitoring
[obnox/samba/samba-obnox.git] / source3 / rpcclient / cmd_netlogon.c
1 /* 
2    Unix SMB/CIFS implementation.
3    RPC pipe client
4
5    Copyright (C) Tim Potter 2000
6    Copyright (C) Guenther Deschner 2008
7
8    This program is free software; you can redistribute it and/or modify
9    it under the terms of the GNU General Public License as published by
10    the Free Software Foundation; either version 3 of the License, or
11    (at your option) any later version.
12    
13    This program is distributed in the hope that it will be useful,
14    but WITHOUT ANY WARRANTY; without even the implied warranty of
15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16    GNU General Public License for more details.
17    
18    You should have received a copy of the GNU General Public License
19    along with this program.  If not, see <http://www.gnu.org/licenses/>.
20 */
21
22 #include "includes.h"
23 #include "rpcclient.h"
24 #include "../libcli/auth/libcli_auth.h"
25 #include "../librpc/gen_ndr/ndr_netlogon.h"
26 #include "../librpc/gen_ndr/ndr_netlogon_c.h"
27 #include "rpc_client/cli_netlogon.h"
28 #include "secrets.h"
29 #include "../libcli/auth/netlogon_creds_cli.h"
30
31 static WERROR cmd_netlogon_logon_ctrl2(struct rpc_pipe_client *cli,
32                                        TALLOC_CTX *mem_ctx, int argc,
33                                        const char **argv)
34 {
35         NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
36         WERROR werr;
37         const char *logon_server = cli->desthost;
38         enum netr_LogonControlCode function_code = NETLOGON_CONTROL_REDISCOVER;
39         uint32_t level = 1;
40         union netr_CONTROL_DATA_INFORMATION data;
41         union netr_CONTROL_QUERY_INFORMATION query;
42         const char *domain = lp_workgroup();
43         struct dcerpc_binding_handle *b = cli->binding_handle;
44         int i;
45 #define fn_code_level(x, item) { x, #x, #item }
46         struct {
47                 enum netr_LogonControlCode code;
48                 const char *name;
49                 const char *argument;
50         } supported_levels[] = {
51                 fn_code_level(NETLOGON_CONTROL_REDISCOVER, domain),
52                 fn_code_level(NETLOGON_CONTROL_TC_QUERY, domain),
53                 fn_code_level(NETLOGON_CONTROL_TRANSPORT_NOTIFY, domain),
54                 fn_code_level(NETLOGON_CONTROL_FIND_USER, user),
55                 fn_code_level(NETLOGON_CONTROL_CHANGE_PASSWORD, domain),
56                 fn_code_level(NETLOGON_CONTROL_TC_VERIFY, domain),
57                 fn_code_level(NETLOGON_CONTROL_SET_DBFLAG, debug_level),
58                 {0, 0, 0}
59         };
60 #undef fn_code_level
61         if ((argc > 5) || (argc < 2)) {
62                 fprintf(stderr, "Usage: %s <logon_server> <function_code> "
63                         "<level:1..4> <argument>\n", argv[0]);
64                 fprintf(stderr, "Supported combinations:\n");
65                 fprintf(stderr, "function_code\targument\n");
66                 for(i=0; supported_levels[i].code; i++) {
67                         fprintf(stderr, "%7d\t\t%s\t(%s)\n",
68                                 supported_levels[i].code,
69                                 supported_levels[i].argument,
70                                 supported_levels[i].name);
71                 }
72                 return WERR_OK;
73         }
74
75         if (argc >= 2) {
76                 logon_server = argv[1];
77         }
78
79         if (argc >= 3) {
80                 function_code = atoi(argv[2]);
81         }
82
83         if (argc >= 4) {
84                 level = atoi(argv[3]);
85         }
86
87         if (argc >= 5) {
88                 domain = argv[4];
89         }
90
91         switch (function_code) {
92                 case NETLOGON_CONTROL_REDISCOVER:
93                 case NETLOGON_CONTROL_TC_QUERY:
94                 case NETLOGON_CONTROL_CHANGE_PASSWORD:
95                 case NETLOGON_CONTROL_TRANSPORT_NOTIFY:
96                 case NETLOGON_CONTROL_TC_VERIFY:
97                         data.domain = domain;
98                         break;
99                 case NETLOGON_CONTROL_FIND_USER:
100                         data.user = domain;
101                         break;
102                 case NETLOGON_CONTROL_SET_DBFLAG:
103                         data.debug_level = atoi(domain);
104                 default:
105                         break;
106         }
107
108         status = dcerpc_netr_LogonControl2(b, mem_ctx,
109                                           logon_server,
110                                           function_code,
111                                           level,
112                                           &data,
113                                           &query,
114                                           &werr);
115         if (!NT_STATUS_IS_OK(status)) {
116                 return ntstatus_to_werror(status);
117         }
118
119         if (!W_ERROR_IS_OK(werr)) {
120                 return werr;
121         }
122
123         /* Display results */
124
125         return werr;
126 }
127
128 static WERROR cmd_netlogon_getanydcname(struct rpc_pipe_client *cli, 
129                                         TALLOC_CTX *mem_ctx, int argc, 
130                                         const char **argv)
131 {
132         const char *dcname = NULL;
133         WERROR werr;
134         NTSTATUS status;
135         int old_timeout;
136         struct dcerpc_binding_handle *b = cli->binding_handle;
137
138         if (argc != 2) {
139                 fprintf(stderr, "Usage: %s domainname\n", argv[0]);
140                 return WERR_OK;
141         }
142
143         /* Make sure to wait for our DC's reply */
144         old_timeout = rpccli_set_timeout(cli, 30000); /* 30 seconds. */
145         rpccli_set_timeout(cli, MAX(old_timeout, 30000)); /* At least 30 sec */
146
147         status = dcerpc_netr_GetAnyDCName(b, mem_ctx,
148                                           cli->desthost,
149                                           argv[1],
150                                           &dcname,
151                                           &werr);
152         rpccli_set_timeout(cli, old_timeout);
153
154         if (!NT_STATUS_IS_OK(status)) {
155                 return ntstatus_to_werror(status);
156         }
157
158         if (!W_ERROR_IS_OK(werr)) {
159                 return werr;
160         }
161
162         /* Display results */
163
164         printf("%s\n", dcname);
165
166         return werr;
167 }
168
169 static WERROR cmd_netlogon_getdcname(struct rpc_pipe_client *cli,
170                                      TALLOC_CTX *mem_ctx, int argc,
171                                      const char **argv)
172 {
173         const char *dcname = NULL;
174         NTSTATUS status;
175         WERROR werr;
176         int old_timeout;
177         struct dcerpc_binding_handle *b = cli->binding_handle;
178
179         if (argc != 2) {
180                 fprintf(stderr, "Usage: %s domainname\n", argv[0]);
181                 return WERR_OK;
182         }
183
184         /* Make sure to wait for our DC's reply */
185         old_timeout = rpccli_set_timeout(cli, 30000); /* 30 seconds. */
186         rpccli_set_timeout(cli, MAX(30000, old_timeout)); /* At least 30 sec */
187
188         status = dcerpc_netr_GetDcName(b, mem_ctx,
189                                        cli->desthost,
190                                        argv[1],
191                                        &dcname,
192                                        &werr);
193         rpccli_set_timeout(cli, old_timeout);
194
195         if (!NT_STATUS_IS_OK(status)) {
196                 return ntstatus_to_werror(status);
197         }
198
199         if (!W_ERROR_IS_OK(werr)) {
200                 return werr;
201         }
202
203         /* Display results */
204
205         printf("%s\n", dcname);
206
207         return werr;
208 }
209
210 static WERROR cmd_netlogon_dsr_getdcname(struct rpc_pipe_client *cli,
211                                          TALLOC_CTX *mem_ctx, int argc,
212                                          const char **argv)
213 {
214         NTSTATUS result;
215         WERROR werr = WERR_OK;
216         uint32_t flags = DS_RETURN_DNS_NAME;
217         const char *server_name = cli->desthost;
218         const char *domain_name;
219         struct GUID domain_guid = GUID_zero();
220         struct GUID site_guid = GUID_zero();
221         struct netr_DsRGetDCNameInfo *info = NULL;
222         struct dcerpc_binding_handle *b = cli->binding_handle;
223
224         if (argc < 2) {
225                 fprintf(stderr, "Usage: %s [domain_name] [domain_guid] "
226                                 "[site_guid] [flags]\n", argv[0]);
227                 return WERR_OK;
228         }
229
230         if (argc >= 2)
231                 domain_name = argv[1];
232
233         if (argc >= 3) {
234                 if (!NT_STATUS_IS_OK(GUID_from_string(argv[2], &domain_guid))) {
235                         return WERR_NOMEM;
236                 }
237         }
238
239         if (argc >= 4) {
240                 if (!NT_STATUS_IS_OK(GUID_from_string(argv[3], &site_guid))) {
241                         return WERR_NOMEM;
242                 }
243         }
244
245         if (argc >= 5)
246                 sscanf(argv[4], "%x", &flags);
247
248         result = dcerpc_netr_DsRGetDCName(b, mem_ctx,
249                                           server_name,
250                                           domain_name,
251                                           &domain_guid,
252                                           &site_guid,
253                                           flags,
254                                           &info,
255                                           &werr);
256         if (!NT_STATUS_IS_OK(result)) {
257                 return ntstatus_to_werror(result);
258         }
259
260         if (W_ERROR_IS_OK(werr)) {
261                 d_printf("DsGetDcName gave: %s\n",
262                 NDR_PRINT_STRUCT_STRING(mem_ctx, netr_DsRGetDCNameInfo, info));
263                 return WERR_OK;
264         }
265
266         printf("rpccli_netlogon_dsr_getdcname returned %s\n",
267                win_errstr(werr));
268
269         return werr;
270 }
271
272 static WERROR cmd_netlogon_dsr_getdcnameex(struct rpc_pipe_client *cli,
273                                            TALLOC_CTX *mem_ctx, int argc,
274                                            const char **argv)
275 {
276         WERROR result;
277         NTSTATUS status;
278         uint32_t flags = DS_RETURN_DNS_NAME;
279         const char *server_name = cli->desthost;
280         const char *domain_name;
281         const char *site_name = NULL;
282         struct GUID domain_guid = GUID_zero();
283         struct netr_DsRGetDCNameInfo *info = NULL;
284         struct dcerpc_binding_handle *b = cli->binding_handle;
285
286         if (argc < 2) {
287                 fprintf(stderr, "Usage: %s [domain_name] [domain_guid] "
288                                 "[site_name] [flags]\n", argv[0]);
289                 return WERR_OK;
290         }
291
292         domain_name = argv[1];
293
294         if (argc >= 3) {
295                 if (!NT_STATUS_IS_OK(GUID_from_string(argv[2], &domain_guid))) {
296                         return WERR_NOMEM;
297                 }
298         }
299
300         if (argc >= 4) {
301                 site_name = argv[3];
302         }
303
304         if (argc >= 5) {
305                 sscanf(argv[4], "%x", &flags);
306         }
307
308         status = dcerpc_netr_DsRGetDCNameEx(b, mem_ctx,
309                                             server_name,
310                                             domain_name,
311                                             &domain_guid,
312                                             site_name,
313                                             flags,
314                                             &info,
315                                             &result);
316         if (!NT_STATUS_IS_OK(status)) {
317                 return ntstatus_to_werror(status);
318         }
319
320         if (!W_ERROR_IS_OK(result)) {
321                 return result;
322         }
323
324         d_printf("DsRGetDCNameEx gave %s\n",
325                 NDR_PRINT_STRUCT_STRING(mem_ctx, netr_DsRGetDCNameInfo, info));
326
327         return result;
328 }
329
330 static WERROR cmd_netlogon_dsr_getdcnameex2(struct rpc_pipe_client *cli,
331                                             TALLOC_CTX *mem_ctx, int argc,
332                                             const char **argv)
333 {
334         WERROR result;
335         NTSTATUS status;
336         uint32_t flags = DS_RETURN_DNS_NAME;
337         const char *server_name = cli->desthost;
338         const char *domain_name = NULL;
339         const char *client_account = NULL;
340         uint32_t mask = 0;
341         const char *site_name = NULL;
342         struct GUID domain_guid = GUID_zero();
343         struct netr_DsRGetDCNameInfo *info = NULL;
344         struct dcerpc_binding_handle *b = cli->binding_handle;
345
346         if (argc < 2) {
347                 fprintf(stderr, "Usage: %s [client_account] [acb_mask] "
348                                 "[domain_name] [domain_guid] [site_name] "
349                                 "[flags]\n", argv[0]);
350                 return WERR_OK;
351         }
352
353         if (argc >= 2) {
354                 client_account = argv[1];
355         }
356
357         if (argc >= 3) {
358                 mask = atoi(argv[2]);
359         }
360
361         if (argc >= 4) {
362                 domain_name = argv[3];
363         }
364
365         if (argc >= 5) {
366                 if (!NT_STATUS_IS_OK(GUID_from_string(argv[4], &domain_guid))) {
367                         return WERR_NOMEM;
368                 }
369         }
370
371         if (argc >= 6) {
372                 site_name = argv[5];
373         }
374
375         if (argc >= 7) {
376                 sscanf(argv[6], "%x", &flags);
377         }
378
379         status = dcerpc_netr_DsRGetDCNameEx2(b, mem_ctx,
380                                              server_name,
381                                              client_account,
382                                              mask,
383                                              domain_name,
384                                              &domain_guid,
385                                              site_name,
386                                              flags,
387                                              &info,
388                                              &result);
389         if (!NT_STATUS_IS_OK(status)) {
390                 return ntstatus_to_werror(status);
391         }
392
393         if (!W_ERROR_IS_OK(result)) {
394                 return result;
395         }
396
397         d_printf("DsRGetDCNameEx2 gave %s\n",
398                 NDR_PRINT_STRUCT_STRING(mem_ctx, netr_DsRGetDCNameInfo, info));
399
400         return result;
401 }
402
403
404 static WERROR cmd_netlogon_dsr_getsitename(struct rpc_pipe_client *cli,
405                                            TALLOC_CTX *mem_ctx, int argc,
406                                            const char **argv)
407 {
408         WERROR werr;
409         NTSTATUS status;
410         const char *sitename = NULL;
411         struct dcerpc_binding_handle *b = cli->binding_handle;
412
413         if (argc != 2) {
414                 fprintf(stderr, "Usage: %s computername\n", argv[0]);
415                 return WERR_OK;
416         }
417
418         status = dcerpc_netr_DsRGetSiteName(b, mem_ctx,
419                                             argv[1],
420                                             &sitename,
421                                             &werr);
422         if (!NT_STATUS_IS_OK(status)) {
423                 return ntstatus_to_werror(status);
424         }
425
426         if (!W_ERROR_IS_OK(werr)) {
427                 printf("rpccli_netlogon_dsr_gesitename returned %s\n",
428                        nt_errstr(werror_to_ntstatus(werr)));
429                 return werr;
430         }
431
432         printf("Computer %s is on Site: %s\n", argv[1], sitename);
433
434         return WERR_OK;
435 }
436
437 static WERROR cmd_netlogon_logon_ctrl(struct rpc_pipe_client *cli,
438                                       TALLOC_CTX *mem_ctx, int argc,
439                                       const char **argv)
440 {
441         NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
442         WERROR werr;
443         const char *logon_server = cli->desthost;
444         enum netr_LogonControlCode function_code = 1;
445         uint32_t level = 1;
446         union netr_CONTROL_QUERY_INFORMATION info;
447         struct dcerpc_binding_handle *b = cli->binding_handle;
448
449         if (argc > 4) {
450                 fprintf(stderr, "Usage: %s <logon_server> <function_code> "
451                         "<level>\n", argv[0]);
452                 return WERR_OK;
453         }
454
455         if (argc >= 2) {
456                 logon_server = argv[1];
457         }
458
459         if (argc >= 3) {
460                 function_code = atoi(argv[2]);
461         }
462
463         if (argc >= 4) {
464                 level = atoi(argv[3]);
465         }
466
467         status = dcerpc_netr_LogonControl(b, mem_ctx,
468                                           logon_server,
469                                           function_code,
470                                           level,
471                                           &info,
472                                           &werr);
473         if (!NT_STATUS_IS_OK(status)) {
474                 return ntstatus_to_werror(status);
475         }
476
477         if (!W_ERROR_IS_OK(werr)) {
478                 return werr;
479         }
480
481         /* Display results */
482
483         return werr;
484 }
485
486 /* Display sam synchronisation information */
487
488 static void display_sam_sync(struct netr_DELTA_ENUM_ARRAY *r)
489 {
490         uint32_t i, j;
491
492         for (i=0; i < r->num_deltas; i++) {
493
494                 union netr_DELTA_UNION u = r->delta_enum[i].delta_union;
495                 union netr_DELTA_ID_UNION id = r->delta_enum[i].delta_id_union;
496
497                 switch (r->delta_enum[i].delta_type) {
498                 case NETR_DELTA_DOMAIN:
499                         printf("Domain: %s\n",
500                                 u.domain->domain_name.string);
501                         break;
502                 case NETR_DELTA_GROUP:
503                         printf("Group: %s\n",
504                                 u.group->group_name.string);
505                         break;
506                 case NETR_DELTA_DELETE_GROUP:
507                         printf("Delete Group: %d\n",
508                                 id.rid);
509                         break;
510                 case NETR_DELTA_RENAME_GROUP:
511                         printf("Rename Group: %s -> %s\n",
512                                 u.rename_group->OldName.string,
513                                 u.rename_group->NewName.string);
514                         break;
515                 case NETR_DELTA_USER:
516                         printf("Account: %s\n",
517                                 u.user->account_name.string);
518                         break;
519                 case NETR_DELTA_DELETE_USER:
520                         printf("Delete User: %d\n",
521                                 id.rid);
522                         break;
523                 case NETR_DELTA_RENAME_USER:
524                         printf("Rename user: %s -> %s\n",
525                                 u.rename_user->OldName.string,
526                                 u.rename_user->NewName.string);
527                         break;
528                 case NETR_DELTA_GROUP_MEMBER:
529                         for (j=0; j < u.group_member->num_rids; j++) {
530                                 printf("rid 0x%x, attrib 0x%08x\n",
531                                         u.group_member->rids[j],
532                                         u.group_member->attribs[j]);
533                         }
534                         break;
535                 case NETR_DELTA_ALIAS:
536                         printf("Alias: %s\n",
537                                 u.alias->alias_name.string);
538                         break;
539                 case NETR_DELTA_DELETE_ALIAS:
540                         printf("Delete Alias: %d\n",
541                                 id.rid);
542                         break;
543                 case NETR_DELTA_RENAME_ALIAS:
544                         printf("Rename alias: %s -> %s\n",
545                                 u.rename_alias->OldName.string,
546                                 u.rename_alias->NewName.string);
547                         break;
548                 case NETR_DELTA_ALIAS_MEMBER:
549                         for (j=0; j < u.alias_member->sids.num_sids; j++) {
550                                 fstring sid_str;
551                                 sid_to_fstring(sid_str,
552                                         u.alias_member->sids.sids[j].sid);
553                                 printf("%s\n", sid_str);
554                         }
555                         break;
556                 case NETR_DELTA_POLICY:
557                         printf("Policy: %s\n",
558                                 sid_string_dbg(id.sid));
559                         break;
560                 case NETR_DELTA_TRUSTED_DOMAIN:
561                         printf("Trusted Domain: %s\n",
562                                 u.trusted_domain->domain_name.string);
563                         break;
564                 case NETR_DELTA_DELETE_TRUST:
565                         printf("Delete Trust: %s\n",
566                                 sid_string_dbg(id.sid));
567                         break;
568                 case NETR_DELTA_ACCOUNT:
569                         printf("Account: %s\n",
570                                 sid_string_dbg(id.sid));
571                         break;
572                 case NETR_DELTA_DELETE_ACCOUNT:
573                         printf("Delete Account: %s\n",
574                                 sid_string_dbg(id.sid));
575                         break;
576                 case NETR_DELTA_SECRET:
577                         printf("Secret: %s\n",
578                                 id.name);
579                         break;
580                 case NETR_DELTA_DELETE_SECRET:
581                         printf("Delete Secret: %s\n",
582                                 id.name);
583                         break;
584                 case NETR_DELTA_DELETE_GROUP2:
585                         printf("Delete Group2: %s\n",
586                                 u.delete_group->account_name);
587                         break;
588                 case NETR_DELTA_DELETE_USER2:
589                         printf("Delete User2: %s\n",
590                                 u.delete_user->account_name);
591                         break;
592                 case NETR_DELTA_MODIFY_COUNT:
593                         printf("sam sequence update: 0x%016llx\n",
594                                 (unsigned long long) *u.modified_count);
595                         break;
596                 default:
597                         printf("unknown delta type 0x%02x\n",
598                                 r->delta_enum[i].delta_type);
599                         break;
600                 }
601         }
602 }
603
604 /* Perform sam synchronisation */
605
606 static NTSTATUS cmd_netlogon_sam_sync(struct rpc_pipe_client *cli,
607                                       TALLOC_CTX *mem_ctx, int argc,
608                                       const char **argv)
609 {
610         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
611         NTSTATUS status;
612         const char *logon_server = cli->desthost;
613         const char *computername = lp_netbios_name();
614         struct netr_Authenticator credential;
615         struct netr_Authenticator return_authenticator;
616         enum netr_SamDatabaseID database_id = SAM_DATABASE_DOMAIN;
617         uint16_t restart_state = 0;
618         uint32_t sync_context = 0;
619         struct dcerpc_binding_handle *b = cli->binding_handle;
620
621         if (argc > 2) {
622                 fprintf(stderr, "Usage: %s [database_id]\n", argv[0]);
623                 return NT_STATUS_OK;
624         }
625
626         if (argc == 2) {
627                 database_id = atoi(argv[1]);
628         }
629
630         /* Synchronise sam database */
631
632         do {
633                 struct netr_DELTA_ENUM_ARRAY *delta_enum_array = NULL;
634                 struct netlogon_creds_CredentialState *creds = NULL;
635
636                 if (rpcclient_netlogon_creds == NULL) {
637                         return NT_STATUS_UNSUCCESSFUL;
638                 }
639
640                 status = netlogon_creds_cli_lock(rpcclient_netlogon_creds,
641                                                  mem_ctx, &creds);
642                 if (!NT_STATUS_IS_OK(status)) {
643                         return status;
644                 }
645
646                 netlogon_creds_client_authenticator(creds, &credential);
647
648                 status = dcerpc_netr_DatabaseSync2(b, mem_ctx,
649                                                    logon_server,
650                                                    computername,
651                                                    &credential,
652                                                    &return_authenticator,
653                                                    database_id,
654                                                    restart_state,
655                                                    &sync_context,
656                                                    &delta_enum_array,
657                                                    0xffff,
658                                                    &result);
659                 if (!NT_STATUS_IS_OK(status)) {
660                         TALLOC_FREE(creds);
661                         return status;
662                 }
663
664                 /* Check returned credentials. */
665                 if (!netlogon_creds_client_check(creds,
666                                                  &return_authenticator.cred)) {
667                         DEBUG(0,("credentials chain check failed\n"));
668                         TALLOC_FREE(creds);
669                         return NT_STATUS_ACCESS_DENIED;
670                 }
671                 TALLOC_FREE(creds);
672
673                 if (NT_STATUS_IS_ERR(result)) {
674                         break;
675                 }
676
677                 /* Display results */
678
679                 display_sam_sync(delta_enum_array);
680
681                 TALLOC_FREE(delta_enum_array);
682
683         } while (NT_STATUS_EQUAL(result, STATUS_MORE_ENTRIES));
684
685         return result;
686 }
687
688 /* Perform sam delta synchronisation */
689
690 static NTSTATUS cmd_netlogon_sam_deltas(struct rpc_pipe_client *cli,
691                                         TALLOC_CTX *mem_ctx, int argc,
692                                         const char **argv)
693 {
694         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
695         NTSTATUS status;
696         uint32_t tmp;
697         const char *logon_server = cli->desthost;
698         const char *computername = lp_netbios_name();
699         struct netr_Authenticator credential;
700         struct netr_Authenticator return_authenticator;
701         enum netr_SamDatabaseID database_id = SAM_DATABASE_DOMAIN;
702         uint64_t sequence_num;
703         struct dcerpc_binding_handle *b = cli->binding_handle;
704
705         if (argc != 3) {
706                 fprintf(stderr, "Usage: %s database_id seqnum\n", argv[0]);
707                 return NT_STATUS_OK;
708         }
709
710         database_id = atoi(argv[1]);
711         tmp = atoi(argv[2]);
712
713         sequence_num = tmp & 0xffff;
714
715         do {
716                 struct netr_DELTA_ENUM_ARRAY *delta_enum_array = NULL;
717                 struct netlogon_creds_CredentialState *creds = NULL;
718
719                 if (rpcclient_netlogon_creds == NULL) {
720                         return NT_STATUS_UNSUCCESSFUL;
721                 }
722
723                 status = netlogon_creds_cli_lock(rpcclient_netlogon_creds,
724                                                  mem_ctx, &creds);
725                 if (!NT_STATUS_IS_OK(status)) {
726                         return status;
727                 }
728
729                 netlogon_creds_client_authenticator(creds, &credential);
730
731                 status = dcerpc_netr_DatabaseDeltas(b, mem_ctx,
732                                                     logon_server,
733                                                     computername,
734                                                     &credential,
735                                                     &return_authenticator,
736                                                     database_id,
737                                                     &sequence_num,
738                                                     &delta_enum_array,
739                                                     0xffff,
740                                                     &result);
741                 if (!NT_STATUS_IS_OK(status)) {
742                         TALLOC_FREE(creds);
743                         return status;
744                 }
745
746                 /* Check returned credentials. */
747                 if (!netlogon_creds_client_check(creds,
748                                                  &return_authenticator.cred)) {
749                         DEBUG(0,("credentials chain check failed\n"));
750                         TALLOC_FREE(creds);
751                         return NT_STATUS_ACCESS_DENIED;
752                 }
753                 TALLOC_FREE(creds);
754
755                 if (NT_STATUS_IS_ERR(result)) {
756                         break;
757                 }
758
759                 /* Display results */
760
761                 display_sam_sync(delta_enum_array);
762
763                 TALLOC_FREE(delta_enum_array);
764
765         } while (NT_STATUS_EQUAL(result, STATUS_MORE_ENTRIES));
766
767         return result;
768 }
769
770 /* Log on a domain user */
771
772 static NTSTATUS cmd_netlogon_sam_logon(struct rpc_pipe_client *cli, 
773                                        TALLOC_CTX *mem_ctx, int argc,
774                                        const char **argv)
775 {
776         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
777         int logon_type = NetlogonNetworkInformation;
778         const char *username, *password;
779         uint32_t logon_param = 0;
780         const char *workstation = NULL;
781         struct netr_SamInfo3 *info3 = NULL;
782
783         /* Check arguments */
784
785         if (argc < 3 || argc > 6) {
786                 fprintf(stderr, "Usage: samlogon <username> <password> [workstation]"
787                         "[logon_type (1 or 2)] [logon_parameter]\n");
788                 return NT_STATUS_OK;
789         }
790
791         username = argv[1];
792         password = argv[2];
793
794         if (argc >= 4) 
795                 workstation = argv[3];
796
797         if (argc >= 5)
798                 sscanf(argv[4], "%i", &logon_type);
799
800         if (argc == 6)
801                 sscanf(argv[5], "%x", &logon_param);
802
803         if (rpcclient_netlogon_creds == NULL) {
804                 result = NT_STATUS_UNSUCCESSFUL;
805                 goto done;
806         }
807
808         /* Perform the sam logon */
809
810         result = rpccli_netlogon_password_logon(rpcclient_netlogon_creds,
811                                                 cli->binding_handle,
812                                                 mem_ctx,
813                                                 logon_param,
814                                                 lp_workgroup(),
815                                                 username,
816                                                 password,
817                                                 workstation,
818                                                 logon_type,
819                                                 &info3);
820         if (!NT_STATUS_IS_OK(result))
821                 goto done;
822
823  done:
824         return result;
825 }
826
827 /* Change the trust account password */
828
829 static NTSTATUS cmd_netlogon_change_trust_pw(struct rpc_pipe_client *cli, 
830                                              TALLOC_CTX *mem_ctx, int argc,
831                                              const char **argv)
832 {
833         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
834
835         /* Check arguments */
836
837         if (argc > 1) {
838                 fprintf(stderr, "Usage: change_trust_pw");
839                 return NT_STATUS_OK;
840         }
841
842         result = trust_pw_change(rpcclient_netlogon_creds,
843                                  rpcclient_msg_ctx,
844                                  cli->binding_handle,
845                                  lp_workgroup(),
846                                  true); /* force */
847         if (!NT_STATUS_IS_OK(result))
848                 goto done;
849
850  done:
851         return result;
852 }
853
854 static WERROR cmd_netlogon_gettrustrid(struct rpc_pipe_client *cli,
855                                        TALLOC_CTX *mem_ctx, int argc,
856                                        const char **argv)
857 {
858         NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
859         WERROR werr = WERR_GENERAL_FAILURE;
860         const char *server_name = cli->desthost;
861         const char *domain_name = lp_workgroup();
862         uint32_t rid = 0;
863         struct dcerpc_binding_handle *b = cli->binding_handle;
864
865         if (argc < 1 || argc > 3) {
866                 fprintf(stderr, "Usage: %s <server_name> <domain_name>\n",
867                         argv[0]);
868                 return WERR_OK;
869         }
870
871         if (argc >= 2) {
872                 server_name = argv[1];
873         }
874
875         if (argc >= 3) {
876                 domain_name = argv[2];
877         }
878
879         status = dcerpc_netr_LogonGetTrustRid(b, mem_ctx,
880                                               server_name,
881                                               domain_name,
882                                               &rid,
883                                               &werr);
884         if (!NT_STATUS_IS_OK(status)) {
885                 werr = ntstatus_to_werror(status);
886                 goto done;
887         }
888
889         if (W_ERROR_IS_OK(werr)) {
890                 printf("Rid: %d\n", rid);
891         }
892  done:
893         return werr;
894 }
895
896 static WERROR cmd_netlogon_dsr_enumtrustdom(struct rpc_pipe_client *cli,
897                                             TALLOC_CTX *mem_ctx, int argc,
898                                             const char **argv)
899 {
900         NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
901         WERROR werr = WERR_GENERAL_FAILURE;
902         const char *server_name = cli->desthost;
903         uint32_t trust_flags = NETR_TRUST_FLAG_IN_FOREST;
904         struct netr_DomainTrustList trusts;
905         struct dcerpc_binding_handle *b = cli->binding_handle;
906
907         if (argc < 1 || argc > 3) {
908                 fprintf(stderr, "Usage: %s <server_name> <trust_flags>\n",
909                         argv[0]);
910                 return WERR_OK;
911         }
912
913         if (argc >= 2) {
914                 server_name = argv[1];
915         }
916
917         if (argc >= 3) {
918                 sscanf(argv[2], "%x", &trust_flags);
919         }
920
921         status = dcerpc_netr_DsrEnumerateDomainTrusts(b, mem_ctx,
922                                                       server_name,
923                                                       trust_flags,
924                                                       &trusts,
925                                                       &werr);
926         if (!NT_STATUS_IS_OK(status)) {
927                 werr = ntstatus_to_werror(status);
928                 goto done;
929         }
930
931         if (W_ERROR_IS_OK(werr)) {
932                 int i;
933
934                 printf("%d domains returned\n", trusts.count);
935
936                 for (i=0; i<trusts.count; i++ ) {
937                         printf("%s (%s)\n",
938                                 trusts.array[i].dns_name,
939                                 trusts.array[i].netbios_name);
940                 }
941         }
942  done:
943         return werr;
944 }
945
946 static WERROR cmd_netlogon_deregisterdnsrecords(struct rpc_pipe_client *cli,
947                                                 TALLOC_CTX *mem_ctx, int argc,
948                                                 const char **argv)
949 {
950         NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
951         WERROR werr = WERR_GENERAL_FAILURE;
952         const char *server_name = cli->desthost;
953         const char *domain = lp_workgroup();
954         const char *dns_host = NULL;
955         struct dcerpc_binding_handle *b = cli->binding_handle;
956
957         if (argc < 1 || argc > 4) {
958                 fprintf(stderr, "Usage: %s <server_name> <domain_name> "
959                         "<dns_host>\n", argv[0]);
960                 return WERR_OK;
961         }
962
963         if (argc >= 2) {
964                 server_name = argv[1];
965         }
966
967         if (argc >= 3) {
968                 domain = argv[2];
969         }
970
971         if (argc >= 4) {
972                 dns_host = argv[3];
973         }
974
975         status = dcerpc_netr_DsrDeregisterDNSHostRecords(b, mem_ctx,
976                                                          server_name,
977                                                          domain,
978                                                          NULL,
979                                                          NULL,
980                                                          dns_host,
981                                                          &werr);
982         if (!NT_STATUS_IS_OK(status)) {
983                 werr = ntstatus_to_werror(status);
984                 goto done;
985         }
986
987         if (W_ERROR_IS_OK(werr)) {
988                 printf("success\n");
989         }
990  done:
991         return werr;
992 }
993
994 static WERROR cmd_netlogon_dsr_getforesttrustinfo(struct rpc_pipe_client *cli,
995                                                   TALLOC_CTX *mem_ctx, int argc,
996                                                   const char **argv)
997 {
998         NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
999         WERROR werr = WERR_GENERAL_FAILURE;
1000         const char *server_name = cli->desthost;
1001         const char *trusted_domain_name = NULL;
1002         struct lsa_ForestTrustInformation *info = NULL;
1003         uint32_t flags = 0;
1004         struct dcerpc_binding_handle *b = cli->binding_handle;
1005
1006         if (argc < 1 || argc > 4) {
1007                 fprintf(stderr, "Usage: %s <server_name> <trusted_domain_name> "
1008                         "<flags>\n", argv[0]);
1009                 return WERR_OK;
1010         }
1011
1012         if (argc >= 2) {
1013                 server_name = argv[1];
1014         }
1015
1016         if (argc >= 3) {
1017                 trusted_domain_name = argv[2];
1018         }
1019
1020         if (argc >= 4) {
1021                 sscanf(argv[3], "%x", &flags);
1022         }
1023
1024         status = dcerpc_netr_DsRGetForestTrustInformation(b, mem_ctx,
1025                                                          server_name,
1026                                                          trusted_domain_name,
1027                                                          flags,
1028                                                          &info,
1029                                                          &werr);
1030         if (!NT_STATUS_IS_OK(status)) {
1031                 werr = ntstatus_to_werror(status);
1032                 goto done;
1033         }
1034
1035         if (W_ERROR_IS_OK(werr)) {
1036                 printf("success\n");
1037         }
1038  done:
1039         return werr;
1040 }
1041
1042 static NTSTATUS cmd_netlogon_enumtrusteddomains(struct rpc_pipe_client *cli,
1043                                                 TALLOC_CTX *mem_ctx, int argc,
1044                                                 const char **argv)
1045 {
1046         NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
1047         NTSTATUS result;
1048         const char *server_name = cli->desthost;
1049         struct netr_Blob blob;
1050         struct dcerpc_binding_handle *b = cli->binding_handle;
1051
1052
1053         if (argc < 1 || argc > 3) {
1054                 fprintf(stderr, "Usage: %s <server_name>\n", argv[0]);
1055                 return NT_STATUS_OK;
1056         }
1057
1058         if (argc >= 2) {
1059                 server_name = argv[1];
1060         }
1061
1062         status = dcerpc_netr_NetrEnumerateTrustedDomains(b, mem_ctx,
1063                                                          server_name,
1064                                                          &blob,
1065                                                          &result);
1066         if (!NT_STATUS_IS_OK(status)) {
1067                 goto done;
1068         }
1069
1070         if (!NT_STATUS_IS_OK(result)) {
1071                 status = result;
1072                 goto done;
1073         }
1074
1075         printf("success\n");
1076         dump_data(1, blob.data, blob.length);
1077  done:
1078         return status;
1079 }
1080
1081 static WERROR cmd_netlogon_enumtrusteddomainsex(struct rpc_pipe_client *cli,
1082                                                 TALLOC_CTX *mem_ctx, int argc,
1083                                                 const char **argv)
1084 {
1085         NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
1086         WERROR werr = WERR_GENERAL_FAILURE;
1087         const char *server_name = cli->desthost;
1088         struct netr_DomainTrustList list;
1089         struct dcerpc_binding_handle *b = cli->binding_handle;
1090
1091         if (argc < 1 || argc > 3) {
1092                 fprintf(stderr, "Usage: %s <server_name>\n", argv[0]);
1093                 return WERR_OK;
1094         }
1095
1096         if (argc >= 2) {
1097                 server_name = argv[1];
1098         }
1099
1100         status = dcerpc_netr_NetrEnumerateTrustedDomainsEx(b, mem_ctx,
1101                                                            server_name,
1102                                                            &list,
1103                                                            &werr);
1104         if (!NT_STATUS_IS_OK(status)) {
1105                 werr = ntstatus_to_werror(status);
1106                 goto done;
1107         }
1108
1109         if (W_ERROR_IS_OK(werr)) {
1110                 printf("success\n");
1111         }
1112  done:
1113         return werr;
1114 }
1115
1116 static WERROR cmd_netlogon_getdcsitecoverage(struct rpc_pipe_client *cli,
1117                                              TALLOC_CTX *mem_ctx, int argc,
1118                                              const char **argv)
1119 {
1120         NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
1121         WERROR werr = WERR_GENERAL_FAILURE;
1122         const char *server_name = cli->desthost;
1123         struct DcSitesCtr *ctr = NULL;
1124         struct dcerpc_binding_handle *b = cli->binding_handle;
1125
1126         if (argc < 1 || argc > 3) {
1127                 fprintf(stderr, "Usage: %s <server_name>\n", argv[0]);
1128                 return WERR_OK;
1129         }
1130
1131         if (argc >= 2) {
1132                 server_name = argv[1];
1133         }
1134
1135         status = dcerpc_netr_DsrGetDcSiteCoverageW(b, mem_ctx,
1136                                                    server_name,
1137                                                    &ctr,
1138                                                    &werr);
1139         if (!NT_STATUS_IS_OK(status)) {
1140                 werr = ntstatus_to_werror(status);
1141                 goto done;
1142         }
1143
1144         if (W_ERROR_IS_OK(werr) && ctr->num_sites) {
1145                 int i;
1146                 printf("sites covered by this DC: %d\n", ctr->num_sites);
1147                 for (i=0; i<ctr->num_sites; i++) {
1148                         printf("%s\n", ctr->sites[i].string);
1149                 }
1150         }
1151  done:
1152         return werr;
1153 }
1154
1155 static NTSTATUS cmd_netlogon_database_redo(struct rpc_pipe_client *cli,
1156                                            TALLOC_CTX *mem_ctx, int argc,
1157                                            const char **argv)
1158 {
1159         NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
1160         NTSTATUS result;
1161         const char *server_name = cli->desthost;
1162         struct netr_Authenticator clnt_creds, srv_cred;
1163         struct netr_DELTA_ENUM_ARRAY *delta_enum_array = NULL;
1164         struct netr_ChangeLogEntry e;
1165         uint32_t rid = 500;
1166         struct dcerpc_binding_handle *b = cli->binding_handle;
1167         struct netlogon_creds_CredentialState *creds = NULL;
1168
1169         if (argc > 2) {
1170                 fprintf(stderr, "Usage: %s <user rid>\n", argv[0]);
1171                 return NT_STATUS_OK;
1172         }
1173
1174         if (argc == 2) {
1175                 sscanf(argv[1], "%d", &rid);
1176         }
1177
1178         if (rpcclient_netlogon_creds == NULL) {
1179                 return NT_STATUS_UNSUCCESSFUL;
1180         }
1181
1182         status = netlogon_creds_cli_lock(rpcclient_netlogon_creds,
1183                                          mem_ctx, &creds);
1184         if (!NT_STATUS_IS_OK(status)) {
1185                 return status;
1186         }
1187
1188         netlogon_creds_client_authenticator(creds, &clnt_creds);
1189
1190         ZERO_STRUCT(e);
1191
1192         e.object_rid            = rid;
1193         e.db_index              = SAM_DATABASE_DOMAIN;
1194         e.delta_type            = NETR_DELTA_USER;
1195
1196         status = dcerpc_netr_DatabaseRedo(b, mem_ctx,
1197                                           server_name,
1198                                           lp_netbios_name(),
1199                                           &clnt_creds,
1200                                           &srv_cred,
1201                                           e,
1202                                           0, /* is calculated automatically */
1203                                           &delta_enum_array,
1204                                           &result);
1205         if (!NT_STATUS_IS_OK(status)) {
1206                 TALLOC_FREE(creds);
1207                 return status;
1208         }
1209
1210         if (!netlogon_creds_client_check(creds, &srv_cred.cred)) {
1211                 DEBUG(0,("credentials chain check failed\n"));
1212                 TALLOC_FREE(creds);
1213                 return NT_STATUS_ACCESS_DENIED;
1214         }
1215         TALLOC_FREE(creds);
1216
1217         return result;
1218 }
1219
1220 static NTSTATUS cmd_netlogon_capabilities(struct rpc_pipe_client *cli,
1221                                           TALLOC_CTX *mem_ctx, int argc,
1222                                           const char **argv)
1223 {
1224         NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
1225         NTSTATUS result;
1226         struct netr_Authenticator credential;
1227         struct netr_Authenticator return_authenticator;
1228         union netr_Capabilities capabilities;
1229         uint32_t level = 1;
1230         struct dcerpc_binding_handle *b = cli->binding_handle;
1231         struct netlogon_creds_CredentialState *creds = NULL;
1232
1233         if (argc > 2) {
1234                 fprintf(stderr, "Usage: %s <level>\n", argv[0]);
1235                 return NT_STATUS_OK;
1236         }
1237
1238         if (argc == 2) {
1239                 level = atoi(argv[1]);
1240         }
1241
1242         ZERO_STRUCT(return_authenticator);
1243
1244         if (rpcclient_netlogon_creds == NULL) {
1245                 return NT_STATUS_UNSUCCESSFUL;
1246         }
1247
1248         status = netlogon_creds_cli_lock(rpcclient_netlogon_creds,
1249                                          mem_ctx, &creds);
1250         if (!NT_STATUS_IS_OK(status)) {
1251                 return status;
1252         }
1253
1254         netlogon_creds_client_authenticator(creds, &credential);
1255
1256         status = dcerpc_netr_LogonGetCapabilities(b, mem_ctx,
1257                                                   cli->desthost,
1258                                                   lp_netbios_name(),
1259                                                   &credential,
1260                                                   &return_authenticator,
1261                                                   level,
1262                                                   &capabilities,
1263                                                   &result);
1264         if (!NT_STATUS_IS_OK(status)) {
1265                 TALLOC_FREE(creds);
1266                 return status;
1267         }
1268
1269         if (!netlogon_creds_client_check(creds,
1270                                          &return_authenticator.cred)) {
1271                 DEBUG(0,("credentials chain check failed\n"));
1272                 TALLOC_FREE(creds);
1273                 return NT_STATUS_ACCESS_DENIED;
1274         }
1275         TALLOC_FREE(creds);
1276
1277         printf("capabilities: 0x%08x\n", capabilities.server_capabilities);
1278
1279         return result;
1280 }
1281
1282 /* List of commands exported by this module */
1283
1284 struct cmd_set netlogon_commands[] = {
1285
1286         { "NETLOGON" },
1287
1288         { "logonctrl2", RPC_RTYPE_WERROR, NULL, cmd_netlogon_logon_ctrl2, &ndr_table_netlogon, NULL, "Logon Control 2",     "" },
1289         { "getanydcname", RPC_RTYPE_WERROR, NULL, cmd_netlogon_getanydcname, &ndr_table_netlogon, NULL, "Get trusted DC name",     "" },
1290         { "getdcname", RPC_RTYPE_WERROR, NULL, cmd_netlogon_getdcname, &ndr_table_netlogon, NULL, "Get trusted PDC name",     "" },
1291         { "dsr_getdcname", RPC_RTYPE_WERROR, NULL, cmd_netlogon_dsr_getdcname, &ndr_table_netlogon, NULL, "Get trusted DC name",     "" },
1292         { "dsr_getdcnameex", RPC_RTYPE_WERROR, NULL, cmd_netlogon_dsr_getdcnameex, &ndr_table_netlogon, NULL, "Get trusted DC name",     "" },
1293         { "dsr_getdcnameex2", RPC_RTYPE_WERROR, NULL, cmd_netlogon_dsr_getdcnameex2, &ndr_table_netlogon, NULL, "Get trusted DC name",     "" },
1294         { "dsr_getsitename", RPC_RTYPE_WERROR, NULL, cmd_netlogon_dsr_getsitename, &ndr_table_netlogon, NULL, "Get sitename",     "" },
1295         { "dsr_getforesttrustinfo", RPC_RTYPE_WERROR, NULL, cmd_netlogon_dsr_getforesttrustinfo, &ndr_table_netlogon, NULL, "Get Forest Trust Info",     "" },
1296         { "logonctrl",  RPC_RTYPE_WERROR, NULL, cmd_netlogon_logon_ctrl, &ndr_table_netlogon, NULL, "Logon Control",       "" },
1297         { "samsync",    RPC_RTYPE_NTSTATUS, cmd_netlogon_sam_sync,    NULL, &ndr_table_netlogon, NULL, "Sam Synchronisation", "", .use_netlogon_creds = true, },
1298         { "samdeltas",  RPC_RTYPE_NTSTATUS, cmd_netlogon_sam_deltas,  NULL, &ndr_table_netlogon, NULL, "Query Sam Deltas",    "", .use_netlogon_creds = true, },
1299         { "samlogon",   RPC_RTYPE_NTSTATUS, cmd_netlogon_sam_logon,   NULL, &ndr_table_netlogon, NULL, "Sam Logon",           "", .use_netlogon_creds = true, },
1300         { "change_trust_pw",   RPC_RTYPE_NTSTATUS, cmd_netlogon_change_trust_pw,   NULL, &ndr_table_netlogon, NULL, "Change Trust Account Password",           "", .use_netlogon_creds = true, },
1301         { "gettrustrid", RPC_RTYPE_WERROR, NULL, cmd_netlogon_gettrustrid, &ndr_table_netlogon, NULL, "Get trust rid",     "" },
1302         { "dsr_enumtrustdom", RPC_RTYPE_WERROR, NULL, cmd_netlogon_dsr_enumtrustdom, &ndr_table_netlogon, NULL, "Enumerate trusted domains",     "" },
1303         { "dsenumdomtrusts",  RPC_RTYPE_WERROR, NULL, cmd_netlogon_dsr_enumtrustdom, &ndr_table_netlogon, NULL, "Enumerate all trusted domains in an AD forest",     "" },
1304         { "deregisterdnsrecords", RPC_RTYPE_WERROR, NULL, cmd_netlogon_deregisterdnsrecords, &ndr_table_netlogon, NULL, "Deregister DNS records",     "" },
1305         { "netrenumtrusteddomains", RPC_RTYPE_NTSTATUS, cmd_netlogon_enumtrusteddomains, NULL, &ndr_table_netlogon, NULL, "Enumerate trusted domains",     "" },
1306         { "netrenumtrusteddomainsex", RPC_RTYPE_WERROR, NULL, cmd_netlogon_enumtrusteddomainsex, &ndr_table_netlogon, NULL, "Enumerate trusted domains",     "" },
1307         { "getdcsitecoverage", RPC_RTYPE_WERROR, NULL, cmd_netlogon_getdcsitecoverage, &ndr_table_netlogon, NULL, "Get the Site-Coverage from a DC",     "" },
1308         { "database_redo", RPC_RTYPE_NTSTATUS, cmd_netlogon_database_redo, NULL, &ndr_table_netlogon, NULL, "Replicate single object from a DC",     "", .use_netlogon_creds = true, },
1309         { "capabilities", RPC_RTYPE_NTSTATUS, cmd_netlogon_capabilities, NULL, &ndr_table_netlogon, NULL, "Return Capabilities",     "", .use_netlogon_creds = true, },
1310
1311         { NULL }
1312 };