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

migration.py (2728B)

      1# Migration test
      2#
      3# Copyright (c) 2019 Red Hat, Inc.
      4#
      5# Authors:
      6#  Cleber Rosa <crosa@redhat.com>
      7#  Caio Carrara <ccarrara@redhat.com>
      8#
      9# This work is licensed under the terms of the GNU GPL, version 2 or
     10# later.  See the COPYING file in the top-level directory.
     11
     12
     13import tempfile
     14from avocado_qemu import Test
     15from avocado import skipUnless
     16
     17from avocado.utils import network
     18from avocado.utils import wait
     19from avocado.utils.path import find_command
     20
     21
     22class Migration(Test):
     23    """
     24    :avocado: tags=migration
     25    """
     26
     27    timeout = 10
     28
     29    @staticmethod
     30    def migration_finished(vm):
     31        return vm.command('query-migrate')['status'] in ('completed', 'failed')
     32
     33    def assert_migration(self, src_vm, dst_vm):
     34        wait.wait_for(self.migration_finished,
     35                      timeout=self.timeout,
     36                      step=0.1,
     37                      args=(src_vm,))
     38        wait.wait_for(self.migration_finished,
     39                      timeout=self.timeout,
     40                      step=0.1,
     41                      args=(dst_vm,))
     42        self.assertEqual(src_vm.command('query-migrate')['status'], 'completed')
     43        self.assertEqual(dst_vm.command('query-migrate')['status'], 'completed')
     44        self.assertEqual(dst_vm.command('query-status')['status'], 'running')
     45        self.assertEqual(src_vm.command('query-status')['status'],'postmigrate')
     46
     47    def do_migrate(self, dest_uri, src_uri=None):
     48        dest_vm = self.get_vm('-incoming', dest_uri)
     49        dest_vm.add_args('-nodefaults')
     50        dest_vm.launch()
     51        if src_uri is None:
     52            src_uri = dest_uri
     53        source_vm = self.get_vm()
     54        source_vm.add_args('-nodefaults')
     55        source_vm.launch()
     56        source_vm.qmp('migrate', uri=src_uri)
     57        self.assert_migration(source_vm, dest_vm)
     58
     59    def _get_free_port(self):
     60        port = network.find_free_port()
     61        if port is None:
     62            self.cancel('Failed to find a free port')
     63        return port
     64
     65
     66    def test_migration_with_tcp_localhost(self):
     67        dest_uri = 'tcp:localhost:%u' % self._get_free_port()
     68        self.do_migrate(dest_uri)
     69
     70    def test_migration_with_unix(self):
     71        with tempfile.TemporaryDirectory(prefix='socket_') as socket_path:
     72            dest_uri = 'unix:%s/qemu-test.sock' % socket_path
     73            self.do_migrate(dest_uri)
     74
     75    @skipUnless(find_command('nc', default=False), "'nc' command not found")
     76    def test_migration_with_exec(self):
     77        """The test works for both netcat-traditional and netcat-openbsd packages."""
     78        free_port = self._get_free_port()
     79        dest_uri = 'exec:nc -l localhost %u' % free_port
     80        src_uri = 'exec:nc localhost %u' % free_port
     81        self.do_migrate(dest_uri, src_uri)