[mirror_qemu.git] / tests / vm / basevm.py

#!/usr/bin/env python
#
# VM testing base class
#
# Copyright 2017 Red Hat Inc.
#
# Authors:
#  Fam Zheng <famz@redhat.com>
#
# This code is licensed under the GPL version 2 or later.  See
# the COPYING file in the top-level directory.
#

from __future__ import print_function
import os
import sys
import logging
import time
import datetime
sys.path.append(os.path.join(os.path.dirname(__file__), "..", "..", "scripts"))
from qemu import QEMUMachine, kvm_available
import subprocess
import hashlib
import optparse
import atexit
import tempfile
import shutil
import multiprocessing
import traceback

SSH_KEY = open(os.path.join(os.path.dirname(__file__),
               "..", "keys", "id_rsa")).read()
SSH_PUB_KEY = open(os.path.join(os.path.dirname(__file__),
                   "..", "keys", "id_rsa.pub")).read()

class BaseVM(object):
    GUEST_USER = "qemu"
    GUEST_PASS = "qemupass"
    ROOT_PASS = "qemupass"

    # The script to run in the guest that builds QEMU
    BUILD_SCRIPT = ""
    # The guest name, to be overridden by subclasses
    name = "#base"
    def __init__(self, debug=False, vcpus=None):
        self._guest = None
        self._tmpdir = os.path.realpath(tempfile.mkdtemp(prefix="vm-test-",
                                                         suffix=".tmp",
                                                         dir="."))
        atexit.register(shutil.rmtree, self._tmpdir)

        self._ssh_key_file = os.path.join(self._tmpdir, "id_rsa")
        open(self._ssh_key_file, "w").write(SSH_KEY)
        subprocess.check_call(["chmod", "600", self._ssh_key_file])

        self._ssh_pub_key_file = os.path.join(self._tmpdir, "id_rsa.pub")
        open(self._ssh_pub_key_file, "w").write(SSH_PUB_KEY)

        self.debug = debug
        self._stderr = sys.stderr
        self._devnull = open(os.devnull, "w")
        if self.debug:
            self._stdout = sys.stdout
        else:
            self._stdout = self._devnull
        self._args = [ \
            "-nodefaults", "-m", "4G",
            "-cpu", "max",
            "-netdev", "user,id=vnet,hostfwd=:127.0.0.1:0-:22",
            "-device", "virtio-net-pci,netdev=vnet",
            "-vnc", "127.0.0.1:0,to=20",
            "-serial", "file:%s" % os.path.join(self._tmpdir, "serial.out")]
        if vcpus:
            self._args += ["-smp", str(vcpus)]
        if kvm_available():
            self._args += ["-enable-kvm"]
        else:
            logging.info("KVM not available, not using -enable-kvm")
        self._data_args = []

    def _download_with_cache(self, url, sha256sum=None):
        def check_sha256sum(fname):
            if not sha256sum:
                return True
            checksum = subprocess.check_output(["sha256sum", fname]).split()[0]
            return sha256sum == checksum

        cache_dir = os.path.expanduser("~/.cache/qemu-vm/download")
        if not os.path.exists(cache_dir):
            os.makedirs(cache_dir)
        fname = os.path.join(cache_dir, hashlib.sha1(url).hexdigest())
        if os.path.exists(fname) and check_sha256sum(fname):
            return fname
        logging.debug("Downloading %s to %s...", url, fname)
        subprocess.check_call(["wget", "-c", url, "-O", fname + ".download"],
                              stdout=self._stdout, stderr=self._stderr)
        os.rename(fname + ".download", fname)
        return fname

    def _ssh_do(self, user, cmd, check, interactive=False):
        ssh_cmd = ["ssh", "-q",
                   "-o", "StrictHostKeyChecking=no",
                   "-o", "UserKnownHostsFile=" + os.devnull,
                   "-o", "ConnectTimeout=1",
                   "-p", self.ssh_port, "-i", self._ssh_key_file]
        if interactive:
            ssh_cmd += ['-t']
        assert not isinstance(cmd, str)
        ssh_cmd += ["%s@127.0.0.1" % user] + list(cmd)
        logging.debug("ssh_cmd: %s", " ".join(ssh_cmd))
        r = subprocess.call(ssh_cmd)
        if check and r != 0:
            raise Exception("SSH command failed: %s" % cmd)
        return r

    def ssh(self, *cmd):
        return self._ssh_do(self.GUEST_USER, cmd, False)

    def ssh_interactive(self, *cmd):
        return self._ssh_do(self.GUEST_USER, cmd, False, True)

    def ssh_root(self, *cmd):
        return self._ssh_do("root", cmd, False)

    def ssh_check(self, *cmd):
        self._ssh_do(self.GUEST_USER, cmd, True)

    def ssh_root_check(self, *cmd):
        self._ssh_do("root", cmd, True)

    def build_image(self, img):
        raise NotImplementedError

    def add_source_dir(self, src_dir):
        name = "data-" + hashlib.sha1(src_dir).hexdigest()[:5]
        tarfile = os.path.join(self._tmpdir, name + ".tar")
        logging.debug("Creating archive %s for src_dir dir: %s", tarfile, src_dir)
        subprocess.check_call(["./scripts/archive-source.sh", tarfile],
                              cwd=src_dir, stdin=self._devnull,
                              stdout=self._stdout, stderr=self._stderr)
        self._data_args += ["-drive",
                            "file=%s,if=none,id=%s,cache=writeback,format=raw" % \
                                    (tarfile, name),
                            "-device",
                            "virtio-blk,drive=%s,serial=%s,bootindex=1" % (name, name)]

    def boot(self, img, extra_args=[]):
        args = self._args + [
            "-device", "VGA",
            "-drive", "file=%s,if=none,id=drive0,cache=writeback" % img,
            "-device", "virtio-blk,drive=drive0,bootindex=0"]
        args += self._data_args + extra_args
        logging.debug("QEMU args: %s", " ".join(args))
        qemu_bin = os.environ.get("QEMU", "qemu-system-x86_64")
        guest = QEMUMachine(binary=qemu_bin, args=args)
        try:
            guest.launch()
        except:
            logging.error("Failed to launch QEMU, command line:")
            logging.error(" ".join([qemu_bin] + args))
            logging.error("Log:")
            logging.error(guest.get_log())
            logging.error("QEMU version >= 2.10 is required")
            raise
        atexit.register(self.shutdown)
        self._guest = guest
        usernet_info = guest.qmp("human-monitor-command",
                                 command_line="info usernet")
        self.ssh_port = None
        for l in usernet_info["return"].splitlines():
            fields = l.split()
            if "TCP[HOST_FORWARD]" in fields and "22" in fields:
                self.ssh_port = l.split()[3]
        if not self.ssh_port:
            raise Exception("Cannot find ssh port from 'info usernet':\n%s" % \
                            usernet_info)

    def wait_ssh(self, seconds=300):
        starttime = datetime.datetime.now()
        guest_up = False
        while (datetime.datetime.now() - starttime).total_seconds() < seconds:
            if self.ssh("exit 0") == 0:
                guest_up = True
                break
            time.sleep(1)
        if not guest_up:
            raise Exception("Timeout while waiting for guest ssh")

    def shutdown(self):
        self._guest.shutdown()

    def wait(self):
        self._guest.wait()

    def qmp(self, *args, **kwargs):
        return self._guest.qmp(*args, **kwargs)

