17c477526SPhilippe Mathieu-Daudé#!/usr/bin/env python3 29dd003a9SVladimir Sementsov-Ogievskiy# group: rw quick 3ddd113beSAndrey Shinkevich# 4ddd113beSAndrey Shinkevich# Test for qcow2 bitmap printed information 5ddd113beSAndrey Shinkevich# 6ddd113beSAndrey Shinkevich# Copyright (c) 2019 Virtuozzo International GmbH 7ddd113beSAndrey Shinkevich# 8ddd113beSAndrey Shinkevich# This program is free software; you can redistribute it and/or modify 9ddd113beSAndrey Shinkevich# it under the terms of the GNU General Public License as published by 10ddd113beSAndrey Shinkevich# the Free Software Foundation; either version 2 of the License, or 11ddd113beSAndrey Shinkevich# (at your option) any later version. 12ddd113beSAndrey Shinkevich# 13ddd113beSAndrey Shinkevich# This program is distributed in the hope that it will be useful, 14ddd113beSAndrey Shinkevich# but WITHOUT ANY WARRANTY; without even the implied warranty of 15ddd113beSAndrey Shinkevich# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 16ddd113beSAndrey Shinkevich# GNU General Public License for more details. 17ddd113beSAndrey Shinkevich# 18ddd113beSAndrey Shinkevich# You should have received a copy of the GNU General Public License 19ddd113beSAndrey Shinkevich# along with this program. If not, see <http://www.gnu.org/licenses/>. 20ddd113beSAndrey Shinkevich# 21ddd113beSAndrey Shinkevich 22ddd113beSAndrey Shinkevichimport iotests 23ddd113beSAndrey Shinkevichimport json 2442eb4591SAndrey Shinkevichimport struct 25*093a13acSJohn Snowfrom iotests import qemu_img_create, qemu_io_log, qemu_img_info, \ 26*093a13acSJohn Snow file_path, img_info_log, log 27ddd113beSAndrey Shinkevich 28eda7a9c5SMax Reitziotests.script_initialize(supported_fmts=['qcow2'], 29b30b8077SVladimir Sementsov-Ogievskiy supported_protocols=['file'], 30b30b8077SVladimir Sementsov-Ogievskiy unsupported_imgopts=['refcount_bits', 'compat']) 31ddd113beSAndrey Shinkevich 32ddd113beSAndrey Shinkevichdisk = file_path('disk') 33ddd113beSAndrey Shinkevichchunk = 256 * 1024 34ddd113beSAndrey Shinkevichbitmap_flag_unknown = 1 << 2 35ddd113beSAndrey Shinkevich# flag_offset = 5*cluster_size + flag_offset_in_bitmap_directory_entry 36ddd113beSAndrey Shinkevichflag_offset = 0x5000f 37ddd113beSAndrey Shinkevich 38ddd113beSAndrey Shinkevich 39ddd113beSAndrey Shinkevichdef print_bitmap(extra_args): 40ddd113beSAndrey Shinkevich log('qemu-img info dump:\n') 41ddd113beSAndrey Shinkevich img_info_log(disk, extra_args=extra_args) 429ebb2b76SJohn Snow result = qemu_img_info('--force-share', disk) 43ddd113beSAndrey Shinkevich if 'bitmaps' in result['format-specific']['data']: 44ddd113beSAndrey Shinkevich bitmaps = result['format-specific']['data']['bitmaps'] 45ddd113beSAndrey Shinkevich log('The same bitmaps in JSON format:') 46ddd113beSAndrey Shinkevich log(bitmaps, indent=2) 47ddd113beSAndrey Shinkevich else: 48ddd113beSAndrey Shinkevich log('No bitmap in JSON format output') 49ddd113beSAndrey Shinkevich 50ddd113beSAndrey Shinkevich 51ddd113beSAndrey Shinkevichdef add_bitmap(bitmap_number, persistent, disabled): 52ddd113beSAndrey Shinkevich granularity = 1 << (13 + bitmap_number) 53ddd113beSAndrey Shinkevich bitmap_name = 'bitmap-' + str(bitmap_number-1) 54ddd113beSAndrey Shinkevich vm = iotests.VM().add_drive(disk) 55ddd113beSAndrey Shinkevich vm.launch() 56ddd113beSAndrey Shinkevich vm.qmp_log('block-dirty-bitmap-add', node='drive0', name=bitmap_name, 57ddd113beSAndrey Shinkevich granularity=granularity, persistent=persistent, 58ddd113beSAndrey Shinkevich disabled=disabled) 59ddd113beSAndrey Shinkevich vm.shutdown() 60ddd113beSAndrey Shinkevich 61ddd113beSAndrey Shinkevich 62ddd113beSAndrey Shinkevichdef write_to_disk(offset, size): 63ddd113beSAndrey Shinkevich write = 'write {} {}'.format(offset, size) 64*093a13acSJohn Snow qemu_io_log('-c', write, disk) 65ddd113beSAndrey Shinkevich 66ddd113beSAndrey Shinkevich 67ddd113beSAndrey Shinkevichdef toggle_flag(offset): 68ddd113beSAndrey Shinkevich with open(disk, "r+b") as f: 69ddd113beSAndrey Shinkevich f.seek(offset, 0) 7042eb4591SAndrey Shinkevich # Read one byte in a way compatible with Python 2 7142eb4591SAndrey Shinkevich flags = struct.unpack("B", f.read(1)) 7242eb4591SAndrey Shinkevich toggled = flags[0] ^ bitmap_flag_unknown 73ddd113beSAndrey Shinkevich f.seek(-1, 1) 7442eb4591SAndrey Shinkevich f.write(struct.pack("B", toggled)) 75ddd113beSAndrey Shinkevich 76ddd113beSAndrey Shinkevich 77ddd113beSAndrey Shinkevichqemu_img_create('-f', iotests.imgfmt, disk, '1M') 78ddd113beSAndrey Shinkevich 79ddd113beSAndrey Shinkevichfor num in range(1, 4): 80ddd113beSAndrey Shinkevich disabled = False 81ddd113beSAndrey Shinkevich if num == 2: 82ddd113beSAndrey Shinkevich disabled = True 83ddd113beSAndrey Shinkevich log('Test {}'.format(num)) 84ddd113beSAndrey Shinkevich add_bitmap(num, num > 1, disabled) 85ddd113beSAndrey Shinkevich write_to_disk((num-1) * chunk, chunk) 86ddd113beSAndrey Shinkevich print_bitmap([]) 87ddd113beSAndrey Shinkevich log('') 88ddd113beSAndrey Shinkevich 89ddd113beSAndrey Shinkevichvm = iotests.VM().add_drive(disk) 90ddd113beSAndrey Shinkevichvm.launch() 91ddd113beSAndrey Shinkevichnum += 1 92ddd113beSAndrey Shinkevichlog('Test {}\nChecking "in-use" flag...'.format(num)) 93ddd113beSAndrey Shinkevichprint_bitmap(['--force-share']) 94ddd113beSAndrey Shinkevichvm.shutdown() 95ddd113beSAndrey Shinkevich 96ddd113beSAndrey Shinkevichnum += 1 97ddd113beSAndrey Shinkevichlog('\nTest {}'.format(num)) 98ddd113beSAndrey Shinkevichqemu_img_create('-f', iotests.imgfmt, disk, '1M') 99ddd113beSAndrey Shinkevichadd_bitmap(1, True, False) 100ddd113beSAndrey Shinkevichlog('Write an unknown bitmap flag \'{}\' into a new QCOW2 image at offset {}' 101ddd113beSAndrey Shinkevich .format(hex(bitmap_flag_unknown), flag_offset)) 102ddd113beSAndrey Shinkevichtoggle_flag(flag_offset) 1038f685ac3SJohn Snowimg_info_log(disk, check=False) 104ddd113beSAndrey Shinkevichtoggle_flag(flag_offset) 105ddd113beSAndrey Shinkevichlog('Unset the unknown bitmap flag \'{}\' in the bitmap directory entry:\n' 106ddd113beSAndrey Shinkevich .format(hex(bitmap_flag_unknown))) 107ddd113beSAndrey Shinkevichimg_info_log(disk) 108ddd113beSAndrey Shinkevichlog('Test complete') 109