1#!/usr/bin/env python3 2# group: auto quick 3# 4# Test shutdown when bitmap is exported through NBD server 5# 6# Copyright (c) 2020 Virtuozzo International GmbH. 7# 8# This program is free software; you can redistribute it and/or modify 9# it under the terms of the GNU General Public License as published by 10# the Free Software Foundation; either version 2 of the License, or 11# (at your option) any later version. 12# 13# This program is distributed in the hope that it will be useful, 14# but WITHOUT ANY WARRANTY; without even the implied warranty of 15# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 16# GNU General Public License for more details. 17# 18# You should have received a copy of the GNU General Public License 19# along with this program. If not, see <http://www.gnu.org/licenses/>. 20# 21 22import iotests 23 24# The test is unrelated to formats, restrict it to qcow2 to avoid extra runs 25iotests.script_initialize( 26 supported_fmts=['qcow2'], 27) 28 29nbd_sock = iotests.file_path('nbd.sock', base_dir=iotests.sock_dir) 30nbd_uri = 'nbd+unix:///disk?socket=' + nbd_sock 31size = 1024 * 1024 32 33vm = iotests.VM() 34vm.launch() 35 36vm.qmp_log('blockdev-add', **{ 37 'node-name': 'disk', 38 'driver': 'null-co', 39 'size': 1024 * 1024, 40}) 41 42vm.qmp_log('block-dirty-bitmap-add', **{ 43 'node': 'disk', 44 'name': 'bitmap0' 45}) 46 47vm.qmp_log('nbd-server-start', **{ 48 'addr': { 49 'type': 'unix', 50 'data': {'path': nbd_sock} 51 } 52}, filters=[iotests.filter_qmp_testfiles]) 53 54vm.qmp_log('nbd-server-add', **{ 55 'device': 'disk', 56 'writable': True, 57 'bitmap': 'bitmap0' 58}) 59 60p = iotests.QemuIoInteractive('-f', 'raw', nbd_uri) 61# wait for connection and check it: 62iotests.log(p.cmd('read 0 512').rstrip(), filters=[iotests.filter_qemu_io]) 63 64vm.shutdown() 65 66p.close() 67