mirror of https://github.com/vitalif/e2fsprogs
141 lines
2.9 KiB
C
141 lines
2.9 KiB
C
/*
|
|
* expand.c --- expand an ext2fs directory
|
|
*
|
|
* Copyright (C) 1993, 1994, 1995, 1996, 1997, 1998, 1999 Theodore Ts'o.
|
|
*
|
|
* %Begin-Header%
|
|
* This file may be redistributed under the terms of the GNU Library
|
|
* General Public License, version 2.
|
|
* %End-Header%
|
|
*/
|
|
|
|
#include "config.h"
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#if HAVE_UNISTD_H
|
|
#include <unistd.h>
|
|
#endif
|
|
|
|
#include "ext2_fs.h"
|
|
#include "ext2fs.h"
|
|
|
|
struct expand_dir_struct {
|
|
int done;
|
|
int newblocks;
|
|
blk64_t goal;
|
|
errcode_t err;
|
|
ext2_ino_t dir;
|
|
};
|
|
|
|
static int expand_dir_proc(ext2_filsys fs,
|
|
blk64_t *blocknr,
|
|
e2_blkcnt_t blockcnt,
|
|
blk64_t ref_block EXT2FS_ATTR((unused)),
|
|
int ref_offset EXT2FS_ATTR((unused)),
|
|
void *priv_data)
|
|
{
|
|
struct expand_dir_struct *es = (struct expand_dir_struct *) priv_data;
|
|
blk64_t new_blk;
|
|
char *block;
|
|
errcode_t retval;
|
|
|
|
if (*blocknr) {
|
|
if (blockcnt >= 0)
|
|
es->goal = *blocknr;
|
|
return 0;
|
|
}
|
|
if (blockcnt &&
|
|
(EXT2FS_B2C(fs, es->goal) == EXT2FS_B2C(fs, es->goal+1)))
|
|
new_blk = es->goal+1;
|
|
else {
|
|
es->goal &= ~EXT2FS_CLUSTER_MASK(fs);
|
|
retval = ext2fs_new_block2(fs, es->goal, 0, &new_blk);
|
|
if (retval) {
|
|
es->err = retval;
|
|
return BLOCK_ABORT;
|
|
}
|
|
es->newblocks++;
|
|
ext2fs_block_alloc_stats2(fs, new_blk, +1);
|
|
}
|
|
if (blockcnt > 0) {
|
|
retval = ext2fs_new_dir_block(fs, 0, 0, &block);
|
|
if (retval) {
|
|
es->err = retval;
|
|
return BLOCK_ABORT;
|
|
}
|
|
es->done = 1;
|
|
retval = ext2fs_write_dir_block4(fs, new_blk, block, 0,
|
|
es->dir);
|
|
} else {
|
|
retval = ext2fs_get_mem(fs->blocksize, &block);
|
|
if (retval) {
|
|
es->err = retval;
|
|
return BLOCK_ABORT;
|
|
}
|
|
memset(block, 0, fs->blocksize);
|
|
retval = io_channel_write_blk64(fs->io, new_blk, 1, block);
|
|
}
|
|
if (blockcnt >= 0)
|
|
es->goal = new_blk;
|
|
if (retval) {
|
|
es->err = retval;
|
|
return BLOCK_ABORT;
|
|
}
|
|
ext2fs_free_mem(&block);
|
|
*blocknr = new_blk;
|
|
|
|
if (es->done)
|
|
return (BLOCK_CHANGED | BLOCK_ABORT);
|
|
else
|
|
return BLOCK_CHANGED;
|
|
}
|
|
|
|
errcode_t ext2fs_expand_dir(ext2_filsys fs, ext2_ino_t dir)
|
|
{
|
|
errcode_t retval;
|
|
struct expand_dir_struct es;
|
|
struct ext2_inode inode;
|
|
|
|
EXT2_CHECK_MAGIC(fs, EXT2_ET_MAGIC_EXT2FS_FILSYS);
|
|
|
|
if (!(fs->flags & EXT2_FLAG_RW))
|
|
return EXT2_ET_RO_FILSYS;
|
|
|
|
if (!fs->block_map)
|
|
return EXT2_ET_NO_BLOCK_BITMAP;
|
|
|
|
retval = ext2fs_check_directory(fs, dir);
|
|
if (retval)
|
|
return retval;
|
|
|
|
es.done = 0;
|
|
es.err = 0;
|
|
es.goal = 0;
|
|
es.newblocks = 0;
|
|
es.dir = dir;
|
|
|
|
retval = ext2fs_block_iterate3(fs, dir, BLOCK_FLAG_APPEND,
|
|
0, expand_dir_proc, &es);
|
|
|
|
if (es.err)
|
|
return es.err;
|
|
if (!es.done)
|
|
return EXT2_ET_EXPAND_DIR_ERR;
|
|
|
|
/*
|
|
* Update the size and block count fields in the inode.
|
|
*/
|
|
retval = ext2fs_read_inode(fs, dir, &inode);
|
|
if (retval)
|
|
return retval;
|
|
|
|
inode.i_size += fs->blocksize;
|
|
ext2fs_iblk_add_blocks(fs, &inode, es.newblocks);
|
|
|
|
retval = ext2fs_write_inode(fs, dir, &inode);
|
|
if (retval)
|
|
return retval;
|
|
|
|
return 0;
|
|
}
|