daemon/lvm.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 <unistd.h>
  25 #include <ctype.h>
  26
  27 #include "daemon.h"
  28 #include "actions.h"
  29
  30 /* LVM actions.  Keep an eye on liblvm, although at the time
  31  * of writing it hasn't progressed very far.
  32  */
  33
  34 static char **
  35 convert_lvm_output (char *out, char *prefix)
  36 {
  37   char *p, *pend;
  38   char **r = NULL;
  39   int size = 0, alloc = 0;
  40   int len;
  41   char buf[256];
  42   char *str;
  43
  44   p = out;
  45   while (p) {
  46     pend = strchr (p, '\n');    /* Get the next line of output. */
  47     if (pend) {
  48       *pend = '\0';
  49       pend++;
  50     }
  51
  52     while (*p && isspace (*p))  /* Skip any leading whitespace. */
  53       p++;
  54
  55     /* Sigh, skip trailing whitespace too.  "pvs", I'm looking at you. */
  56     len = strlen (p)-1;
  57     while (*p && isspace (p[len]))
  58       p[len--] = '\0';
  59
  60     if (!*p) {                  /* Empty line?  Skip it. */
  61       p = pend;
  62       continue;
  63     }
  64
  65     /* Prefix? */
  66     if (prefix) {
  67       snprintf (buf, sizeof buf, "%s%s", prefix, p);
  68       str = buf;
  69     } else
  70       str = p;
  71
  72     if (add_string (&r, &size, &alloc, str) == -1) {
  73       free (out);
  74       return NULL;
  75     }
  76
  77     p = pend;
  78   }
  79
  80   free (out);
  81
  82   if (add_string (&r, &size, &alloc, NULL) == -1)
  83     return NULL;
  84
  85   sort_strings (r, size-1);
  86   return r;
  87 }
  88
  89 char **
  90 do_pvs (void)
  91 {
  92   char *out, *err;
  93   int r;
  94
  95   r = command (&out, &err,
  96                "/sbin/lvm", "pvs", "-o", "pv_name", "--noheadings", NULL);
  97   if (r == -1) {
  98     reply_with_error ("%s", err);
  99     free (out);
 100     free (err);
 101     return NULL;
 102   }
 103
 104   free (err);
 105
 106   return convert_lvm_output (out, NULL);
 107 }
 108
 109 char **
 110 do_vgs (void)
 111 {
 112   char *out, *err;
 113   int r;
 114
 115   r = command (&out, &err,
 116                "/sbin/lvm", "vgs", "-o", "vg_name", "--noheadings", NULL);
 117   if (r == -1) {
 118     reply_with_error ("%s", err);
 119     free (out);
 120     free (err);
 121     return NULL;
 122   }
 123
 124   free (err);
 125
 126   return convert_lvm_output (out, NULL);
 127 }
 128
 129 char **
 130 do_lvs (void)
 131 {
 132   char *out, *err;
 133   int r;
 134
 135   r = command (&out, &err,
 136                "/sbin/lvm", "lvs",
 137                "-o", "vg_name,lv_name", "--noheadings",
 138                "--separator", "/", NULL);
 139   if (r == -1) {
 140     reply_with_error ("%s", err);
 141     free (out);
 142     free (err);
 143     return NULL;
 144   }
 145
 146   free (err);
 147
 148   return convert_lvm_output (out, "/dev/");
 149 }
 150
 151 /* These were so complex to implement that I ended up auto-generating
 152  * the code.  That code is in stubs.c, and it is generated as usual
 153  * by generator.ml.
 154  */
 155 guestfs_int_lvm_pv_list *
 156 do_pvs_full (void)
 157 {
 158   return parse_command_line_pvs ();
 159 }
 160
 161 guestfs_int_lvm_vg_list *
 162 do_vgs_full (void)
 163 {
 164   return parse_command_line_vgs ();
 165 }
 166
 167 guestfs_int_lvm_lv_list *
 168 do_lvs_full (void)
 169 {
 170   return parse_command_line_lvs ();
 171 }
 172
 173 int
 174 do_pvcreate (char *device)
 175 {
 176   char *err;
 177   int r;
 178
 179   RESOLVE_DEVICE (device, return -1);
 180
 181   r = command (NULL, &err,
 182                "/sbin/lvm", "pvcreate", device, NULL);
 183   if (r == -1) {
 184     reply_with_error ("%s", err);
 185     free (err);
 186     return -1;
 187   }
 188
 189   free (err);
 190
 191   udev_settle ();
 192
 193   return 0;
 194 }
 195
 196 int
 197 do_vgcreate (char *volgroup, char **physvols)
 198 {
 199   char *err;
 200   int r, argc, i;
 201   const char **argv;
 202
 203   /* Check they are devices and also do device name translation. */
 204   for (i = 0; physvols[i] != NULL; ++i)
 205     RESOLVE_DEVICE (physvols[i], return -1);
 206
 207   argc = count_strings (physvols) + 3;
 208   argv = malloc (sizeof (char *) * (argc + 1));
 209   if (argv == NULL) {
 210     reply_with_perror ("malloc");
 211     return -1;
 212   }
 213   argv[0] = "/sbin/lvm";
 214   argv[1] = "vgcreate";
 215   argv[2] = volgroup;
 216   for (i = 3; i <= argc; ++i)
 217     argv[i] = physvols[i-3];
 218
 219   r = commandv (NULL, &err, argv);
 220   if (r == -1) {
 221     reply_with_error ("%s", err);
 222     free (err);
 223     return -1;
 224   }
 225
 226   free (err);
 227
 228   udev_settle ();
 229
 230   return 0;
 231 }
 232
 233 int
 234 do_lvcreate (char *logvol, char *volgroup, int mbytes)
 235 {
 236   char *err;
 237   int r;
 238   char size[64];
 239
 240   snprintf (size, sizeof size, "%d", mbytes);
 241
 242   r = command (NULL, &err,
 243                "/sbin/lvm", "lvcreate",
 244                "-L", size, "-n", logvol, volgroup, NULL);
 245   if (r == -1) {
 246     reply_with_error ("%s", err);
 247     free (err);
 248     return -1;
 249   }
 250
 251   free (err);
 252
 253   udev_settle ();
 254
 255   return 0;
 256 }
 257
 258 int
 259 do_lvresize (char *logvol, int mbytes)
 260 {
 261   char *err;
 262   int r;
 263   char size[64];
 264
 265   RESOLVE_DEVICE (logvol, return -1);
 266
 267   snprintf (size, sizeof size, "%d", mbytes);
 268
 269   r = command (NULL, &err,
 270                "/sbin/lvm", "lvresize",
 271                "-L", size, logvol, NULL);
 272   if (r == -1) {
 273     reply_with_error ("lvresize: %s", err);
 274     free (err);
 275     return -1;
 276   }
 277
 278   free (err);
 279   return 0;
 280 }
 281
 282 /* Super-dangerous command used for testing.  It removes all
 283  * LVs, VGs and PVs permanently.
 284  */
 285 int
 286 do_lvm_remove_all (void)
 287 {
 288   char **xs;
 289   int i, r;
 290   char *err;
 291
 292   /* Remove LVs. */
 293   xs = do_lvs ();
 294   if (xs == NULL)
 295     return -1;
 296
 297   for (i = 0; xs[i] != NULL; ++i) {
 298     r = command (NULL, &err, "/sbin/lvm", "lvremove", "-f", xs[i], NULL);
 299     if (r == -1) {
 300       reply_with_error ("lvremove: %s: %s", xs[i], err);
 301       free (err);
 302       free_strings (xs);
 303       return -1;
 304     }
 305     free (err);
 306   }
 307   free_strings (xs);
 308
 309   /* Remove VGs. */
 310   xs = do_vgs ();
 311   if (xs == NULL)
 312     return -1;
 313
 314   for (i = 0; xs[i] != NULL; ++i) {
 315     r = command (NULL, &err, "/sbin/lvm", "vgremove", "-f", xs[i], NULL);
 316     if (r == -1) {
 317       reply_with_error ("vgremove: %s: %s", xs[i], err);
 318       free (err);
 319       free_strings (xs);
 320       return -1;
 321     }
 322     free (err);
 323   }
 324   free_strings (xs);
 325
 326   /* Remove PVs. */
 327   xs = do_pvs ();
 328   if (xs == NULL)
 329     return -1;
 330
 331   for (i = 0; xs[i] != NULL; ++i) {
 332     r = command (NULL, &err, "/sbin/lvm", "pvremove", "-f", xs[i], NULL);
 333     if (r == -1) {
 334       reply_with_error ("pvremove: %s: %s", xs[i], err);
 335       free (err);
 336       free_strings (xs);
 337       return -1;
 338     }
 339     free (err);
 340   }
 341   free_strings (xs);
 342
 343   udev_settle ();
 344
 345   /* There, that was easy, sorry about your data. */
 346   return 0;
 347 }
 348
 349 int
 350 do_lvremove (char *device)
 351 {
 352   char *err;
 353   int r;
 354
 355   RESOLVE_DEVICE (device, return -1);
 356
 357   r = command (NULL, &err,
 358                "/sbin/lvm", "lvremove", "-f", device, NULL);
 359   if (r == -1) {
 360     reply_with_error ("%s", err);
 361     free (err);
 362     return -1;
 363   }
 364
 365   free (err);
 366
 367   udev_settle ();
 368
 369   return 0;
 370 }
 371
 372 int
 373 do_vgremove (char *device)
 374 {
 375   char *err;
 376   int r;
 377
 378   r = command (NULL, &err,
 379                "/sbin/lvm", "vgremove", "-f", device, NULL);
 380   if (r == -1) {
 381     reply_with_error ("%s", err);
 382     free (err);
 383     return -1;
 384   }
 385
 386   free (err);
 387
 388   udev_settle ();
 389
 390   return 0;
 391 }
 392
 393 int
 394 do_pvremove (char *device)
 395 {
 396   char *err;
 397   int r;
 398
 399   RESOLVE_DEVICE (device, return -1);
 400
 401   r = command (NULL, &err,
 402                "/sbin/lvm", "pvremove", "-ff", device, NULL);
 403   if (r == -1) {
 404     reply_with_error ("%s", err);
 405     free (err);
 406     return -1;
 407   }
 408
 409   free (err);
 410
 411   udev_settle ();
 412
 413   return 0;
 414 }
 415
 416 int
 417 do_pvresize (char *device)
 418 {
 419   char *err;
 420   int r;
 421
 422   RESOLVE_DEVICE (device, return -1);
 423
 424   r = command (NULL, &err,
 425                "/sbin/lvm", "pvresize", device, NULL);
 426   if (r == -1) {
 427     reply_with_error ("pvresize: %s: %s", device, err);
 428     free (err);
 429     return -1;
 430   }
 431
 432   free (err);
 433   return 0;
 434 }
 435
 436 int
 437 do_vg_activate (int activate, char **volgroups)
 438 {
 439   char *err;
 440   int r, i, argc;
 441   const char **argv;
 442
 443   argc = count_strings (volgroups) + 4;
 444   argv = malloc (sizeof (char *) * (argc+1));
 445   if (argv == NULL) {
 446     reply_with_perror ("malloc");
 447     return -1;
 448   }
 449
 450   argv[0] = "/sbin/lvm";
 451   argv[1] = "vgchange";
 452   argv[2] = "-a";
 453   argv[3] = activate ? "y" : "n";
 454   for (i = 4; i <= argc; ++i)
 455     argv[i] = volgroups[i-4];
 456
 457   r = commandv (NULL, &err, argv);
 458   if (r == -1) {
 459     reply_with_error ("vgchange: %s", err);
 460     free (err);
 461     return -1;
 462   }
 463
 464   free (err);
 465
 466   udev_settle ();
 467
 468   return 0;
 469 }
 470
 471 int
 472 do_vg_activate_all (int activate)
 473 {
 474   char *empty[] = { NULL };
 475   return do_vg_activate (activate, empty);
 476 }