def parse_args(vm_name):
    parser = optparse.OptionParser(
        description="VM test utility.  Exit codes: "
                    "0 = success, "
                    "1 = command line error, "
                    "2 = environment initialization failed, "
                    "3 = test command failed")
    parser.add_option("--debug", "-D", action="store_true",
                      help="enable debug output")
    parser.add_option("--image", "-i", default="%s.img" % vm_name,
                      help="image file name")
    parser.add_option("--force", "-f", action="store_true",
                      help="force build image even if image exists")
    parser.add_option("--jobs", type=int, default=multiprocessing.cpu_count() / 2,
                      help="number of virtual CPUs")
    parser.add_option("--verbose", "-V", action="store_true",
                      help="Pass V=1 to builds within the guest")
    parser.add_option("--build-image", "-b", action="store_true",
                      help="build image")
    parser.add_option("--build-qemu",
                      help="build QEMU from source in guest")
    parser.add_option("--interactive", "-I", action="store_true",
                      help="Interactively run command")
    parser.add_option("--snapshot", "-s", action="store_true",
                      help="run tests with a snapshot")
    parser.disable_interspersed_args()
    return parser.parse_args()

def main(vmcls):
    try:
        args, argv = parse_args(vmcls.name)
        if not argv and not args.build_qemu and not args.build_image:
            print("Nothing to do?")
            return 1
        logging.basicConfig(level=(logging.DEBUG if args.debug
                                   else logging.WARN))
        vm = vmcls(debug=args.debug, vcpus=args.jobs)
        if args.build_image:
            if os.path.exists(args.image) and not args.force:
                sys.stderr.writelines(["Image file exists: %s\n" % args.image,
                                      "Use --force option to overwrite\n"])
                return 1
            return vm.build_image(args.image)
        if args.build_qemu:
            vm.add_source_dir(args.build_qemu)
            cmd = [vm.BUILD_SCRIPT.format(
                   configure_opts = " ".join(argv),
                   jobs=args.jobs,
                   verbose = "V=1" if args.verbose else "")]
        else:
            cmd = argv
        img = args.image
        if args.snapshot:
            img += ",snapshot=on"
        vm.boot(img)
        vm.wait_ssh()
    except Exception as e:
        if isinstance(e, SystemExit) and e.code == 0:
            return 0
        sys.stderr.write("Failed to prepare guest environment\n")
        traceback.print_exc()
        return 2

    if args.interactive:
        if vm.ssh_interactive(*cmd) == 0:
            return 0
        vm.ssh_interactive()
        return 3
    else:
        if vm.ssh(*cmd) != 0:
            return 3
