Added license information
[kvmfornfv.git] / qemu / block / qed-check.c
1 /*
2  * QEMU Enhanced Disk Format Consistency Check
3  *
4  * Copyright IBM, Corp. 2010
5  *
6  * Authors:
7  *  Stefan Hajnoczi   <stefanha@linux.vnet.ibm.com>
8  *
9  * This work is licensed under the terms of the GNU LGPL, version 2 or later.
10  * See the COPYING.LIB file in the top-level directory.
11  *
12  */
13
14 #include "qed.h"
15
16 typedef struct {
17     BDRVQEDState *s;
18     BdrvCheckResult *result;
19     bool fix;                           /* whether to fix invalid offsets */
20
21     uint64_t nclusters;
22     uint32_t *used_clusters;            /* referenced cluster bitmap */
23
24     QEDRequest request;
25 } QEDCheck;
26
27 static bool qed_test_bit(uint32_t *bitmap, uint64_t n) {
28     return !!(bitmap[n / 32] & (1 << (n % 32)));
29 }
30
31 static void qed_set_bit(uint32_t *bitmap, uint64_t n) {
32     bitmap[n / 32] |= 1 << (n % 32);
33 }
34
35 /**
36  * Set bitmap bits for clusters
37  *
38  * @check:          Check structure
39  * @offset:         Starting offset in bytes
40  * @n:              Number of clusters
41  */
42 static bool qed_set_used_clusters(QEDCheck *check, uint64_t offset,
43                                   unsigned int n)
44 {
45     uint64_t cluster = qed_bytes_to_clusters(check->s, offset);
46     unsigned int corruptions = 0;
47
48     while (n-- != 0) {
49         /* Clusters should only be referenced once */
50         if (qed_test_bit(check->used_clusters, cluster)) {
51             corruptions++;
52         }
53
54         qed_set_bit(check->used_clusters, cluster);
55         cluster++;
56     }
57
58     check->result->corruptions += corruptions;
59     return corruptions == 0;
60 }
61
62 /**
63  * Check an L2 table
64  *
65  * @ret:            Number of invalid cluster offsets
66  */
67 static unsigned int qed_check_l2_table(QEDCheck *check, QEDTable *table)
68 {
69     BDRVQEDState *s = check->s;
70     unsigned int i, num_invalid = 0;
71     uint64_t last_offset = 0;
72
73     for (i = 0; i < s->table_nelems; i++) {
74         uint64_t offset = table->offsets[i];
75
76         if (qed_offset_is_unalloc_cluster(offset) ||
77             qed_offset_is_zero_cluster(offset)) {
78             continue;
79         }
80         check->result->bfi.allocated_clusters++;
81         if (last_offset && (last_offset + s->header.cluster_size != offset)) {
82             check->result->bfi.fragmented_clusters++;
83         }
84         last_offset = offset;
85
86         /* Detect invalid cluster offset */
87         if (!qed_check_cluster_offset(s, offset)) {
88             if (check->fix) {
89                 table->offsets[i] = 0;
90                 check->result->corruptions_fixed++;
91             } else {
92                 check->result->corruptions++;
93             }
94
95             num_invalid++;
96             continue;
97         }
98
99         qed_set_used_clusters(check, offset, 1);
100     }
101
102     return num_invalid;
103 }
104
105 /**
106  * Descend tables and check each cluster is referenced once only
107  */
108 static int qed_check_l1_table(QEDCheck *check, QEDTable *table)
109 {
110     BDRVQEDState *s = check->s;
111     unsigned int i, num_invalid_l1 = 0;
112     int ret, last_error = 0;
113
114     /* Mark L1 table clusters used */
115     qed_set_used_clusters(check, s->header.l1_table_offset,
116                           s->header.table_size);
117
118     for (i = 0; i < s->table_nelems; i++) {
119         unsigned int num_invalid_l2;
120         uint64_t offset = table->offsets[i];
121
122         if (qed_offset_is_unalloc_cluster(offset)) {
123             continue;
124         }
125
126         /* Detect invalid L2 offset */
127         if (!qed_check_table_offset(s, offset)) {
128             /* Clear invalid offset */
129             if (check->fix) {
130                 table->offsets[i] = 0;
131                 check->result->corruptions_fixed++;
132             } else {
133                 check->result->corruptions++;
134             }
135
136             num_invalid_l1++;
137             continue;
138         }
139
140         if (!qed_set_used_clusters(check, offset, s->header.table_size)) {
141             continue; /* skip an invalid table */
142         }
143
144         ret = qed_read_l2_table_sync(s, &check->request, offset);
145         if (ret) {
146             check->result->check_errors++;
147             last_error = ret;
148             continue;
149         }
150
151         num_invalid_l2 = qed_check_l2_table(check,
152                                             check->request.l2_table->table);
153
154         /* Write out fixed L2 table */
155         if (num_invalid_l2 > 0 && check->fix) {
156             ret = qed_write_l2_table_sync(s, &check->request, 0,
157                                           s->table_nelems, false);
158             if (ret) {
159                 check->result->check_errors++;
160                 last_error = ret;
161                 continue;
162             }
163         }
164     }
165
166     /* Drop reference to final table */
167     qed_unref_l2_cache_entry(check->request.l2_table);
168     check->request.l2_table = NULL;
169
170     /* Write out fixed L1 table */
171     if (num_invalid_l1 > 0 && check->fix) {
172         ret = qed_write_l1_table_sync(s, 0, s->table_nelems);
173         if (ret) {
174             check->result->check_errors++;
175             last_error = ret;
176         }
177     }
178
179     return last_error;
180 }
181
182 /**
183  * Check for unreferenced (leaked) clusters
184  */
185 static void qed_check_for_leaks(QEDCheck *check)
186 {
187     BDRVQEDState *s = check->s;
188     uint64_t i;
189
190     for (i = s->header.header_size; i < check->nclusters; i++) {
191         if (!qed_test_bit(check->used_clusters, i)) {
192             check->result->leaks++;
193         }
194     }
195 }
196
197 /**
198  * Mark an image clean once it passes check or has been repaired
199  */
200 static void qed_check_mark_clean(BDRVQEDState *s, BdrvCheckResult *result)
201 {
202     /* Skip if there were unfixable corruptions or I/O errors */
203     if (result->corruptions > 0 || result->check_errors > 0) {
204         return;
205     }
206
207     /* Skip if image is already marked clean */
208     if (!(s->header.features & QED_F_NEED_CHECK)) {
209         return;
210     }
211
212     /* Ensure fixes reach storage before clearing check bit */
213     bdrv_flush(s->bs);
214
215     s->header.features &= ~QED_F_NEED_CHECK;
216     qed_write_header_sync(s);
217 }
218
219 int qed_check(BDRVQEDState *s, BdrvCheckResult *result, bool fix)
220 {
221     QEDCheck check = {
222         .s = s,
223         .result = result,
224         .nclusters = qed_bytes_to_clusters(s, s->file_size),
225         .request = { .l2_table = NULL },
226         .fix = fix,
227     };
228     int ret;
229
230     check.used_clusters = g_try_new0(uint32_t, (check.nclusters + 31) / 32);
231     if (check.nclusters && check.used_clusters == NULL) {
232         return -ENOMEM;
233     }
234
235     check.result->bfi.total_clusters =
236         (s->header.image_size + s->header.cluster_size - 1) /
237             s->header.cluster_size;
238     ret = qed_check_l1_table(&check, s->l1_table);
239     if (ret == 0) {
240         /* Only check for leaks if entire image was scanned successfully */
241         qed_check_for_leaks(&check);
242
243         if (fix) {
244             qed_check_mark_clean(s, result);
245         }
246     }
247
248     g_free(check.used_clusters);
249     return ret;
250 }