63672164186247752f89593c385f9433aac2a830
[invirt/packages/invirt-remote.git] / 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