#!/usr/bin/env python import sys import os.path from subprocess import call def check(b): if not b: exit(1) vg = "xenvg" prefix = "d_" subcommand = sys.argv[1] machine = sys.argv[2] disk = sys.argv[3] lvname = prefix + machine + "_" + disk lvpath = "/dev/" + vg + "/" + lvname def ensureoff(machine): # Make sure the machine is off, but we don't care about errors if it is already off. rv = call(["/usr/sbin/xm", "destroy", prefix + machine]) if subcommand == "lvremove": rv = call(["/sbin/lvremove", "--force", lvpath]) ensureoff(machine) if rv != 0: print >>sys.stderr, "Error removing LV %s\n" %(lvname,) sys.exit(1) elif subcommand == "lvresize": size = sys.argv[4] ensureoff(machine) rv = call(["/sbin/lvresize", "-L", size + "M", lvpath]) if rv != 0: print >>sys.stderr, "Error resizing LV %s\n" %(lvname,) sys.exit(1) elif subcommand == "lvrename": newmachine = sys.argv[4] newlvname = prefix + newmachine + "_" + disk ensureoff(machine) ensureoff(newmachine) rv = call(["/sbin/lvrename", vg, lvname, newlvname]) if rv != 0: print >>sys.stderr, "Error renaming LV %s\n" %(lvname,) sys.exit(1) elif subcommand == "lvcreate": size = sys.argv[4] rv = call(["/sbin/lvcreate", "-L", size + "M", "-n", lvname, vg]) if rv != 0: print >>sys.stderr, "Error creating LV %s\n" %(lvname,) sys.exit(1) elif subcommand == "lvcreate-all": from sipb_xen_database import * import re connect('postgres://sipb-xen@sipb-xen-dev.mit.edu/sipb_xen') for d in Disk.select(): check(re.match('^[A-Za-z0-9]+$', d.guest_device_name)) machine = Machine.get(d.machine_id) check(re.match('^[A-Za-z0-9][A-Za-z0-9._-]*$', machine.name)) lvname = prefix + machine.name + "_" + d.guest_device_name if not os.path.exists("/dev/%s/%s" % (vg, lvname)): # LV doesn't exist print >>sys.stderr, "Creating LV %s..." % (lvname,) rv = call(["/sbin/lvcreate", "-L", str(d.size) + "M", "-n", lvname, vg]) if rv != 0: print >>sys.stderr, "Error creating LV %s\n" %(lvname,) sys.exit(1)