e2image.c revision f38cf3cb34addaa53d1f855d7607b151082a4229
1691925f304f07de950da5a7b45c797f38c542bb8Dan Shi/* 2691925f304f07de950da5a7b45c797f38c542bb8Dan Shi * e2image.c --- Program which writes an image file backing up 3691925f304f07de950da5a7b45c797f38c542bb8Dan Shi * critical metadata for the filesystem. 4691925f304f07de950da5a7b45c797f38c542bb8Dan Shi * 5691925f304f07de950da5a7b45c797f38c542bb8Dan Shi * Copyright 2000, 2001 by Theodore Ts'o. 6691925f304f07de950da5a7b45c797f38c542bb8Dan Shi * 7691925f304f07de950da5a7b45c797f38c542bb8Dan Shi * %Begin-Header% 8691925f304f07de950da5a7b45c797f38c542bb8Dan Shi * This file may be redistributed under the terms of the GNU Public 9691925f304f07de950da5a7b45c797f38c542bb8Dan Shi * License. 10691925f304f07de950da5a7b45c797f38c542bb8Dan Shi * %End-Header% 11691925f304f07de950da5a7b45c797f38c542bb8Dan Shi */ 12691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 13691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#define _LARGEFILE_SOURCE 14322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi#define _LARGEFILE64_SOURCE 15691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 16322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi#include <fcntl.h> 17691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#include <grp.h> 18691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#ifdef HAVE_GETOPT_H 19691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#include <getopt.h> 20691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#else 21691925f304f07de950da5a7b45c797f38c542bb8Dan Shiextern char *optarg; 22691925f304f07de950da5a7b45c797f38c542bb8Dan Shiextern int optind; 23691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#endif 24691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#include <pwd.h> 25691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#include <stdio.h> 26691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#ifdef HAVE_STDLIB_H 27691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#include <stdlib.h> 28691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#endif 29691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#include <string.h> 30691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#include <time.h> 31691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#include <unistd.h> 32691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#include <fcntl.h> 33691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#include <errno.h> 34322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi#include <sys/stat.h> 35322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi#include <sys/types.h> 36322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi 37691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#include "ext2fs/ext2_fs.h" 38691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#include "ext2fs/ext2fs.h" 39691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#include "et/com_err.h" 40691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#include "uuid/uuid.h" 41322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi#include "e2p/e2p.h" 42322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi#include "ext2fs/e2image.h" 43322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi 44322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi#include "../version.h" 45322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi#include "nls-enable.h" 46322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi 47322f4d1060633bb4d463ce7880e36587d5792a8cDan Shiconst char * program_name = "e2image"; 48322f4d1060633bb4d463ce7880e36587d5792a8cDan Shichar * device_name = NULL; 49322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi 50322f4d1060633bb4d463ce7880e36587d5792a8cDan Shistatic void usage(void) 51322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi{ 52322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi fprintf(stderr, _("Usage: %s [-rsI] device image_file\n"), 53322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi program_name); 54322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi exit (1); 55322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi} 56322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi 57322f4d1060633bb4d463ce7880e36587d5792a8cDan Shistatic void write_header(int fd, struct ext2_image_hdr *hdr, int blocksize) 58322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi{ 59322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi char *header_buf; 60322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi int actual; 61322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi 62322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi header_buf = malloc(blocksize); 63322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi if (!header_buf) { 64322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi fputs(_("Couldn't allocate header buffer\n"), stderr); 65322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi exit(1); 66691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 67691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 68691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (lseek(fd, 0, SEEK_SET) < 0) { 69691925f304f07de950da5a7b45c797f38c542bb8Dan Shi perror("lseek while writing header"); 70691925f304f07de950da5a7b45c797f38c542bb8Dan Shi exit(1); 71691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 72691925f304f07de950da5a7b45c797f38c542bb8Dan Shi memset(header_buf, 0, blocksize); 73691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 74691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (hdr) 75691925f304f07de950da5a7b45c797f38c542bb8Dan Shi memcpy(header_buf, hdr, sizeof(struct ext2_image_hdr)); 76691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 77691925f304f07de950da5a7b45c797f38c542bb8Dan Shi actual = write(fd, header_buf, blocksize); 78691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (actual < 0) { 79691925f304f07de950da5a7b45c797f38c542bb8Dan Shi perror("write header"); 80691925f304f07de950da5a7b45c797f38c542bb8Dan Shi exit(1); 81691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 82691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (actual != blocksize) { 83691925f304f07de950da5a7b45c797f38c542bb8Dan Shi fprintf(stderr, _("short write (only %d bytes) for " 84691925f304f07de950da5a7b45c797f38c542bb8Dan Shi "writing image header"), actual); 85691925f304f07de950da5a7b45c797f38c542bb8Dan Shi exit(1); 86322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi } 87691925f304f07de950da5a7b45c797f38c542bb8Dan Shi free(header_buf); 88691925f304f07de950da5a7b45c797f38c542bb8Dan Shi} 89691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 90691925f304f07de950da5a7b45c797f38c542bb8Dan Shistatic void write_image_file(ext2_filsys fs, int fd) 91691925f304f07de950da5a7b45c797f38c542bb8Dan Shi{ 92691925f304f07de950da5a7b45c797f38c542bb8Dan Shi struct ext2_image_hdr hdr; 93691925f304f07de950da5a7b45c797f38c542bb8Dan Shi struct stat st; 94691925f304f07de950da5a7b45c797f38c542bb8Dan Shi errcode_t retval; 95691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 96691925f304f07de950da5a7b45c797f38c542bb8Dan Shi write_header(fd, NULL, fs->blocksize); 97691925f304f07de950da5a7b45c797f38c542bb8Dan Shi memset(&hdr, 0, sizeof(struct ext2_image_hdr)); 98691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 99691925f304f07de950da5a7b45c797f38c542bb8Dan Shi hdr.offset_super = lseek(fd, 0, SEEK_CUR); 100691925f304f07de950da5a7b45c797f38c542bb8Dan Shi retval = ext2fs_image_super_write(fs, fd, 0); 101691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (retval) { 102691925f304f07de950da5a7b45c797f38c542bb8Dan Shi com_err(program_name, retval, _("while writing superblock")); 103691925f304f07de950da5a7b45c797f38c542bb8Dan Shi exit(1); 104691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 105691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 106691925f304f07de950da5a7b45c797f38c542bb8Dan Shi hdr.offset_inode = lseek(fd, 0, SEEK_CUR); 107691925f304f07de950da5a7b45c797f38c542bb8Dan Shi retval = ext2fs_image_inode_write(fs, fd, 108691925f304f07de950da5a7b45c797f38c542bb8Dan Shi (fd != 1) ? IMAGER_FLAG_SPARSEWRITE : 0); 109691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (retval) { 110691925f304f07de950da5a7b45c797f38c542bb8Dan Shi com_err(program_name, retval, _("while writing inode table")); 111691925f304f07de950da5a7b45c797f38c542bb8Dan Shi exit(1); 112691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 113691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 114691925f304f07de950da5a7b45c797f38c542bb8Dan Shi hdr.offset_blockmap = lseek(fd, 0, SEEK_CUR); 115691925f304f07de950da5a7b45c797f38c542bb8Dan Shi retval = ext2fs_image_bitmap_write(fs, fd, 0); 116691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (retval) { 117691925f304f07de950da5a7b45c797f38c542bb8Dan Shi com_err(program_name, retval, _("while writing block bitmap")); 118691925f304f07de950da5a7b45c797f38c542bb8Dan Shi exit(1); 119691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 120691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 121691925f304f07de950da5a7b45c797f38c542bb8Dan Shi hdr.offset_inodemap = lseek(fd, 0, SEEK_CUR); 122691925f304f07de950da5a7b45c797f38c542bb8Dan Shi retval = ext2fs_image_bitmap_write(fs, fd, IMAGER_FLAG_INODEMAP); 123691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (retval) { 124691925f304f07de950da5a7b45c797f38c542bb8Dan Shi com_err(program_name, retval, _("while writing inode bitmap")); 125691925f304f07de950da5a7b45c797f38c542bb8Dan Shi exit(1); 126691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 127691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 128691925f304f07de950da5a7b45c797f38c542bb8Dan Shi hdr.magic_number = EXT2_ET_MAGIC_E2IMAGE; 129691925f304f07de950da5a7b45c797f38c542bb8Dan Shi strcpy(hdr.magic_descriptor, "Ext2 Image 1.0"); 130691925f304f07de950da5a7b45c797f38c542bb8Dan Shi gethostname(hdr.fs_hostname, sizeof(hdr.fs_hostname)); 131691925f304f07de950da5a7b45c797f38c542bb8Dan Shi strncpy(hdr.fs_device_name, device_name, sizeof(hdr.fs_device_name)-1); 132691925f304f07de950da5a7b45c797f38c542bb8Dan Shi hdr.fs_device_name[sizeof(hdr.fs_device_name) - 1] = 0; 133691925f304f07de950da5a7b45c797f38c542bb8Dan Shi hdr.fs_blocksize = fs->blocksize; 134691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 135691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (stat(device_name, &st) == 0) 136691925f304f07de950da5a7b45c797f38c542bb8Dan Shi hdr.fs_device = st.st_rdev; 137691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 138691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (fstat(fd, &st) == 0) { 139691925f304f07de950da5a7b45c797f38c542bb8Dan Shi hdr.image_device = st.st_dev; 140691925f304f07de950da5a7b45c797f38c542bb8Dan Shi hdr.image_inode = st.st_ino; 141691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 142691925f304f07de950da5a7b45c797f38c542bb8Dan Shi memcpy(hdr.fs_uuid, fs->super->s_uuid, sizeof(hdr.fs_uuid)); 143691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 144691925f304f07de950da5a7b45c797f38c542bb8Dan Shi hdr.image_time = time(0); 145691925f304f07de950da5a7b45c797f38c542bb8Dan Shi write_header(fd, &hdr, fs->blocksize); 146691925f304f07de950da5a7b45c797f38c542bb8Dan Shi} 147691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 148691925f304f07de950da5a7b45c797f38c542bb8Dan Shi/* 149691925f304f07de950da5a7b45c797f38c542bb8Dan Shi * These set of functions are used to write a RAW image file. 150691925f304f07de950da5a7b45c797f38c542bb8Dan Shi */ 151691925f304f07de950da5a7b45c797f38c542bb8Dan Shiext2fs_block_bitmap meta_block_map; 152691925f304f07de950da5a7b45c797f38c542bb8Dan Shiext2fs_block_bitmap scramble_block_map; /* Directory blocks to be scrambled */ 153691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 154691925f304f07de950da5a7b45c797f38c542bb8Dan Shistruct process_block_struct { 155691925f304f07de950da5a7b45c797f38c542bb8Dan Shi ext2_ino_t ino; 156691925f304f07de950da5a7b45c797f38c542bb8Dan Shi int is_dir; 157691925f304f07de950da5a7b45c797f38c542bb8Dan Shi}; 158691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 159691925f304f07de950da5a7b45c797f38c542bb8Dan Shi/* 160691925f304f07de950da5a7b45c797f38c542bb8Dan Shi * These subroutines short circuits ext2fs_get_blocks and 161691925f304f07de950da5a7b45c797f38c542bb8Dan Shi * ext2fs_check_directory; we use them since we already have the inode 162691925f304f07de950da5a7b45c797f38c542bb8Dan Shi * structure, so there's no point in letting the ext2fs library read 163691925f304f07de950da5a7b45c797f38c542bb8Dan Shi * the inode again. 164691925f304f07de950da5a7b45c797f38c542bb8Dan Shi */ 165691925f304f07de950da5a7b45c797f38c542bb8Dan Shistatic ino_t stashed_ino = 0; 166691925f304f07de950da5a7b45c797f38c542bb8Dan Shistatic struct ext2_inode *stashed_inode; 167691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 168691925f304f07de950da5a7b45c797f38c542bb8Dan Shistatic errcode_t meta_get_blocks(ext2_filsys fs EXT2FS_ATTR((unused)), 169691925f304f07de950da5a7b45c797f38c542bb8Dan Shi ext2_ino_t ino, 170691925f304f07de950da5a7b45c797f38c542bb8Dan Shi blk_t *blocks) 171691925f304f07de950da5a7b45c797f38c542bb8Dan Shi{ 172691925f304f07de950da5a7b45c797f38c542bb8Dan Shi int i; 173691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 174691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if ((ino != stashed_ino) || !stashed_inode) 175691925f304f07de950da5a7b45c797f38c542bb8Dan Shi return EXT2_ET_CALLBACK_NOTHANDLED; 176691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 177691925f304f07de950da5a7b45c797f38c542bb8Dan Shi for (i=0; i < EXT2_N_BLOCKS; i++) 178691925f304f07de950da5a7b45c797f38c542bb8Dan Shi blocks[i] = stashed_inode->i_block[i]; 179691925f304f07de950da5a7b45c797f38c542bb8Dan Shi return 0; 180691925f304f07de950da5a7b45c797f38c542bb8Dan Shi} 181691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 182691925f304f07de950da5a7b45c797f38c542bb8Dan Shistatic errcode_t meta_check_directory(ext2_filsys fs EXT2FS_ATTR((unused)), 183691925f304f07de950da5a7b45c797f38c542bb8Dan Shi ext2_ino_t ino) 184691925f304f07de950da5a7b45c797f38c542bb8Dan Shi{ 185691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if ((ino != stashed_ino) || !stashed_inode) 186691925f304f07de950da5a7b45c797f38c542bb8Dan Shi return EXT2_ET_CALLBACK_NOTHANDLED; 187691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 188691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (!LINUX_S_ISDIR(stashed_inode->i_mode)) 189691925f304f07de950da5a7b45c797f38c542bb8Dan Shi return EXT2_ET_NO_DIRECTORY; 190691925f304f07de950da5a7b45c797f38c542bb8Dan Shi return 0; 191691925f304f07de950da5a7b45c797f38c542bb8Dan Shi} 192691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 193691925f304f07de950da5a7b45c797f38c542bb8Dan Shistatic errcode_t meta_read_inode(ext2_filsys fs EXT2FS_ATTR((unused)), 194691925f304f07de950da5a7b45c797f38c542bb8Dan Shi ext2_ino_t ino, 195322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi struct ext2_inode *inode) 196322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi{ 197691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if ((ino != stashed_ino) || !stashed_inode) 198691925f304f07de950da5a7b45c797f38c542bb8Dan Shi return EXT2_ET_CALLBACK_NOTHANDLED; 199691925f304f07de950da5a7b45c797f38c542bb8Dan Shi *inode = *stashed_inode; 200322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi return 0; 201322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi} 202322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi 203322f4d1060633bb4d463ce7880e36587d5792a8cDan Shistatic void use_inode_shortcuts(ext2_filsys fs, int bool) 204691925f304f07de950da5a7b45c797f38c542bb8Dan Shi{ 205691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (bool) { 206691925f304f07de950da5a7b45c797f38c542bb8Dan Shi fs->get_blocks = meta_get_blocks; 207322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi fs->check_directory = meta_check_directory; 208322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi fs->read_inode = meta_read_inode; 209322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi stashed_ino = 0; 210322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi } else { 211691925f304f07de950da5a7b45c797f38c542bb8Dan Shi fs->get_blocks = 0; 212691925f304f07de950da5a7b45c797f38c542bb8Dan Shi fs->check_directory = 0; 213691925f304f07de950da5a7b45c797f38c542bb8Dan Shi fs->read_inode = 0; 214322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi } 215322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi} 216322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi 217691925f304f07de950da5a7b45c797f38c542bb8Dan Shistatic int process_dir_block(ext2_filsys fs EXT2FS_ATTR((unused)), 218691925f304f07de950da5a7b45c797f38c542bb8Dan Shi blk_t *block_nr, 219691925f304f07de950da5a7b45c797f38c542bb8Dan Shi e2_blkcnt_t blockcnt EXT2FS_ATTR((unused)), 220322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi blk_t ref_block EXT2FS_ATTR((unused)), 221322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi int ref_offset EXT2FS_ATTR((unused)), 222322f4d1060633bb4d463ce7880e36587d5792a8cDan Shi void *priv_data EXT2FS_ATTR((unused))) 223691925f304f07de950da5a7b45c797f38c542bb8Dan Shi{ 224691925f304f07de950da5a7b45c797f38c542bb8Dan Shi struct process_block_struct *p; 225691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 226691925f304f07de950da5a7b45c797f38c542bb8Dan Shi p = (struct process_block_struct *) priv_data; 227691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 228691925f304f07de950da5a7b45c797f38c542bb8Dan Shi ext2fs_mark_block_bitmap(meta_block_map, *block_nr); 229691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (scramble_block_map && p->is_dir && blockcnt >= 0) 230691925f304f07de950da5a7b45c797f38c542bb8Dan Shi ext2fs_mark_block_bitmap(scramble_block_map, *block_nr); 231691925f304f07de950da5a7b45c797f38c542bb8Dan Shi return 0; 232691925f304f07de950da5a7b45c797f38c542bb8Dan Shi} 233691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 234691925f304f07de950da5a7b45c797f38c542bb8Dan Shistatic int process_file_block(ext2_filsys fs EXT2FS_ATTR((unused)), 235691925f304f07de950da5a7b45c797f38c542bb8Dan Shi blk_t *block_nr, 236691925f304f07de950da5a7b45c797f38c542bb8Dan Shi e2_blkcnt_t blockcnt, 237691925f304f07de950da5a7b45c797f38c542bb8Dan Shi blk_t ref_block EXT2FS_ATTR((unused)), 238691925f304f07de950da5a7b45c797f38c542bb8Dan Shi int ref_offset EXT2FS_ATTR((unused)), 239691925f304f07de950da5a7b45c797f38c542bb8Dan Shi void *priv_data EXT2FS_ATTR((unused))) 240691925f304f07de950da5a7b45c797f38c542bb8Dan Shi{ 241691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (blockcnt < 0) { 242691925f304f07de950da5a7b45c797f38c542bb8Dan Shi ext2fs_mark_block_bitmap(meta_block_map, *block_nr); 243691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 244691925f304f07de950da5a7b45c797f38c542bb8Dan Shi return 0; 245691925f304f07de950da5a7b45c797f38c542bb8Dan Shi} 246691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 247691925f304f07de950da5a7b45c797f38c542bb8Dan Shistatic void mark_table_blocks(ext2_filsys fs) 248691925f304f07de950da5a7b45c797f38c542bb8Dan Shi{ 249691925f304f07de950da5a7b45c797f38c542bb8Dan Shi blk_t first_block, b; 250691925f304f07de950da5a7b45c797f38c542bb8Dan Shi unsigned int i,j; 251691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 252691925f304f07de950da5a7b45c797f38c542bb8Dan Shi first_block = fs->super->s_first_data_block; 253691925f304f07de950da5a7b45c797f38c542bb8Dan Shi /* 254691925f304f07de950da5a7b45c797f38c542bb8Dan Shi * Mark primary superblock 255691925f304f07de950da5a7b45c797f38c542bb8Dan Shi */ 256691925f304f07de950da5a7b45c797f38c542bb8Dan Shi ext2fs_mark_block_bitmap(meta_block_map, first_block); 257691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 258691925f304f07de950da5a7b45c797f38c542bb8Dan Shi /* 259691925f304f07de950da5a7b45c797f38c542bb8Dan Shi * Mark the primary superblock descriptors 260691925f304f07de950da5a7b45c797f38c542bb8Dan Shi */ 261691925f304f07de950da5a7b45c797f38c542bb8Dan Shi for (j = 0; j < fs->desc_blocks; j++) { 262691925f304f07de950da5a7b45c797f38c542bb8Dan Shi ext2fs_mark_block_bitmap(meta_block_map, 263691925f304f07de950da5a7b45c797f38c542bb8Dan Shi ext2fs_descriptor_block_loc(fs, first_block, j)); 264691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 265691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 266691925f304f07de950da5a7b45c797f38c542bb8Dan Shi for (i = 0; i < fs->group_desc_count; i++) { 267691925f304f07de950da5a7b45c797f38c542bb8Dan Shi /* 268691925f304f07de950da5a7b45c797f38c542bb8Dan Shi * Mark the blocks used for the inode table 269691925f304f07de950da5a7b45c797f38c542bb8Dan Shi */ 270691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (fs->group_desc[i].bg_inode_table) { 271691925f304f07de950da5a7b45c797f38c542bb8Dan Shi for (j = 0, b = fs->group_desc[i].bg_inode_table; 272691925f304f07de950da5a7b45c797f38c542bb8Dan Shi j < (unsigned) fs->inode_blocks_per_group; 273691925f304f07de950da5a7b45c797f38c542bb8Dan Shi j++, b++) 274691925f304f07de950da5a7b45c797f38c542bb8Dan Shi ext2fs_mark_block_bitmap(meta_block_map, b); 275691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 276691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 277691925f304f07de950da5a7b45c797f38c542bb8Dan Shi /* 278691925f304f07de950da5a7b45c797f38c542bb8Dan Shi * Mark block used for the block bitmap 279691925f304f07de950da5a7b45c797f38c542bb8Dan Shi */ 280691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (fs->group_desc[i].bg_block_bitmap) { 281691925f304f07de950da5a7b45c797f38c542bb8Dan Shi ext2fs_mark_block_bitmap(meta_block_map, 282691925f304f07de950da5a7b45c797f38c542bb8Dan Shi fs->group_desc[i].bg_block_bitmap); 283691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 284691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 285691925f304f07de950da5a7b45c797f38c542bb8Dan Shi /* 286691925f304f07de950da5a7b45c797f38c542bb8Dan Shi * Mark block used for the inode bitmap 287691925f304f07de950da5a7b45c797f38c542bb8Dan Shi */ 288691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (fs->group_desc[i].bg_inode_bitmap) { 289691925f304f07de950da5a7b45c797f38c542bb8Dan Shi ext2fs_mark_block_bitmap(meta_block_map, 290691925f304f07de950da5a7b45c797f38c542bb8Dan Shi fs->group_desc[i].bg_inode_bitmap); 291691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 292691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 293691925f304f07de950da5a7b45c797f38c542bb8Dan Shi} 294691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 295691925f304f07de950da5a7b45c797f38c542bb8Dan Shi/* 296691925f304f07de950da5a7b45c797f38c542bb8Dan Shi * This function returns 1 if the specified block is all zeros 297691925f304f07de950da5a7b45c797f38c542bb8Dan Shi */ 298691925f304f07de950da5a7b45c797f38c542bb8Dan Shistatic int check_zero_block(char *buf, int blocksize) 299691925f304f07de950da5a7b45c797f38c542bb8Dan Shi{ 300691925f304f07de950da5a7b45c797f38c542bb8Dan Shi char *cp = buf; 301691925f304f07de950da5a7b45c797f38c542bb8Dan Shi int left = blocksize; 302691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 303691925f304f07de950da5a7b45c797f38c542bb8Dan Shi while (left > 0) { 304691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (*cp++) 305691925f304f07de950da5a7b45c797f38c542bb8Dan Shi return 0; 306691925f304f07de950da5a7b45c797f38c542bb8Dan Shi left--; 307691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 308691925f304f07de950da5a7b45c797f38c542bb8Dan Shi return 1; 309691925f304f07de950da5a7b45c797f38c542bb8Dan Shi} 310691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 311691925f304f07de950da5a7b45c797f38c542bb8Dan Shistatic void write_block(int fd, char *buf, int sparse_offset, 312691925f304f07de950da5a7b45c797f38c542bb8Dan Shi int blocksize, blk_t block) 313691925f304f07de950da5a7b45c797f38c542bb8Dan Shi{ 314691925f304f07de950da5a7b45c797f38c542bb8Dan Shi int count; 315691925f304f07de950da5a7b45c797f38c542bb8Dan Shi errcode_t err; 316691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 317691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (sparse_offset) { 318691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#ifdef HAVE_LSEEK64 319691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (lseek64(fd, sparse_offset, SEEK_CUR) < 0) 320691925f304f07de950da5a7b45c797f38c542bb8Dan Shi perror("lseek"); 321691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#else 322691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (lseek(fd, sparse_offset, SEEK_CUR) < 0) 323691925f304f07de950da5a7b45c797f38c542bb8Dan Shi perror("lseek"); 324691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#endif 325691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 326691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (blocksize) { 327691925f304f07de950da5a7b45c797f38c542bb8Dan Shi count = write(fd, buf, blocksize); 328691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (count != blocksize) { 329691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (count == -1) 330691925f304f07de950da5a7b45c797f38c542bb8Dan Shi err = errno; 331691925f304f07de950da5a7b45c797f38c542bb8Dan Shi else 332691925f304f07de950da5a7b45c797f38c542bb8Dan Shi err = 0; 333691925f304f07de950da5a7b45c797f38c542bb8Dan Shi com_err(program_name, err, "error writing block %u", 334691925f304f07de950da5a7b45c797f38c542bb8Dan Shi block); 335691925f304f07de950da5a7b45c797f38c542bb8Dan Shi exit(1); 336691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 337691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 338691925f304f07de950da5a7b45c797f38c542bb8Dan Shi} 339691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 340691925f304f07de950da5a7b45c797f38c542bb8Dan Shiint name_id[256]; 341691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 342691925f304f07de950da5a7b45c797f38c542bb8Dan Shistatic void scramble_dir_block(ext2_filsys fs, blk_t blk, char *buf) 343691925f304f07de950da5a7b45c797f38c542bb8Dan Shi{ 344691925f304f07de950da5a7b45c797f38c542bb8Dan Shi char *p, *end, *cp; 345691925f304f07de950da5a7b45c797f38c542bb8Dan Shi struct ext2_dir_entry_2 *dirent; 346691925f304f07de950da5a7b45c797f38c542bb8Dan Shi int rec_len, id, len; 347691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 348691925f304f07de950da5a7b45c797f38c542bb8Dan Shi end = buf + fs->blocksize; 349691925f304f07de950da5a7b45c797f38c542bb8Dan Shi for (p = buf; p < end-8; p += rec_len) { 350691925f304f07de950da5a7b45c797f38c542bb8Dan Shi dirent = (struct ext2_dir_entry_2 *) p; 351691925f304f07de950da5a7b45c797f38c542bb8Dan Shi rec_len = dirent->rec_len; 352691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#ifdef WORDS_BIGENDIAN 353691925f304f07de950da5a7b45c797f38c542bb8Dan Shi rec_len = ext2fs_swab16(rec_len); 354691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#endif 355691925f304f07de950da5a7b45c797f38c542bb8Dan Shi rec_len = (rec_len || fs->blocksize < 65536) ? 356691925f304f07de950da5a7b45c797f38c542bb8Dan Shi rec_len : 65536; 357691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#if 0 358691925f304f07de950da5a7b45c797f38c542bb8Dan Shi printf("rec_len = %d, name_len = %d\n", rec_len, dirent->name_len); 359691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#endif 360691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (rec_len < 8 || (rec_len % 4) || 361691925f304f07de950da5a7b45c797f38c542bb8Dan Shi (p+rec_len > end)) { 362691925f304f07de950da5a7b45c797f38c542bb8Dan Shi printf("Corrupt directory block %lu: " 363691925f304f07de950da5a7b45c797f38c542bb8Dan Shi "bad rec_len (%d)\n", (unsigned long) blk, 364691925f304f07de950da5a7b45c797f38c542bb8Dan Shi rec_len); 365691925f304f07de950da5a7b45c797f38c542bb8Dan Shi rec_len = end - p; 366691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#ifdef WORDS_BIGENDIAN 367691925f304f07de950da5a7b45c797f38c542bb8Dan Shi dirent->rec_len = ext2fs_swab16(rec_len); 368691925f304f07de950da5a7b45c797f38c542bb8Dan Shi#endif 369691925f304f07de950da5a7b45c797f38c542bb8Dan Shi continue; 370691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 371691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (dirent->name_len + 8 > rec_len) { 372691925f304f07de950da5a7b45c797f38c542bb8Dan Shi printf("Corrupt directory block %lu: " 373691925f304f07de950da5a7b45c797f38c542bb8Dan Shi "bad name_len (%d)\n", (unsigned long) blk, 374691925f304f07de950da5a7b45c797f38c542bb8Dan Shi dirent->name_len); 375691925f304f07de950da5a7b45c797f38c542bb8Dan Shi dirent->name_len = rec_len - 8; 376691925f304f07de950da5a7b45c797f38c542bb8Dan Shi continue; 377691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 378691925f304f07de950da5a7b45c797f38c542bb8Dan Shi cp = p+8; 379691925f304f07de950da5a7b45c797f38c542bb8Dan Shi len = rec_len - dirent->name_len - 8; 380691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (len > 0) 381691925f304f07de950da5a7b45c797f38c542bb8Dan Shi memset(cp+dirent->name_len, 0, len); 382691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (dirent->name_len==1 && cp[0] == '.') 383691925f304f07de950da5a7b45c797f38c542bb8Dan Shi continue; 384691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (dirent->name_len==2 && cp[0] == '.' && cp[1] == '.') 385691925f304f07de950da5a7b45c797f38c542bb8Dan Shi continue; 386691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 387691925f304f07de950da5a7b45c797f38c542bb8Dan Shi memset(cp, 'A', dirent->name_len); 388691925f304f07de950da5a7b45c797f38c542bb8Dan Shi len = dirent->name_len; 389691925f304f07de950da5a7b45c797f38c542bb8Dan Shi id = name_id[len]++; 390691925f304f07de950da5a7b45c797f38c542bb8Dan Shi while ((len > 0) && (id > 0)) { 391691925f304f07de950da5a7b45c797f38c542bb8Dan Shi *cp += id % 26; 392691925f304f07de950da5a7b45c797f38c542bb8Dan Shi id = id / 26; 393691925f304f07de950da5a7b45c797f38c542bb8Dan Shi cp++; 394691925f304f07de950da5a7b45c797f38c542bb8Dan Shi len--; 395691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 396691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 397691925f304f07de950da5a7b45c797f38c542bb8Dan Shi} 398691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 399691925f304f07de950da5a7b45c797f38c542bb8Dan Shistatic void output_meta_data_blocks(ext2_filsys fs, int fd) 400691925f304f07de950da5a7b45c797f38c542bb8Dan Shi{ 401691925f304f07de950da5a7b45c797f38c542bb8Dan Shi errcode_t retval; 402691925f304f07de950da5a7b45c797f38c542bb8Dan Shi blk_t blk; 403691925f304f07de950da5a7b45c797f38c542bb8Dan Shi char *buf, *zero_buf; 404691925f304f07de950da5a7b45c797f38c542bb8Dan Shi int sparse = 0; 405691925f304f07de950da5a7b45c797f38c542bb8Dan Shi 406691925f304f07de950da5a7b45c797f38c542bb8Dan Shi buf = malloc(fs->blocksize); 407691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (!buf) { 408691925f304f07de950da5a7b45c797f38c542bb8Dan Shi com_err(program_name, ENOMEM, "while allocating buffer"); 409691925f304f07de950da5a7b45c797f38c542bb8Dan Shi exit(1); 410691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 411691925f304f07de950da5a7b45c797f38c542bb8Dan Shi zero_buf = malloc(fs->blocksize); 412691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (!zero_buf) { 413691925f304f07de950da5a7b45c797f38c542bb8Dan Shi com_err(program_name, ENOMEM, "while allocating buffer"); 414691925f304f07de950da5a7b45c797f38c542bb8Dan Shi exit(1); 415691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 416691925f304f07de950da5a7b45c797f38c542bb8Dan Shi memset(zero_buf, 0, fs->blocksize); 417691925f304f07de950da5a7b45c797f38c542bb8Dan Shi for (blk = 0; blk < fs->super->s_blocks_count; blk++) { 418691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if ((blk >= fs->super->s_first_data_block) && 419691925f304f07de950da5a7b45c797f38c542bb8Dan Shi ext2fs_test_block_bitmap(meta_block_map, blk)) { 420691925f304f07de950da5a7b45c797f38c542bb8Dan Shi retval = io_channel_read_blk(fs->io, blk, 1, buf); 421691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (retval) { 422691925f304f07de950da5a7b45c797f38c542bb8Dan Shi com_err(program_name, retval, 423691925f304f07de950da5a7b45c797f38c542bb8Dan Shi "error reading block %u", blk); 424691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 425691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (scramble_block_map && 426691925f304f07de950da5a7b45c797f38c542bb8Dan Shi ext2fs_test_block_bitmap(scramble_block_map, blk)) 427691925f304f07de950da5a7b45c797f38c542bb8Dan Shi scramble_dir_block(fs, blk, buf); 428691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if ((fd != 1) && check_zero_block(buf, fs->blocksize)) 429691925f304f07de950da5a7b45c797f38c542bb8Dan Shi goto sparse_write; 430691925f304f07de950da5a7b45c797f38c542bb8Dan Shi write_block(fd, buf, sparse, fs->blocksize, blk); 431691925f304f07de950da5a7b45c797f38c542bb8Dan Shi sparse = 0; 432691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } else { 433691925f304f07de950da5a7b45c797f38c542bb8Dan Shi sparse_write: 434691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (fd == 1) { 435691925f304f07de950da5a7b45c797f38c542bb8Dan Shi write_block(fd, zero_buf, 0, 436691925f304f07de950da5a7b45c797f38c542bb8Dan Shi fs->blocksize, blk); 437691925f304f07de950da5a7b45c797f38c542bb8Dan Shi continue; 438691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 439691925f304f07de950da5a7b45c797f38c542bb8Dan Shi sparse += fs->blocksize; 440691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (sparse >= 1024*1024) { 441691925f304f07de950da5a7b45c797f38c542bb8Dan Shi write_block(fd, 0, sparse, 0, 0); 442691925f304f07de950da5a7b45c797f38c542bb8Dan Shi sparse = 0; 443691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 444691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 445691925f304f07de950da5a7b45c797f38c542bb8Dan Shi } 446691925f304f07de950da5a7b45c797f38c542bb8Dan Shi if (sparse) 447691925f304f07de950da5a7b45c797f38c542bb8Dan Shi write_block(fd, zero_buf, sparse-1, 1, -1); 448691925f304f07de950da5a7b45c797f38c542bb8Dan Shi free(zero_buf); 449691925f304f07de950da5a7b45c797f38c542bb8Dan Shi free(buf); 450} 451 452static void write_raw_image_file(ext2_filsys fs, int fd, int scramble_flag) 453{ 454 struct process_block_struct pb; 455 struct ext2_inode inode; 456 ext2_inode_scan scan; 457 ext2_ino_t ino; 458 errcode_t retval; 459 char * block_buf; 460 461 retval = ext2fs_allocate_block_bitmap(fs, "in-use block map", 462 &meta_block_map); 463 if (retval) { 464 com_err(program_name, retval, "while allocating block bitmap"); 465 exit(1); 466 } 467 468 if (scramble_flag) { 469 retval = ext2fs_allocate_block_bitmap(fs, "scramble block map", 470 &scramble_block_map); 471 if (retval) { 472 com_err(program_name, retval, 473 "while allocating scramble block bitmap"); 474 exit(1); 475 } 476 } 477 478 mark_table_blocks(fs); 479 480 retval = ext2fs_open_inode_scan(fs, 0, &scan); 481 if (retval) { 482 com_err(program_name, retval, _("while opening inode scan")); 483 exit(1); 484 } 485 486 block_buf = malloc(fs->blocksize * 3); 487 if (!block_buf) { 488 com_err(program_name, 0, "Can't allocate block buffer"); 489 exit(1); 490 } 491 492 use_inode_shortcuts(fs, 1); 493 stashed_inode = &inode; 494 while (1) { 495 retval = ext2fs_get_next_inode(scan, &ino, &inode); 496 if (retval == EXT2_ET_BAD_BLOCK_IN_INODE_TABLE) 497 continue; 498 if (retval) { 499 com_err(program_name, retval, 500 _("while getting next inode")); 501 exit(1); 502 } 503 if (ino == 0) 504 break; 505 if (!inode.i_links_count) 506 continue; 507 if (inode.i_file_acl) { 508 ext2fs_mark_block_bitmap(meta_block_map, 509 inode.i_file_acl); 510 } 511 if (!ext2fs_inode_has_valid_blocks(&inode)) 512 continue; 513 514 stashed_ino = ino; 515 pb.ino = ino; 516 pb.is_dir = LINUX_S_ISDIR(inode.i_mode); 517 if (LINUX_S_ISDIR(inode.i_mode) || 518 (LINUX_S_ISLNK(inode.i_mode) && 519 ext2fs_inode_has_valid_blocks(&inode)) || 520 ino == fs->super->s_journal_inum) { 521 retval = ext2fs_block_iterate2(fs, ino, 522 BLOCK_FLAG_READ_ONLY, block_buf, 523 process_dir_block, &pb); 524 if (retval) { 525 com_err(program_name, retval, 526 "while iterating over inode %u", 527 ino); 528 exit(1); 529 } 530 } else { 531 if ((inode.i_flags & EXT4_EXTENTS_FL) || 532 inode.i_block[EXT2_IND_BLOCK] || 533 inode.i_block[EXT2_DIND_BLOCK] || 534 inode.i_block[EXT2_TIND_BLOCK]) { 535 retval = ext2fs_block_iterate2(fs, 536 ino, BLOCK_FLAG_READ_ONLY, block_buf, 537 process_file_block, &pb); 538 if (retval) { 539 com_err(program_name, retval, 540 "while iterating over inode %u", ino); 541 exit(1); 542 } 543 } 544 } 545 } 546 use_inode_shortcuts(fs, 0); 547 output_meta_data_blocks(fs, fd); 548 free(block_buf); 549} 550 551static void install_image(char *device, char *image_fn, int raw_flag) 552{ 553 errcode_t retval; 554 ext2_filsys fs; 555 int open_flag = EXT2_FLAG_IMAGE_FILE; 556 int fd = 0; 557 io_manager io_ptr; 558 io_channel io, image_io; 559 560 if (raw_flag) { 561 com_err(program_name, 0, "Raw images cannot be installed"); 562 exit(1); 563 } 564 565#ifdef CONFIG_TESTIO_DEBUG 566 if (getenv("TEST_IO_FLAGS") || getenv("TEST_IO_BLOCK")) { 567 io_ptr = test_io_manager; 568 test_io_backing_manager = unix_io_manager; 569 } else 570#endif 571 io_ptr = unix_io_manager; 572 573 retval = ext2fs_open (image_fn, open_flag, 0, 0, 574 io_ptr, &fs); 575 if (retval) { 576 com_err (program_name, retval, _("while trying to open %s"), 577 image_fn); 578 exit(1); 579 } 580 581 retval = ext2fs_read_bitmaps (fs); 582 if (retval) { 583 com_err(program_name, retval, "error reading bitmaps"); 584 exit(1); 585 } 586 587#ifdef HAVE_OPEN64 588 fd = open64(image_fn, O_RDONLY); 589#else 590 fd = open(image_fn, O_RDONLY); 591#endif 592 if (fd < 0) { 593 perror(image_fn); 594 exit(1); 595 } 596 597 retval = io_ptr->open(device, IO_FLAG_RW, &io); 598 if (retval) { 599 com_err(device, 0, "while opening device file"); 600 exit(1); 601 } 602 603 image_io = fs->io; 604 605 ext2fs_rewrite_to_io(fs, io); 606 607 if (lseek(fd, fs->image_header->offset_inode, SEEK_SET) < 0) { 608 perror("lseek"); 609 exit(1); 610 } 611 612 retval = ext2fs_image_inode_read(fs, fd, 0); 613 if (retval) { 614 com_err(image_fn, 0, "while restoring the image table"); 615 exit(1); 616 } 617 618 ext2fs_close (fs); 619 exit (0); 620} 621 622int main (int argc, char ** argv) 623{ 624 int c; 625 errcode_t retval; 626 ext2_filsys fs; 627 char *image_fn; 628 int open_flag = 0; 629 int raw_flag = 0; 630 int install_flag = 0; 631 int scramble_flag = 0; 632 int fd = 0; 633 634#ifdef ENABLE_NLS 635 setlocale(LC_MESSAGES, ""); 636 setlocale(LC_CTYPE, ""); 637 bindtextdomain(NLS_CAT_NAME, LOCALEDIR); 638 textdomain(NLS_CAT_NAME); 639#endif 640 fprintf (stderr, "e2image %s (%s)\n", E2FSPROGS_VERSION, 641 E2FSPROGS_DATE); 642 if (argc && *argv) 643 program_name = *argv; 644 add_error_table(&et_ext2_error_table); 645 while ((c = getopt (argc, argv, "rsI")) != EOF) 646 switch (c) { 647 case 'r': 648 raw_flag++; 649 break; 650 case 's': 651 scramble_flag++; 652 break; 653 case 'I': 654 install_flag++; 655 break; 656 default: 657 usage(); 658 } 659 if (optind != argc - 2 ) 660 usage(); 661 device_name = argv[optind]; 662 image_fn = argv[optind+1]; 663 664 if (install_flag) { 665 install_image(device_name, image_fn, raw_flag); 666 exit (0); 667 } 668 669 retval = ext2fs_open (device_name, open_flag, 0, 0, 670 unix_io_manager, &fs); 671 if (retval) { 672 com_err (program_name, retval, _("while trying to open %s"), 673 device_name); 674 fputs(_("Couldn't find valid filesystem superblock.\n"), stdout); 675 exit(1); 676 } 677 678 if (strcmp(image_fn, "-") == 0) 679 fd = 1; 680 else { 681#ifdef HAVE_OPEN64 682 fd = open64(image_fn, O_CREAT|O_TRUNC|O_WRONLY, 0600); 683#else 684 fd = open(image_fn, O_CREAT|O_TRUNC|O_WRONLY, 0600); 685#endif 686 if (fd < 0) { 687 com_err(program_name, errno, 688 _("while trying to open %s"), argv[optind+1]); 689 exit(1); 690 } 691 } 692 693 if (raw_flag) 694 write_raw_image_file(fs, fd, scramble_flag); 695 else 696 write_image_file(fs, fd); 697 698 ext2fs_close (fs); 699 remove_error_table(&et_ext2_error_table); 700 exit (0); 701} 702