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         static char             *cmdstr = "";
 585         const char *server;
 586         struct cli_state        *cli;
 587         fstring                 password="",
 588                                 username="",
 589                 domain="";
 590         static char             *opt_authfile=NULL,
 591                                 *opt_username=NULL,
 592                                 *opt_domain=NULL,
 593                                 *opt_configfile=NULL,
 594                                 *opt_logfile=NULL,
 595                                 *opt_ipaddr=NULL;
 596         pstring                 logfile;
 597         struct cmd_set          **cmd_set;
 598         struct in_addr          server_ip;
 599         NTSTATUS                nt_status;
 600
 601         /* make sure the vars that get altered (4th field) are in
 602            a fixed location or certain compilers complain */
 603         poptContext pc;
 604         struct poptOption long_options[] = {
 605                 POPT_AUTOHELP
 606                 {"authfile",    'A', POPT_ARG_STRING,   &opt_authfile, 'A', "File containing user credentials"},
 607                 {"conf",        's', POPT_ARG_STRING,   &opt_configfile, 's', "Specify an alternative config file"},
 608                 {"nopass",      'N', POPT_ARG_NONE,     &got_pass, 'N', "Don't ask for a password"},
 609                 {"user",        'U', POPT_ARG_STRING,   &opt_username, 'U', "Set the network username"},
 610                 {"workgroup",   'W', POPT_ARG_STRING,   &opt_domain, 'W', "Set the domain name for user account"},
 611                 {"command",     'c', POPT_ARG_STRING,   &cmdstr, 'c', "Execute semicolon separated cmds"},
 612                 {"logfile",     'l', POPT_ARG_STRING,   &opt_logfile, 'l', "Logfile to use instead of stdout"},
 613                 {"dest-ip",     'I', POPT_ARG_STRING,   &opt_ipaddr, 'I', "Specify destination IP address"},
 614                 { NULL, 0, POPT_ARG_INCLUDE_TABLE, popt_common_debug },
 615                 { NULL }
 616         };
 617
 618         setlinebuf(stdout);
 619
 620         /* Parse options */
 621
 622         pc = poptGetContext("rpcclient", argc, (const char **) argv,
 623                             long_options, 0);
 624
 625         if (argc == 1) {
 626                 poptPrintHelp(pc, stderr, 0);
 627                 return 0;
 628         }
 629
 630         while((opt = poptGetNextOpt(pc)) != -1) {
 631                 switch (opt) {
 632                 case 'A':
 633                         /* only get the username, password, and domain from the file */
 634                         read_authfile (opt_authfile, username, password, domain);
 635                         if (strlen (password))
 636                                 got_pass = 1;
 637                         break;
 638
 639                 case 'l':
 640                         slprintf(logfile, sizeof(logfile) - 1, "%s.client",
 641                                  opt_logfile);
 642                         lp_set_logfile(logfile);
 643                         interactive = False;
 644                         break;
 645
 646                 case 's':
 647                         pstrcpy(dyn_CONFIGFILE, opt_configfile);
 648                         break;
 649
 650                 case 'U': {
 651                         char *lp;
 652
 653                         pstrcpy(username,opt_username);
 654
 655                         if ((lp=strchr_m(username,'%'))) {
 656                                 *lp = 0;
 657                                 pstrcpy(password,lp+1);
 658                                 got_pass = 1;
 659                                 memset(strchr_m(opt_username,'%') + 1, 'X',
 660                                        strlen(password));
 661                         }
 662                         break;
 663                 }
 664                 case 'I':
 665                         if ( (server_ip.s_addr=inet_addr(opt_ipaddr)) == INADDR_NONE ) {
 666                                 fprintf(stderr, "%s not a valid IP address\n",
 667                                         opt_ipaddr);
 668                                 return 1;
 669                         }
 670                 case 'W':
 671                         pstrcpy(domain, opt_domain);
 672                         break;
 673                 }
 674         }
 675
 676         /* Get server as remaining unparsed argument.  Print usage if more
 677            than one unparsed argument is present. */
 678
 679         server = poptGetArg(pc);
 680
 681         if (!server || poptGetArg(pc)) {
 682                 poptPrintHelp(pc, stderr, 0);
 683                 return 1;
 684         }
 685
 686         poptFreeContext(pc);
 687
 688         /* the following functions are part of the Samba debugging
 689            facilities.  See lib/debug.c */
 690         setup_logging("rpcclient", interactive);
 691         if (!interactive)
 692                 reopen_logs();
 693
 694         /* Load smb.conf file */
 695
 696         if (!lp_load(dyn_CONFIGFILE,True,False,False))
 697                 fprintf(stderr, "Can't load %s\n", dyn_CONFIGFILE);
 698
 699         load_interfaces();
 700
 701         get_myname((*global_myname)?NULL:global_myname);
 702         strupper(global_myname);
 703
 704         /* Resolve the IP address */
 705
 706         if (!opt_ipaddr && !resolve_name(server, &server_ip, 0x20))  {
 707                 DEBUG(1,("Unable to resolve %s\n", server));
 708                 return 1;
 709         }
 710
 711         /*
 712          * Get password
 713          * from stdin if necessary
 714          */
 715
 716         if (!got_pass) {
 717                 char *pass = getpass("Password:");
 718                 if (pass) {
 719                         fstrcpy(password, pass);
 720                 }
 721         }
 722
 723         if (!strlen(username) && !got_pass)
 724                 get_username(username);
 725
 726         nt_status = cli_full_connection(&cli, global_myname, server,
 727                                         &server_ip, 0,
 728                                         "IPC$", "IPC",
 729                                         username, domain,
 730                                         password, 0);
 731
 732         if (!NT_STATUS_IS_OK(nt_status)) {
 733                 DEBUG(0,("Cannot connect to server.  Error was %s\n", nt_errstr(nt_status)));
 734                 return 1;
 735         }
 736
 737         memset(password,'X',sizeof(password));
 738
 739         /* Load command lists */
 740
 741         cmd_set = rpcclient_command_list;
 742
 743         while(*cmd_set) {
 744                 add_command_set(*cmd_set);
 745                 add_command_set(separator_command);
 746                 cmd_set++;
 747         }
 748
 749         fetch_machine_sid(cli);
 750
 751        /* Do anything specified with -c */
 752         if (cmdstr[0]) {
 753                 char    *cmd;
 754                 char    *p = cmdstr;
 755
 756                 while((cmd=next_command(&p)) != NULL) {
 757                         process_cmd(cli, cmd);
 758                 }
 759
 760                 cli_shutdown(cli);
 761                 return 0;
 762         }
 763
 764         /* Loop around accepting commands */
 765
 766         while(1) {
 767                 pstring prompt;
 768                 char *line;
 769
 770                 slprintf(prompt, sizeof(prompt) - 1, "rpcclient $> ");
 771
 772                 line = smb_readline(prompt, NULL, completion_fn);
 773
 774                 if (line == NULL)
 775                         break;
 776
 777                 if (line[0] != '\n')
 778                         process_cmd(cli, line);
 779         }
 780
 781         cli_shutdown(cli);
 782         return 0;
 783 }