1 /*
2  * e2fsck.c - a consistency checker for the new extended file system.
3  *
4  * Copyright (C) 1993, 1994, 1995, 1996, 1997 Theodore Ts'o.
5  *
6  * %Begin-Header%
7  * This file may be redistributed under the terms of the GNU Public
8  * License.
9  * %End-Header%
10  */
11 
12 #include "config.h"
13 #include <errno.h>
14 
15 #include "e2fsck.h"
16 #include "problem.h"
17 
18 /*
19  * This function allocates an e2fsck context
20  */
e2fsck_allocate_context(e2fsck_t * ret)21 errcode_t e2fsck_allocate_context(e2fsck_t *ret)
22 {
23 	e2fsck_t	context;
24 	errcode_t	retval;
25 	char		*time_env;
26 
27 	retval = ext2fs_get_mem(sizeof(struct e2fsck_struct), &context);
28 	if (retval)
29 		return retval;
30 
31 	memset(context, 0, sizeof(struct e2fsck_struct));
32 
33 	context->process_inode_size = 256;
34 	context->ext_attr_ver = 2;
35 	context->blocks_per_page = 1;
36 	context->htree_slack_percentage = 255;
37 
38 	time_env = getenv("E2FSCK_TIME");
39 	if (time_env)
40 		context->now = (time_t) strtoull(time_env, NULL, 0);
41 	else {
42 		context->now = time(0);
43 		if (context->now < 1262322000) /* January 1 2010 */
44 			context->flags |= E2F_FLAG_TIME_INSANE;
45 	}
46 
47 	*ret = context;
48 	return 0;
49 }
50 
51 /*
52  * This function resets an e2fsck context; it is called when e2fsck
53  * needs to be restarted.
54  */
e2fsck_reset_context(e2fsck_t ctx)55 errcode_t e2fsck_reset_context(e2fsck_t ctx)
56 {
57 	int	i;
58 
59 	ctx->flags &= E2F_RESET_FLAGS;
60 	ctx->lost_and_found = 0;
61 	ctx->bad_lost_and_found = 0;
62 	if (ctx->inode_used_map) {
63 		ext2fs_free_inode_bitmap(ctx->inode_used_map);
64 		ctx->inode_used_map = 0;
65 	}
66 	if (ctx->inode_dir_map) {
67 		ext2fs_free_inode_bitmap(ctx->inode_dir_map);
68 		ctx->inode_dir_map = 0;
69 	}
70 	if (ctx->inode_reg_map) {
71 		ext2fs_free_inode_bitmap(ctx->inode_reg_map);
72 		ctx->inode_reg_map = 0;
73 	}
74 	if (ctx->block_found_map) {
75 		ext2fs_free_block_bitmap(ctx->block_found_map);
76 		ctx->block_found_map = 0;
77 	}
78 	if (ctx->inode_link_info) {
79 		ext2fs_free_icount(ctx->inode_link_info);
80 		ctx->inode_link_info = 0;
81 	}
82 	if (ctx->journal_io) {
83 		if (ctx->fs && ctx->fs->io != ctx->journal_io)
84 			io_channel_close(ctx->journal_io);
85 		ctx->journal_io = 0;
86 	}
87 	if (ctx->fs && ctx->fs->dblist) {
88 		ext2fs_free_dblist(ctx->fs->dblist);
89 		ctx->fs->dblist = 0;
90 	}
91 	e2fsck_free_dir_info(ctx);
92 	e2fsck_free_dx_dir_info(ctx);
93 	if (ctx->refcount) {
94 		ea_refcount_free(ctx->refcount);
95 		ctx->refcount = 0;
96 	}
97 	if (ctx->refcount_extra) {
98 		ea_refcount_free(ctx->refcount_extra);
99 		ctx->refcount_extra = 0;
100 	}
101 	if (ctx->ea_block_quota_blocks) {
102 		ea_refcount_free(ctx->ea_block_quota_blocks);
103 		ctx->ea_block_quota_blocks = 0;
104 	}
105 	if (ctx->ea_block_quota_inodes) {
106 		ea_refcount_free(ctx->ea_block_quota_inodes);
107 		ctx->ea_block_quota_inodes = 0;
108 	}
109 	if (ctx->ea_inode_refs) {
110 		ea_refcount_free(ctx->ea_inode_refs);
111 		ctx->ea_inode_refs = 0;
112 	}
113 	if (ctx->block_dup_map) {
114 		ext2fs_free_block_bitmap(ctx->block_dup_map);
115 		ctx->block_dup_map = 0;
116 	}
117 	if (ctx->block_ea_map) {
118 		ext2fs_free_block_bitmap(ctx->block_ea_map);
119 		ctx->block_ea_map = 0;
120 	}
121 	if (ctx->block_metadata_map) {
122 		ext2fs_free_block_bitmap(ctx->block_metadata_map);
123 		ctx->block_metadata_map = 0;
124 	}
125 	if (ctx->inode_bb_map) {
126 		ext2fs_free_inode_bitmap(ctx->inode_bb_map);
127 		ctx->inode_bb_map = 0;
128 	}
129 	if (ctx->inode_bad_map) {
130 		ext2fs_free_inode_bitmap(ctx->inode_bad_map);
131 		ctx->inode_bad_map = 0;
132 	}
133 	if (ctx->inode_imagic_map) {
134 		ext2fs_free_inode_bitmap(ctx->inode_imagic_map);
135 		ctx->inode_imagic_map = 0;
136 	}
137 	if (ctx->dirs_to_hash) {
138 		ext2fs_u32_list_free(ctx->dirs_to_hash);
139 		ctx->dirs_to_hash = 0;
140 	}
141 
142 	/*
143 	 * Clear the array of invalid meta-data flags
144 	 */
145 	if (ctx->invalid_inode_bitmap_flag) {
146 		ext2fs_free_mem(&ctx->invalid_inode_bitmap_flag);
147 		ctx->invalid_inode_bitmap_flag = 0;
148 	}
149 	if (ctx->invalid_block_bitmap_flag) {
150 		ext2fs_free_mem(&ctx->invalid_block_bitmap_flag);
151 		ctx->invalid_block_bitmap_flag = 0;
152 	}
153 	if (ctx->invalid_inode_table_flag) {
154 		ext2fs_free_mem(&ctx->invalid_inode_table_flag);
155 		ctx->invalid_inode_table_flag = 0;
156 	}
157 	if (ctx->encrypted_dirs) {
158 		ext2fs_u32_list_free(ctx->encrypted_dirs);
159 		ctx->encrypted_dirs = 0;
160 	}
161 	if (ctx->casefolded_dirs) {
162 		ext2fs_u32_list_free(ctx->casefolded_dirs);
163 		ctx->casefolded_dirs = 0;
164 	}
165 	if (ctx->inode_count) {
166 		ext2fs_free_icount(ctx->inode_count);
167 		ctx->inode_count = 0;
168 	}
169 
170 	/* Clear statistic counters */
171 	ctx->fs_directory_count = 0;
172 	ctx->fs_regular_count = 0;
173 	ctx->fs_blockdev_count = 0;
174 	ctx->fs_chardev_count = 0;
175 	ctx->fs_links_count = 0;
176 	ctx->fs_symlinks_count = 0;
177 	ctx->fs_fast_symlinks_count = 0;
178 	ctx->fs_fifo_count = 0;
179 	ctx->fs_total_count = 0;
180 	ctx->fs_badblocks_count = 0;
181 	ctx->fs_sockets_count = 0;
182 	ctx->fs_ind_count = 0;
183 	ctx->fs_dind_count = 0;
184 	ctx->fs_tind_count = 0;
185 	ctx->fs_fragmented = 0;
186 	ctx->fs_fragmented_dir = 0;
187 	ctx->large_files = 0;
188 
189 	for (i=0; i < MAX_EXTENT_DEPTH_COUNT; i++)
190 		ctx->extent_depth_count[i] = 0;
191 
192 	/* Reset the superblock to the user's requested value */
193 	ctx->superblock = ctx->use_superblock;
194 
195 	return 0;
196 }
197 
e2fsck_free_context(e2fsck_t ctx)198 void e2fsck_free_context(e2fsck_t ctx)
199 {
200 	if (!ctx)
201 		return;
202 
203 	e2fsck_reset_context(ctx);
204 	if (ctx->blkid)
205 		blkid_put_cache(ctx->blkid);
206 
207 	if (ctx->profile)
208 		profile_release(ctx->profile);
209 
210 	if (ctx->filesystem_name)
211 		ext2fs_free_mem(&ctx->filesystem_name);
212 
213 	if (ctx->device_name)
214 		ext2fs_free_mem(&ctx->device_name);
215 
216 	if (ctx->log_fn)
217 		free(ctx->log_fn);
218 
219 	if (ctx->logf)
220 		fclose(ctx->logf);
221 
222 	if (ctx->problem_log_fn)
223 		free(ctx->problem_log_fn);
224 
225 	if (ctx->problem_logf) {
226 		fputs("</problem_log>\n", ctx->problem_logf);
227 		fclose(ctx->problem_logf);
228 	}
229 	ext2fs_free_mem(&ctx);
230 }
231 
232 /*
233  * This function runs through the e2fsck passes and calls them all,
234  * returning restart, abort, or cancel as necessary...
235  */
236 typedef void (*pass_t)(e2fsck_t ctx);
237 
238 static pass_t e2fsck_passes[] = {
239 	e2fsck_pass1, e2fsck_pass1e, e2fsck_pass2, e2fsck_pass3,
240 	e2fsck_pass4, e2fsck_pass5, 0 };
241 
e2fsck_run(e2fsck_t ctx)242 int e2fsck_run(e2fsck_t ctx)
243 {
244 	int	i;
245 	pass_t	e2fsck_pass;
246 
247 #ifdef HAVE_SETJMP_H
248 	if (setjmp(ctx->abort_loc)) {
249 		ctx->flags &= ~E2F_FLAG_SETJMP_OK;
250 		return (ctx->flags & E2F_FLAG_RUN_RETURN);
251 	}
252 	ctx->flags |= E2F_FLAG_SETJMP_OK;
253 #endif
254 
255 	for (i=0; (e2fsck_pass = e2fsck_passes[i]); i++) {
256 		if (ctx->flags & E2F_FLAG_RUN_RETURN)
257 			break;
258 		if (e2fsck_mmp_update(ctx->fs))
259 			fatal_error(ctx, 0);
260 		e2fsck_pass(ctx);
261 		if (ctx->progress)
262 			(void) (ctx->progress)(ctx, 0, 0, 0);
263 	}
264 	ctx->flags &= ~E2F_FLAG_SETJMP_OK;
265 
266 	if (ctx->flags & E2F_FLAG_RUN_RETURN)
267 		return (ctx->flags & E2F_FLAG_RUN_RETURN);
268 	return 0;
269 }
270