1beb6b57bSJohn Snow""" 2beb6b57bSJohn SnowQEMU machine module: 3beb6b57bSJohn Snow 4beb6b57bSJohn SnowThe machine module primarily provides the QEMUMachine class, 5beb6b57bSJohn Snowwhich provides facilities for managing the lifetime of a QEMU VM. 6beb6b57bSJohn Snow""" 7beb6b57bSJohn Snow 8beb6b57bSJohn Snow# Copyright (C) 2015-2016 Red Hat Inc. 9beb6b57bSJohn Snow# Copyright (C) 2012 IBM Corp. 10beb6b57bSJohn Snow# 11beb6b57bSJohn Snow# Authors: 12beb6b57bSJohn Snow# Fam Zheng <famz@redhat.com> 13beb6b57bSJohn Snow# 14beb6b57bSJohn Snow# This work is licensed under the terms of the GNU GPL, version 2. See 15beb6b57bSJohn Snow# the COPYING file in the top-level directory. 16beb6b57bSJohn Snow# 17beb6b57bSJohn Snow# Based on qmp.py. 18beb6b57bSJohn Snow# 19beb6b57bSJohn Snow 20beb6b57bSJohn Snowimport errno 21beb6b57bSJohn Snowfrom itertools import chain 225690b437SJohn Snowimport locale 23beb6b57bSJohn Snowimport logging 24beb6b57bSJohn Snowimport os 25beb6b57bSJohn Snowimport shutil 26beb6b57bSJohn Snowimport signal 27beb6b57bSJohn Snowimport socket 28beb6b57bSJohn Snowimport subprocess 29beb6b57bSJohn Snowimport tempfile 30beb6b57bSJohn Snowfrom types import TracebackType 31beb6b57bSJohn Snowfrom typing import ( 32beb6b57bSJohn Snow Any, 33beb6b57bSJohn Snow BinaryIO, 34beb6b57bSJohn Snow Dict, 35beb6b57bSJohn Snow List, 36beb6b57bSJohn Snow Optional, 37beb6b57bSJohn Snow Sequence, 38beb6b57bSJohn Snow Tuple, 39beb6b57bSJohn Snow Type, 4015c3b863SVladimir Sementsov-Ogievskiy TypeVar, 41beb6b57bSJohn Snow) 42beb6b57bSJohn Snow 4337094b6dSJohn Snowfrom qemu.qmp import SocketAddrT 4437094b6dSJohn Snowfrom qemu.qmp.legacy import ( 45a4225303SJohn Snow QEMUMonitorProtocol, 46beb6b57bSJohn Snow QMPMessage, 47beb6b57bSJohn Snow QMPReturnValue, 48beb6b57bSJohn Snow) 49beb6b57bSJohn Snow 50beb6b57bSJohn Snowfrom . import console_socket 51beb6b57bSJohn Snow 52beb6b57bSJohn Snow 53beb6b57bSJohn SnowLOG = logging.getLogger(__name__) 54beb6b57bSJohn Snow 55beb6b57bSJohn Snow 56beb6b57bSJohn Snowclass QEMUMachineError(Exception): 57beb6b57bSJohn Snow """ 58beb6b57bSJohn Snow Exception called when an error in QEMUMachine happens. 59beb6b57bSJohn Snow """ 60beb6b57bSJohn Snow 61beb6b57bSJohn Snow 62beb6b57bSJohn Snowclass QEMUMachineAddDeviceError(QEMUMachineError): 63beb6b57bSJohn Snow """ 64beb6b57bSJohn Snow Exception raised when a request to add a device can not be fulfilled 65beb6b57bSJohn Snow 66beb6b57bSJohn Snow The failures are caused by limitations, lack of information or conflicting 67beb6b57bSJohn Snow requests on the QEMUMachine methods. This exception does not represent 68beb6b57bSJohn Snow failures reported by the QEMU binary itself. 69beb6b57bSJohn Snow """ 70beb6b57bSJohn Snow 71beb6b57bSJohn Snow 7250465f94SJohn Snowclass VMLaunchFailure(QEMUMachineError): 7350465f94SJohn Snow """ 7450465f94SJohn Snow Exception raised when a VM launch was attempted, but failed. 7550465f94SJohn Snow """ 7650465f94SJohn Snow def __init__(self, exitcode: Optional[int], 7750465f94SJohn Snow command: str, output: Optional[str]): 7850465f94SJohn Snow super().__init__(exitcode, command, output) 7950465f94SJohn Snow self.exitcode = exitcode 8050465f94SJohn Snow self.command = command 8150465f94SJohn Snow self.output = output 8250465f94SJohn Snow 8350465f94SJohn Snow def __str__(self) -> str: 8450465f94SJohn Snow ret = '' 8550465f94SJohn Snow if self.__cause__ is not None: 8650465f94SJohn Snow name = type(self.__cause__).__name__ 8750465f94SJohn Snow reason = str(self.__cause__) 8850465f94SJohn Snow if reason: 8950465f94SJohn Snow ret += f"{name}: {reason}" 9050465f94SJohn Snow else: 9150465f94SJohn Snow ret += f"{name}" 9250465f94SJohn Snow ret += '\n' 9350465f94SJohn Snow 9450465f94SJohn Snow if self.exitcode is not None: 9550465f94SJohn Snow ret += f"\tExit code: {self.exitcode}\n" 9650465f94SJohn Snow ret += f"\tCommand: {self.command}\n" 9750465f94SJohn Snow ret += f"\tOutput: {self.output}\n" 9850465f94SJohn Snow return ret 9950465f94SJohn Snow 10050465f94SJohn Snow 101beb6b57bSJohn Snowclass AbnormalShutdown(QEMUMachineError): 102beb6b57bSJohn Snow """ 103beb6b57bSJohn Snow Exception raised when a graceful shutdown was requested, but not performed. 104beb6b57bSJohn Snow """ 105beb6b57bSJohn Snow 106beb6b57bSJohn Snow 10715c3b863SVladimir Sementsov-Ogievskiy_T = TypeVar('_T', bound='QEMUMachine') 10815c3b863SVladimir Sementsov-Ogievskiy 10915c3b863SVladimir Sementsov-Ogievskiy 110beb6b57bSJohn Snowclass QEMUMachine: 111beb6b57bSJohn Snow """ 112beb6b57bSJohn Snow A QEMU VM. 113beb6b57bSJohn Snow 114beb6b57bSJohn Snow Use this object as a context manager to ensure 115beb6b57bSJohn Snow the QEMU process terminates:: 116beb6b57bSJohn Snow 117beb6b57bSJohn Snow with VM(binary) as vm: 118beb6b57bSJohn Snow ... 119beb6b57bSJohn Snow # vm is guaranteed to be shut down here 120beb6b57bSJohn Snow """ 12182e6517dSJohn Snow # pylint: disable=too-many-instance-attributes, too-many-public-methods 122beb6b57bSJohn Snow 123beb6b57bSJohn Snow def __init__(self, 124beb6b57bSJohn Snow binary: str, 125beb6b57bSJohn Snow args: Sequence[str] = (), 126beb6b57bSJohn Snow wrapper: Sequence[str] = (), 127beb6b57bSJohn Snow name: Optional[str] = None, 128beb6b57bSJohn Snow base_temp_dir: str = "/var/tmp", 129beb6b57bSJohn Snow monitor_address: Optional[SocketAddrT] = None, 130beb6b57bSJohn Snow sock_dir: Optional[str] = None, 131beb6b57bSJohn Snow drain_console: bool = False, 132b306e26cSCleber Rosa console_log: Optional[str] = None, 133e2f948a8SEmanuele Giuseppe Esposito log_dir: Optional[str] = None, 134ada73a49SVladimir Sementsov-Ogievskiy qmp_timer: Optional[float] = 30): 135beb6b57bSJohn Snow ''' 136beb6b57bSJohn Snow Initialize a QEMUMachine 137beb6b57bSJohn Snow 138beb6b57bSJohn Snow @param binary: path to the qemu binary 139beb6b57bSJohn Snow @param args: list of extra arguments 140beb6b57bSJohn Snow @param wrapper: list of arguments used as prefix to qemu binary 141beb6b57bSJohn Snow @param name: prefix for socket and log file names (default: qemu-PID) 142beb6b57bSJohn Snow @param base_temp_dir: default location where temp files are created 143beb6b57bSJohn Snow @param monitor_address: address for QMP monitor 144beb6b57bSJohn Snow @param sock_dir: where to create socket (defaults to base_temp_dir) 145beb6b57bSJohn Snow @param drain_console: (optional) True to drain console socket to buffer 146beb6b57bSJohn Snow @param console_log: (optional) path to console log file 147b306e26cSCleber Rosa @param log_dir: where to create and keep log files 148e2f948a8SEmanuele Giuseppe Esposito @param qmp_timer: (optional) default QMP socket timeout 149beb6b57bSJohn Snow @note: Qemu process is not started until launch() is used. 150beb6b57bSJohn Snow ''' 15182e6517dSJohn Snow # pylint: disable=too-many-arguments 15282e6517dSJohn Snow 153beb6b57bSJohn Snow # Direct user configuration 154beb6b57bSJohn Snow 155beb6b57bSJohn Snow self._binary = binary 156beb6b57bSJohn Snow self._args = list(args) 157beb6b57bSJohn Snow self._wrapper = wrapper 158e2f948a8SEmanuele Giuseppe Esposito self._qmp_timer = qmp_timer 159beb6b57bSJohn Snow 160f9922937SPeter Delevoryas self._name = name or f"{id(self):x}" 161bd4c0ef4SMarc-André Lureau self._sock_pair: Optional[Tuple[socket.socket, socket.socket]] = None 16287bf1fe5SJohn Snow self._temp_dir: Optional[str] = None 163beb6b57bSJohn Snow self._base_temp_dir = base_temp_dir 16487bf1fe5SJohn Snow self._sock_dir = sock_dir 165b306e26cSCleber Rosa self._log_dir = log_dir 166beb6b57bSJohn Snow 167beb6b57bSJohn Snow self._monitor_address = monitor_address 168beb6b57bSJohn Snow 169beb6b57bSJohn Snow self._console_log_path = console_log 170beb6b57bSJohn Snow if self._console_log_path: 171beb6b57bSJohn Snow # In order to log the console, buffering needs to be enabled. 172beb6b57bSJohn Snow self._drain_console = True 173beb6b57bSJohn Snow else: 174beb6b57bSJohn Snow self._drain_console = drain_console 175beb6b57bSJohn Snow 176beb6b57bSJohn Snow # Runstate 177beb6b57bSJohn Snow self._qemu_log_path: Optional[str] = None 178beb6b57bSJohn Snow self._qemu_log_file: Optional[BinaryIO] = None 179beb6b57bSJohn Snow self._popen: Optional['subprocess.Popen[bytes]'] = None 180beb6b57bSJohn Snow self._events: List[QMPMessage] = [] 181beb6b57bSJohn Snow self._iolog: Optional[str] = None 182beb6b57bSJohn Snow self._qmp_set = True # Enable QMP monitor by default. 183beb6b57bSJohn Snow self._qmp_connection: Optional[QEMUMonitorProtocol] = None 184beb6b57bSJohn Snow self._qemu_full_args: Tuple[str, ...] = () 185beb6b57bSJohn Snow self._launched = False 186beb6b57bSJohn Snow self._machine: Optional[str] = None 187beb6b57bSJohn Snow self._console_index = 0 188beb6b57bSJohn Snow self._console_set = False 189beb6b57bSJohn Snow self._console_device_type: Optional[str] = None 190beb6b57bSJohn Snow self._console_address = os.path.join( 191f9922937SPeter Delevoryas self.sock_dir, f"{self._name}.con" 192beb6b57bSJohn Snow ) 193beb6b57bSJohn Snow self._console_socket: Optional[socket.socket] = None 194beb6b57bSJohn Snow self._remove_files: List[str] = [] 195beb6b57bSJohn Snow self._user_killed = False 196b9420e4fSJohn Snow self._quit_issued = False 197beb6b57bSJohn Snow 19815c3b863SVladimir Sementsov-Ogievskiy def __enter__(self: _T) -> _T: 199beb6b57bSJohn Snow return self 200beb6b57bSJohn Snow 201beb6b57bSJohn Snow def __exit__(self, 202beb6b57bSJohn Snow exc_type: Optional[Type[BaseException]], 203beb6b57bSJohn Snow exc_val: Optional[BaseException], 204beb6b57bSJohn Snow exc_tb: Optional[TracebackType]) -> None: 205beb6b57bSJohn Snow self.shutdown() 206beb6b57bSJohn Snow 207beb6b57bSJohn Snow def add_monitor_null(self) -> None: 208beb6b57bSJohn Snow """ 209beb6b57bSJohn Snow This can be used to add an unused monitor instance. 210beb6b57bSJohn Snow """ 211beb6b57bSJohn Snow self._args.append('-monitor') 212beb6b57bSJohn Snow self._args.append('null') 213beb6b57bSJohn Snow 21415c3b863SVladimir Sementsov-Ogievskiy def add_fd(self: _T, fd: int, fdset: int, 21515c3b863SVladimir Sementsov-Ogievskiy opaque: str, opts: str = '') -> _T: 216beb6b57bSJohn Snow """ 217beb6b57bSJohn Snow Pass a file descriptor to the VM 218beb6b57bSJohn Snow """ 219beb6b57bSJohn Snow options = ['fd=%d' % fd, 220beb6b57bSJohn Snow 'set=%d' % fdset, 221beb6b57bSJohn Snow 'opaque=%s' % opaque] 222beb6b57bSJohn Snow if opts: 223beb6b57bSJohn Snow options.append(opts) 224beb6b57bSJohn Snow 225beb6b57bSJohn Snow # This did not exist before 3.4, but since then it is 226beb6b57bSJohn Snow # mandatory for our purpose 227beb6b57bSJohn Snow if hasattr(os, 'set_inheritable'): 228beb6b57bSJohn Snow os.set_inheritable(fd, True) 229beb6b57bSJohn Snow 230beb6b57bSJohn Snow self._args.append('-add-fd') 231beb6b57bSJohn Snow self._args.append(','.join(options)) 232beb6b57bSJohn Snow return self 233beb6b57bSJohn Snow 234beb6b57bSJohn Snow def send_fd_scm(self, fd: Optional[int] = None, 235beb6b57bSJohn Snow file_path: Optional[str] = None) -> int: 236beb6b57bSJohn Snow """ 237514d00dfSJohn Snow Send an fd or file_path to the remote via SCM_RIGHTS. 238beb6b57bSJohn Snow 239514d00dfSJohn Snow Exactly one of fd and file_path must be given. If it is 240514d00dfSJohn Snow file_path, the file will be opened read-only and the new file 241514d00dfSJohn Snow descriptor will be sent to the remote. 242beb6b57bSJohn Snow """ 243beb6b57bSJohn Snow if file_path is not None: 244beb6b57bSJohn Snow assert fd is None 245514d00dfSJohn Snow with open(file_path, "rb") as passfile: 246514d00dfSJohn Snow fd = passfile.fileno() 247514d00dfSJohn Snow self._qmp.send_fd_scm(fd) 248beb6b57bSJohn Snow else: 249beb6b57bSJohn Snow assert fd is not None 250514d00dfSJohn Snow self._qmp.send_fd_scm(fd) 251beb6b57bSJohn Snow 252514d00dfSJohn Snow return 0 253beb6b57bSJohn Snow 254beb6b57bSJohn Snow @staticmethod 255beb6b57bSJohn Snow def _remove_if_exists(path: str) -> None: 256beb6b57bSJohn Snow """ 257beb6b57bSJohn Snow Remove file object at path if it exists 258beb6b57bSJohn Snow """ 259beb6b57bSJohn Snow try: 260beb6b57bSJohn Snow os.remove(path) 261beb6b57bSJohn Snow except OSError as exception: 262beb6b57bSJohn Snow if exception.errno == errno.ENOENT: 263beb6b57bSJohn Snow return 264beb6b57bSJohn Snow raise 265beb6b57bSJohn Snow 266beb6b57bSJohn Snow def is_running(self) -> bool: 267beb6b57bSJohn Snow """Returns true if the VM is running.""" 268beb6b57bSJohn Snow return self._popen is not None and self._popen.poll() is None 269beb6b57bSJohn Snow 270beb6b57bSJohn Snow @property 271beb6b57bSJohn Snow def _subp(self) -> 'subprocess.Popen[bytes]': 272beb6b57bSJohn Snow if self._popen is None: 273beb6b57bSJohn Snow raise QEMUMachineError('Subprocess pipe not present') 274beb6b57bSJohn Snow return self._popen 275beb6b57bSJohn Snow 276beb6b57bSJohn Snow def exitcode(self) -> Optional[int]: 277beb6b57bSJohn Snow """Returns the exit code if possible, or None.""" 278beb6b57bSJohn Snow if self._popen is None: 279beb6b57bSJohn Snow return None 280beb6b57bSJohn Snow return self._popen.poll() 281beb6b57bSJohn Snow 282beb6b57bSJohn Snow def get_pid(self) -> Optional[int]: 283beb6b57bSJohn Snow """Returns the PID of the running process, or None.""" 284beb6b57bSJohn Snow if not self.is_running(): 285beb6b57bSJohn Snow return None 286beb6b57bSJohn Snow return self._subp.pid 287beb6b57bSJohn Snow 288beb6b57bSJohn Snow def _load_io_log(self) -> None: 2895690b437SJohn Snow # Assume that the output encoding of QEMU's terminal output is 2905690b437SJohn Snow # defined by our locale. If indeterminate, allow open() to fall 2915690b437SJohn Snow # back to the platform default. 2925690b437SJohn Snow _, encoding = locale.getlocale() 293beb6b57bSJohn Snow if self._qemu_log_path is not None: 2945690b437SJohn Snow with open(self._qemu_log_path, "r", encoding=encoding) as iolog: 295beb6b57bSJohn Snow self._iolog = iolog.read() 296beb6b57bSJohn Snow 297beb6b57bSJohn Snow @property 298beb6b57bSJohn Snow def _base_args(self) -> List[str]: 299beb6b57bSJohn Snow args = ['-display', 'none', '-vga', 'none'] 300beb6b57bSJohn Snow 301beb6b57bSJohn Snow if self._qmp_set: 302bd4c0ef4SMarc-André Lureau if self._sock_pair: 303bd4c0ef4SMarc-André Lureau fd = self._sock_pair[0].fileno() 304bd4c0ef4SMarc-André Lureau os.set_inheritable(fd, True) 305bd4c0ef4SMarc-André Lureau moncdev = f"socket,id=mon,fd={fd}" 306bd4c0ef4SMarc-André Lureau elif isinstance(self._monitor_address, tuple): 307beb6b57bSJohn Snow moncdev = "socket,id=mon,host={},port={}".format( 308beb6b57bSJohn Snow *self._monitor_address 309beb6b57bSJohn Snow ) 310beb6b57bSJohn Snow else: 311beb6b57bSJohn Snow moncdev = f"socket,id=mon,path={self._monitor_address}" 312beb6b57bSJohn Snow args.extend(['-chardev', moncdev, '-mon', 313beb6b57bSJohn Snow 'chardev=mon,mode=control']) 314beb6b57bSJohn Snow 315beb6b57bSJohn Snow if self._machine is not None: 316beb6b57bSJohn Snow args.extend(['-machine', self._machine]) 317beb6b57bSJohn Snow for _ in range(self._console_index): 318beb6b57bSJohn Snow args.extend(['-serial', 'null']) 319beb6b57bSJohn Snow if self._console_set: 320beb6b57bSJohn Snow chardev = ('socket,id=console,path=%s,server=on,wait=off' % 321beb6b57bSJohn Snow self._console_address) 322beb6b57bSJohn Snow args.extend(['-chardev', chardev]) 323beb6b57bSJohn Snow if self._console_device_type is None: 324beb6b57bSJohn Snow args.extend(['-serial', 'chardev:console']) 325beb6b57bSJohn Snow else: 326beb6b57bSJohn Snow device = '%s,chardev=console' % self._console_device_type 327beb6b57bSJohn Snow args.extend(['-device', device]) 328beb6b57bSJohn Snow return args 329beb6b57bSJohn Snow 330555fe0c2SWainer dos Santos Moschetta @property 331555fe0c2SWainer dos Santos Moschetta def args(self) -> List[str]: 332555fe0c2SWainer dos Santos Moschetta """Returns the list of arguments given to the QEMU binary.""" 333555fe0c2SWainer dos Santos Moschetta return self._args 334555fe0c2SWainer dos Santos Moschetta 335beb6b57bSJohn Snow def _pre_launch(self) -> None: 336beb6b57bSJohn Snow if self._console_set: 337beb6b57bSJohn Snow self._remove_files.append(self._console_address) 338beb6b57bSJohn Snow 339beb6b57bSJohn Snow if self._qmp_set: 340bd4c0ef4SMarc-André Lureau sock = None 341bd4c0ef4SMarc-André Lureau if self._monitor_address is None: 342bd4c0ef4SMarc-André Lureau self._sock_pair = socket.socketpair() 343bd4c0ef4SMarc-André Lureau sock = self._sock_pair[1] 3446eeb3de7SJohn Snow if isinstance(self._monitor_address, str): 345beb6b57bSJohn Snow self._remove_files.append(self._monitor_address) 346*7f5f3ae7SJohn Snow 347beb6b57bSJohn Snow self._qmp_connection = QEMUMonitorProtocol( 348*7f5f3ae7SJohn Snow address=self._monitor_address, 349bd4c0ef4SMarc-André Lureau sock=sock, 350*7f5f3ae7SJohn Snow server=bool(self._monitor_address), 351beb6b57bSJohn Snow nickname=self._name 352beb6b57bSJohn Snow ) 353beb6b57bSJohn Snow 354beb6b57bSJohn Snow # NOTE: Make sure any opened resources are *definitely* freed in 355beb6b57bSJohn Snow # _post_shutdown()! 356beb6b57bSJohn Snow # pylint: disable=consider-using-with 357b306e26cSCleber Rosa self._qemu_log_path = os.path.join(self.log_dir, self._name + ".log") 358beb6b57bSJohn Snow self._qemu_log_file = open(self._qemu_log_path, 'wb') 359beb6b57bSJohn Snow 360b1ca9919SJohn Snow self._iolog = None 361b1ca9919SJohn Snow self._qemu_full_args = tuple(chain( 362b1ca9919SJohn Snow self._wrapper, 363b1ca9919SJohn Snow [self._binary], 364b1ca9919SJohn Snow self._base_args, 365b1ca9919SJohn Snow self._args 366b1ca9919SJohn Snow )) 367b1ca9919SJohn Snow 368beb6b57bSJohn Snow def _post_launch(self) -> None: 369bd4c0ef4SMarc-André Lureau if self._sock_pair: 370bd4c0ef4SMarc-André Lureau self._sock_pair[0].close() 371beb6b57bSJohn Snow if self._qmp_connection: 372*7f5f3ae7SJohn Snow if self._sock_pair: 373*7f5f3ae7SJohn Snow self._qmp.connect() 374*7f5f3ae7SJohn Snow else: 375e2f948a8SEmanuele Giuseppe Esposito self._qmp.accept(self._qmp_timer) 376beb6b57bSJohn Snow 377eb7a91d0SEmanuele Giuseppe Esposito def _close_qemu_log_file(self) -> None: 378eb7a91d0SEmanuele Giuseppe Esposito if self._qemu_log_file is not None: 379eb7a91d0SEmanuele Giuseppe Esposito self._qemu_log_file.close() 380eb7a91d0SEmanuele Giuseppe Esposito self._qemu_log_file = None 381eb7a91d0SEmanuele Giuseppe Esposito 382beb6b57bSJohn Snow def _post_shutdown(self) -> None: 383beb6b57bSJohn Snow """ 384beb6b57bSJohn Snow Called to cleanup the VM instance after the process has exited. 385beb6b57bSJohn Snow May also be called after a failed launch. 386beb6b57bSJohn Snow """ 3879cccb330SJohn Snow LOG.debug("Cleaning up after VM process") 38849a608b8SJohn Snow try: 38949a608b8SJohn Snow self._close_qmp_connection() 39049a608b8SJohn Snow except Exception as err: # pylint: disable=broad-except 39149a608b8SJohn Snow LOG.warning( 39249a608b8SJohn Snow "Exception closing QMP connection: %s", 39349a608b8SJohn Snow str(err) if str(err) else type(err).__name__ 39449a608b8SJohn Snow ) 39549a608b8SJohn Snow finally: 39649a608b8SJohn Snow assert self._qmp_connection is None 397beb6b57bSJohn Snow 398eb7a91d0SEmanuele Giuseppe Esposito self._close_qemu_log_file() 399beb6b57bSJohn Snow 400beb6b57bSJohn Snow self._load_io_log() 401beb6b57bSJohn Snow 402beb6b57bSJohn Snow self._qemu_log_path = None 403beb6b57bSJohn Snow 404beb6b57bSJohn Snow if self._temp_dir is not None: 405beb6b57bSJohn Snow shutil.rmtree(self._temp_dir) 406beb6b57bSJohn Snow self._temp_dir = None 407beb6b57bSJohn Snow 408beb6b57bSJohn Snow while len(self._remove_files) > 0: 409beb6b57bSJohn Snow self._remove_if_exists(self._remove_files.pop()) 410beb6b57bSJohn Snow 411beb6b57bSJohn Snow exitcode = self.exitcode() 412beb6b57bSJohn Snow if (exitcode is not None and exitcode < 0 413beb6b57bSJohn Snow and not (self._user_killed and exitcode == -signal.SIGKILL)): 414beb6b57bSJohn Snow msg = 'qemu received signal %i; command: "%s"' 415beb6b57bSJohn Snow if self._qemu_full_args: 416beb6b57bSJohn Snow command = ' '.join(self._qemu_full_args) 417beb6b57bSJohn Snow else: 418beb6b57bSJohn Snow command = '' 419beb6b57bSJohn Snow LOG.warning(msg, -int(exitcode), command) 420beb6b57bSJohn Snow 421b9420e4fSJohn Snow self._quit_issued = False 422beb6b57bSJohn Snow self._user_killed = False 423beb6b57bSJohn Snow self._launched = False 424beb6b57bSJohn Snow 425beb6b57bSJohn Snow def launch(self) -> None: 426beb6b57bSJohn Snow """ 427beb6b57bSJohn Snow Launch the VM and make sure we cleanup and expose the 428beb6b57bSJohn Snow command line/output in case of exception 429beb6b57bSJohn Snow """ 430beb6b57bSJohn Snow 431beb6b57bSJohn Snow if self._launched: 432beb6b57bSJohn Snow raise QEMUMachineError('VM already launched') 433beb6b57bSJohn Snow 434beb6b57bSJohn Snow try: 435beb6b57bSJohn Snow self._launch() 43650465f94SJohn Snow except BaseException as exc: 4371611e6cfSJohn Snow # We may have launched the process but it may 4381611e6cfSJohn Snow # have exited before we could connect via QMP. 4391611e6cfSJohn Snow # Assume the VM didn't launch or is exiting. 4401611e6cfSJohn Snow # If we don't wait for the process, exitcode() may still be 4411611e6cfSJohn Snow # 'None' by the time control is ceded back to the caller. 4421611e6cfSJohn Snow if self._launched: 4431611e6cfSJohn Snow self.wait() 4441611e6cfSJohn Snow else: 445beb6b57bSJohn Snow self._post_shutdown() 446beb6b57bSJohn Snow 44750465f94SJohn Snow if isinstance(exc, Exception): 44850465f94SJohn Snow raise VMLaunchFailure( 44950465f94SJohn Snow exitcode=self.exitcode(), 45050465f94SJohn Snow command=' '.join(self._qemu_full_args), 45150465f94SJohn Snow output=self._iolog 45250465f94SJohn Snow ) from exc 45350465f94SJohn Snow 45450465f94SJohn Snow # Don't wrap 'BaseException'; doing so would downgrade 45550465f94SJohn Snow # that exception. However, we still want to clean up. 456beb6b57bSJohn Snow raise 457beb6b57bSJohn Snow 458beb6b57bSJohn Snow def _launch(self) -> None: 459beb6b57bSJohn Snow """ 460beb6b57bSJohn Snow Launch the VM and establish a QMP connection 461beb6b57bSJohn Snow """ 462beb6b57bSJohn Snow self._pre_launch() 463beb6b57bSJohn Snow LOG.debug('VM launch command: %r', ' '.join(self._qemu_full_args)) 464beb6b57bSJohn Snow 465beb6b57bSJohn Snow # Cleaning up of this subprocess is guaranteed by _do_shutdown. 466beb6b57bSJohn Snow # pylint: disable=consider-using-with 467beb6b57bSJohn Snow self._popen = subprocess.Popen(self._qemu_full_args, 468beb6b57bSJohn Snow stdin=subprocess.DEVNULL, 469beb6b57bSJohn Snow stdout=self._qemu_log_file, 470beb6b57bSJohn Snow stderr=subprocess.STDOUT, 471beb6b57bSJohn Snow shell=False, 472beb6b57bSJohn Snow close_fds=False) 4731611e6cfSJohn Snow self._launched = True 474beb6b57bSJohn Snow self._post_launch() 475beb6b57bSJohn Snow 47649a608b8SJohn Snow def _close_qmp_connection(self) -> None: 47749a608b8SJohn Snow """ 47849a608b8SJohn Snow Close the underlying QMP connection, if any. 47949a608b8SJohn Snow 48049a608b8SJohn Snow Dutifully report errors that occurred while closing, but assume 48149a608b8SJohn Snow that any error encountered indicates an abnormal termination 48249a608b8SJohn Snow process and not a failure to close. 48349a608b8SJohn Snow """ 48449a608b8SJohn Snow if self._qmp_connection is None: 48549a608b8SJohn Snow return 48649a608b8SJohn Snow 48749a608b8SJohn Snow try: 48849a608b8SJohn Snow self._qmp.close() 48949a608b8SJohn Snow except EOFError: 49049a608b8SJohn Snow # EOF can occur as an Exception here when using the Async 49149a608b8SJohn Snow # QMP backend. It indicates that the server closed the 49249a608b8SJohn Snow # stream. If we successfully issued 'quit' at any point, 49349a608b8SJohn Snow # then this was expected. If the remote went away without 49449a608b8SJohn Snow # our permission, it's worth reporting that as an abnormal 49549a608b8SJohn Snow # shutdown case. 49649a608b8SJohn Snow if not (self._user_killed or self._quit_issued): 49749a608b8SJohn Snow raise 49849a608b8SJohn Snow finally: 49949a608b8SJohn Snow self._qmp_connection = None 50049a608b8SJohn Snow 501beb6b57bSJohn Snow def _early_cleanup(self) -> None: 502beb6b57bSJohn Snow """ 503beb6b57bSJohn Snow Perform any cleanup that needs to happen before the VM exits. 504beb6b57bSJohn Snow 5051611e6cfSJohn Snow This method may be called twice upon shutdown, once each by soft 5061611e6cfSJohn Snow and hard shutdown in failover scenarios. 507beb6b57bSJohn Snow """ 508beb6b57bSJohn Snow # If we keep the console socket open, we may deadlock waiting 509beb6b57bSJohn Snow # for QEMU to exit, while QEMU is waiting for the socket to 5109323e79fSPeter Maydell # become writable. 511beb6b57bSJohn Snow if self._console_socket is not None: 5129cccb330SJohn Snow LOG.debug("Closing console socket") 513beb6b57bSJohn Snow self._console_socket.close() 514beb6b57bSJohn Snow self._console_socket = None 515beb6b57bSJohn Snow 516beb6b57bSJohn Snow def _hard_shutdown(self) -> None: 517beb6b57bSJohn Snow """ 518beb6b57bSJohn Snow Perform early cleanup, kill the VM, and wait for it to terminate. 519beb6b57bSJohn Snow 520beb6b57bSJohn Snow :raise subprocess.Timeout: When timeout is exceeds 60 seconds 521beb6b57bSJohn Snow waiting for the QEMU process to terminate. 522beb6b57bSJohn Snow """ 5239cccb330SJohn Snow LOG.debug("Performing hard shutdown") 524beb6b57bSJohn Snow self._early_cleanup() 525beb6b57bSJohn Snow self._subp.kill() 526beb6b57bSJohn Snow self._subp.wait(timeout=60) 527beb6b57bSJohn Snow 528b9420e4fSJohn Snow def _soft_shutdown(self, timeout: Optional[int]) -> None: 529beb6b57bSJohn Snow """ 530beb6b57bSJohn Snow Perform early cleanup, attempt to gracefully shut down the VM, and wait 531beb6b57bSJohn Snow for it to terminate. 532beb6b57bSJohn Snow 533beb6b57bSJohn Snow :param timeout: Timeout in seconds for graceful shutdown. 534beb6b57bSJohn Snow A value of None is an infinite wait. 535beb6b57bSJohn Snow 536beb6b57bSJohn Snow :raise ConnectionReset: On QMP communication errors 537beb6b57bSJohn Snow :raise subprocess.TimeoutExpired: When timeout is exceeded waiting for 538beb6b57bSJohn Snow the QEMU process to terminate. 539beb6b57bSJohn Snow """ 5409cccb330SJohn Snow LOG.debug("Attempting graceful termination") 5419cccb330SJohn Snow 542beb6b57bSJohn Snow self._early_cleanup() 543beb6b57bSJohn Snow 5449cccb330SJohn Snow if self._quit_issued: 5459cccb330SJohn Snow LOG.debug( 5469cccb330SJohn Snow "Anticipating QEMU termination due to prior 'quit' command, " 5479cccb330SJohn Snow "or explicit call to wait()" 5489cccb330SJohn Snow ) 5499cccb330SJohn Snow else: 5509cccb330SJohn Snow LOG.debug("Politely asking QEMU to terminate") 5519cccb330SJohn Snow 552beb6b57bSJohn Snow if self._qmp_connection: 55349a608b8SJohn Snow try: 554b9420e4fSJohn Snow if not self._quit_issued: 55549a608b8SJohn Snow # May raise ExecInterruptedError or StateError if the 55649a608b8SJohn Snow # connection dies or has *already* died. 557b9420e4fSJohn Snow self.qmp('quit') 55849a608b8SJohn Snow finally: 55949a608b8SJohn Snow # Regardless, we want to quiesce the connection. 56049a608b8SJohn Snow self._close_qmp_connection() 5613c6e5e8cSJohn Snow elif not self._quit_issued: 5623c6e5e8cSJohn Snow LOG.debug( 5633c6e5e8cSJohn Snow "Not anticipating QEMU quit and no QMP connection present, " 5643c6e5e8cSJohn Snow "issuing SIGTERM" 5653c6e5e8cSJohn Snow ) 5663c6e5e8cSJohn Snow self._subp.terminate() 567beb6b57bSJohn Snow 568beb6b57bSJohn Snow # May raise subprocess.TimeoutExpired 5699cccb330SJohn Snow LOG.debug( 5709cccb330SJohn Snow "Waiting (timeout=%s) for QEMU process (pid=%s) to terminate", 5719cccb330SJohn Snow timeout, self._subp.pid 5729cccb330SJohn Snow ) 573beb6b57bSJohn Snow self._subp.wait(timeout=timeout) 574beb6b57bSJohn Snow 575b9420e4fSJohn Snow def _do_shutdown(self, timeout: Optional[int]) -> None: 576beb6b57bSJohn Snow """ 577beb6b57bSJohn Snow Attempt to shutdown the VM gracefully; fallback to a hard shutdown. 578beb6b57bSJohn Snow 579beb6b57bSJohn Snow :param timeout: Timeout in seconds for graceful shutdown. 580beb6b57bSJohn Snow A value of None is an infinite wait. 581beb6b57bSJohn Snow 582beb6b57bSJohn Snow :raise AbnormalShutdown: When the VM could not be shut down gracefully. 583beb6b57bSJohn Snow The inner exception will likely be ConnectionReset or 584beb6b57bSJohn Snow subprocess.TimeoutExpired. In rare cases, non-graceful termination 585beb6b57bSJohn Snow may result in its own exceptions, likely subprocess.TimeoutExpired. 586beb6b57bSJohn Snow """ 587beb6b57bSJohn Snow try: 588b9420e4fSJohn Snow self._soft_shutdown(timeout) 589beb6b57bSJohn Snow except Exception as exc: 5909cccb330SJohn Snow if isinstance(exc, subprocess.TimeoutExpired): 5919cccb330SJohn Snow LOG.debug("Timed out waiting for QEMU process to exit") 5929cccb330SJohn Snow LOG.debug("Graceful shutdown failed", exc_info=True) 5939cccb330SJohn Snow LOG.debug("Falling back to hard shutdown") 594beb6b57bSJohn Snow self._hard_shutdown() 595beb6b57bSJohn Snow raise AbnormalShutdown("Could not perform graceful shutdown") \ 596beb6b57bSJohn Snow from exc 597beb6b57bSJohn Snow 598b9420e4fSJohn Snow def shutdown(self, 599beb6b57bSJohn Snow hard: bool = False, 600beb6b57bSJohn Snow timeout: Optional[int] = 30) -> None: 601beb6b57bSJohn Snow """ 602beb6b57bSJohn Snow Terminate the VM (gracefully if possible) and perform cleanup. 603beb6b57bSJohn Snow Cleanup will always be performed. 604beb6b57bSJohn Snow 605beb6b57bSJohn Snow If the VM has not yet been launched, or shutdown(), wait(), or kill() 606beb6b57bSJohn Snow have already been called, this method does nothing. 607beb6b57bSJohn Snow 608beb6b57bSJohn Snow :param hard: When true, do not attempt graceful shutdown, and 609beb6b57bSJohn Snow suppress the SIGKILL warning log message. 610beb6b57bSJohn Snow :param timeout: Optional timeout in seconds for graceful shutdown. 611beb6b57bSJohn Snow Default 30 seconds, A `None` value is an infinite wait. 612beb6b57bSJohn Snow """ 613beb6b57bSJohn Snow if not self._launched: 614beb6b57bSJohn Snow return 615beb6b57bSJohn Snow 6169cccb330SJohn Snow LOG.debug("Shutting down VM appliance; timeout=%s", timeout) 6179cccb330SJohn Snow if hard: 6189cccb330SJohn Snow LOG.debug("Caller requests immediate termination of QEMU process.") 6199cccb330SJohn Snow 620beb6b57bSJohn Snow try: 621beb6b57bSJohn Snow if hard: 622beb6b57bSJohn Snow self._user_killed = True 623beb6b57bSJohn Snow self._hard_shutdown() 624beb6b57bSJohn Snow else: 625b9420e4fSJohn Snow self._do_shutdown(timeout) 626beb6b57bSJohn Snow finally: 627beb6b57bSJohn Snow self._post_shutdown() 628beb6b57bSJohn Snow 629beb6b57bSJohn Snow def kill(self) -> None: 630beb6b57bSJohn Snow """ 631beb6b57bSJohn Snow Terminate the VM forcefully, wait for it to exit, and perform cleanup. 632beb6b57bSJohn Snow """ 633beb6b57bSJohn Snow self.shutdown(hard=True) 634beb6b57bSJohn Snow 635beb6b57bSJohn Snow def wait(self, timeout: Optional[int] = 30) -> None: 636beb6b57bSJohn Snow """ 637beb6b57bSJohn Snow Wait for the VM to power off and perform post-shutdown cleanup. 638beb6b57bSJohn Snow 639beb6b57bSJohn Snow :param timeout: Optional timeout in seconds. Default 30 seconds. 640beb6b57bSJohn Snow A value of `None` is an infinite wait. 641beb6b57bSJohn Snow """ 642b9420e4fSJohn Snow self._quit_issued = True 643b9420e4fSJohn Snow self.shutdown(timeout=timeout) 644beb6b57bSJohn Snow 645beb6b57bSJohn Snow def set_qmp_monitor(self, enabled: bool = True) -> None: 646beb6b57bSJohn Snow """ 647beb6b57bSJohn Snow Set the QMP monitor. 648beb6b57bSJohn Snow 649beb6b57bSJohn Snow @param enabled: if False, qmp monitor options will be removed from 650beb6b57bSJohn Snow the base arguments of the resulting QEMU command 651beb6b57bSJohn Snow line. Default is True. 6525c02c865SJohn Snow 6535c02c865SJohn Snow .. note:: Call this function before launch(). 654beb6b57bSJohn Snow """ 655beb6b57bSJohn Snow self._qmp_set = enabled 656beb6b57bSJohn Snow 657beb6b57bSJohn Snow @property 658beb6b57bSJohn Snow def _qmp(self) -> QEMUMonitorProtocol: 659beb6b57bSJohn Snow if self._qmp_connection is None: 660beb6b57bSJohn Snow raise QEMUMachineError("Attempt to access QMP with no connection") 661beb6b57bSJohn Snow return self._qmp_connection 662beb6b57bSJohn Snow 663beb6b57bSJohn Snow @classmethod 664c7daa57eSVladimir Sementsov-Ogievskiy def _qmp_args(cls, conv_keys: bool, 665c7daa57eSVladimir Sementsov-Ogievskiy args: Dict[str, Any]) -> Dict[str, object]: 666c7daa57eSVladimir Sementsov-Ogievskiy if conv_keys: 667c7daa57eSVladimir Sementsov-Ogievskiy return {k.replace('_', '-'): v for k, v in args.items()} 668c7daa57eSVladimir Sementsov-Ogievskiy 669c7daa57eSVladimir Sementsov-Ogievskiy return args 670beb6b57bSJohn Snow 671beb6b57bSJohn Snow def qmp(self, cmd: str, 6723f3c9b4cSVladimir Sementsov-Ogievskiy args_dict: Optional[Dict[str, object]] = None, 6733f3c9b4cSVladimir Sementsov-Ogievskiy conv_keys: Optional[bool] = None, 674beb6b57bSJohn Snow **args: Any) -> QMPMessage: 675beb6b57bSJohn Snow """ 676beb6b57bSJohn Snow Invoke a QMP command and return the response dict 677beb6b57bSJohn Snow """ 6783f3c9b4cSVladimir Sementsov-Ogievskiy if args_dict is not None: 6793f3c9b4cSVladimir Sementsov-Ogievskiy assert not args 6803f3c9b4cSVladimir Sementsov-Ogievskiy assert conv_keys is None 6813f3c9b4cSVladimir Sementsov-Ogievskiy args = args_dict 6823f3c9b4cSVladimir Sementsov-Ogievskiy conv_keys = False 6833f3c9b4cSVladimir Sementsov-Ogievskiy 6843f3c9b4cSVladimir Sementsov-Ogievskiy if conv_keys is None: 6853f3c9b4cSVladimir Sementsov-Ogievskiy conv_keys = True 6863f3c9b4cSVladimir Sementsov-Ogievskiy 687c7daa57eSVladimir Sementsov-Ogievskiy qmp_args = self._qmp_args(conv_keys, args) 688b9420e4fSJohn Snow ret = self._qmp.cmd(cmd, args=qmp_args) 689b9420e4fSJohn Snow if cmd == 'quit' and 'error' not in ret and 'return' in ret: 690b9420e4fSJohn Snow self._quit_issued = True 691b9420e4fSJohn Snow return ret 692beb6b57bSJohn Snow 693beb6b57bSJohn Snow def command(self, cmd: str, 694beb6b57bSJohn Snow conv_keys: bool = True, 695beb6b57bSJohn Snow **args: Any) -> QMPReturnValue: 696beb6b57bSJohn Snow """ 697beb6b57bSJohn Snow Invoke a QMP command. 698beb6b57bSJohn Snow On success return the response dict. 699beb6b57bSJohn Snow On failure raise an exception. 700beb6b57bSJohn Snow """ 701c7daa57eSVladimir Sementsov-Ogievskiy qmp_args = self._qmp_args(conv_keys, args) 702b9420e4fSJohn Snow ret = self._qmp.command(cmd, **qmp_args) 703b9420e4fSJohn Snow if cmd == 'quit': 704b9420e4fSJohn Snow self._quit_issued = True 705b9420e4fSJohn Snow return ret 706beb6b57bSJohn Snow 707beb6b57bSJohn Snow def get_qmp_event(self, wait: bool = False) -> Optional[QMPMessage]: 708beb6b57bSJohn Snow """ 709beb6b57bSJohn Snow Poll for one queued QMP events and return it 710beb6b57bSJohn Snow """ 711beb6b57bSJohn Snow if self._events: 712beb6b57bSJohn Snow return self._events.pop(0) 713beb6b57bSJohn Snow return self._qmp.pull_event(wait=wait) 714beb6b57bSJohn Snow 715beb6b57bSJohn Snow def get_qmp_events(self, wait: bool = False) -> List[QMPMessage]: 716beb6b57bSJohn Snow """ 717beb6b57bSJohn Snow Poll for queued QMP events and return a list of dicts 718beb6b57bSJohn Snow """ 719beb6b57bSJohn Snow events = self._qmp.get_events(wait=wait) 720beb6b57bSJohn Snow events.extend(self._events) 721beb6b57bSJohn Snow del self._events[:] 722beb6b57bSJohn Snow return events 723beb6b57bSJohn Snow 724beb6b57bSJohn Snow @staticmethod 725beb6b57bSJohn Snow def event_match(event: Any, match: Optional[Any]) -> bool: 726beb6b57bSJohn Snow """ 727beb6b57bSJohn Snow Check if an event matches optional match criteria. 728beb6b57bSJohn Snow 729beb6b57bSJohn Snow The match criteria takes the form of a matching subdict. The event is 730beb6b57bSJohn Snow checked to be a superset of the subdict, recursively, with matching 731beb6b57bSJohn Snow values whenever the subdict values are not None. 732beb6b57bSJohn Snow 733beb6b57bSJohn Snow This has a limitation that you cannot explicitly check for None values. 734beb6b57bSJohn Snow 735beb6b57bSJohn Snow Examples, with the subdict queries on the left: 736beb6b57bSJohn Snow - None matches any object. 737beb6b57bSJohn Snow - {"foo": None} matches {"foo": {"bar": 1}} 738beb6b57bSJohn Snow - {"foo": None} matches {"foo": 5} 739beb6b57bSJohn Snow - {"foo": {"abc": None}} does not match {"foo": {"bar": 1}} 740beb6b57bSJohn Snow - {"foo": {"rab": 2}} matches {"foo": {"bar": 1, "rab": 2}} 741beb6b57bSJohn Snow """ 742beb6b57bSJohn Snow if match is None: 743beb6b57bSJohn Snow return True 744beb6b57bSJohn Snow 745beb6b57bSJohn Snow try: 746beb6b57bSJohn Snow for key in match: 747beb6b57bSJohn Snow if key in event: 748beb6b57bSJohn Snow if not QEMUMachine.event_match(event[key], match[key]): 749beb6b57bSJohn Snow return False 750beb6b57bSJohn Snow else: 751beb6b57bSJohn Snow return False 752beb6b57bSJohn Snow return True 753beb6b57bSJohn Snow except TypeError: 754beb6b57bSJohn Snow # either match or event wasn't iterable (not a dict) 755beb6b57bSJohn Snow return bool(match == event) 756beb6b57bSJohn Snow 757beb6b57bSJohn Snow def event_wait(self, name: str, 758beb6b57bSJohn Snow timeout: float = 60.0, 759beb6b57bSJohn Snow match: Optional[QMPMessage] = None) -> Optional[QMPMessage]: 760beb6b57bSJohn Snow """ 761beb6b57bSJohn Snow event_wait waits for and returns a named event from QMP with a timeout. 762beb6b57bSJohn Snow 763beb6b57bSJohn Snow name: The event to wait for. 764beb6b57bSJohn Snow timeout: QEMUMonitorProtocol.pull_event timeout parameter. 765beb6b57bSJohn Snow match: Optional match criteria. See event_match for details. 766beb6b57bSJohn Snow """ 767beb6b57bSJohn Snow return self.events_wait([(name, match)], timeout) 768beb6b57bSJohn Snow 769beb6b57bSJohn Snow def events_wait(self, 770beb6b57bSJohn Snow events: Sequence[Tuple[str, Any]], 771beb6b57bSJohn Snow timeout: float = 60.0) -> Optional[QMPMessage]: 772beb6b57bSJohn Snow """ 773beb6b57bSJohn Snow events_wait waits for and returns a single named event from QMP. 774beb6b57bSJohn Snow In the case of multiple qualifying events, this function returns the 775beb6b57bSJohn Snow first one. 776beb6b57bSJohn Snow 777beb6b57bSJohn Snow :param events: A sequence of (name, match_criteria) tuples. 778beb6b57bSJohn Snow The match criteria are optional and may be None. 779beb6b57bSJohn Snow See event_match for details. 780beb6b57bSJohn Snow :param timeout: Optional timeout, in seconds. 781beb6b57bSJohn Snow See QEMUMonitorProtocol.pull_event. 782beb6b57bSJohn Snow 783a4225303SJohn Snow :raise asyncio.TimeoutError: 784a4225303SJohn Snow If timeout was non-zero and no matching events were found. 785a4225303SJohn Snow 786beb6b57bSJohn Snow :return: A QMP event matching the filter criteria. 787beb6b57bSJohn Snow If timeout was 0 and no event matched, None. 788beb6b57bSJohn Snow """ 789beb6b57bSJohn Snow def _match(event: QMPMessage) -> bool: 790beb6b57bSJohn Snow for name, match in events: 791beb6b57bSJohn Snow if event['event'] == name and self.event_match(event, match): 792beb6b57bSJohn Snow return True 793beb6b57bSJohn Snow return False 794beb6b57bSJohn Snow 795beb6b57bSJohn Snow event: Optional[QMPMessage] 796beb6b57bSJohn Snow 797beb6b57bSJohn Snow # Search cached events 798beb6b57bSJohn Snow for event in self._events: 799beb6b57bSJohn Snow if _match(event): 800beb6b57bSJohn Snow self._events.remove(event) 801beb6b57bSJohn Snow return event 802beb6b57bSJohn Snow 803beb6b57bSJohn Snow # Poll for new events 804beb6b57bSJohn Snow while True: 805beb6b57bSJohn Snow event = self._qmp.pull_event(wait=timeout) 806beb6b57bSJohn Snow if event is None: 807beb6b57bSJohn Snow # NB: None is only returned when timeout is false-ish. 808a4225303SJohn Snow # Timeouts raise asyncio.TimeoutError instead! 809beb6b57bSJohn Snow break 810beb6b57bSJohn Snow if _match(event): 811beb6b57bSJohn Snow return event 812beb6b57bSJohn Snow self._events.append(event) 813beb6b57bSJohn Snow 814beb6b57bSJohn Snow return None 815beb6b57bSJohn Snow 816beb6b57bSJohn Snow def get_log(self) -> Optional[str]: 817beb6b57bSJohn Snow """ 818beb6b57bSJohn Snow After self.shutdown or failed qemu execution, this returns the output 819beb6b57bSJohn Snow of the qemu process. 820beb6b57bSJohn Snow """ 821beb6b57bSJohn Snow return self._iolog 822beb6b57bSJohn Snow 823beb6b57bSJohn Snow def add_args(self, *args: str) -> None: 824beb6b57bSJohn Snow """ 825beb6b57bSJohn Snow Adds to the list of extra arguments to be given to the QEMU binary 826beb6b57bSJohn Snow """ 827beb6b57bSJohn Snow self._args.extend(args) 828beb6b57bSJohn Snow 829beb6b57bSJohn Snow def set_machine(self, machine_type: str) -> None: 830beb6b57bSJohn Snow """ 831beb6b57bSJohn Snow Sets the machine type 832beb6b57bSJohn Snow 833beb6b57bSJohn Snow If set, the machine type will be added to the base arguments 834beb6b57bSJohn Snow of the resulting QEMU command line. 835beb6b57bSJohn Snow """ 836beb6b57bSJohn Snow self._machine = machine_type 837beb6b57bSJohn Snow 838beb6b57bSJohn Snow def set_console(self, 839beb6b57bSJohn Snow device_type: Optional[str] = None, 840beb6b57bSJohn Snow console_index: int = 0) -> None: 841beb6b57bSJohn Snow """ 842beb6b57bSJohn Snow Sets the device type for a console device 843beb6b57bSJohn Snow 844beb6b57bSJohn Snow If set, the console device and a backing character device will 845beb6b57bSJohn Snow be added to the base arguments of the resulting QEMU command 846beb6b57bSJohn Snow line. 847beb6b57bSJohn Snow 848beb6b57bSJohn Snow This is a convenience method that will either use the provided 849beb6b57bSJohn Snow device type, or default to a "-serial chardev:console" command 850beb6b57bSJohn Snow line argument. 851beb6b57bSJohn Snow 852beb6b57bSJohn Snow The actual setting of command line arguments will be be done at 853beb6b57bSJohn Snow machine launch time, as it depends on the temporary directory 854beb6b57bSJohn Snow to be created. 855beb6b57bSJohn Snow 856beb6b57bSJohn Snow @param device_type: the device type, such as "isa-serial". If 857beb6b57bSJohn Snow None is given (the default value) a "-serial 858beb6b57bSJohn Snow chardev:console" command line argument will 859beb6b57bSJohn Snow be used instead, resorting to the machine's 860beb6b57bSJohn Snow default device type. 861beb6b57bSJohn Snow @param console_index: the index of the console device to use. 862beb6b57bSJohn Snow If not zero, the command line will create 863beb6b57bSJohn Snow 'index - 1' consoles and connect them to 864beb6b57bSJohn Snow the 'null' backing character device. 865beb6b57bSJohn Snow """ 866beb6b57bSJohn Snow self._console_set = True 867beb6b57bSJohn Snow self._console_device_type = device_type 868beb6b57bSJohn Snow self._console_index = console_index 869beb6b57bSJohn Snow 870beb6b57bSJohn Snow @property 871beb6b57bSJohn Snow def console_socket(self) -> socket.socket: 872beb6b57bSJohn Snow """ 873beb6b57bSJohn Snow Returns a socket connected to the console 874beb6b57bSJohn Snow """ 875beb6b57bSJohn Snow if self._console_socket is None: 876beb6b57bSJohn Snow self._console_socket = console_socket.ConsoleSocket( 877beb6b57bSJohn Snow self._console_address, 878beb6b57bSJohn Snow file=self._console_log_path, 879beb6b57bSJohn Snow drain=self._drain_console) 880beb6b57bSJohn Snow return self._console_socket 881beb6b57bSJohn Snow 882beb6b57bSJohn Snow @property 883beb6b57bSJohn Snow def temp_dir(self) -> str: 884beb6b57bSJohn Snow """ 885beb6b57bSJohn Snow Returns a temporary directory to be used for this machine 886beb6b57bSJohn Snow """ 887beb6b57bSJohn Snow if self._temp_dir is None: 888beb6b57bSJohn Snow self._temp_dir = tempfile.mkdtemp(prefix="qemu-machine-", 889beb6b57bSJohn Snow dir=self._base_temp_dir) 890beb6b57bSJohn Snow return self._temp_dir 891b306e26cSCleber Rosa 892b306e26cSCleber Rosa @property 89387bf1fe5SJohn Snow def sock_dir(self) -> str: 89487bf1fe5SJohn Snow """ 89587bf1fe5SJohn Snow Returns the directory used for sockfiles by this machine. 89687bf1fe5SJohn Snow """ 89787bf1fe5SJohn Snow if self._sock_dir: 89887bf1fe5SJohn Snow return self._sock_dir 89987bf1fe5SJohn Snow return self.temp_dir 90087bf1fe5SJohn Snow 90187bf1fe5SJohn Snow @property 902b306e26cSCleber Rosa def log_dir(self) -> str: 903b306e26cSCleber Rosa """ 904b306e26cSCleber Rosa Returns a directory to be used for writing logs 905b306e26cSCleber Rosa """ 906b306e26cSCleber Rosa if self._log_dir is None: 907b306e26cSCleber Rosa return self.temp_dir 908b306e26cSCleber Rosa return self._log_dir 909