X-Git-Url: https://gerrit.fd.io/r/gitweb?p=csit.git;a=blobdiff_plain;f=resources%2Flibraries%2Fpython%2FQemuUtils.py;h=74965ec580511094c12f99f8a7dd06c598414e93;hp=c723878499fe80e0b56fe6211bffff5acb62537b;hb=22417d56aa83eaced43c4ef4473e38a34fdb4dfa;hpb=5d5db63262e5c141e5eb435c65154cee214887af diff --git a/resources/libraries/python/QemuUtils.py b/resources/libraries/python/QemuUtils.py index c723878499..74965ec580 100644 --- a/resources/libraries/python/QemuUtils.py +++ b/resources/libraries/python/QemuUtils.py @@ -1,4 +1,4 @@ -# Copyright (c) 2016 Cisco and/or its affiliates. +# Copyright (c) 2019 Cisco and/or its affiliates. # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at: @@ -13,28 +13,547 @@ """QEMU utilities library.""" +from time import time, sleep +import json +from re import match +# Disable due to pylint bug +# pylint: disable=no-name-in-module,import-error +from distutils.version import StrictVersion + from robot.api import logger -from ssh import SSH -from constants import Constants +from resources.libraries.python.ssh import exec_cmd, exec_cmd_no_error +from resources.libraries.python.Constants import Constants +from resources.libraries.python.DUTSetup import DUTSetup +from resources.libraries.python.topology import NodeType, Topology + +__all__ = ["QemuOptions", "QemuUtils"] + + +class QemuOptions(object): + """QEMU option class. + + The class can handle input parameters or parameters that acts as QEMU + command line parameters. The only variable is a list of dictionaries + where dictionaries can be added multiple times. This emulates the QEMU + behavior where one command line parameter can be used multiple times (1..N). + Example can be device or object (so it is not an issue to have one memory + block of 2G and and second memory block of 512M but from other numa). + + Class does support get value or string representation that will return + space separated, dash prefixed string of key value pairs used for command + line. + """ + + # Use one instance of class per tests. + ROBOT_LIBRARY_SCOPE = 'TEST CASE' + + def __init__(self): + self.variables = list() + + def add(self, variable, value): + """Add parameter to the list. + + :param variable: QEMU parameter. + :param value: Parameter value. + :type variable: str + :type value: str or int + """ + self.variables.append({str(variable): value}) + def get(self, variable): + """Get parameter of variable(s) from list that matches input value. + + :param variable: QEMU parameter to get. + :type variable: str + :returns: List of values or value that matches input parameter. + :rtype: list or str + """ + selected = [d[variable] for d in self.variables if variable in d] + return selected if len(selected) > 1 else selected[0] + + def get_values(self): + """Get all values from dict items in list. + + :returns: List of all dictionary values. + :rtype: list + """ + return [d.values()[0] for d in self.variables] + + def __str__(self): + """Return space separated string of key value pairs. + + :returns: Space separated string of key value pairs. + :rtype: str + """ + return " ".join(["-{k} {v}".format(k=d.keys()[0], v=d.values()[0]) + for d in self.variables]) class QemuUtils(object): """QEMU utilities.""" + # Use one instance of class per tests. + ROBOT_LIBRARY_SCOPE = 'TEST CASE' + + def __init__(self, node=None, qemu_id=1, smp=1, mem=512, vnf=None, + img='/var/lib/vm/vhost-nested.img', bin_path='/usr/bin'): + """Initialize QemuUtil class. + + :param node: Node to run QEMU on. + :param qemu_id: QEMU identifier. + :param smp: Number of virtual SMP units (cores). + :param mem: Amount of memory. + :param vnf: Network function workload. + :param img: QEMU disk image or kernel image path. + :param bin_path: QEMU binary path. + :type node: dict + :type qemu_id: int + :type smp: int + :type mem: int + :type vnf: str + :type img: str + :type bin_path: str + """ + self._vhost_id = 0 + self._vm_info = { + 'type': NodeType.VM, + 'port': 10021 + qemu_id, + 'serial': 4555 + qemu_id, + 'username': 'cisco', + 'password': 'cisco', + 'interfaces': {}, + } + if node: + self.qemu_set_node(node) + # Input Options. + self._opt = QemuOptions() + self._opt.add('qemu_id', qemu_id) + self._opt.add('bin_path', bin_path) + self._opt.add('mem', int(mem)) + self._opt.add('smp', int(smp)) + self._opt.add('img', img) + self._opt.add('vnf', vnf) + # Temporary files. + self._temp = QemuOptions() + self._temp.add('pid', '/var/run/qemu_{id}.pid'.format(id=qemu_id)) + # Computed parameters for QEMU command line. + if '/var/lib/vm/' in img: + self._opt.add('vm_type', 'nestedvm') + self._temp.add('qmp', '/var/run/qmp_{id}.sock'.format(id=qemu_id)) + self._temp.add('qga', '/var/run/qga_{id}.sock'.format(id=qemu_id)) + else: + raise RuntimeError('QEMU: Unknown VM image option!') + self._params = QemuOptions() + self.add_params() + + def add_params(self): + """Set QEMU command line parameters.""" + self.add_default_params() + if self._opt.get('vm_type') == 'nestedvm': + self.add_nestedvm_params() + else: + raise RuntimeError('QEMU: Unsupported VM type!') + + def add_default_params(self): + """Set default QEMU command line parameters.""" + self._params.add('daemonize', '') + self._params.add('nodefaults', '') + self._params.add('name', 'vnf{qemu},debug-threads=on'. + format(qemu=self._opt.get('qemu_id'))) + self._params.add('no-user-config', '') + self._params.add('monitor', 'none') + self._params.add('display', 'none') + self._params.add('vga', 'none') + self._params.add('enable-kvm', '') + self._params.add('pidfile', '{pid}'. + format(pid=self._temp.get('pid'))) + self._params.add('cpu', 'host') + self._params.add('machine', 'pc,accel=kvm,usb=off,mem-merge=off') + self._params.add('smp', '{smp},sockets=1,cores={smp},threads=1'. + format(smp=self._opt.get('smp'))) + self._params.add('object', + 'memory-backend-file,id=mem,size={mem}M,' + 'mem-path=/mnt/huge,share=on'. + format(mem=self._opt.get('mem'))) + self._params.add('m', '{mem}M'. + format(mem=self._opt.get('mem'))) + self._params.add('numa', 'node,memdev=mem') + self._params.add('balloon', 'none') + + def add_nestedvm_params(self): + """Set NestedVM QEMU parameters.""" + self._params.add('net', 'nic,macaddr=52:54:00:00:{qemu:02x}:ff'. + format(qemu=self._opt.get('qemu_id'))) + self._params.add('net', 'user,hostfwd=tcp::{info[port]}-:22'. + format(info=self._vm_info)) + # TODO: Remove try except after fully migrated to Bionic or + # qemu_set_node is removed. + try: + locking = ',file.locking=off'\ + if self.qemu_version(version='2.10') else '' + except AttributeError: + locking = '' + self._params.add('drive', + 'file={img},format=raw,cache=none,if=virtio{locking}'. + format(img=self._opt.get('img'), locking=locking)) + self._params.add('qmp', 'unix:{qmp},server,nowait'. + format(qmp=self._temp.get('qmp'))) + self._params.add('chardev', 'socket,host=127.0.0.1,port={info[serial]},' + 'id=gnc0,server,nowait'.format(info=self._vm_info)) + self._params.add('device', 'isa-serial,chardev=gnc0') + self._params.add('chardev', + 'socket,path={qga},server,nowait,id=qga0'. + format(qga=self._temp.get('qga'))) + self._params.add('device', 'isa-serial,chardev=qga0') + + def qemu_set_node(self, node): + """Set node to run QEMU on. + + :param node: Node to run QEMU on. + :type node: dict + """ + self._node = node + self._vm_info['host'] = node['host'] + if node['port'] != 22: + self._vm_info['host_port'] = node['port'] + self._vm_info['host_username'] = node['username'] + self._vm_info['host_password'] = node['password'] + + def get_qemu_pids(self): + """Get QEMU CPU pids. + + :returns: List of QEMU CPU pids. + :rtype: list of str + """ + command = ("grep -rwl 'CPU' /proc/$(sudo cat {pid})/task/*/comm ". + format(pid=self._temp.get('pid'))) + command += (r"| xargs dirname | sed -e 's/\/.*\///g'") + + stdout, _ = exec_cmd_no_error(self._node, command) + return stdout.splitlines() + + def qemu_set_affinity(self, *host_cpus): + """Set qemu affinity by getting thread PIDs via QMP and taskset to list + of CPU cores. + + :param host_cpus: List of CPU cores. + :type host_cpus: list + """ + try: + qemu_cpus = self.get_qemu_pids() + + if len(qemu_cpus) != len(host_cpus): + raise ValueError('Host CPU count must match Qemu Thread count!') + + for qemu_cpu, host_cpu in zip(qemu_cpus, host_cpus): + command = ('taskset -pc {host_cpu} {thread}'. + format(host_cpu=host_cpu, thread=qemu_cpu)) + message = ('QEMU: Set affinity failed on {host}!'. + format(host=self._node['host'])) + exec_cmd_no_error(self._node, command, sudo=True, + message=message) + except (RuntimeError, ValueError): + self.qemu_kill_all() + raise + + def qemu_set_scheduler_policy(self): + """Set scheduler policy to SCHED_RR with priority 1 for all Qemu CPU + processes. + + :raises RuntimeError: Set scheduler policy failed. + """ + try: + qemu_cpus = self.get_qemu_pids() + + for qemu_cpu in qemu_cpus: + command = ('chrt -r -p 1 {thread}'. + format(thread=qemu_cpu)) + message = ('QEMU: Set SCHED_RR failed on {host}'. + format(host=self._node['host'])) + exec_cmd_no_error(self._node, command, sudo=True, + message=message) + except (RuntimeError, ValueError): + self.qemu_kill_all() + raise + + def qemu_add_vhost_user_if(self, socket, server=True, jumbo_frames=False, + queue_size=None, queues=1): + """Add Vhost-user interface. + + :param socket: Path of the unix socket. + :param server: If True the socket shall be a listening socket. + :param jumbo_frames: Set True if jumbo frames are used in the test. + :param queue_size: Vring queue size. + :param queues: Number of queues. + :type socket: str + :type server: bool + :type jumbo_frames: bool + :type queue_size: int + :type queues: int + """ + self._vhost_id += 1 + self._params.add('chardev', + 'socket,id=char{vhost},path={socket}{server}'. + format(vhost=self._vhost_id, socket=socket, + server=',server' if server is True else '')) + self._params.add('netdev', + 'vhost-user,id=vhost{vhost},' + 'chardev=char{vhost},queues={queues}'. + format(vhost=self._vhost_id, queues=queues)) + mac = ('52:54:00:00:{qemu:02x}:{vhost:02x}'. + format(qemu=self._opt.get('qemu_id'), vhost=self._vhost_id)) + queue_size = ('rx_queue_size={queue_size},tx_queue_size={queue_size}'. + format(queue_size=queue_size)) if queue_size else '' + mbuf = 'on,host_mtu=9200' + self._params.add('device', + 'virtio-net-pci,netdev=vhost{vhost},' + 'mac={mac},mq=on,vectors={vectors},csum=off,gso=off,' + 'guest_tso4=off,guest_tso6=off,guest_ecn=off,' + 'mrg_rxbuf={mbuf},{queue_size}'. + format(vhost=self._vhost_id, mac=mac, + mbuf=mbuf if jumbo_frames else 'off', + queue_size=queue_size, + vectors=(2 * queues + 2))) + + # Add interface MAC and socket to the node dict. + if_data = {'mac_address': mac, 'socket': socket} + if_name = 'vhost{vhost}'.format(vhost=self._vhost_id) + self._vm_info['interfaces'][if_name] = if_data + # Add socket to temporary file list. + self._temp.add(if_name, socket) + + def _qemu_qmp_exec(self, cmd): + """Execute QMP command. + + QMP is JSON based protocol which allows to control QEMU instance. + + :param cmd: QMP command to execute. + :type cmd: str + :returns: Command output in python representation of JSON format. The + { "return": {} } response is QMP's success response. An error + response will contain the "error" keyword instead of "return". + """ + # To enter command mode, the qmp_capabilities command must be issued. + command = ('echo "{{ \\"execute\\": \\"qmp_capabilities\\" }}' + '{{ \\"execute\\": \\"{cmd}\\" }}" | ' + 'sudo -S socat - UNIX-CONNECT:{qmp}'. + format(cmd=cmd, qmp=self._temp.get('qmp'))) + message = ('QMP execute "{cmd}" failed on {host}'. + format(cmd=cmd, host=self._node['host'])) + stdout, _ = exec_cmd_no_error(self._node, command, sudo=False, + message=message) + + # Skip capabilities negotiation messages. + out_list = stdout.splitlines() + if len(out_list) < 3: + raise RuntimeError('Invalid QMP output on {host}'. + format(host=self._node['host'])) + return json.loads(out_list[2]) + + def _qemu_qga_flush(self): + """Flush the QGA parser state.""" + command = ('(printf "\xFF"; sleep 1) | ' + 'sudo -S socat - UNIX-CONNECT:{qga}'. + format(qga=self._temp.get('qga'))) + message = ('QGA flush failed on {host}'.format(host=self._node['host'])) + stdout, _ = exec_cmd_no_error(self._node, command, sudo=False, + message=message) + + return json.loads(stdout.split('\n', 1)[0]) if stdout else dict() + + def _qemu_qga_exec(self, cmd): + """Execute QGA command. + + QGA provide access to a system-level agent via standard QMP commands. + + :param cmd: QGA command to execute. + :type cmd: str + """ + command = ('(echo "{{ \\"execute\\": \\"{cmd}\\" }}"; sleep 1) | ' + 'sudo -S socat - UNIX-CONNECT:{qga}'. + format(cmd=cmd, qga=self._temp.get('qga'))) + message = ('QGA execute "{cmd}" failed on {host}'. + format(cmd=cmd, host=self._node['host'])) + stdout, _ = exec_cmd_no_error(self._node, command, sudo=False, + message=message) + + return json.loads(stdout.split('\n', 1)[0]) if stdout else dict() + + def _wait_until_vm_boot(self, timeout=60): + """Wait until QEMU VM is booted. + + First try to flush qga until there is output. + Then ping QEMU guest agent each 5s until VM booted or timeout. + + :param timeout: Waiting timeout in seconds (optional, default 60s). + :type timeout: int + """ + start = time() + while True: + if time() - start > timeout: + raise RuntimeError('timeout, VM not booted on {host}'. + format(host=self._node['host'])) + out = None + try: + out = self._qemu_qga_flush() + except ValueError: + logger.trace('QGA qga flush unexpected output {out}'. + format(out=out)) + # Empty output - VM not booted yet + if not out: + sleep(5) + else: + break + while True: + if time() - start > timeout: + raise RuntimeError('timeout, VM not booted on {host}'. + format(host=self._node['host'])) + out = None + try: + out = self._qemu_qga_exec('guest-ping') + except ValueError: + logger.trace('QGA guest-ping unexpected output {out}'. + format(out=out)) + # Empty output - VM not booted yet. + if not out: + sleep(5) + # Non-error return - VM booted. + elif out.get('return') is not None: + break + # Skip error and wait. + elif out.get('error') is not None: + sleep(5) + else: + # If there is an unexpected output from QGA guest-info, try + # again until timeout. + logger.trace('QGA guest-ping unexpected output {out}'. + format(out=out)) + + logger.trace('VM booted on {host}'.format(host=self._node['host'])) + + def _update_vm_interfaces(self): + """Update interface names in VM node dict.""" + # Send guest-network-get-interfaces command via QGA, output example: + # {"return": [{"name": "eth0", "hardware-address": "52:54:00:00:04:01"}, + # {"name": "eth1", "hardware-address": "52:54:00:00:04:02"}]}. + out = self._qemu_qga_exec('guest-network-get-interfaces') + interfaces = out.get('return') + mac_name = {} + if not interfaces: + raise RuntimeError('Get VM interface list failed on {host}'. + format(host=self._node['host'])) + # Create MAC-name dict. + for interface in interfaces: + if 'hardware-address' not in interface: + continue + mac_name[interface['hardware-address']] = interface['name'] + # Match interface by MAC and save interface name. + for interface in self._vm_info['interfaces'].values(): + mac = interface.get('mac_address') + if_name = mac_name.get(mac) + if if_name is None: + logger.trace('Interface name for MAC {mac} not found'. + format(mac=mac)) + else: + interface['name'] = if_name + + def qemu_start(self): + """Start QEMU and wait until VM boot. + + :returns: VM node info. + :rtype: dict + """ + DUTSetup.check_huge_page(self._node, '/mnt/huge', self._opt.get('mem')) + + command = ('{bin_path}/qemu-system-{arch} {params}'. + format(bin_path=self._opt.get('bin_path'), + arch=Topology.get_node_arch(self._node), + params=self._params)) + message = ('QEMU: Start failed on {host}!'. + format(host=self._node['host'])) + + try: + exec_cmd_no_error(self._node, command, timeout=300, sudo=True, + message=message) + self._wait_until_vm_boot() + # Update interface names in VM node dict. + self._update_vm_interfaces() + except RuntimeError: + self.qemu_kill_all() + raise + return self._vm_info + + def qemu_kill(self): + """Kill qemu process.""" + exec_cmd(self._node, 'chmod +r {pid}'. + format(pid=self._temp.get('pid')), sudo=True) + exec_cmd(self._node, 'kill -SIGKILL $(cat {pid})'. + format(pid=self._temp.get('pid')), sudo=True) + + for value in self._temp.get_values(): + exec_cmd(self._node, 'rm -f {value}'.format(value=value), sudo=True) + + def qemu_kill_all(self, node=None): + """Kill all qemu processes on DUT node if specified. + + :param node: Node to kill all QEMU processes on. + :type node: dict + """ + if node: + self.qemu_set_node(node) + exec_cmd(self._node, 'pkill -SIGKILL qemu', sudo=True) + + for value in self._temp.get_values(): + exec_cmd(self._node, 'rm -f {value}'.format(value=value), sudo=True) + + def qemu_version(self, version=None): + """Return Qemu version or compare if version is higher than parameter. + + :param version: Version to compare. + :type version: str + :returns: Qemu version or Boolean if version is higher than parameter. + :rtype: str or bool + """ + command = ('{bin_path}/qemu-system-{arch} --version'. + format(bin_path=self._opt.get('bin_path'), + arch=Topology.get_node_arch(self._node))) + try: + stdout, _ = exec_cmd_no_error(self._node, command, sudo=True) + ver = match(r'QEMU emulator version ([\d.]*)', stdout).group(1) + return StrictVersion(ver) > StrictVersion(version) \ + if version else ver + except RuntimeError: + self.qemu_kill_all() + raise + @staticmethod - def build_qemu(node): + def build_qemu(node, force_install=False, apply_patch=False): """Build QEMU from sources. :param node: Node to build QEMU on. + :param force_install: If True, then remove previous build. + :param apply_patch: If True, then apply patches from qemu_patches dir. :type node: dict + :type force_install: bool + :type apply_patch: bool + :raises RuntimeError: If building QEMU failed. """ - ssh = SSH() - ssh.connect(node) + directory = (' --directory={install_dir}{patch}'. + format(install_dir=Constants.QEMU_INSTALL_DIR, + patch='-patch' if apply_patch else '-base')) + version = (' --version={install_version}'. + format(install_version=Constants.QEMU_INSTALL_VERSION)) + force = ' --force' if force_install else '' + patch = ' --patch' if apply_patch else '' + target_list = (' --target-list={arch}-softmmu'. + format(arch=Topology.get_node_arch(node))) - (ret_code, stdout, stderr) = \ - ssh.exec_command('sudo -Sn bash {0}/{1}/qemu_build.sh'.format( - Constants.REMOTE_FW_DIR, Constants.RESOURCES_LIB_SH), 1000) - logger.trace(stdout) - if 0 != int(ret_code): - logger.debug('QEMU build failed {0}'.format(stderr)) - raise RuntimeError('QEMU build failed on {0}'.format(node['host'])) + command = ("sudo -E sh -c " + "'{fw_dir}/{lib_sh}/qemu_build.sh{version}{directory}" + "{force}{patch}{target_list}'". + format(fw_dir=Constants.REMOTE_FW_DIR, + lib_sh=Constants.RESOURCES_LIB_SH, + version=version, directory=directory, force=force, + patch=patch, target_list=target_list)) + message = ('QEMU: Build failed on {host}!'.format(host=node['host'])) + exec_cmd_no_error(node, command, sudo=False, message=message, + timeout=1000)