fish/fish.c

   1 /* guestfish - the filesystem interactive shell
   2  * Copyright (C) 2009-2010 Red Hat Inc.
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software
  16  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17  */
  18
  19 #include <config.h>
  20
  21 #include <stdio.h>
  22 #include <stdlib.h>
  23 #include <string.h>
  24 #include <unistd.h>
  25 #include <fcntl.h>
  26 #include <getopt.h>
  27 #include <signal.h>
  28 #include <assert.h>
  29 #include <sys/types.h>
  30 #include <sys/wait.h>
  31 #include <locale.h>
  32
  33 #ifdef HAVE_LIBREADLINE
  34 #include <readline/readline.h>
  35 #include <readline/history.h>
  36 #endif
  37
  38 #include <guestfs.h>
  39
  40 #include "fish.h"
  41 #include "c-ctype.h"
  42 #include "closeout.h"
  43 #include "progname.h"
  44
  45 struct drv {
  46   struct drv *next;
  47   char *filename;               /* disk filename (for -a or -N options) */
  48   prep_data *data;              /* prepared type (for -N option only) */
  49   char *device;                 /* device inside the appliance */
  50 };
  51
  52 struct mp {
  53   struct mp *next;
  54   char *device;
  55   char *mountpoint;
  56 };
  57
  58 static void add_drives (struct drv *drv);
  59 static void prepare_drives (struct drv *drv);
  60 static void mount_mps (struct mp *mp);
  61 static int launch (void);
  62 static void interactive (void);
  63 static void shell_script (void);
  64 static void script (int prompt);
  65 static void cmdline (char *argv[], int optind, int argc);
  66 static void initialize_readline (void);
  67 static void cleanup_readline (void);
  68 #ifdef HAVE_LIBREADLINE
  69 static void add_history_line (const char *);
  70 #endif
  71 static void print_shell_quote (FILE *stream, const char *str);
  72
  73 /* Currently open libguestfs handle. */
  74 guestfs_h *g;
  75
  76 int read_only = 0;
  77 int quit = 0;
  78 int verbose = 0;
  79 int remote_control_listen = 0;
  80 int remote_control = 0;
  81 int exit_on_error = 1;
  82 int command_num = 0;
  83
  84 static void __attribute__((noreturn))
  85 usage (int status)
  86 {
  87   if (status != EXIT_SUCCESS)
  88     fprintf (stderr, _("Try `%s --help' for more information.\n"),
  89              program_name);
  90   else {
  91     fprintf (stdout,
  92            _("%s: guest filesystem shell\n"
  93              "%s lets you edit virtual machine filesystems\n"
  94              "Copyright (C) 2009 Red Hat Inc.\n"
  95              "Usage:\n"
  96              "  %s [--options] cmd [: cmd : cmd ...]\n"
  97              "  %s -i libvirt-domain\n"
  98              "  %s -i disk-image(s)\n"
  99              "or for interactive use:\n"
 100              "  %s\n"
 101              "or from a shell script:\n"
 102              "  %s <<EOF\n"
 103              "  cmd\n"
 104              "  ...\n"
 105              "  EOF\n"
 106              "Options:\n"
 107              "  -h|--cmd-help        List available commands\n"
 108              "  -h|--cmd-help cmd    Display detailed help on 'cmd'\n"
 109              "  -a|--add image       Add image\n"
 110              "  -D|--no-dest-paths   Don't tab-complete paths from guest fs\n"
 111              "  -f|--file file       Read commands from file\n"
 112              "  -i|--inspector       Run virt-inspector to get disk mountpoints\n"
 113              "  --listen             Listen for remote commands\n"
 114              "  -m|--mount dev[:mnt] Mount dev on mnt (if omitted, /)\n"
 115              "  -n|--no-sync         Don't autosync\n"
 116              "  -N|--new type        Create prepared disk (test1.img, ...)\n"
 117              "  --remote[=pid]       Send commands to remote %s\n"
 118              "  -r|--ro              Mount read-only\n"
 119              "  --selinux            Enable SELinux support\n"
 120              "  -v|--verbose         Verbose messages\n"
 121              "  -x                   Echo each command before executing it\n"
 122              "  -V|--version         Display version and exit\n"
 123              "For more information,  see the manpage %s(1).\n"),
 124              program_name, program_name, program_name,
 125              program_name, program_name, program_name,
 126              program_name, program_name, program_name);
 127   }
 128   exit (status);
 129 }
 130
 131 int
 132 main (int argc, char *argv[])
 133 {
 134   /* Set global program name that is not polluted with libtool artifacts.  */
 135   set_program_name (argv[0]);
 136
 137   atexit (close_stdout);
 138
 139   setlocale (LC_ALL, "");
 140   bindtextdomain (PACKAGE, LOCALEBASEDIR);
 141   textdomain (PACKAGE);
 142
 143   enum { HELP_OPTION = CHAR_MAX + 1 };
 144
 145   static const char *options = "a:Df:h::im:nN:rv?Vx";
 146   static const struct option long_options[] = {
 147     { "add", 1, 0, 'a' },
 148     { "cmd-help", 2, 0, 'h' },
 149     { "file", 1, 0, 'f' },
 150     { "help", 0, 0, HELP_OPTION },
 151     { "inspector", 0, 0, 'i' },
 152     { "listen", 0, 0, 0 },
 153     { "mount", 1, 0, 'm' },
 154     { "new", 1, 0, 'N' },
 155     { "no-dest-paths", 0, 0, 'D' },
 156     { "no-sync", 0, 0, 'n' },
 157     { "remote", 2, 0, 0 },
 158     { "ro", 0, 0, 'r' },
 159     { "selinux", 0, 0, 0 },
 160     { "verbose", 0, 0, 'v' },
 161     { "version", 0, 0, 'V' },
 162     { 0, 0, 0, 0 }
 163   };
 164   struct drv *drvs = NULL;
 165   struct drv *drv;
 166   struct mp *mps = NULL;
 167   struct mp *mp;
 168   char *p, *file = NULL;
 169   int c;
 170   int inspector = 0;
 171   int option_index;
 172   struct sigaction sa;
 173   char next_drive = 'a';
 174   int next_prepared_drive = 1;
 175
 176   initialize_readline ();
 177
 178   memset (&sa, 0, sizeof sa);
 179   sa.sa_handler = SIG_IGN;
 180   sa.sa_flags = SA_RESTART;
 181   sigaction (SIGPIPE, &sa, NULL);
 182
 183   /* guestfs_create is meant to be a lightweight operation, so
 184    * it's OK to do it early here.
 185    */
 186   g = guestfs_create ();
 187   if (g == NULL) {
 188     fprintf (stderr, _("guestfs_create: failed to create handle\n"));
 189     exit (EXIT_FAILURE);
 190   }
 191
 192   guestfs_set_autosync (g, 1);
 193
 194   /* If developing, add ./appliance to the path.  Note that libtools
 195    * interferes with this because uninstalled guestfish is a shell
 196    * script that runs the real program with an absolute path.  Detect
 197    * that too.
 198    *
 199    * BUT if LIBGUESTFS_PATH environment variable is already set by
 200    * the user, then don't override it.
 201    */
 202   if (getenv ("LIBGUESTFS_PATH") == NULL &&
 203       argv[0] &&
 204       (argv[0][0] != '/' || strstr (argv[0], "/.libs/lt-") != NULL))
 205     guestfs_set_path (g, "appliance:" GUESTFS_DEFAULT_PATH);
 206
 207   /* CAUTION: we are careful to modify argv[0] here, only after
 208    * using it just above.
 209    *
 210    * getopt_long uses argv[0], so give it the sanitized name.  Save a copy
 211    * of the original, in case it's needed in virt-inspector mode, below.
 212    */
 213   char *real_argv0 = argv[0];
 214   argv[0] = bad_cast (program_name);
 215
 216   for (;;) {
 217     c = getopt_long (argc, argv, options, long_options, &option_index);
 218     if (c == -1) break;
 219
 220     switch (c) {
 221     case 0:                     /* options which are long only */
 222       if (STREQ (long_options[option_index].name, "listen"))
 223         remote_control_listen = 1;
 224       else if (STREQ (long_options[option_index].name, "remote")) {
 225         if (optarg) {
 226           if (sscanf (optarg, "%d", &remote_control) != 1) {
 227             fprintf (stderr, _("%s: --listen=PID: PID was not a number: %s\n"),
 228                      program_name, optarg);
 229             exit (EXIT_FAILURE);
 230           }
 231         } else {
 232           p = getenv ("GUESTFISH_PID");
 233           if (!p || sscanf (p, "%d", &remote_control) != 1) {
 234             fprintf (stderr, _("%s: remote: $GUESTFISH_PID must be set"
 235                                " to the PID of the remote process\n"),
 236                      program_name);
 237             exit (EXIT_FAILURE);
 238           }
 239         }
 240       } else if (STREQ (long_options[option_index].name, "selinux")) {
 241         guestfs_set_selinux (g, 1);
 242       } else {
 243         fprintf (stderr, _("%s: unknown long option: %s (%d)\n"),
 244                  program_name, long_options[option_index].name, option_index);
 245         exit (EXIT_FAILURE);
 246       }
 247       break;
 248
 249     case 'a':
 250       if (access (optarg, R_OK) != 0) {
 251         perror (optarg);
 252         exit (EXIT_FAILURE);
 253       }
 254       drv = malloc (sizeof (struct drv));
 255       if (!drv) {
 256         perror ("malloc");
 257         exit (EXIT_FAILURE);
 258       }
 259       drv->filename = optarg;
 260       drv->data = NULL;
 261       /* We could fill the device field in, but in fact we
 262        * only use it for the -N option at present.
 263        */
 264       drv->device = NULL;
 265       drv->next = drvs;
 266       drvs = drv;
 267       next_drive++;
 268       break;
 269
 270     case 'N':
 271       if (STRCASEEQ (optarg, "list")) {
 272         list_prepared_drives ();
 273         exit (EXIT_SUCCESS);
 274       }
 275       drv = malloc (sizeof (struct drv));
 276       if (!drv) {
 277         perror ("malloc");
 278         exit (EXIT_FAILURE);
 279       }
 280       if (asprintf (&drv->filename, "test%d.img",
 281                     next_prepared_drive++) == -1) {
 282         perror ("asprintf");
 283         exit (EXIT_FAILURE);
 284       }
 285       drv->data = create_prepared_file (optarg, drv->filename);
 286       if (asprintf (&drv->device, "/dev/sd%c", next_drive++) == -1) {
 287         perror ("asprintf");
 288         exit (EXIT_FAILURE);
 289       }
 290       drv->next = drvs;
 291       drvs = drv;
 292       break;
 293
 294     case 'D':
 295       complete_dest_paths = 0;
 296       break;
 297
 298     case 'f':
 299       if (file) {
 300         fprintf (stderr, _("%s: only one -f parameter can be given\n"),
 301                  program_name);
 302         exit (EXIT_FAILURE);
 303       }
 304       file = optarg;
 305       break;
 306
 307     case 'h':
 308       if (optarg)
 309         display_command (optarg);
 310       else if (argv[optind] && argv[optind][0] != '-')
 311         display_command (argv[optind++]);
 312       else
 313         list_commands ();
 314       exit (EXIT_SUCCESS);
 315
 316     case 'i':
 317       inspector = 1;
 318       break;
 319
 320     case 'm':
 321       mp = malloc (sizeof (struct mp));
 322       if (!mp) {
 323         perror ("malloc");
 324         exit (EXIT_FAILURE);
 325       }
 326       p = strchr (optarg, ':');
 327       if (p) {
 328         *p = '\0';
 329         mp->mountpoint = p+1;
 330       } else
 331         mp->mountpoint = bad_cast ("/");
 332       mp->device = optarg;
 333       mp->next = mps;
 334       mps = mp;
 335       break;
 336
 337     case 'n':
 338       guestfs_set_autosync (g, 0);
 339       break;
 340
 341     case 'r':
 342       read_only = 1;
 343       break;
 344
 345     case 'v':
 346       verbose++;
 347       guestfs_set_verbose (g, verbose);
 348       break;
 349
 350     case 'V':
 351       printf ("%s %s\n", program_name, PACKAGE_VERSION);
 352       exit (EXIT_SUCCESS);
 353
 354     case 'x':
 355       guestfs_set_trace (g, 1);
 356       break;
 357
 358     case HELP_OPTION:
 359       usage (EXIT_SUCCESS);
 360
 361     default:
 362       usage (EXIT_FAILURE);
 363     }
 364   }
 365
 366   /* Inspector mode invalidates most of the other arguments. */
 367   if (inspector) {
 368     if (drvs || mps || remote_control_listen || remote_control ||
 369         guestfs_get_selinux (g)) {
 370       fprintf (stderr, _("%s: cannot use -i option with -a, -m, -N, "
 371                          "--listen, --remote or --selinux\n"),
 372                program_name);
 373       exit (EXIT_FAILURE);
 374     }
 375     if (optind >= argc) {
 376       fprintf (stderr,
 377            _("%s: -i requires a libvirt domain or path(s) to disk image(s)\n"),
 378                program_name);
 379       exit (EXIT_FAILURE);
 380     }
 381
 382     char *cmd;
 383     size_t cmdlen;
 384     FILE *fp = open_memstream (&cmd, &cmdlen);
 385     if (fp == NULL) {
 386       perror ("open_memstream");
 387       exit (EXIT_FAILURE);
 388     }
 389
 390     fprintf (fp, "virt-inspector");
 391     while (optind < argc) {
 392       fputc (' ', fp);
 393       print_shell_quote (fp, argv[optind]);
 394       optind++;
 395     }
 396
 397     if (read_only)
 398       fprintf (fp, " --ro-fish");
 399     else
 400       fprintf (fp, " --fish");
 401
 402     if (fclose (fp) == -1) {
 403       perror ("fclose");
 404       exit (EXIT_FAILURE);
 405     }
 406
 407     if (verbose)
 408       fprintf (stderr,
 409                "%s -i: running: %s\n", program_name, cmd);
 410
 411     FILE *pp = popen (cmd, "r");
 412     if (pp == NULL) {
 413       perror (cmd);
 414       exit (EXIT_FAILURE);
 415     }
 416
 417     char *cmd2;
 418     fp = open_memstream (&cmd2, &cmdlen);
 419     if (fp == NULL) {
 420       perror ("open_memstream");
 421       exit (EXIT_FAILURE);
 422     }
 423
 424     fprintf (fp, "%s", real_argv0);
 425
 426     if (guestfs_get_verbose (g))
 427       fprintf (fp, " -v");
 428     if (!guestfs_get_autosync (g))
 429       fprintf (fp, " -n");
 430     if (guestfs_get_trace (g))
 431       fprintf (fp, " -x");
 432
 433     char *insp = NULL;
 434     size_t insplen;
 435     if (getline (&insp, &insplen, pp) == -1) {
 436       perror (cmd);
 437       exit (EXIT_FAILURE);
 438     }
 439     fprintf (fp, " %s", insp);
 440
 441     if (pclose (pp) == -1) {
 442       perror (cmd);
 443       exit (EXIT_FAILURE);
 444     }
 445
 446     if (fclose (fp) == -1) {
 447       perror ("fclose");
 448       exit (EXIT_FAILURE);
 449     }
 450
 451     if (verbose)
 452       fprintf (stderr,
 453                "%s -i: running: %s\n", program_name, cmd2);
 454
 455     int r = system (cmd2);
 456     if (r == -1) {
 457       perror (cmd2);
 458       exit (EXIT_FAILURE);
 459     }
 460
 461     free (cmd);
 462     free (cmd2);
 463     free (insp);
 464
 465     exit (WEXITSTATUS (r));
 466   }
 467
 468   /* If we've got drives to add, add them now. */
 469   add_drives (drvs);
 470
 471   /* If we've got mountpoints or prepared drives, we must launch the
 472    * guest and mount them.
 473    */
 474   if (next_prepared_drive > 1 || mps != NULL) {
 475     if (launch () == -1) exit (EXIT_FAILURE);
 476     prepare_drives (drvs);
 477     mount_mps (mps);
 478   }
 479
 480   /* Remote control? */
 481   if (remote_control_listen && remote_control) {
 482     fprintf (stderr,
 483              _("%s: cannot use --listen and --remote options at the same time\n"),
 484              program_name);
 485     exit (EXIT_FAILURE);
 486   }
 487
 488   if (remote_control_listen) {
 489     if (optind < argc) {
 490       fprintf (stderr,
 491                _("%s: extra parameters on the command line with --listen flag\n"),
 492                program_name);
 493       exit (EXIT_FAILURE);
 494     }
 495     if (file) {
 496       fprintf (stderr,
 497                _("%s: cannot use --listen and --file options at the same time\n"),
 498                program_name);
 499       exit (EXIT_FAILURE);
 500     }
 501     rc_listen ();
 502   }
 503
 504   /* -f (file) parameter? */
 505   if (file) {
 506     close (0);
 507     if (open (file, O_RDONLY) == -1) {
 508       perror (file);
 509       exit (EXIT_FAILURE);
 510     }
 511   }
 512
 513   /* Interactive, shell script, or command(s) on the command line? */
 514   if (optind >= argc) {
 515     if (isatty (0))
 516       interactive ();
 517     else
 518       shell_script ();
 519   }
 520   else
 521     cmdline (argv, optind, argc);
 522
 523   cleanup_readline ();
 524
 525   exit (EXIT_SUCCESS);
 526 }
 527
 528 void
 529 pod2text (const char *name, const char *shortdesc, const char *str)
 530 {
 531   FILE *fp;
 532
 533   fp = popen ("pod2text", "w");
 534   if (fp == NULL) {
 535     /* pod2text failed, maybe not found, so let's just print the
 536      * source instead, since that's better than doing nothing.
 537      */
 538     printf ("%s - %s\n\n%s\n", name, shortdesc, str);
 539     return;
 540   }
 541   fprintf (fp, "=head1 NAME\n\n%s - %s\n\n", name, shortdesc);
 542   fputs (str, fp);
 543   pclose (fp);
 544 }
 545
 546 /* List is built in reverse order, so mount them in reverse order. */
 547 static void
 548 mount_mps (struct mp *mp)
 549 {
 550   int r;
 551
 552   if (mp) {
 553     mount_mps (mp->next);
 554
 555     /* Don't use guestfs_mount here because that will default to mount
 556      * options -o sync,noatime.  For more information, see guestfs(3)
 557      * section "LIBGUESTFS GOTCHAS".
 558      */
 559     const char *options = read_only ? "ro" : "";
 560     r = guestfs_mount_options (g, options, mp->device, mp->mountpoint);
 561     if (r == -1)
 562       exit (EXIT_FAILURE);
 563   }
 564 }
 565
 566 static void
 567 add_drives (struct drv *drv)
 568 {
 569   int r;
 570
 571   if (drv) {
 572     add_drives (drv->next);
 573
 574     if (drv->data /* -N option is not affected by --ro */ || !read_only)
 575       r = guestfs_add_drive (g, drv->filename);
 576     else
 577       r = guestfs_add_drive_ro (g, drv->filename);
 578     if (r == -1)
 579       exit (EXIT_FAILURE);
 580   }
 581 }
 582
 583 static void
 584 prepare_drives (struct drv *drv)
 585 {
 586   if (drv) {
 587     prepare_drives (drv->next);
 588     if (drv->data)
 589       prepare_drive (drv->filename, drv->data, drv->device);
 590   }
 591 }
 592
 593 static int
 594 launch (void)
 595 {
 596   if (guestfs_is_config (g)) {
 597     if (guestfs_launch (g) == -1)
 598       return -1;
 599   }
 600   return 0;
 601 }
 602
 603 static void
 604 interactive (void)
 605 {
 606   script (1);
 607 }
 608
 609 static void
 610 shell_script (void)
 611 {
 612   script (0);
 613 }
 614
 615 #define FISH "><fs> "
 616
 617 static char *line_read = NULL;
 618
 619 static char *
 620 rl_gets (int prompt)
 621 {
 622 #ifdef HAVE_LIBREADLINE
 623
 624   if (prompt) {
 625     if (line_read) {
 626       free (line_read);
 627       line_read = NULL;
 628     }
 629
 630     line_read = readline (prompt ? FISH : "");
 631
 632     if (line_read && *line_read)
 633       add_history_line (line_read);
 634
 635     return line_read;
 636   }
 637
 638 #endif /* HAVE_LIBREADLINE */
 639
 640   static char buf[8192];
 641   int len;
 642
 643   if (prompt) printf (FISH);
 644   line_read = fgets (buf, sizeof buf, stdin);
 645
 646   if (line_read) {
 647     len = strlen (line_read);
 648     if (len > 0 && buf[len-1] == '\n') buf[len-1] = '\0';
 649   }
 650
 651   return line_read;
 652 }
 653
 654 static void
 655 script (int prompt)
 656 {
 657   char *buf;
 658   char *cmd;
 659   char *p, *pend;
 660   char *argv[64];
 661   int len;
 662   int global_exit_on_error = !prompt;
 663   int tilde_candidate;
 664
 665   if (prompt)
 666     printf (_("\n"
 667               "Welcome to guestfish, the libguestfs filesystem interactive shell for\n"
 668               "editing virtual machine filesystems.\n"
 669               "\n"
 670               "Type: 'help' for a list of commands\n"
 671               "      'man' to read the manual\n"
 672               "      'quit' to quit the shell\n"
 673               "\n"));
 674
 675   while (!quit) {
 676     char *pipe = NULL;
 677
 678     exit_on_error = global_exit_on_error;
 679
 680     buf = rl_gets (prompt);
 681     if (!buf) {
 682       quit = 1;
 683       break;
 684     }
 685
 686     /* Skip any initial whitespace before the command. */
 687   again:
 688     while (*buf && c_isspace (*buf))
 689       buf++;
 690
 691     if (!*buf) continue;
 692
 693     /* If the next character is '#' then this is a comment. */
 694     if (*buf == '#') continue;
 695
 696     /* If the next character is '!' then pass the whole lot to system(3). */
 697     if (*buf == '!') {
 698       int r;
 699
 700       r = system (buf+1);
 701       if (exit_on_error) {
 702         if (r == -1 ||
 703             (WIFSIGNALED (r) &&
 704              (WTERMSIG (r) == SIGINT || WTERMSIG (r) == SIGQUIT)) ||
 705             WEXITSTATUS (r) != 0)
 706           exit (EXIT_FAILURE);
 707       }
 708       continue;
 709     }
 710
 711     /* If the next character is '-' allow the command to fail without
 712      * exiting on error (just for this one command though).
 713      */
 714     if (*buf == '-') {
 715       exit_on_error = 0;
 716       buf++;
 717       goto again;
 718     }
 719
 720     /* Get the command (cannot be quoted). */
 721     len = strcspn (buf, " \t");
 722
 723     if (len == 0) continue;
 724
 725     cmd = buf;
 726     unsigned int i = 0;
 727     if (buf[len] == '\0') {
 728       argv[0] = NULL;
 729       goto got_command;
 730     }
 731
 732     buf[len] = '\0';
 733     p = &buf[len+1];
 734     p += strspn (p, " \t");
 735
 736     /* Get the parameters. */
 737     while (*p && i < sizeof argv / sizeof argv[0]) {
 738       tilde_candidate = 0;
 739
 740       /* Parameters which start with quotes or pipes are treated
 741        * specially.  Bare parameters are delimited by whitespace.
 742        */
 743       if (*p == '"') {
 744         p++;
 745         len = strcspn (p, "\"");
 746         if (p[len] == '\0') {
 747           fprintf (stderr, _("%s: unterminated double quote\n"), program_name);
 748           if (exit_on_error) exit (EXIT_FAILURE);
 749           goto next_command;
 750         }
 751         if (p[len+1] && (p[len+1] != ' ' && p[len+1] != '\t')) {
 752           fprintf (stderr,
 753                    _("%s: command arguments not separated by whitespace\n"),
 754                    program_name);
 755           if (exit_on_error) exit (EXIT_FAILURE);
 756           goto next_command;
 757         }
 758         p[len] = '\0';
 759         pend = p[len+1] ? &p[len+2] : &p[len+1];
 760       } else if (*p == '\'') {
 761         p++;
 762         len = strcspn (p, "'");
 763         if (p[len] == '\0') {
 764           fprintf (stderr, _("%s: unterminated single quote\n"), program_name);
 765           if (exit_on_error) exit (EXIT_FAILURE);
 766           goto next_command;
 767         }
 768         if (p[len+1] && (p[len+1] != ' ' && p[len+1] != '\t')) {
 769           fprintf (stderr,
 770                    _("%s: command arguments not separated by whitespace\n"),
 771                    program_name);
 772           if (exit_on_error) exit (EXIT_FAILURE);
 773           goto next_command;
 774         }
 775         p[len] = '\0';
 776         pend = p[len+1] ? &p[len+2] : &p[len+1];
 777       } else if (*p == '|') {
 778         *p = '\0';
 779         pipe = p+1;
 780         continue;
 781         /*
 782       } else if (*p == '[') {
 783         int c = 1;
 784         p++;
 785         pend = p;
 786         while (*pend && c != 0) {
 787           if (*pend == '[') c++;
 788           else if (*pend == ']') c--;
 789           pend++;
 790         }
 791         if (c != 0) {
 792           fprintf (stderr,
 793                    _("%s: unterminated \"[...]\" sequence\n"), program_name);
 794           if (exit_on_error) exit (EXIT_FAILURE);
 795           goto next_command;
 796         }
 797         if (*pend && (*pend != ' ' && *pend != '\t')) {
 798           fprintf (stderr,
 799                    _("%s: command arguments not separated by whitespace\n"),
 800                    program_name);
 801           if (exit_on_error) exit (EXIT_FAILURE);
 802           goto next_command;
 803         }
 804         *(pend-1) = '\0';
 805         */
 806       } else if (*p != ' ' && *p != '\t') {
 807         /* If the first character is a ~ then note that this parameter
 808          * is a candidate for ~username expansion.  NB this does not
 809          * apply to quoted parameters.
 810          */
 811         tilde_candidate = *p == '~';
 812         len = strcspn (p, " \t");
 813         if (p[len]) {
 814           p[len] = '\0';
 815           pend = &p[len+1];
 816         } else
 817           pend = &p[len];
 818       } else {
 819         fprintf (stderr, _("%s: internal error parsing string at '%s'\n"),
 820                  program_name, p);
 821         abort ();
 822       }
 823
 824       if (!tilde_candidate)
 825         argv[i] = p;
 826       else
 827         argv[i] = try_tilde_expansion (p);
 828       i++;
 829       p = pend;
 830
 831       if (*p)
 832         p += strspn (p, " \t");
 833     }
 834
 835     if (i == sizeof argv / sizeof argv[0]) {
 836       fprintf (stderr, _("%s: too many arguments\n"), program_name);
 837       if (exit_on_error) exit (EXIT_FAILURE);
 838       goto next_command;
 839     }
 840
 841     argv[i] = NULL;
 842
 843   got_command:
 844     if (issue_command (cmd, argv, pipe) == -1) {
 845       if (exit_on_error) exit (EXIT_FAILURE);
 846     }
 847
 848   next_command:;
 849   }
 850   if (prompt) printf ("\n");
 851 }
 852
 853 static void
 854 cmdline (char *argv[], int optind, int argc)
 855 {
 856   const char *cmd;
 857   char **params;
 858
 859   exit_on_error = 1;
 860
 861   if (optind >= argc) return;
 862
 863   cmd = argv[optind++];
 864   if (STREQ (cmd, ":")) {
 865     fprintf (stderr, _("%s: empty command on command line\n"), program_name);
 866     exit (EXIT_FAILURE);
 867   }
 868
 869   /* Allow -cmd on the command line to mean (temporarily) override
 870    * the normal exit on error (RHBZ#578407).
 871    */
 872   if (cmd[0] == '-') {
 873     exit_on_error = 0;
 874     cmd++;
 875   }
 876
 877   params = &argv[optind];
 878
 879   /* Search for end of command list or ":" ... */
 880   while (optind < argc && STRNEQ (argv[optind], ":"))
 881     optind++;
 882
 883   if (optind == argc) {
 884     if (issue_command (cmd, params, NULL) == -1 && exit_on_error)
 885         exit (EXIT_FAILURE);
 886   } else {
 887     argv[optind] = NULL;
 888     if (issue_command (cmd, params, NULL) == -1 && exit_on_error)
 889       exit (EXIT_FAILURE);
 890     cmdline (argv, optind+1, argc);
 891   }
 892 }
 893
 894 int
 895 issue_command (const char *cmd, char *argv[], const char *pipecmd)
 896 {
 897   int argc;
 898   int stdout_saved_fd = -1;
 899   int pid = 0;
 900   int i, r;
 901
 902   /* This counts the commands issued, starting at 1. */
 903   command_num++;
 904
 905   /* For | ... commands.  Annoyingly we can't use popen(3) here. */
 906   if (pipecmd) {
 907     int fd[2];
 908
 909     if (fflush (stdout) == EOF) {
 910       perror ("failed to flush standard output");
 911       return -1;
 912     }
 913     if (pipe (fd) < 0) {
 914       perror ("pipe failed");
 915       return -1;
 916     }
 917     pid = fork ();
 918     if (pid == -1) {
 919       perror ("fork");
 920       return -1;
 921     }
 922
 923     if (pid == 0) {             /* Child process. */
 924       close (fd[1]);
 925       if (dup2 (fd[0], 0) < 0) {
 926         perror ("dup2 of stdin failed");
 927         _exit (1);
 928       }
 929
 930       r = system (pipecmd);
 931       if (r == -1) {
 932         perror (pipecmd);
 933         _exit (1);
 934       }
 935       _exit (WEXITSTATUS (r));
 936     }
 937
 938     if ((stdout_saved_fd = dup (1)) < 0) {
 939       perror ("failed to dup stdout");
 940       return -1;
 941     }
 942     close (fd[0]);
 943     if (dup2 (fd[1], 1) < 0) {
 944       perror ("failed to dup stdout");
 945       close (stdout_saved_fd);
 946       return -1;
 947     }
 948     close (fd[1]);
 949   }
 950
 951   for (argc = 0; argv[argc] != NULL; ++argc)
 952     ;
 953
 954   /* If --remote was set, then send this command to a remote process. */
 955   if (remote_control)
 956     r = rc_remote (remote_control, cmd, argc, argv, exit_on_error);
 957
 958   /* Otherwise execute it locally. */
 959   else if (STRCASEEQ (cmd, "help")) {
 960     if (argc == 0)
 961       list_commands ();
 962     else
 963       display_command (argv[0]);
 964     r = 0;
 965   }
 966   else if (STRCASEEQ (cmd, "quit") ||
 967            STRCASEEQ (cmd, "exit") ||
 968            STRCASEEQ (cmd, "q")) {
 969     quit = 1;
 970     r = 0;
 971   }
 972   else if (STRCASEEQ (cmd, "alloc") ||
 973            STRCASEEQ (cmd, "allocate"))
 974     r = do_alloc (cmd, argc, argv);
 975   else if (STRCASEEQ (cmd, "echo"))
 976     r = do_echo (cmd, argc, argv);
 977   else if (STRCASEEQ (cmd, "edit") ||
 978            STRCASEEQ (cmd, "vi") ||
 979            STRCASEEQ (cmd, "emacs"))
 980     r = do_edit (cmd, argc, argv);
 981   else if (STRCASEEQ (cmd, "lcd"))
 982     r = do_lcd (cmd, argc, argv);
 983   else if (STRCASEEQ (cmd, "glob"))
 984     r = do_glob (cmd, argc, argv);
 985   else if (STRCASEEQ (cmd, "man") ||
 986            STRCASEEQ (cmd, "manual"))
 987     r = do_man (cmd, argc, argv);
 988   else if (STRCASEEQ (cmd, "more") ||
 989            STRCASEEQ (cmd, "less"))
 990     r = do_more (cmd, argc, argv);
 991   else if (STRCASEEQ (cmd, "reopen"))
 992     r = do_reopen (cmd, argc, argv);
 993   else if (STRCASEEQ (cmd, "sparse"))
 994     r = do_sparse (cmd, argc, argv);
 995   else if (STRCASEEQ (cmd, "time"))
 996     r = do_time (cmd, argc, argv);
 997   else
 998     r = run_action (cmd, argc, argv);
 999
1000   /* Always flush stdout after every command, so that messages, results
1001    * etc appear immediately.
1002    */
1003   if (fflush (stdout) == EOF) {
1004     perror ("failed to flush standard output");
1005     return -1;
1006   }
1007
1008   if (pipecmd) {
1009     close (1);
1010     if (dup2 (stdout_saved_fd, 1) < 0) {
1011       perror ("failed to dup2 standard output");
1012       r = -1;
1013     }
1014     close (stdout_saved_fd);
1015     if (waitpid (pid, NULL, 0) < 0) {
1016       perror ("waiting for command to complete");
1017       r = -1;
1018     }
1019   }
1020
1021   return r;
1022 }
1023
1024 void
1025 list_builtin_commands (void)
1026 {
1027   /* help, man and quit should appear at the top */
1028   printf ("%-20s %s\n",
1029           "help", _("display a list of commands or help on a command"));
1030   printf ("%-20s %s\n",
1031           "man", _("read the manual"));
1032   printf ("%-20s %s\n",
1033           "quit", _("quit guestfish"));
1034
1035   printf ("%-20s %s\n",
1036           "alloc", _("allocate an image"));
1037   printf ("%-20s %s\n",
1038           "echo", _("display a line of text"));
1039   printf ("%-20s %s\n",
1040           "edit", _("edit a file in the image"));
1041   printf ("%-20s %s\n",
1042           "lcd", _("local change directory"));
1043   printf ("%-20s %s\n",
1044           "glob", _("expand wildcards in command"));
1045   printf ("%-20s %s\n",
1046           "more", _("view a file in the pager"));
1047   printf ("%-20s %s\n",
1048           "reopen", _("close and reopen libguestfs handle"));
1049   printf ("%-20s %s\n",
1050           "sparse", _("allocate a sparse image file"));
1051   printf ("%-20s %s\n",
1052           "time", _("measure time taken to run command"));
1053
1054   /* actions are printed after this (see list_commands) */
1055 }
1056
1057 void
1058 display_builtin_command (const char *cmd)
1059 {
1060   /* help for actions is auto-generated, see display_command */
1061
1062   if (STRCASEEQ (cmd, "alloc") ||
1063       STRCASEEQ (cmd, "allocate"))
1064     printf (_("alloc - allocate an image\n"
1065               "     alloc <filename> <size>\n"
1066               "\n"
1067               "    This creates an empty (zeroed) file of the given size,\n"
1068               "    and then adds so it can be further examined.\n"
1069               "\n"
1070               "    For more advanced image creation, see qemu-img utility.\n"
1071               "\n"
1072               "    Size can be specified using standard suffixes, eg. '1M'.\n"
1073               ));
1074   else if (STRCASEEQ (cmd, "echo"))
1075     printf (_("echo - display a line of text\n"
1076               "     echo [<params> ...]\n"
1077               "\n"
1078               "    This echos the parameters to the terminal.\n"));
1079   else if (STRCASEEQ (cmd, "edit") ||
1080            STRCASEEQ (cmd, "vi") ||
1081            STRCASEEQ (cmd, "emacs"))
1082     printf (_("edit - edit a file in the image\n"
1083               "     edit <filename>\n"
1084               "\n"
1085               "    This is used to edit a file.\n"
1086               "\n"
1087               "    It is the equivalent of (and is implemented by)\n"
1088               "    running \"cat\", editing locally, and then \"write\".\n"
1089               "\n"
1090               "    Normally it uses $EDITOR, but if you use the aliases\n"
1091               "    \"vi\" or \"emacs\" you will get those editors.\n"
1092               "\n"
1093               "    NOTE: This will not work reliably for large files\n"
1094               "    (> 2 MB) or binary files containing \\0 bytes.\n"));
1095   else if (STRCASEEQ (cmd, "lcd"))
1096     printf (_("lcd - local change directory\n"
1097               "    lcd <directory>\n"
1098               "\n"
1099               "    Change guestfish's current directory. This command is\n"
1100               "    useful if you want to download files to a particular\n"
1101               "    place.\n"));
1102   else if (STRCASEEQ (cmd, "glob"))
1103     printf (_("glob - expand wildcards in command\n"
1104               "    glob <command> [<args> ...]\n"
1105               "\n"
1106               "    Glob runs <command> with wildcards expanded in any\n"
1107               "    command args.  Note that the command is run repeatedly\n"
1108               "    once for each expanded argument.\n"));
1109   else if (STRCASEEQ (cmd, "man") ||
1110            STRCASEEQ (cmd, "manual"))
1111     printf (_("man - read the manual\n"
1112               "    man\n"
1113               "\n"
1114               "    Opens the manual page for guestfish.\n"));
1115   else if (STRCASEEQ (cmd, "help"))
1116     printf (_("help - display a list of commands or help on a command\n"
1117               "     help cmd\n"
1118               "     help\n"));
1119   else if (STRCASEEQ (cmd, "more") ||
1120            STRCASEEQ (cmd, "less"))
1121     printf (_("more - view a file in the pager\n"
1122               "     more <filename>\n"
1123               "\n"
1124               "    This is used to view a file in the pager.\n"
1125               "\n"
1126               "    It is the equivalent of (and is implemented by)\n"
1127               "    running \"cat\" and using the pager.\n"
1128               "\n"
1129               "    Normally it uses $PAGER, but if you use the alias\n"
1130               "    \"less\" then it always uses \"less\".\n"
1131               "\n"
1132               "    NOTE: This will not work reliably for large files\n"
1133               "    (> 2 MB) or binary files containing \\0 bytes.\n"));
1134   else if (STRCASEEQ (cmd, "quit") ||
1135            STRCASEEQ (cmd, "exit") ||
1136            STRCASEEQ (cmd, "q"))
1137     printf (_("quit - quit guestfish\n"
1138               "     quit\n"));
1139   else if (STRCASEEQ (cmd, "reopen"))
1140     printf (_("reopen - close and reopen the libguestfs handle\n"
1141               "     reopen\n"
1142               "\n"
1143               "Close and reopen the libguestfs handle.  It is not necessary to use\n"
1144               "this normally, because the handle is closed properly when guestfish\n"
1145               "exits.  However this is occasionally useful for testing.\n"));
1146   else if (STRCASEEQ (cmd, "sparse"))
1147     printf (_("sparse - allocate a sparse image file\n"
1148               "     sparse <filename> <size>\n"
1149               "\n"
1150               "    This creates an empty sparse file of the given size,\n"
1151               "    and then adds so it can be further examined.\n"
1152               "\n"
1153               "    In all respects it works the same as the 'alloc'\n"
1154               "    command, except that the image file is allocated\n"
1155               "    sparsely, which means that disk blocks are not assigned\n"
1156               "    to the file until they are needed.  Sparse disk files\n"
1157               "    only use space when written to, but they are slower\n"
1158               "    and there is a danger you could run out of real disk\n"
1159               "    space during a write operation.\n"
1160               "\n"
1161               "    For more advanced image creation, see qemu-img utility.\n"
1162               "\n"
1163               "    Size can be specified using standard suffixes, eg. '1M'.\n"
1164               ));
1165   else if (STRCASEEQ (cmd, "time"))
1166     printf (_("time - measure time taken to run command\n"
1167               "    time <command> [<args> ...]\n"
1168               "\n"
1169               "    This runs <command> as usual, and prints the elapsed\n"
1170               "    time afterwards.\n"));
1171   else
1172     fprintf (stderr, _("%s: command not known, use -h to list all commands\n"),
1173              cmd);
1174 }
1175
1176 /* This is printed when the user types in an unknown command for the
1177  * first command issued.  A common case is the user doing:
1178  *   guestfish disk.img
1179  * expecting guestfish to open 'disk.img' (in fact, this tried to
1180  * run a command 'disk.img').
1181  */
1182 void
1183 extended_help_message (void)
1184 {
1185   fprintf (stderr,
1186            _("Did you mean to open a disk image?  guestfish -a disk.img\n"
1187              "For a list of commands:             guestfish -h\n"
1188              "For complete documentation:         man guestfish\n"));
1189 }
1190
1191 void
1192 free_strings (char **argv)
1193 {
1194   int argc;
1195
1196   for (argc = 0; argv[argc] != NULL; ++argc)
1197     free (argv[argc]);
1198   free (argv);
1199 }
1200
1201 int
1202 count_strings (char *const *argv)
1203 {
1204   int c;
1205
1206   for (c = 0; argv[c]; ++c)
1207     ;
1208   return c;
1209 }
1210
1211 void
1212 print_strings (char *const *argv)
1213 {
1214   int argc;
1215
1216   for (argc = 0; argv[argc] != NULL; ++argc)
1217     printf ("%s\n", argv[argc]);
1218 }
1219
1220 void
1221 print_table (char *const *argv)
1222 {
1223   int i;
1224
1225   for (i = 0; argv[i] != NULL; i += 2)
1226     printf ("%s: %s\n", argv[i], argv[i+1]);
1227 }
1228
1229 int
1230 is_true (const char *str)
1231 {
1232   return
1233     STRCASENEQ (str, "0") &&
1234     STRCASENEQ (str, "f") &&
1235     STRCASENEQ (str, "false") &&
1236     STRCASENEQ (str, "n") &&
1237     STRCASENEQ (str, "no");
1238 }
1239
1240 /* Free strings from a non-NULL terminated char** */
1241 static void
1242 free_n_strings (char **str, size_t len)
1243 {
1244   size_t i;
1245
1246   for (i = 0; i < len; i++) {
1247     free (str[i]);
1248   }
1249   free (str);
1250 }
1251
1252 char **
1253 parse_string_list (const char *str)
1254 {
1255   char **argv = NULL;
1256   size_t argv_len = 0;
1257
1258   /* Current position pointer */
1259   const char *p = str;
1260
1261   /* Token might be simple:
1262    *  Token
1263    * or be quoted:
1264    *  'This is a single token'
1265    * or contain embedded single-quoted sections:
1266    *  This' is a sing'l'e to'ken
1267    *
1268    * The latter may seem over-complicated, but it's what a normal shell does.
1269    * Not doing it risks surprising somebody.
1270    *
1271    * This outer loop is over complete tokens.
1272    */
1273   while (*p) {
1274     char *tok = NULL;
1275     size_t tok_len = 0;
1276
1277     /* Skip leading whitespace */
1278     p += strspn (p, " \t");
1279
1280     char in_quote = 0;
1281
1282     /* This loop is over token 'fragments'. A token can be in multiple bits if
1283      * it contains single quotes. We also treat both sides of an escaped quote
1284      * as separate fragments because we can't just copy it: we have to remove
1285      * the \.
1286      */
1287     while (*p && (!c_isblank (*p) || in_quote)) {
1288       const char *end = p;
1289
1290       /* Check if the fragment starts with a quote */
1291       if ('\'' == *p) {
1292         /* Toggle in_quote */
1293         in_quote = !in_quote;
1294
1295         /* Skip the quote */
1296         p++; end++;
1297       }
1298
1299       /* If we're in a quote, look for an end quote */
1300       if (in_quote) {
1301         end += strcspn (end, "'");
1302       }
1303
1304       /* Otherwise, look for whitespace or a quote */
1305       else {
1306         end += strcspn (end, " \t'");
1307       }
1308
1309       /* Grow the token to accommodate the fragment */
1310       size_t tok_end = tok_len;
1311       tok_len += end - p;
1312       char *tok_new = realloc (tok, tok_len + 1);
1313       if (NULL == tok_new) {
1314         perror ("realloc");
1315         free_n_strings (argv, argv_len);
1316         free (tok);
1317         exit (EXIT_FAILURE);
1318       }
1319       tok = tok_new;
1320
1321       /* Check if we stopped on an escaped quote */
1322       if ('\'' == *end && end != p && *(end-1) == '\\') {
1323         /* Add everything before \' to the token */
1324         memcpy (&tok[tok_end], p, end - p - 1);
1325
1326         /* Add the quote */
1327         tok[tok_len-1] = '\'';
1328
1329         /* Already processed the quote */
1330         p = end + 1;
1331       }
1332
1333       else {
1334         /* Add the whole fragment */
1335         memcpy (&tok[tok_end], p, end - p);
1336
1337         p = end;
1338       }
1339     }
1340
1341     /* We've reached the end of a token. We shouldn't still be in quotes. */
1342     if (in_quote) {
1343       fprintf (stderr, _("Runaway quote in string \"%s\"\n"), str);
1344
1345       free_n_strings (argv, argv_len);
1346
1347       return NULL;
1348     }
1349
1350     /* Add this token if there is one. There might not be if there was
1351      * whitespace at the end of the input string */
1352     if (tok) {
1353       /* Add the NULL terminator */
1354       tok[tok_len] = '\0';
1355
1356       /* Add the argument to the argument list */
1357       argv_len++;
1358       char **argv_new = realloc (argv, sizeof (*argv) * argv_len);
1359       if (NULL == argv_new) {
1360         perror ("realloc");
1361         free_n_strings (argv, argv_len-1);
1362         free (tok);
1363         exit (EXIT_FAILURE);
1364       }
1365       argv = argv_new;
1366
1367       argv[argv_len-1] = tok;
1368     }
1369   }
1370
1371   /* NULL terminate the argument list */
1372   argv_len++;
1373   char **argv_new = realloc (argv, sizeof (*argv) * argv_len);
1374   if (NULL == argv_new) {
1375     perror ("realloc");
1376     free_n_strings (argv, argv_len-1);
1377     exit (EXIT_FAILURE);
1378   }
1379   argv = argv_new;
1380
1381   argv[argv_len-1] = NULL;
1382
1383   return argv;
1384 }
1385
1386 #ifdef HAVE_LIBREADLINE
1387 static char histfile[1024];
1388 static int nr_history_lines = 0;
1389 #endif
1390
1391 static void
1392 initialize_readline (void)
1393 {
1394 #ifdef HAVE_LIBREADLINE
1395   const char *home;
1396
1397   home = getenv ("HOME");
1398   if (home) {
1399     snprintf (histfile, sizeof histfile, "%s/.guestfish", home);
1400     using_history ();
1401     (void) read_history (histfile);
1402   }
1403
1404   rl_readline_name = "guestfish";
1405   rl_attempted_completion_function = do_completion;
1406 #endif
1407 }
1408
1409 static void
1410 cleanup_readline (void)
1411 {
1412 #ifdef HAVE_LIBREADLINE
1413   int fd;
1414
1415   if (histfile[0] != '\0') {
1416     fd = open (histfile, O_WRONLY|O_CREAT, 0644);
1417     if (fd == -1) {
1418       perror (histfile);
1419       return;
1420     }
1421     close (fd);
1422
1423 #ifdef HAVE_APPEND_HISTORY
1424     (void) append_history (nr_history_lines, histfile);
1425 #else
1426     (void) write_history (histfile);
1427 #endif
1428   }
1429 #endif
1430 }
1431
1432 #ifdef HAVE_LIBREADLINE
1433 static void
1434 add_history_line (const char *line)
1435 {
1436   add_history (line);
1437   nr_history_lines++;
1438 }
1439 #endif
1440
1441 int
1442 xwrite (int fd, const void *v_buf, size_t len)
1443 {
1444   int r;
1445   const char *buf = v_buf;
1446
1447   while (len > 0) {
1448     r = write (fd, buf, len);
1449     if (r == -1) {
1450       perror ("write");
1451       return -1;
1452     }
1453     buf += r;
1454     len -= r;
1455   }
1456
1457   return 0;
1458 }
1459
1460 /* Resolve the special "win:..." form for Windows-specific paths.
1461  * This always returns a newly allocated string which is freed by the
1462  * caller function in "cmds.c".
1463  */
1464 char *
1465 resolve_win_path (const char *path)
1466 {
1467   char *ret;
1468   size_t i;
1469
1470   if (STRCASENEQLEN (path, "win:", 4)) {
1471     ret = strdup (path);
1472     if (ret == NULL)
1473       perror ("strdup");
1474     return ret;
1475   }
1476
1477   path += 4;
1478
1479   /* Drop drive letter, if it's "C:". */
1480   if (STRCASEEQLEN (path, "c:", 2))
1481     path += 2;
1482
1483   if (!*path) {
1484     ret = strdup ("/");
1485     if (ret == NULL)
1486       perror ("strdup");
1487     return ret;
1488   }
1489
1490   ret = strdup (path);
1491   if (ret == NULL) {
1492     perror ("strdup");
1493     return NULL;
1494   }
1495
1496   /* Blindly convert any backslashes into forward slashes.  Is this good? */
1497   for (i = 0; i < strlen (ret); ++i)
1498     if (ret[i] == '\\')
1499       ret[i] = '/';
1500
1501   char *t = guestfs_case_sensitive_path (g, ret);
1502   free (ret);
1503   ret = t;
1504
1505   return ret;
1506 }
1507
1508 /* Resolve the special FileIn paths ("-" or "-<<END" or filename).
1509  * The caller (cmds.c) will call free_file_in after the command has
1510  * run which should clean up resources.
1511  */
1512 static char *file_in_heredoc (const char *endmarker);
1513 static char *file_in_tmpfile = NULL;
1514
1515 char *
1516 file_in (const char *arg)
1517 {
1518   char *ret;
1519
1520   if (STREQ (arg, "-")) {
1521     ret = strdup ("/dev/stdin");
1522     if (!ret) {
1523       perror ("strdup");
1524       return NULL;
1525     }
1526   }
1527   else if (STRPREFIX (arg, "-<<")) {
1528     const char *endmarker = &arg[3];
1529     if (*endmarker == '\0') {
1530       fprintf (stderr, "%s: missing end marker in -<< expression\n",
1531                program_name);
1532       return NULL;
1533     }
1534     ret = file_in_heredoc (endmarker);
1535     if (ret == NULL)
1536       return NULL;
1537   }
1538   else {
1539     ret = strdup (arg);
1540     if (!ret) {
1541       perror ("strdup");
1542       return NULL;
1543     }
1544   }
1545
1546   return ret;
1547 }
1548
1549 static char *
1550 file_in_heredoc (const char *endmarker)
1551 {
1552   static const char template[] = "/tmp/heredocXXXXXX";
1553   file_in_tmpfile = strdup (template);
1554   if (file_in_tmpfile == NULL) {
1555     perror ("strdup");
1556     return NULL;
1557   }
1558
1559   int fd = mkstemp (file_in_tmpfile);
1560   if (fd == -1) {
1561     perror ("mkstemp");
1562     goto error1;
1563   }
1564
1565   size_t markerlen = strlen (endmarker);
1566
1567   char buffer[BUFSIZ];
1568   int write_error = 0;
1569   while (fgets (buffer, sizeof buffer, stdin) != NULL) {
1570     /* Look for "END"<EOF> or "END\n" in input. */
1571     size_t blen = strlen (buffer);
1572     if (STREQLEN (buffer, endmarker, markerlen) &&
1573         (blen == markerlen ||
1574          (blen == markerlen+1 && buffer[markerlen] == '\n')))
1575       goto found_end;
1576
1577     if (xwrite (fd, buffer, blen) == -1) {
1578       if (!write_error) perror ("write");
1579       write_error = 1;
1580       /* continue reading up to the end marker */
1581     }
1582   }
1583
1584   /* Reached EOF of stdin without finding the end marker, which
1585    * is likely to be an error.
1586    */
1587   fprintf (stderr, "%s: end of input reached without finding '%s'\n",
1588            program_name, endmarker);
1589   goto error2;
1590
1591  found_end:
1592   if (write_error) {
1593     close (fd);
1594     goto error2;
1595   }
1596
1597   if (close (fd) == -1) {
1598     perror ("close");
1599     goto error2;
1600   }
1601
1602   return file_in_tmpfile;
1603
1604  error2:
1605   unlink (file_in_tmpfile);
1606
1607  error1:
1608   free (file_in_tmpfile);
1609   file_in_tmpfile = NULL;
1610   return NULL;
1611 }
1612
1613 void
1614 free_file_in (char *s)
1615 {
1616   if (file_in_tmpfile) {
1617     if (unlink (file_in_tmpfile) == -1)
1618       perror (file_in_tmpfile);
1619     file_in_tmpfile = NULL;
1620   }
1621
1622   /* Free the device or file name which was strdup'd in file_in().
1623    * Note it's not immediately clear, but for -<< heredocs,
1624    * s == file_in_tmpfile, so this frees up that buffer.
1625    */
1626   free (s);
1627 }
1628
1629 /* Resolve the special FileOut paths ("-" or filename).
1630  * The caller (cmds.c) will call free (str) after the command has run.
1631  */
1632 char *
1633 file_out (const char *arg)
1634 {
1635   char *ret;
1636
1637   if (STREQ (arg, "-"))
1638     ret = strdup ("/dev/stdout");
1639   else
1640     ret = strdup (arg);
1641
1642   if (!ret) {
1643     perror ("strdup");
1644     return NULL;
1645   }
1646   return ret;
1647 }
1648
1649 static void
1650 print_shell_quote (FILE *stream, const char *str)
1651 {
1652 #define SAFE(c) (c_isalnum((c)) ||                                      \
1653                  (c) == '/' || (c) == '-' || (c) == '_' || (c) == '.')
1654   int i;
1655
1656   for (i = 0; str[i]; ++i) {
1657     if (!SAFE(str[i]))
1658       putc ('\\', stream);
1659     putc (str[i], stream);
1660   }
1661 }