apex/undercloud/undercloud.py

   1 ##############################################################################
   2 # Copyright (c) 2017 Tim Rozet (trozet@redhat.com) and others.
   3 #
   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 import libvirt
  11 import logging
  12 import os
  13 import platform
  14 import shutil
  15 import subprocess
  16 import time
  17
  18 from apex.virtual import utils as virt_utils
  19 from apex.virtual import configure_vm as vm_lib
  20 from apex.common import constants
  21 from apex.common import utils
  22
  23
  24 class ApexUndercloudException(Exception):
  25     pass
  26
  27
  28 class Undercloud:
  29     """
  30     This class represents an Apex Undercloud VM
  31     """
  32     def __init__(self, image_path, template_path,
  33                  root_pw=None, external_network=False,
  34                  image_name='undercloud.qcow2'):
  35         self.ip = None
  36         self.root_pw = root_pw
  37         self.external_net = external_network
  38         self.volume = os.path.join(constants.LIBVIRT_VOLUME_PATH,
  39                                    'undercloud.qcow2')
  40         self.image_path = image_path
  41         self.image_name = image_name
  42         self.template_path = template_path
  43         self.vm = None
  44         if Undercloud._get_vm():
  45             logging.error("Undercloud VM already exists.  Please clean "
  46                           "before creating")
  47             raise ApexUndercloudException("Undercloud VM already exists!")
  48         self.create()
  49
  50     @staticmethod
  51     def _get_vm():
  52         conn = libvirt.open('qemu:///system')
  53         try:
  54             vm = conn.lookupByName('undercloud')
  55             return vm
  56         except libvirt.libvirtError:
  57             logging.debug("No undercloud VM exists")
  58
  59     def create(self):
  60         networks = ['admin']
  61         if self.external_net:
  62             networks.append('external')
  63         console = 'ttyAMA0' if platform.machine() == 'aarch64' else 'ttyS0'
  64         root = 'vda' if platform.machine() == 'aarch64' else 'sda'
  65
  66         self.vm = vm_lib.create_vm(name='undercloud',
  67                                    image=self.volume,
  68                                    baremetal_interfaces=networks,
  69                                    direct_boot='overcloud-full',
  70                                    kernel_args=['console={}'.format(console),
  71                                                 'root=/dev/{}'.format(root)],
  72                                    default_network=True,
  73                                    template_dir=self.template_path)
  74         self.setup_volumes()
  75         self.inject_auth()
  76
  77     def _set_ip(self):
  78         ip_out = self.vm.interfaceAddresses(
  79             libvirt.VIR_DOMAIN_INTERFACE_ADDRESSES_SRC_LEASE, 0)
  80         if ip_out:
  81             for (name, val) in ip_out.items():
  82                 for ipaddr in val['addrs']:
  83                     if ipaddr['type'] == libvirt.VIR_IP_ADDR_TYPE_IPV4:
  84                         self.ip = ipaddr['addr']
  85                         return True
  86
  87     def start(self):
  88         """
  89         Start Undercloud VM
  90         :return: None
  91         """
  92         if self.vm.isActive():
  93             logging.info("Undercloud already started")
  94         else:
  95             logging.info("Starting undercloud")
  96             self.vm.create()
  97             # give 10 seconds to come up
  98             time.sleep(10)
  99         # set IP
 100         for x in range(5):
 101             if self._set_ip():
 102                 logging.info("Undercloud started.  IP Address: {}".format(
 103                     self.ip))
 104                 break
 105             logging.debug("Did not find undercloud IP in {} "
 106                           "attempts...".format(x))
 107             time.sleep(10)
 108         else:
 109             logging.error("Cannot find IP for Undercloud")
 110             raise ApexUndercloudException(
 111                 "Unable to find IP for undercloud.  Check if VM booted "
 112                 "correctly")
 113
 114     def configure(self, net_settings, playbook, apex_temp_dir):
 115         """
 116         Configures undercloud VM
 117         :param net_setings: Network settings for deployment
 118         :param playbook: playbook to use to configure undercloud
 119         :param apex_temp_dir: temporary apex directory to hold configs/logs
 120         :return: None
 121         """
 122
 123         logging.info("Configuring Undercloud...")
 124         # run ansible
 125         ansible_vars = Undercloud.generate_config(net_settings)
 126         ansible_vars['apex_temp_dir'] = apex_temp_dir
 127         try:
 128             utils.run_ansible(ansible_vars, playbook, host=self.ip,
 129                               user='stack')
 130         except subprocess.CalledProcessError:
 131             logging.error(
 132                 "Failed to install undercloud..."
 133                 "please check log: {}".format(os.path.join(
 134                     apex_temp_dir, 'apex-undercloud-install.log')))
 135             raise ApexUndercloudException('Failed to install undercloud')
 136         logging.info("Undercloud installed!")
 137
 138     def setup_volumes(self):
 139         for img_file in ('overcloud-full.vmlinuz', 'overcloud-full.initrd',
 140                          self.image_name):
 141             src_img = os.path.join(self.image_path, img_file)
 142             if img_file == self.image_name:
 143                 dest_img = os.path.join(constants.LIBVIRT_VOLUME_PATH,
 144                                         'undercloud.qcow2')
 145             else:
 146                 dest_img = os.path.join(constants.LIBVIRT_VOLUME_PATH,
 147                                         img_file)
 148             if not os.path.isfile(src_img):
 149                 raise ApexUndercloudException(
 150                     "Required source file does not exist:{}".format(src_img))
 151             if os.path.exists(dest_img):
 152                 os.remove(dest_img)
 153             shutil.copyfile(src_img, dest_img)
 154             shutil.chown(dest_img, user='qemu', group='qemu')
 155             os.chmod(dest_img, 0o0744)
 156         # TODO(trozet):check if resize needed right now size is 50gb
 157         # there is a lib called vminspect which has some dependencies and is
 158         # not yet available in pip.  Consider switching to this lib later.
 159
 160     def inject_auth(self):
 161         virt_ops = list()
 162         # virt-customize keys/pws
 163         if self.root_pw:
 164             pw_op = "password:{}".format(self.root_pw)
 165             virt_ops.append({constants.VIRT_PW: pw_op})
 166         # ssh key setup
 167         virt_ops.append({constants.VIRT_RUN_CMD:
 168                         'mkdir -p /root/.ssh'})
 169         virt_ops.append({constants.VIRT_UPLOAD:
 170                          '/root/.ssh/id_rsa.pub:/root/.ssh/authorized_keys'})
 171         run_cmds = [
 172             'chmod 600 /root/.ssh/authorized_keys',
 173             'restorecon /root/.ssh/authorized_keys',
 174             'cp /root/.ssh/authorized_keys /home/stack/.ssh/',
 175             'chown stack:stack /home/stack/.ssh/authorized_keys',
 176             'chmod 600 /home/stack/.ssh/authorized_keys'
 177         ]
 178         for cmd in run_cmds:
 179             virt_ops.append({constants.VIRT_RUN_CMD: cmd})
 180         virt_utils.virt_customize(virt_ops, self.volume)
 181
 182     @staticmethod
 183     def generate_config(ns):
 184         """
 185         Generates a dictionary of settings for configuring undercloud
 186         :param ns: network settings to derive undercloud settings
 187         :return: dictionary of settings
 188         """
 189
 190         ns_admin = ns['networks']['admin']
 191         intro_range = ns['apex']['networks']['admin']['introspection_range']
 192         config = dict()
 193         config['undercloud_config'] = [
 194             "enable_ui false",
 195             "undercloud_update_packages false",
 196             "undercloud_debug false",
 197             "inspection_extras false",
 198             "undercloud_hostname undercloud.{}".format(ns['dns-domain']),
 199             "local_ip {}/{}".format(str(ns_admin['installer_vm']['ip']),
 200                                     str(ns_admin['cidr']).split('/')[1]),
 201             "network_gateway {}".format(str(ns_admin['installer_vm']['ip'])),
 202             "network_cidr {}".format(str(ns_admin['cidr'])),
 203             "dhcp_start {}".format(str(ns_admin['dhcp_range'][0])),
 204             "dhcp_end {}".format(str(ns_admin['dhcp_range'][1])),
 205             "inspection_iprange {}".format(','.join(intro_range))
 206         ]
 207
 208         config['ironic_config'] = [
 209             "disk_utils iscsi_verify_attempts 30",
 210             "disk_partitioner check_device_max_retries 40"
 211         ]
 212
 213         config['nova_config'] = [
 214             "dns_domain {}".format(ns['dns-domain']),
 215             "dhcp_domain {}".format(ns['dns-domain'])
 216         ]
 217
 218         config['neutron_config'] = [
 219             "dns_domain {}".format(ns['dns-domain']),
 220         ]
 221         # FIXME(trozet): possible bug here with not using external network
 222         ns_external = ns['networks']['external'][0]
 223         config['external_network'] = {
 224             "vlan": ns_external['installer_vm']['vlan'],
 225             "ip": ns_external['installer_vm']['ip'],
 226             "prefix": str(ns_external['cidr']).split('/')[1],
 227             "enabled": ns_external['enabled']
 228         }
 229
 230         # Check if this is an ARM deployment
 231         config['aarch64'] = platform.machine() == 'aarch64'
 232
 233         return config