summaryrefslogtreecommitdiffstats
path: root/tests/vm/centos.aarch64
blob: 2de7ef6992c0f74878530ca46479e5d6b524267b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
#!/usr/bin/env python3
#
# Centos aarch64 image
#
# Copyright 2020 Linaro
#
# Authors:
#  Robert Foley <robert.foley@linaro.org>
#  Originally based on ubuntu.aarch64
#
# This code is licensed under the GPL version 2 or later.  See
# the COPYING file in the top-level directory.
#

import os
import sys
import subprocess
import basevm
import time
import traceback
import aarch64vm


DEFAULT_CONFIG = {
    'cpu'          : "max",
    'machine'      : "virt,gic-version=max",
    'install_cmds' : (
        "dnf config-manager --set-enabled powertools, "
        "dnf config-manager --add-repo=https://download.docker.com/linux/centos/docker-ce.repo, "
        "dnf install -y make ninja-build git python38 gcc gcc-c++ flex bison "\
            "glib2-devel perl pixman-devel zlib-devel docker-ce.aarch64, "
        "systemctl enable docker, "
    ),
    # We increase beyond the default time since during boot
    # it can take some time (many seconds) to log into the VM.
    'ssh_timeout'  : 60,
}


class CentosAarch64VM(basevm.BaseVM):
    name = "centos8.aarch64"
    arch = "aarch64"
    image_name = "CentOS-Stream-GenericCloud-8-20220125.1.aarch64.qcow2"
    image_link = "https://cloud.centos.org/centos/8-stream/aarch64/images/"
    image_link += image_name
    BUILD_SCRIPT = """
        set -e;
        cd $(mktemp -d);
        export SRC_ARCHIVE=/dev/vdb;
        sudo chmod a+r $SRC_ARCHIVE;
        tar -xf $SRC_ARCHIVE;
        ./configure {configure_opts};
        make --output-sync {target} -j{jobs} {verbose};
    """

    def boot(self, img, extra_args=None):
        aarch64vm.create_flash_images(self._tmpdir, self._efi_aarch64)
        default_args = aarch64vm.get_pflash_args(self._tmpdir)
        if extra_args:
            extra_args.extend(default_args)
        else:
            extra_args = default_args
        # We always add these performance tweaks
        # because without them, we boot so slowly that we
        # can time out finding the boot efi device.
        if '-smp' not in extra_args and \
           '-smp' not in self._config['extra_args'] and \
           '-smp' not in self._args:
            # Only add if not already there to give caller option to change it.
            extra_args.extend(["-smp", "8"])
        # We have overridden boot() since aarch64 has additional parameters.
        # Call down to the base class method.
        super(CentosAarch64VM, self).boot(img, extra_args=extra_args)

    def build_image(self, img):
        cimg = self._download_with_cache(self.image_link)
        img_tmp = img + ".tmp"
        subprocess.run(['cp', '-f', cimg, img_tmp])
        self.exec_qemu_img("resize", img_tmp, "50G")
        self.boot(img_tmp, extra_args = ["-cdrom", self.gen_cloud_init_iso()])
        self.wait_ssh(wait_root=True)
        self.ssh_root_check("touch /etc/cloud/cloud-init.disabled")

        # If the user chooses *not* to do the second phase,
        # then we will jump right to the graceful shutdown
        if self._config['install_cmds'] != "":
            install_cmds = self._config['install_cmds'].split(',')
            for cmd in install_cmds:
                self.ssh_root_check(cmd)

        self.ssh_root("poweroff")
        self.wait()
        os.rename(img_tmp, img)
        print("image creation complete: {}".format(img))
        return 0


if __name__ == "__main__":
    defaults = aarch64vm.get_config_defaults(CentosAarch64VM, DEFAULT_CONFIG)
    sys.exit(basevm.main(CentosAarch64VM, defaults))