fish/copy.c

   1 /* guestfish - the filesystem interactive shell
   2  * Copyright (C) 2010-2011 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 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 <limits.h>
  26 #include <sys/types.h>
  27 #include <sys/stat.h>
  28 #include <sys/wait.h>
  29
  30 #include "fish.h"
  31
  32 static int make_tar_from_local (const char *local);
  33 static int make_tar_output (const char *local, const char *basename);
  34 static int split_path (char *buf, size_t buf_size, const char *path, const char **dirname, const char **basename);
  35
  36 int
  37 run_copy_in (const char *cmd, size_t argc, char *argv[])
  38 {
  39   if (argc < 2) {
  40     fprintf (stderr,
  41              _("use 'copy-in <local> [<local>...] <remotedir>' to copy files into the image\n"));
  42     return -1;
  43   }
  44
  45   /* Remote directory is always the last arg. */
  46   char *remote = argv[argc-1];
  47
  48   /* Allow win: prefix on remote. */
  49   remote = win_prefix (remote);
  50   if (remote == NULL)
  51     return -1;
  52
  53   int nr_locals = argc-1;
  54
  55   int remote_is_dir = guestfs_is_dir (g, remote);
  56   if (remote_is_dir == -1) {
  57     free (remote);
  58     return -1;
  59   }
  60
  61   if (!remote_is_dir) {
  62     fprintf (stderr, _("copy-in: target '%s' is not a directory\n"), remote);
  63     free (remote);
  64     return -1;
  65   }
  66
  67   /* Upload each local one at a time using tar-in. */
  68   int i;
  69   for (i = 0; i < nr_locals; ++i) {
  70     int fd = make_tar_from_local (argv[i]);
  71     if (fd == -1) {
  72       free (remote);
  73       return -1;
  74     }
  75
  76     char fdbuf[64];
  77     snprintf (fdbuf, sizeof fdbuf, "/dev/fd/%d", fd);
  78
  79     int r = guestfs_tar_in (g, fdbuf, remote);
  80
  81     if (close (fd) == -1) {
  82       perror ("close (tar-from-local subprocess)");
  83       r = -1;
  84     }
  85
  86     int status;
  87     if (wait (&status) == -1) {
  88       perror ("wait (tar-from-local subprocess)");
  89       free (remote);
  90       return -1;
  91     }
  92     if (!(WIFEXITED (status) && WEXITSTATUS (status) == 0)) {
  93       free (remote);
  94       return -1;
  95     }
  96
  97     if (r == -1) {
  98       free (remote);
  99       return -1;
 100     }
 101   }
 102
 103   free (remote);
 104
 105   return 0;
 106 }
 107
 108 static void tar_create (const char *dir, const char *path) __attribute__((noreturn));
 109
 110 /* This creates a subprocess which feeds a tar file back to the
 111  * main guestfish process.
 112  */
 113 static int
 114 make_tar_from_local (const char *local)
 115 {
 116   int fd[2];
 117
 118   if (pipe (fd) == -1) {
 119     perror ("pipe");
 120     return -1;
 121   }
 122
 123   pid_t pid = fork ();
 124   if (pid == -1) {
 125     perror ("fork");
 126     return -1;
 127   }
 128
 129   if (pid > 0) {                /* Parent */
 130     close (fd[1]);
 131     return fd[0];
 132   }
 133
 134   /* Child. */
 135   close (fd[0]);
 136   dup2 (fd[1], 1);
 137   close (fd[1]);
 138
 139   char buf[PATH_MAX];
 140   const char *dirname, *basename;
 141   if (split_path (buf, sizeof buf, local, &dirname, &basename) == -1)
 142     _exit (EXIT_FAILURE);
 143
 144   tar_create (dirname, basename);
 145 }
 146
 147 /* Split path into directory name and base name, using the buffer
 148  * provided as a working area.  If there is no directory name
 149  * (eg. path == "/") then this can return dirname as NULL.
 150  */
 151 static int
 152 split_path (char *buf, size_t buf_size,
 153             const char *path, const char **dirname, const char **basename)
 154 {
 155   size_t len = strlen (path);
 156   if (len == 0 || len > buf_size - 1) {
 157     fprintf (stderr, _("error: argument is zero length or longer than maximum permitted\n"));
 158     return -1;
 159   }
 160
 161   strcpy (buf, path);
 162
 163   if (len >= 2 && buf[len-1] == '/') {
 164     buf[len-1] = '\0';
 165     len--;
 166   }
 167
 168   char *p = strrchr (buf, '/');
 169   if (p && p > buf) {           /* "foo/bar" */
 170     *p = '\0';
 171     p++;
 172     if (dirname) *dirname = buf;
 173     if (basename) *basename = p;
 174   } else if (p && p == buf) {   /* "/foo" */
 175     if (dirname) *dirname = "/";
 176     if (basename) *basename = buf+1;
 177   } else {
 178     if (dirname) *dirname = NULL;
 179     if (basename) *basename = buf;
 180   }
 181
 182   return 0;
 183 }
 184
 185 static void
 186 tar_create (const char *dir, const char *path)
 187 {
 188   if (dir)
 189     execlp ("tar", "tar", "-C", dir, "-cf", "-", path, NULL);
 190   else
 191     execlp ("tar", "tar", "-cf", "-", path, NULL);
 192
 193   perror ("execlp: tar");
 194   _exit (EXIT_FAILURE);
 195 }
 196
 197 int
 198 run_copy_out (const char *cmd, size_t argc, char *argv[])
 199 {
 200   if (argc < 2) {
 201     fprintf (stderr,
 202              _("use 'copy-out <remote> [<remote>...] <localdir>' to copy files out of the image\n"));
 203     return -1;
 204   }
 205
 206   /* Local directory is always the last arg. */
 207   const char *local = argv[argc-1];
 208   int nr_remotes = argc-1;
 209
 210   struct stat statbuf;
 211   if (stat (local, &statbuf) == -1 ||
 212       ! (S_ISDIR (statbuf.st_mode))) {
 213     fprintf (stderr, _("copy-out: target '%s' is not a directory\n"), local);
 214     return -1;
 215   }
 216
 217   /* Download each remote one at a time using tar-out. */
 218   int i, r;
 219   for (i = 0; i < nr_remotes; ++i) {
 220     char *remote = argv[i];
 221
 222     /* Allow win:... prefix on remotes. */
 223     remote = win_prefix (remote);
 224     if (remote == NULL)
 225       return -1;
 226
 227     /* If the remote is a file, download it.  If it's a directory,
 228      * create the directory in local first before using tar-out.
 229      */
 230     r = guestfs_is_file (g, remote);
 231     if (r == -1) {
 232       free (remote);
 233       return -1;
 234     }
 235     if (r == 1) {               /* is file */
 236       char buf[PATH_MAX];
 237       const char *basename;
 238       if (split_path (buf, sizeof buf, remote, NULL, &basename) == -1) {
 239         free (remote);
 240         return -1;
 241       }
 242
 243       char filename[PATH_MAX];
 244       snprintf (filename, sizeof filename, "%s/%s", local, basename);
 245       if (guestfs_download (g, remote, filename) == -1) {
 246         free (remote);
 247         return -1;
 248       }
 249     }
 250     else {                      /* not a regular file */
 251       r = guestfs_is_dir (g, remote);
 252       if (r == -1) {
 253         free (remote);
 254         return -1;
 255       }
 256
 257       if (r == 0) {
 258         fprintf (stderr, _("copy-out: '%s' is not a file or directory\n"),
 259                  remote);
 260         free (remote);
 261         return -1;
 262       }
 263
 264       char buf[PATH_MAX];
 265       const char *basename;
 266       if (split_path (buf, sizeof buf, remote, NULL, &basename) == -1) {
 267         free (remote);
 268         return -1;
 269       }
 270
 271       int fd = make_tar_output (local, basename);
 272       if (fd == -1) {
 273         free (remote);
 274         return -1;
 275       }
 276
 277       char fdbuf[64];
 278       snprintf (fdbuf, sizeof fdbuf, "/dev/fd/%d", fd);
 279
 280       int r = guestfs_tar_out (g, remote, fdbuf);
 281
 282       if (close (fd) == -1) {
 283         perror ("close (tar-output subprocess)");
 284         free (remote);
 285         r = -1;
 286       }
 287
 288       int status;
 289       if (wait (&status) == -1) {
 290         perror ("wait (tar-output subprocess)");
 291         free (remote);
 292         return -1;
 293       }
 294       if (!(WIFEXITED (status) && WEXITSTATUS (status) == 0)) {
 295         free (remote);
 296         return -1;
 297       }
 298
 299       if (r == -1) {
 300         free (remote);
 301         return -1;
 302       }
 303     }
 304
 305     free (remote);
 306   }
 307
 308   return 0;
 309 }
 310
 311 /* This creates a subprocess which takes a tar file from the main
 312  * guestfish process and unpacks it into the 'local/basename'
 313  * directory.
 314  */
 315 static int
 316 make_tar_output (const char *local, const char *basename)
 317 {
 318   int fd[2];
 319
 320   if (pipe (fd) == -1) {
 321     perror ("pipe");
 322     return -1;
 323   }
 324
 325   pid_t pid = fork ();
 326   if (pid == -1) {
 327     perror ("fork");
 328     return -1;
 329   }
 330
 331   if (pid > 0) {                /* Parent */
 332     close (fd[0]);
 333     return fd[1];
 334   }
 335
 336   /* Child. */
 337   close (fd[1]);
 338   dup2 (fd[0], 0);
 339   close (fd[0]);
 340
 341   if (chdir (local) == -1) {
 342     perror (local);
 343     _exit (EXIT_FAILURE);
 344   }
 345
 346   mkdir (basename, 0777);
 347
 348   if (chdir (basename) == -1) {
 349     perror (basename);
 350     _exit (EXIT_FAILURE);
 351   }
 352
 353   execlp ("tar", "tar", "-xf", "-", NULL);
 354   perror ("execlp: tar");
 355   _exit (EXIT_FAILURE);
 356 }