#!/usr/bin/env python import sys import time import os import random import string from subprocess import call, PIPE, Popen from invirt.config import structs as config def check(b): if not b: exit(1) vg = "xenvg" prefix = "d_" subcommand = sys.argv[1] 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], stderr=PIPE) if subcommand == "lvcreate-all": from invirt import database import re database.connect() 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) else: machine = sys.argv[2] disk = sys.argv[3] lvname = prefix + machine + "_" + disk lvpath = "/dev/" + vg + "/" + lvname if subcommand == "lvremove": def error(): print >>sys.stderr, "Error removing LV %s\n" % lvname sys.exit(1) # Rename the LV to something else so we can wipe it before reusing # the space while True: new_lvname = "old_%s_%s" % (lvname, ''.join(random.choice(string.ascii_letters) for i in xrange(6))) new_lvpath = "/dev/%s/%s" % (vg, new_lvname) p = Popen(["/sbin/lvrename", lvpath, new_lvpath], stdout=PIPE, stderr=PIPE) rv = p.wait() if rv == 5 and 'already exists in volume group' in p.stderr.read(): continue elif rv != 0: error() else: break ensureoff(machine) # Fork. The child process wipes the LV and then deletes # it. There's not really anything sane to do with errors (since # this is running non-interactively), so let's just drop them on # the floor for now. if os.fork() == 0: call(["/usr/bin/ionice", "-c", "2", "-n", "7", "/bin/dd", "if=/dev/zero", "of=%s" % new_lvpath]) call(["/sbin/lvchange", "-a", "n", new_lvpath]) call(["/sbin/lvchange", "-a", "ey", new_lvpath]) call(["/sbin/lvremove", "--force", new_lvpath]) elif subcommand == "lvresize": size = sys.argv[4] ensureoff(machine) p = Popen(["/sbin/lvresize", "-L", size + "M", lvpath], stdin=PIPE, stderr=PIPE) print >> p.stdin, 'y' err = p.stderr.read() if p.wait() != 0 and 'matches existing size' not in err: print >> sys.stderr, "Error resizing LV %s:\n" %(lvname,) print >> sys.stderr, err sys.exit(1) print >> sys.stderr, err 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)