Brad Bishop | c342db3 | 2019-05-15 21:57:59 -0400 | [diff] [blame] | 1 | # |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame] | 2 | # Copyright (C) 2016 Intel Corporation |
Brad Bishop | c342db3 | 2019-05-15 21:57:59 -0400 | [diff] [blame] | 3 | # |
| 4 | # SPDX-License-Identifier: MIT |
| 5 | # |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame] | 6 | |
| 7 | import os |
| 8 | import sys |
| 9 | import signal |
| 10 | import time |
Andrew Geissler | 82c905d | 2020-04-13 13:39:40 -0500 | [diff] [blame^] | 11 | from collections import defaultdict |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame] | 12 | |
| 13 | from .ssh import OESSHTarget |
| 14 | from oeqa.utils.qemurunner import QemuRunner |
| 15 | |
Brad Bishop | 316dfdd | 2018-06-25 12:45:53 -0400 | [diff] [blame] | 16 | supported_fstypes = ['ext3', 'ext4', 'cpio.gz', 'wic'] |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame] | 17 | |
| 18 | class OEQemuTarget(OESSHTarget): |
Brad Bishop | 977dc1a | 2019-02-06 16:01:43 -0500 | [diff] [blame] | 19 | def __init__(self, logger, server_ip, timeout=300, user='root', |
Brad Bishop | 1932369 | 2019-04-05 15:28:33 -0400 | [diff] [blame] | 20 | port=None, machine='', rootfs='', kernel='', kvm=False, slirp=False, |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame] | 21 | dump_dir='', dump_host_cmds='', display='', bootlog='', |
Andrew Geissler | 82c905d | 2020-04-13 13:39:40 -0500 | [diff] [blame^] | 22 | tmpdir='', dir_image='', boottime=60, serial_ports=2, |
| 23 | boot_patterns = defaultdict(str), ovmf=False, **kwargs): |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame] | 24 | |
Brad Bishop | 977dc1a | 2019-02-06 16:01:43 -0500 | [diff] [blame] | 25 | super(OEQemuTarget, self).__init__(logger, None, server_ip, timeout, |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame] | 26 | user, port) |
| 27 | |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame] | 28 | self.server_ip = server_ip |
Andrew Geissler | 82c905d | 2020-04-13 13:39:40 -0500 | [diff] [blame^] | 29 | self.server_port = 0 |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame] | 30 | self.machine = machine |
| 31 | self.rootfs = rootfs |
| 32 | self.kernel = kernel |
| 33 | self.kvm = kvm |
Andrew Geissler | 82c905d | 2020-04-13 13:39:40 -0500 | [diff] [blame^] | 34 | self.ovmf = ovmf |
Brad Bishop | 1932369 | 2019-04-05 15:28:33 -0400 | [diff] [blame] | 35 | self.use_slirp = slirp |
Andrew Geissler | 82c905d | 2020-04-13 13:39:40 -0500 | [diff] [blame^] | 36 | self.boot_patterns = boot_patterns |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame] | 37 | |
| 38 | self.runner = QemuRunner(machine=machine, rootfs=rootfs, tmpdir=tmpdir, |
| 39 | deploy_dir_image=dir_image, display=display, |
| 40 | logfile=bootlog, boottime=boottime, |
Brad Bishop | 1932369 | 2019-04-05 15:28:33 -0400 | [diff] [blame] | 41 | use_kvm=kvm, use_slirp=slirp, dump_dir=dump_dir, |
Andrew Geissler | 82c905d | 2020-04-13 13:39:40 -0500 | [diff] [blame^] | 42 | dump_host_cmds=dump_host_cmds, logger=logger, |
| 43 | serial_ports=serial_ports, boot_patterns = boot_patterns, |
| 44 | use_ovmf=ovmf) |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame] | 45 | |
Brad Bishop | 1932369 | 2019-04-05 15:28:33 -0400 | [diff] [blame] | 46 | def start(self, params=None, extra_bootparams=None, runqemuparams=''): |
| 47 | if self.use_slirp and not self.server_ip: |
| 48 | self.logger.error("Could not start qemu with slirp without server ip - provide 'TEST_SERVER_IP'") |
| 49 | raise RuntimeError("FAILED to start qemu - check the task log and the boot log") |
| 50 | if self.runner.start(params, extra_bootparams=extra_bootparams, runqemuparams=runqemuparams): |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame] | 51 | self.ip = self.runner.ip |
Brad Bishop | 1932369 | 2019-04-05 15:28:33 -0400 | [diff] [blame] | 52 | if self.use_slirp: |
| 53 | target_ip_port = self.runner.ip.split(':') |
| 54 | if len(target_ip_port) == 2: |
| 55 | target_ip = target_ip_port[0] |
| 56 | port = target_ip_port[1] |
| 57 | self.ip = target_ip |
| 58 | self.ssh = self.ssh + ['-p', port] |
| 59 | self.scp = self.scp + ['-P', port] |
| 60 | else: |
| 61 | self.logger.error("Could not get host machine port to connect qemu with slirp, ssh will not be " |
| 62 | "able to connect to qemu with slirp") |
| 63 | if self.runner.server_ip: |
| 64 | self.server_ip = self.runner.server_ip |
Brad Bishop | 6e60e8b | 2018-02-01 10:27:11 -0500 | [diff] [blame] | 65 | else: |
| 66 | self.stop() |
| 67 | raise RuntimeError("FAILED to start qemu - check the task log and the boot log") |
| 68 | |
| 69 | def stop(self): |
| 70 | self.runner.stop() |