tests/acceptance/virtio-gpu.py

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