daemon/ext2.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 <unistd.h>
  24 #include <ctype.h>
  25
  26 #include "../src/guestfs_protocol.h"
  27 #include "daemon.h"
  28 #include "actions.h"
  29
  30 char **
  31 do_tune2fs_l (const char *device)
  32 {
  33   int r;
  34   char *out, *err;
  35   char *p, *pend, *colon;
  36   char **ret = NULL;
  37   int size = 0, alloc = 0;
  38
  39   IS_DEVICE (device, NULL);
  40
  41   r = command (&out, &err, "/sbin/tune2fs", "-l", device, NULL);
  42   if (r == -1) {
  43     reply_with_error ("tune2fs: %s", err);
  44     free (err);
  45     free (out);
  46     return NULL;
  47   }
  48   free (err);
  49
  50   p = out;
  51
  52   /* Discard the first line if it contains "tune2fs ...". */
  53   if (strncmp (p, "tune2fs ", 8) == 0) {
  54     p = strchr (p, '\n');
  55     if (p) p++;
  56     else {
  57       reply_with_error ("tune2fs: truncated output");
  58       free (out);
  59       return NULL;
  60     }
  61   }
  62
  63   /* Read the lines and split into "key: value". */
  64   while (*p) {
  65     pend = strchrnul (p, '\n');
  66     if (*pend == '\n') {
  67       *pend = '\0';
  68       pend++;
  69     }
  70
  71     if (!*p) continue;
  72
  73     colon = strchr (p, ':');
  74     if (colon) {
  75       *colon = '\0';
  76
  77       do { colon++; } while (*colon && isspace (*colon));
  78
  79       if (add_string (&ret, &size, &alloc, p) == -1) {
  80         free (out);
  81         return NULL;
  82       }
  83       if (strcmp (colon, "<none>") == 0 ||
  84           strcmp (colon, "<not available>") == 0 ||
  85           strcmp (colon, "(none)") == 0) {
  86         if (add_string (&ret, &size, &alloc, "") == -1) {
  87           free (out);
  88           return NULL;
  89         }
  90       } else {
  91         if (add_string (&ret, &size, &alloc, colon) == -1) {
  92           free (out);
  93           return NULL;
  94         }
  95       }
  96     }
  97     else {
  98       if (add_string (&ret, &size, &alloc, p) == -1) {
  99         free (out);
 100         return NULL;
 101       }
 102       if (add_string (&ret, &size, &alloc, "") == -1) {
 103         free (out);
 104         return NULL;
 105       }
 106     }
 107
 108     p = pend;
 109   }
 110
 111   free (out);
 112
 113   if (add_string (&ret, &size, &alloc, NULL) == -1)
 114     return NULL;
 115
 116   return ret;
 117 }
 118
 119 int
 120 do_set_e2label (const char *device, const char *label)
 121 {
 122   int r;
 123   char *err;
 124
 125   r = command (NULL, &err, "/sbin/e2label", device, label, NULL);
 126   if (r == -1) {
 127     reply_with_error ("e2label: %s", err);
 128     free (err);
 129     return -1;
 130   }
 131
 132   free (err);
 133   return 0;
 134 }
 135
 136 char *
 137 do_get_e2label (const char *device)
 138 {
 139   int r, len;
 140   char *out, *err;
 141
 142   r = command (&out, &err, "/sbin/e2label", device, NULL);
 143   if (r == -1) {
 144     reply_with_error ("e2label: %s", err);
 145     free (out);
 146     free (err);
 147     return NULL;
 148   }
 149
 150   free (err);
 151
 152   /* Remove any trailing \n from the label. */
 153   len = strlen (out);
 154   if (len > 0 && out[len-1] == '\n')
 155     out[len-1] = '\0';
 156
 157   return out;                   /* caller frees */
 158 }
 159
 160 int
 161 do_set_e2uuid (const char *device, const char *uuid)
 162 {
 163   int r;
 164   char *err;
 165
 166   r = command (NULL, &err, "/sbin/tune2fs", "-U", uuid, device, NULL);
 167   if (r == -1) {
 168     reply_with_error ("tune2fs -U: %s", err);
 169     free (err);
 170     return -1;
 171   }
 172
 173   free (err);
 174   return 0;
 175 }
 176
 177 char *
 178 do_get_e2uuid (const char *device)
 179 {
 180   int r;
 181   char *out, *err, *p, *q;
 182
 183   /* It's not so straightforward to get the volume UUID.  We have
 184    * to use tune2fs -l and then look for a particular string in
 185    * the output.
 186    */
 187
 188   r = command (&out, &err, "/sbin/tune2fs", "-l", device, NULL);
 189   if (r == -1) {
 190     reply_with_error ("tune2fs -l: %s", err);
 191     free (out);
 192     free (err);
 193     return NULL;
 194   }
 195
 196   free (err);
 197
 198   /* Look for /\nFilesystem UUID:\s+/ in the output. */
 199   p = strstr (out, "\nFilesystem UUID:");
 200   if (p == NULL) {
 201     reply_with_error ("no Filesystem UUID in the output of tune2fs -l");
 202     free (out);
 203     return NULL;
 204   }
 205
 206   p += 17;
 207   while (*p && isspace (*p))
 208     p++;
 209   if (!*p) {
 210     reply_with_error ("malformed Filesystem UUID in the output of tune2fs -l");
 211     free (out);
 212     return NULL;
 213   }
 214
 215   /* Now 'p' hopefully points to the start of the UUID. */
 216   q = p;
 217   while (*q && (isxdigit (*q) || *q == '-'))
 218     q++;
 219   if (!*q) {
 220     reply_with_error ("malformed Filesystem UUID in the output of tune2fs -l");
 221     free (out);
 222     return NULL;
 223   }
 224
 225   *q = '\0';
 226
 227   p = strdup (p);
 228   if (!p) {
 229     reply_with_perror ("strdup");
 230     free (out);
 231     return NULL;
 232   }
 233
 234   free (out);
 235   return p;                     /* caller frees */
 236 }
 237
 238 int
 239 do_resize2fs (const char *device)
 240 {
 241   char *err;
 242   int r;
 243
 244   IS_DEVICE (device, -1);
 245
 246   r = command (NULL, &err, "/sbin/resize2fs", device, NULL);
 247   if (r == -1) {
 248     reply_with_error ("resize2fs: %s", err);
 249     free (err);
 250     return -1;
 251   }
 252
 253   free (err);
 254   return 0;
 255 }