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
  26 #include "daemon.h"
  27 #include "c-ctype.h"
  28 #include "actions.h"
  29 #include "optgroups.h"
  30
  31 int
  32 optgroup_lvm2_available (void)
  33 {
  34   return prog_exists ("lvm");
  35 }
  36
  37 /* LVM actions.  Keep an eye on liblvm, although at the time
  38  * of writing it hasn't progressed very far.
  39  */
  40
  41 static char **
  42 convert_lvm_output (char *out, const char *prefix)
  43 {
  44   char *p, *pend;
  45   char **r = NULL;
  46   int size = 0, alloc = 0;
  47   int len;
  48   char buf[256];
  49   char *str;
  50
  51   p = out;
  52   while (p) {
  53     pend = strchr (p, '\n');    /* Get the next line of output. */
  54     if (pend) {
  55       *pend = '\0';
  56       pend++;
  57     }
  58
  59     while (*p && c_isspace (*p))        /* Skip any leading whitespace. */
  60       p++;
  61
  62     /* Sigh, skip trailing whitespace too.  "pvs", I'm looking at you. */
  63     len = strlen (p)-1;
  64     while (*p && c_isspace (p[len]))
  65       p[len--] = '\0';
  66
  67     if (!*p) {                  /* Empty line?  Skip it. */
  68       p = pend;
  69       continue;
  70     }
  71
  72     /* Prefix? */
  73     if (prefix) {
  74       snprintf (buf, sizeof buf, "%s%s", prefix, p);
  75       str = buf;
  76     } else
  77       str = p;
  78
  79     if (add_string (&r, &size, &alloc, str) == -1) {
  80       free (out);
  81       return NULL;
  82     }
  83
  84     p = pend;
  85   }
  86
  87   free (out);
  88
  89   if (add_string (&r, &size, &alloc, NULL) == -1)
  90     return NULL;
  91
  92   sort_strings (r, size-1);
  93   return r;
  94 }
  95
  96 char **
  97 do_pvs (void)
  98 {
  99   char *out, *err;
 100   int r;
 101
 102   r = command (&out, &err,
 103                "lvm", "pvs", "-o", "pv_name", "--noheadings", NULL);
 104   if (r == -1) {
 105     reply_with_error ("%s", err);
 106     free (out);
 107     free (err);
 108     return NULL;
 109   }
 110
 111   free (err);
 112
 113   return convert_lvm_output (out, NULL);
 114 }
 115
 116 char **
 117 do_vgs (void)
 118 {
 119   char *out, *err;
 120   int r;
 121
 122   r = command (&out, &err,
 123                "lvm", "vgs", "-o", "vg_name", "--noheadings", NULL);
 124   if (r == -1) {
 125     reply_with_error ("%s", err);
 126     free (out);
 127     free (err);
 128     return NULL;
 129   }
 130
 131   free (err);
 132
 133   return convert_lvm_output (out, NULL);
 134 }
 135
 136 char **
 137 do_lvs (void)
 138 {
 139   char *out, *err;
 140   int r;
 141
 142   r = command (&out, &err,
 143                "lvm", "lvs",
 144                "-o", "vg_name,lv_name", "--noheadings",
 145                "--separator", "/", NULL);
 146   if (r == -1) {
 147     reply_with_error ("%s", err);
 148     free (out);
 149     free (err);
 150     return NULL;
 151   }
 152
 153   free (err);
 154
 155   return convert_lvm_output (out, "/dev/");
 156 }
 157
 158 /* These were so complex to implement that I ended up auto-generating
 159  * the code.  That code is in stubs.c, and it is generated as usual
 160  * by generator.ml.
 161  */
 162 guestfs_int_lvm_pv_list *
 163 do_pvs_full (void)
 164 {
 165   return parse_command_line_pvs ();
 166 }
 167
 168 guestfs_int_lvm_vg_list *
 169 do_vgs_full (void)
 170 {
 171   return parse_command_line_vgs ();
 172 }
 173
 174 guestfs_int_lvm_lv_list *
 175 do_lvs_full (void)
 176 {
 177   return parse_command_line_lvs ();
 178 }
 179
 180 int
 181 do_pvcreate (const char *device)
 182 {
 183   char *err;
 184   int r;
 185
 186   r = command (NULL, &err,
 187                "lvm", "pvcreate", device, NULL);
 188   if (r == -1) {
 189     reply_with_error ("%s", err);
 190     free (err);
 191     return -1;
 192   }
 193
 194   free (err);
 195
 196   udev_settle ();
 197
 198   return 0;
 199 }
 200
 201 int
 202 do_vgcreate (const char *volgroup, char *const *physvols)
 203 {
 204   char *err;
 205   int r, argc, i;
 206   const char **argv;
 207
 208   argc = count_strings (physvols) + 3;
 209   argv = malloc (sizeof (char *) * (argc + 1));
 210   if (argv == NULL) {
 211     reply_with_perror ("malloc");
 212     return -1;
 213   }
 214   argv[0] = "lvm";
 215   argv[1] = "vgcreate";
 216   argv[2] = volgroup;
 217   for (i = 3; i <= argc; ++i)
 218     argv[i] = physvols[i-3];
 219
 220   r = commandv (NULL, &err, (const char * const*) argv);
 221   if (r == -1) {
 222     reply_with_error ("%s", err);
 223     free (err);
 224     return -1;
 225   }
 226
 227   free (err);
 228
 229   udev_settle ();
 230
 231   return 0;
 232 }
 233
 234 int
 235 do_lvcreate (const char *logvol, const char *volgroup, int mbytes)
 236 {
 237   char *err;
 238   int r;
 239   char size[64];
 240
 241   snprintf (size, sizeof size, "%d", mbytes);
 242
 243   r = command (NULL, &err,
 244                "lvm", "lvcreate",
 245                "-L", size, "-n", logvol, volgroup, NULL);
 246   if (r == -1) {
 247     reply_with_error ("%s", err);
 248     free (err);
 249     return -1;
 250   }
 251
 252   free (err);
 253
 254   udev_settle ();
 255
 256   return 0;
 257 }
 258
 259 int
 260 do_lvresize (const char *logvol, int mbytes)
 261 {
 262   char *err;
 263   int r;
 264   char size[64];
 265
 266   snprintf (size, sizeof size, "%d", mbytes);
 267
 268   r = command (NULL, &err,
 269                "lvm", "lvresize",
 270                "-L", size, logvol, NULL);
 271   if (r == -1) {
 272     reply_with_error ("%s", err);
 273     free (err);
 274     return -1;
 275   }
 276
 277   free (err);
 278   return 0;
 279 }
 280
 281 int
 282 do_lvresize_free (const char *logvol, int percent)
 283 {
 284   char *err;
 285   int r;
 286
 287   if (percent < 0 || percent > 100) {
 288     reply_with_error ("percentage must be [0..100] (was %d)", percent);
 289     return -1;
 290   }
 291
 292   char size[64];
 293   snprintf (size, sizeof size, "+%d%%FREE", percent);
 294
 295   r = command (NULL, &err,
 296                "lvm", "lvresize", "-l", size, logvol, NULL);
 297   if (r == -1) {
 298     reply_with_error ("%s", err);
 299     free (err);
 300     return -1;
 301   }
 302
 303   free (err);
 304   return 0;
 305 }
 306
 307 /* Super-dangerous command used for testing.  It removes all
 308  * LVs, VGs and PVs permanently.
 309  */
 310 int
 311 do_lvm_remove_all (void)
 312 {
 313   char **xs;
 314   int i, r;
 315   char *err;
 316
 317   /* Remove LVs. */
 318   xs = do_lvs ();
 319   if (xs == NULL)
 320     return -1;
 321
 322   for (i = 0; xs[i] != NULL; ++i) {
 323     r = command (NULL, &err, "lvm", "lvremove", "-f", xs[i], NULL);
 324     if (r == -1) {
 325       reply_with_error ("lvremove: %s: %s", xs[i], err);
 326       free (err);
 327       free_strings (xs);
 328       return -1;
 329     }
 330     free (err);
 331   }
 332   free_strings (xs);
 333
 334   /* Remove VGs. */
 335   xs = do_vgs ();
 336   if (xs == NULL)
 337     return -1;
 338
 339   for (i = 0; xs[i] != NULL; ++i) {
 340     r = command (NULL, &err, "lvm", "vgremove", "-f", xs[i], NULL);
 341     if (r == -1) {
 342       reply_with_error ("vgremove: %s: %s", xs[i], err);
 343       free (err);
 344       free_strings (xs);
 345       return -1;
 346     }
 347     free (err);
 348   }
 349   free_strings (xs);
 350
 351   /* Remove PVs. */
 352   xs = do_pvs ();
 353   if (xs == NULL)
 354     return -1;
 355
 356   for (i = 0; xs[i] != NULL; ++i) {
 357     r = command (NULL, &err, "lvm", "pvremove", "-f", xs[i], NULL);
 358     if (r == -1) {
 359       reply_with_error ("pvremove: %s: %s", xs[i], err);
 360       free (err);
 361       free_strings (xs);
 362       return -1;
 363     }
 364     free (err);
 365   }
 366   free_strings (xs);
 367
 368   udev_settle ();
 369
 370   /* There, that was easy, sorry about your data. */
 371   return 0;
 372 }
 373
 374 int
 375 do_lvremove (const char *device)
 376 {
 377   char *err;
 378   int r;
 379
 380   r = command (NULL, &err,
 381                "lvm", "lvremove", "-f", device, NULL);
 382   if (r == -1) {
 383     reply_with_error ("%s", err);
 384     free (err);
 385     return -1;
 386   }
 387
 388   free (err);
 389
 390   udev_settle ();
 391
 392   return 0;
 393 }
 394
 395 int
 396 do_vgremove (const char *device)
 397 {
 398   char *err;
 399   int r;
 400
 401   r = command (NULL, &err,
 402                "lvm", "vgremove", "-f", 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_pvremove (const char *device)
 418 {
 419   char *err;
 420   int r;
 421
 422   r = command (NULL, &err,
 423                "lvm", "pvremove", "-ff", device, NULL);
 424   if (r == -1) {
 425     reply_with_error ("%s", err);
 426     free (err);
 427     return -1;
 428   }
 429
 430   free (err);
 431
 432   udev_settle ();
 433
 434   return 0;
 435 }
 436
 437 int
 438 do_pvresize (const char *device)
 439 {
 440   char *err;
 441   int r;
 442
 443   r = command (NULL, &err,
 444                "lvm", "pvresize", device, NULL);
 445   if (r == -1) {
 446     reply_with_error ("%s: %s", device, err);
 447     free (err);
 448     return -1;
 449   }
 450
 451   free (err);
 452   return 0;
 453 }
 454
 455 int
 456 do_vg_activate (int activate, char *const *volgroups)
 457 {
 458   char *err;
 459   int r, i, argc;
 460   const char **argv;
 461
 462   argc = count_strings (volgroups) + 4;
 463   argv = malloc (sizeof (char *) * (argc+1));
 464   if (argv == NULL) {
 465     reply_with_perror ("malloc");
 466     return -1;
 467   }
 468
 469   argv[0] = "lvm";
 470   argv[1] = "vgchange";
 471   argv[2] = "-a";
 472   argv[3] = activate ? "y" : "n";
 473   for (i = 4; i <= argc; ++i)
 474     argv[i] = volgroups[i-4];
 475
 476   r = commandv (NULL, &err, (const char * const*) argv);
 477   if (r == -1) {
 478     reply_with_error ("vgchange: %s", err);
 479     free (err);
 480     return -1;
 481   }
 482
 483   free (err);
 484
 485   udev_settle ();
 486
 487   return 0;
 488 }
 489
 490 int
 491 do_vg_activate_all (int activate)
 492 {
 493   char *empty[] = { NULL };
 494   return do_vg_activate (activate, empty);
 495 }
 496
 497 int
 498 do_lvrename (const char *logvol, const char *newlogvol)
 499 {
 500   char *err;
 501   int r;
 502
 503   r = command (NULL, &err,
 504                "lvm", "lvrename",
 505                logvol, newlogvol, NULL);
 506   if (r == -1) {
 507     reply_with_error ("%s -> %s: %s", logvol, newlogvol, err);
 508     free (err);
 509     return -1;
 510   }
 511
 512   free (err);
 513
 514   udev_settle ();
 515
 516   return 0;
 517 }
 518
 519 int
 520 do_vgrename (const char *volgroup, const char *newvolgroup)
 521 {
 522   char *err;
 523   int r;
 524
 525   r = command (NULL, &err,
 526                "lvm", "vgrename",
 527                volgroup, newvolgroup, NULL);
 528   if (r == -1) {
 529     reply_with_error ("%s -> %s: %s", volgroup, newvolgroup, err);
 530     free (err);
 531     return -1;
 532   }
 533
 534   free (err);
 535
 536   udev_settle ();
 537
 538   return 0;
 539 }
 540
 541 static char *
 542 get_lvm_field (const char *cmd, const char *field, const char *device)
 543 {
 544   char *out;
 545   char *err;
 546   int r = command (&out, &err,
 547                    "lvm", cmd,
 548                    "--unbuffered", "--noheadings", "-o", field,
 549                    device, NULL);
 550   if (r == -1) {
 551     reply_with_error ("%s: %s", device, err);
 552     free (out);
 553     free (err);
 554     return NULL;
 555   }
 556
 557   free (err);
 558
 559   trim (out);
 560   return out;                   /* Caller frees. */
 561 }
 562
 563 char *
 564 do_pvuuid (const char *device)
 565 {
 566   return get_lvm_field ("pvs", "pv_uuid", device);
 567 }
 568
 569 char *
 570 do_vguuid (const char *vgname)
 571 {
 572   return get_lvm_field ("vgs", "vg_uuid", vgname);
 573 }
 574
 575 char *
 576 do_lvuuid (const char *device)
 577 {
 578   return get_lvm_field ("lvs", "lv_uuid", device);
 579 }
 580
 581 static char **
 582 get_lvm_fields (const char *cmd, const char *field, const char *device)
 583 {
 584   char *out;
 585   char *err;
 586   int r = command (&out, &err,
 587                    "lvm", cmd,
 588                    "--unbuffered", "--noheadings", "-o", field,
 589                    device, NULL);
 590   if (r == -1) {
 591     reply_with_error ("%s: %s", device, err);
 592     free (out);
 593     free (err);
 594     return NULL;
 595   }
 596
 597   free (err);
 598
 599   char **ret = split_lines (out);
 600   free (out);
 601
 602   if (ret == NULL)
 603     return NULL;
 604
 605   size_t i;
 606   for (i = 0; ret[i] != NULL; ++i)
 607     trim (ret[i]);
 608
 609   return ret;
 610 }
 611
 612 char **
 613 do_vgpvuuids (const char *vgname)
 614 {
 615   return get_lvm_fields ("vgs", "pv_uuid", vgname);
 616 }
 617
 618 char **
 619 do_vglvuuids (const char *vgname)
 620 {
 621   return get_lvm_fields ("vgs", "lv_uuid", vgname);
 622 }
 623
 624 int
 625 do_vgscan (void)
 626 {
 627   char *err;
 628   int r;
 629
 630   r = command (NULL, &err,
 631                "lvm", "vgscan", NULL);
 632   if (r == -1) {
 633     reply_with_error ("%s", err);
 634     free (err);
 635     return -1;
 636   }
 637
 638   free (err);
 639   return 0;
 640 }