xref: /openbmc/qemu/tests/qemu-iotests/163 (revision 7d87775f) 
1#!/usr/bin/env python3
2# group: rw
3#
4# Tests for shrinking images
5#
6# Copyright (c) 2016-2017 Parallels 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 os, random, iotests, struct, qcow2, sys
23from iotests import qemu_img, qemu_io, image_size
24
25test_img = os.path.join(iotests.test_dir, 'test.img')
26check_img = os.path.join(iotests.test_dir, 'check.img')
27
28def size_to_int(str):
29    suff = ['B', 'K', 'M', 'G', 'T']
30    return int(str[:-1]) * 1024**suff.index(str[-1:])
31
32class ShrinkBaseClass(iotests.QMPTestCase):
33    image_len = '128M'
34    shrink_size = '10M'
35    chunk_size = '16M'
36    refcount_bits = '16'
37
38    def __qcow2_check(self, filename):
39        entry_bits = 3
40        entry_size = 1 << entry_bits
41        l1_mask = 0x00fffffffffffe00
42        div_roundup = lambda n, d: (n + d - 1) // d
43
44        def split_by_n(data, n):
45            for x in range(0, len(data), n):
46                yield struct.unpack('>Q', data[x:x + n])[0] & l1_mask
47
48        def check_l1_table(h, l1_data):
49            l1_list = list(split_by_n(l1_data, entry_size))
50            real_l1_size = div_roundup(h.size,
51                                       1 << (h.cluster_bits*2 - entry_size))
52            used, unused = l1_list[:real_l1_size], l1_list[real_l1_size:]
53
54            self.assertTrue(len(used) != 0, "Verifying l1 table content")
55            self.assertFalse(any(unused), "Verifying l1 table content")
56
57        def check_reftable(fd, h, reftable):
58            for offset in split_by_n(reftable, entry_size):
59                if offset != 0:
60                    fd.seek(offset)
61                    cluster = fd.read(1 << h.cluster_bits)
62                    self.assertTrue(any(cluster), "Verifying reftable content")
63
64        with open(filename, "rb") as fd:
65            h = qcow2.QcowHeader(fd)
66
67            fd.seek(h.l1_table_offset)
68            l1_table = fd.read(h.l1_size << entry_bits)
69
70            fd.seek(h.refcount_table_offset)
71            reftable = fd.read(h.refcount_table_clusters << h.cluster_bits)
72
73            check_l1_table(h, l1_table)
74            check_reftable(fd, h, reftable)
75
76    def __raw_check(self, filename):
77        pass
78
79    image_check = {
80        'qcow2' : __qcow2_check,
81        'raw' : __raw_check
82    }
83
84    def setUp(self):
85        if iotests.imgfmt == 'raw':
86            qemu_img('create', '-f', iotests.imgfmt, test_img, self.image_len)
87            qemu_img('create', '-f', iotests.imgfmt, check_img,
88                     self.shrink_size)
89        else:
90            qemu_img('create', '-f', iotests.imgfmt,
91                     '-o', 'cluster_size=' + self.cluster_size +
92                     ',refcount_bits=' + self.refcount_bits,
93                     test_img, self.image_len)
94            qemu_img('create', '-f', iotests.imgfmt,
95                     '-o', 'cluster_size=%s'% self.cluster_size,
96                     check_img, self.shrink_size)
97        qemu_io('-c', 'write -P 0xff 0 ' + self.shrink_size, check_img)
98
99    def tearDown(self):
100        os.remove(test_img)
101        os.remove(check_img)
102
103    def image_verify(self):
104        self.assertEqual(image_size(test_img), image_size(check_img),
105                         "Verifying image size")
106        self.image_check[iotests.imgfmt](self, test_img)
107
108        if iotests.imgfmt == 'raw':
109            return
110        qemu_img('check', test_img)
111
112    def test_empty_image(self):
113        qemu_img('resize',  '-f', iotests.imgfmt, '--shrink', test_img,
114                 self.shrink_size)
115
116        qemu_io('-c', f"read -P 0x00 0 {self.shrink_size}", test_img)
117
118        self.image_verify()
119
120    def test_sequential_write(self):
121        for offs in range(0, size_to_int(self.image_len),
122                          size_to_int(self.chunk_size)):
123            qemu_io('-c', 'write -P 0xff %d %s' % (offs, self.chunk_size),
124                    test_img)
125
126        qemu_img('resize',  '-f', iotests.imgfmt, '--shrink', test_img,
127                 self.shrink_size)
128
129        qemu_img("compare", test_img, check_img)
130
131        self.image_verify()
132
133    def test_random_write(self):
134        offs_list = list(range(0, size_to_int(self.image_len),
135                               size_to_int(self.chunk_size)))
136        random.shuffle(offs_list)
137        for offs in offs_list:
138            qemu_io('-c', 'write -P 0xff %d %s' % (offs, self.chunk_size),
139                    test_img)
140
141        qemu_img('resize',  '-f', iotests.imgfmt, '--shrink', test_img,
142                 self.shrink_size)
143
144        qemu_img("compare", test_img, check_img)
145
146        self.image_verify()
147
148class TestShrink512(ShrinkBaseClass):
149    image_len = '3M'
150    shrink_size = '1M'
151    chunk_size = '256K'
152    cluster_size = '512'
153    refcount_bits = '64'
154
155class TestShrink64K(ShrinkBaseClass):
156    cluster_size = '64K'
157
158class TestShrink1M(ShrinkBaseClass):
159    cluster_size = '1M'
160    refcount_bits = '1'
161
162ShrinkBaseClass = None
163
164if __name__ == '__main__':
165    iotests.main(supported_fmts=['raw', 'qcow2'],
166                 supported_protocols=['file'],
167                 unsupported_imgopts=['compat'])
168