Command line and interactive shell parsing, prompts etc.

2 files changed, 210 insertions, 17 deletions

diff --git a/fish/fish.c b/fish/fish.c
index 85543004..a2525162 100644
--- a/fish/fish.c
+++ b/fish/fish.c
@@ -1,4 +1,4 @@
-/* guestfish - the filesystem shell
+/* guestfish - the filesystem interactive shell
  * Copyright (C) 2009 Red Hat Inc. 
  *
  * This program is free software; you can redistribute it and/or modify
@@ -28,6 +28,37 @@
 
 #include "fish.h"
 
+struct mp {
+  struct mp *next;
+  char *device;
+  char *mountpoint;
+};
+
+static void mount_mps (struct mp *mp);
+static void interactive (void);
+static void shell_script (void);
+static void script (int prompt);
+static void cmdline (char *argv[], int optind, int argc);
+static void issue_command (const char *cmd, char *argv[]);
+
+/* Currently open libguestfs handle. */
+guestfs_h *g;
+int g_launched = 0;
+
+int quit = 0;
+
+void
+launch (void)
+{
+  if (!g_launched) {
+    if (guestfs_launch (g) == -1)
+      exit (1);
+    if (guestfs_wait_ready (g) == -1)
+      exit (1);
+    g_launched = 1;
+  }
+}
+
 static void
 usage (void)
 {
@@ -36,7 +67,7 @@ usage (void)
 	   "guestfish lets you edit virtual machine filesystems\n"
 	   "Copyright (C) 2009 Red Hat Inc.\n"
 	   "Usage:\n"
-	   "  guestfish [--options] [cmd]\n"
+	   "  guestfish [--options] cmd [: cmd ...]\n"
 	   "or for interactive use:\n"
 	   "  guestfish\n"
 	   "or from a shell script:\n"
@@ -47,25 +78,53 @@ usage (void)
 	   "Options:\n"
 	   "  -h|--cmd-help       List available commands\n"
 	   "  -h|--cmd-help cmd   Display detailed help on 'cmd'\n"
-	   "For more information, see the manual page guestfish(1).\n");
+	   "  -a image            Add image\n"
+	   "  -m dev[:mnt]        Mount dev on mnt (if omitted, /)\n"
+	   /*"  --ro|-r             All mounts are read-only\n"*/
+	   "  -v|--verbose        Verbose messages\n"
+	   "For more information, see the manpage guestfish(1).\n");
 }
 
 int
 main (int argc, char *argv[])
 {
-  static const char *options = "h::?";
+  static const char *options = "a:h::m:v?";
   static struct option long_options[] = {
+    { "add", 1, 0, 'a' },
     { "cmd-help", 2, 0, 'h' },
     { "help", 0, 0, '?' },
+    { "mount", 1, 0, 'm' },
+    { "verbose", 0, 0, 'v' },
     { 0, 0, 0, 0 }
   };
+  struct mp *mps = NULL;
+  struct mp *mp;
+  char *p;
   int c;
 
+  /* guestfs_create is meant to be a lightweight operation, so
+   * it's OK to do it early here.
+   */
+  g = guestfs_create ();
+  if (g == NULL) {
+    fprintf (stderr, "guestfs_create: failed to create handle\n");
+    exit (1);
+  }
+
   for (;;) {
     c = getopt_long (argc, argv, options, long_options, NULL);
     if (c == -1) break;
 
     switch (c) {
+    case 'a':
+      if (access (optarg, R_OK) != 0) {
+	perror (optarg);
+	exit (1);
+      }
+      if (guestfs_add_drive (g, optarg) == -1)
+	exit (1);
+      break;
+
     case 'h':
       if (optarg)
 	display_command (optarg);
@@ -75,6 +134,23 @@ main (int argc, char *argv[])
 	list_commands ();
       exit (0);
 
+    case 'm':
+      mp = malloc (sizeof (struct mp));
+      if (!mp) {
+	perror ("malloc");
+	exit (1);
+      }
+      p = strchr (optarg, ':');
+      if (p) {
+	*p = '\0';
+	mp->mountpoint = p+1;
+      } else
+	mp->mountpoint = "/";
+      mp->device = optarg;
+      mp->next = mps;
+      mps = mp->next;
+      break;
+
     case '?':
       usage ();
       exit (0);
@@ -85,21 +161,21 @@ main (int argc, char *argv[])
     }
   }
 
-  if (optind < argc) {
-    usage ();
-    exit (1);
+  /* If we've got mountpoints, we must launch the guest and mount them. */
+  if (mps != NULL) {
+    launch ();
+    mount_mps (mps);
   }
 
-
-
-
-
-
-
-
-
-
-
+  /* Interactive, shell script, or command(s) on the command line? */
+  if (optind >= argc) {
+    if (isatty (0))
+      interactive ();
+    else
+      shell_script ();
+  }
+  else
+    cmdline (argv, optind, argc);
 
   exit (0);
 }
@@ -123,3 +199,115 @@ pod2text (const char *heading, const char *str)
   fputs (str, fp);
   pclose (fp);
 }
