Whamcloud - gitweb
e2fsck: fix memory leaks with pfsck enabled
[tools/e2fsprogs.git] / e2fsck / e2fsck.c
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  */
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  */
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_casefold_map) {
79                 ext2fs_free_block_bitmap(ctx->inode_casefold_map);
80                 ctx->inode_casefold_map = 0;
81         }
82         if (ctx->inodes_to_rebuild) {
83                 ext2fs_free_inode_bitmap(ctx->inodes_to_rebuild);
84                 ctx->inodes_to_rebuild = 0;
85         }
86         if (ctx->inode_link_info) {
87                 ext2fs_free_icount(ctx->inode_link_info);
88                 ctx->inode_link_info = 0;
89         }
90         if (ctx->journal_io) {
91                 if (ctx->fs && ctx->fs->io != ctx->journal_io)
92                         io_channel_close(ctx->journal_io);
93                 ctx->journal_io = 0;
94         }
95         if (ctx->fs && ctx->fs->dblist) {
96                 ext2fs_free_dblist(ctx->fs->dblist);
97                 ctx->fs->dblist = 0;
98         }
99         e2fsck_free_dir_info(ctx);
100         e2fsck_free_dx_dir_info(ctx);
101         if (ctx->refcount) {
102                 ea_refcount_free(ctx->refcount);
103                 ctx->refcount = 0;
104         }
105         if (ctx->refcount_extra) {
106                 ea_refcount_free(ctx->refcount_extra);
107                 ctx->refcount_extra = 0;
108         }
109         if (ctx->refcount_orig) {
110                 ea_refcount_free(ctx->refcount_orig);
111                 ctx->refcount_orig = 0;
112         }
113         if (ctx->ea_block_quota_blocks) {
114                 ea_refcount_free(ctx->ea_block_quota_blocks);
115                 ctx->ea_block_quota_blocks = 0;
116         }
117         if (ctx->ea_block_quota_inodes) {
118                 ea_refcount_free(ctx->ea_block_quota_inodes);
119                 ctx->ea_block_quota_inodes = 0;
120         }
121         if (ctx->ea_inode_refs) {
122                 ea_refcount_free(ctx->ea_inode_refs);
123                 ctx->ea_inode_refs = 0;
124         }
125         if (ctx->block_dup_map) {
126                 ext2fs_free_block_bitmap(ctx->block_dup_map);
127                 ctx->block_dup_map = 0;
128         }
129         if (ctx->block_ea_map) {
130                 ext2fs_free_block_bitmap(ctx->block_ea_map);
131                 ctx->block_ea_map = 0;
132         }
133         if (ctx->block_metadata_map) {
134                 ext2fs_free_block_bitmap(ctx->block_metadata_map);
135                 ctx->block_metadata_map = 0;
136         }
137         if (ctx->inode_bb_map) {
138                 ext2fs_free_inode_bitmap(ctx->inode_bb_map);
139                 ctx->inode_bb_map = 0;
140         }
141         if (ctx->inode_bad_map) {
142                 ext2fs_free_inode_bitmap(ctx->inode_bad_map);
143                 ctx->inode_bad_map = 0;
144         }
145         if (ctx->inode_imagic_map) {
146                 ext2fs_free_inode_bitmap(ctx->inode_imagic_map);
147                 ctx->inode_imagic_map = 0;
148         }
149         if (ctx->dirs_to_hash) {
150                 ext2fs_u32_list_free(ctx->dirs_to_hash);
151                 ctx->dirs_to_hash = 0;
152         }
153         destroy_encrypted_file_info(ctx);
154
155         /*
156          * Clear the array of invalid meta-data flags
157          */
158         if (ctx->invalid_inode_bitmap_flag) {
159                 ext2fs_free_mem(&ctx->invalid_inode_bitmap_flag);
160                 ctx->invalid_inode_bitmap_flag = 0;
161         }
162         if (ctx->invalid_block_bitmap_flag) {
163                 ext2fs_free_mem(&ctx->invalid_block_bitmap_flag);
164                 ctx->invalid_block_bitmap_flag = 0;
165         }
166         if (ctx->invalid_inode_table_flag) {
167                 ext2fs_free_mem(&ctx->invalid_inode_table_flag);
168                 ctx->invalid_inode_table_flag = 0;
169         }
170         if (ctx->casefolded_dirs) {
171                 ext2fs_u32_list_free(ctx->casefolded_dirs);
172                 ctx->casefolded_dirs = 0;
173         }
174         if (ctx->inode_count) {
175                 ext2fs_free_icount(ctx->inode_count);
176                 ctx->inode_count = 0;
177         }
178
179         /* Clear statistic counters */
180         ctx->fs_directory_count = 0;
181         ctx->fs_regular_count = 0;
182         ctx->fs_blockdev_count = 0;
183         ctx->fs_chardev_count = 0;
184         ctx->fs_links_count = 0;
185         ctx->fs_symlinks_count = 0;
186         ctx->fs_fast_symlinks_count = 0;
187         ctx->fs_fifo_count = 0;
188         ctx->fs_total_count = 0;
189         ctx->fs_badblocks_count = 0;
190         ctx->fs_sockets_count = 0;
191         ctx->fs_ind_count = 0;
192         ctx->fs_dind_count = 0;
193         ctx->fs_tind_count = 0;
194         ctx->fs_fragmented = 0;
195         ctx->fs_fragmented_dir = 0;
196         ctx->large_files = 0;
197         ctx->large_dirs = 0;
198 #ifdef HAVE_PTHREAD
199         ctx->fs_need_locking = 0;
200 #endif
201
202         for (i=0; i < MAX_EXTENT_DEPTH_COUNT; i++)
203                 ctx->extent_depth_count[i] = 0;
204
205         /* Reset the superblock to the user's requested value */
206         ctx->superblock = ctx->use_superblock;
207
208         return 0;
209 }
210
211 void e2fsck_free_context(e2fsck_t ctx)
212 {
213         if (!ctx)
214                 return;
215
216         e2fsck_reset_context(ctx);
217         if (ctx->blkid)
218                 blkid_put_cache(ctx->blkid);
219
220         if (ctx->profile)
221                 profile_release(ctx->profile);
222
223         if (ctx->filesystem_name)
224                 ext2fs_free_mem(&ctx->filesystem_name);
225
226         if (ctx->device_name)
227                 ext2fs_free_mem(&ctx->device_name);
228
229         if (ctx->log_fn)
230                 free(ctx->log_fn);
231
232         if (ctx->logf)
233                 fclose(ctx->logf);
234
235         if (ctx->problem_log_fn)
236                 free(ctx->problem_log_fn);
237
238         if (ctx->problem_logf) {
239                 fputs("</problem_log>\n", ctx->problem_logf);
240                 fclose(ctx->problem_logf);
241         }
242         ext2fs_free_mem(&ctx);
243 }
244
245 /*
246  * This function runs through the e2fsck passes and calls them all,
247  * returning restart, abort, or cancel as necessary...
248  */
249 typedef void (*pass_t)(e2fsck_t ctx);
250
251 static pass_t e2fsck_passes[] = {
252         e2fsck_pass1, e2fsck_pass1e, e2fsck_pass2, e2fsck_pass3,
253         e2fsck_pass4, e2fsck_pass5, 0 };
254
255 int e2fsck_run(e2fsck_t ctx)
256 {
257         int     i;
258         pass_t  e2fsck_pass;
259
260 #ifdef HAVE_SETJMP_H
261         if (setjmp(ctx->abort_loc)) {
262                 ctx->flags &= ~E2F_FLAG_SETJMP_OK;
263                 return (ctx->flags & E2F_FLAG_RUN_RETURN);
264         }
265         ctx->flags |= E2F_FLAG_SETJMP_OK;
266 #endif
267
268         for (i=0; (e2fsck_pass = e2fsck_passes[i]); i++) {
269                 if (ctx->flags & E2F_FLAG_RUN_RETURN)
270                         break;
271                 if (e2fsck_mmp_update(ctx->fs))
272                         fatal_error(ctx, 0);
273                 e2fsck_pass(ctx);
274                 if (ctx->progress)
275                         (void) (ctx->progress)(ctx, 0, 0, 0);
276         }
277         ctx->flags &= ~E2F_FLAG_SETJMP_OK;
278
279         if (ctx->flags & E2F_FLAG_RUN_RETURN)
280                 return (ctx->flags & E2F_FLAG_RUN_RETURN);
281         return 0;
282 }