static int alloc_refcount_block(BlockDriverState *bs,
                                int64_t cluster_index, void **refcount_block)
{
    BDRVQcow2State *s = bs->opaque;
    unsigned int refcount_table_index;
    int64_t ret;
    BLKDBG_EVENT(bs->file, BLKDBG_REFBLOCK_ALLOC);
    /* Find the refcount block for the given cluster */
    refcount_table_index = cluster_index >> s->refcount_block_bits;
    if (refcount_table_index < s->refcount_table_size) {
        uint64_t refcount_block_offset =
            s->refcount_table[refcount_table_index] & REFT_OFFSET_MASK;
        /* If it's already there, we're done */
        if (refcount_block_offset) {
            if (offset_into_cluster(s, refcount_block_offset)) {
                qcow2_signal_corruption(bs, true, -1, -1, "Refblock offset %#"
                                        PRIx64 " unaligned (reftable index: "
                                        "%#x)", refcount_block_offset,
                                        refcount_table_index);
             return load_refcount_block(bs, refcount_block_offset,
                                        refcount_block);
    /*
     * If we came here, we need to allocate something. Something is at least
     * a cluster for the new refcount block. It may also include a new refcount
     * table if the old refcount table is too small.
     *
     * Note that allocating clusters here needs some special care:
     *
     * - We can't use the normal qcow2_alloc_clusters(), it would try to
     *   increase the refcount and very likely we would end up with an endless
     *   recursion. Instead we must place the refcount blocks in a way that
     *   they can describe them themselves.
     *
     * - We need to consider that at this point we are inside update_refcounts
     *   and potentially doing an initial refcount increase. This means that
     *   some clusters have already been allocated by the caller, but their
     *   refcount isn't accurate yet. If we allocate clusters for metadata, we
     *   need to return -EAGAIN to signal the caller that it needs to restart
     *   the search for free clusters.
     *
     * - alloc_clusters_noref and qcow2_free_clusters may load a different
     *   refcount block into the cache
     */
    *refcount_block = NULL;
    /* We write to the refcount table, so we might depend on L2 tables */
    ret = qcow2_cache_flush(bs, s->l2_table_cache);
    if (ret < 0) {
        return ret;
    /* Allocate the refcount block itself and mark it as used */
    int64_t new_block = alloc_clusters_noref(bs, s->cluster_size);
    if (new_block < 0) {
        return new_block;
#ifdef DEBUG_ALLOC2
    fprintf(stderr, "qcow2: Allocate refcount block %d for %" PRIx64
        " at %" PRIx64 "\n",
        refcount_table_index, cluster_index << s->cluster_bits, new_block);
#endif
    if (in_same_refcount_block(s, new_block, cluster_index << s->cluster_bits)) {
        /* Zero the new refcount block before updating it */
        ret = qcow2_cache_get_empty(bs, s->refcount_block_cache, new_block,
                                    refcount_block);
        if (ret < 0) {
            goto fail;
        memset(*refcount_block, 0, s->cluster_size);
        /* The block describes itself, need to update the cache */
        int block_index = (new_block >> s->cluster_bits) &
            (s->refcount_block_size - 1);
        s->set_refcount(*refcount_block, block_index, 1);
    } else {
        /* Described somewhere else. This can recurse at most twice before we
         * arrive at a block that describes itself. */
        ret = update_refcount(bs, new_block, s->cluster_size, 1, false,
                              QCOW2_DISCARD_NEVER);
        if (ret < 0) {
            goto fail;
        ret = qcow2_cache_flush(bs, s->refcount_block_cache);
        if (ret < 0) {
            goto fail;
        /* Initialize the new refcount block only after updating its refcount,
         * update_refcount uses the refcount cache itself */
        ret = qcow2_cache_get_empty(bs, s->refcount_block_cache, new_block,
                                    refcount_block);
        if (ret < 0) {
            goto fail;
        memset(*refcount_block, 0, s->cluster_size);
    /* Now the new refcount block needs to be written to disk */
    BLKDBG_EVENT(bs->file, BLKDBG_REFBLOCK_ALLOC_WRITE);
    qcow2_cache_entry_mark_dirty(bs, s->refcount_block_cache, *refcount_block);
    ret = qcow2_cache_flush(bs, s->refcount_block_cache);
    if (ret < 0) {
        goto fail;
    /* If the refcount table is big enough, just hook the block up there */
    if (refcount_table_index < s->refcount_table_size) {
        uint64_t data64 = cpu_to_be64(new_block);
        BLKDBG_EVENT(bs->file, BLKDBG_REFBLOCK_ALLOC_HOOKUP);
        ret = bdrv_pwrite_sync(bs->file,
            s->refcount_table_offset + refcount_table_index * sizeof(uint64_t),
            &data64, sizeof(data64));
        if (ret < 0) {
            goto fail;
        s->refcount_table[refcount_table_index] = new_block;
        /* If there's a hole in s->refcount_table then it can happen
         * that refcount_table_index < s->max_refcount_table_index */
        s->max_refcount_table_index =
            MAX(s->max_refcount_table_index, refcount_table_index);
        /* The new refcount block may be where the caller intended to put its
         * data, so let it restart the search. */
        return -EAGAIN;
    qcow2_cache_put(bs, s->refcount_block_cache, refcount_block);
    /*
     * If we come here, we need to grow the refcount table. Again, a new
     * refcount table needs some space and we can't simply allocate to avoid
     * endless recursion.
     *
     * Therefore let's grab new refcount blocks at the end of the image, which
     * will describe themselves and the new refcount table. This way we can
     * reference them only in the new table and do the switch to the new
     * refcount table at once without producing an inconsistent state in
     * between.
     */
    BLKDBG_EVENT(bs->file, BLKDBG_REFTABLE_GROW);
    /* Calculate the number of refcount blocks needed so far; this will be the
     * basis for calculating the index of the first cluster used for the
     * self-describing refcount structures which we are about to create.
     *
     * Because we reached this point, there cannot be any refcount entries for
     * cluster_index or higher indices yet. However, because new_block has been
     * allocated to describe that cluster (and it will assume this role later
     * on), we cannot use that index; also, new_block may actually have a higher
     * cluster index than cluster_index, so it needs to be taken into account
     * here (and 1 needs to be added to its value because that cluster is used).
     */
    uint64_t blocks_used = DIV_ROUND_UP(MAX(cluster_index + 1,
                                            (new_block >> s->cluster_bits) + 1),
                                        s->refcount_block_size);
    /* Create the new refcount table and blocks */
    uint64_t meta_offset = (blocks_used * s->refcount_block_size) *
        s->cluster_size;
    ret = qcow2_refcount_area(bs, meta_offset, 0, false,
                              refcount_table_index, new_block);
    if (ret < 0) {
        return ret;
    ret = load_refcount_block(bs, new_block, refcount_block);
    if (ret < 0) {
        return ret;
    /* If we were trying to do the initial refcount update for some cluster
     * allocation, we might have used the same clusters to store newly
     * allocated metadata. Make the caller search some new space. */
    return -EAGAIN;
fail:
    if (*refcount_block != NULL) {
        qcow2_cache_put(bs, s->refcount_block_cache, refcount_block);
    return ret;
static int alloc_refcount_block(BlockDriverState *bs,
                                int64_t cluster_index, void **refcount_block)
{
    BDRVQcow2State *s = bs->opaque;
    unsigned int refcount_table_index;
    int64_t ret;
    BLKDBG_EVENT(bs->file, BLKDBG_REFBLOCK_ALLOC);
    
    refcount_table_index = cluster_index >> s->refcount_block_bits;
    if (refcount_table_index < s->refcount_table_size) {
        uint64_t refcount_block_offset =
            s->refcount_table[refcount_table_index] & REFT_OFFSET_MASK;
        
        if (refcount_block_offset) {
            if (offset_into_cluster(s, refcount_block_offset)) {
                qcow2_signal_corruption(bs, true, -1, -1, "Refblock offset %#"
                                        PRIx64 " unaligned (reftable index: "
                                        "%#x)", refcount_block_offset,
                                        refcount_table_index);
             return load_refcount_block(bs, refcount_block_offset,
                                        refcount_block);
    





















    *refcount_block = NULL;
    
    ret = qcow2_cache_flush(bs, s->l2_table_cache);
    if (ret < 0) {
        return ret;
    
    int64_t new_block = alloc_clusters_noref(bs, s->cluster_size);
    if (new_block < 0) {
        return new_block;
#ifdef DEBUG_ALLOC2
    fprintf(stderr, "qcow2: Allocate refcount block %d for %" PRIx64
        " at %" PRIx64 "\n",
        refcount_table_index, cluster_index << s->cluster_bits, new_block);
#endif
    if (in_same_refcount_block(s, new_block, cluster_index << s->cluster_bits)) {
        
        ret = qcow2_cache_get_empty(bs, s->refcount_block_cache, new_block,
                                    refcount_block);
        if (ret < 0) {
            goto fail;
        memset(*refcount_block, 0, s->cluster_size);
        
        int block_index = (new_block >> s->cluster_bits) &
            (s->refcount_block_size - 1);
        s->set_refcount(*refcount_block, block_index, 1);
    } else {
        

        ret = update_refcount(bs, new_block, s->cluster_size, 1, false,
                              QCOW2_DISCARD_NEVER);
        if (ret < 0) {
            goto fail;
        ret = qcow2_cache_flush(bs, s->refcount_block_cache);
        if (ret < 0) {
            goto fail;
        

        ret = qcow2_cache_get_empty(bs, s->refcount_block_cache, new_block,
                                    refcount_block);
        if (ret < 0) {
            goto fail;
        memset(*refcount_block, 0, s->cluster_size);
    
    BLKDBG_EVENT(bs->file, BLKDBG_REFBLOCK_ALLOC_WRITE);
    qcow2_cache_entry_mark_dirty(bs, s->refcount_block_cache, *refcount_block);
    ret = qcow2_cache_flush(bs, s->refcount_block_cache);
    if (ret < 0) {
        goto fail;
    
    if (refcount_table_index < s->refcount_table_size) {
        uint64_t data64 = cpu_to_be64(new_block);
        BLKDBG_EVENT(bs->file, BLKDBG_REFBLOCK_ALLOC_HOOKUP);
        ret = bdrv_pwrite_sync(bs->file,
            s->refcount_table_offset + refcount_table_index * sizeof(uint64_t),
            &data64, sizeof(data64));
        if (ret < 0) {
            goto fail;
        s->refcount_table[refcount_table_index] = new_block;
        

        s->max_refcount_table_index =
            MAX(s->max_refcount_table_index, refcount_table_index);
        

        return -EAGAIN;
    qcow2_cache_put(bs, s->refcount_block_cache, refcount_block);
    










    BLKDBG_EVENT(bs->file, BLKDBG_REFTABLE_GROW);
    










    uint64_t blocks_used = DIV_ROUND_UP(MAX(cluster_index + 1,
                                            (new_block >> s->cluster_bits) + 1),
                                        s->refcount_block_size);
    
    uint64_t meta_offset = (blocks_used * s->refcount_block_size) *
        s->cluster_size;
    ret = qcow2_refcount_area(bs, meta_offset, 0, false,
                              refcount_table_index, new_block);
    if (ret < 0) {
        return ret;
    ret = load_refcount_block(bs, new_block, refcount_block);
    if (ret < 0) {
        return ret;
    


    return -EAGAIN;
fail:
    if (*refcount_block != NULL) {
        qcow2_cache_put(bs, s->refcount_block_cache, refcount_block);
    return ret;
static int FUNC_0(BlockDriverState *VAR_0,
                                int64_t VAR_1, void **VAR_2)
{
    BDRVQcow2State *s = VAR_0->opaque;
    unsigned int VAR_3;
    int64_t ret;
    BLKDBG_EVENT(VAR_0->file, BLKDBG_REFBLOCK_ALLOC);
    
    VAR_3 = VAR_1 >> s->refcount_block_bits;
    if (VAR_3 < s->refcount_table_size) {
        uint64_t refcount_block_offset =
            s->refcount_table[VAR_3] & REFT_OFFSET_MASK;
        
        if (refcount_block_offset) {
            if (offset_into_cluster(s, refcount_block_offset)) {
                qcow2_signal_corruption(VAR_0, true, -1, -1, "Refblock offset %#"
                                        PRIx64 " unaligned (reftable index: "
                                        "%#x)", refcount_block_offset,
                                        VAR_3);
             return load_refcount_block(VAR_0, refcount_block_offset,
                                        VAR_2);
    





















    *VAR_2 = NULL;
    
    ret = qcow2_cache_flush(VAR_0, s->l2_table_cache);
    if (ret < 0) {
        return ret;
    
    int64_t new_block = alloc_clusters_noref(VAR_0, s->cluster_size);
    if (new_block < 0) {
        return new_block;
#ifdef DEBUG_ALLOC2
    fprintf(stderr, "qcow2: Allocate refcount block %d for %" PRIx64
        " at %" PRIx64 "\n",
        VAR_3, VAR_1 << s->cluster_bits, new_block);
#endif
    if (in_same_refcount_block(s, new_block, VAR_1 << s->cluster_bits)) {
        
        ret = qcow2_cache_get_empty(VAR_0, s->refcount_block_cache, new_block,
                                    VAR_2);
        if (ret < 0) {
            goto fail;
        memset(*VAR_2, 0, s->cluster_size);
        
        int VAR_4 = (new_block >> s->cluster_bits) &
            (s->refcount_block_size - 1);
        s->set_refcount(*VAR_2, VAR_4, 1);
    } else {
        

        ret = update_refcount(VAR_0, new_block, s->cluster_size, 1, false,
                              QCOW2_DISCARD_NEVER);
        if (ret < 0) {
            goto fail;
        ret = qcow2_cache_flush(VAR_0, s->refcount_block_cache);
        if (ret < 0) {
            goto fail;
        

        ret = qcow2_cache_get_empty(VAR_0, s->refcount_block_cache, new_block,
                                    VAR_2);
        if (ret < 0) {
            goto fail;
        memset(*VAR_2, 0, s->cluster_size);
    
    BLKDBG_EVENT(VAR_0->file, BLKDBG_REFBLOCK_ALLOC_WRITE);
    qcow2_cache_entry_mark_dirty(VAR_0, s->refcount_block_cache, *VAR_2);
    ret = qcow2_cache_flush(VAR_0, s->refcount_block_cache);
    if (ret < 0) {
        goto fail;
    
    if (VAR_3 < s->refcount_table_size) {
        uint64_t data64 = cpu_to_be64(new_block);
        BLKDBG_EVENT(VAR_0->file, BLKDBG_REFBLOCK_ALLOC_HOOKUP);
        ret = bdrv_pwrite_sync(VAR_0->file,
            s->refcount_table_offset + VAR_3 * sizeof(uint64_t),
            &data64, sizeof(data64));
        if (ret < 0) {
            goto fail;
        s->refcount_table[VAR_3] = new_block;
        

        s->max_refcount_table_index =
            MAX(s->max_refcount_table_index, VAR_3);
        

        return -EAGAIN;
    qcow2_cache_put(VAR_0, s->refcount_block_cache, VAR_2);
    










    BLKDBG_EVENT(VAR_0->file, BLKDBG_REFTABLE_GROW);
    










    uint64_t blocks_used = DIV_ROUND_UP(MAX(VAR_1 + 1,
                                            (new_block >> s->cluster_bits) + 1),
                                        s->refcount_block_size);
    
    uint64_t meta_offset = (blocks_used * s->refcount_block_size) *
        s->cluster_size;
    ret = qcow2_refcount_area(VAR_0, meta_offset, 0, false,
                              VAR_3, new_block);
    if (ret < 0) {
        return ret;
    ret = load_refcount_block(VAR_0, new_block, VAR_2);
    if (ret < 0) {
        return ret;
    


    return -EAGAIN;
fail:
    if (*VAR_2 != NULL) {
        qcow2_cache_put(VAR_0, s->refcount_block_cache, VAR_2);
    return ret;
static int FUNC_0(BlockDriverState *VAR_0,
int64_t VAR_1, void **VAR_2)
{ BDRVQcow2State *s = VAR_0->opaque; unsigned int VAR_3; int64_t ret; BLKDBG_EVENT(VAR_0->file, BLKDBG_REFBLOCK_ALLOC); VAR_3 = VAR_1 >> s->refcount_block_bits; if (VAR_3 < s->refcount_table_size) { uint64_t refcount_block_offset =
s->refcount_table[VAR_3] & REFT_OFFSET_MASK; if (refcount_block_offset) { if (offset_into_cluster(s, refcount_block_offset)) { qcow2_signal_corruption(VAR_0, true, -1, -1, "Refblock offset %#"
PRIx64 " unaligned (reftable index: "
"%#x)", refcount_block_offset,
VAR_3); return load_refcount_block(VAR_0, refcount_block_offset,
VAR_2); *VAR_2 = NULL; ret = qcow2_cache_flush(VAR_0, s->l2_table_cache); if (ret < 0) { return ret; int64_t new_block = alloc_clusters_noref(VAR_0, s->cluster_size); if (new_block < 0) { return new_block; #ifdef DEBUG_ALLOC2
fprintf(stderr, "qcow2: Allocate refcount block %d for %" PRIx64
" at %" PRIx64 "\n",
VAR_3, VAR_1 << s->cluster_bits, new_block); #endif
if (in_same_refcount_block(s, new_block, VAR_1 << s->cluster_bits)) { ret = qcow2_cache_get_empty(VAR_0, s->refcount_block_cache, new_block,
VAR_2); if (ret < 0) { goto fail; memset(*VAR_2, 0, s->cluster_size); int VAR_4 = (new_block >> s->cluster_bits) &
(s->refcount_block_size - 1); s->set_refcount(*VAR_2, VAR_4, 1); } else { ret = update_refcount(VAR_0, new_block, s->cluster_size, 1, false,
QCOW2_DISCARD_NEVER); if (ret < 0) { goto fail; ret = qcow2_cache_flush(VAR_0, s->refcount_block_cache); if (ret < 0) { goto fail; ret = qcow2_cache_get_empty(VAR_0, s->refcount_block_cache, new_block,
VAR_2); if (ret < 0) { goto fail; memset(*VAR_2, 0, s->cluster_size); BLKDBG_EVENT(VAR_0->file, BLKDBG_REFBLOCK_ALLOC_WRITE); qcow2_cache_entry_mark_dirty(VAR_0, s->refcount_block_cache, *VAR_2); ret = qcow2_cache_flush(VAR_0, s->refcount_block_cache); if (ret < 0) { goto fail; if (VAR_3 < s->refcount_table_size) { uint64_t data64 = cpu_to_be64(new_block); BLKDBG_EVENT(VAR_0->file, BLKDBG_REFBLOCK_ALLOC_HOOKUP); ret = bdrv_pwrite_sync(VAR_0->file,
s->refcount_table_offset + VAR_3 * sizeof(uint64_t),
&data64, sizeof(data64)); if (ret < 0) { goto fail; s->refcount_table[VAR_3] = new_block; s->max_refcount_table_index =
MAX(s->max_refcount_table_index, VAR_3); return -EAGAIN; qcow2_cache_put(VAR_0, s->refcount_block_cache, VAR_2); BLKDBG_EVENT(VAR_0->file, BLKDBG_REFTABLE_GROW); uint64_t blocks_used = DIV_ROUND_UP(MAX(VAR_1 + 1,
(new_block >> s->cluster_bits) + 1),
s->refcount_block_size); uint64_t meta_offset = (blocks_used * s->refcount_block_size) *
s->cluster_size; ret = qcow2_refcount_area(VAR_0, meta_offset, 0, false,
VAR_3, new_block); if (ret < 0) { return ret; ret = load_refcount_block(VAR_0, new_block, VAR_2); if (ret < 0) { return ret; return -EAGAIN; fail:
if (*VAR_2 != NULL) { qcow2_cache_put(VAR_0, s->refcount_block_cache, VAR_2); return ret;