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_io('-f', iotests.imgfmt, '-c', 'write 512M 1M', source_img_path) 38 iotests.qemu_img_create('-f', iotests.imgfmt, dest_img_path, img_size) 39 40 iotests.log('Launching VMs...') 41 (source_vm.add_drive(source_img_path) 42 .launch()) 43 (dest_vm.add_drive(dest_img_path) 44 .add_incoming('unix:{0}'.format(migration_sock_path)) 45 .launch()) 46 47 source_vm.qmp_log('block-dirty-bitmap-add', node='drive0', name='bitmap0') 48 49 iotests.log('Launching NBD server on destination...') 50 iotests.log(dest_vm.qmp('nbd-server-start', addr={'type': 'unix', 'data': {'path': nbd_sock_path}})) 51 iotests.log(dest_vm.qmp('nbd-server-add', device='drive0', writable=True)) 52 53 iotests.log('Starting `drive-mirror` on source...') 54 iotests.log(source_vm.qmp( 55 'drive-mirror', 56 device='drive0', 57 target='nbd+unix:///drive0?socket={0}'.format(nbd_sock_path), 58 sync='full', 59 format='raw', # always raw, the server handles the format 60 mode='existing', 61 job_id='mirror-job0')) 62 63 iotests.log('Waiting for `drive-mirror` to complete...') 64 iotests.log(source_vm.event_wait('BLOCK_JOB_READY'), 65 filters=[iotests.filter_qmp_event, 66 iotests.filter_block_job]) 67 68 iotests.log('Starting migration...') 69 capabilities = [{'capability': 'events', 'state': True}, 70 {'capability': 'dirty-bitmaps', 'state': True}] 71 source_vm.qmp('migrate-set-capabilities', capabilities=capabilities) 72 dest_vm.qmp('migrate-set-capabilities', capabilities=capabilities) 73 iotests.log(source_vm.qmp('migrate', uri='unix:{0}'.format(migration_sock_path))) 74 75 source_vm.qmp_log('migrate-start-postcopy') 76 77 while True: 78 event1 = source_vm.event_wait('MIGRATION') 79 if event1['data']['status'] == 'postcopy-active': 80 # This event is racy, it depends do we really do postcopy or bitmap 81 # was migrated during downtime (and no data to migrate in postcopy 82 # phase). So, don't log it. 83 continue 84 iotests.log(event1, filters=[iotests.filter_qmp_event]) 85 if event1['data']['status'] in ('completed', 'failed'): 86 iotests.log('Gracefully ending the `drive-mirror` job on source...') 87 iotests.log(source_vm.qmp('block-job-cancel', device='mirror-job0')) 88 break 89 90 while True: 91 event2 = source_vm.event_wait('BLOCK_JOB_COMPLETED') 92 iotests.log(event2, filters=[iotests.filter_qmp_event, 93 iotests.filter_block_job]) 94 if event2['event'] == 'BLOCK_JOB_COMPLETED': 95 iotests.log('Stopping the NBD server on destination...') 96 iotests.log(dest_vm.qmp('nbd-server-stop')) 97 break 98 99 iotests.log('Wait for migration completion on target...') 100 migr_events = (('MIGRATION', {'data': {'status': 'completed'}}), 101 ('MIGRATION', {'data': {'status': 'failed'}})) 102 event = dest_vm.events_wait(migr_events) 103 iotests.log(event, filters=[iotests.filter_qmp_event]) 104 105 iotests.log('Check bitmaps on source:') 106 iotests.log(source_vm.qmp('query-block')['return'][0]['inserted']['dirty-bitmaps']) 107 108 iotests.log('Check bitmaps on target:') 109 iotests.log(dest_vm.qmp('query-block')['return'][0]['inserted']['dirty-bitmaps']) 110