#!/usr/bin/env python3 """ Script to test virtualization functionality Copyright (C) 2013, 2014 Canonical Ltd. Authors Jeff Marcom Daniel Manrique Jeff Lane This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License version 3, as published by the Free Software Foundation. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program. If not, see . """ from argparse import ArgumentParser import os import logging import lsb_release import requests import shlex from subprocess import ( Popen, PIPE, DEVNULL, CalledProcessError, check_output, call ) import sys import tempfile import tarfile import time import urllib.request from urllib.parse import urlparse from uuid import uuid4 DEFAULT_TIMEOUT = 500 # The "TAR" type is a tarball that contains both # a disk image and a kernel binary. This is useful # on architectures that don't (yet) have a bootloader # in the disk image that we can chain to, and instead # we need to have qemu load boot files externally CLOUD_IMAGE_TYPE_TAR = 1 CLOUD_IMAGE_TYPE_DISK = 2 QEMU_DISK_TYPE_SD = 1 QEMU_DISK_TYPE_VIRTIO = 2 QEMU_DISK_TYPE_VIRTIO_BLK = 3 QEMU_ARCH_CONFIG = { 'arm64': { 'cloudimg_type': CLOUD_IMAGE_TYPE_TAR, 'cloudimg_arch': 'arm64', 'qemu_bin': 'qemu-system-aarch64', 'qemu_disk_type': QEMU_DISK_TYPE_VIRTIO_BLK, 'qemu_extra_args': [ '-cpu', 'host', '-enable-kvm', ], }, 'armhf': { 'cloudimg_type': CLOUD_IMAGE_TYPE_TAR, 'cloudimg_arch': 'armhf', 'qemu_bin': 'qemu-system-arm', 'qemu_disk_type': QEMU_DISK_TYPE_VIRTIO_BLK, 'qemu_extra_args': [ '-machine', 'virt', '-cpu', 'host', '-enable-kvm', '-serial', 'stdio', ], }, 'amd64': { 'cloudimg_type': CLOUD_IMAGE_TYPE_DISK, 'cloudimg_arch': 'amd64', 'qemu_bin': 'qemu-system-x86_64', 'qemu_disk_type': QEMU_DISK_TYPE_VIRTIO, 'qemu_extra_args': [ '-machine', 'accel=kvm:tcg', ], }, 'i386': { 'cloudimg_type': CLOUD_IMAGE_TYPE_DISK, 'cloudimg_arch': 'i386', 'qemu_bin': 'qemu-system-x86_64', 'qemu_disk_type': QEMU_DISK_TYPE_VIRTIO, 'qemu_extra_args': [ '-machine', 'accel=kvm:tcg', ], }, 'ppc64el': { 'cloudimg_type': CLOUD_IMAGE_TYPE_DISK, 'cloudimg_arch': 'ppc64el', 'qemu_bin': 'qemu-system-ppc64', 'qemu_disk_type': QEMU_DISK_TYPE_VIRTIO, 'qemu_extra_args': [ '-enable-kvm', '-machine', 'pseries,usb=off', '-cpu', 'host', ], }, 's390x': { 'cloudimg_type': CLOUD_IMAGE_TYPE_DISK, 'cloudimg_arch': 's390x', 'qemu_bin': 'qemu-system-s390x', 'qemu_disk_type': QEMU_DISK_TYPE_VIRTIO, 'qemu_extra_args': [ '-enable-kvm', '-machine', 's390-ccw-virtio', ], }, } class QemuRunner(object): def __init__(self, arch): self.arch = arch self.config = QEMU_ARCH_CONFIG[arch] self.drive_id = 0 # Parameters common to all architectures self.params = [ self.config['qemu_bin'], "-m", "1024", "-display", "none", "-nographic", "-net", "nic", "-net", "user,net=10.0.0.0/8,host=10.0.0.1,hostfwd=tcp::2222-:22", ] # If arch is arm64, add the machine type for gicv3, or default to old # type if self.arch == 'arm64': (self.config['qemu_extra_args']. extend(['-machine', 'virt,gic_version=host'])) # Add any architecture-specific parameters if 'qemu_extra_args' in self.config: self.params = self.params + self.config['qemu_extra_args'] self.append = [] if self.config['cloudimg_type'] == CLOUD_IMAGE_TYPE_TAR: self.append = self.append + [ 'console=ttyAMA0', 'earlyprintk=serial', 'ro', 'rootfstype=ext4', 'root=LABEL=cloudimg-rootfs', 'rootdelay=10', ] def add_boot_files(self, kernel=None, initrd=None, dtb=None): if kernel: self.params = self.params + ['-kernel', kernel] if initrd: self.params = self.params + ['-initrd', initrd] if dtb: self.params = self.params + ['-dtb', dtb] def add_drive(self, cloudimg): drive = ["-drive"] if self.config['qemu_disk_type'] == QEMU_DISK_TYPE_SD: drive = drive + ["file=%s,if=sd,cache=writeback" % (cloudimg)] elif self.config['qemu_disk_type'] == QEMU_DISK_TYPE_VIRTIO: drive = drive + ["file=%s,if=virtio" % (cloudimg)] elif self.config['qemu_disk_type'] == QEMU_DISK_TYPE_VIRTIO_BLK: drive = drive + ["file=%s,if=none,id=disk.%d" % (cloudimg, self.drive_id)] drive = drive + ["-device", "virtio-blk-device,drive=disk.%d" % (self.drive_id)] self.params = self.params + drive self.drive_id = self.drive_id + 1 def get_params(self): params = self.params if self.append: params = params + ['-append', '"%s"' % (" ".join(self.append))] return params class KVMTest(object): def __init__(self, image=None, timeout=500, debug_file=None): self.image = image self.timeout = timeout self.debug_file = debug_file self.arch = check_output(['dpkg', '--print-architecture'], universal_newlines=True).strip() self.qemu_config = QEMU_ARCH_CONFIG[self.arch] self.release = lsb_release.get_distro_information()["CODENAME"] def url_to_path(self, image_path): """ Test the provided image path to determine if it's a URL or or a simple file path """ url = urlparse(image_path) if url.scheme == '' or url.scheme == 'file': # Gives us path wheter we specify a filesystem path or a file URL logging.debug("Cloud image exists locally at %s" % url.path) return url.path elif url.scheme == 'http' or url.scheme == 'ftp': # Gives us the stuff needed to build the URL to download the image return self.download_image(image_path) def get_image_name_and_url(self, image_url=None): """ Build a URL for official Ubuntu images hosted either at cloud-images.ubuntu.com or on a maas server hosting a mirror of cloud-images.ubuntu.com """ def _construct_filename(alt_pattern=None, initial_url=None): if self.qemu_config['cloudimg_type'] == CLOUD_IMAGE_TYPE_TAR: cloud_iso = "%s-server-cloudimg-%s.tar_gz" % ( self.release, self.qemu_config['cloudimg_arch']) elif alt_pattern == "modern": # LP 1635345 - yakkety and beyond have a new naming scheme cloud_iso = "%s-server-cloudimg-%s.img" % ( self.release, self.qemu_config['cloudimg_arch']) elif self.qemu_config['cloudimg_type'] == CLOUD_IMAGE_TYPE_DISK: cloud_iso = "%s-server-cloudimg-%s-disk1.img" % ( self.release, self.qemu_config['cloudimg_arch']) elif initial_url: # LP 1662580 - if we pass a full URL, assume the last piece is # the filname and return that. cloud_iso = initial_url.split('/')[-1] else: logging.error("Unknown cloud image type") sys.exit(1) return cloud_iso def _construct_url(initial_url, cloud_iso): return "/".join((initial_url, cloud_iso)) def _test_cloud_url(url): # test our URL to make sure it's reachable try: ret = requests.head(url) except OSError as e: logging.error("Unable to connect to {}".format(url)) logging.error(" * Message: {}".format(e.with_traceback(None))) return False if ret.status_code != 200: return False else: return True if image_url is None: # If we have not specified a URL to get our images from, default # to ubuntu.com cloud_iso = _construct_filename() initial_url = "/".join(("http://cloud-images.ubuntu.com", self.release, "current")) full_url = _construct_url(initial_url, cloud_iso) # Test our URL and rebuild with alternate name if not _test_cloud_url(full_url): logging.warn("Cloud Image URL not valid: %s" % full_url) logging.warn(" * This means we could not reach the remote " "file. Retrying with a different filename " "schema.") cloud_iso = _construct_filename("modern") full_url = _construct_url(initial_url, cloud_iso) # retest one more time then exit if it still fails if not _test_cloud_url(full_url): logging.error("Cloud URL is not valid: %s" % full_url) logging.error(" * It appears that there is a problem " "finding the expected file. Check the URL " "noted above.") sys.exit(1) else: url = urlparse(image_url) if ( url.path.endswith('/') or url.path == '' or not (url.path.endswith(".img") or url.path.endswith(".tar_gz")) ): # If we have a relative URL (local copies of official images) # http://192.168.0.1/ or http://192.168.0.1/images/ cloud_iso = _construct_filename() full_url = _construct_url(image_url.rstrip("/"), cloud_iso) if not _test_cloud_url(full_url): logging.warn("Cloud Image URL not valid: %s" % full_url) logging.warn(" * This means we could not reach the remote " "file. Retrying with a different filename " "schema.") cloud_iso = _construct_filename("modern") full_url = _construct_url(image_url.rstrip("/"), cloud_iso) if not _test_cloud_url(full_url): logging.error("Cloud URL is not valid: %s" % full_url) logging.error(" * It appears that there is a problem " "finding the expected file. Check the " "URL noted above.") sys.exit(1) else: # Assume anything else is an absolute URL to a remote server if not _test_cloud_url(image_url): logging.error("Cloud Image URL invalid: %s" % image_url) logging.error(" * Check the URL and ensure it is correct") sys.exit(1) else: full_url = image_url cloud_iso = _construct_filename(initial_url=full_url) return full_url, cloud_iso def download_image(self, image_url=None): """ Downloads Cloud image for same release as host machine """ if image_url is None: full_url, cloud_iso = self.get_image_name_and_url() else: full_url, cloud_iso = self.get_image_name_and_url(image_url) logging.debug("Acquiring cloud image from: {}".format(full_url)) # Attempt download try: resp = urllib.request.urlretrieve(full_url, cloud_iso) except (IOError, OSError, urllib.error.HTTPError, urllib.error.URLError) as exception: logging.error("Failed download of image from %s: %s", image_url, exception) return False # Unpack img file from tar if self.qemu_config['cloudimg_type'] == CLOUD_IMAGE_TYPE_TAR: cloud_iso_tgz = tarfile.open(cloud_iso) cloud_iso = cloud_iso.replace('tar_gz', 'img') cloud_iso_tgz.extract(cloud_iso) if not os.path.isfile(cloud_iso): return False return cloud_iso def boot_image(self, data_disk): """ Attempts to boot the newly created qcow image using the config data defined in config.iso """ logging.debug("Attempting boot for:{}".format(data_disk)) qemu = QemuRunner(self.arch) # Assume that a tar type image is not self-bootable, so # therefore requires explicit bootfiles (otherwise, why # not just use the disk format directly? if self.qemu_config['cloudimg_type'] == CLOUD_IMAGE_TYPE_TAR: for dir in ['/boot', '/']: kernel = os.path.join(dir, 'vmlinuz') initrd = os.path.join(dir, 'initrd.img') if os.path.isfile(kernel): qemu.add_boot_files(kernel=kernel, initrd=initrd) break qemu.add_drive(data_disk) # Should we attach the cloud config disk if os.path.isfile("seed.iso"): logging.debug("Attaching Cloud config disk") qemu.add_drive("seed.iso") params = qemu.get_params() logging.debug("Using params:{}".format(" ".join(params))) logging.info("Storing VM console output in {}".format( os.path.realpath(self.debug_file))) # Open VM STDERR/STDOUT log file for writing try: file = open(self.debug_file, 'w') except IOError: logging.error("Failed creating file:{}".format(self.debug_file)) return False # Start Virtual machine self.process = Popen( params, stdin=PIPE, stderr=file, stdout=file, universal_newlines=True, shell=False) def create_cloud_disk(self): """ Generate Cloud meta data and creates an iso object to be mounted as virtual device to instance during boot. """ user_data = """\ #cloud-config runcmd: - [ sh, -c, echo "========= CERTIFICATION TEST =========" ] power_state: mode: halt message: Bye timeout: 480 final_message: CERTIFICATION BOOT COMPLETE """ meta_data = """\ { echo instance-id: iid-local01; echo local-hostname, certification; } """ for file in ['user-data', 'meta-data']: logging.debug("Creating cloud %s", file) with open(file, "wt") as data_file: os.fchmod(data_file.fileno(), 0o777) data_file.write(vars()[file.replace("-", "_")]) # Create Data ISO hosting user & meta cloud config data try: iso_build = check_output( ['genisoimage', '-output', 'seed.iso', '-volid', 'cidata', '-joliet', '-rock', 'user-data', 'meta-data'], universal_newlines=True) except CalledProcessError as exception: logging.exception("Cloud data disk creation failed") def log_check(self, stream): if "CERTIFICATION BOOT COMPLETE" in stream: return 0 else: return 1 def start(self): if self.arch == 'arm64': # lp:1548539 - For arm64, we need to make sure we're using qemu # later than 2.0.0 to enable gic_version functionality logging.debug('Checking QEMU version for arm64 arch') cmd = 'apt-cache policy qemu-system-arm | grep Installed' installed_version = (check_output(['/bin/bash', '-c', cmd]). decode().split(':', 1)[1].strip()) cmd = ('dpkg --compare-versions \"2.0.0\" \"lt\" \"{}\"' .format(installed_version)) retcode = call(['/bin/bash', '-c', cmd]) if retcode != 0: logging.error('arm64 needs qemu-system version later than ' '2.0.0') return 1 logging.debug('Starting KVM Test') status = 1 # Create temp directory: date = time.strftime("%b_%d_%Y_") with tempfile.TemporaryDirectory("_kvm_test", date) as temp_dir: os.chmod(temp_dir, 0o744) os.chdir(temp_dir) if not self.image: logging.debug('No image specified, downloading one now.') # Download cloud image self.image = self.download_image() else: logging.debug('Cloud image location specified: %s' % self.image) self.image = self.url_to_path(self.image) if self.image and os.path.isfile(self.image): if "cloud" in self.image: # Will assume we need to supply cloud meta data # for instance boot to be successful self.create_cloud_disk() # Boot Virtual Machine instance = self.boot_image(self.image) # If running in console, reset console window to regain # control from VM Serial I/0 if sys.stdout.isatty(): call('reset') # Check to be sure VM boot was successful self.elapsed_time = 0 status = 1 while self.elapsed_time <= self.timeout: # Check log every 30 seconds to see if the VM boots with open(self.debug_file, 'r') as debug_file: status = self.log_check(debug_file.read()) if status == 0: logging.info("Booted successfully.") break else: # Sleep 30 seconds and log check again time.sleep(30) self.elapsed_time += 30 else: # Finally, if we didn't get the Success message by now try # one more time and return 1 if we still haven't booted if status != 0: with open(self.debug_file, 'r') as debug_file: stream = debug_file.read() status = self.log_check(stream) if status == 0: logging.info("Booted successfully.") else: logging.error("KVM instance failed to boot.") logging.error("Console output".center(72, "=")) logging.error(stream) self.process.terminate() elif not self.image: logging.error("Could not find downloaded image") else: logging.error("Could not find: {}".format(self.image)) return status class RunCommand(object): """ Runs a command and can return all needed info: * stdout * stderr * return code * original command Convenince class to avoid the same repetitive code to run shell commands. """ def __init__(self, cmd=None): self.stdout = None self.stderr = None self.returncode = None self.cmd = cmd self.run(self.cmd) def run(self, cmd): proc = Popen(shlex.split(cmd), stdout=PIPE, stderr=PIPE, stdin=DEVNULL, universal_newlines=True) self.stdout, self.stderr = proc.communicate() self.returncode = proc.returncode class LXDTest(object): def __init__(self, template=None, rootfs=None): self.rootfs_url = rootfs self.template_url = template self.rootfs_tarball = None self.template_tarball = None self.name = 'testbed' self.image_alias = uuid4().hex self.default_remote = "ubuntu:" self.os_version = lsb_release.get_distro_information()["RELEASE"] def run_command(self, cmd): task = RunCommand(cmd) if task.returncode != 0: logging.error('Command {} returnd a code of {}'.format( task.cmd, task.returncode)) logging.error(' STDOUT: {}'.format(task.stdout)) logging.error(' STDERR: {}'.format(task.stderr)) return False else: logging.debug('Command {}:'.format(task.cmd)) if task.stdout != '': logging.debug(' STDOUT: {}'.format(task.stdout)) elif task.stderr != '': logging.debug(' STDERR: {}'.format(task.stderr)) else: logging.debug(' Command returned no output') return True def setup(self): # Initialize LXD result = True logging.debug("Attempting to initialize LXD") # TODO: Need a method to see if LXD is already initialized if not self.run_command('lxd init --auto'): logging.debug('Error encounterd while initializing LXD') result = False # Retrieve and insert LXD images if self.template_url is not None: logging.debug("Downloading template.") targetfile = urlparse(self.template_url).path.split('/')[-1] filename = os.path.join('/tmp', targetfile) if not os.path.isfile(filename): self.template_tarball = self.download_images(self.template_url, filename) if not self.template_tarball: logging.error("Unable to download {} from " "{}".format(self.template_tarball, self.template_url)) logging.error("Aborting") result = False else: logging.debug("Template file {} already exists. " "Skipping Download.".format(filename)) self.template_tarball = filename if self.rootfs_url is not None: logging.debug("Downloading rootfs.") targetfile = urlparse(self.rootfs_url).path.split('/')[-1] filename = os.path.join('/tmp', targetfile) if not os.path.isfile(filename): self.rootfs_tarball = self.download_images(self.rootfs_url, filename) if not self.rootfs_tarball: logging.error("Unable to download {} from{}".format( self.rootfs_tarball, self.rootfs_url)) logging.error("Aborting") result = False else: logging.debug("Template file {} already exists. " "Skipping Download.".format(filename)) self.rootfs_tarball = filename # Insert images if self.template_url is not None and self.rootfs_url is not None: logging.debug("Importing images into LXD") cmd = 'lxc image import {} rootfs {} --alias {}'.format( self.template_tarball, self.rootfs_tarball, self.image_alias) result = self.run_command(cmd) if not result: logging.error('Error encountered while attempting to ' 'import images into LXD') result = False else: logging.debug("No local image available, attempting to " "import from default remote.") retry = 2 cmd = 'lxc image copy {}{} local: --alias {}'.format( self.default_remote, self.os_version, self.image_alias) result = self.run_command(cmd) while not result and retry > 0: logging.error('Error encountered while attempting to ' 'import images from default remote.') logging.error('Retrying up to {} times.'.format(retry)) result = self.run_command(cmd) retry -= 1 return result def download_images(self, url, filename): """ Downloads LXD files for same release as host machine """ # TODO: Clean this up to use a non-internet simplestream on MAAS server logging.debug("Attempting download of {} from {}".format(filename, url)) try: resp = urllib.request.urlretrieve(url, filename) except (IOError, OSError, urllib.error.HTTPError, urllib.error.URLError) as exception: logging.error("Failed download of image from %s: %s", url, exception) return False except ValueError as verr: logging.error("Invalid URL %s" % url) logging.error("%s" % verr) return False if not os.path.isfile(filename): logging.warn("Can not find {}".format(filename)) return False return filename def cleanup(self): """ Clean up test files an containers created """ logging.debug('Cleaning up images and containers created during test') self.run_command('lxc image delete {}'.format(self.image_alias)) self.run_command('lxc delete --force {}'.format(self.name)) def start(self): """ Creates a container and performs the test """ result = self.setup() if not result: logging.error("One or more setup stages failed.") return False # Create container logging.debug("Launching container") if not self.run_command('lxc launch {} {}'.format(self.image_alias, self.name)): return False logging.debug("Container listing:") cmd = ("lxc list") if not self.run_command(cmd): return False logging.debug("Testing container") cmd = ("lxc exec {} dd if=/dev/urandom of=testdata.txt " "bs=1024 count=1000".format(self.name)) if not self.run_command(cmd): return False return True def test_lxd(args): logging.debug("Executing LXD Test") template = None rootfs = None # First in priority are environment variables. if 'LXD_TEMPLATE' in os.environ: template = os.environ['LXD_TEMPLATE'] if 'LXD_ROOTFS' in os.environ: rootfs = os.environ['LXD_ROOTFS'] # Finally, highest-priority are command line arguments. if args.template: template = args.template if args.rootfs: rootfs = args.rootfs lxd_test = LXDTest(template, rootfs) result = lxd_test.start() lxd_test.cleanup() if result: print("PASS: Container was succssfully started and checked") sys.exit(0) else: print("FAIL: Container was not started and checked") sys.exit(1) def test_kvm(args): logging.debug("Executing KVM Test") image = "" timeout = "" # First in priority are environment variables. if 'KVM_TIMEOUT' in os.environ: try: timeout = float(os.environ['KVM_TIMEOUT']) except ValueError as err: logging.warning("TIMEOUT env variable: %s" % err) timeout = DEFAULT_TIMEOUT if 'KVM_IMAGE' in os.environ: image = os.environ['KVM_IMAGE'] # Finally, highest-priority are command line arguments. if args.timeout: timeout = args.timeout elif not timeout: timeout = DEFAULT_TIMEOUT if args.image: image = args.image kvm_test = KVMTest(image, timeout, args.log_file) # If arch is ppc64el, disable smt if kvm_test.arch == 'ppc64el': os.system("/usr/sbin/ppc64_cpu --smt=off") result = kvm_test.start() # If arch is ppc64el, re-enable smt if kvm_test.arch == 'ppc64el': os.system("/usr/sbin/ppc64_cpu --smt=on") sys.exit(result) def main(): parser = ArgumentParser(description="Virtualization Test") subparsers = parser.add_subparsers() # Main cli options kvm_test_parser = subparsers.add_parser( 'kvm', help=("Run kvm virtualization test")) lxd_test_parser = subparsers.add_parser( 'lxd', help=("Run the LXD validation test")) parser.add_argument('--debug', dest='log_level', action="store_const", const=logging.DEBUG, default=logging.INFO) # Sub test options kvm_test_parser.add_argument( '-i', '--image', type=str, default=None) kvm_test_parser.add_argument( '-t', '--timeout', type=int) kvm_test_parser.add_argument( '-l', '--log-file', default='virt_debug', help="Location for debugging output log. Defaults to %(default)s.") kvm_test_parser.set_defaults(func=test_kvm) lxd_test_parser.add_argument( '--template', type=str, default=None) lxd_test_parser.add_argument( '--rootfs', type=str, default=None) lxd_test_parser.set_defaults(func=test_lxd) args = parser.parse_args() try: logging.basicConfig(level=args.log_level) except AttributeError: pass # avoids exception when trying to run without specifying 'kvm' # silence normal output from requests module logging.getLogger("requests").setLevel(logging.WARNING) logging.getLogger("urllib3").setLevel(logging.WARNING) # Verify args try: args.func(args) except AttributeError: parser.print_help() return 1 if __name__ == "__main__": sys.exit(main())