2
* e2image.c --- Program which writes an image file backing up
3
* critical metadata for the filesystem.
5
* Copyright 2000, 2001 by Theodore Ts'o.
8
* This file may be redistributed under the terms of the GNU Public
13
#define _LARGEFILE_SOURCE
14
#define _LARGEFILE64_SOURCE
36
#include <sys/types.h>
39
#include "ext2fs/ext2_fs.h"
40
#include "ext2fs/ext2fs.h"
41
#include "et/com_err.h"
42
#include "uuid/uuid.h"
44
#include "ext2fs/e2image.h"
45
#include "ext2fs/qcow2.h"
47
#include "../version.h"
48
#include "nls-enable.h"
50
#define QCOW_OFLAG_COPIED (1LL << 63)
53
const char * program_name = "e2image";
54
char * device_name = NULL;
56
static void lseek_error_and_exit(int errnum)
62
static blk64_t align_offset(blk64_t offset, int n)
64
return (offset + n - 1) & ~(n - 1);
67
static int get_bits_from_size(size_t size)
75
/* Not a power of two */
85
static void usage(void)
87
fprintf(stderr, _("Usage: %s [-rsIQ] device image_file\n"),
92
static void generic_write(int fd, void *buf, int blocksize, blk64_t block)
94
int count, free_buf = 0;
102
err = ext2fs_get_arrayzero(1, blocksize, &buf);
104
com_err(program_name, err, "while allocating buffer");
109
count = write(fd, buf, blocksize);
110
if (count != blocksize) {
117
com_err(program_name, err, "error writing block %llu",
120
com_err(program_name, err, "error in write()");
125
ext2fs_free_mem(&buf);
128
static void write_header(int fd, void *hdr, int hdr_size, int wrt_size)
134
if (hdr_size > wrt_size) {
135
fprintf(stderr, _("Error: header size is bigger than "
139
ret = ext2fs_get_mem(wrt_size, &header_buf);
141
fputs(_("Couldn't allocate header buffer\n"), stderr);
145
if (ext2fs_llseek(fd, 0, SEEK_SET) < 0) {
146
perror("ext2fs_llseek while writing header");
149
memset(header_buf, 0, wrt_size);
152
memcpy(header_buf, hdr, hdr_size);
154
generic_write(fd, header_buf, wrt_size, 0);
156
ext2fs_free_mem(&header_buf);
159
static void write_image_file(ext2_filsys fs, int fd)
161
struct ext2_image_hdr hdr;
165
write_header(fd, NULL, fs->blocksize, fs->blocksize);
166
memset(&hdr, 0, sizeof(struct ext2_image_hdr));
168
hdr.offset_super = ext2fs_llseek(fd, 0, SEEK_CUR);
169
retval = ext2fs_image_super_write(fs, fd, 0);
171
com_err(program_name, retval, _("while writing superblock"));
175
hdr.offset_inode = ext2fs_llseek(fd, 0, SEEK_CUR);
176
retval = ext2fs_image_inode_write(fs, fd,
177
(fd != 1) ? IMAGER_FLAG_SPARSEWRITE : 0);
179
com_err(program_name, retval, _("while writing inode table"));
183
hdr.offset_blockmap = ext2fs_llseek(fd, 0, SEEK_CUR);
184
retval = ext2fs_image_bitmap_write(fs, fd, 0);
186
com_err(program_name, retval, _("while writing block bitmap"));
190
hdr.offset_inodemap = ext2fs_llseek(fd, 0, SEEK_CUR);
191
retval = ext2fs_image_bitmap_write(fs, fd, IMAGER_FLAG_INODEMAP);
193
com_err(program_name, retval, _("while writing inode bitmap"));
197
hdr.magic_number = EXT2_ET_MAGIC_E2IMAGE;
198
strcpy(hdr.magic_descriptor, "Ext2 Image 1.0");
199
gethostname(hdr.fs_hostname, sizeof(hdr.fs_hostname));
200
strncpy(hdr.fs_device_name, device_name, sizeof(hdr.fs_device_name)-1);
201
hdr.fs_device_name[sizeof(hdr.fs_device_name) - 1] = 0;
202
hdr.fs_blocksize = fs->blocksize;
204
if (stat(device_name, &st) == 0)
205
hdr.fs_device = st.st_rdev;
207
if (fstat(fd, &st) == 0) {
208
hdr.image_device = st.st_dev;
209
hdr.image_inode = st.st_ino;
211
memcpy(hdr.fs_uuid, fs->super->s_uuid, sizeof(hdr.fs_uuid));
213
hdr.image_time = time(0);
214
write_header(fd, &hdr, fs->blocksize, fs->blocksize);
218
* These set of functions are used to write a RAW image file.
220
ext2fs_block_bitmap meta_block_map;
221
ext2fs_block_bitmap scramble_block_map; /* Directory blocks to be scrambled */
222
blk64_t meta_blocks_count;
224
struct process_block_struct {
230
* These subroutines short circuits ext2fs_get_blocks and
231
* ext2fs_check_directory; we use them since we already have the inode
232
* structure, so there's no point in letting the ext2fs library read
235
static ino_t stashed_ino = 0;
236
static struct ext2_inode *stashed_inode;
238
static errcode_t meta_get_blocks(ext2_filsys fs EXT2FS_ATTR((unused)),
244
if ((ino != stashed_ino) || !stashed_inode)
245
return EXT2_ET_CALLBACK_NOTHANDLED;
247
for (i=0; i < EXT2_N_BLOCKS; i++)
248
blocks[i] = stashed_inode->i_block[i];
252
static errcode_t meta_check_directory(ext2_filsys fs EXT2FS_ATTR((unused)),
255
if ((ino != stashed_ino) || !stashed_inode)
256
return EXT2_ET_CALLBACK_NOTHANDLED;
258
if (!LINUX_S_ISDIR(stashed_inode->i_mode))
259
return EXT2_ET_NO_DIRECTORY;
263
static errcode_t meta_read_inode(ext2_filsys fs EXT2FS_ATTR((unused)),
265
struct ext2_inode *inode)
267
if ((ino != stashed_ino) || !stashed_inode)
268
return EXT2_ET_CALLBACK_NOTHANDLED;
269
*inode = *stashed_inode;
273
static void use_inode_shortcuts(ext2_filsys fs, int bool)
276
fs->get_blocks = meta_get_blocks;
277
fs->check_directory = meta_check_directory;
278
fs->read_inode = meta_read_inode;
282
fs->check_directory = 0;
287
static int process_dir_block(ext2_filsys fs EXT2FS_ATTR((unused)),
289
e2_blkcnt_t blockcnt EXT2FS_ATTR((unused)),
290
blk64_t ref_block EXT2FS_ATTR((unused)),
291
int ref_offset EXT2FS_ATTR((unused)),
292
void *priv_data EXT2FS_ATTR((unused)))
294
struct process_block_struct *p;
296
p = (struct process_block_struct *) priv_data;
298
ext2fs_mark_block_bitmap2(meta_block_map, *block_nr);
300
if (scramble_block_map && p->is_dir && blockcnt >= 0)
301
ext2fs_mark_block_bitmap2(scramble_block_map, *block_nr);
305
static int process_file_block(ext2_filsys fs EXT2FS_ATTR((unused)),
307
e2_blkcnt_t blockcnt,
308
blk64_t ref_block EXT2FS_ATTR((unused)),
309
int ref_offset EXT2FS_ATTR((unused)),
310
void *priv_data EXT2FS_ATTR((unused)))
313
ext2fs_mark_block_bitmap2(meta_block_map, *block_nr);
319
static void mark_table_blocks(ext2_filsys fs)
321
blk64_t first_block, b;
324
first_block = fs->super->s_first_data_block;
326
* Mark primary superblock
328
ext2fs_mark_block_bitmap2(meta_block_map, first_block);
332
* Mark the primary superblock descriptors
334
for (j = 0; j < fs->desc_blocks; j++) {
335
ext2fs_mark_block_bitmap2(meta_block_map,
336
ext2fs_descriptor_block_loc2(fs, first_block, j));
338
meta_blocks_count += fs->desc_blocks;
340
for (i = 0; i < fs->group_desc_count; i++) {
342
* Mark the blocks used for the inode table
344
if (!ext2fs_bg_flags_test(fs, i, EXT2_BG_INODE_UNINIT) &&
345
ext2fs_inode_table_loc(fs, i)) {
346
unsigned int end = (unsigned) fs->inode_blocks_per_group;
347
/* skip unused blocks */
348
if (EXT2_HAS_RO_COMPAT_FEATURE(fs->super,
349
EXT4_FEATURE_RO_COMPAT_GDT_CSUM))
350
end -= (ext2fs_bg_itable_unused(fs, i) /
351
EXT2_INODES_PER_BLOCK(fs->super));
352
for (j = 0, b = ext2fs_inode_table_loc(fs, i);
355
ext2fs_mark_block_bitmap2(meta_block_map, b);
361
* Mark block used for the block bitmap
363
if (!ext2fs_bg_flags_test(fs, i, EXT2_BG_BLOCK_UNINIT) &&
364
ext2fs_block_bitmap_loc(fs, i)) {
365
ext2fs_mark_block_bitmap2(meta_block_map,
366
ext2fs_block_bitmap_loc(fs, i));
371
* Mark block used for the inode bitmap
373
if (!ext2fs_bg_flags_test(fs, i, EXT2_BG_INODE_UNINIT) &&
374
ext2fs_inode_bitmap_loc(fs, i)) {
375
ext2fs_mark_block_bitmap2(meta_block_map,
376
ext2fs_inode_bitmap_loc(fs, i));
383
* This function returns 1 if the specified block is all zeros
385
static int check_zero_block(char *buf, int blocksize)
388
int left = blocksize;
398
static void write_block(int fd, char *buf, int sparse_offset,
399
int blocksize, blk64_t block)
404
ret = ext2fs_llseek(fd, sparse_offset, SEEK_CUR);
407
lseek_error_and_exit(errno);
408
generic_write(fd, buf, blocksize, block);
413
#define EXT4_MAX_REC_LEN ((1<<16)-1)
415
static void scramble_dir_block(ext2_filsys fs, blk64_t blk, char *buf)
418
struct ext2_dir_entry_2 *dirent;
419
unsigned int rec_len;
422
end = buf + fs->blocksize;
423
for (p = buf; p < end-8; p += rec_len) {
424
dirent = (struct ext2_dir_entry_2 *) p;
425
rec_len = dirent->rec_len;
426
#ifdef WORDS_BIGENDIAN
427
rec_len = ext2fs_swab16(rec_len);
429
if (rec_len == EXT4_MAX_REC_LEN || rec_len == 0)
430
rec_len = fs->blocksize;
432
rec_len = (rec_len & 65532) | ((rec_len & 3) << 16);
434
printf("rec_len = %d, name_len = %d\n", rec_len, dirent->name_len);
436
if (rec_len < 8 || (rec_len % 4) ||
438
printf("Corrupt directory block %lu: "
439
"bad rec_len (%d)\n", (unsigned long) blk,
442
(void) ext2fs_set_rec_len(fs, rec_len,
443
(struct ext2_dir_entry *) dirent);
444
#ifdef WORDS_BIGENDIAN
445
dirent->rec_len = ext2fs_swab16(dirent->rec_len);
449
if (dirent->name_len + 8 > rec_len) {
450
printf("Corrupt directory block %lu: "
451
"bad name_len (%d)\n", (unsigned long) blk,
453
dirent->name_len = rec_len - 8;
457
len = rec_len - dirent->name_len - 8;
459
memset(cp+dirent->name_len, 0, len);
460
if (dirent->name_len==1 && cp[0] == '.')
462
if (dirent->name_len==2 && cp[0] == '.' && cp[1] == '.')
465
memset(cp, 'A', dirent->name_len);
466
len = dirent->name_len;
468
while ((len > 0) && (id > 0)) {
477
static void output_meta_data_blocks(ext2_filsys fs, int fd)
481
char *buf, *zero_buf;
484
retval = ext2fs_get_mem(fs->blocksize, &buf);
486
com_err(program_name, retval, "while allocating buffer");
489
retval = ext2fs_get_memzero(fs->blocksize, &zero_buf);
491
com_err(program_name, retval, "while allocating buffer");
494
for (blk = 0; blk < ext2fs_blocks_count(fs->super); blk++) {
495
if ((blk >= fs->super->s_first_data_block) &&
496
ext2fs_test_block_bitmap2(meta_block_map, blk)) {
497
retval = io_channel_read_blk64(fs->io, blk, 1, buf);
499
com_err(program_name, retval,
500
"error reading block %llu", blk);
502
if (scramble_block_map &&
503
ext2fs_test_block_bitmap2(scramble_block_map, blk))
504
scramble_dir_block(fs, blk, buf);
505
if ((fd != 1) && check_zero_block(buf, fs->blocksize))
507
write_block(fd, buf, sparse, fs->blocksize, blk);
512
write_block(fd, zero_buf, 0,
516
sparse += fs->blocksize;
517
if (sparse > 1024*1024) {
518
write_block(fd, 0, 1024*1024, 0, 0);
523
#ifdef HAVE_FTRUNCATE64
525
ext2_loff_t offset = ext2fs_llseek(fd, sparse, SEEK_CUR);
528
lseek_error_and_exit(errno);
529
if (ftruncate64(fd, offset) < 0)
530
write_block(fd, zero_buf, -1, 1, -1);
534
write_block(fd, zero_buf, sparse-1, 1, -1);
536
ext2fs_free_mem(&zero_buf);
537
ext2fs_free_mem(&buf);
540
static void init_l1_table(struct ext2_super_block *sb,
541
struct ext2_qcow2_image *image)
546
ret = ext2fs_get_arrayzero(image->l1_size, sizeof(__u64), &l1_table);
548
com_err(program_name, ret, "while allocating l1 table");
552
image->l1_table = l1_table;
555
static void init_l2_cache(struct ext2_qcow2_image *image)
557
unsigned int count, i;
558
struct ext2_qcow2_l2_cache *cache;
559
struct ext2_qcow2_l2_table *table;
562
ret = ext2fs_get_arrayzero(1, sizeof(struct ext2_qcow2_l2_cache),
567
count = (image->l1_size > L2_CACHE_PREALLOC) ? L2_CACHE_PREALLOC :
570
cache->count = count;
572
cache->next_offset = image->l2_offset;
574
for (i = 0; i < count; i++) {
575
ret = ext2fs_get_arrayzero(1,
576
sizeof(struct ext2_qcow2_l2_table), &table);
580
ret = ext2fs_get_arrayzero(image->l2_size,
581
sizeof(__u64), &table->data);
585
table->next = cache->free_head;
586
cache->free_head = table;
589
image->l2_cache = cache;
593
com_err(program_name, ret, "while allocating l2 cache");
597
static void put_l2_cache(struct ext2_qcow2_image *image)
599
struct ext2_qcow2_l2_cache *cache = image->l2_cache;
600
struct ext2_qcow2_l2_table *tmp, *table;
605
table = cache->free_head;
606
cache->free_head = NULL;
611
ext2fs_free_mem(&tmp->data);
612
ext2fs_free_mem(&tmp);
615
if (cache->free != cache->count) {
616
fprintf(stderr, "Warning: There are still tables in the "
617
"cache while putting the cache, data will "
618
"be lost so the image may not be valid.\n");
619
table = cache->used_head;
620
cache->used_head = NULL;
624
ext2fs_free_mem(&cache);
627
static int init_refcount(struct ext2_qcow2_image *img, blk64_t table_offset)
629
struct ext2_qcow2_refcount *ref;
630
blk64_t table_clusters;
633
ref = &(img->refcount);
636
* One refcount block addresses 2048 clusters, one refcount table
637
* addresses cluster/sizeof(__u64) refcount blocks, and we need
638
* to address meta_blocks_count clusters + qcow2 metadata clusters
641
table_clusters = meta_blocks_count + (table_offset >>
643
table_clusters >>= (img->cluster_bits + 6 - 1);
644
table_clusters = (table_clusters == 0) ? 1 : table_clusters;
646
ref->refcount_table_offset = table_offset;
647
ref->refcount_table_clusters = table_clusters;
648
ref->refcount_table_index = 0;
649
ref->refcount_block_index = 0;
651
/* Allocate refcount table */
652
ret = ext2fs_get_arrayzero(ref->refcount_table_clusters,
653
img->cluster_size, &ref->refcount_table);
657
/* Allocate refcount block */
658
ret = ext2fs_get_arrayzero(1, img->cluster_size, &ref->refcount_block);
660
ext2fs_free_mem(&ref->refcount_table);
665
static int initialize_qcow2_image(int fd, ext2_filsys fs,
666
struct ext2_qcow2_image *image)
668
struct ext2_qcow2_hdr *header;
669
blk64_t total_size, offset;
670
int shift, l2_bits, header_size, l1_size, ret;
671
int cluster_bits = get_bits_from_size(fs->blocksize);
672
struct ext2_super_block *sb = fs->super;
674
/* Allocate header */
675
ret = ext2fs_get_memzero(sizeof(struct ext2_qcow2_hdr), &header);
679
total_size = ext2fs_blocks_count(sb) << cluster_bits;
680
image->cluster_size = fs->blocksize;
681
image->l2_size = 1 << (cluster_bits - 3);
682
image->cluster_bits = cluster_bits;
685
header->magic = ext2fs_cpu_to_be32(QCOW_MAGIC);
686
header->version = ext2fs_cpu_to_be32(QCOW_VERSION);
687
header->size = ext2fs_cpu_to_be64(total_size);
688
header->cluster_bits = ext2fs_cpu_to_be32(cluster_bits);
690
header_size = (sizeof(struct ext2_qcow2_hdr) + 7) & ~7;
691
offset = align_offset(header_size, image->cluster_size);
693
header->l1_table_offset = ext2fs_cpu_to_be64(offset);
694
image->l1_offset = offset;
696
l2_bits = cluster_bits - 3;
697
shift = cluster_bits + l2_bits;
698
l1_size = ((total_size + (1LL << shift) - 1) >> shift);
699
header->l1_size = ext2fs_cpu_to_be32(l1_size);
700
image->l1_size = l1_size;
702
/* Make space for L1 table */
703
offset += align_offset(l1_size * sizeof(blk64_t), image->cluster_size);
705
/* Initialize refcounting */
706
ret = init_refcount(image, offset);
708
ext2fs_free_mem(&header);
711
header->refcount_table_offset = ext2fs_cpu_to_be64(offset);
712
header->refcount_table_clusters =
713
ext2fs_cpu_to_be32(image->refcount.refcount_table_clusters);
714
offset += image->cluster_size;
715
offset += image->refcount.refcount_table_clusters <<
718
/* Make space for L2 tables */
719
image->l2_offset = offset;
720
offset += image->cluster_size;
722
/* Make space for first refcount block */
723
image->refcount.refcount_block_offset = offset;
726
/* Initialize l1 and l2 tables */
727
init_l1_table(sb, image);
728
init_l2_cache(image);
733
static void free_qcow2_image(struct ext2_qcow2_image *img)
739
ext2fs_free_mem(&img->hdr);
742
ext2fs_free_mem(&img->l1_table);
744
if (img->refcount.refcount_table)
745
ext2fs_free_mem(&img->refcount.refcount_table);
746
if (img->refcount.refcount_block)
747
ext2fs_free_mem(&img->refcount.refcount_block);
751
ext2fs_free_mem(&img);
755
* Put table from used list (used_head) into free list (free_head).
756
* l2_table is used to return pointer to the next used table (used_head).
758
static void put_used_table(struct ext2_qcow2_image *img,
759
struct ext2_qcow2_l2_table **l2_table)
761
struct ext2_qcow2_l2_cache *cache = img->l2_cache;
762
struct ext2_qcow2_l2_table *table;
764
table = cache->used_head;
765
cache->used_head = table->next;
769
cache->used_tail = NULL;
771
/* Clean the table for case we will need to use it again */
772
memset(table->data, 0, img->cluster_size);
773
table->next = cache->free_head;
774
cache->free_head = table;
778
*l2_table = cache->used_head;
781
static void flush_l2_cache(struct ext2_qcow2_image *image)
785
struct ext2_qcow2_l2_cache *cache = image->l2_cache;
786
struct ext2_qcow2_l2_table *table = cache->used_head;
789
/* Store current position */
790
if ((offset = ext2fs_llseek(fd, 0, SEEK_CUR)) < 0)
791
lseek_error_and_exit(errno);
794
while (cache->free < cache->count) {
795
if (seek != table->offset) {
796
if (ext2fs_llseek(fd, table->offset, SEEK_SET) < 0)
797
lseek_error_and_exit(errno);
798
seek = table->offset;
801
generic_write(fd, (char *)table->data, image->cluster_size , 0);
802
put_used_table(image, &table);
803
seek += image->cluster_size;
806
/* Restore previous position */
807
if (ext2fs_llseek(fd, offset, SEEK_SET) < 0)
808
lseek_error_and_exit(errno);
812
* Get first free table (from free_head) and put it into tail of used list
814
* l2_table is used to return pointer to moved table.
815
* Returns 1 if the cache is full, 0 otherwise.
817
static void get_free_table(struct ext2_qcow2_image *image,
818
struct ext2_qcow2_l2_table **l2_table)
820
struct ext2_qcow2_l2_table *table;
821
struct ext2_qcow2_l2_cache *cache = image->l2_cache;
823
if (0 == cache->free)
824
flush_l2_cache(image);
826
table = cache->free_head;
828
cache->free_head = table->next;
830
if (cache->used_tail)
831
cache->used_tail->next = table;
833
/* First item in the used list */
834
cache->used_head = table;
836
cache->used_tail = table;
842
static int add_l2_item(struct ext2_qcow2_image *img, blk64_t blk,
843
blk64_t data, blk64_t next)
845
struct ext2_qcow2_l2_cache *cache = img->l2_cache;
846
struct ext2_qcow2_l2_table *table = cache->used_tail;
847
blk64_t l1_index = blk / img->l2_size;
848
blk64_t l2_index = blk & (img->l2_size - 1);
852
* Need to create new table if it does not exist,
855
if (!table || (table->l1_index != l1_index)) {
856
get_free_table(img, &table);
857
table->l1_index = l1_index;
858
table->offset = cache->next_offset;
859
cache->next_offset = next;
860
img->l1_table[l1_index] =
861
ext2fs_cpu_to_be64(table->offset | QCOW_OFLAG_COPIED);
865
table->data[l2_index] = ext2fs_cpu_to_be64(data | QCOW_OFLAG_COPIED);
869
static int update_refcount(int fd, struct ext2_qcow2_image *img,
870
blk64_t offset, blk64_t rfblk_pos)
872
struct ext2_qcow2_refcount *ref;
876
ref = &(img->refcount);
877
table_index = offset >> (2 * img->cluster_bits - 1);
880
* Need to create new refcount block when the offset addresses
881
* another item in the refcount table
883
if (table_index != ref->refcount_table_index) {
885
if (ext2fs_llseek(fd, ref->refcount_block_offset, SEEK_SET) < 0)
886
lseek_error_and_exit(errno);
888
generic_write(fd, (char *)ref->refcount_block,
889
img->cluster_size, 0);
890
memset(ref->refcount_block, 0, img->cluster_size);
892
ref->refcount_table[ref->refcount_table_index] =
893
ext2fs_cpu_to_be64(ref->refcount_block_offset);
894
ref->refcount_block_offset = rfblk_pos;
895
ref->refcount_block_index = 0;
896
ref->refcount_table_index = table_index;
901
* We are relying on the fact that we are creating the qcow2
902
* image sequentially, hence we will always allocate refcount
903
* block items sequentialy.
905
ref->refcount_block[ref->refcount_block_index] = ext2fs_cpu_to_be16(1);
906
ref->refcount_block_index++;
910
static int sync_refcount(int fd, struct ext2_qcow2_image *img)
912
struct ext2_qcow2_refcount *ref;
914
ref = &(img->refcount);
916
ref->refcount_table[ref->refcount_table_index] =
917
ext2fs_cpu_to_be64(ref->refcount_block_offset);
918
if (ext2fs_llseek(fd, ref->refcount_table_offset, SEEK_SET) < 0)
919
lseek_error_and_exit(errno);
920
generic_write(fd, (char *)ref->refcount_table,
921
ref->refcount_table_clusters << img->cluster_bits, 0);
923
if (ext2fs_llseek(fd, ref->refcount_block_offset, SEEK_SET) < 0)
924
lseek_error_and_exit(errno);
925
generic_write(fd, (char *)ref->refcount_block, img->cluster_size, 0);
929
static void output_qcow2_meta_data_blocks(ext2_filsys fs, int fd)
932
blk64_t blk, offset, size, end;
934
struct ext2_qcow2_image *img;
935
unsigned int header_size;
937
/* allocate struct ext2_qcow2_image */
938
retval = ext2fs_get_mem(sizeof(struct ext2_qcow2_image), &img);
940
com_err(program_name, retval,
941
"while allocating ext2_qcow2_image");
945
retval = initialize_qcow2_image(fd, fs, img);
947
com_err(program_name, retval,
948
"while initializing ext2_qcow2_image");
951
header_size = align_offset(sizeof(struct ext2_qcow2_hdr),
953
write_header(fd, img->hdr, sizeof(struct ext2_qcow2_hdr), header_size);
955
/* Refcount all qcow2 related metadata up to refcount_block_offset */
956
end = img->refcount.refcount_block_offset;
957
if (ext2fs_llseek(fd, end, SEEK_SET) < 0)
958
lseek_error_and_exit(errno);
959
blk = end + img->cluster_size;
960
for (offset = 0; offset <= end; offset += img->cluster_size) {
961
if (update_refcount(fd, img, offset, blk)) {
962
blk += img->cluster_size;
964
* If we create new refcount block, we need to refcount
967
end += img->cluster_size;
970
if (ext2fs_llseek(fd, offset, SEEK_SET) < 0)
971
lseek_error_and_exit(errno);
973
retval = ext2fs_get_mem(fs->blocksize, &buf);
975
com_err(program_name, retval, "while allocating buffer");
978
/* Write qcow2 data blocks */
979
for (blk = 0; blk < ext2fs_blocks_count(fs->super); blk++) {
980
if ((blk >= fs->super->s_first_data_block) &&
981
ext2fs_test_block_bitmap2(meta_block_map, blk)) {
982
retval = io_channel_read_blk64(fs->io, blk, 1, buf);
984
com_err(program_name, retval,
985
"error reading block %llu", blk);
988
if (scramble_block_map &&
989
ext2fs_test_block_bitmap2(scramble_block_map, blk))
990
scramble_dir_block(fs, blk, buf);
991
if (check_zero_block(buf, fs->blocksize))
994
if (update_refcount(fd, img, offset, offset)) {
995
/* Make space for another refcount block */
996
offset += img->cluster_size;
997
if (ext2fs_llseek(fd, offset, SEEK_SET) < 0)
998
lseek_error_and_exit(errno);
1000
* We have created the new refcount block, this
1001
* means that we need to refcount it as well.
1002
* So the previous update_refcount refcounted
1003
* the block itself and now we are going to
1004
* create refcount for data. New refcount
1005
* block should not be created!
1007
if (update_refcount(fd, img, offset, offset)) {
1008
fprintf(stderr, "Programming error: "
1009
"multiple sequential refcount "
1010
"blocks created!\n");
1015
generic_write(fd, buf, fs->blocksize, 0);
1017
if (add_l2_item(img, blk, offset,
1018
offset + img->cluster_size)) {
1019
offset += img->cluster_size;
1020
if (update_refcount(fd, img, offset,
1021
offset + img->cluster_size)) {
1022
offset += img->cluster_size;
1023
if (update_refcount(fd, img, offset,
1026
"Programming error: multiple sequential refcount "
1027
"blocks created!\n");
1031
offset += img->cluster_size;
1032
if (ext2fs_llseek(fd, offset, SEEK_SET) < 0)
1033
lseek_error_and_exit(errno);
1037
offset += img->cluster_size;
1040
update_refcount(fd, img, offset, offset);
1041
flush_l2_cache(img);
1042
sync_refcount(fd, img);
1045
if (ext2fs_llseek(fd, img->l1_offset, SEEK_SET) < 0)
1046
lseek_error_and_exit(errno);
1047
size = img->l1_size * sizeof(__u64);
1048
generic_write(fd, (char *)img->l1_table, size, 0);
1050
ext2fs_free_mem(&buf);
1051
free_qcow2_image(img);
1054
static void write_raw_image_file(ext2_filsys fs, int fd, int type, int flags)
1056
struct process_block_struct pb;
1057
struct ext2_inode inode;
1058
ext2_inode_scan scan;
1063
meta_blocks_count = 0;
1064
retval = ext2fs_allocate_block_bitmap(fs, "in-use block map",
1067
com_err(program_name, retval, "while allocating block bitmap");
1071
if (flags & E2IMAGE_SCRAMBLE_FLAG) {
1072
retval = ext2fs_allocate_block_bitmap(fs, "scramble block map",
1073
&scramble_block_map);
1075
com_err(program_name, retval,
1076
"while allocating scramble block bitmap");
1081
mark_table_blocks(fs);
1083
retval = ext2fs_open_inode_scan(fs, 0, &scan);
1085
com_err(program_name, retval, _("while opening inode scan"));
1089
retval = ext2fs_get_mem(fs->blocksize * 3, &block_buf);
1091
com_err(program_name, 0, "Can't allocate block buffer");
1095
use_inode_shortcuts(fs, 1);
1096
stashed_inode = &inode;
1098
retval = ext2fs_get_next_inode(scan, &ino, &inode);
1099
if (retval == EXT2_ET_BAD_BLOCK_IN_INODE_TABLE)
1102
com_err(program_name, retval,
1103
_("while getting next inode"));
1108
if (!inode.i_links_count)
1110
if (ext2fs_file_acl_block(fs, &inode)) {
1111
ext2fs_mark_block_bitmap2(meta_block_map,
1112
ext2fs_file_acl_block(fs, &inode));
1113
meta_blocks_count++;
1115
if (!ext2fs_inode_has_valid_blocks2(fs, &inode))
1120
pb.is_dir = LINUX_S_ISDIR(inode.i_mode);
1121
if (LINUX_S_ISDIR(inode.i_mode) ||
1122
(LINUX_S_ISLNK(inode.i_mode) &&
1123
ext2fs_inode_has_valid_blocks2(fs, &inode)) ||
1124
ino == fs->super->s_journal_inum) {
1125
retval = ext2fs_block_iterate3(fs, ino,
1126
BLOCK_FLAG_READ_ONLY, block_buf,
1127
process_dir_block, &pb);
1129
com_err(program_name, retval,
1130
"while iterating over inode %u",
1135
if ((inode.i_flags & EXT4_EXTENTS_FL) ||
1136
inode.i_block[EXT2_IND_BLOCK] ||
1137
inode.i_block[EXT2_DIND_BLOCK] ||
1138
inode.i_block[EXT2_TIND_BLOCK]) {
1139
retval = ext2fs_block_iterate3(fs,
1140
ino, BLOCK_FLAG_READ_ONLY, block_buf,
1141
process_file_block, &pb);
1143
com_err(program_name, retval,
1144
"while iterating over inode %u", ino);
1150
use_inode_shortcuts(fs, 0);
1152
if (type & E2IMAGE_QCOW2)
1153
output_qcow2_meta_data_blocks(fs, fd);
1155
output_meta_data_blocks(fs, fd);
1157
ext2fs_free_mem(&block_buf);
1158
ext2fs_close_inode_scan(scan);
1159
ext2fs_free_block_bitmap(meta_block_map);
1160
if (type & E2IMAGE_SCRAMBLE_FLAG)
1161
ext2fs_free_block_bitmap(scramble_block_map);
1164
static void install_image(char *device, char *image_fn, int type)
1168
int open_flag = EXT2_FLAG_IMAGE_FILE;
1174
com_err(program_name, 0, "Raw and qcow2 images cannot"
1179
#ifdef CONFIG_TESTIO_DEBUG
1180
if (getenv("TEST_IO_FLAGS") || getenv("TEST_IO_BLOCK")) {
1181
io_ptr = test_io_manager;
1182
test_io_backing_manager = unix_io_manager;
1185
io_ptr = unix_io_manager;
1187
retval = ext2fs_open (image_fn, open_flag, 0, 0,
1190
com_err (program_name, retval, _("while trying to open %s"),
1195
retval = ext2fs_read_bitmaps (fs);
1197
com_err(program_name, retval, "error reading bitmaps");
1201
fd = ext2fs_open_file(image_fn, O_RDONLY, 0);
1207
retval = io_ptr->open(device, IO_FLAG_RW, &io);
1209
com_err(device, 0, "while opening device file");
1213
ext2fs_rewrite_to_io(fs, io);
1215
if (ext2fs_llseek(fd, fs->image_header->offset_inode, SEEK_SET) < 0) {
1216
perror("ext2fs_llseek");
1220
retval = ext2fs_image_inode_read(fs, fd, 0);
1222
com_err(image_fn, 0, "while restoring the image table");
1230
static struct ext2_qcow2_hdr *check_qcow2_image(int *fd, char *name)
1233
*fd = ext2fs_open_file(name, O_RDONLY, 0600);
1237
return qcow2_read_header(*fd);
1240
int main (int argc, char ** argv)
1246
struct ext2_qcow2_hdr *header = NULL;
1247
int open_flag = EXT2_FLAG_64BITS;
1255
setlocale(LC_MESSAGES, "");
1256
setlocale(LC_CTYPE, "");
1257
bindtextdomain(NLS_CAT_NAME, LOCALEDIR);
1258
textdomain(NLS_CAT_NAME);
1259
set_com_err_gettext(gettext);
1261
fprintf (stderr, "e2image %s (%s)\n", E2FSPROGS_VERSION,
1264
program_name = *argv;
1265
add_error_table(&et_ext2_error_table);
1266
while ((c = getopt(argc, argv, "rsIQ")) != EOF)
1269
flags |= E2IMAGE_INSTALL_FLAG;
1274
img_type |= E2IMAGE_QCOW2;
1279
img_type |= E2IMAGE_RAW;
1282
flags |= E2IMAGE_SCRAMBLE_FLAG;
1287
if (optind != argc - 2 )
1289
device_name = argv[optind];
1290
image_fn = argv[optind+1];
1292
if (flags & E2IMAGE_INSTALL_FLAG) {
1293
install_image(device_name, image_fn, img_type);
1297
if (img_type & E2IMAGE_RAW) {
1298
header = check_qcow2_image(&qcow2_fd, device_name);
1300
flags |= E2IMAGE_IS_QCOW2_FLAG;
1305
retval = ext2fs_open (device_name, open_flag, 0, 0,
1306
unix_io_manager, &fs);
1308
com_err (program_name, retval, _("while trying to open %s"),
1310
fputs(_("Couldn't find valid filesystem superblock.\n"), stdout);
1315
if (strcmp(image_fn, "-") == 0)
1318
fd = ext2fs_open_file(image_fn, O_CREAT|O_TRUNC|O_WRONLY, 0600);
1320
com_err(program_name, errno,
1321
_("while trying to open %s"), argv[optind+1]);
1326
if ((img_type & E2IMAGE_QCOW2) && (fd == 1)) {
1327
com_err(program_name, 0, "QCOW2 image can not be written to "
1332
if (flags & E2IMAGE_IS_QCOW2_FLAG) {
1333
ret = qcow2_write_raw_image(qcow2_fd, fd, header);
1335
if (ret == -QCOW_COMPRESSED)
1336
fprintf(stderr, "Image (%s) is compressed\n",
1338
if (ret == -QCOW_ENCRYPTED)
1339
fprintf(stderr, "Image (%s) is encrypted\n",
1341
com_err(program_name, ret,
1342
_("while trying to convert qcow2 image"
1343
" (%s) into raw image (%s)"),
1344
device_name, image_fn);
1351
write_raw_image_file(fs, fd, img_type, flags);
1353
write_image_file(fs, fd);
1361
remove_error_table(&et_ext2_error_table);