Merge remote-tracking branch 'remotes/mst/tags/for_upstream' into staging
[qemu.git] / tests / qemu-iotests / 148
1 #!/usr/bin/env python3
2 # group: rw quick
3 #
4 # Test the rate limit of QMP events
5 #
6 # Copyright (C) 2016 Igalia, S.L.
7 # Author: Alberto Garcia <berto@igalia.com>
8 #
9 # This program is free software; you can redistribute it and/or modify
10 # it under the terms of the GNU General Public License as published by
11 # the Free Software Foundation; either version 2 of the License, or
12 # (at your option) any later version.
13 #
14 # This program is distributed in the hope that it will be useful,
15 # but WITHOUT ANY WARRANTY; without even the implied warranty of
16 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17 # GNU General Public License for more details.
18 #
19 # You should have received a copy of the GNU General Public License
20 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
21 #
22
23 import os
24 import iotests
25
26 imgs = (os.path.join(iotests.test_dir, 'quorum0.img'),
27         os.path.join(iotests.test_dir, 'quorum1.img'),
28         os.path.join(iotests.test_dir, 'quorum2.img'))
29
30 img_conf = (os.path.join(iotests.test_dir, 'quorum0.conf'),
31             os.path.join(iotests.test_dir, 'quorum1.conf'),
32             os.path.join(iotests.test_dir, 'quorum2.conf'))
33
34 event_rate = 1000000000
35 sector_size = 512
36 offset = 10
37
38 class TestQuorumEvents(iotests.QMPTestCase):
39     read_pattern = 'quorum'
40
41     def create_blkdebug_file(self, blkdebug_file, bad_sector):
42         file = open(blkdebug_file, 'w')
43         file.write('''
44 [inject-error]
45 event = "read_aio"
46 errno = "5"
47 sector = "%d"
48 ''' % bad_sector)
49         file.close()
50
51     @iotests.skip_if_unsupported(['quorum'])
52     def setUp(self):
53         driveopts = ['driver=quorum', 'vote-threshold=2']
54         driveopts.append('read-pattern=%s' % self.read_pattern)
55         for i in range(len(imgs)):
56             iotests.qemu_img('create', '-f', iotests.imgfmt, imgs[i], '1M')
57             self.create_blkdebug_file(img_conf[i], i + offset)
58             driveopts.append('children.%d.driver=%s' % (i, iotests.imgfmt))
59             driveopts.append('children.%d.file.driver=blkdebug' % i)
60             driveopts.append('children.%d.file.config=%s' % (i, img_conf[i]))
61             driveopts.append('children.%d.file.image.filename=%s' % (i, imgs[i]))
62             driveopts.append('children.%d.node-name=img%d' % (i, i))
63         self.vm = iotests.VM()
64         self.vm.add_drive(None, opts = ','.join(driveopts))
65         self.vm.launch()
66
67     def tearDown(self):
68         self.vm.shutdown()
69         for i in range(len(imgs)):
70             os.remove(imgs[i])
71             os.remove(img_conf[i])
72
73     def do_check_event(self, node, sector = 0):
74         if node == None:
75             self.assertEqual(self.vm.get_qmp_event(), None)
76             return
77
78         for event in self.vm.get_qmp_events(wait=True):
79             if event['event'] == 'QUORUM_REPORT_BAD':
80                 self.assert_qmp(event, 'data/node-name', node)
81                 self.assert_qmp(event, 'data/sector-num', sector)
82
83     def testQuorum(self):
84         # Generate an error and get an event
85         self.vm.hmp_qemu_io("drive0", "aio_read %d %d" %
86                             (offset * sector_size, sector_size))
87         self.vm.qtest("clock_step 10")
88         self.do_check_event('img0', offset)
89
90         # I/O errors in the same child: only one event is emitted
91         delay = 10
92         for i in range(3):
93             self.vm.hmp_qemu_io("drive0", "aio_read %d %d" %
94                                 (offset * sector_size, sector_size))
95             self.vm.qtest("clock_step %d" % delay)
96             self.do_check_event(None)
97
98         # Wait enough so the event is finally emitted
99         self.vm.qtest("clock_step %d" % (2 * event_rate))
100         self.do_check_event('img0', offset)
101
102         # I/O errors in the same child: all events are emitted
103         delay = 2 * event_rate
104         for i in range(3):
105             self.vm.hmp_qemu_io("drive0", "aio_read %d %d" %
106                                 (offset * sector_size, sector_size))
107             self.vm.qtest("clock_step %d" % delay)
108             self.do_check_event('img0', offset)
109
110         # I/O errors in different children: all events are emitted
111         delay = 10
112         for i in range(len(imgs)):
113             self.vm.hmp_qemu_io("drive0", "aio_read %d %d" %
114                                 ((offset + i) * sector_size, sector_size))
115             self.vm.qtest("clock_step %d" % delay)
116             # In fifo mode only errors in the first child are detected
117             if i > 0 and self.read_pattern == 'fifo':
118                 self.do_check_event(None)
119             else:
120                 self.do_check_event('img%d' % i, offset + i)
121
122         # I/O errors in different children: all events are emitted
123         delay = 2 * event_rate
124         for i in range(len(imgs)):
125             self.vm.hmp_qemu_io("drive0", "aio_read %d %d" %
126                                 ((offset + i) * sector_size, sector_size))
127             self.vm.qtest("clock_step %d" % delay)
128             # In fifo mode only errors in the first child are detected
129             if i > 0 and self.read_pattern == 'fifo':
130                 self.do_check_event(None)
131             else:
132                 self.do_check_event('img%d' % i, offset + i)
133
134         # No more pending events
135         self.do_check_event(None)
136
137 class TestFifoQuorumEvents(TestQuorumEvents):
138     read_pattern = 'fifo'
139
140 if __name__ == '__main__':
141     iotests.verify_quorum()
142     iotests.main(supported_fmts=["raw"],
143                  supported_protocols=["file"])