Commit	Line	Data
ff2ebff0 FZ	1	#!/usr/bin/env python
	2	#
	3	# VM testing base class
	4	#
	5	# Copyright 2017 Red Hat Inc.
	6	#
	7	# Authors:
	8	# Fam Zheng <famz@redhat.com>
	9	#
	10	# This code is licensed under the GPL version 2 or later. See
	11	# the COPYING file in the top-level directory.
	12	#
	13
f03868bd	14	from __future__ import print_function
ff2ebff0 FZ	15	import os
	16	import sys
	17	import logging
	18	import time
	19	import datetime
	20	sys.path.append(os.path.join(os.path.dirname(__file__), "..", "..", "scripts"))
b59b82ed	21	from qemu import QEMUMachine, kvm_available
ff2ebff0 FZ	22	import subprocess
	23	import hashlib
	24	import optparse
	25	import atexit
	26	import tempfile
	27	import shutil
	28	import multiprocessing
	29	import traceback
	30
	31	SSH_KEY = open(os.path.join(os.path.dirname(__file__),
	32	"..", "keys", "id_rsa")).read()
	33	SSH_PUB_KEY = open(os.path.join(os.path.dirname(__file__),
	34	"..", "keys", "id_rsa.pub")).read()
	35
	36	class BaseVM(object):
	37	GUEST_USER = "qemu"
	38	GUEST_PASS = "qemupass"
	39	ROOT_PASS = "qemupass"
	40
	41	# The script to run in the guest that builds QEMU
	42	BUILD_SCRIPT = ""
	43	# The guest name, to be overridden by subclasses
	44	name = "#base"
	45	def __init__(self, debug=False, vcpus=None):
	46	self._guest = None
	47	self._tmpdir = os.path.realpath(tempfile.mkdtemp(prefix="vm-test-",
	48	suffix=".tmp",
	49	dir="."))
	50	atexit.register(shutil.rmtree, self._tmpdir)
	51
	52	self._ssh_key_file = os.path.join(self._tmpdir, "id_rsa")
	53	open(self._ssh_key_file, "w").write(SSH_KEY)
	54	subprocess.check_call(["chmod", "600", self._ssh_key_file])
	55
	56	self._ssh_pub_key_file = os.path.join(self._tmpdir, "id_rsa.pub")
	57	open(self._ssh_pub_key_file, "w").write(SSH_PUB_KEY)
	58
	59	self.debug = debug
	60	self._stderr = sys.stderr
	61	self._devnull = open(os.devnull, "w")
	62	if self.debug:
	63	self._stdout = sys.stdout
	64	else:
	65	self._stdout = self._devnull
	66	self._args = [ \
eb2712f5	67	"-nodefaults", "-m", "4G",
b33bd859	68	"-cpu", "max",
ff2ebff0 FZ	69	"-netdev", "user,id=vnet,hostfwd=:127.0.0.1:0-:22",
	70	"-device", "virtio-net-pci,netdev=vnet",
	71	"-vnc", "127.0.0.1:0,to=20",
	72	"-serial", "file:%s" % os.path.join(self._tmpdir, "serial.out")]
	73	if vcpus:
	74	self._args += ["-smp", str(vcpus)]
