daemon/link.c

   1 /* libguestfs - the guestfsd daemon
   2  * Copyright (C) 2009 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 <fcntl.h>
  25 #include <unistd.h>
  26 #include <limits.h>
  27
  28 #include "daemon.h"
  29 #include "actions.h"
  30
  31 char *
  32 do_readlink (const char *path)
  33 {
  34   ssize_t r;
  35   char *ret;
  36   char link[PATH_MAX];
  37
  38   CHROOT_IN;
  39   r = readlink (path, link, sizeof link);
  40   CHROOT_OUT;
  41   if (r == -1) {
  42     reply_with_perror ("readlink");
  43     return NULL;
  44   }
  45
  46   ret = strndup (link, r);
  47   if (ret == NULL) {
  48     reply_with_perror ("strndup");
  49     return NULL;
  50   }
  51
  52   return ret;                   /* caller frees */
  53 }
  54
  55 char **
  56 do_readlinklist (const char *path, char *const *names)
  57 {
  58   int fd_cwd;
  59   size_t i;
  60   ssize_t r;
  61   char link[PATH_MAX];
  62   const char *str;
  63   char **ret = NULL;
  64   int size = 0, alloc = 0;
  65
  66   CHROOT_IN;
  67   fd_cwd = open (path, O_RDONLY | O_DIRECTORY);
  68   CHROOT_OUT;
  69
  70   if (fd_cwd == -1) {
  71     reply_with_perror ("open: %s", path);
  72     return NULL;
  73   }
  74
  75   for (i = 0; names[i] != NULL; ++i) {
  76     r = readlinkat (fd_cwd, names[i], link, sizeof link);
  77     if (r >= PATH_MAX) {
  78       reply_with_perror ("readlinkat: returned link is too long");
  79       close (fd_cwd);
  80       return NULL;
  81     }
  82     /* Because of the way this function is intended to be used,
  83      * we actually expect to see errors here, and they are not fatal.
  84      */
  85     if (r >= 0) {
  86       link[r] = '\0';
  87       str = link;
  88     } else
  89       str = "";
  90     if (add_string (&ret, &size, &alloc, str) == -1) {
  91       close (fd_cwd);
  92       return NULL;
  93     }
  94   }
  95
  96   close (fd_cwd);
  97
  98   if (add_string (&ret, &size, &alloc, NULL) == -1)
  99     return NULL;
 100
 101   return ret;
 102 }
 103
 104 static int
 105 _link (const char *flag, int symbolic, const char *target, const char *linkname)
 106 {
 107   int r;
 108   char *err;
 109   char *buf_linkname;
 110   char *buf_target;
 111
 112   /* Prefix linkname with sysroot. */
 113   buf_linkname = sysroot_path (linkname);
 114   if (!buf_linkname) {
 115     reply_with_perror ("malloc");
 116     return -1;
 117   }
 118
 119   /* Only prefix target if it's _not_ a symbolic link, and if
 120    * the target is absolute.  Note that the resulting link will
 121    * always be "broken" from the p.o.v. of the appliance, ie:
 122    * /a -> /b but the path as seen here is /sysroot/b
 123    */
 124   buf_target = NULL;
 125   if (!symbolic && target[0] == '/') {
 126     buf_target = sysroot_path (target);
 127     if (!buf_target) {
 128       reply_with_perror ("malloc");
 129       free (buf_linkname);
 130       return -1;
 131     }
 132   }
 133
 134   if (flag)
 135     r = command (NULL, &err,
 136                  "ln", flag, "--", /* target could begin with '-' */
 137                  buf_target ? : target, buf_linkname, NULL);
 138   else
 139     r = command (NULL, &err,
 140                  "ln", "--",
 141                  buf_target ? : target, buf_linkname, NULL);
 142   free (buf_linkname);
 143   free (buf_target);
 144   if (r == -1) {
 145     reply_with_error ("ln%s%s: %s: %s: %s",
 146                       flag ? " " : "",
 147                       flag ? : "",
 148                       target, linkname, err);
 149     free (err);
 150     return -1;
 151   }
 152
 153   free (err);
 154
 155   return 0;
 156 }
 157
 158 int
 159 do_ln (const char *target, const char *linkname)
 160 {
 161   return _link (NULL, 0, target, linkname);
 162 }
 163
 164 int
 165 do_ln_f (const char *target, const char *linkname)
 166 {
 167   return _link ("-f", 0, target, linkname);
 168 }
 169
 170 int
 171 do_ln_s (const char *target, const char *linkname)
 172 {
 173   return _link ("-s", 1, target, linkname);
 174 }
 175
 176 int
 177 do_ln_sf (const char *target, const char *linkname)
 178 {
 179   return _link ("-sf", 1, target, linkname);
 180 }