9c22f59374bc26ac9123f2075c40275ba9a1412a
[invirt/packages/invirt-remote.git] / host / usr / sbin / invirt-listvms
1 #!/usr/bin/python
2
3 import sys
4 import time
5 from xen.lowlevel import xs
6
7 xsc = xs.xs()
8
9 def live_vms():
10     domids = set(xsc.ls('', '/local/domain'))
11     domids.discard('0')
12
13     vms = dict()
14
15     for domid in domids:
16         try:
17             name, data = get_dom(int(domid))
18         except (xs.Error, TypeError):
19             continue # went down since we started
20         if name is not None and name.startswith('d_'):
21             name = name[2:]
22             vms[name] = data
23     return vms
24
25 def get_dom(domid):
26     name = xsc.read('', '/local/domain/%d/name' % domid)
27     data = dict()
28     data['domid'] = domid
29     # presence of a graphical console
30     data['console'] = xsc.read('', '/local/domain/%d/device/vfb/0/state' % domid)
31     # uptime
32     data['vm'] = xsc.read('', '/local/domain/%d/vm' % domid)
33     data['start_time'] = float(xsc.read('', '%s/start_time' % data['vm']))
34     data['uptime'] = time.time()-data['start_time']
35     # CD or autoinstall, if we're using either
36     vbds = xsc.ls('', '/local/domain/%d/device/vbd' % domid)
37     for vbd in vbds:
38         dev = '/local/domain/%d/device/vbd/%s' % (domid, vbd)
39         dev_type = xsc.read('', '%s/device-type' % dev)
40         dev_params = xsc.read('', '%s/params' % xsc.read('', '%s/backend' % dev))
41         if dev_type == 'cdrom':
42             data['cdrom'] = dev_params
43         elif dev_type == 'disk' and dev_params == '/dev/xenvg/s_install_hda':
44             data['autoinstall'] = True
45     if 'auto=true' in xsc.read('', '%s/image/cmdline' % data['vm']):
46         data['autoinstall'] = True
47
48     return name, data
49
50 if __name__ == '__main__':
51     vms = live_vms()
52     if '--json' in sys.argv[1:]:
53         import cjson
54         print cjson.encode(vms)
55     elif '--pickle' in sys.argv[1:]:
56         import cPickle
57         cPickle.dump(vms, sys.stdout, cPickle.HIGHEST_PROTOCOL)
58     else:
59         import yaml
60         print yaml.dump(vms, Dumper=yaml.CSafeDumper, default_flow_style=False)