xref: /qemu/tests/qemu-iotests/281 (revision d201cf7a) 
1#!/usr/bin/env python3
2# group: rw quick
3#
4# Test cases for blockdev + IOThread interactions
5#
6# Copyright (C) 2019 Red Hat, Inc.
7#
8# This program is free software; you can redistribute it and/or modify
9# it under the terms of the GNU General Public License as published by
10# the Free Software Foundation; either version 2 of the License, or
11# (at your option) any later version.
12#
13# This program is distributed in the hope that it will be useful,
14# but WITHOUT ANY WARRANTY; without even the implied warranty of
15# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16# GNU General Public License for more details.
17#
18# You should have received a copy of the GNU General Public License
19# along with this program.  If not, see <http://www.gnu.org/licenses/>.
20#
21
22import os
23import iotests
24from iotests import qemu_img
25
26image_len = 64 * 1024 * 1024
27
28# Test for RHBZ#1782175
29class TestDirtyBitmapIOThread(iotests.QMPTestCase):
30    drive0_img = os.path.join(iotests.test_dir, 'drive0.img')
31    images = { 'drive0': drive0_img }
32
33    def setUp(self):
34        for name in self.images:
35            qemu_img('create', '-f', iotests.imgfmt,
36                     self.images[name], str(image_len))
37
38        self.vm = iotests.VM()
39        self.vm.add_object('iothread,id=iothread0')
40
41        for name in self.images:
42            self.vm.add_blockdev('driver=file,filename=%s,node-name=file_%s'
43                                 % (self.images[name], name))
44            self.vm.add_blockdev('driver=qcow2,file=file_%s,node-name=%s'
45                                 % (name, name))
46
47        self.vm.launch()
48        self.vm.qmp('x-blockdev-set-iothread',
49                    node_name='drive0', iothread='iothread0',
50                    force=True)
51
52    def tearDown(self):
53        self.vm.shutdown()
54        for name in self.images:
55            os.remove(self.images[name])
56
57    def test_add_dirty_bitmap(self):
58        result = self.vm.qmp(
59            'block-dirty-bitmap-add',
60            node='drive0',
61            name='bitmap1',
62            persistent=True,
63        )
64
65        self.assert_qmp(result, 'return', {})
66
67
68# Test for RHBZ#1746217 & RHBZ#1773517
69class TestNBDMirrorIOThread(iotests.QMPTestCase):
70    nbd_sock = os.path.join(iotests.sock_dir, 'nbd.sock')
71    drive0_img = os.path.join(iotests.test_dir, 'drive0.img')
72    mirror_img = os.path.join(iotests.test_dir, 'mirror.img')
73    images = { 'drive0': drive0_img, 'mirror': mirror_img }
74
75    def setUp(self):
76        for name in self.images:
77            qemu_img('create', '-f', iotests.imgfmt,
78                     self.images[name], str(image_len))
79
80        self.vm_src = iotests.VM(path_suffix='src')
81        self.vm_src.add_object('iothread,id=iothread0')
82        self.vm_src.add_blockdev('driver=file,filename=%s,node-name=file0'
83                          % (self.drive0_img))
84        self.vm_src.add_blockdev('driver=qcow2,file=file0,node-name=drive0')
85        self.vm_src.launch()
86        self.vm_src.qmp('x-blockdev-set-iothread',
87                        node_name='drive0', iothread='iothread0',
88                        force=True)
89
90        self.vm_tgt = iotests.VM(path_suffix='tgt')
91        self.vm_tgt.add_object('iothread,id=iothread0')
92        self.vm_tgt.add_blockdev('driver=file,filename=%s,node-name=file0'
93                          % (self.mirror_img))
94        self.vm_tgt.add_blockdev('driver=qcow2,file=file0,node-name=drive0')
95        self.vm_tgt.launch()
96        self.vm_tgt.qmp('x-blockdev-set-iothread',
97                        node_name='drive0', iothread='iothread0',
98                        force=True)
99
100    def tearDown(self):
101        self.vm_src.shutdown()
102        self.vm_tgt.shutdown()
103        for name in self.images:
104            os.remove(self.images[name])
105
106    def test_nbd_mirror(self):
107        result = self.vm_tgt.qmp(
108            'nbd-server-start',
109            addr={
110                'type': 'unix',
111                'data': { 'path': self.nbd_sock }
112            }
113        )
114        self.assert_qmp(result, 'return', {})
115
116        result = self.vm_tgt.qmp(
117            'nbd-server-add',
118            device='drive0',
119            writable=True
120        )
121        self.assert_qmp(result, 'return', {})
122
123        result = self.vm_src.qmp(
124            'drive-mirror',
125            device='drive0',
126            target='nbd+unix:///drive0?socket=' + self.nbd_sock,
127            sync='full',
128            mode='existing',
129            speed=64*1024*1024,
130            job_id='j1'
131        )
132        self.assert_qmp(result, 'return', {})
133
134        self.vm_src.event_wait(name="BLOCK_JOB_READY")
135
136
137# Test for RHBZ#1779036
138class TestExternalSnapshotAbort(iotests.QMPTestCase):
139    drive0_img = os.path.join(iotests.test_dir, 'drive0.img')
140    snapshot_img = os.path.join(iotests.test_dir, 'snapshot.img')
141    images = { 'drive0': drive0_img, 'snapshot': snapshot_img }
142
143    def setUp(self):
144        for name in self.images:
145            qemu_img('create', '-f', iotests.imgfmt,
146                     self.images[name], str(image_len))
147
148        self.vm = iotests.VM()
149        self.vm.add_object('iothread,id=iothread0')
150        self.vm.add_blockdev('driver=file,filename=%s,node-name=file0'
151                          % (self.drive0_img))
152        self.vm.add_blockdev('driver=qcow2,file=file0,node-name=drive0')
153        self.vm.launch()
154        self.vm.qmp('x-blockdev-set-iothread',
155                    node_name='drive0', iothread='iothread0',
156                    force=True)
157
158    def tearDown(self):
159        self.vm.shutdown()
160        for name in self.images:
161            os.remove(self.images[name])
162
163    def test_external_snapshot_abort(self):
164        # Use a two actions transaction with a bogus values on the second
165        # one to trigger an abort of the transaction.
166        result = self.vm.qmp('transaction', actions=[
167            {
168                'type': 'blockdev-snapshot-sync',
169                'data': { 'node-name': 'drive0',
170                          'snapshot-file': self.snapshot_img,
171                          'snapshot-node-name': 'snap1',
172                          'mode': 'absolute-paths',
173                          'format': 'qcow2' }
174            },
175            {
176                'type': 'blockdev-snapshot-sync',
177                'data': { 'node-name': 'drive0',
178                          'snapshot-file': '/fakesnapshot',
179                          'snapshot-node-name': 'snap2',
180                          'mode': 'absolute-paths',
181                          'format': 'qcow2' }
182            },
183        ])
184
185        # Crashes on failure, we expect this error.
186        self.assert_qmp(result, 'error/class', 'GenericError')
187
188
189# Test for RHBZ#1782111
190class TestBlockdevBackupAbort(iotests.QMPTestCase):
191    drive0_img = os.path.join(iotests.test_dir, 'drive0.img')
192    drive1_img = os.path.join(iotests.test_dir, 'drive1.img')
193    snap0_img = os.path.join(iotests.test_dir, 'snap0.img')
194    snap1_img = os.path.join(iotests.test_dir, 'snap1.img')
195    images = { 'drive0': drive0_img,
196               'drive1': drive1_img,
197               'snap0': snap0_img,
198               'snap1': snap1_img }
199
200    def setUp(self):
201        for name in self.images:
202            qemu_img('create', '-f', iotests.imgfmt,
203                     self.images[name], str(image_len))
204
205        self.vm = iotests.VM()
206        self.vm.add_object('iothread,id=iothread0')
207        self.vm.add_device('virtio-scsi,iothread=iothread0')
208
209        for name in self.images:
210            self.vm.add_blockdev('driver=file,filename=%s,node-name=file_%s'
211                                 % (self.images[name], name))
212            self.vm.add_blockdev('driver=qcow2,file=file_%s,node-name=%s'
213                                 % (name, name))
214
215        self.vm.add_device('scsi-hd,drive=drive0')
216        self.vm.add_device('scsi-hd,drive=drive1')
217        self.vm.launch()
218
219    def tearDown(self):
220        self.vm.shutdown()
221        for name in self.images:
222            os.remove(self.images[name])
223
224    def test_blockdev_backup_abort(self):
225        # Use a two actions transaction with a bogus values on the second
226        # one to trigger an abort of the transaction.
227        result = self.vm.qmp('transaction', actions=[
228            {
229                'type': 'blockdev-backup',
230                'data': { 'device': 'drive0',
231                          'target': 'snap0',
232                          'sync': 'full',
233                          'job-id': 'j1' }
234            },
235            {
236                'type': 'blockdev-backup',
237                'data': { 'device': 'drive1',
238                          'target': 'snap1',
239                          'sync': 'full' }
240            },
241        ])
242
243        # Hangs on failure, we expect this error.
244        self.assert_qmp(result, 'error/class', 'GenericError')
245
246if __name__ == '__main__':
247    iotests.main(supported_fmts=['qcow2'],
248                 supported_protocols=['file'],
249                 unsupported_imgopts=['compat'])
250