+
+/* List is built in reverse order, so mount them in reverse order. */
+static void
+mount_mps (struct mp *mp)
+{
+  if (mp) {
+    mount_mps (mp->next);
+    if (guestfs_mount (g, mp->device, mp->mountpoint) == -1)
+      exit (1);
+  }
+}
+
+static void
+interactive (void)
+{
+  script (1);
+}
+
+static void
+shell_script (void)
+{
+  script (0);
+}
+
+static void
+script (int prompt)
+{
+  char buf[8192];
+  char *cmd;
+  char *argv[64];
+  int len, i;
+
+  if (prompt)
+    printf ("Welcome to guestfish, the libguestfs filesystem interactive shell for\n"
+	    "editing virtual machine filesystems.\n"
+	    "\n"
+	    "Type: 'help' for help with commands\n"
+	    "      'quit' to quit the shell\n"
+	    "\n");
+
+  while (!quit) {
+    if (prompt) printf ("><fs> ");
+    if (fgets (buf, sizeof buf, stdin) == NULL) {
+      quit = 1;
+      break;
+    }
+
+    len = strlen (buf);
+    if (len > 0 && buf[len-1] == '\n') buf[len-1] = '\0';
+
+    /* Split the buffer up at whitespace. */
+    cmd = strtok (buf, " \t");
+    if (cmd == NULL)
+      continue;
+
+    i = 0;
+    while (i < sizeof argv / sizeof argv[0] &&
+	   (argv[i] = strtok (NULL, " \t")) != NULL)
+      i++;
+    if (i == sizeof argv / sizeof argv[0]) {
+      fprintf (stderr, "guestfish: too many arguments in command\n");
+      exit (1);
+    }
+
+    issue_command (cmd, argv);
+  }
+  if (prompt) printf ("\n");
+}
+
+static void
+cmdline (char *argv[], int optind, int argc)
+{
+  const char *cmd;
+  char **params;
+
+  if (optind >= argc) return;
+
+  cmd = argv[optind++];
+  if (strcmp (cmd, ":") == 0) {
+    fprintf (stderr, "guestfish: empty command on command line\n");
+    exit (1);
+  }
+  params = &argv[optind];
+
+  /* Search for end of command list or ":" ... */
+  while (optind < argc && strcmp (argv[optind], ":") != 0)
+    optind++;
+
+  if (optind == argc)
+    issue_command (cmd, params);
+  else {
+    argv[optind] = NULL;
+    issue_command (cmd, params);
+    cmdline (argv, optind+1, argc);
+  }
+}
+
+static void
+issue_command (const char *cmd, char *argv[])
+{
+  int i;
+
+  fprintf (stderr, "cmd = %s", cmd);
+  for (i = 0; argv[i] != NULL; ++i)
+    fprintf (stderr, ", arg[%d]=%s", i, argv[i]);
+  fprintf (stderr, "\n");
+
+
+
+
+
+}
diff --git a/fish/fish.h b/fish/fish.h
index 3e811703..09a9ef27 100644
--- a/fish/fish.h
+++ b/fish/fish.h
@@ -19,7 +19,12 @@
 #ifndef FISH_H
 #define FISH_H
 
+#include <guestfs.h>
+
 /* in fish.c */
+extern guestfs_h *g;
+extern int g_launched;
+extern int quit;
 extern void pod2text (const char *heading, const char *body);
 
 /* in cmds.c (auto-generated) */