host/usr/sbin/invirt-lvm

   1 #!/usr/bin/env python
   2
   3 import sys
   4 import os.path
   5 from subprocess import call, PIPE, Popen
   6 from invirt.config import structs as config
   7
   8 def check(b):
   9     if not b:
  10         exit(1)
  11
  12 vg = "xenvg"
  13 prefix = "d_"
  14
  15 subcommand = sys.argv[1]
  16
  17 def ensureoff(machine):
  18     # Make sure the machine is off, but we don't care about errors if it is already off.
  19     rv = call(["/usr/sbin/xm", "destroy", prefix + machine],
  20               stderr=PIPE)
  21
  22 if subcommand == "lvcreate-all":
  23     from invirt import database
  24     import re
  25     database.connect()
  26     for d in Disk.select():
  27         check(re.match('^[A-Za-z0-9]+$', d.guest_device_name))
  28         machine = Machine.get(d.machine_id)
  29         check(re.match('^[A-Za-z0-9][A-Za-z0-9._-]*$', machine.name))
  30         lvname = prefix + machine.name + "_" + d.guest_device_name
  31         if not os.path.exists("/dev/%s/%s" % (vg, lvname)):
  32             # LV doesn't exist
  33             print >>sys.stderr, "Creating LV %s..." % (lvname,)
  34             rv = call(["/sbin/lvcreate", "-L", str(d.size) + "M", "-n", lvname, vg])
  35             if rv != 0:
  36                 print >>sys.stderr, "Error creating LV %s\n" %(lvname,)
  37                 sys.exit(1)
  38 else:
  39     machine = sys.argv[2]
  40     disk = sys.argv[3]
  41     lvname = prefix + machine + "_" + disk
  42     lvpath = "/dev/" + vg + "/" + lvname
  43 if subcommand == "lvremove":
  44     def error():
  45         print >>sys.stderr, "Error removing LV %s\n" % lvname
  46         sys.exit(1)
  47     # I know this is the wrong answer, but sometimes the first
  48     # lvchange -a n fails for no particularly good reason, so this is
  49     # a pretty good workaround
  50     call(["/sbin/lvchange", "-a", "n", lvpath])
  51     rv = call(["/sbin/lvchange", "-a", "n", lvpath])
  52     if rv != 0:
  53         error()
  54     rv = call(["/sbin/lvchange", "-a", "ey", lvpath])
  55     if rv != 0:
  56         error()
  57     rv = call(["/sbin/lvremove", "--force", lvpath])
  58     if rv != 0:
  59         error()
  60     ensureoff(machine)
  61 elif subcommand == "lvresize":
  62     size = sys.argv[4]
  63     ensureoff(machine)
  64     p = Popen(["/sbin/lvresize", "-L", size + "M", lvpath],
  65               stdin=PIPE, stderr=PIPE)
  66     print >> p.stdin, 'y'
  67     err = p.stderr.read()
  68     if p.wait() != 0 and 'matches existing size' not in err:
  69         print >> sys.stderr, "Error resizing LV %s:\n" %(lvname,)
  70         print >> sys.stderr, err
  71         sys.exit(1)
  72     print >> sys.stderr, err
  73 elif subcommand == "lvrename":
  74     newmachine = sys.argv[4]
  75     newlvname = prefix + newmachine + "_" + disk
  76     ensureoff(machine)
  77     ensureoff(newmachine)
  78     rv = call(["/sbin/lvrename", vg, lvname, newlvname])
  79     if rv != 0:
  80         print >>sys.stderr, "Error renaming LV %s\n" %(lvname,)
  81         sys.exit(1)
  82 elif subcommand == "lvcreate":
  83     size = sys.argv[4]
  84     rv = call(["/sbin/lvcreate", "-L", size + "M", "-n", lvname, vg])
  85     if rv != 0:
  86         print >>sys.stderr, "Error creating LV %s\n" %(lvname,)
  87         sys.exit(1)
  88
  89