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