xref: /qemu/tests/functional/test_intel_iommu.py (revision 513823e7521a09ed7ad1e32e6454bac3b2cbf52d) 
1#!/usr/bin/env python3
2#
3# INTEL_IOMMU Functional tests
4#
5# Copyright (c) 2021 Red Hat, Inc.
6#
7# Author:
8#  Eric Auger <eric.auger@redhat.com>
9#
10# This work is licensed under the terms of the GNU GPL, version 2 or
11# later.  See the COPYING file in the top-level directory.
12
13import hashlib
14import urllib.request
15
16from qemu_test import LinuxKernelTest, Asset, exec_command_and_wait_for_pattern
17from qemu_test.utils import get_usernet_hostfwd_port
18
19
20class IntelIOMMU(LinuxKernelTest):
21
22    ASSET_KERNEL = Asset(
23        ('https://archives.fedoraproject.org/pub/archive/fedora/linux/releases'
24         '/31/Server/x86_64/os/images/pxeboot/vmlinuz'),
25        'd4738d03dbbe083ca610d0821d0a8f1488bebbdccef54ce33e3adb35fda00129')
26
27    ASSET_INITRD = Asset(
28        ('https://archives.fedoraproject.org/pub/archive/fedora/linux/releases'
29         '/31/Server/x86_64/os/images/pxeboot/initrd.img'),
30        '277cd6c7adf77c7e63d73bbb2cded8ef9e2d3a2f100000e92ff1f8396513cd8b')
31
32    ASSET_DISKIMAGE = Asset(
33        ('https://archives.fedoraproject.org/pub/archive/fedora/linux/releases'
34         '/31/Cloud/x86_64/images/Fedora-Cloud-Base-31-1.9.x86_64.qcow2'),
35        'e3c1b309d9203604922d6e255c2c5d098a309c2d46215d8fc026954f3c5c27a0')
36
37    DEFAULT_KERNEL_PARAMS = ('root=/dev/vda1 console=ttyS0 net.ifnames=0 '
38                             'quiet rd.rescue ')
39    GUEST_PORT = 8080
40    IOMMU_ADDON = ',iommu_platform=on,disable-modern=off,disable-legacy=on'
41    kernel_path = None
42    initrd_path = None
43    kernel_params = None
44
45    def add_common_args(self, path):
46        self.vm.add_args('-drive', f'file={path},if=none,id=drv0,snapshot=on')
47        self.vm.add_args('-device', 'virtio-blk-pci,bus=pcie.0,' +
48                         'drive=drv0,id=virtio-disk0,bootindex=1,'
49                         'werror=stop,rerror=stop' + self.IOMMU_ADDON)
50        self.vm.add_args('-device', 'virtio-gpu-pci' + self.IOMMU_ADDON)
51
52        self.vm.add_args('-netdev',
53                         'user,id=n1,hostfwd=tcp:127.0.0.1:0-:%d' %
54                         self.GUEST_PORT)
55        self.vm.add_args('-device',
56                         'virtio-net-pci,netdev=n1' + self.IOMMU_ADDON)
57
58        self.vm.add_args('-device', 'virtio-rng-pci,rng=rng0')
59        self.vm.add_args('-object',
60                         'rng-random,id=rng0,filename=/dev/urandom')
61        self.vm.add_args("-m", "1G")
62        self.vm.add_args("-accel", "kvm")
63
64    def common_vm_setup(self):
65        self.set_machine('q35')
66        self.require_accelerator("kvm")
67        self.require_netdev('user')
68
69        self.kernel_path = self.ASSET_KERNEL.fetch()
70        self.initrd_path = self.ASSET_INITRD.fetch()
71        image_path = self.ASSET_DISKIMAGE.fetch()
72        self.add_common_args(image_path)
73        self.kernel_params = self.DEFAULT_KERNEL_PARAMS
74
75    def run_and_check(self):
76        if self.kernel_path:
77            self.vm.add_args('-kernel', self.kernel_path,
78                             '-append', self.kernel_params,
79                             '-initrd', self.initrd_path)
80        self.vm.set_console()
81        self.vm.launch()
82        self.wait_for_console_pattern('Entering emergency mode.')
83        prompt = '# '
84        self.wait_for_console_pattern(prompt)
85
86        # Copy a file (checked later), umount afterwards to drop disk cache:
87        exec_command_and_wait_for_pattern(self, 'mount /dev/vda1 /sysroot',
88                                          prompt)
89        filename = '/boot/initramfs-5.3.7-301.fc31.x86_64.img'
90        exec_command_and_wait_for_pattern(self, (f'cp /sysroot{filename}'
91                                                 ' /sysroot/root/data'),
92                                          prompt)
93        exec_command_and_wait_for_pattern(self, 'umount /sysroot', prompt)
94
95        # Switch from initrd to the cloud image filesystem:
96        exec_command_and_wait_for_pattern(self, 'mount /dev/vda1 /sysroot',
97                                          prompt)
98        exec_command_and_wait_for_pattern(self,
99                ('for d in dev proc sys run ; do '
100                 'mount -o bind /$d /sysroot/$d ; done'), prompt)
101        exec_command_and_wait_for_pattern(self, 'chroot /sysroot', prompt)
102
103        # Checking for IOMMU enablement:
104        self.log.info("Checking whether IOMMU has been enabled...")
105        exec_command_and_wait_for_pattern(self, 'cat /proc/cmdline',
106                                          'intel_iommu=on')
107        self.wait_for_console_pattern(prompt)
108        exec_command_and_wait_for_pattern(self, 'dmesg | grep DMAR:',
109                                          'IOMMU enabled')
110        self.wait_for_console_pattern(prompt)
111        exec_command_and_wait_for_pattern(self,
112                                    'find /sys/kernel/iommu_groups/ -type l',
113                                    'devices/0000:00:')
114        self.wait_for_console_pattern(prompt)
115
116        # Check hard disk device via sha256sum:
117        self.log.info("Checking hard disk...")
118        hashsum = '0dc7472f879be70b2f3daae279e3ae47175ffe249691e7d97f47222b65b8a720'
119        exec_command_and_wait_for_pattern(self, 'sha256sum ' + filename,
120                                          hashsum)
121        self.wait_for_console_pattern(prompt)
122        exec_command_and_wait_for_pattern(self, 'sha256sum /root/data',
123                                          hashsum)
124        self.wait_for_console_pattern(prompt)
125
126        # Check virtio-net via HTTP:
127        exec_command_and_wait_for_pattern(self, 'dhclient eth0', prompt)
128        exec_command_and_wait_for_pattern(self,
129                        f'python3 -m http.server {self.GUEST_PORT} & sleep 1',
130                        f'Serving HTTP on 0.0.0.0 port {self.GUEST_PORT}')
131        hl = hashlib.sha256()
132        hostport = get_usernet_hostfwd_port(self.vm)
133        url = f'http://localhost:{hostport}{filename}'
134        self.log.info(f'Downloading {url} ...')
135        with urllib.request.urlopen(url) as response:
136            while True:
137                chunk = response.read(1 << 20)
138                if not chunk:
139                    break
140                hl.update(chunk)
141
142        digest = hl.hexdigest()
143        self.log.info(f'sha256sum of download is {digest}.')
144        self.assertEqual(digest, hashsum)
145
146    def test_intel_iommu(self):
147        self.common_vm_setup()
148        self.vm.add_args('-device', 'intel-iommu,intremap=on')
149        self.vm.add_args('-machine', 'kernel_irqchip=split')
150        self.kernel_params += 'intel_iommu=on'
151        self.run_and_check()
152
153    def test_intel_iommu_strict(self):
154        self.common_vm_setup()
155        self.vm.add_args('-device', 'intel-iommu,intremap=on')
156        self.vm.add_args('-machine', 'kernel_irqchip=split')
157        self.kernel_params += 'intel_iommu=on,strict'
158        self.run_and_check()
159
160    def test_intel_iommu_strict_cm(self):
161        self.common_vm_setup()
162        self.vm.add_args('-device', 'intel-iommu,intremap=on,caching-mode=on')
163        self.vm.add_args('-machine', 'kernel_irqchip=split')
164        self.kernel_params += 'intel_iommu=on,strict'
165        self.run_and_check()
166
167    def test_intel_iommu_pt(self):
168        self.common_vm_setup()
169        self.vm.add_args('-device', 'intel-iommu,intremap=on')
170        self.vm.add_args('-machine', 'kernel_irqchip=split')
171        self.kernel_params += 'intel_iommu=on iommu=pt'
172        self.run_and_check()
173
174if __name__ == '__main__':
175    LinuxKernelTest.main()
176