1/*
2 * e2image.c --- Program which writes an image file backing up
3 * critical metadata for the filesystem.
4 *
5 * Copyright 2000, 2001 by Theodore Ts'o.
6 *
7 * %Begin-Header%
8 * This file may be redistributed under the terms of the GNU Public
9 * License.
10 * %End-Header%
11 */
12
13#define _LARGEFILE_SOURCE
14#define _LARGEFILE64_SOURCE
15
16#include <fcntl.h>
17#include <grp.h>
18#ifdef HAVE_GETOPT_H
19#include <getopt.h>
20#else
21extern char *optarg;
22extern int optind;
23#endif
24#include <pwd.h>
25#include <stdio.h>
26#include <stdlib.h>
27#include <string.h>
28#include <time.h>
29#include <unistd.h>
30#include <fcntl.h>
31#include <errno.h>
32#include <sys/stat.h>
33#include <sys/types.h>
34
35#include "ext2fs/ext2_fs.h"
36#include "ext2fs/ext2fs.h"
37#include "et/com_err.h"
38#include "uuid/uuid.h"
39#include "e2p/e2p.h"
40#include "ext2fs/e2image.h"
41
42#include "../version.h"
43#include "nls-enable.h"
44
45const char * program_name = "e2image";
46char * device_name = NULL;
47
48static void usage(void)
49{
50	fprintf(stderr, _("Usage: %s [-rsI] device image_file\n"),
51		program_name);
52	exit (1);
53}
54
55static void write_header(int fd, struct ext2_image_hdr *hdr, int blocksize)
56{
57	char *header_buf;
58	int actual;
59
60	header_buf = malloc(blocksize);
61	if (!header_buf) {
62		fputs(_("Couldn't allocate header buffer\n"), stderr);
63		exit(1);
64	}
65
66	if (lseek(fd, 0, SEEK_SET) < 0) {
67		perror("lseek while writing header");
68		exit(1);
69	}
70	memset(header_buf, 0, blocksize);
71
72	if (hdr)
73		memcpy(header_buf, hdr, sizeof(struct ext2_image_hdr));
74
75	actual = write(fd, header_buf, blocksize);
76	if (actual < 0) {
77		perror("write header");
78		exit(1);
79	}
80	if (actual != blocksize) {
81		fprintf(stderr, _("short write (only %d bytes) for "
82				  "writing image header"), actual);
83		exit(1);
84	}
85	free(header_buf);
86}
87
88static void write_image_file(ext2_filsys fs, int fd)
89{
90	struct ext2_image_hdr	hdr;
91	struct stat		st;
92	errcode_t		retval;
93
94	write_header(fd, NULL, fs->blocksize);
95	memset(&hdr, 0, sizeof(struct ext2_image_hdr));
96
97	hdr.offset_super = lseek(fd, 0, SEEK_CUR);
98	retval = ext2fs_image_super_write(fs, fd, 0);
99	if (retval) {
100		com_err(program_name, retval, _("while writing superblock"));
101		exit(1);
102	}
103
104	hdr.offset_inode = lseek(fd, 0, SEEK_CUR);
105	retval = ext2fs_image_inode_write(fs, fd,
106				  (fd != 1) ? IMAGER_FLAG_SPARSEWRITE : 0);
107	if (retval) {
108		com_err(program_name, retval, _("while writing inode table"));
109		exit(1);
110	}
111
112	hdr.offset_blockmap = lseek(fd, 0, SEEK_CUR);
113	retval = ext2fs_image_bitmap_write(fs, fd, 0);
114	if (retval) {
115		com_err(program_name, retval, _("while writing block bitmap"));
116		exit(1);
117	}
118
119	hdr.offset_inodemap = lseek(fd, 0, SEEK_CUR);
120	retval = ext2fs_image_bitmap_write(fs, fd, IMAGER_FLAG_INODEMAP);
121	if (retval) {
122		com_err(program_name, retval, _("while writing inode bitmap"));
123		exit(1);
124	}
125
126	hdr.magic_number = EXT2_ET_MAGIC_E2IMAGE;
127	strcpy(hdr.magic_descriptor, "Ext2 Image 1.0");
128	gethostname(hdr.fs_hostname, sizeof(hdr.fs_hostname));
129	strncpy(hdr.fs_device_name, device_name, sizeof(hdr.fs_device_name)-1);
130	hdr.fs_device_name[sizeof(hdr.fs_device_name) - 1] = 0;
131	hdr.fs_blocksize = fs->blocksize;
132
133	if (stat(device_name, &st) == 0)
134		hdr.fs_device = st.st_rdev;
135
136	if (fstat(fd, &st) == 0) {
137		hdr.image_device = st.st_dev;
138		hdr.image_inode = st.st_ino;
139	}
140	memcpy(hdr.fs_uuid, fs->super->s_uuid, sizeof(hdr.fs_uuid));
141
142	hdr.image_time = time(0);
143	write_header(fd, &hdr, fs->blocksize);
144}
145
146/*
147 * These set of functions are used to write a RAW image file.
148 */
149ext2fs_block_bitmap meta_block_map;
150ext2fs_block_bitmap scramble_block_map;	/* Directory blocks to be scrambled */
151
152struct process_block_struct {
153	ext2_ino_t	ino;
154	int		is_dir;
155};
156
157/*
158 * These subroutines short circuits ext2fs_get_blocks and
159 * ext2fs_check_directory; we use them since we already have the inode
160 * structure, so there's no point in letting the ext2fs library read
161 * the inode again.
162 */
163static ino_t stashed_ino = 0;
164static struct ext2_inode *stashed_inode;
165
166static errcode_t meta_get_blocks(ext2_filsys fs EXT2FS_ATTR((unused)),
167				 ext2_ino_t ino,
168				 blk_t *blocks)
169{
170	int	i;
171
172	if ((ino != stashed_ino) || !stashed_inode)
173		return EXT2_ET_CALLBACK_NOTHANDLED;
174
175	for (i=0; i < EXT2_N_BLOCKS; i++)
176		blocks[i] = stashed_inode->i_block[i];
177	return 0;
178}
179
180static errcode_t meta_check_directory(ext2_filsys fs EXT2FS_ATTR((unused)),
181				      ext2_ino_t ino)
182{
183	if ((ino != stashed_ino) || !stashed_inode)
184		return EXT2_ET_CALLBACK_NOTHANDLED;
185
186	if (!LINUX_S_ISDIR(stashed_inode->i_mode))
187		return EXT2_ET_NO_DIRECTORY;
188	return 0;
189}
190
191static errcode_t meta_read_inode(ext2_filsys fs EXT2FS_ATTR((unused)),
192				 ext2_ino_t ino,
193				 struct ext2_inode *inode)
194{
195	if ((ino != stashed_ino) || !stashed_inode)
196		return EXT2_ET_CALLBACK_NOTHANDLED;
197	*inode = *stashed_inode;
198	return 0;
199}
200
201static void use_inode_shortcuts(ext2_filsys fs, int bool)
202{
203	if (bool) {
204		fs->get_blocks = meta_get_blocks;
205		fs->check_directory = meta_check_directory;
206		fs->read_inode = meta_read_inode;
207		stashed_ino = 0;
208	} else {
209		fs->get_blocks = 0;
210		fs->check_directory = 0;
211		fs->read_inode = 0;
212	}
213}
214
215static int process_dir_block(ext2_filsys fs EXT2FS_ATTR((unused)),
216			     blk_t *block_nr,
217			     e2_blkcnt_t blockcnt EXT2FS_ATTR((unused)),
218			     blk_t ref_block EXT2FS_ATTR((unused)),
219			     int ref_offset EXT2FS_ATTR((unused)),
220			     void *priv_data EXT2FS_ATTR((unused)))
221{
222	struct process_block_struct *p;
223
224	p = (struct process_block_struct *) priv_data;
225
226	ext2fs_mark_block_bitmap(meta_block_map, *block_nr);
227	if (scramble_block_map && p->is_dir && blockcnt >= 0)
228		ext2fs_mark_block_bitmap(scramble_block_map, *block_nr);
229	return 0;
230}
231
232static int process_file_block(ext2_filsys fs EXT2FS_ATTR((unused)),
233			      blk_t *block_nr,
234			      e2_blkcnt_t blockcnt,
235			      blk_t ref_block EXT2FS_ATTR((unused)),
236			      int ref_offset EXT2FS_ATTR((unused)),
237			      void *priv_data EXT2FS_ATTR((unused)))
238{
239	if (blockcnt < 0) {
240		ext2fs_mark_block_bitmap(meta_block_map, *block_nr);
241	}
242	return 0;
243}
244
245static void mark_table_blocks(ext2_filsys fs)
246{
247	blk_t	first_block, b;
248	unsigned int	i,j;
249
250	first_block = fs->super->s_first_data_block;
251	/*
252	 * Mark primary superblock
253	 */
254	ext2fs_mark_block_bitmap(meta_block_map, first_block);
255
256	/*
257	 * Mark the primary superblock descriptors
258	 */
259	for (j = 0; j < fs->desc_blocks; j++) {
260		ext2fs_mark_block_bitmap(meta_block_map,
261			 ext2fs_descriptor_block_loc(fs, first_block, j));
262	}
263
264	for (i = 0; i < fs->group_desc_count; i++) {
265		/*
266		 * Mark the blocks used for the inode table
267		 */
268		if (fs->group_desc[i].bg_inode_table) {
269			for (j = 0, b = fs->group_desc[i].bg_inode_table;
270			     j < (unsigned) fs->inode_blocks_per_group;
271			     j++, b++)
272				ext2fs_mark_block_bitmap(meta_block_map, b);
273		}
274
275		/*
276		 * Mark block used for the block bitmap
277		 */
278		if (fs->group_desc[i].bg_block_bitmap) {
279			ext2fs_mark_block_bitmap(meta_block_map,
280				     fs->group_desc[i].bg_block_bitmap);
281		}
282
283		/*
284		 * Mark block used for the inode bitmap
285		 */
286		if (fs->group_desc[i].bg_inode_bitmap) {
287			ext2fs_mark_block_bitmap(meta_block_map,
288				 fs->group_desc[i].bg_inode_bitmap);
289		}
290	}
291}
292
293/*
294 * This function returns 1 if the specified block is all zeros
295 */
296static int check_zero_block(char *buf, int blocksize)
297{
298	char	*cp = buf;
299	int	left = blocksize;
300
301	while (left > 0) {
302		if (*cp++)
303			return 0;
304		left--;
305	}
306	return 1;
307}
308
309static void write_block(int fd, char *buf, int sparse_offset,
310			int blocksize, blk_t block)
311{
312	int		count;
313	errcode_t	err;
314
315	if (sparse_offset) {
316#ifdef HAVE_LSEEK64
317		if (lseek64(fd, sparse_offset, SEEK_CUR) < 0)
318			perror("lseek");
319#else
320		if (lseek(fd, sparse_offset, SEEK_CUR) < 0)
321			perror("lseek");
322#endif
323	}
324	if (blocksize) {
325		count = write(fd, buf, blocksize);
326		if (count != blocksize) {
327			if (count == -1)
328				err = errno;
329			else
330				err = 0;
331			com_err(program_name, err, "error writing block %u",
332				block);
333			exit(1);
334		}
335	}
336}
337
338int name_id[256];
339
340static void scramble_dir_block(ext2_filsys fs, blk_t blk, char *buf)
341{
342	char *p, *end, *cp;
343	struct ext2_dir_entry_2 *dirent;
344	int rec_len, id, len;
345
346	end = buf + fs->blocksize;
347	for (p = buf; p < end-8; p += rec_len) {
348		dirent = (struct ext2_dir_entry_2 *) p;
349		rec_len = dirent->rec_len;
350#ifdef EXT2FS_ENABLE_SWAPFS
351		if (fs->flags & EXT2_FLAG_SWAP_BYTES)
352			rec_len = ext2fs_swab16(rec_len);
353#endif
354#if 0
355		printf("rec_len = %d, name_len = %d\n", rec_len, dirent->name_len);
356#endif
357		if (rec_len < 8 || (rec_len % 4) ||
358		    (p+rec_len > end)) {
359			printf("Corrupt directory block %lu: "
360			       "bad rec_len (%d)\n", (unsigned long) blk,
361			       rec_len);
362			rec_len = end - p;
363#ifdef EXT2FS_ENABLE_SWAPFS
364			if (fs->flags & EXT2_FLAG_SWAP_BYTES)
365				dirent->rec_len = ext2fs_swab16(rec_len);
366#endif
367			continue;
368		}
369		if (dirent->name_len + 8 > rec_len) {
370			printf("Corrupt directory block %lu: "
371			       "bad name_len (%d)\n", (unsigned long) blk,
372			       dirent->name_len);
373			dirent->name_len = rec_len - 8;
374			continue;
375		}
376		cp = p+8;
377		len = rec_len - dirent->name_len - 8;
378		if (len > 0)
379			memset(cp+dirent->name_len, 0, len);
380		if (dirent->name_len==1 && cp[0] == '.')
381			continue;
382		if (dirent->name_len==2 && cp[0] == '.' && cp[1] == '.')
383			continue;
384
385		memset(cp, 'A', dirent->name_len);
386		len = dirent->name_len;
387		id = name_id[len]++;
388		while ((len > 0) && (id > 0)) {
389			*cp += id % 26;
390			id = id / 26;
391			cp++;
392			len--;
393		}
394	}
395}
396
397static void output_meta_data_blocks(ext2_filsys fs, int fd)
398{
399	errcode_t	retval;
400	blk_t		blk;
401	char		*buf, *zero_buf;
402	int		sparse = 0;
403
404	buf = malloc(fs->blocksize);
405	if (!buf) {
406		com_err(program_name, ENOMEM, "while allocating buffer");
407		exit(1);
408	}
409	zero_buf = malloc(fs->blocksize);
410	if (!zero_buf) {
411		com_err(program_name, ENOMEM, "while allocating buffer");
412		exit(1);
413	}
414	memset(zero_buf, 0, fs->blocksize);
415	for (blk = 0; blk < fs->super->s_blocks_count; blk++) {
416		if ((blk >= fs->super->s_first_data_block) &&
417		    ext2fs_test_block_bitmap(meta_block_map, blk)) {
418			retval = io_channel_read_blk(fs->io, blk, 1, buf);
419			if (retval) {
420				com_err(program_name, retval,
421					"error reading block %u", blk);
422			}
423			if (scramble_block_map &&
424			    ext2fs_test_block_bitmap(scramble_block_map, blk))
425				scramble_dir_block(fs, blk, buf);
426			if ((fd != 1) && check_zero_block(buf, fs->blocksize))
427				goto sparse_write;
428			write_block(fd, buf, sparse, fs->blocksize, blk);
429			sparse = 0;
430		} else {
431		sparse_write:
432			if (fd == 1) {
433				write_block(fd, zero_buf, 0,
434					    fs->blocksize, blk);
435				continue;
436			}
437			sparse += fs->blocksize;
438			if (sparse >= 1024*1024) {
439				write_block(fd, 0, sparse, 0, 0);
440				sparse = 0;
441			}
442		}
443	}
444	if (sparse)
445		write_block(fd, zero_buf, sparse-1, 1, -1);
446	free(zero_buf);
447	free(buf);
448}
449
450static void write_raw_image_file(ext2_filsys fs, int fd, int scramble_flag)
451{
452	struct process_block_struct	pb;
453	struct ext2_inode		inode;
454	ext2_inode_scan			scan;
455	ext2_ino_t			ino;
456	errcode_t			retval;
457	char *				block_buf;
458
459	retval = ext2fs_allocate_block_bitmap(fs, "in-use block map",
460					      &meta_block_map);
461	if (retval) {
462		com_err(program_name, retval, "while allocating block bitmap");
463		exit(1);
464	}
465
466	if (scramble_flag) {
467		retval = ext2fs_allocate_block_bitmap(fs, "scramble block map",
468						      &scramble_block_map);
469		if (retval) {
470			com_err(program_name, retval,
471				"while allocating scramble block bitmap");
472			exit(1);
473		}
474	}
475
476	mark_table_blocks(fs);
477
478	retval = ext2fs_open_inode_scan(fs, 0, &scan);
479	if (retval) {
480		com_err(program_name, retval, _("while opening inode scan"));
481		exit(1);
482	}
483
484	block_buf = malloc(fs->blocksize * 3);
485	if (!block_buf) {
486		com_err(program_name, 0, "Can't allocate block buffer");
487		exit(1);
488	}
489
490	use_inode_shortcuts(fs, 1);
491	stashed_inode = &inode;
492	while (1) {
493		retval = ext2fs_get_next_inode(scan, &ino, &inode);
494		if (retval == EXT2_ET_BAD_BLOCK_IN_INODE_TABLE)
495			continue;
496		if (retval) {
497			com_err(program_name, retval,
498				_("while getting next inode"));
499			exit(1);
500		}
501		if (ino == 0)
502			break;
503		if (!inode.i_links_count)
504			continue;
505		if (inode.i_file_acl) {
506			ext2fs_mark_block_bitmap(meta_block_map,
507						 inode.i_file_acl);
508		}
509		if (!ext2fs_inode_has_valid_blocks(&inode))
510			continue;
511
512		stashed_ino = ino;
513		pb.ino = ino;
514		pb.is_dir = LINUX_S_ISDIR(inode.i_mode);
515		if (LINUX_S_ISDIR(inode.i_mode) ||
516		    (LINUX_S_ISLNK(inode.i_mode) &&
517		     ext2fs_inode_has_valid_blocks(&inode)) ||
518		    ino == fs->super->s_journal_inum) {
519			retval = ext2fs_block_iterate2(fs, ino, 0,
520				       block_buf, process_dir_block, &pb);
521			if (retval) {
522				com_err(program_name, retval,
523					"while iterating over inode %u",
524					ino);
525				exit(1);
526			}
527		} else {
528			if (inode.i_block[EXT2_IND_BLOCK] ||
529			    inode.i_block[EXT2_DIND_BLOCK] ||
530			    inode.i_block[EXT2_TIND_BLOCK]) {
531				retval = ext2fs_block_iterate2(fs,
532				       ino, 0, block_buf,
533				       process_file_block, &pb);
534				if (retval) {
535					com_err(program_name, retval,
536					"while iterating over inode %u", ino);
537					exit(1);
538				}
539			}
540		}
541	}
542	use_inode_shortcuts(fs, 0);
543	output_meta_data_blocks(fs, fd);
544	free(block_buf);
545}
546
547static void install_image(char *device, char *image_fn, int raw_flag)
548{
549	errcode_t retval;
550	ext2_filsys fs;
551	int open_flag = EXT2_FLAG_IMAGE_FILE;
552	int fd = 0;
553	io_manager	io_ptr;
554	io_channel	io, image_io;
555
556	if (raw_flag) {
557		com_err(program_name, 0, "Raw images cannot be installed");
558		exit(1);
559	}
560
561#ifdef CONFIG_TESTIO_DEBUG
562	io_ptr = test_io_manager;
563	test_io_backing_manager = unix_io_manager;
564#else
565	io_ptr = unix_io_manager;
566#endif
567
568	retval = ext2fs_open (image_fn, open_flag, 0, 0,
569			      io_ptr, &fs);
570        if (retval) {
571		com_err (program_name, retval, _("while trying to open %s"),
572			 image_fn);
573		exit(1);
574	}
575
576	retval = ext2fs_read_bitmaps (fs);
577	if (retval) {
578		com_err(program_name, retval, "error reading bitmaps");
579		exit(1);
580	}
581
582#ifdef HAVE_OPEN64
583	fd = open64(image_fn, O_RDONLY);
584#else
585	fd = open(image_fn, O_RDONLY);
586#endif
587	if (fd < 0) {
588		perror(image_fn);
589		exit(1);
590	}
591
592	retval = io_ptr->open(device, IO_FLAG_RW, &io);
593	if (retval) {
594		com_err(device, 0, "while opening device file");
595		exit(1);
596	}
597
598	image_io = fs->io;
599
600	ext2fs_rewrite_to_io(fs, io);
601
602	if (lseek(fd, fs->image_header->offset_inode, SEEK_SET) < 0) {
603		perror("lseek");
604		exit(1);
605	}
606
607	retval = ext2fs_image_inode_read(fs, fd, 0);
608	if (retval) {
609		com_err(image_fn, 0, "while restoring the image table");
610		exit(1);
611	}
612
613	ext2fs_close (fs);
614	exit (0);
615}
616
617int main (int argc, char ** argv)
618{
619	int c;
620	errcode_t retval;
621	ext2_filsys fs;
622	char *image_fn;
623	int open_flag = 0;
624	int raw_flag = 0;
625	int install_flag = 0;
626	int scramble_flag = 0;
627	int fd = 0;
628
629#ifdef ENABLE_NLS
630	setlocale(LC_MESSAGES, "");
631	setlocale(LC_CTYPE, "");
632	bindtextdomain(NLS_CAT_NAME, LOCALEDIR);
633	textdomain(NLS_CAT_NAME);
634#endif
635	fprintf (stderr, "e2image %s (%s)\n", E2FSPROGS_VERSION,
636		 E2FSPROGS_DATE);
637	if (argc && *argv)
638		program_name = *argv;
639	add_error_table(&et_ext2_error_table);
640	while ((c = getopt (argc, argv, "rsI")) != EOF)
641		switch (c) {
642		case 'r':
643			raw_flag++;
644			break;
645		case 's':
646			scramble_flag++;
647			break;
648		case 'I':
649			install_flag++;
650			break;
651		default:
652			usage();
653		}
654	if (optind != argc - 2 )
655		usage();
656	device_name = argv[optind];
657	image_fn = argv[optind+1];
658
659	if (install_flag) {
660		install_image(device_name, image_fn, raw_flag);
661		exit (0);
662	}
663
664	retval = ext2fs_open (device_name, open_flag, 0, 0,
665			      unix_io_manager, &fs);
666        if (retval) {
667		com_err (program_name, retval, _("while trying to open %s"),
668			 device_name);
669		fputs(_("Couldn't find valid filesystem superblock.\n"), stdout);
670		exit(1);
671	}
672
673	if (strcmp(image_fn, "-") == 0)
674		fd = 1;
675	else {
676#ifdef HAVE_OPEN64
677		fd = open64(image_fn, O_CREAT|O_TRUNC|O_WRONLY, 0600);
678#else
679		fd = open(image_fn, O_CREAT|O_TRUNC|O_WRONLY, 0600);
680#endif
681		if (fd < 0) {
682			com_err(program_name, errno,
683				_("while trying to open %s"), argv[optind+1]);
684			exit(1);
685		}
686	}
687
688	if (raw_flag)
689		write_raw_image_file(fs, fd, scramble_flag);
690	else
691		write_image_file(fs, fd);
692
693	ext2fs_close (fs);
694	remove_error_table(&et_ext2_error_table);
695	exit (0);
696}
697