Whamcloud - gitweb
Tighten up profile parsing code by combining functions
[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 <errno.h>
13
14 #include "e2fsck.h"
15 #include "problem.h"
16
17 /*
18  * This function allocates an e2fsck context 
19  */
20 errcode_t e2fsck_allocate_context(e2fsck_t *ret)
21 {
22         e2fsck_t        context;
23         errcode_t       retval;
24         char            *time_env;
25
26         retval = ext2fs_get_mem(sizeof(struct e2fsck_struct), &context);
27         if (retval)
28                 return retval;
29         
30         memset(context, 0, sizeof(struct e2fsck_struct));
31
32         context->process_inode_size = 256;
33         context->ext_attr_ver = 2;
34         
35         time_env = getenv("E2FSCK_TIME");
36         if (time_env)
37                 context->now = strtoul(time_env, NULL, 0);
38         else
39                 context->now = time(0);
40
41         *ret = context;
42         return 0;
43 }
44
45 /*
46  * This function resets an e2fsck context; it is called when e2fsck
47  * needs to be restarted.
48  */
49 errcode_t e2fsck_reset_context(e2fsck_t ctx)
50 {
51         ctx->flags = 0;
52         ctx->lost_and_found = 0;
53         ctx->bad_lost_and_found = 0;
54         if (ctx->inode_used_map) {
55                 ext2fs_free_inode_bitmap(ctx->inode_used_map);
56                 ctx->inode_used_map = 0;
57         }
58         if (ctx->inode_dir_map) {
59                 ext2fs_free_inode_bitmap(ctx->inode_dir_map);
60                 ctx->inode_dir_map = 0;
61         }
62         if (ctx->inode_reg_map) {
63                 ext2fs_free_inode_bitmap(ctx->inode_reg_map);
64                 ctx->inode_reg_map = 0;
65         }
66         if (ctx->block_found_map) {
67                 ext2fs_free_block_bitmap(ctx->block_found_map);
68                 ctx->block_found_map = 0;
69         }
70         if (ctx->inode_link_info) {
71                 ext2fs_free_icount(ctx->inode_link_info);
72                 ctx->inode_link_info = 0;
73         }
74         if (ctx->journal_io) {
75                 if (ctx->fs && ctx->fs->io != ctx->journal_io)
76                         io_channel_close(ctx->journal_io);
77                 ctx->journal_io = 0;
78         }
79         if (ctx->fs && ctx->fs->dblist) {
80                 ext2fs_free_dblist(ctx->fs->dblist);
81                 ctx->fs->dblist = 0;
82         }
83         e2fsck_free_dir_info(ctx);
84 #ifdef ENABLE_HTREE
85         e2fsck_free_dx_dir_info(ctx);
86 #endif
87         if (ctx->refcount) {
88                 ea_refcount_free(ctx->refcount);
89                 ctx->refcount = 0;
90         }
91         if (ctx->refcount_extra) {
92                 ea_refcount_free(ctx->refcount_extra);
93                 ctx->refcount_extra = 0;
94         }
95         if (ctx->block_dup_map) {
96                 ext2fs_free_block_bitmap(ctx->block_dup_map);
97                 ctx->block_dup_map = 0;
98         }
99         if (ctx->block_ea_map) {
100                 ext2fs_free_block_bitmap(ctx->block_ea_map);
101                 ctx->block_ea_map = 0;
102         }
103         if (ctx->inode_bb_map) {
104                 ext2fs_free_inode_bitmap(ctx->inode_bb_map);
105                 ctx->inode_bb_map = 0;
106         }
107         if (ctx->inode_bad_map) {
108                 ext2fs_free_inode_bitmap(ctx->inode_bad_map);
109                 ctx->inode_bad_map = 0;
110         }
111         if (ctx->inode_imagic_map) {
112                 ext2fs_free_inode_bitmap(ctx->inode_imagic_map);
113                 ctx->inode_imagic_map = 0;
114         }
115         if (ctx->dirs_to_hash) {
116                 ext2fs_u32_list_free(ctx->dirs_to_hash);
117                 ctx->dirs_to_hash = 0;
118         }
119
120         /*
121          * Clear the array of invalid meta-data flags
122          */
123         if (ctx->invalid_inode_bitmap_flag) {
124                 ext2fs_free_mem(&ctx->invalid_inode_bitmap_flag);
125                 ctx->invalid_inode_bitmap_flag = 0;
126         }
127         if (ctx->invalid_block_bitmap_flag) {
128                 ext2fs_free_mem(&ctx->invalid_block_bitmap_flag);
129                 ctx->invalid_block_bitmap_flag = 0;
130         }
131         if (ctx->invalid_inode_table_flag) {
132                 ext2fs_free_mem(&ctx->invalid_inode_table_flag);
133                 ctx->invalid_inode_table_flag = 0;
134         }
135
136         /* Clear statistic counters */
137         ctx->fs_directory_count = 0;
138         ctx->fs_regular_count = 0;
139         ctx->fs_blockdev_count = 0;
140         ctx->fs_chardev_count = 0;
141         ctx->fs_links_count = 0;
142         ctx->fs_symlinks_count = 0;
143         ctx->fs_fast_symlinks_count = 0;
144         ctx->fs_fifo_count = 0;
145         ctx->fs_total_count = 0;
146         ctx->fs_badblocks_count = 0;
147         ctx->fs_sockets_count = 0;
148         ctx->fs_ind_count = 0;
149         ctx->fs_dind_count = 0;
150         ctx->fs_tind_count = 0;
151         ctx->fs_fragmented = 0;
152         ctx->large_files = 0;
153
154         /* Reset the superblock to the user's requested value */
155         ctx->superblock = ctx->use_superblock;
156         
157         return 0;
158 }
159
160 void e2fsck_free_context(e2fsck_t ctx)
161 {
162         if (!ctx)
163                 return;
164         
165         e2fsck_reset_context(ctx);
166         if (ctx->blkid)
167                 blkid_put_cache(ctx->blkid);
168
169         if (ctx->profile)
170                 profile_release(ctx->profile);
171                         
172         ext2fs_free_mem(&ctx);
173 }
174
175 /*
176  * This function runs through the e2fsck passes and calls them all,
177  * returning restart, abort, or cancel as necessary...
178  */
179 typedef void (*pass_t)(e2fsck_t ctx);
180
181 pass_t e2fsck_passes[] = {
182         e2fsck_pass1, e2fsck_pass2, e2fsck_pass3, e2fsck_pass4,
183         e2fsck_pass5, 0 };
184
185 #define E2F_FLAG_RUN_RETURN     (E2F_FLAG_SIGNAL_MASK|E2F_FLAG_RESTART)
186
187 int e2fsck_run(e2fsck_t ctx)
188 {
189         int     i;
190         pass_t  e2fsck_pass;
191
192 #ifdef HAVE_SETJMP_H
193         if (setjmp(ctx->abort_loc)) {
194                 ctx->flags &= ~E2F_FLAG_SETJMP_OK;
195                 return (ctx->flags & E2F_FLAG_RUN_RETURN);
196         }
197         ctx->flags |= E2F_FLAG_SETJMP_OK;
198 #endif
199                 
200         for (i=0; (e2fsck_pass = e2fsck_passes[i]); i++) {
201                 if (ctx->flags & E2F_FLAG_RUN_RETURN)
202                         break;
203                 e2fsck_pass(ctx);
204                 if (ctx->progress)
205                         (void) (ctx->progress)(ctx, 0, 0, 0);
206         }
207         ctx->flags &= ~E2F_FLAG_SETJMP_OK;
208         
209         if (ctx->flags & E2F_FLAG_RUN_RETURN)
210                 return (ctx->flags & E2F_FLAG_RUN_RETURN);
211         return 0;
212 }