c47c1507e5dbb341a403a1cce8a73d77c513ec6e
[bbaumbach/samba-autobuild/.git] / source3 / rpcclient / rpcclient.c
1 /* 
2    Unix SMB/CIFS implementation.
3    RPC pipe client
4
5    Copyright (C) Tim Potter 2000-2001
6    Copyright (C) Martin Pool 2003
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 "popt_common.h"
24 #include "rpcclient.h"
25 #include "../libcli/auth/libcli_auth.h"
26 #include "../librpc/gen_ndr/ndr_lsa_c.h"
27 #include "rpc_client/cli_lsarpc.h"
28 #include "../librpc/gen_ndr/ndr_netlogon.h"
29 #include "rpc_client/cli_netlogon.h"
30 #include "../libcli/smbreadline/smbreadline.h"
31 #include "../libcli/security/security.h"
32 #include "passdb.h"
33 #include "libsmb/libsmb.h"
34 #include "auth/gensec/gensec.h"
35 #include "../libcli/smb/smbXcli_base.h"
36 #include "messages.h"
37
38 enum pipe_auth_type_spnego {
39         PIPE_AUTH_TYPE_SPNEGO_NONE = 0,
40         PIPE_AUTH_TYPE_SPNEGO_NTLMSSP,
41         PIPE_AUTH_TYPE_SPNEGO_KRB5
42 };
43
44 struct dom_sid domain_sid;
45
46 static enum dcerpc_AuthType pipe_default_auth_type = DCERPC_AUTH_TYPE_NONE;
47 static enum pipe_auth_type_spnego pipe_default_auth_spnego_type = 0;
48 static enum dcerpc_AuthLevel pipe_default_auth_level = DCERPC_AUTH_LEVEL_NONE;
49 static unsigned int timeout = 0;
50 static enum dcerpc_transport_t default_transport = NCACN_NP;
51
52 struct messaging_context *rpcclient_msg_ctx;
53 struct user_auth_info *rpcclient_auth_info;
54 struct cli_state *rpcclient_cli_state;
55 struct netlogon_creds_cli_context *rpcclient_netlogon_creds;
56
57 /* List to hold groups of commands.
58  *
59  * Commands are defined in a list of arrays: arrays are easy to
60  * statically declare, and lists are easier to dynamically extend.
61  */
62
63 static struct cmd_list {
64         struct cmd_list *prev, *next;
65         struct cmd_set *cmd_set;
66 } *cmd_list;
67
68 /****************************************************************************
69 handle completion of commands for readline
70 ****************************************************************************/
71 static char **completion_fn(const char *text, int start, int end)
72 {
73 #define MAX_COMPLETIONS 1000
74         char **matches;
75         int i, count=0;
76         struct cmd_list *commands = cmd_list;
77
78 #if 0   /* JERRY */
79         /* FIXME!!!  -- what to do when completing argument? */
80         /* for words not at the start of the line fallback 
81            to filename completion */
82         if (start) 
83                 return NULL;
84 #endif
85
86         /* make sure we have a list of valid commands */
87         if (!commands) {
88                 return NULL;
89         }
90
91         matches = SMB_MALLOC_ARRAY(char *, MAX_COMPLETIONS);
92         if (!matches) {
93                 return NULL;
94         }
95
96         matches[count++] = SMB_STRDUP(text);
97         if (!matches[0]) {
98                 SAFE_FREE(matches);
99                 return NULL;
100         }
101
102         while (commands && count < MAX_COMPLETIONS-1) {
103                 if (!commands->cmd_set) {
104                         break;
105                 }
106
107                 for (i=0; commands->cmd_set[i].name; i++) {
108                         if ((strncmp(text, commands->cmd_set[i].name, strlen(text)) == 0) &&
109                                 (( commands->cmd_set[i].returntype == RPC_RTYPE_NTSTATUS &&
110                         commands->cmd_set[i].ntfn ) || 
111                       ( commands->cmd_set[i].returntype == RPC_RTYPE_WERROR &&
112                         commands->cmd_set[i].wfn))) {
113                                 matches[count] = SMB_STRDUP(commands->cmd_set[i].name);
114                                 if (!matches[count]) {
115                                         for (i = 0; i < count; i++) {
116                                                 SAFE_FREE(matches[count]);
117                                         }
118                                         SAFE_FREE(matches);
119                                         return NULL;
120                                 }
121                                 count++;
122                         }
123                 }
124                 commands = commands->next;
125         }
126
127         if (count == 2) {
128                 SAFE_FREE(matches[0]);
129                 matches[0] = SMB_STRDUP(matches[1]);
130         }
131         matches[count] = NULL;
132         return matches;
133 }
134
135 static char *next_command (char **cmdstr)
136 {
137         char *command;
138         char                    *p;
139
140         if (!cmdstr || !(*cmdstr))
141                 return NULL;
142
143         p = strchr_m(*cmdstr, ';');
144         if (p)
145                 *p = '\0';
146         command = SMB_STRDUP(*cmdstr);
147         if (p)
148                 *cmdstr = p + 1;
149         else
150                 *cmdstr = NULL;
151
152         return command;
153 }
154
155 /* Fetch the SID for this computer */
156
157 static void fetch_machine_sid(struct cli_state *cli)
158 {
159         struct policy_handle pol;
160         NTSTATUS result = NT_STATUS_OK, status;
161         static bool got_domain_sid;
162         TALLOC_CTX *mem_ctx;
163         struct rpc_pipe_client *lsapipe = NULL;
164         union lsa_PolicyInformation *info = NULL;
165         struct dcerpc_binding_handle *b;
166
167         if (got_domain_sid) return;
168
169         if (!(mem_ctx=talloc_init("fetch_machine_sid"))) {
170                 DEBUG(0,("fetch_machine_sid: talloc_init returned NULL!\n"));
171                 goto error;
172         }
173
174         result = cli_rpc_pipe_open_noauth(cli, &ndr_table_lsarpc,
175                                           &lsapipe);
176         if (!NT_STATUS_IS_OK(result)) {
177                 fprintf(stderr, "could not initialise lsa pipe. Error was %s\n", nt_errstr(result) );
178                 goto error;
179         }
180
181         b = lsapipe->binding_handle;
182
183         result = rpccli_lsa_open_policy(lsapipe, mem_ctx, True, 
184                                      SEC_FLAG_MAXIMUM_ALLOWED,
185                                      &pol);
186         if (!NT_STATUS_IS_OK(result)) {
187                 goto error;
188         }
189
190         status = dcerpc_lsa_QueryInfoPolicy(b, mem_ctx,
191                                             &pol,
192                                             LSA_POLICY_INFO_ACCOUNT_DOMAIN,
193                                             &info,
194                                             &result);
195         if (!NT_STATUS_IS_OK(status)) {
196                 result = status;
197                 goto error;
198         }
199         if (!NT_STATUS_IS_OK(result)) {
200                 goto error;
201         }
202
203         got_domain_sid = True;
204         sid_copy(&domain_sid, info->account_domain.sid);
205
206         dcerpc_lsa_Close(b, mem_ctx, &pol, &result);
207         TALLOC_FREE(lsapipe);
208         talloc_destroy(mem_ctx);
209
210         return;
211
212  error:
213
214         if (lsapipe) {
215                 TALLOC_FREE(lsapipe);
216         }
217
218         fprintf(stderr, "could not obtain sid for domain %s\n", cli->domain);
219
220         if (!NT_STATUS_IS_OK(result)) {
221                 fprintf(stderr, "error: %s\n", nt_errstr(result));
222         }
223
224         exit(1);
225 }
226
227 /* List the available commands on a given pipe */
228
229 static NTSTATUS cmd_listcommands(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
230                                  int argc, const char **argv)
231 {
232         struct cmd_list *tmp;
233         struct cmd_set *tmp_set;
234         int i;
235
236         /* Usage */
237
238         if (argc != 2) {
239                 printf("Usage: %s <pipe>\n", argv[0]);
240                 return NT_STATUS_OK;
241         }
242
243         /* Help on one command */
244
245         for (tmp = cmd_list; tmp; tmp = tmp->next) 
246         {
247                 tmp_set = tmp->cmd_set;
248
249                 if (!strcasecmp_m(argv[1], tmp_set->name))
250                 {
251                         printf("Available commands on the %s pipe:\n\n", tmp_set->name);
252
253                         i = 0;
254                         tmp_set++;
255                         while(tmp_set->name) {
256                                 printf("%30s", tmp_set->name);
257                                 tmp_set++;
258                                 i++;
259                                 if (i%3 == 0)
260                                         printf("\n");
261                         }
262
263                         /* drop out of the loop */
264                         break;
265                 }
266         }
267         printf("\n\n");
268
269         return NT_STATUS_OK;
270 }
271
272 /* Display help on commands */
273
274 static NTSTATUS cmd_help(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
275                          int argc, const char **argv)
276 {
277         struct cmd_list *tmp;
278         struct cmd_set *tmp_set;
279
280         /* Usage */
281
282         if (argc > 2) {
283                 printf("Usage: %s [command]\n", argv[0]);
284                 return NT_STATUS_OK;
285         }
286
287         /* Help on one command */
288
289         if (argc == 2) {
290                 for (tmp = cmd_list; tmp; tmp = tmp->next) {
291
292                         tmp_set = tmp->cmd_set;
293
294                         while(tmp_set->name) {
295                                 if (strequal(argv[1], tmp_set->name)) {
296                                         if (tmp_set->usage &&
297                                             tmp_set->usage[0])
298                                                 printf("%s\n", tmp_set->usage);
299                                         else
300                                                 printf("No help for %s\n", tmp_set->name);
301
302                                         return NT_STATUS_OK;
303                                 }
304
305                                 tmp_set++;
306                         }
307                 }
308
309                 printf("No such command: %s\n", argv[1]);
310                 return NT_STATUS_OK;
311         }
312
313         /* List all commands */
314
315         for (tmp = cmd_list; tmp; tmp = tmp->next) {
316
317                 tmp_set = tmp->cmd_set;
318
319                 while(tmp_set->name) {
320
321                         printf("%15s\t\t%s\n", tmp_set->name,
322                                tmp_set->description ? tmp_set->description:
323                                "");
324
325                         tmp_set++;
326                 }
327         }
328
329         return NT_STATUS_OK;
330 }
331
332 /* Change the debug level */
333
334 static NTSTATUS cmd_debuglevel(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
335                                int argc, const char **argv)
336 {
337         if (argc > 2) {
338                 printf("Usage: %s [debuglevel]\n", argv[0]);
339                 return NT_STATUS_OK;
340         }
341
342         if (argc == 2) {
343                 lp_set_cmdline("log level", argv[1]);
344         }
345
346         printf("debuglevel is %d\n", DEBUGLEVEL);
347
348         return NT_STATUS_OK;
349 }
350
351 static NTSTATUS cmd_quit(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
352                          int argc, const char **argv)
353 {
354         exit(0);
355         return NT_STATUS_OK; /* NOTREACHED */
356 }
357
358 static NTSTATUS cmd_set_ss_level(void)
359 {
360         struct cmd_list *tmp;
361
362         /* Close any existing connections not at this level. */
363
364         for (tmp = cmd_list; tmp; tmp = tmp->next) {
365                 struct cmd_set *tmp_set;
366
367                 for (tmp_set = tmp->cmd_set; tmp_set->name; tmp_set++) {
368                         if (tmp_set->rpc_pipe == NULL) {
369                                 continue;
370                         }
371
372                         if ((tmp_set->rpc_pipe->auth->auth_type
373                              != pipe_default_auth_type)
374                             || (tmp_set->rpc_pipe->auth->auth_level
375                                 != pipe_default_auth_level)) {
376                                 TALLOC_FREE(tmp_set->rpc_pipe);
377                                 tmp_set->rpc_pipe = NULL;
378                         }
379                 }
380         }
381         return NT_STATUS_OK;
382 }
383
384 static NTSTATUS cmd_set_transport(void)
385 {
386         struct cmd_list *tmp;
387
388         /* Close any existing connections not at this level. */
389
390         for (tmp = cmd_list; tmp; tmp = tmp->next) {
391                 struct cmd_set *tmp_set;
392
393                 for (tmp_set = tmp->cmd_set; tmp_set->name; tmp_set++) {
394                         if (tmp_set->rpc_pipe == NULL) {
395                                 continue;
396                         }
397
398                         if (tmp_set->rpc_pipe->transport->transport != default_transport) {
399                                 TALLOC_FREE(tmp_set->rpc_pipe);
400                                 tmp_set->rpc_pipe = NULL;
401                         }
402                 }
403         }
404         return NT_STATUS_OK;
405 }
406
407 static NTSTATUS cmd_sign(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
408                          int argc, const char **argv)
409 {
410         const char *p = "[KRB5|KRB5_SPNEGO|NTLMSSP|NTLMSSP_SPNEGO|SCHANNEL]";
411         const char *type = "NTLMSSP";
412
413         pipe_default_auth_level = DCERPC_AUTH_LEVEL_INTEGRITY;
414         pipe_default_auth_type = DCERPC_AUTH_TYPE_NTLMSSP;
415
416         if (argc > 2) {
417                 printf("Usage: %s %s\n", argv[0], p);
418                 return NT_STATUS_OK;
419         }
420
421         if (argc == 2) {
422                 type = argv[1];
423                 if (strequal(type, "KRB5")) {
424                         pipe_default_auth_type = DCERPC_AUTH_TYPE_KRB5;
425                 } else if (strequal(type, "KRB5_SPNEGO")) {
426                         pipe_default_auth_type = DCERPC_AUTH_TYPE_SPNEGO;
427                         pipe_default_auth_spnego_type = PIPE_AUTH_TYPE_SPNEGO_KRB5;
428                 } else if (strequal(type, "NTLMSSP")) {
429                         pipe_default_auth_type = DCERPC_AUTH_TYPE_NTLMSSP;
430                 } else if (strequal(type, "NTLMSSP_SPNEGO")) {
431                         pipe_default_auth_type = DCERPC_AUTH_TYPE_SPNEGO;
432                         pipe_default_auth_spnego_type = PIPE_AUTH_TYPE_SPNEGO_NTLMSSP;
433                 } else if (strequal(type, "SCHANNEL")) {
434                         pipe_default_auth_type = DCERPC_AUTH_TYPE_SCHANNEL;
435                 } else {
436                         printf("unknown type %s\n", type);
437                         printf("Usage: %s %s\n", argv[0], p);
438                         return NT_STATUS_INVALID_LEVEL;
439                 }
440         }
441
442         d_printf("Setting %s - sign\n", type);
443
444         return cmd_set_ss_level();
445 }
446
447 static NTSTATUS cmd_seal(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
448                          int argc, const char **argv)
449 {
450         const char *p = "[KRB5|KRB5_SPNEGO|NTLMSSP|NTLMSSP_SPNEGO|SCHANNEL]";
451         const char *type = "NTLMSSP";
452
453         pipe_default_auth_level = DCERPC_AUTH_LEVEL_PRIVACY;
454         pipe_default_auth_type = DCERPC_AUTH_TYPE_NTLMSSP;
455
456         if (argc > 2) {
457                 printf("Usage: %s %s\n", argv[0], p);
458                 return NT_STATUS_OK;
459         }
460
461         if (argc == 2) {
462                 type = argv[1];
463                 if (strequal(type, "KRB5")) {
464                         pipe_default_auth_type = DCERPC_AUTH_TYPE_KRB5;
465                 } else if (strequal(type, "KRB5_SPNEGO")) {
466                         pipe_default_auth_type = DCERPC_AUTH_TYPE_SPNEGO;
467                         pipe_default_auth_spnego_type = PIPE_AUTH_TYPE_SPNEGO_KRB5;
468                 } else if (strequal(type, "NTLMSSP")) {
469                         pipe_default_auth_type = DCERPC_AUTH_TYPE_NTLMSSP;
470                 } else if (strequal(type, "NTLMSSP_SPNEGO")) {
471                         pipe_default_auth_type = DCERPC_AUTH_TYPE_SPNEGO;
472                         pipe_default_auth_spnego_type = PIPE_AUTH_TYPE_SPNEGO_NTLMSSP;
473                 } else if (strequal(type, "SCHANNEL")) {
474                         pipe_default_auth_type = DCERPC_AUTH_TYPE_SCHANNEL;
475                 } else {
476                         printf("unknown type %s\n", type);
477                         printf("Usage: %s %s\n", argv[0], p);
478                         return NT_STATUS_INVALID_LEVEL;
479                 }
480         }
481
482         d_printf("Setting %s - sign and seal\n", type);
483
484         return cmd_set_ss_level();
485 }
486
487 static NTSTATUS cmd_timeout(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
488                             int argc, const char **argv)
489 {
490         struct cmd_list *tmp;
491
492         if (argc > 2) {
493                 printf("Usage: %s timeout\n", argv[0]);
494                 return NT_STATUS_OK;
495         }
496
497         if (argc == 2) {
498                 timeout = atoi(argv[1]);
499
500                 for (tmp = cmd_list; tmp; tmp = tmp->next) {
501
502                         struct cmd_set *tmp_set;
503
504                         for (tmp_set = tmp->cmd_set; tmp_set->name; tmp_set++) {
505                                 if (tmp_set->rpc_pipe == NULL) {
506                                         continue;
507                                 }
508
509                                 rpccli_set_timeout(tmp_set->rpc_pipe, timeout);
510                         }
511                 }
512         }
513
514         printf("timeout is %d\n", timeout);
515
516         return NT_STATUS_OK;
517 }
518
519
520 static NTSTATUS cmd_none(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
521                          int argc, const char **argv)
522 {
523         pipe_default_auth_level = DCERPC_AUTH_LEVEL_NONE;
524         pipe_default_auth_type = DCERPC_AUTH_TYPE_NONE;
525         pipe_default_auth_spnego_type = PIPE_AUTH_TYPE_SPNEGO_NONE;
526
527         return cmd_set_ss_level();
528 }
529
530 static NTSTATUS cmd_schannel(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
531                              int argc, const char **argv)
532 {
533         d_printf("Setting schannel - sign and seal\n");
534         pipe_default_auth_level = DCERPC_AUTH_LEVEL_PRIVACY;
535         pipe_default_auth_type = DCERPC_AUTH_TYPE_SCHANNEL;
536
537         return cmd_set_ss_level();
538 }
539
540 static NTSTATUS cmd_schannel_sign(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
541                              int argc, const char **argv)
542 {
543         d_printf("Setting schannel - sign only\n");
544         pipe_default_auth_level = DCERPC_AUTH_LEVEL_INTEGRITY;
545         pipe_default_auth_type = DCERPC_AUTH_TYPE_SCHANNEL;
546
547         return cmd_set_ss_level();
548 }
549
550 static NTSTATUS cmd_choose_transport(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
551                                      int argc, const char **argv)
552 {
553         NTSTATUS status;
554
555         if (argc != 2) {
556                 printf("Usage: %s [NCACN_NP|NCACN_IP_TCP]\n", argv[0]);
557                 return NT_STATUS_OK;
558         }
559
560         if (strequal(argv[1], "NCACN_NP")) {
561                 default_transport = NCACN_NP;
562         } else if (strequal(argv[1], "NCACN_IP_TCP")) {
563                 default_transport = NCACN_IP_TCP;
564         } else {
565                 printf("transport type: %s unknown or not supported\n", argv[1]);
566                 return NT_STATUS_NOT_SUPPORTED;
567         }
568
569         status = cmd_set_transport();
570         if (!NT_STATUS_IS_OK(status)) {
571                 return status;
572         }
573
574         printf("default transport is now: %s\n", argv[1]);
575
576         return NT_STATUS_OK;
577 }
578
579 /* Built in rpcclient commands */
580
581 static struct cmd_set rpcclient_commands[] = {
582
583         { "GENERAL OPTIONS" },
584
585         { "help", RPC_RTYPE_NTSTATUS, cmd_help, NULL,     NULL, NULL,   "Get help on commands", "[command]" },
586         { "?",  RPC_RTYPE_NTSTATUS, cmd_help, NULL,       NULL, NULL,   "Get help on commands", "[command]" },
587         { "debuglevel", RPC_RTYPE_NTSTATUS, cmd_debuglevel, NULL,   NULL,       NULL, "Set debug level", "level" },
588         { "debug", RPC_RTYPE_NTSTATUS, cmd_debuglevel, NULL,   NULL,    NULL, "Set debug level", "level" },
589         { "list",       RPC_RTYPE_NTSTATUS, cmd_listcommands, NULL, NULL,       NULL, "List available commands on <pipe>", "pipe" },
590         { "exit", RPC_RTYPE_NTSTATUS, cmd_quit, NULL,   NULL,   NULL,   "Exit program", "" },
591         { "quit", RPC_RTYPE_NTSTATUS, cmd_quit, NULL,     NULL, NULL, "Exit program", "" },
592         { "sign", RPC_RTYPE_NTSTATUS, cmd_sign, NULL,     NULL, NULL, "Force RPC pipe connections to be signed", "" },
593         { "seal", RPC_RTYPE_NTSTATUS, cmd_seal, NULL,     NULL, NULL, "Force RPC pipe connections to be sealed", "" },
594         { "schannel", RPC_RTYPE_NTSTATUS, cmd_schannel, NULL,     NULL, NULL,   "Force RPC pipe connections to be sealed with 'schannel'.  Assumes valid machine account to this domain controller.", "" },
595         { "schannelsign", RPC_RTYPE_NTSTATUS, cmd_schannel_sign, NULL,    NULL, NULL, "Force RPC pipe connections to be signed (not sealed) with 'schannel'.  Assumes valid machine account to this domain controller.", "" },
596         { "timeout", RPC_RTYPE_NTSTATUS, cmd_timeout, NULL,       NULL, NULL, "Set timeout (in milliseconds) for RPC operations", "" },
597         { "transport", RPC_RTYPE_NTSTATUS, cmd_choose_transport, NULL,    NULL, NULL, "Choose ncacn transport for RPC operations", "" },
598         { "none", RPC_RTYPE_NTSTATUS, cmd_none, NULL,     NULL, NULL, "Force RPC pipe connections to have no special properties", "" },
599
600         { NULL }
601 };
602
603 static struct cmd_set separator_command[] = {
604         { "---------------", MAX_RPC_RETURN_TYPE, NULL, NULL,   NULL, NULL, "----------------------" },
605         { NULL }
606 };
607
608
609 /* Various pipe commands */
610
611 extern struct cmd_set lsarpc_commands[];
612 extern struct cmd_set samr_commands[];
613 extern struct cmd_set spoolss_commands[];
614 extern struct cmd_set netlogon_commands[];
615 extern struct cmd_set srvsvc_commands[];
616 extern struct cmd_set dfs_commands[];
617 extern struct cmd_set ds_commands[];
618 extern struct cmd_set echo_commands[];
619 extern struct cmd_set epmapper_commands[];
620 extern struct cmd_set shutdown_commands[];
621 extern struct cmd_set test_commands[];
622 extern struct cmd_set wkssvc_commands[];
623 extern struct cmd_set ntsvcs_commands[];
624 extern struct cmd_set drsuapi_commands[];
625 extern struct cmd_set eventlog_commands[];
626 extern struct cmd_set winreg_commands[];
627 extern struct cmd_set fss_commands[];
628
629 static struct cmd_set *rpcclient_command_list[] = {
630         rpcclient_commands,
631         lsarpc_commands,
632         ds_commands,
633         samr_commands,
634         spoolss_commands,
635         netlogon_commands,
636         srvsvc_commands,
637         dfs_commands,
638         echo_commands,
639         epmapper_commands,
640         shutdown_commands,
641         test_commands,
642         wkssvc_commands,
643         ntsvcs_commands,
644         drsuapi_commands,
645         eventlog_commands,
646         winreg_commands,
647         fss_commands,
648         NULL
649 };
650
651 static void add_command_set(struct cmd_set *cmd_set)
652 {
653         struct cmd_list *entry;
654
655         if (!(entry = SMB_MALLOC_P(struct cmd_list))) {
656                 DEBUG(0, ("out of memory\n"));
657                 return;
658         }
659
660         ZERO_STRUCTP(entry);
661
662         entry->cmd_set = cmd_set;
663         DLIST_ADD(cmd_list, entry);
664 }
665
666
667 /**
668  * Call an rpcclient function, passing an argv array.
669  *
670  * @param cmd Command to run, as a single string.
671  **/
672 static NTSTATUS do_cmd(struct cli_state *cli,
673                        struct user_auth_info *auth_info,
674                        struct cmd_set *cmd_entry,
675                        struct dcerpc_binding *binding,
676                        int argc, const char **argv)
677 {
678         NTSTATUS ntresult;
679         WERROR wresult;
680         bool ok;
681
682         TALLOC_CTX *mem_ctx;
683
684         /* Create mem_ctx */
685
686         if (!(mem_ctx = talloc_stackframe())) {
687                 DEBUG(0, ("talloc_init() failed\n"));
688                 return NT_STATUS_NO_MEMORY;
689         }
690
691         /* Open pipe */
692
693         if ((cmd_entry->table != NULL) && (cmd_entry->rpc_pipe == NULL)) {
694                 switch (pipe_default_auth_type) {
695                 case DCERPC_AUTH_TYPE_NONE:
696                         ntresult = cli_rpc_pipe_open_noauth_transport(
697                                 cli, default_transport,
698                                 cmd_entry->table,
699                                 &cmd_entry->rpc_pipe);
700                         break;
701                 case DCERPC_AUTH_TYPE_SPNEGO:
702                 {
703                         /* won't happen, but if it does it will fail in cli_rpc_pipe_open_spnego() eventually */
704                         const char *oid = "INVALID";
705                         switch (pipe_default_auth_spnego_type) {
706                         case PIPE_AUTH_TYPE_SPNEGO_NTLMSSP:
707                                 oid = GENSEC_OID_NTLMSSP;
708                                 break;
709                         case PIPE_AUTH_TYPE_SPNEGO_KRB5:
710                                 oid = GENSEC_OID_KERBEROS5;
711                                 break;
712                         case PIPE_AUTH_TYPE_SPNEGO_NONE:
713                                 break;
714                         }
715                         ntresult = cli_rpc_pipe_open_spnego(
716                                 cli, cmd_entry->table,
717                                 default_transport,
718                                 oid,
719                                 pipe_default_auth_level,
720                                 smbXcli_conn_remote_name(cli->conn),
721                                 get_cmdline_auth_info_domain(auth_info),
722                                 get_cmdline_auth_info_username(auth_info),
723                                 get_cmdline_auth_info_password(auth_info),
724                                 &cmd_entry->rpc_pipe);
725                         break;
726                 }
727                 case DCERPC_AUTH_TYPE_NTLMSSP:
728                 case DCERPC_AUTH_TYPE_KRB5:
729                         ntresult = cli_rpc_pipe_open_generic_auth(
730                                 cli, cmd_entry->table,
731                                 default_transport,
732                                 pipe_default_auth_type,
733                                 pipe_default_auth_level,
734                                 smbXcli_conn_remote_name(cli->conn),
735                                 get_cmdline_auth_info_domain(auth_info),
736                                 get_cmdline_auth_info_username(auth_info),
737                                 get_cmdline_auth_info_password(auth_info),
738                                 &cmd_entry->rpc_pipe);
739                         break;
740                 case DCERPC_AUTH_TYPE_SCHANNEL:
741                         TALLOC_FREE(rpcclient_netlogon_creds);
742                         ntresult = cli_rpc_pipe_open_schannel(
743                                 cli, rpcclient_msg_ctx,
744                                 cmd_entry->table,
745                                 default_transport,
746                                 pipe_default_auth_level,
747                                 get_cmdline_auth_info_domain(auth_info),
748                                 &cmd_entry->rpc_pipe,
749                                 talloc_autofree_context(),
750                                 &rpcclient_netlogon_creds);
751                         break;
752                 default:
753                         DEBUG(0, ("Could not initialise %s. Invalid "
754                                   "auth type %u\n",
755                                   cmd_entry->table->name,
756                                   pipe_default_auth_type ));
757                         talloc_free(mem_ctx);
758                         return NT_STATUS_UNSUCCESSFUL;
759                 }
760                 if (!NT_STATUS_IS_OK(ntresult)) {
761                         DEBUG(0, ("Could not initialise %s. Error was %s\n",
762                                   cmd_entry->table->name,
763                                   nt_errstr(ntresult) ));
764                         talloc_free(mem_ctx);
765                         return ntresult;
766                 }
767
768                 ok = ndr_syntax_id_equal(&cmd_entry->table->syntax_id,
769                                          &ndr_table_netlogon.syntax_id);
770                 if (rpcclient_netlogon_creds == NULL && ok) {
771                         const char *dc_name = cmd_entry->rpc_pipe->desthost;
772                         const char *domain = get_cmdline_auth_info_domain(auth_info);
773                         enum netr_SchannelType sec_chan_type = 0;
774                         const char *_account_name = NULL;
775                         const char *account_name = NULL;
776                         struct samr_Password current_nt_hash;
777                         struct samr_Password *previous_nt_hash = NULL;
778
779                         if (!get_trust_pw_hash(get_cmdline_auth_info_domain(auth_info),
780                                                current_nt_hash.hash, &_account_name,
781                                                &sec_chan_type))
782                         {
783                                 DEBUG(0, ("Failed to fetch trust password for %s to connect to %s.\n",
784                                           get_cmdline_auth_info_domain(auth_info),
785                                           cmd_entry->table->name));
786                                 TALLOC_FREE(cmd_entry->rpc_pipe);
787                                 talloc_free(mem_ctx);
788                                 return NT_STATUS_CANT_ACCESS_DOMAIN_INFO;
789                         }
790
791                         account_name = talloc_asprintf(mem_ctx, "%s$", _account_name);
792                         if (account_name == NULL) {
793                                 DEBUG(0, ("Out of memory creating account name to connect to %s.\n",
794                                           cmd_entry->table->name));
795                                 TALLOC_FREE(cmd_entry->rpc_pipe);
796                                 SAFE_FREE(previous_nt_hash);
797                                 TALLOC_FREE(mem_ctx);
798                                 return NT_STATUS_NO_MEMORY;
799                         }
800
801                         ntresult = rpccli_create_netlogon_creds(dc_name,
802                                                 domain,
803                                                 account_name,
804                                                 sec_chan_type,
805                                                 rpcclient_msg_ctx,
806                                                 talloc_autofree_context(),
807                                                 &rpcclient_netlogon_creds);
808                         if (!NT_STATUS_IS_OK(ntresult)) {
809                                 DEBUG(0, ("Could not initialise credentials for %s.\n",
810                                           cmd_entry->table->name));
811                                 TALLOC_FREE(cmd_entry->rpc_pipe);
812                                 SAFE_FREE(previous_nt_hash);
813                                 TALLOC_FREE(mem_ctx);
814                                 return ntresult;
815                         }
816
817                         ntresult = rpccli_setup_netlogon_creds(cli,
818                                                         rpcclient_netlogon_creds,
819                                                         false, /* force_reauth */
820                                                         current_nt_hash,
821                                                         previous_nt_hash);
822                         SAFE_FREE(previous_nt_hash);
823                         if (!NT_STATUS_IS_OK(ntresult)) {
824                                 DEBUG(0, ("Could not initialise credentials for %s.\n",
825                                           cmd_entry->table->name));
826                                 TALLOC_FREE(cmd_entry->rpc_pipe);
827                                 TALLOC_FREE(rpcclient_netlogon_creds);
828                                 TALLOC_FREE(mem_ctx);
829                                 return ntresult;
830                         }
831                 }
832         }
833
834         /* Run command */
835
836         if ( cmd_entry->returntype == RPC_RTYPE_NTSTATUS ) {
837                 ntresult = cmd_entry->ntfn(cmd_entry->rpc_pipe, mem_ctx, argc, argv);
838                 if (!NT_STATUS_IS_OK(ntresult)) {
839                         printf("result was %s\n", nt_errstr(ntresult));
840                 }
841         } else {
842                 wresult = cmd_entry->wfn(cmd_entry->rpc_pipe, mem_ctx, argc, argv);
843                 /* print out the DOS error */
844                 if (!W_ERROR_IS_OK(wresult)) {
845                         printf( "result was %s\n", win_errstr(wresult));
846                 }
847                 ntresult = W_ERROR_IS_OK(wresult)?NT_STATUS_OK:NT_STATUS_UNSUCCESSFUL;
848         }
849
850         /* Cleanup */
851
852         talloc_free(mem_ctx);
853
854         return ntresult;
855 }
856
857
858 /**
859  * Process a command entered at the prompt or as part of -c
860  *
861  * @returns The NTSTATUS from running the command.
862  **/
863 static NTSTATUS process_cmd(struct user_auth_info *auth_info,
864                             struct cli_state *cli,
865                             struct dcerpc_binding *binding,
866                             char *cmd)
867 {
868         struct cmd_list *temp_list;
869         NTSTATUS result = NT_STATUS_OK;
870         int ret;
871         int argc;
872         const char **argv = NULL;
873
874         if ((ret = poptParseArgvString(cmd, &argc, &argv)) != 0) {
875                 fprintf(stderr, "rpcclient: %s\n", poptStrerror(ret));
876                 return NT_STATUS_UNSUCCESSFUL;
877         }
878
879
880         /* Walk through a dlist of arrays of commands. */
881         for (temp_list = cmd_list; temp_list; temp_list = temp_list->next) {
882                 struct cmd_set *temp_set = temp_list->cmd_set;
883
884                 while (temp_set->name) {
885                         if (strequal(argv[0], temp_set->name)) {
886                                 if (!(temp_set->returntype == RPC_RTYPE_NTSTATUS && temp_set->ntfn ) &&
887                          !(temp_set->returntype == RPC_RTYPE_WERROR && temp_set->wfn )) {
888                                         fprintf (stderr, "Invalid command\n");
889                                         goto out_free;
890                                 }
891
892                                 result = do_cmd(cli, auth_info, temp_set,
893                                                 binding, argc, argv);
894
895                                 goto out_free;
896                         }
897                         temp_set++;
898                 }
899         }
900
901         if (argv[0]) {
902                 printf("command not found: %s\n", argv[0]);
903         }
904
905 out_free:
906 /* moved to do_cmd()
907         if (!NT_STATUS_IS_OK(result)) {
908                 printf("result was %s\n", nt_errstr(result));
909         }
910 */
911
912         /* NOTE: popt allocates the whole argv, including the
913          * strings, as a single block.  So a single free is
914          * enough to release it -- we don't free the
915          * individual strings.  rtfm. */
916         free(argv);
917
918         return result;
919 }
920
921
922 /* Main function */
923
924  int main(int argc, char *argv[])
925 {
926         const char **const_argv = discard_const_p(const char *, argv);
927         int                     opt;
928         static char             *cmdstr = NULL;
929         const char *server;
930         struct cli_state        *cli = NULL;
931         static char             *opt_ipaddr=NULL;
932         struct cmd_set          **cmd_set;
933         struct sockaddr_storage server_ss;
934         NTSTATUS                nt_status;
935         static int              opt_port = 0;
936         int result = 0;
937         TALLOC_CTX *frame = talloc_stackframe();
938         uint32_t flags = 0;
939         struct dcerpc_binding *binding = NULL;
940         const char *binding_string = NULL;
941         char *user, *domain, *q;
942
943         /* make sure the vars that get altered (4th field) are in
944            a fixed location or certain compilers complain */
945         poptContext pc;
946         struct poptOption long_options[] = {
947                 POPT_AUTOHELP
948                 {"command",     'c', POPT_ARG_STRING,   &cmdstr, 'c', "Execute semicolon separated cmds", "COMMANDS"},
949                 {"dest-ip", 'I', POPT_ARG_STRING,   &opt_ipaddr, 'I', "Specify destination IP address", "IP"},
950                 {"port", 'p', POPT_ARG_INT,   &opt_port, 'p', "Specify port number", "PORT"},
951                 POPT_COMMON_SAMBA
952                 POPT_COMMON_CONNECTION
953                 POPT_COMMON_CREDENTIALS
954                 POPT_TABLEEND
955         };
956
957         load_case_tables();
958
959         zero_sockaddr(&server_ss);
960
961         setlinebuf(stdout);
962
963         /* the following functions are part of the Samba debugging
964            facilities.  See lib/debug.c */
965         setup_logging("rpcclient", DEBUG_STDOUT);
966
967         rpcclient_auth_info = user_auth_info_init(frame);
968         if (rpcclient_auth_info == NULL) {
969                 exit(1);
970         }
971         popt_common_set_auth_info(rpcclient_auth_info);
972
973         /* Parse options */
974
975         pc = poptGetContext("rpcclient", argc, const_argv,
976                             long_options, 0);
977
978         if (argc == 1) {
979                 poptPrintHelp(pc, stderr, 0);
980                 goto done;
981         }
982
983         while((opt = poptGetNextOpt(pc)) != -1) {
984                 switch (opt) {
985
986                 case 'I':
987                         if (!interpret_string_addr(&server_ss,
988                                                 opt_ipaddr,
989                                                 AI_NUMERICHOST)) {
990                                 fprintf(stderr, "%s not a valid IP address\n",
991                                         opt_ipaddr);
992                                 result = 1;
993                                 goto done;
994                         }
995                 }
996         }
997
998         /* Get server as remaining unparsed argument.  Print usage if more
999            than one unparsed argument is present. */
1000
1001         server = poptGetArg(pc);
1002
1003         if (!server || poptGetArg(pc)) {
1004                 poptPrintHelp(pc, stderr, 0);
1005                 result = 1;
1006                 goto done;
1007         }
1008
1009         poptFreeContext(pc);
1010         popt_burn_cmdline_password(argc, argv);
1011
1012         if (!init_names()) {
1013                 result = 1;
1014                 goto done;
1015         }
1016
1017         /* Load smb.conf file */
1018
1019         if (!lp_load_global(get_dyn_CONFIGFILE()))
1020                 fprintf(stderr, "Can't load %s\n", get_dyn_CONFIGFILE());
1021
1022         /* We must load interfaces after we load the smb.conf */
1023         load_interfaces();
1024
1025         rpcclient_msg_ctx = messaging_init(talloc_autofree_context(),
1026                         samba_tevent_context_init(talloc_autofree_context()));
1027
1028         /*
1029          * Get password
1030          * from stdin if necessary
1031          */
1032
1033         if (get_cmdline_auth_info_use_machine_account(rpcclient_auth_info) &&
1034             !set_cmdline_auth_info_machine_account_creds(rpcclient_auth_info)) {
1035                 result = 1;
1036                 goto done;
1037         }
1038
1039         set_cmdline_auth_info_getpass(rpcclient_auth_info);
1040
1041         if ((server[0] == '/' && server[1] == '/') ||
1042                         (server[0] == '\\' && server[1] ==  '\\')) {
1043                 server += 2;
1044         }
1045
1046         nt_status = dcerpc_parse_binding(frame, server, &binding);
1047
1048         if (!NT_STATUS_IS_OK(nt_status)) {
1049
1050                 binding_string = talloc_asprintf(frame, "ncacn_np:%s",
1051                                                  strip_hostname(server));
1052                 if (!binding_string) {
1053                         result = 1;
1054                         goto done;
1055                 }
1056
1057                 nt_status = dcerpc_parse_binding(frame, binding_string, &binding);
1058                 if (!NT_STATUS_IS_OK(nt_status)) {
1059                         result = -1;
1060                         goto done;
1061                 }
1062         }
1063
1064         if (binding->transport == NCA_UNKNOWN) {
1065                 binding->transport = NCACN_NP;
1066         }
1067
1068         if (binding->flags & DCERPC_CONNECT) {
1069                 pipe_default_auth_level = DCERPC_AUTH_LEVEL_CONNECT;
1070                 pipe_default_auth_type = DCERPC_AUTH_TYPE_NTLMSSP;
1071         }
1072         if (binding->flags & DCERPC_SIGN) {
1073                 pipe_default_auth_level = DCERPC_AUTH_LEVEL_INTEGRITY;
1074                 pipe_default_auth_type = DCERPC_AUTH_TYPE_NTLMSSP;
1075         }
1076         if (binding->flags & DCERPC_SEAL) {
1077                 pipe_default_auth_level = DCERPC_AUTH_LEVEL_PRIVACY;
1078                 pipe_default_auth_type = DCERPC_AUTH_TYPE_NTLMSSP;
1079         }
1080         if (binding->flags & DCERPC_AUTH_SPNEGO) {
1081                 pipe_default_auth_type = DCERPC_AUTH_TYPE_SPNEGO;
1082                 pipe_default_auth_spnego_type = PIPE_AUTH_TYPE_SPNEGO_NTLMSSP;
1083         }
1084         if (binding->flags & DCERPC_AUTH_NTLM) {
1085                 if (pipe_default_auth_type == DCERPC_AUTH_TYPE_SPNEGO) {
1086                         pipe_default_auth_spnego_type = PIPE_AUTH_TYPE_SPNEGO_NTLMSSP;
1087                 } else {
1088                         pipe_default_auth_type = DCERPC_AUTH_TYPE_NTLMSSP;
1089                 }
1090         }
1091         if (binding->flags & DCERPC_AUTH_KRB5) {
1092                 if (pipe_default_auth_type == DCERPC_AUTH_TYPE_SPNEGO) {
1093                         pipe_default_auth_spnego_type = PIPE_AUTH_TYPE_SPNEGO_KRB5;
1094                 } else {
1095                         pipe_default_auth_type = DCERPC_AUTH_TYPE_KRB5;
1096                 }
1097         }
1098         if (pipe_default_auth_type != DCERPC_AUTH_TYPE_NONE) {
1099                 /* If neither Integrity or Privacy are requested then
1100                  * Use just Connect level */
1101                 if (pipe_default_auth_level == DCERPC_AUTH_LEVEL_NONE) {
1102                         pipe_default_auth_level = DCERPC_AUTH_LEVEL_CONNECT;
1103                 }
1104         }
1105
1106         if (get_cmdline_auth_info_use_kerberos(rpcclient_auth_info)) {
1107                 flags |= CLI_FULL_CONNECTION_USE_KERBEROS |
1108                          CLI_FULL_CONNECTION_FALLBACK_AFTER_KERBEROS;
1109         }
1110         if (get_cmdline_auth_info_use_ccache(rpcclient_auth_info)) {
1111                 flags |= CLI_FULL_CONNECTION_USE_CCACHE;
1112         }
1113
1114         user = talloc_strdup(frame, get_cmdline_auth_info_username(rpcclient_auth_info));
1115         SMB_ASSERT(user != NULL);
1116         domain = talloc_strdup(frame, lp_workgroup());
1117         SMB_ASSERT(domain != NULL);
1118         set_cmdline_auth_info_domain(rpcclient_auth_info, domain);
1119
1120         if ((q = strchr_m(user,'\\'))) {
1121                 *q = 0;
1122                 set_cmdline_auth_info_domain(rpcclient_auth_info, user);
1123                 set_cmdline_auth_info_username(rpcclient_auth_info, q+1);
1124         }
1125
1126
1127         nt_status = cli_full_connection(&cli, lp_netbios_name(), binding->host,
1128                                         opt_ipaddr ? &server_ss : NULL, opt_port,
1129                                         "IPC$", "IPC",
1130                                         get_cmdline_auth_info_username(rpcclient_auth_info),
1131                                         get_cmdline_auth_info_domain(rpcclient_auth_info),
1132                                         get_cmdline_auth_info_password(rpcclient_auth_info),
1133                                         flags,
1134                                         get_cmdline_auth_info_signing_state(rpcclient_auth_info));
1135
1136         if (!NT_STATUS_IS_OK(nt_status)) {
1137                 DEBUG(0,("Cannot connect to server.  Error was %s\n", nt_errstr(nt_status)));
1138                 result = 1;
1139                 goto done;
1140         }
1141
1142         if (get_cmdline_auth_info_smb_encrypt(rpcclient_auth_info)) {
1143                 nt_status = cli_cm_force_encryption(cli,
1144                                         get_cmdline_auth_info_username(rpcclient_auth_info),
1145                                         get_cmdline_auth_info_password(rpcclient_auth_info),
1146                                         get_cmdline_auth_info_domain(rpcclient_auth_info),
1147                                         "IPC$");
1148                 if (!NT_STATUS_IS_OK(nt_status)) {
1149                         result = 1;
1150                         goto done;
1151                 }
1152         }
1153
1154 #if 0   /* COMMENT OUT FOR TESTING */
1155         memset(cmdline_auth_info.password,'X',sizeof(cmdline_auth_info.password));
1156 #endif
1157
1158         /* Load command lists */
1159         rpcclient_cli_state = cli;
1160         timeout = cli_set_timeout(cli, 10000);
1161
1162         cmd_set = rpcclient_command_list;
1163
1164         while(*cmd_set) {
1165                 add_command_set(*cmd_set);
1166                 add_command_set(separator_command);
1167                 cmd_set++;
1168         }
1169
1170         default_transport = binding->transport;
1171
1172         fetch_machine_sid(cli);
1173
1174        /* Do anything specified with -c */
1175         if (cmdstr && cmdstr[0]) {
1176                 char    *cmd;
1177                 char    *p = cmdstr;
1178
1179                 result = 0;
1180
1181                 while((cmd=next_command(&p)) != NULL) {
1182                         NTSTATUS cmd_result = process_cmd(rpcclient_auth_info, cli,
1183                                                           binding, cmd);
1184                         SAFE_FREE(cmd);
1185                         result = NT_STATUS_IS_ERR(cmd_result);
1186                 }
1187
1188                 goto done;
1189         }
1190
1191         /* Loop around accepting commands */
1192
1193         while(1) {
1194                 char *line = NULL;
1195
1196                 line = smb_readline("rpcclient $> ", NULL, completion_fn);
1197
1198                 if (line == NULL)
1199                         break;
1200
1201                 if (line[0] != '\n')
1202                         process_cmd(rpcclient_auth_info, cli, binding, line);
1203                 SAFE_FREE(line);
1204         }
1205
1206 done:
1207         rpcclient_cli_state = NULL;
1208         if (cli != NULL) {
1209                 cli_shutdown(cli);
1210         }
1211         TALLOC_FREE(frame);
1212         return result;
1213 }