b59b82ed	75	if kvm_available():
ff2ebff0 FZ	76	self._args += ["-enable-kvm"]
	77	else:
	78	logging.info("KVM not available, not using -enable-kvm")
	79	self._data_args = []
	80
	81	def _download_with_cache(self, url, sha256sum=None):
	82	def check_sha256sum(fname):
	83	if not sha256sum:
	84	return True
	85	checksum = subprocess.check_output(["sha256sum", fname]).split()[0]
	86	return sha256sum == checksum
	87
	88	cache_dir = os.path.expanduser("~/.cache/qemu-vm/download")
	89	if not os.path.exists(cache_dir):
	90	os.makedirs(cache_dir)
	91	fname = os.path.join(cache_dir, hashlib.sha1(url).hexdigest())
	92	if os.path.exists(fname) and check_sha256sum(fname):
	93	return fname
	94	logging.debug("Downloading %s to %s...", url, fname)
	95	subprocess.check_call(["wget", "-c", url, "-O", fname + ".download"],
	96	stdout=self._stdout, stderr=self._stderr)
	97	os.rename(fname + ".download", fname)
	98	return fname
	99
	100	def _ssh_do(self, user, cmd, check, interactive=False):
	101	ssh_cmd = ["ssh", "-q",
	102	"-o", "StrictHostKeyChecking=no",
	103	"-o", "UserKnownHostsFile=" + os.devnull,
	104	"-o", "ConnectTimeout=1",
	105	"-p", self.ssh_port, "-i", self._ssh_key_file]
	106	if interactive:
	107	ssh_cmd += ['-t']
	108	assert not isinstance(cmd, str)
	109	ssh_cmd += ["%s@127.0.0.1" % user] + list(cmd)
	110	logging.debug("ssh_cmd: %s", " ".join(ssh_cmd))
726c9a3b	111	r = subprocess.call(ssh_cmd)
ff2ebff0 FZ	112	if check and r != 0:
	113	raise Exception("SSH command failed: %s" % cmd)
	114	return r
	115
	116	def ssh(self, *cmd):
	117	return self._ssh_do(self.GUEST_USER, cmd, False)
	118
	119	def ssh_interactive(self, *cmd):
	120	return self._ssh_do(self.GUEST_USER, cmd, False, True)
	121
	122	def ssh_root(self, *cmd):
	123	return self._ssh_do("root", cmd, False)
	124
	125	def ssh_check(self, *cmd):
	126	self._ssh_do(self.GUEST_USER, cmd, True)
	127
	128	def ssh_root_check(self, *cmd):
	129	self._ssh_do("root", cmd, True)
	130
	131	def build_image(self, img):
	132	raise NotImplementedError
	133
	134	def add_source_dir(self, src_dir):
	135	name = "data-" + hashlib.sha1(src_dir).hexdigest()[:5]
	136	tarfile = os.path.join(self._tmpdir, name + ".tar")
	137	logging.debug("Creating archive %s for src_dir dir: %s", tarfile, src_dir)
	138	subprocess.check_call(["./scripts/archive-source.sh", tarfile],
	139	cwd=src_dir, stdin=self._devnull,
	140	stdout=self._stdout, stderr=self._stderr)
	141	self._data_args += ["-drive",
	142	"file=%s,if=none,id=%s,cache=writeback,format=raw" % \
	143	(tarfile, name),
	144	"-device",
	145	"virtio-blk,drive=%s,serial=%s,bootindex=1" % (name, name)]
	146
	147	def boot(self, img, extra_args=[]):
	148	args = self._args + [
	149	"-device", "VGA",
	150	"-drive", "file=%s,if=none,id=drive0,cache=writeback" % img,
	151	"-device", "virtio-blk,drive=drive0,bootindex=0"]
	152	args += self._data_args + extra_args
	153	logging.debug("QEMU args: %s", " ".join(args))
	154	qemu_bin = os.environ.get("QEMU", "qemu-system-x86_64")
	155	guest = QEMUMachine(binary=qemu_bin, args=args)
	156	try:
	157	guest.launch()
	158	except:
	159	logging.error("Failed to launch QEMU, command line:")
	160	logging.error(" ".join([qemu_bin] + args))
	161	logging.error("Log:")
	162	logging.error(guest.get_log())
	163	logging.error("QEMU version >= 2.10 is required")
	164	raise
	165	atexit.register(self.shutdown)
	166	self._guest = guest
	167	usernet_info = guest.qmp("human-monitor-command",
	168	command_line="info usernet")
	169	self.ssh_port = None
	170	for l in usernet_info["return"].splitlines():
	171	fields = l.split()
	172	if "TCP[HOST_FORWARD]" in fields and "22" in fields:
	173	self.ssh_port = l.split()[3]
	174	if not self.ssh_port:
	175	raise Exception("Cannot find ssh port from 'info usernet':\n%s" % \
176	usernet_info)
177
6b699ae1	178	def wait_ssh(self, seconds=300):
ff2ebff0 FZ	179	starttime = datetime.datetime.now()
	180	guest_up = False
	181	while (datetime.datetime.now() - starttime).total_seconds() < seconds:
	182	if self.ssh("exit 0") == 0:
	183	guest_up = True
	184	break
	185	time.sleep(1)
	186	if not guest_up:
	187	raise Exception("Timeout while waiting for guest ssh")
	188
	189	def shutdown(self):
	190	self._guest.shutdown()
	191
	192	def wait(self):
	193	self._guest.wait()
	194
	195	def qmp(self, args, *kwargs):
	196	return self._guest.qmp(args, *kwargs)
	197
	198	def parse_args(vm_name):
	199	parser = optparse.OptionParser(
	200	description="VM test utility. Exit codes: "
	201	"0 = success, "
	202	"1 = command line error, "
	203	"2 = environment initialization failed, "
	204	"3 = test command failed")
	205	parser.add_option("--debug", "-D", action="store_true",
	206	help="enable debug output")
	207	parser.add_option("--image", "-i", default="%s.img" % vm_name,
	208	help="image file name")
	209	parser.add_option("--force", "-f", action="store_true",
	210	help="force build image even if image exists")
	211	parser.add_option("--jobs", type=int, default=multiprocessing.cpu_count() / 2,
	212	help="number of virtual CPUs")
