Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 1 | /* |
| 2 | * alloc.c --- allocate new inodes, blocks for ext2fs |
| 3 | * |
Theodore Ts'o | 21c84b7 | 1997-04-29 16:15:03 +0000 | [diff] [blame] | 4 | * Copyright (C) 1993, 1994, 1995, 1996 Theodore Ts'o. |
| 5 | * |
| 6 | * %Begin-Header% |
Theodore Ts'o | 543547a | 2010-05-17 21:31:56 -0400 | [diff] [blame] | 7 | * This file may be redistributed under the terms of the GNU Library |
| 8 | * General Public License, version 2. |
Theodore Ts'o | 21c84b7 | 1997-04-29 16:15:03 +0000 | [diff] [blame] | 9 | * %End-Header% |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 10 | */ |
| 11 | |
Theodore Ts'o | d1154eb | 2011-09-18 17:34:37 -0400 | [diff] [blame] | 12 | #include "config.h" |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 13 | #include <stdio.h> |
Theodore Ts'o | 4cbe8af | 1997-08-10 23:07:40 +0000 | [diff] [blame] | 14 | #if HAVE_UNISTD_H |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 15 | #include <unistd.h> |
Theodore Ts'o | 4cbe8af | 1997-08-10 23:07:40 +0000 | [diff] [blame] | 16 | #endif |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 17 | #include <time.h> |
Theodore Ts'o | 30fab29 | 1997-10-25 22:37:42 +0000 | [diff] [blame] | 18 | #include <string.h> |
Theodore Ts'o | 1d2ff46 | 1997-10-19 23:00:21 +0000 | [diff] [blame] | 19 | #if HAVE_SYS_STAT_H |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 20 | #include <sys/stat.h> |
Theodore Ts'o | 1d2ff46 | 1997-10-19 23:00:21 +0000 | [diff] [blame] | 21 | #endif |
| 22 | #if HAVE_SYS_TYPES_H |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 23 | #include <sys/types.h> |
Theodore Ts'o | 1d2ff46 | 1997-10-19 23:00:21 +0000 | [diff] [blame] | 24 | #endif |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 25 | |
Theodore Ts'o | b5abe6f | 1998-01-19 14:47:53 +0000 | [diff] [blame] | 26 | #include "ext2_fs.h" |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 27 | #include "ext2fs.h" |
| 28 | |
| 29 | /* |
Theodore Ts'o | c71d781 | 2008-08-22 02:52:12 -0400 | [diff] [blame] | 30 | * Check for uninit block bitmaps and deal with them appropriately |
| 31 | */ |
Theodore Ts'o | 03fa6f8 | 2008-11-16 10:03:00 -0500 | [diff] [blame] | 32 | static void check_block_uninit(ext2_filsys fs, ext2fs_block_bitmap map, |
Valerie Aurora Henson | 8f82ef9 | 2009-08-05 00:27:10 -0400 | [diff] [blame] | 33 | dgrp_t group) |
Theodore Ts'o | c71d781 | 2008-08-22 02:52:12 -0400 | [diff] [blame] | 34 | { |
Theodore Ts'o | 03fa6f8 | 2008-11-16 10:03:00 -0500 | [diff] [blame] | 35 | blk_t i; |
Jose R. Santos | 20f2ccb | 2009-07-11 21:29:30 -0400 | [diff] [blame] | 36 | blk64_t blk, super_blk, old_desc_blk, new_desc_blk; |
Theodore Ts'o | c71d781 | 2008-08-22 02:52:12 -0400 | [diff] [blame] | 37 | int old_desc_blocks; |
| 38 | |
| 39 | if (!(EXT2_HAS_RO_COMPAT_FEATURE(fs->super, |
| 40 | EXT4_FEATURE_RO_COMPAT_GDT_CSUM)) || |
Theodore Ts'o | cd65a24 | 2009-10-25 21:42:12 -0400 | [diff] [blame] | 41 | !(ext2fs_bg_flags_test(fs, group, EXT2_BG_BLOCK_UNINIT))) |
Theodore Ts'o | c71d781 | 2008-08-22 02:52:12 -0400 | [diff] [blame] | 42 | return; |
| 43 | |
Theodore Ts'o | 027b057 | 2013-01-03 13:42:38 -0500 | [diff] [blame] | 44 | blk = ext2fs_group_first_block2(fs, group); |
Theodore Ts'o | c71d781 | 2008-08-22 02:52:12 -0400 | [diff] [blame] | 45 | |
Jose R. Santos | 20f2ccb | 2009-07-11 21:29:30 -0400 | [diff] [blame] | 46 | ext2fs_super_and_bgd_loc2(fs, group, &super_blk, |
| 47 | &old_desc_blk, &new_desc_blk, 0); |
Theodore Ts'o | c71d781 | 2008-08-22 02:52:12 -0400 | [diff] [blame] | 48 | |
| 49 | if (fs->super->s_feature_incompat & |
| 50 | EXT2_FEATURE_INCOMPAT_META_BG) |
| 51 | old_desc_blocks = fs->super->s_first_meta_bg; |
| 52 | else |
| 53 | old_desc_blocks = fs->desc_blocks + fs->super->s_reserved_gdt_blocks; |
| 54 | |
Theodore Ts'o | b0ecb78 | 2011-06-10 18:51:58 -0400 | [diff] [blame] | 55 | for (i=0; i < fs->super->s_blocks_per_group; i++, blk++) |
| 56 | ext2fs_fast_unmark_block_bitmap2(map, blk); |
| 57 | |
Theodore Ts'o | 027b057 | 2013-01-03 13:42:38 -0500 | [diff] [blame] | 58 | blk = ext2fs_group_first_block2(fs, group); |
Theodore Ts'o | c71d781 | 2008-08-22 02:52:12 -0400 | [diff] [blame] | 59 | for (i=0; i < fs->super->s_blocks_per_group; i++, blk++) { |
| 60 | if ((blk == super_blk) || |
| 61 | (old_desc_blk && old_desc_blocks && |
| 62 | (blk >= old_desc_blk) && |
| 63 | (blk < old_desc_blk + old_desc_blocks)) || |
| 64 | (new_desc_blk && (blk == new_desc_blk)) || |
Valerie Aurora Henson | d7cca6b | 2009-10-25 21:43:47 -0400 | [diff] [blame] | 65 | (blk == ext2fs_block_bitmap_loc(fs, group)) || |
| 66 | (blk == ext2fs_inode_bitmap_loc(fs, group)) || |
| 67 | (blk >= ext2fs_inode_table_loc(fs, group) && |
| 68 | (blk < ext2fs_inode_table_loc(fs, group) |
Theodore Ts'o | c71d781 | 2008-08-22 02:52:12 -0400 | [diff] [blame] | 69 | + fs->inode_blocks_per_group))) |
Valerie Aurora Henson | 8f82ef9 | 2009-08-05 00:27:10 -0400 | [diff] [blame] | 70 | ext2fs_fast_mark_block_bitmap2(map, blk); |
Theodore Ts'o | c71d781 | 2008-08-22 02:52:12 -0400 | [diff] [blame] | 71 | } |
Eric Sandeen | e633b58 | 2009-10-25 21:41:32 -0400 | [diff] [blame] | 72 | ext2fs_bg_flags_clear(fs, group, EXT2_BG_BLOCK_UNINIT); |
Theodore Ts'o | c71d781 | 2008-08-22 02:52:12 -0400 | [diff] [blame] | 73 | ext2fs_group_desc_csum_set(fs, group); |
Theodore Ts'o | 538eb87 | 2012-03-25 22:24:10 -0400 | [diff] [blame] | 74 | ext2fs_mark_super_dirty(fs); |
| 75 | ext2fs_mark_bb_dirty(fs); |
Theodore Ts'o | c71d781 | 2008-08-22 02:52:12 -0400 | [diff] [blame] | 76 | } |
| 77 | |
| 78 | /* |
| 79 | * Check for uninit inode bitmaps and deal with them appropriately |
| 80 | */ |
Theodore Ts'o | 03fa6f8 | 2008-11-16 10:03:00 -0500 | [diff] [blame] | 81 | static void check_inode_uninit(ext2_filsys fs, ext2fs_inode_bitmap map, |
Theodore Ts'o | c71d781 | 2008-08-22 02:52:12 -0400 | [diff] [blame] | 82 | dgrp_t group) |
| 83 | { |
Theodore Ts'o | 03fa6f8 | 2008-11-16 10:03:00 -0500 | [diff] [blame] | 84 | ext2_ino_t i, ino; |
Theodore Ts'o | c71d781 | 2008-08-22 02:52:12 -0400 | [diff] [blame] | 85 | |
| 86 | if (!(EXT2_HAS_RO_COMPAT_FEATURE(fs->super, |
| 87 | EXT4_FEATURE_RO_COMPAT_GDT_CSUM)) || |
Theodore Ts'o | cd65a24 | 2009-10-25 21:42:12 -0400 | [diff] [blame] | 88 | !(ext2fs_bg_flags_test(fs, group, EXT2_BG_INODE_UNINIT))) |
Theodore Ts'o | c71d781 | 2008-08-22 02:52:12 -0400 | [diff] [blame] | 89 | return; |
| 90 | |
| 91 | ino = (group * fs->super->s_inodes_per_group) + 1; |
| 92 | for (i=0; i < fs->super->s_inodes_per_group; i++, ino++) |
Valerie Aurora Henson | 8f82ef9 | 2009-08-05 00:27:10 -0400 | [diff] [blame] | 93 | ext2fs_fast_unmark_inode_bitmap2(map, ino); |
Theodore Ts'o | c71d781 | 2008-08-22 02:52:12 -0400 | [diff] [blame] | 94 | |
Eric Sandeen | e633b58 | 2009-10-25 21:41:32 -0400 | [diff] [blame] | 95 | ext2fs_bg_flags_clear(fs, group, EXT2_BG_INODE_UNINIT); |
Theodore Ts'o | 538eb87 | 2012-03-25 22:24:10 -0400 | [diff] [blame] | 96 | ext2fs_group_desc_csum_set(fs, group); |
| 97 | ext2fs_mark_ib_dirty(fs); |
| 98 | ext2fs_mark_super_dirty(fs); |
Theodore Ts'o | c71d781 | 2008-08-22 02:52:12 -0400 | [diff] [blame] | 99 | check_block_uninit(fs, fs->block_map, group); |
| 100 | } |
| 101 | |
| 102 | /* |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 103 | * Right now, just search forward from the parent directory's block |
| 104 | * group to find the next free inode. |
| 105 | * |
| 106 | * Should have a special policy for directories. |
| 107 | */ |
Theodore Ts'o | efc6f62 | 2008-08-27 23:07:54 -0400 | [diff] [blame] | 108 | errcode_t ext2fs_new_inode(ext2_filsys fs, ext2_ino_t dir, |
Theodore Ts'o | 5443492 | 2003-12-07 01:28:50 -0500 | [diff] [blame] | 109 | int mode EXT2FS_ATTR((unused)), |
Theodore Ts'o | 31dbecd | 2001-01-11 04:54:39 +0000 | [diff] [blame] | 110 | ext2fs_inode_bitmap map, ext2_ino_t *ret) |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 111 | { |
Sami Liedes | c1a1e7f | 2012-03-10 22:36:12 +0200 | [diff] [blame] | 112 | ext2_ino_t start_inode = 0; |
| 113 | ext2_ino_t i, ino_in_group, upto, first_zero; |
| 114 | errcode_t retval; |
| 115 | dgrp_t group; |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 116 | |
Theodore Ts'o | f3db356 | 1997-04-26 13:34:30 +0000 | [diff] [blame] | 117 | EXT2_CHECK_MAGIC(fs, EXT2_ET_MAGIC_EXT2FS_FILSYS); |
Theodore Ts'o | efc6f62 | 2008-08-27 23:07:54 -0400 | [diff] [blame] | 118 | |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 119 | if (!map) |
| 120 | map = fs->inode_map; |
| 121 | if (!map) |
| 122 | return EXT2_ET_NO_INODE_BITMAP; |
Theodore Ts'o | efc6f62 | 2008-08-27 23:07:54 -0400 | [diff] [blame] | 123 | |
Sami Liedes | c1a1e7f | 2012-03-10 22:36:12 +0200 | [diff] [blame] | 124 | if (dir > 0) { |
| 125 | group = (dir - 1) / EXT2_INODES_PER_GROUP(fs->super); |
| 126 | start_inode = (group * EXT2_INODES_PER_GROUP(fs->super)) + 1; |
| 127 | } |
Theodore Ts'o | 7f88b04 | 1997-04-26 14:48:50 +0000 | [diff] [blame] | 128 | if (start_inode < EXT2_FIRST_INODE(fs->super)) |
| 129 | start_inode = EXT2_FIRST_INODE(fs->super); |
Theodore Ts'o | a93d406 | 2009-01-22 15:55:49 -0500 | [diff] [blame] | 130 | if (start_inode > fs->super->s_inodes_count) |
| 131 | return EXT2_ET_INODE_ALLOC_FAIL; |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 132 | i = start_inode; |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 133 | do { |
Sami Liedes | c1a1e7f | 2012-03-10 22:36:12 +0200 | [diff] [blame] | 134 | ino_in_group = (i - 1) % EXT2_INODES_PER_GROUP(fs->super); |
| 135 | group = (i - 1) / EXT2_INODES_PER_GROUP(fs->super); |
Theodore Ts'o | c71d781 | 2008-08-22 02:52:12 -0400 | [diff] [blame] | 136 | |
Sami Liedes | c1a1e7f | 2012-03-10 22:36:12 +0200 | [diff] [blame] | 137 | check_inode_uninit(fs, map, group); |
| 138 | upto = i + (EXT2_INODES_PER_GROUP(fs->super) - ino_in_group); |
| 139 | if (i < start_inode && upto >= start_inode) |
| 140 | upto = start_inode - 1; |
| 141 | if (upto > fs->super->s_inodes_count) |
| 142 | upto = fs->super->s_inodes_count; |
| 143 | |
| 144 | retval = ext2fs_find_first_zero_inode_bitmap2(map, i, upto, |
| 145 | &first_zero); |
| 146 | if (retval == 0) { |
| 147 | i = first_zero; |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 148 | break; |
Sami Liedes | 7555677 | 2012-03-22 19:42:38 -0400 | [diff] [blame] | 149 | } |
Sami Liedes | c1a1e7f | 2012-03-10 22:36:12 +0200 | [diff] [blame] | 150 | if (retval != ENOENT) |
| 151 | return EXT2_ET_INODE_ALLOC_FAIL; |
| 152 | i = upto + 1; |
| 153 | if (i > fs->super->s_inodes_count) |
| 154 | i = EXT2_FIRST_INODE(fs->super); |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 155 | } while (i != start_inode); |
Theodore Ts'o | efc6f62 | 2008-08-27 23:07:54 -0400 | [diff] [blame] | 156 | |
Valerie Aurora Henson | 8f82ef9 | 2009-08-05 00:27:10 -0400 | [diff] [blame] | 157 | if (ext2fs_test_inode_bitmap2(map, i)) |
Theodore Ts'o | 1f0b6c1 | 1997-10-31 06:07:47 +0000 | [diff] [blame] | 158 | return EXT2_ET_INODE_ALLOC_FAIL; |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 159 | *ret = i; |
| 160 | return 0; |
| 161 | } |
| 162 | |
| 163 | /* |
| 164 | * Stupid algorithm --- we now just search forward starting from the |
| 165 | * goal. Should put in a smarter one someday.... |
| 166 | */ |
Jose R. Santos | 8a5e181 | 2009-06-01 16:15:40 -0400 | [diff] [blame] | 167 | errcode_t ext2fs_new_block2(ext2_filsys fs, blk64_t goal, |
| 168 | ext2fs_block_bitmap map, blk64_t *ret) |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 169 | { |
Jose R. Santos | 8a5e181 | 2009-06-01 16:15:40 -0400 | [diff] [blame] | 170 | blk64_t i; |
Theodore Ts'o | b0ecb78 | 2011-06-10 18:51:58 -0400 | [diff] [blame] | 171 | int c_ratio; |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 172 | |
Theodore Ts'o | f3db356 | 1997-04-26 13:34:30 +0000 | [diff] [blame] | 173 | EXT2_CHECK_MAGIC(fs, EXT2_ET_MAGIC_EXT2FS_FILSYS); |
| 174 | |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 175 | if (!map) |
| 176 | map = fs->block_map; |
| 177 | if (!map) |
| 178 | return EXT2_ET_NO_BLOCK_BITMAP; |
Valerie Aurora Henson | 4efbac6 | 2009-09-07 20:46:34 -0400 | [diff] [blame] | 179 | if (!goal || (goal >= ext2fs_blocks_count(fs->super))) |
Theodore Ts'o | a29f4d3 | 1997-04-29 21:26:48 +0000 | [diff] [blame] | 180 | goal = fs->super->s_first_data_block; |
| 181 | i = goal; |
Theodore Ts'o | b0ecb78 | 2011-06-10 18:51:58 -0400 | [diff] [blame] | 182 | c_ratio = 1 << ext2fs_get_bitmap_granularity(map); |
Theodore Ts'o | 81c6377 | 2011-07-07 22:29:00 -0400 | [diff] [blame] | 183 | if (c_ratio > 1) |
| 184 | goal &= ~EXT2FS_CLUSTER_MASK(fs); |
Theodore Ts'o | c71d781 | 2008-08-22 02:52:12 -0400 | [diff] [blame] | 185 | check_block_uninit(fs, map, |
| 186 | (i - fs->super->s_first_data_block) / |
| 187 | EXT2_BLOCKS_PER_GROUP(fs->super)); |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 188 | do { |
Theodore Ts'o | c71d781 | 2008-08-22 02:52:12 -0400 | [diff] [blame] | 189 | if (((i - fs->super->s_first_data_block) % |
| 190 | EXT2_BLOCKS_PER_GROUP(fs->super)) == 0) |
| 191 | check_block_uninit(fs, map, |
| 192 | (i - fs->super->s_first_data_block) / |
| 193 | EXT2_BLOCKS_PER_GROUP(fs->super)); |
| 194 | |
Valerie Aurora Henson | 8f82ef9 | 2009-08-05 00:27:10 -0400 | [diff] [blame] | 195 | if (!ext2fs_fast_test_block_bitmap2(map, i)) { |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 196 | *ret = i; |
| 197 | return 0; |
| 198 | } |
Theodore Ts'o | b0ecb78 | 2011-06-10 18:51:58 -0400 | [diff] [blame] | 199 | i = (i + c_ratio) & ~(c_ratio - 1); |
Valerie Aurora Henson | 4efbac6 | 2009-09-07 20:46:34 -0400 | [diff] [blame] | 200 | if (i >= ext2fs_blocks_count(fs->super)) |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 201 | i = fs->super->s_first_data_block; |
| 202 | } while (i != goal); |
Theodore Ts'o | 1f0b6c1 | 1997-10-31 06:07:47 +0000 | [diff] [blame] | 203 | return EXT2_ET_BLOCK_ALLOC_FAIL; |
Theodore Ts'o | c555aeb | 1997-10-25 04:16:53 +0000 | [diff] [blame] | 204 | } |
| 205 | |
Jose R. Santos | 8a5e181 | 2009-06-01 16:15:40 -0400 | [diff] [blame] | 206 | errcode_t ext2fs_new_block(ext2_filsys fs, blk_t goal, |
| 207 | ext2fs_block_bitmap map, blk_t *ret) |
| 208 | { |
| 209 | errcode_t retval; |
| 210 | blk64_t val; |
| 211 | retval = ext2fs_new_block2(fs, goal, map, &val); |
| 212 | if (!retval) |
| 213 | *ret = (blk_t) val; |
| 214 | return retval; |
| 215 | } |
| 216 | |
Theodore Ts'o | c555aeb | 1997-10-25 04:16:53 +0000 | [diff] [blame] | 217 | /* |
Theodore Ts'o | 30fab29 | 1997-10-25 22:37:42 +0000 | [diff] [blame] | 218 | * This function zeros out the allocated block, and updates all of the |
| 219 | * appropriate filesystem records. |
Theodore Ts'o | c555aeb | 1997-10-25 04:16:53 +0000 | [diff] [blame] | 220 | */ |
Jose R. Santos | 8a5e181 | 2009-06-01 16:15:40 -0400 | [diff] [blame] | 221 | errcode_t ext2fs_alloc_block2(ext2_filsys fs, blk64_t goal, |
| 222 | char *block_buf, blk64_t *ret) |
Theodore Ts'o | c555aeb | 1997-10-25 04:16:53 +0000 | [diff] [blame] | 223 | { |
| 224 | errcode_t retval; |
Jose R. Santos | 8a5e181 | 2009-06-01 16:15:40 -0400 | [diff] [blame] | 225 | blk64_t block; |
Theodore Ts'o | 30fab29 | 1997-10-25 22:37:42 +0000 | [diff] [blame] | 226 | char *buf = 0; |
Theodore Ts'o | c555aeb | 1997-10-25 04:16:53 +0000 | [diff] [blame] | 227 | |
Theodore Ts'o | 30fab29 | 1997-10-25 22:37:42 +0000 | [diff] [blame] | 228 | if (!block_buf) { |
Theodore Ts'o | c4e3d3f | 2003-08-01 09:41:07 -0400 | [diff] [blame] | 229 | retval = ext2fs_get_mem(fs->blocksize, &buf); |
Theodore Ts'o | 7b4e453 | 1997-10-26 03:41:24 +0000 | [diff] [blame] | 230 | if (retval) |
| 231 | return retval; |
Theodore Ts'o | 30fab29 | 1997-10-25 22:37:42 +0000 | [diff] [blame] | 232 | block_buf = buf; |
| 233 | } |
| 234 | memset(block_buf, 0, fs->blocksize); |
Theodore Ts'o | c555aeb | 1997-10-25 04:16:53 +0000 | [diff] [blame] | 235 | |
Theodore Ts'o | f5c562e | 2008-06-02 17:21:37 -0400 | [diff] [blame] | 236 | if (fs->get_alloc_block) { |
Jose R. Santos | 8a5e181 | 2009-06-01 16:15:40 -0400 | [diff] [blame] | 237 | retval = (fs->get_alloc_block)(fs, goal, &block); |
Theodore Ts'o | f5c562e | 2008-06-02 17:21:37 -0400 | [diff] [blame] | 238 | if (retval) |
| 239 | goto fail; |
Theodore Ts'o | f5c562e | 2008-06-02 17:21:37 -0400 | [diff] [blame] | 240 | } else { |
| 241 | if (!fs->block_map) { |
| 242 | retval = ext2fs_read_block_bitmap(fs); |
| 243 | if (retval) |
| 244 | goto fail; |
| 245 | } |
| 246 | |
Jose R. Santos | 8a5e181 | 2009-06-01 16:15:40 -0400 | [diff] [blame] | 247 | retval = ext2fs_new_block2(fs, goal, 0, &block); |
Theodore Ts'o | 30fab29 | 1997-10-25 22:37:42 +0000 | [diff] [blame] | 248 | if (retval) |
| 249 | goto fail; |
| 250 | } |
| 251 | |
Jose R. Santos | 8a5e181 | 2009-06-01 16:15:40 -0400 | [diff] [blame] | 252 | retval = io_channel_write_blk64(fs->io, block, 1, block_buf); |
Theodore Ts'o | 30fab29 | 1997-10-25 22:37:42 +0000 | [diff] [blame] | 253 | if (retval) |
| 254 | goto fail; |
Theodore Ts'o | efc6f62 | 2008-08-27 23:07:54 -0400 | [diff] [blame] | 255 | |
Jose R. Santos | 8a5e181 | 2009-06-01 16:15:40 -0400 | [diff] [blame] | 256 | ext2fs_block_alloc_stats2(fs, block, +1); |
Theodore Ts'o | 30fab29 | 1997-10-25 22:37:42 +0000 | [diff] [blame] | 257 | *ret = block; |
Theodore Ts'o | 30fab29 | 1997-10-25 22:37:42 +0000 | [diff] [blame] | 258 | |
| 259 | fail: |
| 260 | if (buf) |
Theodore Ts'o | c4e3d3f | 2003-08-01 09:41:07 -0400 | [diff] [blame] | 261 | ext2fs_free_mem(&buf); |
Theodore Ts'o | 30fab29 | 1997-10-25 22:37:42 +0000 | [diff] [blame] | 262 | return retval; |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 263 | } |
| 264 | |
Jose R. Santos | 8a5e181 | 2009-06-01 16:15:40 -0400 | [diff] [blame] | 265 | errcode_t ext2fs_alloc_block(ext2_filsys fs, blk_t goal, |
| 266 | char *block_buf, blk_t *ret) |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 267 | { |
Jose R. Santos | 8a5e181 | 2009-06-01 16:15:40 -0400 | [diff] [blame] | 268 | errcode_t retval; |
| 269 | blk64_t val; |
| 270 | retval = ext2fs_alloc_block2(fs, goal, block_buf, &val); |
| 271 | if (!retval) |
| 272 | *ret = (blk_t) val; |
| 273 | return retval; |
| 274 | } |
| 275 | |
| 276 | errcode_t ext2fs_get_free_blocks2(ext2_filsys fs, blk64_t start, blk64_t finish, |
| 277 | int num, ext2fs_block_bitmap map, blk64_t *ret) |
| 278 | { |
Andreas Dilger | 96367ad | 2011-06-15 22:17:38 -0400 | [diff] [blame] | 279 | blk64_t b = start; |
Theodore Ts'o | b0ecb78 | 2011-06-10 18:51:58 -0400 | [diff] [blame] | 280 | int c_ratio; |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 281 | |
Theodore Ts'o | f3db356 | 1997-04-26 13:34:30 +0000 | [diff] [blame] | 282 | EXT2_CHECK_MAGIC(fs, EXT2_ET_MAGIC_EXT2FS_FILSYS); |
| 283 | |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 284 | if (!map) |
| 285 | map = fs->block_map; |
| 286 | if (!map) |
| 287 | return EXT2_ET_NO_BLOCK_BITMAP; |
| 288 | if (!b) |
| 289 | b = fs->super->s_first_data_block; |
| 290 | if (!finish) |
| 291 | finish = start; |
| 292 | if (!num) |
| 293 | num = 1; |
Theodore Ts'o | b0ecb78 | 2011-06-10 18:51:58 -0400 | [diff] [blame] | 294 | c_ratio = 1 << ext2fs_get_bitmap_granularity(map); |
| 295 | b &= ~(c_ratio - 1); |
| 296 | finish &= ~(c_ratio -1); |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 297 | do { |
Valerie Aurora Henson | 4efbac6 | 2009-09-07 20:46:34 -0400 | [diff] [blame] | 298 | if (b+num-1 > ext2fs_blocks_count(fs->super)) |
Theodore Ts'o | 21c84b7 | 1997-04-29 16:15:03 +0000 | [diff] [blame] | 299 | b = fs->super->s_first_data_block; |
Valerie Aurora Henson | 8f82ef9 | 2009-08-05 00:27:10 -0400 | [diff] [blame] | 300 | if (ext2fs_fast_test_block_bitmap_range2(map, b, num)) { |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 301 | *ret = b; |
| 302 | return 0; |
| 303 | } |
Theodore Ts'o | b0ecb78 | 2011-06-10 18:51:58 -0400 | [diff] [blame] | 304 | b += c_ratio; |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 305 | } while (b != finish); |
Theodore Ts'o | 1f0b6c1 | 1997-10-31 06:07:47 +0000 | [diff] [blame] | 306 | return EXT2_ET_BLOCK_ALLOC_FAIL; |
Theodore Ts'o | 3839e65 | 1997-04-26 13:21:57 +0000 | [diff] [blame] | 307 | } |
| 308 | |
Jose R. Santos | 8a5e181 | 2009-06-01 16:15:40 -0400 | [diff] [blame] | 309 | errcode_t ext2fs_get_free_blocks(ext2_filsys fs, blk_t start, blk_t finish, |
| 310 | int num, ext2fs_block_bitmap map, blk_t *ret) |
| 311 | { |
| 312 | errcode_t retval; |
| 313 | blk64_t val; |
| 314 | retval = ext2fs_get_free_blocks2(fs, start, finish, num, map, &val); |
| 315 | if(!retval) |
| 316 | *ret = (blk_t) val; |
| 317 | return retval; |
| 318 | } |
| 319 | |
Theodore Ts'o | efc6f62 | 2008-08-27 23:07:54 -0400 | [diff] [blame] | 320 | void ext2fs_set_alloc_block_callback(ext2_filsys fs, |
Theodore Ts'o | f5c562e | 2008-06-02 17:21:37 -0400 | [diff] [blame] | 321 | errcode_t (*func)(ext2_filsys fs, |
| 322 | blk64_t goal, |
| 323 | blk64_t *ret), |
| 324 | errcode_t (**old)(ext2_filsys fs, |
| 325 | blk64_t goal, |
| 326 | blk64_t *ret)) |
| 327 | { |
| 328 | if (!fs || fs->magic != EXT2_ET_MAGIC_EXT2FS_FILSYS) |
| 329 | return; |
| 330 | |
| 331 | if (old) |
| 332 | *old = fs->get_alloc_block; |
| 333 | |
| 334 | fs->get_alloc_block = func; |
| 335 | } |