lib/ext2fs/alloc_stats.c

   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 Public
   8  * License.
   9  * %End-Header%
  10  *
  11  */
  12
  13 #include <stdio.h>
  14
  15 #include "ext2_fs.h"
  16 #include "ext2fs.h"
  17
  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 (inuse > 0)
  24                 ext2fs_mark_inode_bitmap(fs->inode_map, ino);
  25         else
  26                 ext2fs_unmark_inode_bitmap(fs->inode_map, ino);
  27         fs->group_desc[group].bg_free_inodes_count -= inuse;
  28         if (isdir)
  29                 fs->group_desc[group].bg_used_dirs_count += inuse;
  30
  31         /* We don't strictly need to be clearing the uninit flag if inuse < 0
  32          * (i.e. freeing inodes) but it also means something is bad. */
  33         fs->group_desc[group].bg_flags &= ~EXT2_BG_INODE_UNINIT;
  34         if (EXT2_HAS_RO_COMPAT_FEATURE(fs->super,
  35                                        EXT4_FEATURE_RO_COMPAT_GDT_CSUM)) {
  36                 ext2_ino_t first_unused_inode = fs->super->s_inodes_per_group -
  37                         fs->group_desc[group].bg_itable_unused +
  38                         group * fs->super->s_inodes_per_group + 1;
  39
  40                 if (ino >= first_unused_inode)
  41                         fs->group_desc[group].bg_itable_unused =
  42                                 group * fs->super->s_inodes_per_group +
  43                                 fs->super->s_inodes_per_group - ino;
  44                 ext2fs_group_desc_csum_set(fs, group);
  45         }
  46
  47         fs->super->s_free_inodes_count -= inuse;
  48         ext2fs_mark_super_dirty(fs);
  49         ext2fs_mark_ib_dirty(fs);
  50 }
  51
  52 void ext2fs_inode_alloc_stats(ext2_filsys fs, ext2_ino_t ino, int inuse)
  53 {
  54         ext2fs_inode_alloc_stats2(fs, ino, inuse, 0);
  55 }
  56
  57 void ext2fs_block_alloc_stats(ext2_filsys fs, blk_t blk, int inuse)
  58 {
  59         int     group = ext2fs_group_of_blk(fs, blk);
  60
  61         if (inuse > 0)
  62                 ext2fs_mark_block_bitmap(fs->block_map, blk);
  63         else
  64                 ext2fs_unmark_block_bitmap(fs->block_map, blk);
  65         fs->group_desc[group].bg_free_blocks_count -= inuse;
  66         fs->group_desc[group].bg_flags &= ~EXT2_BG_BLOCK_UNINIT;
  67         ext2fs_group_desc_csum_set(fs, group);
  68
  69         fs->super->s_free_blocks_count -= inuse;
  70         ext2fs_mark_super_dirty(fs);
  71         ext2fs_mark_bb_dirty(fs);
  72         if (fs->block_alloc_stats)
  73                 (fs->block_alloc_stats)(fs, (blk64_t) blk, inuse);
  74 }
  75
  76 void ext2fs_set_block_alloc_stats_callback(ext2_filsys fs,
  77                                            void (*func)(ext2_filsys fs,
  78                                                         blk64_t blk,
  79                                                         int inuse),
  80                                            void (**old)(ext2_filsys fs,
  81                                                         blk64_t blk,
  82                                                         int inuse))
  83 {
  84         if (!fs || fs->magic != EXT2_ET_MAGIC_EXT2FS_FILSYS)
  85                 return;
  86         if (old)
  87                 *old = fs->block_alloc_stats;
  88
  89         fs->block_alloc_stats = func;
  90 }