Home | History | Annotate | Download | only in ext2fs
      1 /*
      2  * bmove.c --- Move blocks around to make way for a particular
      3  * 	filesystem structure.
      4  *
      5  * Copyright (C) 1997 Theodore Ts'o.  This file may be redistributed
      6  * under the terms of the GNU Public License.
      7  */
      8 
      9 #include <stdio.h>
     10 #include <string.h>
     11 #if HAVE_UNISTD_H
     12 #include <unistd.h>
     13 #endif
     14 #if HAVE_SYS_TYPES_H
     15 #include <sys/types.h>
     16 #endif
     17 #if HAVE_SYS_TIME_H
     18 #include <sys/time.h>
     19 #endif
     20 
     21 #include "ext2_fs.h"
     22 #include "ext2fsP.h"
     23 
     24 struct process_block_struct {
     25 	ext2_ino_t		ino;
     26 	struct ext2_inode *	inode;
     27 	ext2fs_block_bitmap	reserve;
     28 	ext2fs_block_bitmap	alloc_map;
     29 	errcode_t		error;
     30 	char			*buf;
     31 	int			add_dir;
     32 	int			flags;
     33 };
     34 
     35 static int process_block(ext2_filsys fs, blk_t	*block_nr,
     36 			 e2_blkcnt_t blockcnt, blk_t ref_block,
     37 			 int ref_offset, void *priv_data)
     38 {
     39 	struct process_block_struct *pb;
     40 	errcode_t	retval;
     41 	int		ret;
     42 	blk_t		block, orig;
     43 
     44 	pb = (struct process_block_struct *) priv_data;
     45 	block = orig = *block_nr;
     46 	ret = 0;
     47 
     48 	/*
     49 	 * Let's see if this is one which we need to relocate
     50 	 */
     51 	if (ext2fs_test_block_bitmap(pb->reserve, block)) {
     52 		do {
     53 			if (++block >= fs->super->s_blocks_count)
     54 				block = fs->super->s_first_data_block;
     55 			if (block == orig) {
     56 				pb->error = EXT2_ET_BLOCK_ALLOC_FAIL;
     57 				return BLOCK_ABORT;
     58 			}
     59 		} while (ext2fs_test_block_bitmap(pb->reserve, block) ||
     60 			 ext2fs_test_block_bitmap(pb->alloc_map, block));
     61 
     62 		retval = io_channel_read_blk(fs->io, orig, 1, pb->buf);
     63 		if (retval) {
     64 			pb->error = retval;
     65 			return BLOCK_ABORT;
     66 		}
     67 		retval = io_channel_write_blk(fs->io, block, 1, pb->buf);
     68 		if (retval) {
     69 			pb->error = retval;
     70 			return BLOCK_ABORT;
     71 		}
     72 		*block_nr = block;
     73 		ext2fs_mark_block_bitmap(pb->alloc_map, block);
     74 		ret = BLOCK_CHANGED;
     75 		if (pb->flags & EXT2_BMOVE_DEBUG)
     76 			printf("ino=%ld, blockcnt=%lld, %u->%u\n", pb->ino,
     77 			       blockcnt, orig, block);
     78 	}
     79 	if (pb->add_dir) {
     80 		retval = ext2fs_add_dir_block(fs->dblist, pb->ino,
     81 					      block, (int) blockcnt);
     82 		if (retval) {
     83 			pb->error = retval;
     84 			ret |= BLOCK_ABORT;
     85 		}
     86 	}
     87 	return ret;
     88 }
     89 
     90 errcode_t ext2fs_move_blocks(ext2_filsys fs,
     91 			     ext2fs_block_bitmap reserve,
     92 			     ext2fs_block_bitmap alloc_map,
     93 			     int flags)
     94 {
     95 	ext2_ino_t	ino;
     96 	struct ext2_inode inode;
     97 	errcode_t	retval;
     98 	struct process_block_struct pb;
     99 	ext2_inode_scan	scan;
    100 	char		*block_buf;
    101 
    102 	retval = ext2fs_open_inode_scan(fs, 0, &scan);
    103 	if (retval)
    104 		return retval;
    105 
    106 	pb.reserve = reserve;
    107 	pb.error = 0;
    108 	pb.alloc_map = alloc_map ? alloc_map : fs->block_map;
    109 	pb.flags = flags;
    110 
    111 	retval = ext2fs_get_array(4, fs->blocksize, &block_buf);
    112 	if (retval)
    113 		return retval;
    114 	pb.buf = block_buf + fs->blocksize * 3;
    115 
    116 	/*
    117 	 * If GET_DBLIST is set in the flags field, then we should
    118 	 * gather directory block information while we're doing the
    119 	 * block move.
    120 	 */
    121 	if (flags & EXT2_BMOVE_GET_DBLIST) {
    122 		if (fs->dblist) {
    123 			ext2fs_free_dblist(fs->dblist);
    124 			fs->dblist = NULL;
    125 		}
    126 		retval = ext2fs_init_dblist(fs, 0);
    127 		if (retval)
    128 			return retval;
    129 	}
    130 
    131 	retval = ext2fs_get_next_inode(scan, &ino, &inode);
    132 	if (retval)
    133 		return retval;
    134 
    135 	while (ino) {
    136 		if ((inode.i_links_count == 0) ||
    137 		    !ext2fs_inode_has_valid_blocks(&inode))
    138 			goto next;
    139 
    140 		pb.ino = ino;
    141 		pb.inode = &inode;
    142 
    143 		pb.add_dir = (LINUX_S_ISDIR(inode.i_mode) &&
    144 			      flags & EXT2_BMOVE_GET_DBLIST);
    145 
    146 		retval = ext2fs_block_iterate2(fs, ino, 0, block_buf,
    147 					      process_block, &pb);
    148 		if (retval)
    149 			return retval;
    150 		if (pb.error)
    151 			return pb.error;
    152 
    153 	next:
    154 		retval = ext2fs_get_next_inode(scan, &ino, &inode);
    155 		if (retval == EXT2_ET_BAD_BLOCK_IN_INODE_TABLE)
    156 			goto next;
    157 	}
    158 	return 0;
    159 }
    160 
    161