xref: /qemu/tests/avocado/virtio_check_params.py (revision b2a3cbb8) 
1#
2# Test virtio-scsi and virtio-blk queue settings for all machine types
3#
4# Copyright (c) 2019 Virtuozzo International GmbH
5#
6# This program is free software; you can redistribute it and/or modify
7# it under the terms of the GNU General Public License as published by
8# the Free Software Foundation; either version 2 of the License, or
9# (at your option) any later version.
10#
11# This program is distributed in the hope that it will be useful,
12# but WITHOUT ANY WARRANTY; without even the implied warranty of
13# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14# GNU General Public License for more details.
15#
16# You should have received a copy of the GNU General Public License
17# along with this program.  If not, see <http://www.gnu.org/licenses/>.
18#
19
20import sys
21import os
22import re
23import logging
24
25from qemu.machine import QEMUMachine
26from avocado_qemu import QemuSystemTest
27from avocado import skip
28
29#list of machine types and virtqueue properties to test
30VIRTIO_SCSI_PROPS = {'seg_max_adjust': 'seg_max_adjust'}
31VIRTIO_BLK_PROPS = {'seg_max_adjust': 'seg-max-adjust'}
32
33DEV_TYPES = {'virtio-scsi-pci': VIRTIO_SCSI_PROPS,
34             'virtio-blk-pci': VIRTIO_BLK_PROPS}
35
36VM_DEV_PARAMS = {'virtio-scsi-pci': ['-device', 'virtio-scsi-pci,id=scsi0'],
37                 'virtio-blk-pci': ['-device',
38                                    'virtio-blk-pci,id=scsi0,drive=drive0',
39                                    '-drive',
40                                    'driver=null-co,id=drive0,if=none']}
41
42
43class VirtioMaxSegSettingsCheck(QemuSystemTest):
44    @staticmethod
45    def make_pattern(props):
46        pattern_items = ['{0} = \w+'.format(prop) for prop in props]
47        return '|'.join(pattern_items)
48
49    def query_virtqueue(self, vm, dev_type_name):
50        query_ok = False
51        error = None
52        props = None
53
54        output = vm.command('human-monitor-command',
55                            command_line = 'info qtree')
56        props_list = DEV_TYPES[dev_type_name].values();
57        pattern = self.make_pattern(props_list)
58        res = re.findall(pattern, output)
59
60        if len(res) != len(props_list):
61            props_list = set(props_list)
62            res = set(res)
63            not_found = props_list.difference(res)
64            not_found = ', '.join(not_found)
65            error = '({0}): The following properties not found: {1}'\
66                     .format(dev_type_name, not_found)
67        else:
68            query_ok = True
69            props = dict()
70            for prop in res:
71                p = prop.split(' = ')
72                props[p[0]] = p[1]
73        return query_ok, props, error
74
75    def check_mt(self, mt, dev_type_name):
76        mt['device'] = dev_type_name # Only for the debug() call.
77        logger = logging.getLogger('machine')
78        logger.debug(mt)
79        with QEMUMachine(self.qemu_bin) as vm:
80            vm.set_machine(mt["name"])
81            vm.add_args('-nodefaults')
82            for s in VM_DEV_PARAMS[dev_type_name]:
83                vm.add_args(s)
84            try:
85                vm.launch()
86                query_ok, props, error = self.query_virtqueue(vm, dev_type_name)
87            except:
88                query_ok = False
89                error = sys.exc_info()[0]
90
91        if not query_ok:
92            self.fail('machine type {0}: {1}'.format(mt['name'], error))
93
94        for prop_name, prop_val in props.items():
95            expected_val = mt[prop_name]
96            self.assertEqual(expected_val, prop_val)
97
98    @staticmethod
99    def seg_max_adjust_enabled(mt):
100        # machine types >= 5.0 should have seg_max_adjust = true
101        # others seg_max_adjust = false
102        mt = mt.split("-")
103
104        # machine types with one line name and name like pc-x.x
105        if len(mt) <= 2:
106            return False
107
108        # machine types like pc-<chip_name>-x.x[.x]
109        ver = mt[2]
110        ver = ver.split(".");
111
112        # versions >= 5.0 goes with seg_max_adjust enabled
113        major = int(ver[0])
114
115        if major >= 5:
116            return True
117        return False
118
119    @skip("break multi-arch CI")
120    def test_machine_types(self):
121        # collect all machine types except 'none', 'isapc', 'microvm'
122        with QEMUMachine(self.qemu_bin) as vm:
123            vm.launch()
124            machines = [m['name'] for m in vm.command('query-machines')]
125            vm.shutdown()
126        machines.remove('none')
127        machines.remove('isapc')
128        machines.remove('microvm')
129
130        for dev_type in DEV_TYPES:
131            # create the list of machine types and their parameters.
132            mtypes = list()
133            for m in machines:
134                if self.seg_max_adjust_enabled(m):
135                    enabled = 'true'
136                else:
137                    enabled = 'false'
138                mtypes.append({'name': m,
139                               DEV_TYPES[dev_type]['seg_max_adjust']: enabled})
140
141            # test each machine type for a device type
142            for mt in mtypes:
143                self.check_mt(mt, dev_type)
144