cachepc-qemu

Fork of AMDESE/qemu with changes for cachepc side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-qemu
Log | Files | Refs | Submodules | LICENSE | sfeed.txt

virtio-gpu.py (5133B)

      1# virtio-gpu tests
      2#
      3# This work is licensed under the terms of the GNU GPL, version 2 or
      4# later.  See the COPYING file in the top-level directory.
      5
      6
      7from avocado_qemu import Test
      8from avocado_qemu import BUILD_DIR
      9from avocado_qemu import wait_for_console_pattern
     10from avocado_qemu import exec_command_and_wait_for_pattern
     11from avocado_qemu import is_readable_executable_file
     12
     13from qemu.utils import kvm_available
     14
     15import os
     16import socket
     17import subprocess
     18
     19
     20def pick_default_vug_bin():
     21    relative_path = "./contrib/vhost-user-gpu/vhost-user-gpu"
     22    if is_readable_executable_file(relative_path):
     23        return relative_path
     24
     25    bld_dir_path = os.path.join(BUILD_DIR, relative_path)
     26    if is_readable_executable_file(bld_dir_path):
     27        return bld_dir_path
     28
     29
     30class VirtioGPUx86(Test):
     31    """
     32    :avocado: tags=virtio-gpu
     33    :avocado: tags=arch:x86_64
     34    :avocado: tags=cpu:host
     35    """
     36
     37    KERNEL_COMMAND_LINE = "printk.time=0 console=ttyS0 rdinit=/bin/bash"
     38    KERNEL_URL = (
     39        "https://archives.fedoraproject.org/pub/fedora"
     40        "/linux/releases/33/Everything/x86_64/os/images"
     41        "/pxeboot/vmlinuz"
     42    )
     43    KERNEL_HASH = '1433cfe3f2ffaa44de4ecfb57ec25dc2399cdecf'
     44    INITRD_URL = (
     45        "https://archives.fedoraproject.org/pub/fedora"
     46        "/linux/releases/33/Everything/x86_64/os/images"
     47        "/pxeboot/initrd.img"
     48    )
     49    INITRD_HASH = 'c828d68a027b53e5220536585efe03412332c2d9'
     50
     51    def wait_for_console_pattern(self, success_message, vm=None):
     52        wait_for_console_pattern(
     53            self,
     54            success_message,
     55            failure_message="Kernel panic - not syncing",
     56            vm=vm,
     57        )
     58
     59    def test_virtio_vga_virgl(self):
     60        """
     61        :avocado: tags=device:virtio-vga-gl
     62        """
     63        # FIXME: should check presence of virtio, virgl etc
     64        self.require_accelerator('kvm')
     65
     66        kernel_path = self.fetch_asset(self.KERNEL_URL, self.KERNEL_HASH)
     67        initrd_path = self.fetch_asset(self.INITRD_URL, self.INITRD_HASH)
     68
     69        self.vm.set_console()
     70        self.vm.add_args("-m", "2G")
     71        self.vm.add_args("-machine", "pc,accel=kvm")
     72        self.vm.add_args("-device", "virtio-vga-gl")
     73        self.vm.add_args("-display", "egl-headless")
     74        self.vm.add_args(
     75            "-kernel",
     76            kernel_path,
     77            "-initrd",
     78            initrd_path,
     79            "-append",
     80            self.KERNEL_COMMAND_LINE,
     81        )
     82        try:
     83            self.vm.launch()
     84        except:
     85            # TODO: probably fails because we are missing the VirGL features
     86            self.cancel("VirGL not enabled?")
     87
     88        self.wait_for_console_pattern("as init process")
     89        exec_command_and_wait_for_pattern(
     90            self, "/usr/sbin/modprobe virtio_gpu", ""
     91        )
     92        self.wait_for_console_pattern("features: +virgl +edid")
     93
     94    def test_vhost_user_vga_virgl(self):
     95        """
     96        :avocado: tags=device:vhost-user-vga
     97        """
     98        # FIXME: should check presence of vhost-user-gpu, virgl, memfd etc
     99        self.require_accelerator('kvm')
    100
    101        vug = pick_default_vug_bin()
    102        if not vug:
    103            self.cancel("Could not find vhost-user-gpu")
    104
    105        kernel_path = self.fetch_asset(self.KERNEL_URL, self.KERNEL_HASH)
    106        initrd_path = self.fetch_asset(self.INITRD_URL, self.INITRD_HASH)
    107
    108        # Create socketpair to connect proxy and remote processes
    109        qemu_sock, vug_sock = socket.socketpair(
    110            socket.AF_UNIX, socket.SOCK_STREAM
    111        )
    112        os.set_inheritable(qemu_sock.fileno(), True)
    113        os.set_inheritable(vug_sock.fileno(), True)
    114
    115        self._vug_log_path = os.path.join(
    116            self.logdir, "vhost-user-gpu.log"
    117        )
    118        self._vug_log_file = open(self._vug_log_path, "wb")
    119        self.log.info('Complete vhost-user-gpu.log file can be '
    120                      'found at %s', self._vug_log_path)
    121
    122        vugp = subprocess.Popen(
    123            [vug, "--virgl", "--fd=%d" % vug_sock.fileno()],
    124            stdin=subprocess.DEVNULL,
    125            stdout=self._vug_log_file,
    126            stderr=subprocess.STDOUT,
    127            shell=False,
    128            close_fds=False,
    129        )
    130
    131        self.vm.set_console()
    132        self.vm.add_args("-m", "2G")
    133        self.vm.add_args("-object", "memory-backend-memfd,id=mem,size=2G")
    134        self.vm.add_args("-machine", "pc,memory-backend=mem,accel=kvm")
    135        self.vm.add_args("-chardev", "socket,id=vug,fd=%d" % qemu_sock.fileno())
    136        self.vm.add_args("-device", "vhost-user-vga,chardev=vug")
    137        self.vm.add_args("-display", "egl-headless")
    138        self.vm.add_args(
    139            "-kernel",
    140            kernel_path,
    141            "-initrd",
    142            initrd_path,
    143            "-append",
    144            self.KERNEL_COMMAND_LINE,
    145        )
    146        self.vm.launch()
    147        self.wait_for_console_pattern("as init process")
    148        exec_command_and_wait_for_pattern(
    149            self, "/usr/sbin/modprobe virtio_gpu", ""
    150        )
    151        self.wait_for_console_pattern("features: +virgl -edid")
    152        self.vm.shutdown()
    153        qemu_sock.close()
    154        vugp.terminate()
    155        vugp.wait()