xref: /openbmc/qemu/tests/qemu-iotests/194 (revision 2e1cacfb) 
1#!/usr/bin/env python3
2# group: rw migration quick
3#
4# Copyright (C) 2017 Red Hat, Inc.
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# Creator/Owner: Stefan Hajnoczi <stefanha@redhat.com>
20#
21# Non-shared storage migration test using NBD server and drive-mirror
22
23import iotests
24
25iotests.script_initialize(supported_fmts=['qcow2', 'qed', 'raw'],
26                          supported_platforms=['linux'])
27
28with iotests.FilePath('source.img') as source_img_path, \
29     iotests.FilePath('dest.img') as dest_img_path, \
30     iotests.FilePath('migration.sock', 'nbd.sock', base_dir=iotests.sock_dir) \
31        as (migration_sock_path, nbd_sock_path), \
32     iotests.VM('source') as source_vm, \
33     iotests.VM('dest') as dest_vm:
34
35    img_size = '1G'
36    iotests.qemu_img_create('-f', iotests.imgfmt, source_img_path, img_size)
37    iotests.qemu_img_create('-f', iotests.imgfmt, dest_img_path, img_size)
38
39    iotests.log('Launching VMs...')
40    (source_vm.add_drive(source_img_path)
41              .launch())
42    (dest_vm.add_drive(dest_img_path)
43            .add_incoming('unix:{0}'.format(migration_sock_path))
44            .launch())
45
46    source_vm.qmp_log('block-dirty-bitmap-add', node='drive0', name='bitmap0')
47
48    iotests.log('Launching NBD server on destination...')
49    iotests.log(dest_vm.qmp('nbd-server-start', addr={'type': 'unix', 'data': {'path': nbd_sock_path}}))
50    iotests.log(dest_vm.qmp('nbd-server-add', device='drive0', writable=True))
51
52    iotests.log('Starting `drive-mirror` on source...')
53    iotests.log(source_vm.qmp(
54                  'drive-mirror',
55                  device='drive0',
56                  target='nbd+unix:///drive0?socket={0}'.format(nbd_sock_path),
57                  sync='full',
58                  format='raw', # always raw, the server handles the format
59                  mode='existing',
60                  job_id='mirror-job0'))
61
62    iotests.log('Waiting for `drive-mirror` to complete...')
63    iotests.log(source_vm.event_wait('BLOCK_JOB_READY'),
64                filters=[iotests.filter_qmp_event])
65
66    iotests.log('Starting migration...')
67    capabilities = [{'capability': 'events', 'state': True},
68                    {'capability': 'dirty-bitmaps', 'state': True}]
69    source_vm.qmp('migrate-set-capabilities', capabilities=capabilities)
70    dest_vm.qmp('migrate-set-capabilities', capabilities=capabilities)
71    iotests.log(source_vm.qmp('migrate', uri='unix:{0}'.format(migration_sock_path)))
72
73    source_vm.qmp_log('migrate-start-postcopy')
74
75    while True:
76        event1 = source_vm.event_wait('MIGRATION')
77        if event1['data']['status'] == 'postcopy-active':
78            # This event is racy, it depends do we really do postcopy or bitmap
79            # was migrated during downtime (and no data to migrate in postcopy
80            # phase). So, don't log it.
81            continue
82        iotests.log(event1, filters=[iotests.filter_qmp_event])
83        if event1['data']['status'] in ('completed', 'failed'):
84            iotests.log('Gracefully ending the `drive-mirror` job on source...')
85            iotests.log(source_vm.qmp('block-job-cancel', device='mirror-job0'))
86            break
87
88    while True:
89        event2 = source_vm.event_wait('BLOCK_JOB_COMPLETED')
90        iotests.log(event2, filters=[iotests.filter_qmp_event])
91        if event2['event'] == 'BLOCK_JOB_COMPLETED':
92            iotests.log('Stopping the NBD server on destination...')
93            iotests.log(dest_vm.qmp('nbd-server-stop'))
94            break
95
96    iotests.log('Wait for migration completion on target...')
97    migr_events = (('MIGRATION', {'data': {'status': 'completed'}}),
98                   ('MIGRATION', {'data': {'status': 'failed'}}))
99    event = dest_vm.events_wait(migr_events)
100    iotests.log(event, filters=[iotests.filter_qmp_event])
101
102    iotests.log('Check bitmaps on source:')
103    iotests.log(source_vm.qmp('query-block')['return'][0]['inserted']['dirty-bitmaps'])
104
105    iotests.log('Check bitmaps on target:')
106    iotests.log(dest_vm.qmp('query-block')['return'][0]['inserted']['dirty-bitmaps'])
107