1cb5c6cd2SMax Reitz#!/usr/bin/env python3 2*9dd003a9SVladimir Sementsov-Ogievskiy# group: migration 3cb5c6cd2SMax Reitz# 4cb5c6cd2SMax Reitz# Copyright (C) 2020 Red Hat, Inc. 5cb5c6cd2SMax Reitz# 6cb5c6cd2SMax Reitz# Tests for dirty bitmaps migration with node aliases 7cb5c6cd2SMax Reitz# 8cb5c6cd2SMax Reitz# This program is free software; you can redistribute it and/or modify 9cb5c6cd2SMax Reitz# it under the terms of the GNU General Public License as published by 10cb5c6cd2SMax Reitz# the Free Software Foundation; either version 2 of the License, or 11cb5c6cd2SMax Reitz# (at your option) any later version. 12cb5c6cd2SMax Reitz# 13cb5c6cd2SMax Reitz# This program is distributed in the hope that it will be useful, 14cb5c6cd2SMax Reitz# but WITHOUT ANY WARRANTY; without even the implied warranty of 15cb5c6cd2SMax Reitz# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 16cb5c6cd2SMax Reitz# GNU General Public License for more details. 17cb5c6cd2SMax Reitz# 18cb5c6cd2SMax Reitz# You should have received a copy of the GNU General Public License 19cb5c6cd2SMax Reitz# along with this program. If not, see <http://www.gnu.org/licenses/>. 20cb5c6cd2SMax Reitz# 21cb5c6cd2SMax Reitz 22cb5c6cd2SMax Reitzimport os 23cb5c6cd2SMax Reitzimport random 24cb5c6cd2SMax Reitzimport re 25cb5c6cd2SMax Reitzfrom typing import Dict, List, Optional, Union 26cb5c6cd2SMax Reitzimport iotests 27cb5c6cd2SMax Reitzimport qemu 28cb5c6cd2SMax Reitz 29cb5c6cd2SMax ReitzBlockBitmapMapping = List[Dict[str, Union[str, List[Dict[str, str]]]]] 30cb5c6cd2SMax Reitz 31cb5c6cd2SMax Reitzassert iotests.sock_dir is not None 32cb5c6cd2SMax Reitzmig_sock = os.path.join(iotests.sock_dir, 'mig_sock') 33cb5c6cd2SMax Reitz 34cb5c6cd2SMax Reitz 35cb5c6cd2SMax Reitzclass TestDirtyBitmapMigration(iotests.QMPTestCase): 36cb5c6cd2SMax Reitz src_node_name: str = '' 37cb5c6cd2SMax Reitz dst_node_name: str = '' 38cb5c6cd2SMax Reitz src_bmap_name: str = '' 39cb5c6cd2SMax Reitz dst_bmap_name: str = '' 40cb5c6cd2SMax Reitz 41cb5c6cd2SMax Reitz def setUp(self) -> None: 42cb5c6cd2SMax Reitz self.vm_a = iotests.VM(path_suffix='-a') 43cb5c6cd2SMax Reitz self.vm_a.add_blockdev(f'node-name={self.src_node_name},' 44cb5c6cd2SMax Reitz 'driver=null-co') 45cb5c6cd2SMax Reitz self.vm_a.launch() 46cb5c6cd2SMax Reitz 47cb5c6cd2SMax Reitz self.vm_b = iotests.VM(path_suffix='-b') 48cb5c6cd2SMax Reitz self.vm_b.add_blockdev(f'node-name={self.dst_node_name},' 49cb5c6cd2SMax Reitz 'driver=null-co') 50cb5c6cd2SMax Reitz self.vm_b.add_incoming(f'unix:{mig_sock}') 51cb5c6cd2SMax Reitz self.vm_b.launch() 52cb5c6cd2SMax Reitz 53cb5c6cd2SMax Reitz result = self.vm_a.qmp('block-dirty-bitmap-add', 54cb5c6cd2SMax Reitz node=self.src_node_name, 55cb5c6cd2SMax Reitz name=self.src_bmap_name) 56cb5c6cd2SMax Reitz self.assert_qmp(result, 'return', {}) 57cb5c6cd2SMax Reitz 58cb5c6cd2SMax Reitz # Dirty some random megabytes 59cb5c6cd2SMax Reitz for _ in range(9): 60cb5c6cd2SMax Reitz mb_ofs = random.randrange(1024) 61cb5c6cd2SMax Reitz self.vm_a.hmp_qemu_io(self.src_node_name, f'discard {mb_ofs}M 1M') 62cb5c6cd2SMax Reitz 63cb5c6cd2SMax Reitz result = self.vm_a.qmp('x-debug-block-dirty-bitmap-sha256', 64cb5c6cd2SMax Reitz node=self.src_node_name, 65cb5c6cd2SMax Reitz name=self.src_bmap_name) 66cb5c6cd2SMax Reitz self.bitmap_hash_reference = result['return']['sha256'] 67cb5c6cd2SMax Reitz 68cb5c6cd2SMax Reitz caps = [{'capability': name, 'state': True} 69cb5c6cd2SMax Reitz for name in ('dirty-bitmaps', 'events')] 70cb5c6cd2SMax Reitz 71cb5c6cd2SMax Reitz for vm in (self.vm_a, self.vm_b): 72cb5c6cd2SMax Reitz result = vm.qmp('migrate-set-capabilities', capabilities=caps) 73cb5c6cd2SMax Reitz self.assert_qmp(result, 'return', {}) 74cb5c6cd2SMax Reitz 75cb5c6cd2SMax Reitz def tearDown(self) -> None: 76cb5c6cd2SMax Reitz self.vm_a.shutdown() 77cb5c6cd2SMax Reitz self.vm_b.shutdown() 78cb5c6cd2SMax Reitz try: 79cb5c6cd2SMax Reitz os.remove(mig_sock) 80cb5c6cd2SMax Reitz except OSError: 81cb5c6cd2SMax Reitz pass 82cb5c6cd2SMax Reitz 83cb5c6cd2SMax Reitz def check_bitmap(self, bitmap_name_valid: bool) -> None: 84cb5c6cd2SMax Reitz result = self.vm_b.qmp('x-debug-block-dirty-bitmap-sha256', 85cb5c6cd2SMax Reitz node=self.dst_node_name, 86cb5c6cd2SMax Reitz name=self.dst_bmap_name) 87cb5c6cd2SMax Reitz if bitmap_name_valid: 88cb5c6cd2SMax Reitz self.assert_qmp(result, 'return/sha256', 89cb5c6cd2SMax Reitz self.bitmap_hash_reference) 90cb5c6cd2SMax Reitz else: 91cb5c6cd2SMax Reitz self.assert_qmp(result, 'error/desc', 92cb5c6cd2SMax Reitz f"Dirty bitmap '{self.dst_bmap_name}' not found") 93cb5c6cd2SMax Reitz 94cb5c6cd2SMax Reitz def migrate(self, bitmap_name_valid: bool = True, 95cb5c6cd2SMax Reitz migration_success: bool = True) -> None: 96cb5c6cd2SMax Reitz result = self.vm_a.qmp('migrate', uri=f'unix:{mig_sock}') 97cb5c6cd2SMax Reitz self.assert_qmp(result, 'return', {}) 98cb5c6cd2SMax Reitz 99cb5c6cd2SMax Reitz with iotests.Timeout(5, 'Timeout waiting for migration to complete'): 100cb5c6cd2SMax Reitz self.assertEqual(self.vm_a.wait_migration('postmigrate'), 101cb5c6cd2SMax Reitz migration_success) 102cb5c6cd2SMax Reitz self.assertEqual(self.vm_b.wait_migration('running'), 103cb5c6cd2SMax Reitz migration_success) 104cb5c6cd2SMax Reitz 105cb5c6cd2SMax Reitz if migration_success: 106cb5c6cd2SMax Reitz self.check_bitmap(bitmap_name_valid) 107cb5c6cd2SMax Reitz 108cb5c6cd2SMax Reitz def verify_dest_error(self, msg: Optional[str]) -> None: 109cb5c6cd2SMax Reitz """ 110cb5c6cd2SMax Reitz Check whether the given error message is present in vm_b's log. 111cb5c6cd2SMax Reitz (vm_b is shut down to do so.) 112cb5c6cd2SMax Reitz If @msg is None, check that there has not been any error. 113cb5c6cd2SMax Reitz """ 114cb5c6cd2SMax Reitz self.vm_b.shutdown() 115cb5c6cd2SMax Reitz if msg is None: 116cb5c6cd2SMax Reitz self.assertNotIn('qemu-system-', self.vm_b.get_log()) 117cb5c6cd2SMax Reitz else: 118cb5c6cd2SMax Reitz self.assertIn(msg, self.vm_b.get_log()) 119cb5c6cd2SMax Reitz 120cb5c6cd2SMax Reitz @staticmethod 121cb5c6cd2SMax Reitz def mapping(node_name: str, node_alias: str, 122cb5c6cd2SMax Reitz bitmap_name: str, bitmap_alias: str) -> BlockBitmapMapping: 123cb5c6cd2SMax Reitz return [{ 124cb5c6cd2SMax Reitz 'node-name': node_name, 125cb5c6cd2SMax Reitz 'alias': node_alias, 126cb5c6cd2SMax Reitz 'bitmaps': [{ 127cb5c6cd2SMax Reitz 'name': bitmap_name, 128cb5c6cd2SMax Reitz 'alias': bitmap_alias 129cb5c6cd2SMax Reitz }] 130cb5c6cd2SMax Reitz }] 131cb5c6cd2SMax Reitz 132cb5c6cd2SMax Reitz def set_mapping(self, vm: iotests.VM, mapping: BlockBitmapMapping, 133cb5c6cd2SMax Reitz error: Optional[str] = None) -> None: 134cb5c6cd2SMax Reitz """ 135cb5c6cd2SMax Reitz Invoke migrate-set-parameters on @vm to set the given @mapping. 136cb5c6cd2SMax Reitz Check for success if @error is None, or verify the error message 137cb5c6cd2SMax Reitz if it is not. 138cb5c6cd2SMax Reitz On success, verify that "info migrate_parameters" on HMP returns 139cb5c6cd2SMax Reitz our mapping. (Just to check its formatting code.) 140cb5c6cd2SMax Reitz """ 141cb5c6cd2SMax Reitz result = vm.qmp('migrate-set-parameters', 142cb5c6cd2SMax Reitz block_bitmap_mapping=mapping) 143cb5c6cd2SMax Reitz 144cb5c6cd2SMax Reitz if error is None: 145cb5c6cd2SMax Reitz self.assert_qmp(result, 'return', {}) 146cb5c6cd2SMax Reitz 147cb5c6cd2SMax Reitz result = vm.qmp('human-monitor-command', 148cb5c6cd2SMax Reitz command_line='info migrate_parameters') 149cb5c6cd2SMax Reitz 150cb5c6cd2SMax Reitz m = re.search(r'^block-bitmap-mapping:\r?(\n .*)*\n', 151cb5c6cd2SMax Reitz result['return'], flags=re.MULTILINE) 152cb5c6cd2SMax Reitz hmp_mapping = m.group(0).replace('\r', '') if m else None 153cb5c6cd2SMax Reitz 154cb5c6cd2SMax Reitz self.assertEqual(hmp_mapping, self.to_hmp_mapping(mapping)) 155cb5c6cd2SMax Reitz else: 156cb5c6cd2SMax Reitz self.assert_qmp(result, 'error/desc', error) 157cb5c6cd2SMax Reitz 158cb5c6cd2SMax Reitz @staticmethod 159cb5c6cd2SMax Reitz def to_hmp_mapping(mapping: BlockBitmapMapping) -> str: 160cb5c6cd2SMax Reitz result = 'block-bitmap-mapping:\n' 161cb5c6cd2SMax Reitz 162cb5c6cd2SMax Reitz for node in mapping: 163cb5c6cd2SMax Reitz result += f" '{node['node-name']}' -> '{node['alias']}'\n" 164cb5c6cd2SMax Reitz 165cb5c6cd2SMax Reitz assert isinstance(node['bitmaps'], list) 166cb5c6cd2SMax Reitz for bitmap in node['bitmaps']: 167cb5c6cd2SMax Reitz result += f" '{bitmap['name']}' -> '{bitmap['alias']}'\n" 168cb5c6cd2SMax Reitz 169cb5c6cd2SMax Reitz return result 170cb5c6cd2SMax Reitz 171cb5c6cd2SMax Reitz 172cb5c6cd2SMax Reitzclass TestAliasMigration(TestDirtyBitmapMigration): 173cb5c6cd2SMax Reitz src_node_name = 'node0' 174cb5c6cd2SMax Reitz dst_node_name = 'node0' 175cb5c6cd2SMax Reitz src_bmap_name = 'bmap0' 176cb5c6cd2SMax Reitz dst_bmap_name = 'bmap0' 177cb5c6cd2SMax Reitz 178cb5c6cd2SMax Reitz def test_migration_without_alias(self) -> None: 179cb5c6cd2SMax Reitz self.migrate(self.src_node_name == self.dst_node_name and 180cb5c6cd2SMax Reitz self.src_bmap_name == self.dst_bmap_name) 181cb5c6cd2SMax Reitz 182cb5c6cd2SMax Reitz # Check for error message on the destination 183cb5c6cd2SMax Reitz if self.src_node_name != self.dst_node_name: 184cb5c6cd2SMax Reitz self.verify_dest_error(f"Cannot find " 185cb5c6cd2SMax Reitz f"device={self.src_node_name} nor " 186cb5c6cd2SMax Reitz f"node_name={self.src_node_name}") 187cb5c6cd2SMax Reitz else: 188cb5c6cd2SMax Reitz self.verify_dest_error(None) 189cb5c6cd2SMax Reitz 190cb5c6cd2SMax Reitz def test_alias_on_src_migration(self) -> None: 191cb5c6cd2SMax Reitz mapping = self.mapping(self.src_node_name, self.dst_node_name, 192cb5c6cd2SMax Reitz self.src_bmap_name, self.dst_bmap_name) 193cb5c6cd2SMax Reitz 194cb5c6cd2SMax Reitz self.set_mapping(self.vm_a, mapping) 195cb5c6cd2SMax Reitz self.migrate() 196cb5c6cd2SMax Reitz self.verify_dest_error(None) 197cb5c6cd2SMax Reitz 198cb5c6cd2SMax Reitz def test_alias_on_dst_migration(self) -> None: 199cb5c6cd2SMax Reitz mapping = self.mapping(self.dst_node_name, self.src_node_name, 200cb5c6cd2SMax Reitz self.dst_bmap_name, self.src_bmap_name) 201cb5c6cd2SMax Reitz 202cb5c6cd2SMax Reitz self.set_mapping(self.vm_b, mapping) 203cb5c6cd2SMax Reitz self.migrate() 204cb5c6cd2SMax Reitz self.verify_dest_error(None) 205cb5c6cd2SMax Reitz 206cb5c6cd2SMax Reitz def test_alias_on_both_migration(self) -> None: 207cb5c6cd2SMax Reitz src_map = self.mapping(self.src_node_name, 'node-alias', 208cb5c6cd2SMax Reitz self.src_bmap_name, 'bmap-alias') 209cb5c6cd2SMax Reitz 210cb5c6cd2SMax Reitz dst_map = self.mapping(self.dst_node_name, 'node-alias', 211cb5c6cd2SMax Reitz self.dst_bmap_name, 'bmap-alias') 212cb5c6cd2SMax Reitz 213cb5c6cd2SMax Reitz self.set_mapping(self.vm_a, src_map) 214cb5c6cd2SMax Reitz self.set_mapping(self.vm_b, dst_map) 215cb5c6cd2SMax Reitz self.migrate() 216cb5c6cd2SMax Reitz self.verify_dest_error(None) 217cb5c6cd2SMax Reitz 218cb5c6cd2SMax Reitz 219cb5c6cd2SMax Reitzclass TestNodeAliasMigration(TestAliasMigration): 220cb5c6cd2SMax Reitz src_node_name = 'node-src' 221cb5c6cd2SMax Reitz dst_node_name = 'node-dst' 222cb5c6cd2SMax Reitz 223cb5c6cd2SMax Reitz 224cb5c6cd2SMax Reitzclass TestBitmapAliasMigration(TestAliasMigration): 225cb5c6cd2SMax Reitz src_bmap_name = 'bmap-src' 226cb5c6cd2SMax Reitz dst_bmap_name = 'bmap-dst' 227cb5c6cd2SMax Reitz 228cb5c6cd2SMax Reitz 229cb5c6cd2SMax Reitzclass TestFullAliasMigration(TestAliasMigration): 230cb5c6cd2SMax Reitz src_node_name = 'node-src' 231cb5c6cd2SMax Reitz dst_node_name = 'node-dst' 232cb5c6cd2SMax Reitz src_bmap_name = 'bmap-src' 233cb5c6cd2SMax Reitz dst_bmap_name = 'bmap-dst' 234cb5c6cd2SMax Reitz 235cb5c6cd2SMax Reitz 236cb5c6cd2SMax Reitzclass TestLongBitmapNames(TestAliasMigration): 237cb5c6cd2SMax Reitz # Giving long bitmap names is OK, as long as there is a short alias for 238cb5c6cd2SMax Reitz # migration 239cb5c6cd2SMax Reitz src_bmap_name = 'a' * 512 240cb5c6cd2SMax Reitz dst_bmap_name = 'b' * 512 241cb5c6cd2SMax Reitz 242cb5c6cd2SMax Reitz # Skip all tests that do not use the intermediate alias 243cb5c6cd2SMax Reitz def test_migration_without_alias(self) -> None: 244cb5c6cd2SMax Reitz pass 245cb5c6cd2SMax Reitz 246cb5c6cd2SMax Reitz def test_alias_on_src_migration(self) -> None: 247cb5c6cd2SMax Reitz pass 248cb5c6cd2SMax Reitz 249cb5c6cd2SMax Reitz def test_alias_on_dst_migration(self) -> None: 250cb5c6cd2SMax Reitz pass 251cb5c6cd2SMax Reitz 252cb5c6cd2SMax Reitz 253cb5c6cd2SMax Reitzclass TestBlockBitmapMappingErrors(TestDirtyBitmapMigration): 254cb5c6cd2SMax Reitz src_node_name = 'node0' 255cb5c6cd2SMax Reitz dst_node_name = 'node0' 256cb5c6cd2SMax Reitz src_bmap_name = 'bmap0' 257cb5c6cd2SMax Reitz dst_bmap_name = 'bmap0' 258cb5c6cd2SMax Reitz 259cb5c6cd2SMax Reitz """ 260cb5c6cd2SMax Reitz Note that mapping nodes or bitmaps that do not exist is not an error. 261cb5c6cd2SMax Reitz """ 262cb5c6cd2SMax Reitz 263cb5c6cd2SMax Reitz def test_non_injective_node_mapping(self) -> None: 264cb5c6cd2SMax Reitz mapping: BlockBitmapMapping = [ 265cb5c6cd2SMax Reitz { 266cb5c6cd2SMax Reitz 'node-name': 'node0', 267cb5c6cd2SMax Reitz 'alias': 'common-alias', 268cb5c6cd2SMax Reitz 'bitmaps': [{ 269cb5c6cd2SMax Reitz 'name': 'bmap0', 270cb5c6cd2SMax Reitz 'alias': 'bmap-alias0' 271cb5c6cd2SMax Reitz }] 272cb5c6cd2SMax Reitz }, 273cb5c6cd2SMax Reitz { 274cb5c6cd2SMax Reitz 'node-name': 'node1', 275cb5c6cd2SMax Reitz 'alias': 'common-alias', 276cb5c6cd2SMax Reitz 'bitmaps': [{ 277cb5c6cd2SMax Reitz 'name': 'bmap1', 278cb5c6cd2SMax Reitz 'alias': 'bmap-alias1' 279cb5c6cd2SMax Reitz }] 280cb5c6cd2SMax Reitz } 281cb5c6cd2SMax Reitz ] 282cb5c6cd2SMax Reitz 283cb5c6cd2SMax Reitz self.set_mapping(self.vm_a, mapping, 284cb5c6cd2SMax Reitz "Invalid mapping given for block-bitmap-mapping: " 285cb5c6cd2SMax Reitz "The node alias 'common-alias' is used twice") 286cb5c6cd2SMax Reitz 287cb5c6cd2SMax Reitz def test_non_injective_bitmap_mapping(self) -> None: 288cb5c6cd2SMax Reitz mapping: BlockBitmapMapping = [{ 289cb5c6cd2SMax Reitz 'node-name': 'node0', 290cb5c6cd2SMax Reitz 'alias': 'node-alias0', 291cb5c6cd2SMax Reitz 'bitmaps': [ 292cb5c6cd2SMax Reitz { 293cb5c6cd2SMax Reitz 'name': 'bmap0', 294cb5c6cd2SMax Reitz 'alias': 'common-alias' 295cb5c6cd2SMax Reitz }, 296cb5c6cd2SMax Reitz { 297cb5c6cd2SMax Reitz 'name': 'bmap1', 298cb5c6cd2SMax Reitz 'alias': 'common-alias' 299cb5c6cd2SMax Reitz } 300cb5c6cd2SMax Reitz ] 301cb5c6cd2SMax Reitz }] 302cb5c6cd2SMax Reitz 303cb5c6cd2SMax Reitz self.set_mapping(self.vm_a, mapping, 304cb5c6cd2SMax Reitz "Invalid mapping given for block-bitmap-mapping: " 305cb5c6cd2SMax Reitz "The bitmap alias 'node-alias0'/'common-alias' is " 306cb5c6cd2SMax Reitz "used twice") 307cb5c6cd2SMax Reitz 308cb5c6cd2SMax Reitz def test_ambiguous_node_mapping(self) -> None: 309cb5c6cd2SMax Reitz mapping: BlockBitmapMapping = [ 310cb5c6cd2SMax Reitz { 311cb5c6cd2SMax Reitz 'node-name': 'node0', 312cb5c6cd2SMax Reitz 'alias': 'node-alias0', 313cb5c6cd2SMax Reitz 'bitmaps': [{ 314cb5c6cd2SMax Reitz 'name': 'bmap0', 315cb5c6cd2SMax Reitz 'alias': 'bmap-alias0' 316cb5c6cd2SMax Reitz }] 317cb5c6cd2SMax Reitz }, 318cb5c6cd2SMax Reitz { 319cb5c6cd2SMax Reitz 'node-name': 'node0', 320cb5c6cd2SMax Reitz 'alias': 'node-alias1', 321cb5c6cd2SMax Reitz 'bitmaps': [{ 322cb5c6cd2SMax Reitz 'name': 'bmap0', 323cb5c6cd2SMax Reitz 'alias': 'bmap-alias0' 324cb5c6cd2SMax Reitz }] 325cb5c6cd2SMax Reitz } 326cb5c6cd2SMax Reitz ] 327cb5c6cd2SMax Reitz 328cb5c6cd2SMax Reitz self.set_mapping(self.vm_a, mapping, 329cb5c6cd2SMax Reitz "Invalid mapping given for block-bitmap-mapping: " 330cb5c6cd2SMax Reitz "The node name 'node0' is mapped twice") 331cb5c6cd2SMax Reitz 332cb5c6cd2SMax Reitz def test_ambiguous_bitmap_mapping(self) -> None: 333cb5c6cd2SMax Reitz mapping: BlockBitmapMapping = [{ 334cb5c6cd2SMax Reitz 'node-name': 'node0', 335cb5c6cd2SMax Reitz 'alias': 'node-alias0', 336cb5c6cd2SMax Reitz 'bitmaps': [ 337cb5c6cd2SMax Reitz { 338cb5c6cd2SMax Reitz 'name': 'bmap0', 339cb5c6cd2SMax Reitz 'alias': 'bmap-alias0' 340cb5c6cd2SMax Reitz }, 341cb5c6cd2SMax Reitz { 342cb5c6cd2SMax Reitz 'name': 'bmap0', 343cb5c6cd2SMax Reitz 'alias': 'bmap-alias1' 344cb5c6cd2SMax Reitz } 345cb5c6cd2SMax Reitz ] 346cb5c6cd2SMax Reitz }] 347cb5c6cd2SMax Reitz 348cb5c6cd2SMax Reitz self.set_mapping(self.vm_a, mapping, 349cb5c6cd2SMax Reitz "Invalid mapping given for block-bitmap-mapping: " 350cb5c6cd2SMax Reitz "The bitmap 'node0'/'bmap0' is mapped twice") 351cb5c6cd2SMax Reitz 352cb5c6cd2SMax Reitz def test_migratee_node_is_not_mapped_on_src(self) -> None: 353cb5c6cd2SMax Reitz self.set_mapping(self.vm_a, []) 354cb5c6cd2SMax Reitz # Should just ignore all bitmaps on unmapped nodes 355cb5c6cd2SMax Reitz self.migrate(False) 356cb5c6cd2SMax Reitz self.verify_dest_error(None) 357cb5c6cd2SMax Reitz 358cb5c6cd2SMax Reitz def test_migratee_node_is_not_mapped_on_dst(self) -> None: 359cb5c6cd2SMax Reitz self.set_mapping(self.vm_b, []) 360cb5c6cd2SMax Reitz self.migrate(False) 361cb5c6cd2SMax Reitz self.verify_dest_error(f"Unknown node alias '{self.src_node_name}'") 362cb5c6cd2SMax Reitz 363cb5c6cd2SMax Reitz def test_migratee_bitmap_is_not_mapped_on_src(self) -> None: 364cb5c6cd2SMax Reitz mapping: BlockBitmapMapping = [{ 365cb5c6cd2SMax Reitz 'node-name': self.src_node_name, 366cb5c6cd2SMax Reitz 'alias': self.dst_node_name, 367cb5c6cd2SMax Reitz 'bitmaps': [] 368cb5c6cd2SMax Reitz }] 369cb5c6cd2SMax Reitz 370cb5c6cd2SMax Reitz self.set_mapping(self.vm_a, mapping) 371cb5c6cd2SMax Reitz # Should just ignore all unmapped bitmaps 372cb5c6cd2SMax Reitz self.migrate(False) 373cb5c6cd2SMax Reitz self.verify_dest_error(None) 374cb5c6cd2SMax Reitz 375cb5c6cd2SMax Reitz def test_migratee_bitmap_is_not_mapped_on_dst(self) -> None: 376cb5c6cd2SMax Reitz mapping: BlockBitmapMapping = [{ 377cb5c6cd2SMax Reitz 'node-name': self.dst_node_name, 378cb5c6cd2SMax Reitz 'alias': self.src_node_name, 379cb5c6cd2SMax Reitz 'bitmaps': [] 380cb5c6cd2SMax Reitz }] 381cb5c6cd2SMax Reitz 382cb5c6cd2SMax Reitz self.set_mapping(self.vm_b, mapping) 383cb5c6cd2SMax Reitz self.migrate(False) 384cb5c6cd2SMax Reitz self.verify_dest_error(f"Unknown bitmap alias " 385cb5c6cd2SMax Reitz f"'{self.src_bmap_name}' " 386cb5c6cd2SMax Reitz f"on node '{self.dst_node_name}' " 387cb5c6cd2SMax Reitz f"(alias '{self.src_node_name}')") 388cb5c6cd2SMax Reitz 389cb5c6cd2SMax Reitz def test_unused_mapping_on_dst(self) -> None: 390cb5c6cd2SMax Reitz # Let the source not send any bitmaps 391cb5c6cd2SMax Reitz self.set_mapping(self.vm_a, []) 392cb5c6cd2SMax Reitz 393cb5c6cd2SMax Reitz # Establish some mapping on the destination 394cb5c6cd2SMax Reitz self.set_mapping(self.vm_b, []) 395cb5c6cd2SMax Reitz 396cb5c6cd2SMax Reitz # The fact that there is a mapping on B without any bitmaps 397cb5c6cd2SMax Reitz # being received should be fine, not fatal 398cb5c6cd2SMax Reitz self.migrate(False) 399cb5c6cd2SMax Reitz self.verify_dest_error(None) 400cb5c6cd2SMax Reitz 401cb5c6cd2SMax Reitz def test_non_wellformed_node_alias(self) -> None: 402cb5c6cd2SMax Reitz alias = '123-foo' 403cb5c6cd2SMax Reitz 404cb5c6cd2SMax Reitz mapping: BlockBitmapMapping = [{ 405cb5c6cd2SMax Reitz 'node-name': self.src_node_name, 406cb5c6cd2SMax Reitz 'alias': alias, 407cb5c6cd2SMax Reitz 'bitmaps': [] 408cb5c6cd2SMax Reitz }] 409cb5c6cd2SMax Reitz 410cb5c6cd2SMax Reitz self.set_mapping(self.vm_a, mapping, 411cb5c6cd2SMax Reitz f"Invalid mapping given for block-bitmap-mapping: " 412cb5c6cd2SMax Reitz f"The node alias '{alias}' is not well-formed") 413cb5c6cd2SMax Reitz 414cb5c6cd2SMax Reitz def test_node_alias_too_long(self) -> None: 415cb5c6cd2SMax Reitz alias = 'a' * 256 416cb5c6cd2SMax Reitz 417cb5c6cd2SMax Reitz mapping: BlockBitmapMapping = [{ 418cb5c6cd2SMax Reitz 'node-name': self.src_node_name, 419cb5c6cd2SMax Reitz 'alias': alias, 420cb5c6cd2SMax Reitz 'bitmaps': [] 421cb5c6cd2SMax Reitz }] 422cb5c6cd2SMax Reitz 423cb5c6cd2SMax Reitz self.set_mapping(self.vm_a, mapping, 424cb5c6cd2SMax Reitz f"Invalid mapping given for block-bitmap-mapping: " 425cb5c6cd2SMax Reitz f"The node alias '{alias}' is longer than 255 bytes") 426cb5c6cd2SMax Reitz 427cb5c6cd2SMax Reitz def test_bitmap_alias_too_long(self) -> None: 428cb5c6cd2SMax Reitz alias = 'a' * 256 429cb5c6cd2SMax Reitz 430cb5c6cd2SMax Reitz mapping = self.mapping(self.src_node_name, self.dst_node_name, 431cb5c6cd2SMax Reitz self.src_bmap_name, alias) 432cb5c6cd2SMax Reitz 433cb5c6cd2SMax Reitz self.set_mapping(self.vm_a, mapping, 434cb5c6cd2SMax Reitz f"Invalid mapping given for block-bitmap-mapping: " 435cb5c6cd2SMax Reitz f"The bitmap alias '{alias}' is longer than 255 " 436cb5c6cd2SMax Reitz f"bytes") 437cb5c6cd2SMax Reitz 438cb5c6cd2SMax Reitz def test_bitmap_name_too_long(self) -> None: 439cb5c6cd2SMax Reitz name = 'a' * 256 440cb5c6cd2SMax Reitz 441cb5c6cd2SMax Reitz result = self.vm_a.qmp('block-dirty-bitmap-add', 442cb5c6cd2SMax Reitz node=self.src_node_name, 443cb5c6cd2SMax Reitz name=name) 444cb5c6cd2SMax Reitz self.assert_qmp(result, 'return', {}) 445cb5c6cd2SMax Reitz 446cb5c6cd2SMax Reitz self.migrate(False, False) 447cb5c6cd2SMax Reitz 448cb5c6cd2SMax Reitz # Check for the error in the source's log 449cb5c6cd2SMax Reitz self.vm_a.shutdown() 450cb5c6cd2SMax Reitz self.assertIn(f"Cannot migrate bitmap '{name}' on node " 451cb5c6cd2SMax Reitz f"'{self.src_node_name}': Name is longer than 255 bytes", 452cb5c6cd2SMax Reitz self.vm_a.get_log()) 453cb5c6cd2SMax Reitz 454cb5c6cd2SMax Reitz # Expect abnormal shutdown of the destination VM because of 455cb5c6cd2SMax Reitz # the failed migration 456cb5c6cd2SMax Reitz try: 457cb5c6cd2SMax Reitz self.vm_b.shutdown() 458cb5c6cd2SMax Reitz except qemu.machine.AbnormalShutdown: 459cb5c6cd2SMax Reitz pass 460cb5c6cd2SMax Reitz 461cb5c6cd2SMax Reitz def test_aliased_bitmap_name_too_long(self) -> None: 462cb5c6cd2SMax Reitz # Longer than the maximum for bitmap names 463cb5c6cd2SMax Reitz self.dst_bmap_name = 'a' * 1024 464cb5c6cd2SMax Reitz 465cb5c6cd2SMax Reitz mapping = self.mapping(self.dst_node_name, self.src_node_name, 466cb5c6cd2SMax Reitz self.dst_bmap_name, self.src_bmap_name) 467cb5c6cd2SMax Reitz 468cb5c6cd2SMax Reitz # We would have to create this bitmap during migration, and 469cb5c6cd2SMax Reitz # that would fail, because the name is too long. Better to 470cb5c6cd2SMax Reitz # catch it early. 471cb5c6cd2SMax Reitz self.set_mapping(self.vm_b, mapping, 472cb5c6cd2SMax Reitz f"Invalid mapping given for block-bitmap-mapping: " 473cb5c6cd2SMax Reitz f"The bitmap name '{self.dst_bmap_name}' is longer " 474cb5c6cd2SMax Reitz f"than 1023 bytes") 475cb5c6cd2SMax Reitz 476cb5c6cd2SMax Reitz def test_node_name_too_long(self) -> None: 477cb5c6cd2SMax Reitz # Longer than the maximum for node names 478cb5c6cd2SMax Reitz self.dst_node_name = 'a' * 32 479cb5c6cd2SMax Reitz 480cb5c6cd2SMax Reitz mapping = self.mapping(self.dst_node_name, self.src_node_name, 481cb5c6cd2SMax Reitz self.dst_bmap_name, self.src_bmap_name) 482cb5c6cd2SMax Reitz 483cb5c6cd2SMax Reitz # During migration, this would appear simply as a node that 484cb5c6cd2SMax Reitz # cannot be found. Still better to catch impossible node 485cb5c6cd2SMax Reitz # names early (similar to test_non_wellformed_node_alias). 486cb5c6cd2SMax Reitz self.set_mapping(self.vm_b, mapping, 487cb5c6cd2SMax Reitz f"Invalid mapping given for block-bitmap-mapping: " 488cb5c6cd2SMax Reitz f"The node name '{self.dst_node_name}' is longer " 489cb5c6cd2SMax Reitz f"than 31 bytes") 490cb5c6cd2SMax Reitz 491cb5c6cd2SMax Reitz 492cb5c6cd2SMax Reitzclass TestCrossAliasMigration(TestDirtyBitmapMigration): 493cb5c6cd2SMax Reitz """ 494cb5c6cd2SMax Reitz Swap aliases, both to see that qemu does not get confused, and 495cb5c6cd2SMax Reitz that we can migrate multiple things at once. 496cb5c6cd2SMax Reitz 497cb5c6cd2SMax Reitz So we migrate this: 498cb5c6cd2SMax Reitz node-a.bmap-a -> node-b.bmap-b 499cb5c6cd2SMax Reitz node-a.bmap-b -> node-b.bmap-a 500cb5c6cd2SMax Reitz node-b.bmap-a -> node-a.bmap-b 501cb5c6cd2SMax Reitz node-b.bmap-b -> node-a.bmap-a 502cb5c6cd2SMax Reitz """ 503cb5c6cd2SMax Reitz 504cb5c6cd2SMax Reitz src_node_name = 'node-a' 505cb5c6cd2SMax Reitz dst_node_name = 'node-b' 506cb5c6cd2SMax Reitz src_bmap_name = 'bmap-a' 507cb5c6cd2SMax Reitz dst_bmap_name = 'bmap-b' 508cb5c6cd2SMax Reitz 509cb5c6cd2SMax Reitz def setUp(self) -> None: 510cb5c6cd2SMax Reitz TestDirtyBitmapMigration.setUp(self) 511cb5c6cd2SMax Reitz 512cb5c6cd2SMax Reitz # Now create another block device and let both have two bitmaps each 513cb5c6cd2SMax Reitz result = self.vm_a.qmp('blockdev-add', 514cb5c6cd2SMax Reitz node_name='node-b', driver='null-co') 515cb5c6cd2SMax Reitz self.assert_qmp(result, 'return', {}) 516cb5c6cd2SMax Reitz 517cb5c6cd2SMax Reitz result = self.vm_b.qmp('blockdev-add', 518cb5c6cd2SMax Reitz node_name='node-a', driver='null-co') 519cb5c6cd2SMax Reitz self.assert_qmp(result, 'return', {}) 520cb5c6cd2SMax Reitz 521cb5c6cd2SMax Reitz bmaps_to_add = (('node-a', 'bmap-b'), 522cb5c6cd2SMax Reitz ('node-b', 'bmap-a'), 523cb5c6cd2SMax Reitz ('node-b', 'bmap-b')) 524cb5c6cd2SMax Reitz 525cb5c6cd2SMax Reitz for (node, bmap) in bmaps_to_add: 526cb5c6cd2SMax Reitz result = self.vm_a.qmp('block-dirty-bitmap-add', 527cb5c6cd2SMax Reitz node=node, name=bmap) 528cb5c6cd2SMax Reitz self.assert_qmp(result, 'return', {}) 529cb5c6cd2SMax Reitz 530cb5c6cd2SMax Reitz @staticmethod 531cb5c6cd2SMax Reitz def cross_mapping() -> BlockBitmapMapping: 532cb5c6cd2SMax Reitz return [ 533cb5c6cd2SMax Reitz { 534cb5c6cd2SMax Reitz 'node-name': 'node-a', 535cb5c6cd2SMax Reitz 'alias': 'node-b', 536cb5c6cd2SMax Reitz 'bitmaps': [ 537cb5c6cd2SMax Reitz { 538cb5c6cd2SMax Reitz 'name': 'bmap-a', 539cb5c6cd2SMax Reitz 'alias': 'bmap-b' 540cb5c6cd2SMax Reitz }, 541cb5c6cd2SMax Reitz { 542cb5c6cd2SMax Reitz 'name': 'bmap-b', 543cb5c6cd2SMax Reitz 'alias': 'bmap-a' 544cb5c6cd2SMax Reitz } 545cb5c6cd2SMax Reitz ] 546cb5c6cd2SMax Reitz }, 547cb5c6cd2SMax Reitz { 548cb5c6cd2SMax Reitz 'node-name': 'node-b', 549cb5c6cd2SMax Reitz 'alias': 'node-a', 550cb5c6cd2SMax Reitz 'bitmaps': [ 551cb5c6cd2SMax Reitz { 552cb5c6cd2SMax Reitz 'name': 'bmap-b', 553cb5c6cd2SMax Reitz 'alias': 'bmap-a' 554cb5c6cd2SMax Reitz }, 555cb5c6cd2SMax Reitz { 556cb5c6cd2SMax Reitz 'name': 'bmap-a', 557cb5c6cd2SMax Reitz 'alias': 'bmap-b' 558cb5c6cd2SMax Reitz } 559cb5c6cd2SMax Reitz ] 560cb5c6cd2SMax Reitz } 561cb5c6cd2SMax Reitz ] 562cb5c6cd2SMax Reitz 563cb5c6cd2SMax Reitz def verify_dest_has_all_bitmaps(self) -> None: 564cb5c6cd2SMax Reitz bitmaps = self.vm_b.query_bitmaps() 565cb5c6cd2SMax Reitz 566cb5c6cd2SMax Reitz # Extract and sort bitmap names 567cb5c6cd2SMax Reitz for node in bitmaps: 568cb5c6cd2SMax Reitz bitmaps[node] = sorted((bmap['name'] for bmap in bitmaps[node])) 569cb5c6cd2SMax Reitz 570cb5c6cd2SMax Reitz self.assertEqual(bitmaps, 571cb5c6cd2SMax Reitz {'node-a': ['bmap-a', 'bmap-b'], 572cb5c6cd2SMax Reitz 'node-b': ['bmap-a', 'bmap-b']}) 573cb5c6cd2SMax Reitz 574cb5c6cd2SMax Reitz def test_alias_on_src(self) -> None: 575cb5c6cd2SMax Reitz self.set_mapping(self.vm_a, self.cross_mapping()) 576cb5c6cd2SMax Reitz 577cb5c6cd2SMax Reitz # Checks that node-a.bmap-a was migrated to node-b.bmap-b, and 578cb5c6cd2SMax Reitz # that is enough 579cb5c6cd2SMax Reitz self.migrate() 580cb5c6cd2SMax Reitz self.verify_dest_has_all_bitmaps() 581cb5c6cd2SMax Reitz self.verify_dest_error(None) 582cb5c6cd2SMax Reitz 583cb5c6cd2SMax Reitz def test_alias_on_dst(self) -> None: 584cb5c6cd2SMax Reitz self.set_mapping(self.vm_b, self.cross_mapping()) 585cb5c6cd2SMax Reitz 586cb5c6cd2SMax Reitz # Checks that node-a.bmap-a was migrated to node-b.bmap-b, and 587cb5c6cd2SMax Reitz # that is enough 588cb5c6cd2SMax Reitz self.migrate() 589cb5c6cd2SMax Reitz self.verify_dest_has_all_bitmaps() 590cb5c6cd2SMax Reitz self.verify_dest_error(None) 591cb5c6cd2SMax Reitz 592cb5c6cd2SMax Reitz 593cb5c6cd2SMax Reitzif __name__ == '__main__': 594cb5c6cd2SMax Reitz iotests.main(supported_protocols=['file']) 595