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 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
  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 #ifndef OMIT_COM_ERR
  24         if (ino > fs->super->s_inodes_count) {
  25                 com_err("ext2fs_inode_alloc_stats2", 0,
  26                         "Illegal inode number: %lu", (unsigned long) ino);
  27                 return;
  28         }
  29 #endif
  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 (EXT2_HAS_RO_COMPAT_FEATURE(fs->super,
  42                                        EXT4_FEATURE_RO_COMPAT_GDT_CSUM)) {
  43                 ext2_ino_t first_unused_inode = fs->super->s_inodes_per_group -
  44                         ext2fs_bg_itable_unused(fs, group) +
  45                         group * fs->super->s_inodes_per_group + 1;
  46
  47                 if (ino >= first_unused_inode)
  48                         ext2fs_bg_itable_unused_set(fs, group, group * fs->super->s_inodes_per_group + fs->super->s_inodes_per_group - ino);
  49                 ext2fs_group_desc_csum_set(fs, group);
  50         }
  51
  52         fs->super->s_free_inodes_count -= inuse;
  53         ext2fs_mark_super_dirty(fs);
  54         ext2fs_mark_ib_dirty(fs);
  55 }
  56
  57 void ext2fs_inode_alloc_stats(ext2_filsys fs, ext2_ino_t ino, int inuse)
  58 {
  59         ext2fs_inode_alloc_stats2(fs, ino, inuse, 0);
  60 }
  61
  62 void ext2fs_block_alloc_stats2(ext2_filsys fs, blk64_t blk, int inuse)
  63 {
  64         int     group = ext2fs_group_of_blk2(fs, blk);
  65
  66 #ifndef OMIT_COM_ERR
  67         if (blk >= ext2fs_blocks_count(fs->super)) {
  68                 com_err("ext2fs_block_alloc_stats", 0,
  69                         "Illegal block number: %lu", (unsigned long) blk);
  70                 return;
  71         }
  72 #endif
  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 * 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
  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
  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
 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                 blk_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 * n);
 144                 blk += n;
 145                 num -= n;
 146         }
 147         ext2fs_mark_super_dirty(fs);
 148         ext2fs_mark_bb_dirty(fs);
 149 }