Forgot to remove old usage() function
[gd/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
7    This program is free software; you can redistribute it and/or modify
8    it under the terms of the GNU General Public License as published by
9    the Free Software Foundation; either version 2 of the License, or
10    (at your option) any later version.
11    
12    This program is distributed in the hope that it will be useful,
13    but WITHOUT ANY WARRANTY; without even the implied warranty of
14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15    GNU General Public License for more details.
16    
17    You should have received a copy of the GNU General Public License
18    along with this program; if not, write to the Free Software
19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20 */
21
22 #include "includes.h"
23 #include "rpcclient.h"
24
25 DOM_SID domain_sid;
26
27 /* List to hold groups of commands */
28
29 static struct cmd_list {
30         struct cmd_list *prev, *next;
31         struct cmd_set *cmd_set;
32 } *cmd_list;
33
34 /****************************************************************************
35 handle completion of commands for readline
36 ****************************************************************************/
37 static char **completion_fn(char *text, int start, int end)
38 {
39 #define MAX_COMPLETIONS 100
40         char **matches;
41         int i, count=0;
42         struct cmd_list *commands = cmd_list;
43
44 #if 0   /* JERRY */
45         /* FIXME!!!  -- what to do when completing argument? */
46         /* for words not at the start of the line fallback 
47            to filename completion */
48         if (start) 
49                 return NULL;
50 #endif
51
52         /* make sure we have a list of valid commands */
53         if (!commands) 
54                 return NULL;
55
56         matches = (char **)malloc(sizeof(matches[0])*MAX_COMPLETIONS);
57         if (!matches) return NULL;
58
59         matches[count++] = strdup(text);
60         if (!matches[0]) return NULL;
61
62         while (commands && count < MAX_COMPLETIONS-1) 
63         {
64                 if (!commands->cmd_set)
65                         break;
66                 
67                 for (i=0; commands->cmd_set[i].name; i++)
68                 {
69                         if ((strncmp(text, commands->cmd_set[i].name, strlen(text)) == 0) &&
70                                 commands->cmd_set[i].fn) 
71                         {
72                                 matches[count] = strdup(commands->cmd_set[i].name);
73                                 if (!matches[count]) 
74                                         return NULL;
75                                 count++;
76                         }
77                 }
78                 
79                 commands = commands->next;
80                 
81         }
82
83         if (count == 2) {
84                 SAFE_FREE(matches[0]);
85                 matches[0] = strdup(matches[1]);
86         }
87         matches[count] = NULL;
88         return matches;
89 }
90
91 /***********************************************************************
92  * read in username/password credentials from a file
93  */
94 static void read_authfile (
95         char *filename, 
96         char* username, 
97         char* password, 
98         char* domain
99 )
100 {
101         FILE *auth;
102         fstring buf;
103         uint16 len = 0;
104         char *ptr, *val, *param;
105                                
106         if ((auth=sys_fopen(filename, "r")) == NULL)
107         {
108                 printf ("ERROR: Unable to open credentials file!\n");
109                 return;
110         }
111                                 
112         while (!feof(auth))
113         {  
114                 /* get a line from the file */
115                 if (!fgets (buf, sizeof(buf), auth))
116                         continue;
117                 
118                 len = strlen(buf);
119                 
120                 /* skip empty lines */                  
121                 if ((len) && (buf[len-1]=='\n'))
122                 {
123                         buf[len-1] = '\0';
124                         len--;
125                 }       
126                 if (len == 0)
127                         continue;
128                                         
129                 /* break up the line into parameter & value.
130                    will need to eat a little whitespace possibly */
131                 param = buf;
132                 if (!(ptr = strchr_m(buf, '=')))
133                         continue;
134                 val = ptr+1;
135                 *ptr = '\0';
136                                         
137                 /* eat leading white space */
138                 while ((*val!='\0') && ((*val==' ') || (*val=='\t')))
139                         val++;
140                                         
141                 if (strwicmp("password", param) == 0)
142                         fstrcpy (password, val);
143                 else if (strwicmp("username", param) == 0)
144                         fstrcpy (username, val);
145                 else if (strwicmp("domain", param) == 0)
146                         fstrcpy (domain, val);
147                                                 
148                 memset(buf, 0, sizeof(buf));
149         }
150         fclose(auth);
151         
152         return;
153 }
154
155 static char* next_command (char** cmdstr)
156 {
157         static pstring          command;
158         char                    *p;
159         
160         if (!cmdstr || !(*cmdstr))
161                 return NULL;
162         
163         p = strchr_m(*cmdstr, ';');
164         if (p)
165                 *p = '\0';
166         pstrcpy(command, *cmdstr);
167         *cmdstr = p;
168         
169         return command;
170 }
171
172 static void get_username (char *username)
173 {
174         if (getenv("USER"))
175                 pstrcpy(username,getenv("USER"));
176  
177         if (*username == 0 && getenv("LOGNAME"))
178                 pstrcpy(username,getenv("LOGNAME"));
179  
180         if (*username == 0) {
181                 pstrcpy(username,"GUEST");
182         }
183
184         return;
185 }
186
187 /* Fetch the SID for this computer */
188
189 static void fetch_machine_sid(struct cli_state *cli)
190 {
191         POLICY_HND pol;
192         NTSTATUS result = NT_STATUS_OK;
193         uint32 info_class = 5;
194         fstring domain_name;
195         static BOOL got_domain_sid;
196         TALLOC_CTX *mem_ctx;
197
198         if (got_domain_sid) return;
199
200         if (!(mem_ctx=talloc_init()))
201         {
202                 DEBUG(0,("fetch_machine_sid: talloc_init returned NULL!\n"));
203                 goto error;
204         }
205
206
207         if (!cli_nt_session_open (cli, PIPE_LSARPC)) {
208                 fprintf(stderr, "could not initialise lsa pipe\n");
209                 goto error;
210         }
211         
212         result = cli_lsa_open_policy(cli, mem_ctx, True, 
213                                      SEC_RIGHTS_MAXIMUM_ALLOWED,
214                                      &pol);
215         if (!NT_STATUS_IS_OK(result)) {
216                 goto error;
217         }
218
219         result = cli_lsa_query_info_policy(cli, mem_ctx, &pol, info_class, 
220                                            domain_name, &domain_sid);
221         if (!NT_STATUS_IS_OK(result)) {
222                 goto error;
223         }
224
225         got_domain_sid = True;
226
227         cli_lsa_close(cli, mem_ctx, &pol);
228         cli_nt_session_close(cli);
229         talloc_destroy(mem_ctx);
230
231         return;
232
233  error:
234         fprintf(stderr, "could not obtain sid for domain %s\n", cli->domain);
235
236         if (!NT_STATUS_IS_OK(result)) {
237                 fprintf(stderr, "error: %s\n", nt_errstr(result));
238         }
239
240         exit(1);
241 }
242
243 /* List the available commands on a given pipe */
244
245 static NTSTATUS cmd_listcommands(struct cli_state *cli, TALLOC_CTX *mem_ctx,
246                          int argc, char **argv)
247 {
248         struct cmd_list *tmp;
249         struct cmd_set *tmp_set;
250         int i;
251
252         /* Usage */
253
254         if (argc != 2) {
255                 printf("Usage: %s <pipe>\n", argv[0]);
256                 return NT_STATUS_OK;
257         }
258
259         /* Help on one command */
260
261         for (tmp = cmd_list; tmp; tmp = tmp->next) 
262         {
263                 tmp_set = tmp->cmd_set;
264                 
265                 if (!StrCaseCmp(argv[1], tmp_set->name))
266                 {
267                         printf("Available commands on the %s pipe:\n\n", tmp_set->name);
268
269                         i = 0;
270                         tmp_set++;
271                         while(tmp_set->name) {
272                                 printf("%20s", tmp_set->name);
273                                 tmp_set++;
274                                 i++;
275                                 if (i%4 == 0)
276                                         printf("\n");
277                         }
278                         
279                         /* drop out of the loop */
280                         break;
281                 }
282         }
283         printf("\n\n");
284
285         return NT_STATUS_OK;
286 }
287
288 /* Display help on commands */
289
290 static NTSTATUS cmd_help(struct cli_state *cli, TALLOC_CTX *mem_ctx,
291                          int argc, char **argv)
292 {
293         struct cmd_list *tmp;
294         struct cmd_set *tmp_set;
295
296         /* Usage */
297
298         if (argc > 2) {
299                 printf("Usage: %s [command]\n", argv[0]);
300                 return NT_STATUS_OK;
301         }
302
303         /* Help on one command */
304
305         if (argc == 2) {
306                 for (tmp = cmd_list; tmp; tmp = tmp->next) {
307                         
308                         tmp_set = tmp->cmd_set;
309
310                         while(tmp_set->name) {
311                                 if (strequal(argv[1], tmp_set->name)) {
312                                         if (tmp_set->usage &&
313                                             tmp_set->usage[0])
314                                                 printf("%s\n", tmp_set->usage);
315                                         else
316                                                 printf("No help for %s\n", tmp_set->name);
317
318                                         return NT_STATUS_OK;
319                                 }
320
321                                 tmp_set++;
322                         }
323                 }
324
325                 printf("No such command: %s\n", argv[1]);
326                 return NT_STATUS_OK;
327         }
328
329         /* List all commands */
330
331         for (tmp = cmd_list; tmp; tmp = tmp->next) {
332
333                 tmp_set = tmp->cmd_set;
334
335                 while(tmp_set->name) {
336
337                         printf("%15s\t\t%s\n", tmp_set->name,
338                                tmp_set->description ? tmp_set->description:
339                                "");
340
341                         tmp_set++;
342                 }
343         }
344
345         return NT_STATUS_OK;
346 }
347
348 /* Change the debug level */
349
350 static NTSTATUS cmd_debuglevel(struct cli_state *cli, TALLOC_CTX *mem_ctx,
351                                int argc, char **argv)
352 {
353         if (argc > 2) {
354                 printf("Usage: %s [debuglevel]\n", argv[0]);
355                 return NT_STATUS_OK;
356         }
357
358         if (argc == 2) {
359                 DEBUGLEVEL = atoi(argv[1]);
360         }
361
362         printf("debuglevel is %d\n", DEBUGLEVEL);
363
364         return NT_STATUS_OK;
365 }
366
367 static NTSTATUS cmd_quit(struct cli_state *cli, TALLOC_CTX *mem_ctx,
368                          int argc, char **argv)
369 {
370         exit(0);
371         return NT_STATUS_OK; /* NOTREACHED */
372 }
373
374 /* Build in rpcclient commands */
375
376 static struct cmd_set rpcclient_commands[] = {
377
378         { "GENERAL OPTIONS" },
379
380         { "help",       cmd_help,       NULL,   "Get help on commands", "[command]" },
381         { "?",          cmd_help,       NULL,   "Get help on commands", "[command]" },
382         { "debuglevel", cmd_debuglevel, NULL,   "Set debug level", "level" },
383         { "list",       cmd_listcommands, NULL, "List available commands on <pipe>", "pipe" },
384         { "exit",       cmd_quit,       NULL,   "Exit program", "" },
385         { "quit",       cmd_quit,       NULL,   "Exit program", "" },
386
387         { NULL }
388 };
389
390 static struct cmd_set separator_command[] = {
391         { "---------------", NULL,      NULL,   "----------------------" },
392         { NULL }
393 };
394
395
396 /* Various pipe commands */
397
398 extern struct cmd_set lsarpc_commands[];
399 extern struct cmd_set samr_commands[];
400 extern struct cmd_set spoolss_commands[];
401 extern struct cmd_set netlogon_commands[];
402 extern struct cmd_set srvsvc_commands[];
403 extern struct cmd_set dfs_commands[];
404 extern struct cmd_set reg_commands[];
405
406 static struct cmd_set *rpcclient_command_list[] = {
407         rpcclient_commands,
408         lsarpc_commands,
409         samr_commands,
410         spoolss_commands,
411         netlogon_commands,
412         srvsvc_commands,
413         dfs_commands,
414         reg_commands,
415         NULL
416 };
417
418 static void add_command_set(struct cmd_set *cmd_set)
419 {
420         struct cmd_list *entry;
421
422         if (!(entry = (struct cmd_list *)malloc(sizeof(struct cmd_list)))) {
423                 DEBUG(0, ("out of memory\n"));
424                 return;
425         }
426
427         ZERO_STRUCTP(entry);
428
429         entry->cmd_set = cmd_set;
430         DLIST_ADD(cmd_list, entry);
431 }
432
433 static NTSTATUS do_cmd(struct cli_state *cli, struct cmd_set *cmd_entry, 
434                        char *cmd)
435 {
436         char *p = cmd, **argv = NULL;
437         NTSTATUS result = NT_STATUS_UNSUCCESSFUL;
438         pstring buf;
439         int argc = 0, i;
440
441         /* Count number of arguments first time through the loop then
442            allocate memory and strdup them. */
443
444  again:
445         while(next_token(&p, buf, " ", sizeof(buf))) {
446                 if (argv) {
447                         argv[argc] = strdup(buf);
448                 }
449                 
450                 argc++;
451         }
452                                 
453         if (!argv) {
454
455                 /* Create argument list */
456
457                 argv = (char **)malloc(sizeof(char *) * argc);
458                 memset(argv, 0, sizeof(char *) * argc);
459
460                 if (!argv) {
461                         fprintf(stderr, "out of memory\n");
462                         result = NT_STATUS_NO_MEMORY;
463                         goto done;
464                 }
465                                         
466                 p = cmd;
467                 argc = 0;
468                                         
469                 goto again;
470         }
471
472         /* Call the function */
473
474         if (cmd_entry->fn) {
475                 TALLOC_CTX *mem_ctx;
476
477                 /* Create mem_ctx */
478
479                 if (!(mem_ctx = talloc_init())) {
480                         DEBUG(0, ("talloc_init() failed\n"));
481                         goto done;
482                 }
483
484                 /* Open pipe */
485
486                 if (cmd_entry->pipe)
487                         if (!cli_nt_session_open(cli, cmd_entry->pipe)) {
488                                 DEBUG(0, ("Could not initialise %s\n",
489                                           cmd_entry->pipe));
490                                 goto done;
491                         }
492
493                 /* Run command */
494
495                 result = cmd_entry->fn(cli, mem_ctx, argc, argv);
496
497                 /* Cleanup */
498
499                 if (cmd_entry->pipe)
500                         cli_nt_session_close(cli);
501
502                 talloc_destroy(mem_ctx);
503
504         } else {
505                 fprintf (stderr, "Invalid command\n");
506                 goto done;
507         }
508
509  done:
510                                                 
511         /* Cleanup */
512
513         if (argv) {
514                 for (i = 0; i < argc; i++)
515                         SAFE_FREE(argv[i]);
516         
517                 SAFE_FREE(argv);
518         }
519         
520         return result;
521 }
522
523 /* Process a command entered at the prompt or as part of -c */
524
525 static NTSTATUS process_cmd(struct cli_state *cli, char *cmd)
526 {
527         struct cmd_list *temp_list;
528         BOOL found = False;
529         pstring buf;
530         char *p = cmd;
531         NTSTATUS result = NT_STATUS_OK;
532         int len = 0;
533
534         if (cmd[strlen(cmd) - 1] == '\n')
535                 cmd[strlen(cmd) - 1] = '\0';
536
537         if (!next_token(&p, buf, " ", sizeof(buf))) {
538                 return NT_STATUS_OK;
539         }
540
541         /* strip the trainly \n if it exsists */
542         len = strlen(buf);
543         if (buf[len-1] == '\n')
544                 buf[len-1] = '\0';
545
546         /* Search for matching commands */
547
548         for (temp_list = cmd_list; temp_list; temp_list = temp_list->next) {
549                 struct cmd_set *temp_set = temp_list->cmd_set;
550
551                 while(temp_set->name) {
552                         if (strequal(buf, temp_set->name)) {
553                                 found = True;
554                                 result = do_cmd(cli, temp_set, cmd);
555
556                                 goto done;
557                         }
558                         temp_set++;
559                 }
560         }
561
562  done:
563         if (!found && buf[0]) {
564                 printf("command not found: %s\n", buf);
565                 return NT_STATUS_OK;
566         }
567
568         if (!NT_STATUS_IS_OK(result)) {
569                 printf("result was %s\n", nt_errstr(result));
570         }
571
572         return result;
573 }
574
575
576 /* Main function */
577
578  int main(int argc, char *argv[])
579 {
580         extern pstring          global_myname;
581         static int              got_pass = 0;
582         BOOL                    interactive = True;
583         int                     opt;
584         int                     olddebug;
585         static char             *cmdstr = "";
586         const char *server;
587         struct cli_state        *cli;
588         fstring                 password="",
589                                 username="",
590                 domain="";
591         static char             *opt_authfile=NULL,
592                                 *opt_username=NULL,
593                                 *opt_domain=NULL,
594                                 *opt_configfile=NULL,
595                                 *opt_logfile=NULL,
596                                 *opt_ipaddr=NULL;
597         pstring                 logfile;
598         struct cmd_set          **cmd_set;
599         struct in_addr          server_ip;
600         NTSTATUS                nt_status;
601         extern BOOL             AllowDebugChange;
602
603         /* make sure the vars that get altered (4th field) are in
604            a fixed location or certain compilers complain */
605         poptContext pc;
606         struct poptOption long_options[] = {
607                 POPT_AUTOHELP
608                 {"authfile",    'A', POPT_ARG_STRING,   &opt_authfile, 'A', "File containing user credentials"},
609                 {"conf",        's', POPT_ARG_STRING,   &opt_configfile, 's', "Specify an alternative config file"},
610                 {"nopass",      'N', POPT_ARG_NONE,     &got_pass, 'N', "Don't ask for a password"},
611                 {"user",        'U', POPT_ARG_STRING,   &opt_username, 'U', "Set the network username"},
612                 {"workgroup",   'W', POPT_ARG_STRING,   &opt_domain, 'W', "Set the domain name for user account"},
613                 {"command",     'c', POPT_ARG_STRING,   &cmdstr, 'c', "Execute semicolon separated cmds"},
614                 {"logfile",     'l', POPT_ARG_STRING,   &opt_logfile, 'l', "Logfile to use instead of stdout"},
615                 {"dest-ip",     'I', POPT_ARG_STRING,   &opt_ipaddr, 'I', "Specify destination IP address"},
616                 { NULL, 0, POPT_ARG_INCLUDE_TABLE, popt_common_debug },
617                 { NULL }
618         };
619
620         setlinebuf(stdout);
621
622         DEBUGLEVEL = 1;
623         AllowDebugChange = False;
624
625         /* Parse options */
626
627         pc = poptGetContext("rpcclient", argc, (const char **) argv,
628                             long_options, 0);
629
630         if (argc == 1) {
631                 poptPrintHelp(pc, stderr, 0);
632                 return 0;
633         }
634         
635         while((opt = poptGetNextOpt(pc)) != -1) {
636                 switch (opt) {
637                 case 'A':
638                         /* only get the username, password, and domain from the file */
639                         read_authfile (opt_authfile, username, password, domain);
640                         if (strlen (password))
641                                 got_pass = 1;
642                         break;
643                         
644                 case 'l':
645                         slprintf(logfile, sizeof(logfile) - 1, "%s.client", 
646                                  opt_logfile);
647                         lp_set_logfile(logfile);
648                         interactive = False;
649                         break;
650                         
651                 case 's':
652                         pstrcpy(dyn_CONFIGFILE, opt_configfile);
653                         break;
654                         
655                 case 'U': {
656                         char *lp;
657
658                         pstrcpy(username,opt_username);
659
660                         if ((lp=strchr_m(username,'%'))) {
661                                 *lp = 0;
662                                 pstrcpy(password,lp+1);
663                                 got_pass = 1;
664                                 memset(strchr_m(opt_username,'%') + 1, 'X',
665                                        strlen(password));
666                         }
667                         break;
668                 }
669                 case 'I':
670                         if (!inet_aton(opt_ipaddr, &server_ip)) {
671                                 fprintf(stderr, "%s not a valid IP address\n",
672                                         opt_ipaddr);
673                                 return 1;
674                         }
675                 case 'W':
676                         pstrcpy(domain, opt_domain);
677                         break;
678                 }
679         }
680
681         /* Get server as remaining unparsed argument.  Print usage if more
682            than one unparsed argument is present. */
683
684         server = poptGetArg(pc);
685         
686         if (!server || poptGetArg(pc)) {
687                 poptPrintHelp(pc, stderr, 0);
688                 return 1;
689         }
690
691         poptFreeContext(pc);
692
693         /* the following functions are part of the Samba debugging
694            facilities.  See lib/debug.c */
695         setup_logging("rpcclient", interactive);
696         if (!interactive) 
697                 reopen_logs();
698         
699         /* Load smb.conf file */
700         /* FIXME!  How to get this DEBUGLEVEL to last over lp_load()? */
701         olddebug = DEBUGLEVEL;
702         if (!lp_load(dyn_CONFIGFILE,True,False,False)) {
703                 fprintf(stderr, "Can't load %s\n", dyn_CONFIGFILE);
704         }
705         DEBUGLEVEL = olddebug;
706
707         load_interfaces();
708
709         get_myname((*global_myname)?NULL:global_myname);
710         strupper(global_myname);
711
712         /* Resolve the IP address */
713
714         if (!opt_ipaddr && !resolve_name(server, &server_ip, 0x20))  {
715                 DEBUG(1,("Unable to resolve %s\n", server));
716                 return 1;
717         }
718         
719         /*
720          * Get password
721          * from stdin if necessary
722          */
723
724         if (!got_pass) {
725                 char *pass = getpass("Password:");
726                 if (pass) {
727                         fstrcpy(password, pass);
728                 }
729         }
730         
731         if (!strlen(username) && !got_pass)
732                 get_username(username);
733                 
734         nt_status = cli_full_connection(&cli, global_myname, server, 
735                                         &server_ip, 0,
736                                         "IPC$", "IPC",  
737                                         username, domain,
738                                         password, 0);
739         
740         if (!NT_STATUS_IS_OK(nt_status)) {
741                 DEBUG(0,("Cannot connect to server.  Error was %s\n", nt_errstr(nt_status)));
742                 return 1;
743         }
744
745         memset(password,'X',sizeof(password));
746
747         /* Load command lists */
748
749         cmd_set = rpcclient_command_list;
750
751         while(*cmd_set) {
752                 add_command_set(*cmd_set);
753                 add_command_set(separator_command);
754                 cmd_set++;
755         }
756
757         fetch_machine_sid(cli);
758  
759        /* Do anything specified with -c */
760         if (cmdstr[0]) {
761                 char    *cmd;
762                 char    *p = cmdstr;
763  
764                 while((cmd=next_command(&p)) != NULL) {
765                         process_cmd(cli, cmd);
766                 }
767                 
768                 cli_shutdown(cli);
769                 return 0;
770         }
771
772         /* Loop around accepting commands */
773
774         while(1) {
775                 pstring prompt;
776                 char *line;
777
778                 slprintf(prompt, sizeof(prompt) - 1, "rpcclient $> ");
779
780                 line = smb_readline(prompt, NULL, completion_fn);
781
782                 if (line == NULL)
783                         break;
784
785                 if (line[0] != '\n')
786                         process_cmd(cli, line);
787         }
788         
789         cli_shutdown(cli);
790         return 0;
791 }