1#!/usr/bin/env python3 2# 3# Copyright (C) 2019 Red Hat, Inc. 4# 5# This program is free software; you can redistribute it and/or modify 6# it under the terms of the GNU General Public License as published by 7# the Free Software Foundation; either version 2 of the License, or 8# (at your option) any later version. 9# 10# This program is distributed in the hope that it will be useful, 11# but WITHOUT ANY WARRANTY; without even the implied warranty of 12# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 13# GNU General Public License for more details. 14# 15# You should have received a copy of the GNU General Public License 16# along with this program. If not, see <http://www.gnu.org/licenses/>. 17# 18# Creator/Owner: Kevin Wolf <kwolf@redhat.com> 19# 20# Some tests for short backing files and short overlays 21 22import iotests 23 24iotests.script_initialize(supported_fmts=['qcow2'], 25 supported_platforms=['linux']) 26 27size_short = 1 * 1024 * 1024 28size_long = 2 * 1024 * 1024 29size_diff = size_long - size_short 30 31def create_chain() -> None: 32 iotests.qemu_img_log('create', '-f', iotests.imgfmt, base, 33 str(size_long)) 34 iotests.qemu_img_log('create', '-f', iotests.imgfmt, '-b', base, 35 '-F', iotests.imgfmt, mid, str(size_short)) 36 iotests.qemu_img_log('create', '-f', iotests.imgfmt, '-b', mid, 37 '-F', iotests.imgfmt, top, str(size_long)) 38 39 iotests.qemu_io_log('-c', 'write -P 1 0 %d' % size_long, base) 40 41def create_vm() -> iotests.VM: 42 vm = iotests.VM() 43 vm.add_blockdev('file,filename=%s,node-name=base-file' % base) 44 vm.add_blockdev('%s,file=base-file,node-name=base' % iotests.imgfmt) 45 vm.add_blockdev('file,filename=%s,node-name=mid-file' % mid) 46 vm.add_blockdev('%s,file=mid-file,node-name=mid,backing=base' 47 % iotests.imgfmt) 48 vm.add_drive(top, 'backing=mid,node-name=top') 49 return vm 50 51with iotests.FilePath('base') as base, \ 52 iotests.FilePath('mid') as mid, \ 53 iotests.FilePath('top') as top: 54 55 iotests.log('== Commit tests ==') 56 57 create_chain() 58 59 iotests.log('=== Check visible data ===') 60 61 iotests.qemu_io_log('-c', 'read -P 1 0 %d' % size_short, top) 62 iotests.qemu_io_log('-c', 'read -P 0 %d %d' % (size_short, size_diff), top) 63 64 iotests.log('=== Checking allocation status ===') 65 66 iotests.qemu_io_log('-c', 'alloc 0 %d' % size_short, 67 '-c', 'alloc %d %d' % (size_short, size_diff), 68 base) 69 70 iotests.qemu_io_log('-c', 'alloc 0 %d' % size_short, 71 '-c', 'alloc %d %d' % (size_short, size_diff), 72 mid) 73 74 iotests.qemu_io_log('-c', 'alloc 0 %d' % size_short, 75 '-c', 'alloc %d %d' % (size_short, size_diff), 76 top) 77 78 iotests.log('=== Checking map ===') 79 80 iotests.qemu_img_log('map', '--output=json', base) 81 iotests.qemu_img_log('map', '--output=human', base) 82 iotests.qemu_img_log('map', '--output=json', mid) 83 iotests.qemu_img_log('map', '--output=human', mid) 84 iotests.qemu_img_log('map', '--output=json', top) 85 iotests.qemu_img_log('map', '--output=human', top) 86 87 iotests.log('=== Testing qemu-img commit (top -> mid) ===') 88 89 iotests.qemu_img_log('commit', top) 90 iotests.img_info_log(mid) 91 iotests.qemu_io_log('-c', 'read -P 1 0 %d' % size_short, mid) 92 iotests.qemu_io_log('-c', 'read -P 0 %d %d' % (size_short, size_diff), mid) 93 94 iotests.log('=== Testing HMP commit (top -> mid) ===') 95 96 create_chain() 97 with create_vm() as vm: 98 vm.launch() 99 vm.qmp_log('human-monitor-command', command_line='commit drive0') 100 101 iotests.img_info_log(mid) 102 iotests.qemu_io_log('-c', 'read -P 1 0 %d' % size_short, mid) 103 iotests.qemu_io_log('-c', 'read -P 0 %d %d' % (size_short, size_diff), mid) 104 105 iotests.log('=== Testing QMP active commit (top -> mid) ===') 106 107 create_chain() 108 with create_vm() as vm: 109 vm.launch() 110 vm.qmp_log('block-commit', device='top', base_node='mid', 111 job_id='job0', auto_dismiss=False) 112 vm.run_job('job0', wait=5) 113 114 iotests.img_info_log(mid) 115 iotests.qemu_io_log('-c', 'read -P 1 0 %d' % size_short, mid) 116 iotests.qemu_io_log('-c', 'read -P 0 %d %d' % (size_short, size_diff), mid) 117 118 iotests.log('=== Testing qemu-img commit (top -> base) ===') 119 120 create_chain() 121 iotests.qemu_img_log('commit', '-b', base, top) 122 iotests.img_info_log(base) 123 iotests.qemu_io_log('-c', 'read -P 1 0 %d' % size_short, base) 124 iotests.qemu_io_log('-c', 'read -P 0 %d %d' % (size_short, size_diff), base) 125 126 iotests.log('=== Testing QMP active commit (top -> base) ===') 127 128 create_chain() 129 with create_vm() as vm: 130 vm.launch() 131 vm.qmp_log('block-commit', device='top', base_node='base', 132 job_id='job0', auto_dismiss=False) 133 vm.run_job('job0', wait=5) 134 135 iotests.img_info_log(mid) 136 iotests.qemu_io_log('-c', 'read -P 1 0 %d' % size_short, base) 137 iotests.qemu_io_log('-c', 'read -P 0 %d %d' % (size_short, size_diff), base) 138 139 iotests.log('== Resize tests ==') 140 141 # Use different sizes for different allocation modes: 142 # 143 # We want to have at least one test where 32 bit truncation in the size of 144 # the overlapping area becomes visible. This is covered by the 145 # prealloc='off' case (1G to 6G is an overlap of 5G). 146 # 147 # However, we can only do this for modes that don't preallocate data 148 # because otherwise we might run out of space on the test host. 149 # 150 # We also want to test some unaligned combinations. 151 for (prealloc, base_size, top_size_old, top_size_new, off) in [ 152 ('off', '6G', '1G', '8G', '5G'), 153 ('metadata', '32G', '30G', '33G', '31G'), 154 ('falloc', '10M', '5M', '15M', '9M'), 155 ('full', '16M', '8M', '12M', '11M'), 156 ('off', '384k', '253k', '512k', '253k'), 157 ('off', '400k', '256k', '512k', '336k'), 158 ('off', '512k', '256k', '500k', '436k')]: 159 160 iotests.log('=== preallocation=%s ===' % prealloc) 161 iotests.qemu_img_log('create', '-f', iotests.imgfmt, base, base_size) 162 iotests.qemu_img_log('create', '-f', iotests.imgfmt, '-b', base, 163 '-F', iotests.imgfmt, top, top_size_old) 164 iotests.qemu_io_log('-c', 'write -P 1 %s 64k' % off, base) 165 166 # After this, top_size_old to base_size should be allocated/zeroed. 167 # 168 # In theory, leaving base_size to top_size_new unallocated would be 169 # correct, but in practice, if we zero out anything, we zero out 170 # everything up to top_size_new. 171 iotests.qemu_img_log('resize', '-f', iotests.imgfmt, 172 '--preallocation', prealloc, top, top_size_new) 173 iotests.qemu_io_log('-c', 'read -P 0 %s 64k' % off, top) 174 iotests.qemu_io_log('-c', 'map', top) 175 iotests.qemu_img_log('map', '--output=json', top) 176