src/import/qcow2-util.c

edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering/***
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering  This file is part of systemd.
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering  Copyright 2015 Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering  systemd is free software; you can redistribute it and/or modify it
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering  under the terms of the GNU Lesser General Public License as published by
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering  the Free Software Foundation; either version 2.1 of the License, or
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering  (at your option) any later version.
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering  systemd is distributed in the hope that it will be useful, but
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering  WITHOUT ANY WARRANTY; without even the implied warranty of
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering  Lesser General Public License for more details.
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering  You should have received a copy of the GNU Lesser General Public License
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering  along with systemd; If not, see <http://www.gnu.org/licenses/>.
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering***/
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering#include <zlib.h>
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
b5efdb8af40ea759a1ea584c1bc44ecc81dd00ceLennart Poettering#include "alloc-util.h"
1c7dd82563ff2e71a067aea20d2acb2d0553644bLennart Poettering#include "btrfs-util.h"
b5efdb8af40ea759a1ea584c1bc44ecc81dd00ceLennart Poettering#include "qcow2-util.h"
b5efdb8af40ea759a1ea584c1bc44ecc81dd00ceLennart Poettering#include "sparse-endian.h"
b5efdb8af40ea759a1ea584c1bc44ecc81dd00ceLennart Poettering#include "util.h"
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering#define QCOW2_MAGIC 0x514649fb
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering#define QCOW2_COPIED (1ULL << 63)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering#define QCOW2_COMPRESSED (1ULL << 62)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering#define QCOW2_ZERO (1ULL << 0)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poetteringtypedef struct _packed_ Header {
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering      be32_t magic;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering      be32_t version;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering      be64_t backing_file_offset;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering      be32_t backing_file_size;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering      be32_t cluster_bits;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering      be64_t size;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering      be32_t crypt_method;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering      be32_t l1_size;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering      be64_t l1_table_offset;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering      be64_t refcount_table_offset;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering      be32_t refcount_table_clusters;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering      be32_t nb_snapshots;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering      be64_t snapshots_offset;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering      /* The remainder is only present on QCOW3 */
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering      be64_t incompatible_features;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering      be64_t compatible_features;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering      be64_t autoclear_features;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering      be32_t refcount_order;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering      be32_t header_length;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering} Header;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering#define HEADER_MAGIC(header) be32toh((header)->magic)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering#define HEADER_VERSION(header) be32toh((header)->version)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering#define HEADER_CLUSTER_BITS(header) be32toh((header)->cluster_bits)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering#define HEADER_CLUSTER_SIZE(header) (1ULL << HEADER_CLUSTER_BITS(header))
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering#define HEADER_L2_BITS(header) (HEADER_CLUSTER_BITS(header) - 3)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering#define HEADER_SIZE(header) be64toh((header)->size)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering#define HEADER_CRYPT_METHOD(header) be32toh((header)->crypt_method)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering#define HEADER_L1_SIZE(header) be32toh((header)->l1_size)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering#define HEADER_L2_SIZE(header) (HEADER_CLUSTER_SIZE(header)/sizeof(uint64_t))
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering#define HEADER_L1_TABLE_OFFSET(header) be64toh((header)->l1_table_offset)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poetteringstatic uint32_t HEADER_HEADER_LENGTH(const Header *h) {
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (HEADER_VERSION(h) < 3)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return offsetof(Header, incompatible_features);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        return be32toh(h->header_length);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering}
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poetteringstatic int copy_cluster(
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                int sfd, uint64_t soffset,
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                int dfd, uint64_t doffset,
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                uint64_t cluster_size,
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                void *buffer) {
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        ssize_t l;
1c7dd82563ff2e71a067aea20d2acb2d0553644bLennart Poettering        int r;
1c7dd82563ff2e71a067aea20d2acb2d0553644bLennart Poettering
1c7dd82563ff2e71a067aea20d2acb2d0553644bLennart Poettering        r = btrfs_clone_range(sfd, soffset, dfd, doffset, cluster_size);
1c7dd82563ff2e71a067aea20d2acb2d0553644bLennart Poettering        if (r >= 0)
1c7dd82563ff2e71a067aea20d2acb2d0553644bLennart Poettering                return r;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        l = pread(sfd, buffer, cluster_size, soffset);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (l < 0)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -errno;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if ((uint64_t) l != cluster_size)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -EIO;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        l = pwrite(dfd, buffer, cluster_size, doffset);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (l < 0)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -errno;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if ((uint64_t) l != cluster_size)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -EIO;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        return 0;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering}
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poetteringstatic int decompress_cluster(
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                int sfd, uint64_t soffset,
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                int dfd, uint64_t doffset,
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                uint64_t compressed_size,
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                uint64_t cluster_size,
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                void *buffer1,
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                void *buffer2) {
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        _cleanup_free_ void *large_buffer = NULL;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        z_stream s = {};
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        uint64_t sz;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        ssize_t l;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        int r;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (compressed_size > cluster_size) {
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                /* The usual cluster buffer doesn't suffice, let's
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                 * allocate a larger one, temporarily */
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                large_buffer = malloc(compressed_size);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                if (!large_buffer)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                        return -ENOMEM;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                buffer1 = large_buffer;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        }
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        l = pread(sfd, buffer1, compressed_size, soffset);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (l < 0)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -errno;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if ((uint64_t) l != compressed_size)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -EIO;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        s.next_in = buffer1;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        s.avail_in = compressed_size;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        s.next_out = buffer2;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        s.avail_out = cluster_size;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        r = inflateInit2(&s, -12);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (r != Z_OK)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -EIO;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        r = inflate(&s, Z_FINISH);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        sz = (uint8_t*) s.next_out - (uint8_t*) buffer2;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        inflateEnd(&s);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (r != Z_STREAM_END || sz != cluster_size)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -EIO;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        l = pwrite(dfd, buffer2, cluster_size, doffset);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (l < 0)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -errno;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if ((uint64_t) l != cluster_size)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -EIO;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        return 0;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering}
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poetteringstatic int normalize_offset(
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                const Header *header,
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                uint64_t p,
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                uint64_t *ret,
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                bool *compressed,
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                uint64_t *compressed_size) {
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        uint64_t q;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        q = be64toh(p);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (q & QCOW2_COMPRESSED) {
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                uint64_t sz, csize_shift, csize_mask;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                if (!compressed)
15411c0cb1192799b37ec8f25d6f30e8d7292fc6David Herrmann                        return -EOPNOTSUPP;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                csize_shift = 64 - 2 - (HEADER_CLUSTER_BITS(header) - 8);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                csize_mask = (1ULL << (HEADER_CLUSTER_BITS(header) - 8)) - 1;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                sz = (((q >> csize_shift) & csize_mask) + 1) * 512 - (q & 511);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                q &= ((1ULL << csize_shift) - 1);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                if (compressed_size)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                        *compressed_size = sz;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                *compressed = true;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        } else {
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                if (compressed)  {
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                        *compressed = false;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                        *compressed_size = 0;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                }
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                if (q & QCOW2_ZERO) {
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                        /* We make no distinction between zero blocks and holes */
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                        *ret = 0;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                        return 0;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                }
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                q &= ~QCOW2_COPIED;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        }
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        *ret = q;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        return q > 0;  /* returns positive if not a hole */
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering}
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poetteringstatic int verify_header(const Header *header) {
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        assert(header);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (HEADER_MAGIC(header) != QCOW2_MAGIC)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -EBADMSG;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (HEADER_VERSION(header) != 2 &&
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering            HEADER_VERSION(header) != 3)
15411c0cb1192799b37ec8f25d6f30e8d7292fc6David Herrmann                return -EOPNOTSUPP;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (HEADER_CRYPT_METHOD(header) != 0)
15411c0cb1192799b37ec8f25d6f30e8d7292fc6David Herrmann                return -EOPNOTSUPP;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (HEADER_CLUSTER_BITS(header) < 9) /* 512K */
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -EBADMSG;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (HEADER_CLUSTER_BITS(header) > 21) /* 2MB */
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -EBADMSG;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (HEADER_SIZE(header) % HEADER_CLUSTER_SIZE(header) != 0)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -EBADMSG;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (HEADER_L1_SIZE(header) > 32*1024*1024) /* 32MB */
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -EBADMSG;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (HEADER_VERSION(header) == 3) {
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                if (header->incompatible_features != 0)
15411c0cb1192799b37ec8f25d6f30e8d7292fc6David Herrmann                        return -EOPNOTSUPP;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                if (HEADER_HEADER_LENGTH(header) < sizeof(Header))
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                        return -EBADMSG;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        }
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        return 0;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering}
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poetteringint qcow2_convert(int qcow2_fd, int raw_fd) {
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        _cleanup_free_ void *buffer1 = NULL, *buffer2 = NULL;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        _cleanup_free_ be64_t *l1_table = NULL, *l2_table = NULL;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        uint64_t sz, i;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        Header header;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        ssize_t l;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        int r;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        l = pread(qcow2_fd, &header, sizeof(header), 0);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (l < 0)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -errno;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (l != sizeof(header))
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -EIO;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        r = verify_header(&header);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (r < 0)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return r;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        l1_table = new(be64_t, HEADER_L1_SIZE(&header));
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (!l1_table)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -ENOMEM;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        l2_table = malloc(HEADER_CLUSTER_SIZE(&header));
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (!l2_table)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -ENOMEM;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        buffer1 = malloc(HEADER_CLUSTER_SIZE(&header));
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (!buffer1)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -ENOMEM;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        buffer2 = malloc(HEADER_CLUSTER_SIZE(&header));
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (!buffer2)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -ENOMEM;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        /* Empty the file if it exists, we rely on zero bits */
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (ftruncate(raw_fd, 0) < 0)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -errno;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (ftruncate(raw_fd, HEADER_SIZE(&header)) < 0)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -errno;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        sz = sizeof(uint64_t) * HEADER_L1_SIZE(&header);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        l = pread(qcow2_fd, l1_table, sz, HEADER_L1_TABLE_OFFSET(&header));
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (l < 0)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -errno;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if ((uint64_t) l != sz)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -EIO;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        for (i = 0; i < HEADER_L1_SIZE(&header); i ++) {
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                uint64_t l2_begin, j;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                r = normalize_offset(&header, l1_table[i], &l2_begin, NULL, NULL);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                if (r < 0)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                        return r;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                if (r == 0)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                        continue;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                l = pread(qcow2_fd, l2_table, HEADER_CLUSTER_SIZE(&header), l2_begin);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                if (l < 0)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                        return -errno;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                if ((uint64_t) l != HEADER_CLUSTER_SIZE(&header))
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                        return -EIO;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                for (j = 0; j < HEADER_L2_SIZE(&header); j++) {
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                        uint64_t data_begin, p, compressed_size;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                        bool compressed;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                        p = ((i << HEADER_L2_BITS(&header)) + j) << HEADER_CLUSTER_BITS(&header);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                        r = normalize_offset(&header, l2_table[j], &data_begin, &compressed, &compressed_size);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                        if (r < 0)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                                return r;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                        if (r == 0)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                                continue;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                        if (compressed)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                                r = decompress_cluster(
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                                                qcow2_fd, data_begin,
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                                                raw_fd, p,
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                                                compressed_size, HEADER_CLUSTER_SIZE(&header),
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                                                buffer1, buffer2);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                        else
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                                r = copy_cluster(
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                                                qcow2_fd, data_begin,
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                                                raw_fd, p,
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                                                HEADER_CLUSTER_SIZE(&header), buffer1);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                        if (r < 0)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                                return r;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                }
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        }
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        return 0;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering}
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poetteringint qcow2_detect(int fd) {
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        be32_t id;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        ssize_t l;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        l = pread(fd, &id, sizeof(id), 0);
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (l < 0)
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -errno;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        if (l != sizeof(id))
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering                return -EIO;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering        return htobe32(QCOW2_MAGIC) == id;
edce2aed3aa93b84f7b4c70412bdb665da2977b0Lennart Poettering}