1 /*
2 * alloc_stats.c --- Update allocation statistics for ext2fs
3 *
4 * Copyright (C) 2001 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 "config.h"
13 #include <stdio.h>
14
15 #include "ext2_fs.h"
16 #include "ext2fs.h"
17
ext2fs_inode_alloc_stats2(ext2_filsys fs,ext2_ino_t ino,int inuse,int isdir)18 void ext2fs_inode_alloc_stats2(ext2_filsys fs, ext2_ino_t ino,
19 int inuse, int isdir)
20 {
21 int group = ext2fs_group_of_ino(fs, ino);
22
23 if (ino > fs->super->s_inodes_count) {
24 #ifndef OMIT_COM_ERR
25 com_err("ext2fs_inode_alloc_stats2", 0,
26 "Illegal inode number: %lu", (unsigned long) ino);
27 #endif
28 return;
29 }
30 if (inuse > 0)
31 ext2fs_mark_inode_bitmap2(fs->inode_map, ino);
32 else
33 ext2fs_unmark_inode_bitmap2(fs->inode_map, ino);
34 ext2fs_bg_free_inodes_count_set(fs, group, ext2fs_bg_free_inodes_count(fs, group) - inuse);
35 if (isdir)
36 ext2fs_bg_used_dirs_count_set(fs, group, ext2fs_bg_used_dirs_count(fs, group) + inuse);
37
38 /* We don't strictly need to be clearing the uninit flag if inuse < 0
39 * (i.e. freeing inodes) but it also means something is bad. */
40 ext2fs_bg_flags_clear(fs, group, EXT2_BG_INODE_UNINIT);
41 if (ext2fs_has_group_desc_csum(fs)) {
42 ext2_ino_t first_unused_inode = fs->super->s_inodes_per_group -
43 ext2fs_bg_itable_unused(fs, group) +
44 group * fs->super->s_inodes_per_group + 1;
45
46 if (ino >= first_unused_inode)
47 ext2fs_bg_itable_unused_set(fs, group, group * fs->super->s_inodes_per_group + fs->super->s_inodes_per_group - ino);
48 ext2fs_group_desc_csum_set(fs, group);
49 }
50
51 fs->super->s_free_inodes_count -= inuse;
52 ext2fs_mark_super_dirty(fs);
53 ext2fs_mark_ib_dirty(fs);
54 }
55
ext2fs_inode_alloc_stats(ext2_filsys fs,ext2_ino_t ino,int inuse)56 void ext2fs_inode_alloc_stats(ext2_filsys fs, ext2_ino_t ino, int inuse)
57 {
58 ext2fs_inode_alloc_stats2(fs, ino, inuse, 0);
59 }
60
ext2fs_block_alloc_stats2(ext2_filsys fs,blk64_t blk,int inuse)61 void ext2fs_block_alloc_stats2(ext2_filsys fs, blk64_t blk, int inuse)
62 {
63 int group = ext2fs_group_of_blk2(fs, blk);
64
65 if (blk < fs->super->s_first_data_block ||
66 blk >= ext2fs_blocks_count(fs->super)) {
67 #ifndef OMIT_COM_ERR
68 com_err("ext2fs_block_alloc_stats", 0,
69 "Illegal block number: %lu", (unsigned long) blk);
70 #endif
71 return;
72 }
73 if (inuse > 0)
74 ext2fs_mark_block_bitmap2(fs->block_map, blk);
75 else
76 ext2fs_unmark_block_bitmap2(fs->block_map, blk);
77 ext2fs_bg_free_blocks_count_set(fs, group, ext2fs_bg_free_blocks_count(fs, group) - inuse);
78 ext2fs_bg_flags_clear(fs, group, EXT2_BG_BLOCK_UNINIT);
79 ext2fs_group_desc_csum_set(fs, group);
80
81 ext2fs_free_blocks_count_add(fs->super,
82 -inuse * (blk64_t) EXT2FS_CLUSTER_RATIO(fs));
83 ext2fs_mark_super_dirty(fs);
84 ext2fs_mark_bb_dirty(fs);
85 if (fs->block_alloc_stats)
86 (fs->block_alloc_stats)(fs, (blk64_t) blk, inuse);
87 }
88
ext2fs_block_alloc_stats(ext2_filsys fs,blk_t blk,int inuse)89 void ext2fs_block_alloc_stats(ext2_filsys fs, blk_t blk, int inuse)
90 {
91 ext2fs_block_alloc_stats2(fs, blk, inuse);
92 }
93
ext2fs_set_block_alloc_stats_callback(ext2_filsys fs,void (* func)(ext2_filsys fs,blk64_t blk,int inuse),void (** old)(ext2_filsys fs,blk64_t blk,int inuse))94 void ext2fs_set_block_alloc_stats_callback(ext2_filsys fs,
95 void (*func)(ext2_filsys fs,
96 blk64_t blk,
97 int inuse),
98 void (**old)(ext2_filsys fs,
99 blk64_t blk,
100 int inuse))
101 {
102 if (!fs || fs->magic != EXT2_ET_MAGIC_EXT2FS_FILSYS)
103 return;
104 if (old)
105 *old = fs->block_alloc_stats;
106
107 fs->block_alloc_stats = func;
108 }
109
ext2fs_block_alloc_stats_range(ext2_filsys fs,blk64_t blk,blk_t num,int inuse)110 void ext2fs_block_alloc_stats_range(ext2_filsys fs, blk64_t blk,
111 blk_t num, int inuse)
112 {
113 #ifndef OMIT_COM_ERR
114 if (blk + num > ext2fs_blocks_count(fs->super)) {
115 com_err("ext2fs_block_alloc_stats_range", 0,
116 "Illegal block range: %llu (%u) ",
117 (unsigned long long) blk, num);
118 return;
119 }
120 #endif
121 if (inuse == 0)
122 return;
123 if (inuse > 0) {
124 ext2fs_mark_block_bitmap_range2(fs->block_map, blk, num);
125 inuse = 1;
126 } else {
127 ext2fs_unmark_block_bitmap_range2(fs->block_map, blk, num);
128 inuse = -1;
129 }
130 while (num) {
131 int group = ext2fs_group_of_blk2(fs, blk);
132 blk64_t last_blk = ext2fs_group_last_block2(fs, group);
133 blk64_t n = num;
134
135 if (blk + num > last_blk)
136 n = last_blk - blk + 1;
137
138 ext2fs_bg_free_blocks_count_set(fs, group,
139 ext2fs_bg_free_blocks_count(fs, group) -
140 inuse*n/EXT2FS_CLUSTER_RATIO(fs));
141 ext2fs_bg_flags_clear(fs, group, EXT2_BG_BLOCK_UNINIT);
142 ext2fs_group_desc_csum_set(fs, group);
143 ext2fs_free_blocks_count_add(fs->super, -inuse * (blk64_t) n);
144 blk += n;
145 num -= n;
146 }
147 ext2fs_mark_super_dirty(fs);
148 ext2fs_mark_bb_dirty(fs);
149 if (fs->block_alloc_stats_range)
150 (fs->block_alloc_stats_range)(fs, blk, num, inuse);
151 }
152
ext2fs_set_block_alloc_stats_range_callback(ext2_filsys fs,void (* func)(ext2_filsys fs,blk64_t blk,blk_t num,int inuse),void (** old)(ext2_filsys fs,blk64_t blk,blk_t num,int inuse))153 void ext2fs_set_block_alloc_stats_range_callback(ext2_filsys fs,
154 void (*func)(ext2_filsys fs, blk64_t blk,
155 blk_t num, int inuse),
156 void (**old)(ext2_filsys fs, blk64_t blk,
157 blk_t num, int inuse))
158 {
159 if (!fs || fs->magic != EXT2_ET_MAGIC_EXT2FS_FILSYS)
160 return;
161 if (old)
162 *old = fs->block_alloc_stats_range;
163
164 fs->block_alloc_stats_range = func;
165 }
166