qcow2: Let inc_refcounts() return -errno

As of a future patch, inc_refcounts() will have to throw errors which
are generally signaled by returning -errno. Therefore, let it return an
integer which is either 0 for success or -errno and handle the -errno
case in all callers.

Signed-off-by: Max Reitz <mreitz@redhat.com>
Signed-off-by: Kevin Wolf <kwolf@redhat.com>
This commit is contained in:
Max Reitz 2014-10-22 14:09:35 +02:00 committed by Kevin Wolf
parent ad27390c85
commit fef4d3d564
1 changed files with 60 additions and 31 deletions

View File

@ -1081,17 +1081,18 @@ fail:
* *
* Modifies the number of errors in res. * Modifies the number of errors in res.
*/ */
static void inc_refcounts(BlockDriverState *bs, static int inc_refcounts(BlockDriverState *bs,
BdrvCheckResult *res, BdrvCheckResult *res,
uint16_t *refcount_table, uint16_t *refcount_table,
int64_t refcount_table_size, int64_t refcount_table_size,
int64_t offset, int64_t size) int64_t offset, int64_t size)
{ {
BDRVQcowState *s = bs->opaque; BDRVQcowState *s = bs->opaque;
uint64_t start, last, cluster_offset, k; uint64_t start, last, cluster_offset, k;
if (size <= 0) if (size <= 0) {
return; return 0;
}
start = start_of_cluster(s, offset); start = start_of_cluster(s, offset);
last = start_of_cluster(s, offset + size - 1); last = start_of_cluster(s, offset + size - 1);
@ -1111,6 +1112,8 @@ static void inc_refcounts(BlockDriverState *bs,
} }
} }
} }
return 0;
} }
/* Flags for check_refcounts_l1() and check_refcounts_l2() */ /* Flags for check_refcounts_l1() and check_refcounts_l2() */
@ -1165,8 +1168,11 @@ static int check_refcounts_l2(BlockDriverState *bs, BdrvCheckResult *res,
nb_csectors = ((l2_entry >> s->csize_shift) & nb_csectors = ((l2_entry >> s->csize_shift) &
s->csize_mask) + 1; s->csize_mask) + 1;
l2_entry &= s->cluster_offset_mask; l2_entry &= s->cluster_offset_mask;
inc_refcounts(bs, res, refcount_table, refcount_table_size, ret = inc_refcounts(bs, res, refcount_table, refcount_table_size,
l2_entry & ~511, nb_csectors * 512); l2_entry & ~511, nb_csectors * 512);
if (ret < 0) {
goto fail;
}
if (flags & CHECK_FRAG_INFO) { if (flags & CHECK_FRAG_INFO) {
res->bfi.allocated_clusters++; res->bfi.allocated_clusters++;
@ -1201,8 +1207,11 @@ static int check_refcounts_l2(BlockDriverState *bs, BdrvCheckResult *res,
} }
/* Mark cluster as used */ /* Mark cluster as used */
inc_refcounts(bs, res, refcount_table,refcount_table_size, ret = inc_refcounts(bs, res, refcount_table, refcount_table_size,
offset, s->cluster_size); offset, s->cluster_size);
if (ret < 0) {
goto fail;
}
/* Correct offsets are cluster aligned */ /* Correct offsets are cluster aligned */
if (offset_into_cluster(s, offset)) { if (offset_into_cluster(s, offset)) {
@ -1245,19 +1254,20 @@ static int check_refcounts_l1(BlockDriverState *bs,
int flags) int flags)
{ {
BDRVQcowState *s = bs->opaque; BDRVQcowState *s = bs->opaque;
uint64_t *l1_table, l2_offset, l1_size2; uint64_t *l1_table = NULL, l2_offset, l1_size2;
int i, ret; int i, ret;
l1_size2 = l1_size * sizeof(uint64_t); l1_size2 = l1_size * sizeof(uint64_t);
/* Mark L1 table as used */ /* Mark L1 table as used */
inc_refcounts(bs, res, refcount_table, refcount_table_size, ret = inc_refcounts(bs, res, refcount_table, refcount_table_size,
l1_table_offset, l1_size2); l1_table_offset, l1_size2);
if (ret < 0) {
goto fail;
}
/* Read L1 table entries from disk */ /* Read L1 table entries from disk */
if (l1_size2 == 0) { if (l1_size2 > 0) {
l1_table = NULL;
} else {
l1_table = g_try_malloc(l1_size2); l1_table = g_try_malloc(l1_size2);
if (l1_table == NULL) { if (l1_table == NULL) {
ret = -ENOMEM; ret = -ENOMEM;
@ -1280,8 +1290,11 @@ static int check_refcounts_l1(BlockDriverState *bs,
if (l2_offset) { if (l2_offset) {
/* Mark L2 table as used */ /* Mark L2 table as used */
l2_offset &= L1E_OFFSET_MASK; l2_offset &= L1E_OFFSET_MASK;
inc_refcounts(bs, res, refcount_table, refcount_table_size, ret = inc_refcounts(bs, res, refcount_table, refcount_table_size,
l2_offset, s->cluster_size); l2_offset, s->cluster_size);
if (ret < 0) {
goto fail;
}
/* L2 tables are cluster aligned */ /* L2 tables are cluster aligned */
if (offset_into_cluster(s, l2_offset)) { if (offset_into_cluster(s, l2_offset)) {
@ -1550,6 +1563,7 @@ static int check_refblocks(BlockDriverState *bs, BdrvCheckResult *res,
{ {
BDRVQcowState *s = bs->opaque; BDRVQcowState *s = bs->opaque;
int64_t i; int64_t i;
int ret;
for(i = 0; i < s->refcount_table_size; i++) { for(i = 0; i < s->refcount_table_size; i++) {
uint64_t offset, cluster; uint64_t offset, cluster;
@ -1572,8 +1586,11 @@ static int check_refblocks(BlockDriverState *bs, BdrvCheckResult *res,
} }
if (offset != 0) { if (offset != 0) {
inc_refcounts(bs, res, *refcount_table, *nb_clusters, ret = inc_refcounts(bs, res, *refcount_table, *nb_clusters,
offset, s->cluster_size); offset, s->cluster_size);
if (ret < 0) {
return ret;
}
if ((*refcount_table)[cluster] != 1) { if ((*refcount_table)[cluster] != 1) {
fprintf(stderr, "%s refcount block %" PRId64 fprintf(stderr, "%s refcount block %" PRId64
" refcount=%d\n", " refcount=%d\n",
@ -1602,8 +1619,11 @@ static int check_refblocks(BlockDriverState *bs, BdrvCheckResult *res,
sizeof(**refcount_table)); sizeof(**refcount_table));
} }
(*refcount_table)[cluster]--; (*refcount_table)[cluster]--;
inc_refcounts(bs, res, *refcount_table, *nb_clusters, ret = inc_refcounts(bs, res, *refcount_table, *nb_clusters,
new_offset, s->cluster_size); new_offset, s->cluster_size);
if (ret < 0) {
return ret;
}
res->corruptions_fixed++; res->corruptions_fixed++;
} else { } else {
@ -1635,8 +1655,11 @@ static int calculate_refcounts(BlockDriverState *bs, BdrvCheckResult *res,
} }
/* header */ /* header */
inc_refcounts(bs, res, *refcount_table, *nb_clusters, ret = inc_refcounts(bs, res, *refcount_table, *nb_clusters,
0, s->cluster_size); 0, s->cluster_size);
if (ret < 0) {
return ret;
}
/* current L1 table */ /* current L1 table */
ret = check_refcounts_l1(bs, res, *refcount_table, *nb_clusters, ret = check_refcounts_l1(bs, res, *refcount_table, *nb_clusters,
@ -1649,18 +1672,24 @@ static int calculate_refcounts(BlockDriverState *bs, BdrvCheckResult *res,
for (i = 0; i < s->nb_snapshots; i++) { for (i = 0; i < s->nb_snapshots; i++) {
sn = s->snapshots + i; sn = s->snapshots + i;
ret = check_refcounts_l1(bs, res, *refcount_table, *nb_clusters, ret = check_refcounts_l1(bs, res, *refcount_table, *nb_clusters,
sn->l1_table_offset, sn->l1_size, 0); sn->l1_table_offset, sn->l1_size, 0);
if (ret < 0) { if (ret < 0) {
return ret; return ret;
} }
} }
inc_refcounts(bs, res, *refcount_table, *nb_clusters, ret = inc_refcounts(bs, res, *refcount_table, *nb_clusters,
s->snapshots_offset, s->snapshots_size); s->snapshots_offset, s->snapshots_size);
if (ret < 0) {
return ret;
}
/* refcount data */ /* refcount data */
inc_refcounts(bs, res, *refcount_table, *nb_clusters, ret = inc_refcounts(bs, res, *refcount_table, *nb_clusters,
s->refcount_table_offset, s->refcount_table_offset,
s->refcount_table_size * sizeof(uint64_t)); s->refcount_table_size * sizeof(uint64_t));
if (ret < 0) {
return ret;
}
return check_refblocks(bs, res, fix, refcount_table, nb_clusters); return check_refblocks(bs, res, fix, refcount_table, nb_clusters);
} }