cachepc-qemu

Fork of AMDESE/qemu with changes for cachepc side-channel attack
git clone https://git.sinitax.com/sinitax/cachepc-qemu
Log | Files | Refs | Submodules | LICENSE | sfeed.txt

194 (4380B)

      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_pipe('create', '-f', iotests.imgfmt, source_img_path, img_size)
     37    iotests.qemu_img_pipe('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        iotests.log(event1, filters=[iotests.filter_qmp_event])
     78        if event1['data']['status'] in ('completed', 'failed'):
     79            iotests.log('Gracefully ending the `drive-mirror` job on source...')
     80            iotests.log(source_vm.qmp('block-job-cancel', device='mirror-job0'))
     81            break
     82
     83    while True:
     84        event2 = source_vm.event_wait('BLOCK_JOB_COMPLETED')
     85        iotests.log(event2, filters=[iotests.filter_qmp_event])
     86        if event2['event'] == 'BLOCK_JOB_COMPLETED':
     87            iotests.log('Stopping the NBD server on destination...')
     88            iotests.log(dest_vm.qmp('nbd-server-stop'))
     89            break
     90
     91    iotests.log('Wait for migration completion on target...')
     92    migr_events = (('MIGRATION', {'data': {'status': 'completed'}}),
     93                   ('MIGRATION', {'data': {'status': 'failed'}}))
     94    event = dest_vm.events_wait(migr_events)
     95    iotests.log(event, filters=[iotests.filter_qmp_event])
     96
     97    iotests.log('Check bitmaps on source:')
     98    iotests.log(source_vm.qmp('query-block')['return'][0]['inserted']['dirty-bitmaps'])
     99
    100    iotests.log('Check bitmaps on target:')
    101    iotests.log(dest_vm.qmp('query-block')['return'][0]['inserted']['dirty-bitmaps'])