deploy/dha_adapters/libvirt_adapter.py

   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 ###############################################################################
   9
  10
  11 from lxml import etree
  12 from hardware_adapter import HardwareAdapter
  13 import tempfile
  14
  15 from common import (
  16     log,
  17     exec_cmd,
  18     err,
  19     delete,
  20 )
  21
  22 DEV = {'pxe': 'network',
  23        'disk': 'hd',
  24        'iso': 'cdrom'}
  25
  26
  27 class LibvirtAdapter(HardwareAdapter):
  28
  29     def __init__(self, yaml_path):
  30         super(LibvirtAdapter, self).__init__(yaml_path)
  31         self.parser = etree.XMLParser(remove_blank_text=True)
  32
  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)
  39
  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)
  46
  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)
  51
  52     def translate(self, boot_order_list):
  53         translated = []
  54         for boot_dev in boot_order_list:
  55             if boot_dev in DEV:
  56                 translated.append(DEV[boot_dev])
  57             else:
  58                 err('Boot device %s not recognized' % boot_dev)
  59         return translated
  60
  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')
  69         for o in os:
  70             for bootelem in ['boot', 'bootmenu']:
  71                 boot = o.xpath(bootelem)
  72                 for b in boot:
  73                     o.remove(b)
  74             for dev in boot_order_list:
  75                 b = etree.Element('boot')
  76                 b.set('dev', dev)
  77                 o.append(b)
  78             bmenu = etree.Element('bootmenu')
  79             bmenu.set('enable', 'no')
  80             o.append(bmenu)
  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)
  86         delete(temp_dir)
  87
  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')
  93         for disk in disks:
  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]
  99                     delete(disk_file)
 100                     exec_cmd('fallocate -l %s %s' % (disk_size, disk_file))
 101
 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)
 107
 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))
 113
 114     def get_node_pxe_mac(self, node_id):
 115         mac_list = []
 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')
 122             for mac in macs:
 123                 mac_list.append(mac.get('address').lower())
 124         return mac_list
 125
 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')
 130         for disk in disks:
 131             if disk.get('device') == device_type:
 132                 targets = disk.xpath('target')
 133                 for target in targets:
 134                     device = target.get('dev')
 135                     if device:
 136                         return device
 137
 138     def get_virt_net_conf_dir(self):
 139         return self.dha_struct['virtNetConfDir']