diff --git a/image/main.c b/image/main.c index b29e68f8..171072c4 100644 --- a/image/main.c +++ b/image/main.c @@ -45,6 +45,19 @@ #define MAX_WORKER_THREADS (32) +const struct dump_version dump_versions[] = { + /* + * The original format, which only supports tree blocks and free space + * cache dump. + */ + { .version = 0, + .max_pending_size = SZ_256K, + .magic_cpu = 0xbd5c25e27295668bULL, + .extra_sb_flags = 1 }, +}; + +const struct dump_version *current_version = &dump_versions[0]; + struct async_work { struct list_head list; struct list_head ordered; @@ -406,7 +419,7 @@ static void meta_cluster_init(struct metadump_struct *md, u64 start) md->num_items = 0; md->num_ready = 0; header = &md->cluster.header; - header->magic = cpu_to_le64(HEADER_MAGIC); + header->magic = cpu_to_le64(current_version->magic_cpu); header->bytenr = cpu_to_le64(start); header->nritems = cpu_to_le32(0); header->compress = md->compress_level > 0 ? @@ -718,7 +731,7 @@ static int add_extent(u64 start, u64 size, struct metadump_struct *md, { int ret; if (md->data != data || - md->pending_size + size > MAX_PENDING_SIZE || + md->pending_size + size > current_version->max_pending_size || md->pending_start + md->pending_size != start) { ret = flush_pending(md, 0); if (ret) @@ -1047,7 +1060,8 @@ static void update_super_old(u8 *buffer) u32 sectorsize = btrfs_super_sectorsize(super); u64 flags = btrfs_super_flags(super); - flags |= BTRFS_SUPER_FLAG_METADUMP; + if (current_version->extra_sb_flags) + flags |= BTRFS_SUPER_FLAG_METADUMP; btrfs_set_super_flags(super, flags); key = (struct btrfs_disk_key *)(super->sys_chunk_array); @@ -1147,7 +1161,8 @@ finish: if (mdres->clear_space_cache) btrfs_set_super_cache_generation(super, 0); - flags |= BTRFS_SUPER_FLAG_METADUMP_V2; + if (current_version->extra_sb_flags) + flags |= BTRFS_SUPER_FLAG_METADUMP_V2; btrfs_set_super_flags(super, flags); btrfs_set_super_sys_array_size(super, new_array_size); btrfs_set_super_num_devices(super, 1); @@ -1337,7 +1352,7 @@ static void *restore_worker(void *data) u8 *outbuf; int outfd; int ret; - int compress_size = MAX_PENDING_SIZE * 4; + int compress_size = current_version->max_pending_size * 4; outfd = fileno(mdres->out); buffer = malloc(compress_size); @@ -1490,6 +1505,42 @@ static void mdrestore_destroy(struct mdrestore_struct *mdres, int num_threads) free(mdres->original_super); } +static int detect_version(FILE *in) +{ + struct meta_cluster *cluster; + u8 buf[BLOCK_SIZE]; + bool found = false; + int i; + int ret; + + if (fseek(in, 0, SEEK_SET) < 0) { + error("seek failed: %m"); + return -errno; + } + ret = fread(buf, BLOCK_SIZE, 1, in); + if (!ret) { + error("failed to read header"); + return -EIO; + } + + fseek(in, 0, SEEK_SET); + cluster = (struct meta_cluster *)buf; + for (i = 0; i < ARRAY_SIZE(dump_versions); i++) { + if (le64_to_cpu(cluster->header.magic) == + dump_versions[i].magic_cpu) { + found = true; + current_version = &dump_versions[i]; + break; + } + } + + if (!found) { + error("unrecognized header format"); + return -EINVAL; + } + return 0; +} + static int mdrestore_init(struct mdrestore_struct *mdres, FILE *in, FILE *out, int old_restore, int num_threads, int fixup_offset, @@ -1497,6 +1548,9 @@ static int mdrestore_init(struct mdrestore_struct *mdres, { int i, ret = 0; + ret = detect_version(in); + if (ret < 0) + return ret; memset(mdres, 0, sizeof(*mdres)); pthread_cond_init(&mdres->cond, NULL); pthread_mutex_init(&mdres->mutex, NULL); @@ -1850,7 +1904,7 @@ static int search_for_chunk_blocks(struct mdrestore_struct *mdres) u64 current_cluster = 0, bytenr; u64 item_bytenr; u32 bufsize, nritems, i; - u32 max_size = MAX_PENDING_SIZE * 2; + u32 max_size = current_version->max_pending_size * 2; u8 *buffer, *tmp = NULL; int ret = 0; @@ -1903,7 +1957,7 @@ static int search_for_chunk_blocks(struct mdrestore_struct *mdres) ret = 0; header = &cluster->header; - if (le64_to_cpu(header->magic) != HEADER_MAGIC || + if (le64_to_cpu(header->magic) != current_version->magic_cpu || le64_to_cpu(header->bytenr) != current_cluster) { error("bad header in metadump image"); ret = -EIO; @@ -2102,7 +2156,7 @@ static int build_chunk_tree(struct mdrestore_struct *mdres, ret = 0; header = &cluster->header; - if (le64_to_cpu(header->magic) != HEADER_MAGIC || + if (le64_to_cpu(header->magic) != current_version->magic_cpu || le64_to_cpu(header->bytenr) != 0) { error("bad header in metadump image"); return -EIO; @@ -2675,7 +2729,7 @@ static int restore_metadump(const char *input, FILE *out, int old_restore, break; header = &cluster->header; - if (le64_to_cpu(header->magic) != HEADER_MAGIC || + if (le64_to_cpu(header->magic) != current_version->magic_cpu || le64_to_cpu(header->bytenr) != bytenr) { error("bad header in metadump image"); ret = -EIO; diff --git a/image/metadump.h b/image/metadump.h index 57bc3bf2..bcffbd47 100644 --- a/image/metadump.h +++ b/image/metadump.h @@ -22,8 +22,6 @@ #include "kernel-lib/list.h" #include "kernel-shared/ctree.h" -#define HEADER_MAGIC 0xbd5c25e27295668bULL -#define MAX_PENDING_SIZE SZ_256K #define BLOCK_SIZE SZ_1K #define BLOCK_MASK (BLOCK_SIZE - 1) @@ -33,6 +31,16 @@ #define COMPRESS_NONE 0 #define COMPRESS_ZLIB 1 +struct dump_version { + u64 magic_cpu; + int version; + int max_pending_size; + unsigned int extra_sb_flags:1; +}; + +extern const struct dump_version dump_versions[]; +const extern struct dump_version *current_version; + struct meta_cluster_item { __le64 bytenr; __le32 size;