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
22 DEV = {'pxe': 'network',
27 class LibvirtAdapter(HardwareAdapter):
29 def __init__(self, yaml_path):
30 super(LibvirtAdapter, self).__init__(yaml_path)
31 self.parser = etree.XMLParser(remove_blank_text=True)
33 def node_power_off(self, node_id):
34 vm_name = self.get_node_property(node_id, 'libvirtName')
35 log('Power OFF Node %s' % vm_name)
36 state = exec_cmd('virsh domstate %s' % vm_name)
37 if state == 'running':
38 exec_cmd('virsh destroy %s' % vm_name, False)
40 def node_power_on(self, node_id):
41 vm_name = self.get_node_property(node_id, 'libvirtName')
42 log('Power ON Node %s' % vm_name)
43 state = exec_cmd('virsh domstate %s' % vm_name)
44 if state == 'shut off':
45 exec_cmd('virsh start %s' % vm_name)
47 def node_reset(self, node_id):
48 vm_name = self.get_node_property(node_id, 'libvirtName')
49 log('Reset Node %s' % vm_name)
50 exec_cmd('virsh reset %s' % vm_name)
52 def translate(self, boot_order_list):
54 for boot_dev in boot_order_list:
56 translated.append(DEV[boot_dev])
58 err('Boot device %s not recognized' % boot_dev)
61 def node_set_boot_order(self, node_id, boot_order_list):
62 boot_order_list = self.translate(boot_order_list)
63 vm_name = self.get_node_property(node_id, 'libvirtName')
64 temp_dir = tempfile.mkdtemp()
65 log('Set boot order %s on Node %s' % (boot_order_list, vm_name))
66 resp = exec_cmd('virsh dumpxml %s' % vm_name)
67 xml_dump = etree.fromstring(resp, self.parser)
68 os = xml_dump.xpath('/domain/os')
70 for bootelem in ['boot', 'bootmenu']:
71 boot = o.xpath(bootelem)
74 for dev in boot_order_list:
75 b = etree.Element('boot')
78 bmenu = etree.Element('bootmenu')
79 bmenu.set('enable', 'no')
81 tree = etree.ElementTree(xml_dump)
82 xml_file = temp_dir + '/%s.xml' % vm_name
83 with open(xml_file, 'w') as f:
84 tree.write(f, pretty_print=True, xml_declaration=True)
85 exec_cmd('virsh define %s' % xml_file)
88 def node_zero_mbr(self, node_id):
89 vm_name = self.get_node_property(node_id, 'libvirtName')
90 resp = exec_cmd('virsh dumpxml %s' % vm_name)
91 xml_dump = etree.fromstring(resp)
92 disks = xml_dump.xpath('/domain/devices/disk')
94 if disk.get('device') == 'disk':
95 sources = disk.xpath('source')
96 for source in sources:
97 disk_file = source.get('file')
98 disk_size = exec_cmd('ls -l %s' % disk_file).split()[4]
100 exec_cmd('fallocate -l %s %s' % (disk_size, disk_file))
102 def node_eject_iso(self, node_id):
103 vm_name = self.get_node_property(node_id, 'libvirtName')
104 device = self.get_name_of_device(vm_name, 'cdrom')
105 exec_cmd('virsh change-media %s --eject %s --config --live'
106 % (vm_name, device), False)
108 def node_insert_iso(self, node_id, iso_file):
109 vm_name = self.get_node_property(node_id, 'libvirtName')
110 device = self.get_name_of_device(vm_name, 'cdrom')
111 exec_cmd('virsh change-media %s --insert %s %s'
112 % (vm_name, device, iso_file))
114 def get_node_pxe_mac(self, node_id):
116 vm_name = self.get_node_property(node_id, 'libvirtName')
117 resp = exec_cmd('virsh dumpxml %s' % vm_name)
118 xml_dump = etree.fromstring(resp)
119 interfaces = xml_dump.xpath('/domain/devices/interface')
120 for interface in interfaces:
121 macs = interface.xpath('mac')
123 mac_list.append(mac.get('address').lower())
126 def get_name_of_device(self, vm_name, device_type):
127 resp = exec_cmd('virsh dumpxml %s' % vm_name)
128 xml_dump = etree.fromstring(resp)
129 disks = xml_dump.xpath('/domain/devices/disk')
131 if disk.get('device') == device_type:
132 targets = disk.xpath('target')
133 for target in targets:
134 device = target.get('dev')
138 def get_virt_net_conf_dir(self):
139 return self.dha_struct['virtNetConfDir']