forked from brl/citadel
231 lines
8.8 KiB
Python
231 lines
8.8 KiB
Python
|
# Copyright (C) 2013 Intel Corporation
|
||
|
|
#
|
||
|
|
# Released under the MIT license (see COPYING.MIT)
|
||
|
|
|
||
|
|
# This module is used by testimage.bbclass for setting up and controlling a target machine.
|
||
|
|
|
||
|
|
import os
|
||
|
|
import shutil
|
||
|
|
import subprocess
|
||
|
|
import bb
|
||
|
|
import traceback
|
||
|
|
import sys
|
||
|
|
import logging
|
||
|
|
from oeqa.utils.sshcontrol import SSHControl
|
||
|
|
from oeqa.utils.qemurunner import QemuRunner
|
||
|
|
from oeqa.utils.qemutinyrunner import QemuTinyRunner
|
||
|
|
from oeqa.utils.dump import TargetDumper
|
||
|
|
from oeqa.controllers.testtargetloader import TestTargetLoader
|
||
|
|
from abc import ABCMeta, abstractmethod
|
||
|
|
|
||
|
|
class BaseTarget(object, metaclass=ABCMeta):
|
||
|
|
|
||
|
|
supported_image_fstypes = []
|
||
|
|
|
||
|
|
def __init__(self, d, logger):
|
||
|
|
self.connection = None
|
||
|
|
self.ip = None
|
||
|
|
self.server_ip = None
|
||
|
|
self.datetime = d.getVar('DATETIME')
|
||
|
|
self.testdir = d.getVar("TEST_LOG_DIR")
|
||
|
|
self.pn = d.getVar("PN")
|
||
|
|
self.logger = logger
|
||
|
|
|
||
|
|
@abstractmethod
|
||
|
|
def deploy(self):
|
||
|
|
|
||
|
|
self.sshlog = os.path.join(self.testdir, "ssh_target_log.%s" % self.datetime)
|
||
|
|
sshloglink = os.path.join(self.testdir, "ssh_target_log")
|
||
|
|
if os.path.islink(sshloglink):
|
||
|
|
os.unlink(sshloglink)
|
||
|
|
os.symlink(self.sshlog, sshloglink)
|
||
|
|
self.logger.info("SSH log file: %s" % self.sshlog)
|
||
|
|
|
||
|
|
@abstractmethod
|
||
|
|
def start(self, params=None, ssh=True, extra_bootparams=None):
|
||
|
|
pass
|
||
|
|
|
||
|
|
@abstractmethod
|
||
|
|
def stop(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def get_extra_files(self):
|
||
|
|
return None
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def match_image_fstype(self, d, image_fstypes=None):
|
||
|
|
if not image_fstypes:
|
||
|
|
image_fstypes = d.getVar('IMAGE_FSTYPES').split(' ')
|
||
|
|
possible_image_fstypes = [fstype for fstype in self.supported_image_fstypes if fstype in image_fstypes]
|
||
|
|
if possible_image_fstypes:
|
||
|
|
return possible_image_fstypes[0]
|
||
|
|
else:
|
||
|
|
return None
|
||
|
|
|
||
|
|
def get_image_fstype(self, d):
|
||
|
|
image_fstype = self.match_image_fstype(d)
|
||
|
|
if image_fstype:
|
||
|
|
return image_fstype
|
||
|
|
else:
|
||
|
|
bb.fatal("IMAGE_FSTYPES should contain a Target Controller supported image fstype: %s " % ', '.join(map(str, self.supported_image_fstypes)))
|
||
|
|
|
||
|
|
def restart(self, params=None):
|
||
|
|
self.stop()
|
||
|
|
self.start(params)
|
||
|
|
|
||
|
|
def run(self, cmd, timeout=None):
|
||
|
|
return self.connection.run(cmd, timeout)
|
||
|
|
|
||
|
|
def copy_to(self, localpath, remotepath):
|
||
|
|
return self.connection.copy_to(localpath, remotepath)
|
||
|
|
|
||
|
|
def copy_from(self, remotepath, localpath):
|
||
|
|
return self.connection.copy_from(remotepath, localpath)
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
class QemuTarget(BaseTarget):
|
||
|
|
|
||
|
|
supported_image_fstypes = ['ext3', 'ext4', 'cpio.gz', 'wic']
|
||
|
|
|
||
|
|
def __init__(self, d, logger, image_fstype=None):
|
||
|
|
|
||
|
|
super(QemuTarget, self).__init__(d, logger)
|
||
|
|
|
||
|
|
self.rootfs = ''
|
||
|
|
self.kernel = ''
|
||
|
|
self.image_fstype = ''
|
||
|
|
|
||
|
|
if d.getVar('FIND_ROOTFS') == '1':
|
||
|
|
self.image_fstype = image_fstype or self.get_image_fstype(d)
|
||
|
|
self.rootfs = os.path.join(d.getVar("DEPLOY_DIR_IMAGE"), d.getVar("IMAGE_LINK_NAME") + '.' + self.image_fstype)
|
||
|
|
self.kernel = os.path.join(d.getVar("DEPLOY_DIR_IMAGE"), d.getVar("KERNEL_IMAGETYPE", False) + '-' + d.getVar('MACHINE', False) + '.bin')
|
||
|
|
self.qemulog = os.path.join(self.testdir, "qemu_boot_log.%s" % self.datetime)
|
||
|
|
dump_target_cmds = d.getVar("testimage_dump_target")
|
||
|
|
dump_host_cmds = d.getVar("testimage_dump_host")
|
||
|
|
dump_dir = d.getVar("TESTIMAGE_DUMP_DIR")
|
||
|
|
qemu_use_kvm = d.getVar("QEMU_USE_KVM")
|
||
|
|
if qemu_use_kvm and \
|
||
|
|
(qemu_use_kvm == "True" and "x86" in d.getVar("MACHINE") or \
|
||
|
|
d.getVar("MACHINE") in qemu_use_kvm.split()):
|
||
|
|
use_kvm = True
|
||
|
|
else:
|
||
|
|
use_kvm = False
|
||
|
|
|
||
|
|
# Log QemuRunner log output to a file
|
||
|
|
import oe.path
|
||
|
|
bb.utils.mkdirhier(self.testdir)
|
||
|
|
self.qemurunnerlog = os.path.join(self.testdir, 'qemurunner_log.%s' % self.datetime)
|
||
|
|
loggerhandler = logging.FileHandler(self.qemurunnerlog)
|
||
|
|
loggerhandler.setFormatter(logging.Formatter("%(levelname)s: %(message)s"))
|
||
|
|
self.logger.addHandler(loggerhandler)
|
||
|
|
oe.path.symlink(os.path.basename(self.qemurunnerlog), os.path.join(self.testdir, 'qemurunner_log'), force=True)
|
||
|
|
|
||
|
|
if d.getVar("DISTRO") == "poky-tiny":
|
||
|
|
self.runner = QemuTinyRunner(machine=d.getVar("MACHINE"),
|
||
|
|
rootfs=self.rootfs,
|
||
|
|
tmpdir = d.getVar("TMPDIR"),
|
||
|
|
deploy_dir_image = d.getVar("DEPLOY_DIR_IMAGE"),
|
||
|
|
display = d.getVar("BB_ORIGENV", False).getVar("DISPLAY"),
|
||
|
|
logfile = self.qemulog,
|
||
|
|
kernel = self.kernel,
|
||
|
|
boottime = int(d.getVar("TEST_QEMUBOOT_TIMEOUT")),
|
||
|
|
logger = logger)
|
||
|
|
else:
|
||
|
|
self.runner = QemuRunner(machine=d.getVar("MACHINE"),
|
||
|
|
rootfs=self.rootfs,
|
||
|
|
tmpdir = d.getVar("TMPDIR"),
|
||
|
|
deploy_dir_image = d.getVar("DEPLOY_DIR_IMAGE"),
|
||
|
|
display = d.getVar("BB_ORIGENV", False).getVar("DISPLAY"),
|
||
|
|
logfile = self.qemulog,
|
||
|
|
boottime = int(d.getVar("TEST_QEMUBOOT_TIMEOUT")),
|
||
|
|
use_kvm = use_kvm,
|
||
|
|
dump_dir = dump_dir,
|
||
|
|
dump_host_cmds = d.getVar("testimage_dump_host"),
|
||
|
|
logger = logger)
|
||
|
|
|
||
|
|
self.target_dumper = TargetDumper(dump_target_cmds, dump_dir, self.runner)
|
||
|
|
|
||
|
|
def deploy(self):
|
||
|
|
bb.utils.mkdirhier(self.testdir)
|
||
|
|
|
||
|
|
qemuloglink = os.path.join(self.testdir, "qemu_boot_log")
|
||
|
|
if os.path.islink(qemuloglink):
|
||
|
|
os.unlink(qemuloglink)
|
||
|
|
os.symlink(self.qemulog, qemuloglink)
|
||
|
|
|
||
|
|
self.logger.info("rootfs file: %s" % self.rootfs)
|
||
|
|
self.logger.info("Qemu log file: %s" % self.qemulog)
|
||
|
|
super(QemuTarget, self).deploy()
|
||
|
|
|
||
|
|
def start(self, params=None, ssh=True, extra_bootparams='', runqemuparams='', launch_cmd='', discard_writes=True):
|
||
|
|
if launch_cmd:
|
||
|
|
start = self.runner.launch(get_ip=ssh, launch_cmd=launch_cmd)
|
||
|
|
else:
|
||
|
|
start = self.runner.start(params, get_ip=ssh, extra_bootparams=extra_bootparams, runqemuparams=runqemuparams, discard_writes=discard_writes)
|
||
|
|
|
||
|
|
if start:
|
||
|
|
if ssh:
|
||
|
|
self.ip = self.runner.ip
|
||
|
|
self.server_ip = self.runner.server_ip
|
||
|
|
self.connection = SSHControl(ip=self.ip, logfile=self.sshlog)
|
||
|
|
else:
|
||
|
|
self.stop()
|
||
|
|
if os.path.exists(self.qemulog):
|
||
|
|
with open(self.qemulog, 'r') as f:
|
||
|
|
bb.error("Qemu log output from %s:\n%s" % (self.qemulog, f.read()))
|
||
|
|
raise bb.build.FuncFailed("%s - FAILED to start qemu - check the task log and the boot log" % self.pn)
|
||
|
|
|
||
|
|
def check(self):
|
||
|
|
return self.runner.is_alive()
|
||
|
|
|
||
|
|
def stop(self):
|
||
|
|
self.runner.stop()
|
||
|
|
self.connection = None
|
||
|
|
self.ip = None
|
||
|
|
self.server_ip = None
|
||
|
|
|
||
|
|
def restart(self, params=None):
|
||
|
|
if self.runner.restart(params):
|
||
|
|
self.ip = self.runner.ip
|
||
|
|
self.server_ip = self.runner.server_ip
|
||
|
|
self.connection = SSHControl(ip=self.ip, logfile=self.sshlog)
|
||
|
|
else:
|
||
|
|
raise bb.build.FuncFailed("%s - FAILED to re-start qemu - check the task log and the boot log" % self.pn)
|
||
|
|
|
||
|
|
def run_serial(self, command, timeout=5):
|
||
|
|
return self.runner.run_serial(command, timeout=timeout)
|
||
|
|
|
||
|
|
|
||
|
|
class SimpleRemoteTarget(BaseTarget):
|
||
|
|
|
||
|
|
def __init__(self, d):
|
||
|
|
super(SimpleRemoteTarget, self).__init__(d)
|
||
|
|
addr = d.getVar("TEST_TARGET_IP") or bb.fatal('Please set TEST_TARGET_IP with the IP address of the machine you want to run the tests on.')
|
||
|
|
self.ip = addr.split(":")[0]
|
||
|
|
try:
|
||
|
|
self.port = addr.split(":")[1]
|
||
|
|
except IndexError:
|
||
|
|
self.port = None
|
||
|
|
self.logger.info("Target IP: %s" % self.ip)
|
||
|
|
self.server_ip = d.getVar("TEST_SERVER_IP")
|
||
|
|
if not self.server_ip:
|
||
|
|
try:
|
||
|
|
self.server_ip = subprocess.check_output(['ip', 'route', 'get', self.ip ]).split("\n")[0].split()[-1]
|
||
|
|
except Exception as e:
|
||
|
|
bb.fatal("Failed to determine the host IP address (alternatively you can set TEST_SERVER_IP with the IP address of this machine): %s" % e)
|
||
|
|
self.logger.info("Server IP: %s" % self.server_ip)
|
||
|
|
|
||
|
|
def deploy(self):
|
||
|
|
super(SimpleRemoteTarget, self).deploy()
|
||
|
|
|
||
|
|
def start(self, params=None, ssh=True, extra_bootparams=None):
|
||
|
|
if ssh:
|
||
|
|
self.connection = SSHControl(self.ip, logfile=self.sshlog, port=self.port)
|
||
|
|
|
||
|
|
def stop(self):
|
||
|
|
self.connection = None
|
||
|
|
self.ip = None
|
||
|
|
self.server_ip = None
|