2 Unix SMB/CIFS implementation.
5 Copyright (C) Tim Potter 2000-2001
6 Copyright (C) Martin Pool 2003
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 2 of the License, or
11 (at your option) any later version.
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.
18 You should have received a copy of the GNU General Public License
19 along with this program; if not, write to the Free Software
20 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
24 #include "rpcclient.h"
28 static enum pipe_auth_type pipe_default_auth_type = PIPE_AUTH_TYPE_NONE;
29 static enum pipe_auth_level pipe_default_auth_level = PIPE_AUTH_LEVEL_NONE;
31 /* List to hold groups of commands.
33 * Commands are defined in a list of arrays: arrays are easy to
34 * statically declare, and lists are easier to dynamically extend.
37 static struct cmd_list {
38 struct cmd_list *prev, *next;
39 struct cmd_set *cmd_set;
42 /****************************************************************************
43 handle completion of commands for readline
44 ****************************************************************************/
45 static char **completion_fn(const char *text, int start, int end)
47 #define MAX_COMPLETIONS 100
50 struct cmd_list *commands = cmd_list;
53 /* FIXME!!! -- what to do when completing argument? */
54 /* for words not at the start of the line fallback
55 to filename completion */
60 /* make sure we have a list of valid commands */
64 matches = SMB_MALLOC_ARRAY(char *, MAX_COMPLETIONS);
65 if (!matches) return NULL;
67 matches[count++] = SMB_STRDUP(text);
68 if (!matches[0]) return NULL;
70 while (commands && count < MAX_COMPLETIONS-1)
72 if (!commands->cmd_set)
75 for (i=0; commands->cmd_set[i].name; i++)
77 if ((strncmp(text, commands->cmd_set[i].name, strlen(text)) == 0) &&
78 (( commands->cmd_set[i].returntype == RPC_RTYPE_NTSTATUS &&
79 commands->cmd_set[i].ntfn ) ||
80 ( commands->cmd_set[i].returntype == RPC_RTYPE_WERROR &&
81 commands->cmd_set[i].wfn)))
83 matches[count] = SMB_STRDUP(commands->cmd_set[i].name);
90 commands = commands->next;
95 SAFE_FREE(matches[0]);
96 matches[0] = SMB_STRDUP(matches[1]);
98 matches[count] = NULL;
102 static char* next_command (char** cmdstr)
104 static pstring command;
107 if (!cmdstr || !(*cmdstr))
110 p = strchr_m(*cmdstr, ';');
113 pstrcpy(command, *cmdstr);
122 /* Fetch the SID for this computer */
124 static void fetch_machine_sid(struct cli_state *cli)
127 NTSTATUS result = NT_STATUS_OK;
128 uint32 info_class = 5;
129 char *domain_name = NULL;
130 static BOOL got_domain_sid;
132 DOM_SID *dom_sid = NULL;
133 struct rpc_pipe_client *lsapipe = NULL;
135 if (got_domain_sid) return;
137 if (!(mem_ctx=talloc_init("fetch_machine_sid"))) {
138 DEBUG(0,("fetch_machine_sid: talloc_init returned NULL!\n"));
142 if ((lsapipe = cli_rpc_pipe_open_noauth(cli, PI_LSARPC, &result)) == NULL) {
143 fprintf(stderr, "could not initialise lsa pipe. Error was %s\n", nt_errstr(result) );
147 result = rpccli_lsa_open_policy(lsapipe, mem_ctx, True,
148 SEC_RIGHTS_MAXIMUM_ALLOWED,
150 if (!NT_STATUS_IS_OK(result)) {
154 result = rpccli_lsa_query_info_policy(lsapipe, mem_ctx, &pol, info_class,
155 &domain_name, &dom_sid);
156 if (!NT_STATUS_IS_OK(result)) {
160 got_domain_sid = True;
161 sid_copy( &domain_sid, dom_sid );
163 rpccli_lsa_close(lsapipe, mem_ctx, &pol);
164 cli_rpc_pipe_close(lsapipe);
165 talloc_destroy(mem_ctx);
172 cli_rpc_pipe_close(lsapipe);
175 fprintf(stderr, "could not obtain sid for domain %s\n", cli->domain);
177 if (!NT_STATUS_IS_OK(result)) {
178 fprintf(stderr, "error: %s\n", nt_errstr(result));
184 /* List the available commands on a given pipe */
186 static NTSTATUS cmd_listcommands(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
187 int argc, const char **argv)
189 struct cmd_list *tmp;
190 struct cmd_set *tmp_set;
196 printf("Usage: %s <pipe>\n", argv[0]);
200 /* Help on one command */
202 for (tmp = cmd_list; tmp; tmp = tmp->next)
204 tmp_set = tmp->cmd_set;
206 if (!StrCaseCmp(argv[1], tmp_set->name))
208 printf("Available commands on the %s pipe:\n\n", tmp_set->name);
212 while(tmp_set->name) {
213 printf("%30s", tmp_set->name);
220 /* drop out of the loop */
229 /* Display help on commands */
231 static NTSTATUS cmd_help(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
232 int argc, const char **argv)
234 struct cmd_list *tmp;
235 struct cmd_set *tmp_set;
240 printf("Usage: %s [command]\n", argv[0]);
244 /* Help on one command */
247 for (tmp = cmd_list; tmp; tmp = tmp->next) {
249 tmp_set = tmp->cmd_set;
251 while(tmp_set->name) {
252 if (strequal(argv[1], tmp_set->name)) {
253 if (tmp_set->usage &&
255 printf("%s\n", tmp_set->usage);
257 printf("No help for %s\n", tmp_set->name);
266 printf("No such command: %s\n", argv[1]);
270 /* List all commands */
272 for (tmp = cmd_list; tmp; tmp = tmp->next) {
274 tmp_set = tmp->cmd_set;
276 while(tmp_set->name) {
278 printf("%15s\t\t%s\n", tmp_set->name,
279 tmp_set->description ? tmp_set->description:
289 /* Change the debug level */
291 static NTSTATUS cmd_debuglevel(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
292 int argc, const char **argv)
295 printf("Usage: %s [debuglevel]\n", argv[0]);
300 DEBUGLEVEL = atoi(argv[1]);
303 printf("debuglevel is %d\n", DEBUGLEVEL);
308 static NTSTATUS cmd_quit(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
309 int argc, const char **argv)
312 return NT_STATUS_OK; /* NOTREACHED */
315 static NTSTATUS cmd_set_ss_level(void)
317 struct cmd_list *tmp;
319 /* Close any existing connections not at this level. */
321 for (tmp = cmd_list; tmp; tmp = tmp->next) {
322 struct cmd_set *tmp_set;
324 for (tmp_set = tmp->cmd_set; tmp_set->name; tmp_set++) {
325 if (tmp_set->rpc_pipe == NULL) {
329 if (tmp_set->rpc_pipe->auth.auth_type != pipe_default_auth_type ||
330 tmp_set->rpc_pipe->auth.auth_level != pipe_default_auth_level) {
331 cli_rpc_pipe_close(tmp_set->rpc_pipe);
332 tmp_set->rpc_pipe = NULL;
339 static NTSTATUS cmd_sign(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
340 int argc, const char **argv)
342 pipe_default_auth_level = PIPE_AUTH_LEVEL_INTEGRITY;
343 pipe_default_auth_type = PIPE_AUTH_TYPE_NTLMSSP;
346 printf("Usage: %s [NTLMSSP|NTLMSSP_SPNEGO|SCHANNEL]\n", argv[0]);
351 if (strequal(argv[1], "NTLMSSP")) {
352 pipe_default_auth_type = PIPE_AUTH_TYPE_NTLMSSP;
353 } else if (strequal(argv[1], "NTLMSSP_SPNEGO")) {
354 pipe_default_auth_type = PIPE_AUTH_TYPE_SPNEGO_NTLMSSP;
355 } else if (strequal(argv[1], "SCHANNEL")) {
356 pipe_default_auth_type = PIPE_AUTH_TYPE_SCHANNEL;
358 printf("unknown type %s\n", argv[1]);
359 return NT_STATUS_INVALID_LEVEL;
363 printf("debuglevel is %d\n", DEBUGLEVEL);
364 return cmd_set_ss_level();
367 static NTSTATUS cmd_seal(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
368 int argc, const char **argv)
370 pipe_default_auth_level = PIPE_AUTH_LEVEL_PRIVACY;
371 pipe_default_auth_type = PIPE_AUTH_TYPE_NTLMSSP;
374 printf("Usage: %s [NTLMSSP|NTLMSSP_SPNEGO|SCHANNEL]\n", argv[0]);
379 if (strequal(argv[1], "NTLMSSP")) {
380 pipe_default_auth_type = PIPE_AUTH_TYPE_NTLMSSP;
381 } else if (strequal(argv[1], "NTLMSSP_SPNEGO")) {
382 pipe_default_auth_type = PIPE_AUTH_TYPE_SPNEGO_NTLMSSP;
383 } else if (strequal(argv[1], "SCHANNEL")) {
384 pipe_default_auth_type = PIPE_AUTH_TYPE_SCHANNEL;
386 printf("unknown type %s\n", argv[1]);
387 return NT_STATUS_INVALID_LEVEL;
390 return cmd_set_ss_level();
393 static NTSTATUS cmd_none(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
394 int argc, const char **argv)
396 pipe_default_auth_level = PIPE_AUTH_LEVEL_NONE;
397 pipe_default_auth_type = PIPE_AUTH_TYPE_NONE;
399 return cmd_set_ss_level();
402 static NTSTATUS cmd_schannel(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
403 int argc, const char **argv)
405 d_printf("Setting schannel - sign and seal\n");
406 pipe_default_auth_level = PIPE_AUTH_LEVEL_PRIVACY;
407 pipe_default_auth_type = PIPE_AUTH_TYPE_SCHANNEL;
409 return cmd_set_ss_level();
412 static NTSTATUS cmd_schannel_sign(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
413 int argc, const char **argv)
415 d_printf("Setting schannel - sign only\n");
416 pipe_default_auth_level = PIPE_AUTH_LEVEL_INTEGRITY;
417 pipe_default_auth_type = PIPE_AUTH_TYPE_SCHANNEL;
419 return cmd_set_ss_level();
423 /* Built in rpcclient commands */
425 static struct cmd_set rpcclient_commands[] = {
427 { "GENERAL OPTIONS" },
429 { "help", RPC_RTYPE_NTSTATUS, cmd_help, NULL, -1, NULL, "Get help on commands", "[command]" },
430 { "?", RPC_RTYPE_NTSTATUS, cmd_help, NULL, -1, NULL, "Get help on commands", "[command]" },
431 { "debuglevel", RPC_RTYPE_NTSTATUS, cmd_debuglevel, NULL, -1, NULL, "Set debug level", "level" },
432 { "list", RPC_RTYPE_NTSTATUS, cmd_listcommands, NULL, -1, NULL, "List available commands on <pipe>", "pipe" },
433 { "exit", RPC_RTYPE_NTSTATUS, cmd_quit, NULL, -1, NULL, "Exit program", "" },
434 { "quit", RPC_RTYPE_NTSTATUS, cmd_quit, NULL, -1, NULL, "Exit program", "" },
435 { "sign", RPC_RTYPE_NTSTATUS, cmd_sign, NULL, -1, NULL, "Force RPC pipe connections to be signed", "" },
436 { "seal", RPC_RTYPE_NTSTATUS, cmd_seal, NULL, -1, NULL, "Force RPC pipe connections to be sealed", "" },
437 { "schannel", RPC_RTYPE_NTSTATUS, cmd_schannel, NULL, -1, NULL, "Force RPC pipe connections to be sealed with 'schannel'. Assumes valid machine account to this domain controller.", "" },
438 { "schannelsign", RPC_RTYPE_NTSTATUS, cmd_schannel_sign, NULL, -1, NULL, "Force RPC pipe connections to be signed (not sealed) with 'schannel'. Assumes valid machine account to this domain controller.", "" },
439 { "none", RPC_RTYPE_NTSTATUS, cmd_none, NULL, -1, NULL, "Force RPC pipe connections to have no special properties", "" },
444 static struct cmd_set separator_command[] = {
445 { "---------------", MAX_RPC_RETURN_TYPE, NULL, NULL, -1, NULL, "----------------------" },
450 /* Various pipe commands */
452 extern struct cmd_set lsarpc_commands[];
453 extern struct cmd_set samr_commands[];
454 extern struct cmd_set spoolss_commands[];
455 extern struct cmd_set netlogon_commands[];
456 extern struct cmd_set srvsvc_commands[];
457 extern struct cmd_set dfs_commands[];
458 extern struct cmd_set reg_commands[];
459 extern struct cmd_set ds_commands[];
460 extern struct cmd_set echo_commands[];
461 extern struct cmd_set shutdown_commands[];
462 extern struct cmd_set test_commands[];
464 static struct cmd_set *rpcclient_command_list[] = {
480 static void add_command_set(struct cmd_set *cmd_set)
482 struct cmd_list *entry;
484 if (!(entry = SMB_MALLOC_P(struct cmd_list))) {
485 DEBUG(0, ("out of memory\n"));
491 entry->cmd_set = cmd_set;
492 DLIST_ADD(cmd_list, entry);
497 * Call an rpcclient function, passing an argv array.
499 * @param cmd Command to run, as a single string.
501 static NTSTATUS do_cmd(struct cli_state *cli,
502 struct cmd_set *cmd_entry,
503 int argc, char **argv)
512 if (!(mem_ctx = talloc_init("do_cmd"))) {
513 DEBUG(0, ("talloc_init() failed\n"));
514 return NT_STATUS_NO_MEMORY;
519 if (cmd_entry->pipe_idx != -1 && cmd_entry->rpc_pipe == NULL) {
520 switch (pipe_default_auth_type) {
521 case PIPE_AUTH_TYPE_NONE:
522 cmd_entry->rpc_pipe = cli_rpc_pipe_open_noauth(cli,
526 case PIPE_AUTH_TYPE_SPNEGO_NTLMSSP:
527 cmd_entry->rpc_pipe = cli_rpc_pipe_open_spnego_ntlmssp(cli,
529 pipe_default_auth_level,
531 cmdline_auth_info.username,
532 cmdline_auth_info.password,
535 case PIPE_AUTH_TYPE_NTLMSSP:
536 cmd_entry->rpc_pipe = cli_rpc_pipe_open_ntlmssp(cli,
538 pipe_default_auth_level,
540 cmdline_auth_info.username,
541 cmdline_auth_info.password,
544 case PIPE_AUTH_TYPE_SCHANNEL:
545 cmd_entry->rpc_pipe = cli_rpc_pipe_open_schannel(cli,
547 pipe_default_auth_level,
552 DEBUG(0, ("Could not initialise %s. Invalid auth type %u\n",
553 cli_get_pipe_name(cmd_entry->pipe_idx),
554 pipe_default_auth_type ));
555 return NT_STATUS_UNSUCCESSFUL;
557 if (!cmd_entry->rpc_pipe) {
558 DEBUG(0, ("Could not initialise %s. Error was %s\n",
559 cli_get_pipe_name(cmd_entry->pipe_idx),
560 nt_errstr(ntresult) ));
564 if (cmd_entry->pipe_idx == PI_NETLOGON) {
565 uint32 neg_flags = NETLOGON_NEG_AUTH2_FLAGS;
566 uint32 sec_channel_type;
567 uchar trust_password[16];
569 if (!secrets_fetch_trust_account_password(lp_workgroup(),
571 NULL, &sec_channel_type)) {
572 return NT_STATUS_UNSUCCESSFUL;
575 ntresult = rpccli_netlogon_setup_creds(cmd_entry->rpc_pipe,
576 cli->desthost, /* server name */
577 lp_workgroup(), /* domain */
578 global_myname(), /* client name */
579 global_myname(), /* machine account name */
584 if (!NT_STATUS_IS_OK(ntresult)) {
585 DEBUG(0, ("Could not initialise credentials for %s.\n",
586 cli_get_pipe_name(cmd_entry->pipe_idx)));
594 if ( cmd_entry->returntype == RPC_RTYPE_NTSTATUS ) {
595 ntresult = cmd_entry->ntfn(cmd_entry->rpc_pipe, mem_ctx, argc, (const char **) argv);
596 if (!NT_STATUS_IS_OK(ntresult)) {
597 printf("result was %s\n", nt_errstr(ntresult));
600 wresult = cmd_entry->wfn(cmd_entry->rpc_pipe, mem_ctx, argc, (const char **) argv);
601 /* print out the DOS error */
602 if (!W_ERROR_IS_OK(wresult)) {
603 printf( "result was %s\n", dos_errstr(wresult));
605 ntresult = W_ERROR_IS_OK(wresult)?NT_STATUS_OK:NT_STATUS_UNSUCCESSFUL;
610 talloc_destroy(mem_ctx);
617 * Process a command entered at the prompt or as part of -c
619 * @returns The NTSTATUS from running the command.
621 static NTSTATUS process_cmd(struct cli_state *cli, char *cmd)
623 struct cmd_list *temp_list;
624 NTSTATUS result = NT_STATUS_OK;
629 if ((ret = poptParseArgvString(cmd, &argc, (const char ***) &argv)) != 0) {
630 fprintf(stderr, "rpcclient: %s\n", poptStrerror(ret));
631 return NT_STATUS_UNSUCCESSFUL;
635 /* Walk through a dlist of arrays of commands. */
636 for (temp_list = cmd_list; temp_list; temp_list = temp_list->next) {
637 struct cmd_set *temp_set = temp_list->cmd_set;
639 while (temp_set->name) {
640 if (strequal(argv[0], temp_set->name)) {
641 if (!(temp_set->returntype == RPC_RTYPE_NTSTATUS && temp_set->ntfn ) &&
642 !(temp_set->returntype == RPC_RTYPE_WERROR && temp_set->wfn )) {
643 fprintf (stderr, "Invalid command\n");
647 result = do_cmd(cli, temp_set, argc, argv);
656 printf("command not found: %s\n", argv[0]);
661 if (!NT_STATUS_IS_OK(result)) {
662 printf("result was %s\n", nt_errstr(result));
667 /* NOTE: popt allocates the whole argv, including the
668 * strings, as a single block. So a single free is
669 * enough to release it -- we don't free the
670 * individual strings. rtfm. */
680 int main(int argc, char *argv[])
682 BOOL interactive = True;
684 static char *cmdstr = NULL;
686 struct cli_state *cli;
687 static char *opt_ipaddr=NULL;
688 struct cmd_set **cmd_set;
689 struct in_addr server_ip;
691 static int opt_port = 0;
693 /* make sure the vars that get altered (4th field) are in
694 a fixed location or certain compilers complain */
696 struct poptOption long_options[] = {
698 {"command", 'c', POPT_ARG_STRING, &cmdstr, 'c', "Execute semicolon separated cmds", "COMMANDS"},
699 {"dest-ip", 'I', POPT_ARG_STRING, &opt_ipaddr, 'I', "Specify destination IP address", "IP"},
700 {"port", 'p', POPT_ARG_INT, &opt_port, 'p', "Specify port number", "PORT"},
702 POPT_COMMON_CONNECTION
703 POPT_COMMON_CREDENTIALS
709 ZERO_STRUCT(server_ip);
713 /* the following functions are part of the Samba debugging
714 facilities. See lib/debug.c */
715 setup_logging("rpcclient", interactive);
721 pc = poptGetContext("rpcclient", argc, (const char **) argv,
725 poptPrintHelp(pc, stderr, 0);
729 while((opt = poptGetNextOpt(pc)) != -1) {
733 if ( (server_ip.s_addr=inet_addr(opt_ipaddr)) == INADDR_NONE ) {
734 fprintf(stderr, "%s not a valid IP address\n",
741 /* Get server as remaining unparsed argument. Print usage if more
742 than one unparsed argument is present. */
744 server = poptGetArg(pc);
746 if (!server || poptGetArg(pc)) {
747 poptPrintHelp(pc, stderr, 0);
758 /* Load smb.conf file */
760 if (!lp_load(dyn_CONFIGFILE,True,False,False,True))
761 fprintf(stderr, "Can't load %s\n", dyn_CONFIGFILE);
765 * from stdin if necessary
768 if (!cmdline_auth_info.got_pass) {
769 char *pass = getpass("Password:");
771 pstrcpy(cmdline_auth_info.password, pass);
775 nt_status = cli_full_connection(&cli, global_myname(), server,
776 opt_ipaddr ? &server_ip : NULL, opt_port,
778 cmdline_auth_info.username,
780 cmdline_auth_info.password,
781 cmdline_auth_info.use_kerberos ? CLI_FULL_CONNECTION_USE_KERBEROS : 0,
782 cmdline_auth_info.signing_state,NULL);
784 if (!NT_STATUS_IS_OK(nt_status)) {
785 DEBUG(0,("Cannot connect to server. Error was %s\n", nt_errstr(nt_status)));
789 #if 0 /* COMMENT OUT FOR TESTING */
790 memset(cmdline_auth_info.password,'X',sizeof(cmdline_auth_info.password));
793 /* Load command lists */
795 cmd_set = rpcclient_command_list;
798 add_command_set(*cmd_set);
799 add_command_set(separator_command);
803 fetch_machine_sid(cli);
805 /* Do anything specified with -c */
806 if (cmdstr && cmdstr[0]) {
811 while((cmd=next_command(&p)) != NULL) {
812 NTSTATUS cmd_result = process_cmd(cli, cmd);
813 result = NT_STATUS_IS_ERR(cmd_result);
820 /* Loop around accepting commands */
826 slprintf(prompt, sizeof(prompt) - 1, "rpcclient $> ");
828 line = smb_readline(prompt, NULL, completion_fn);
834 process_cmd(cli, line);