1 files changed, 307 insertions, 0 deletions
diff --git a/fish/copy.c b/fish/copy.c
new file mode 100644
index 00000000..789e430d
--- /dev/null
+++ b/fish/copy.c
@@ -0,0 +1,307 @@
+/* guestfish - the filesystem interactive shell
+ * Copyright (C) 2010 Red Hat Inc.
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ */
+
+#include <config.h>
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <assert.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <sys/wait.h>
+
+#include "fish.h"
+
+static int make_tar_from_local (const char *local);
+static int make_tar_output (const char *local, const char *basename);
+static int split_path (char *buf, size_t buf_size, const char *path, char **dirname, char **basename);
+
+int
+do_copy_in (const char *cmd, int argc, char *argv[])
+{
+  if (argc < 2) {
+    fprintf (stderr,
+             _("use 'copy-in <local> [<local>...] <remotedir>' to copy files into the image\n"));
+    return -1;
+  }
+
+  /* Remote directory is always the last arg. */
+  const char *remote = argv[argc-1];
+  int nr_locals = argc-1;
+
+  int remote_is_dir = guestfs_is_dir (g, remote);
+  if (remote_is_dir == -1)
+    return -1;
+
+  if (!remote_is_dir) {
+    fprintf (stderr, _("copy-in: target '%s' is not a directory\n"), remote);
+    return -1;
+  }
+
+  /* Upload each local one at a time using tar-in. */
+  int i;
+  for (i = 0; i < nr_locals; ++i) {
+    int fd = make_tar_from_local (argv[i]);
+    if (fd == -1)
+      return -1;
+
+    char fdbuf[64];
+    snprintf (fdbuf, sizeof fdbuf, "/dev/fd/%d", fd);
+
+    int r = guestfs_tar_in (g, fdbuf, remote);
+
+    if (close (fd) == -1) {
+      perror ("close (tar-from-local subprocess)");
+      r = -1;
+    }
+
+    int status;
+    if (wait (&status) == -1) {
+      perror ("wait (tar-from-local subprocess)");
+      return -1;
+    }
+    if (!(WIFEXITED (status) && WEXITSTATUS (status) == 0))
+      return -1;
+
+    if (r == -1)
+      return -1;
+  }
+
+  return 0;
+}
+
+static void tar_create (const char *dir, const char *path) __attribute__((noreturn));
+
+/* This creates a subprocess which feeds a tar file back to the
+ * main guestfish process.
+ */
+static int
+make_tar_from_local (const char *local)
+{
+  int fd[2];
+
+  if (pipe (fd) == -1) {
+    perror ("pipe");
+    return -1;
+  }
+
+  pid_t pid = fork ();
+  if (pid == -1) {
+    perror ("fork");
+    return -1;
+  }
+
+  if (pid > 0) {                /* Parent */
+    close (fd[1]);
+    return fd[0];
+  }
+
+  /* Child. */
+  close (fd[0]);
+  dup2 (fd[1], 1);
+  close (fd[1]);
+
+  char buf[PATH_MAX];
+  char *dirname, *basename;
+  if (split_path (buf, sizeof buf, local, &dirname, &basename) == -1)
+    _exit (EXIT_FAILURE);
+
+  tar_create (dirname, basename);
+}
+
+/* Split path into directory name and base name, using the buffer
+ * provided as a working area.  If there is no directory name
+ * (eg. path == "/") then this can return dirname as NULL.
+ */
+static int
+split_path (char *buf, size_t buf_size,
+            const char *path, char **dirname, char **basename)
+{
+  size_t len = strlen (path);
+  if (len == 0 || len > buf_size - 1) {
+    fprintf (stderr, _("error: argument is zero length or longer than maximum permitted\n"));
+    return -1;
+  }
+
+  strcpy (buf, path);
+
+  if (len >= 2 && buf[len-1] == '/') {
+    buf[len-1] = '\0';
+    len--;
+  }
+
+  char *p = strrchr (buf, '/');
+  if (p && p > buf) {
+    *p = '\0';
+    p++;
+    if (dirname) *dirname = buf;
+    if (basename) *basename = p;
+  } else {
+    if (dirname) *dirname = NULL;
+    if (basename) *basename = buf;
+  }
+
+  return 0;
+}
+
+static void
+tar_create (const char *dir, const char *path)
+{
+  if (dir)
+    execlp ("tar", "tar", "-C", dir, "-cf", "-", path, NULL);
+  else
+    execlp ("tar", "tar", "-cf", "-", path, NULL);
+
+  perror ("execlp: tar");
+  _exit (EXIT_FAILURE);
+}
+
+int
+do_copy_out (const char *cmd, int argc, char *argv[])
+{
+  if (argc < 2) {
+    fprintf (stderr,
+             _("use 'copy-out <remote> [<remote>...] <localdir>' to copy files out of the image\n"));
+    return -1;
+  }
+
+  /* Local directory is always the last arg. */
+  const char *local = argv[argc-1];
+  int nr_remotes = argc-1;
+
+  struct stat statbuf;
+  if (stat (local, &statbuf) == -1 ||
+      ! (S_ISDIR (statbuf.st_mode))) {
+    fprintf (stderr, _("copy-in: target '%s' is not a directory\n"), local);
+    return -1;
+  }
+
+  /* Download each remote one at a time using tar-out. */
+  int i, r;
+  for (i = 0; i < nr_remotes; ++i) {
+    /* If the remote is a file, download it.  If it's a directory,
+     * create the directory in local first before using tar-out.
+     */
+    r = guestfs_is_file (g, argv[i]);
+    if (r == -1)
+      return -1;
+    if (r == 1) {               /* is file */
+      char buf[PATH_MAX];
+      char *basename;
+      if (split_path (buf, sizeof buf, argv[i], NULL, &basename) == -1)
+        return -1;
+
+      char filename[PATH_MAX];
+      snprintf (filename, sizeof filename, "%s/%s", local, basename);
+      if (guestfs_download (g, argv[i], filename) == -1)
+        return -1;
+    }
+    else {                      /* not a regular file */
+      r = guestfs_is_dir (g, argv[i]);
+      if (r == -1)
+        return -1;
+
+      if (r == 0) {
+        fprintf (stderr, _("copy-out: '%s' is not a file or directory\n"),
+                 argv[i]);
+        return -1;
+      }
+
+      char buf[PATH_MAX];
+      char *basename;
+      if (split_path (buf, sizeof buf, argv[i], NULL, &basename) == -1)
+        return -1;
+
+      int fd = make_tar_output (local, basename);
+      if (fd == -1)
+        return -1;
+
+      char fdbuf[64];
+      snprintf (fdbuf, sizeof fdbuf, "/dev/fd/%d", fd);
+
+      int r = guestfs_tar_out (g, argv[i], fdbuf);
+
+      if (close (fd) == -1) {
+        perror ("close (tar-output subprocess)");
+        r = -1;
+      }
+
+      int status;
+      if (wait (&status) == -1) {
+        perror ("wait (tar-output subprocess)");
+        return -1;
+      }
+      if (!(WIFEXITED (status) && WEXITSTATUS (status) == 0))
+        return -1;
+
+      if (r == -1)
+        return -1;
+    }
+  }
+
+  return 0;
+}
+
+/* This creates a subprocess which takes a tar file from the main
+ * guestfish process and unpacks it into the 'local/basename'
+ * directory.
+ */
+static int
+make_tar_output (const char *local, const char *basename)
+{
+  int fd[2];
+
+  if (pipe (fd) == -1) {
+    perror ("pipe");
+    return -1;
+  }
+
+  pid_t pid = fork ();
+  if (pid == -1) {
+    perror ("fork");
+    return -1;
+  }
+
+  if (pid > 0) {                /* Parent */
+    close (fd[0]);
+    return fd[1];
+  }
+
+  /* Child. */
+  close (fd[1]);
+  dup2 (fd[0], 0);
+  close (fd[0]);
+
+  if (chdir (local) == -1) {
+    perror (local);
+    _exit (EXIT_FAILURE);
+  }
+
+  mkdir (basename, 0777);
+
+  if (chdir (basename) == -1) {
+    perror (basename);
+    _exit (EXIT_FAILURE);
+  }
+
+  execlp ("tar", "tar", "-xf", "-", NULL);
+  perror ("execlp: tar");
+  _exit (EXIT_FAILURE);
+}