41e3340a PM	213	parser.add_option("--verbose", "-V", action="store_true",
41e3340a PM	214	help="Pass V=1 to builds within the guest")
ff2ebff0 FZ	215	parser.add_option("--build-image", "-b", action="store_true",
	216	help="build image")
	217	parser.add_option("--build-qemu",
	218	help="build QEMU from source in guest")
	219	parser.add_option("--interactive", "-I", action="store_true",
	220	help="Interactively run command")
983c2a77 FZ	221	parser.add_option("--snapshot", "-s", action="store_true",
983c2a77 FZ	222	help="run tests with a snapshot")
ff2ebff0 FZ	223	parser.disable_interspersed_args()
	224	return parser.parse_args()
	225
	226	def main(vmcls):
	227	try:
	228	args, argv = parse_args(vmcls.name)
	229	if not argv and not args.build_qemu and not args.build_image:
f03868bd	230	print("Nothing to do?")
ff2ebff0	231	return 1
fb3b4e6d EH	232	logging.basicConfig(level=(logging.DEBUG if args.debug
fb3b4e6d EH	233	else logging.WARN))
ff2ebff0 FZ	234	vm = vmcls(debug=args.debug, vcpus=args.jobs)
	235	if args.build_image:
	236	if os.path.exists(args.image) and not args.force:
	237	sys.stderr.writelines(["Image file exists: %s\n" % args.image,
	238	"Use --force option to overwrite\n"])
	239	return 1
	240	return vm.build_image(args.image)
	241	if args.build_qemu:
	242	vm.add_source_dir(args.build_qemu)
	243	cmd = [vm.BUILD_SCRIPT.format(
	244	configure_opts = " ".join(argv),
41e3340a PM	245	jobs=args.jobs,
41e3340a PM	246	verbose = "V=1" if args.verbose else "")]
ff2ebff0 FZ	247	else:
ff2ebff0 FZ	248	cmd = argv
983c2a77 FZ	249	img = args.image
	250	if args.snapshot:
	251	img += ",snapshot=on"
	252	vm.boot(img)
ff2ebff0 FZ	253	vm.wait_ssh()
	254	except Exception as e:
	255	if isinstance(e, SystemExit) and e.code == 0:
	256	return 0
	257	sys.stderr.write("Failed to prepare guest environment\n")
	258	traceback.print_exc()
	259	return 2
	260
	261	if args.interactive:
	262	if vm.ssh_interactive(*cmd) == 0:
	263	return 0
	264	vm.ssh_interactive()
	265	return 3
	266	else:
	267	if vm.ssh(*cmd) != 0:
	268	return 3