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

196 (2250B)

      1#!/usr/bin/env python3
      2# group: rw quick migration
      3#
      4# Test clearing unknown autoclear_features flag by qcow2 after
      5# migration. This test mimics migration to older qemu.
      6#
      7# Copyright (c) 2017 Parallels International GmbH
      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
     23import os
     24import iotests
     25from iotests import qemu_img
     26
     27disk = os.path.join(iotests.test_dir, 'disk')
     28migfile = os.path.join(iotests.test_dir, 'migfile')
     29
     30class TestInvalidateAutoclear(iotests.QMPTestCase):
     31
     32    def tearDown(self):
     33        self.vm_a.shutdown()
     34        self.vm_b.shutdown()
     35        os.remove(disk)
     36        os.remove(migfile)
     37
     38    def setUp(self):
     39        qemu_img('create', '-f', iotests.imgfmt, disk, '1M')
     40
     41        self.vm_a = iotests.VM(path_suffix='a').add_drive(disk)
     42        self.vm_a.launch()
     43
     44        self.vm_b = iotests.VM(path_suffix='b').add_drive(disk)
     45        self.vm_b.add_incoming("exec: cat '" + migfile + "'")
     46
     47    def test_migration(self):
     48        result = self.vm_a.qmp('migrate', uri='exec:cat>' + migfile)
     49        self.assert_qmp(result, 'return', {});
     50        self.assertNotEqual(self.vm_a.event_wait("STOP"), None)
     51
     52        with open(disk, 'r+b') as f:
     53            f.seek(88) # first byte of autoclear_features field
     54            f.write(b'\xff')
     55
     56        self.vm_b.launch()
     57        while True:
     58            result = self.vm_b.qmp('query-status')
     59            if result['return']['status'] == 'running':
     60                break
     61
     62        with open(disk, 'rb') as f:
     63            f.seek(88)
     64            self.assertEqual(f.read(1), b'\x00')
     65
     66if __name__ == '__main__':
     67    iotests.main(supported_fmts=['qcow2'],
     68                 supported_protocols=['file'])