files/usr/sbin/sipb-xen-lvm

   1 #!/usr/bin/env python
   2
   3 import sys
   4 import os.path
   5 from subprocess import call
   6
   7 def check(b):
   8     if not b:
   9         exit(1)
  10
  11 vg = "xenvg"
  12 prefix = "d_"
  13
  14 subcommand = sys.argv[1]
  15
  16 def ensureoff(machine):
  17     # Make sure the machine is off, but we don't care about errors if it is already off.
  18     try:
  19         rv = call(["/usr/sbin/xm", "destroy", prefix + machine])
  20     except:
  21         pass
  22
  23 if subcommand == "lvcreate-all":
  24     from sipb_xen_database import *
  25     import re
  26     connect('postgres://sipb-xen@sipb-xen-dev.mit.edu/sipb_xen')
  27     for d in Disk.select():
  28         check(re.match('^[A-Za-z0-9]+$', d.guest_device_name))
  29         machine = Machine.get(d.machine_id)
  30         check(re.match('^[A-Za-z0-9][A-Za-z0-9._-]*$', machine.name))
  31         lvname = prefix + machine.name + "_" + d.guest_device_name
  32         if not os.path.exists("/dev/%s/%s" % (vg, lvname)):
  33             # LV doesn't exist
  34             print >>sys.stderr, "Creating LV %s..." % (lvname,)
  35             rv = call(["/sbin/lvcreate", "-L", str(d.size) + "M", "-n", lvname, vg])
  36             if rv != 0:
  37                 print >>sys.stderr, "Error creating LV %s\n" %(lvname,)
  38                 sys.exit(1)
  39 else:
  40     machine = sys.argv[2]
  41     disk = sys.argv[3]
  42     lvname = prefix + machine + "_" + disk
  43     lvpath = "/dev/" + vg + "/" + lvname
  44 if subcommand == "lvremove":
  45     rv = call(["/sbin/lvremove", "--force", lvpath])
  46     ensureoff(machine)
  47     if rv != 0:
  48         print >>sys.stderr, "Error removing LV %s\n" %(lvname,)
  49         sys.exit(1)
  50 elif subcommand == "lvresize":
  51     size = sys.argv[4]
  52     ensureoff(machine)
  53     rv = call(["/sbin/lvresize", "-L", size + "M", lvpath])
  54     if rv != 0:
  55         print >>sys.stderr, "Error resizing LV %s\n" %(lvname,)
  56         sys.exit(1)
  57 elif subcommand == "lvrename":
  58     newmachine = sys.argv[4]
  59     newlvname = prefix + newmachine + "_" + disk
  60     ensureoff(machine)
  61     ensureoff(newmachine)
  62     rv = call(["/sbin/lvrename", vg, lvname, newlvname])
  63     if rv != 0:
  64         print >>sys.stderr, "Error renaming LV %s\n" %(lvname,)
  65         sys.exit(1)
  66 elif subcommand == "lvcreate":
  67     size = sys.argv[4]
  68     rv = call(["/sbin/lvcreate", "-L", size + "M", "-n", lvname, vg])
  69     if rv != 0:
  70         print >>sys.stderr, "Error creating LV %s\n" %(lvname,)
  71         sys.exit(1)
  72
  73