1 /*
2 * expand.c --- expand an ext2fs directory
3 *
4 * Copyright (C) 1993, 1994, 1995, 1996, 1997, 1998, 1999 Theodore Ts'o.
5 *
6 * %Begin-Header%
7 * This file may be redistributed under the terms of the GNU Library
8 * General Public License, version 2.
9 * %End-Header%
10 */
11
12 #include <stdio.h>
13 #include <string.h>
14 #if HAVE_UNISTD_H
15 #include <unistd.h>
16 #endif
17
18 #include "ext2_fs.h"
19 #include "ext2fs.h"
20
21 struct expand_dir_struct {
22 int done;
23 int newblocks;
24 blk64_t goal;
25 errcode_t err;
26 };
27
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)28 static int expand_dir_proc(ext2_filsys fs,
29 blk64_t *blocknr,
30 e2_blkcnt_t blockcnt,
31 blk64_t ref_block EXT2FS_ATTR((unused)),
32 int ref_offset EXT2FS_ATTR((unused)),
33 void *priv_data)
34 {
35 struct expand_dir_struct *es = (struct expand_dir_struct *) priv_data;
36 blk64_t new_blk;
37 char *block;
38 errcode_t retval;
39
40 if (*blocknr) {
41 if (blockcnt >= 0)
42 es->goal = *blocknr;
43 return 0;
44 }
45 if (blockcnt &&
46 (EXT2FS_B2C(fs, es->goal) == EXT2FS_B2C(fs, es->goal+1)))
47 new_blk = es->goal+1;
48 else {
49 es->goal &= ~EXT2FS_CLUSTER_MASK(fs);
50 retval = ext2fs_new_block2(fs, es->goal, 0, &new_blk);
51 if (retval) {
52 es->err = retval;
53 return BLOCK_ABORT;
54 }
55 es->newblocks++;
56 ext2fs_block_alloc_stats2(fs, new_blk, +1);
57 }
58 if (blockcnt > 0) {
59 retval = ext2fs_new_dir_block(fs, 0, 0, &block);
60 if (retval) {
61 es->err = retval;
62 return BLOCK_ABORT;
63 }
64 es->done = 1;
65 retval = ext2fs_write_dir_block(fs, new_blk, block);
66 } else {
67 retval = ext2fs_get_mem(fs->blocksize, &block);
68 if (retval) {
69 es->err = retval;
70 return BLOCK_ABORT;
71 }
72 memset(block, 0, fs->blocksize);
73 retval = io_channel_write_blk64(fs->io, new_blk, 1, block);
74 }
75 if (blockcnt >= 0)
76 es->goal = new_blk;
77 if (retval) {
78 es->err = retval;
79 return BLOCK_ABORT;
80 }
81 ext2fs_free_mem(&block);
82 *blocknr = new_blk;
83
84 if (es->done)
85 return (BLOCK_CHANGED | BLOCK_ABORT);
86 else
87 return BLOCK_CHANGED;
88 }
89
ext2fs_expand_dir(ext2_filsys fs,ext2_ino_t dir)90 errcode_t ext2fs_expand_dir(ext2_filsys fs, ext2_ino_t dir)
91 {
92 errcode_t retval;
93 struct expand_dir_struct es;
94 struct ext2_inode inode;
95
96 EXT2_CHECK_MAGIC(fs, EXT2_ET_MAGIC_EXT2FS_FILSYS);
97
98 if (!(fs->flags & EXT2_FLAG_RW))
99 return EXT2_ET_RO_FILSYS;
100
101 if (!fs->block_map)
102 return EXT2_ET_NO_BLOCK_BITMAP;
103
104 retval = ext2fs_check_directory(fs, dir);
105 if (retval)
106 return retval;
107
108 es.done = 0;
109 es.err = 0;
110 es.goal = 0;
111 es.newblocks = 0;
112
113 retval = ext2fs_block_iterate3(fs, dir, BLOCK_FLAG_APPEND,
114 0, expand_dir_proc, &es);
115
116 if (es.err)
117 return es.err;
118 if (!es.done)
119 return EXT2_ET_EXPAND_DIR_ERR;
120
121 /*
122 * Update the size and block count fields in the inode.
123 */
124 retval = ext2fs_read_inode(fs, dir, &inode);
125 if (retval)
126 return retval;
127
128 inode.i_size += fs->blocksize;
129 ext2fs_iblk_add_blocks(fs, &inode, es.newblocks);
130
131 retval = ext2fs_write_inode(fs, dir, &inode);
132 if (retval)
133 return retval;
134
135 return 0;
136 }
137