1 ###############################################################################
2 # Copyright (c) 2015 Ericsson AB and others.
3 # szilard.cserey@ericsson.com
4 # All rights reserved. This program and the accompanying materials
5 # are made available under the terms of the Apache License, Version 2.0
6 # which accompanies this distribution, and is available at
7 # http://www.apache.org/licenses/LICENSE-2.0
8 ###############################################################################
11 from lxml import etree
12 from hardware_adapter import HardwareAdapter
21 DEV = {'pxe': 'network',
26 class LibvirtAdapter(HardwareAdapter):
28 def __init__(self, yaml_path):
29 super(LibvirtAdapter, self).__init__(yaml_path)
30 self.parser = etree.XMLParser(remove_blank_text=True)
32 def node_power_off(self, node_id):
33 vm_name = self.get_node_property(node_id, 'libvirtName')
34 log('Power OFF Node %s' % vm_name)
35 state = exec_cmd('virsh domstate %s' % vm_name)
36 if state == 'running':
37 exec_cmd('virsh destroy %s' % vm_name, False)
39 def node_power_on(self, node_id):
40 vm_name = self.get_node_property(node_id, 'libvirtName')
41 log('Power ON Node %s' % vm_name)
42 state = exec_cmd('virsh domstate %s' % vm_name)
43 if state == 'shut off':
44 exec_cmd('virsh start %s' % vm_name)
46 def node_reset(self, node_id):
47 vm_name = self.get_node_property(node_id, 'libvirtName')
48 log('Reset Node %s' % vm_name)
49 exec_cmd('virsh reset %s' % vm_name)
51 def translate(self, boot_order_list):
53 for boot_dev in boot_order_list:
55 translated.append(DEV[boot_dev])
57 err('Boot device %s not recognized' % boot_dev)
60 def node_set_boot_order(self, node_id, boot_order_list):
61 boot_order_list = self.translate(boot_order_list)
62 vm_name = self.get_node_property(node_id, 'libvirtName')
63 temp_dir = exec_cmd('mktemp -d')
64 log('Set boot order %s on Node %s' % (boot_order_list, vm_name))
65 resp = exec_cmd('virsh dumpxml %s' % vm_name)
66 xml_dump = etree.fromstring(resp, self.parser)
67 os = xml_dump.xpath('/domain/os')
69 for bootelem in ['boot', 'bootmenu']:
70 boot = o.xpath(bootelem)
73 for dev in boot_order_list:
74 b = etree.Element('boot')
77 bmenu = etree.Element('bootmenu')
78 bmenu.set('enable', 'no')
80 tree = etree.ElementTree(xml_dump)
81 xml_file = temp_dir + '/%s.xml' % vm_name
82 with open(xml_file, 'w') as f:
83 tree.write(f, pretty_print=True, xml_declaration=True)
84 exec_cmd('virsh define %s' % xml_file)
87 def node_zero_mbr(self, node_id):
88 vm_name = self.get_node_property(node_id, 'libvirtName')
89 resp = exec_cmd('virsh dumpxml %s' % vm_name)
90 xml_dump = etree.fromstring(resp)
91 disks = xml_dump.xpath('/domain/devices/disk')
93 if disk.get('device') == 'disk':
94 sources = disk.xpath('source')
95 for source in sources:
96 disk_file = source.get('file')
97 disk_size = exec_cmd('ls -l %s' % disk_file).split()[4]
99 exec_cmd('fallocate -l %s %s' % (disk_size, disk_file))
101 def node_eject_iso(self, node_id):
102 vm_name = self.get_node_property(node_id, 'libvirtName')
103 device = self.get_name_of_device(vm_name, 'cdrom')
104 exec_cmd('virsh change-media %s --eject %s --config --live'
105 % (vm_name, device), False)
107 def node_insert_iso(self, node_id, iso_file):
108 vm_name = self.get_node_property(node_id, 'libvirtName')
109 device = self.get_name_of_device(vm_name, 'cdrom')
110 exec_cmd('virsh change-media %s --insert %s %s'
111 % (vm_name, device, iso_file))
113 def get_node_pxe_mac(self, node_id):
115 vm_name = self.get_node_property(node_id, 'libvirtName')
116 resp = exec_cmd('virsh dumpxml %s' % vm_name)
117 xml_dump = etree.fromstring(resp)
118 interfaces = xml_dump.xpath('/domain/devices/interface')
119 for interface in interfaces:
120 macs = interface.xpath('mac')
122 mac_list.append(mac.get('address').lower())
125 def get_name_of_device(self, vm_name, device_type):
126 resp = exec_cmd('virsh dumpxml %s' % vm_name)
127 xml_dump = etree.fromstring(resp)
128 disks = xml_dump.xpath('/domain/devices/disk')
130 if disk.get('device') == device_type:
131 targets = disk.xpath('target')
132 for target in targets:
133 device = target.get('dev')
137 def get_virt_net_conf_dir(self):
138 return self.dha_struct['virtNetConfDir']