Refactor FEATURE_RO to FRO
[lwext4.git] / lwext4 / ext4_balloc.c
index 71074585af8b7c3096b86f34f987256812aa10b9..c3b2ce0acee4bb4dd75cc71a1ee4e25a29dd7446 100644 (file)
@@ -42,6 +42,7 @@
 #include "ext4_config.h"
 #include "ext4_balloc.h"
 #include "ext4_super.h"
+#include "ext4_crc32c.h"
 #include "ext4_block_group.h"
 #include "ext4_fs.h"
 #include "ext4_bitmap.h"
@@ -53,9 +54,9 @@
  * @return Block group index
  */
 uint32_t ext4_balloc_get_bgid_of_block(struct ext4_sblock *s,
-                                      uint32_t baddr)
+                                      uint64_t baddr)
 {
-       if (ext4_get32(s, first_data_block))
+       if (ext4_get32(s, first_data_block) && baddr)
                baddr--;
 
        return baddr / ext4_get32(s, blocks_per_group);
@@ -66,10 +67,10 @@ uint32_t ext4_balloc_get_bgid_of_block(struct ext4_sblock *s,
  * @param bgid block group index
  * @return Block address
  */
-uint32_t ext4_balloc_get_block_of_bgid(struct ext4_sblock *s,
+uint64_t ext4_balloc_get_block_of_bgid(struct ext4_sblock *s,
                                       uint32_t bgid)
 {
-       uint32_t baddr = 0;
+       uint64_t baddr = 0;
        if (ext4_get32(s, first_data_block))
                baddr++;
 
@@ -77,7 +78,46 @@ uint32_t ext4_balloc_get_block_of_bgid(struct ext4_sblock *s,
        return baddr;
 }
 
-int ext4_balloc_free_block(struct ext4_inode_ref *inode_ref, uint32_t baddr)
+static uint32_t ext4_balloc_bitmap_csum(struct ext4_sblock *sb,
+                                       void *bitmap)
+{
+       uint32_t checksum = 0;
+       if (ext4_sb_has_feature_read_only(sb,
+                               EXT4_FRO_COM_METADATA_CSUM)) {
+               uint32_t blocks_per_group =
+                       ext4_get32(sb, blocks_per_group);
+
+               /* First calculate crc32 checksum against fs uuid */
+               checksum = ext4_crc32c(~0, sb->uuid, sizeof(sb->uuid));
+               /* Then calculate crc32 checksum against block_group_desc */
+               checksum = ext4_crc32c(checksum, bitmap,
+                                    blocks_per_group / 8);
+       }
+       return checksum;
+}
+
+/*
+ * BIG FAT NOTES:
+ *       Currently we do not verify the checksum of bitmaps.
+ */
+
+void ext4_balloc_set_bitmap_csum(struct ext4_sblock *sb,
+                                struct ext4_bgroup *bg,
+                                void *bitmap)
+{
+       int desc_size = ext4_sb_get_desc_size(sb);
+       uint32_t checksum = ext4_balloc_bitmap_csum(sb, bitmap);
+       uint16_t lo_checksum = to_le16(checksum & 0xFFFF),
+                hi_checksum = to_le16(checksum >> 16);
+       
+       /* See if we need to assign a 32bit checksum */
+       bg->block_bitmap_csum_lo = lo_checksum;
+       if (desc_size == EXT4_MAX_BLOCK_GROUP_DESCRIPTOR_SIZE)
+               bg->block_bitmap_csum_hi = hi_checksum;
+
+}
+
+int ext4_balloc_free_block(struct ext4_inode_ref *inode_ref, ext4_fsblk_t baddr)
 {
        struct ext4_fs *fs = inode_ref->fs;
        struct ext4_sblock *sb = &fs->sb;
@@ -92,7 +132,7 @@ int ext4_balloc_free_block(struct ext4_inode_ref *inode_ref, uint32_t baddr)
                return rc;
 
        /* Load block with bitmap */
-       uint32_t bitmap_block_addr =
+       ext4_fsblk_t bitmap_block_addr =
            ext4_bg_get_block_bitmap(bg_ref.block_group, sb);
 
        struct ext4_block bitmap_block;
@@ -105,6 +145,8 @@ int ext4_balloc_free_block(struct ext4_inode_ref *inode_ref, uint32_t baddr)
 
        /* Modify bitmap */
        ext4_bmap_bit_clr(bitmap_block.data, index_in_group);
+       ext4_balloc_set_bitmap_csum(sb, bg_ref.block_group,
+                                   bitmap_block.data);
        bitmap_block.dirty = true;
 
        /* Release block with bitmap */
@@ -140,7 +182,7 @@ int ext4_balloc_free_block(struct ext4_inode_ref *inode_ref, uint32_t baddr)
        return ext4_fs_put_block_group_ref(&bg_ref);
 }
 
-int ext4_balloc_free_blocks(struct ext4_inode_ref *inode_ref, uint32_t first,
+int ext4_balloc_free_blocks(struct ext4_inode_ref *inode_ref, ext4_fsblk_t first,
                            uint32_t count)
 {
        int rc = EOK;
@@ -175,7 +217,7 @@ int ext4_balloc_free_blocks(struct ext4_inode_ref *inode_ref, uint32_t first,
                    ext4_fs_baddr2_index_in_group(sb, first);
 
                /* Load block with bitmap */
-               uint32_t bitmap_block_addr =
+               ext4_fsblk_t bitmap_block_addr =
                    ext4_bg_get_block_bitmap(bg_ref.block_group, sb);
 
                struct ext4_block bitmap_block;
@@ -195,6 +237,8 @@ int ext4_balloc_free_blocks(struct ext4_inode_ref *inode_ref, uint32_t first,
                /* Modify bitmap */
                ext4_bmap_bits_free(bitmap_block.data, index_in_group_first,
                                    free_cnt);
+               ext4_balloc_set_bitmap_csum(sb, bg_ref.block_group,
+                                           bitmap_block.data);
                bitmap_block.dirty = true;
 
                count -= free_cnt;
@@ -242,96 +286,16 @@ int ext4_balloc_free_blocks(struct ext4_inode_ref *inode_ref, uint32_t first,
        return rc;
 }
 
-/**@brief Compute 'goal' for allocation algorithm.
- * @param inode_ref Reference to inode, to allocate block for
- * @param goal
- * @return error code
- */
-static int ext4_balloc_find_goal(struct ext4_inode_ref *inode_ref,
-                                uint32_t *goal)
+int ext4_balloc_alloc_block(struct ext4_inode_ref *inode_ref,
+                           ext4_fsblk_t goal,
+                           ext4_fsblk_t *fblock)
 {
-       struct ext4_sblock *sb = &inode_ref->fs->sb;
-       *goal = 0;
-
-       uint64_t inode_size = ext4_inode_get_size(sb, inode_ref->inode);
-       uint32_t block_size = ext4_sb_get_block_size(sb);
-       uint32_t inode_block_count = inode_size / block_size;
-
-       if (inode_size % block_size != 0)
-               inode_block_count++;
-
-       /* If inode has some blocks, get last block address + 1 */
-       if (inode_block_count > 0) {
-               int rc = ext4_fs_get_inode_data_block_index(
-                   inode_ref, inode_block_count - 1, goal);
-               if (rc != EOK)
-                       return rc;
-
-               if (*goal != 0) {
-                       (*goal)++;
-                       return rc;
-               }
-
-               /* If goal == 0, sparse file -> continue */
-       }
-
-       /* Identify block group of inode */
-
-       uint32_t inodes_per_group = ext4_get32(sb, inodes_per_group);
-       uint32_t block_group = (inode_ref->index - 1) / inodes_per_group;
-       block_size = ext4_sb_get_block_size(sb);
-
-       /* Load block group reference */
-       struct ext4_block_group_ref bg_ref;
-       int rc =
-           ext4_fs_get_block_group_ref(inode_ref->fs, block_group, &bg_ref);
-       if (rc != EOK)
-               return rc;
-
-       /* Compute indexes */
-       uint32_t block_group_count = ext4_block_group_cnt(sb);
-       uint32_t inode_table_first_block =
-           ext4_bg_get_inode_table_first_block(bg_ref.block_group, sb);
-       uint16_t inode_table_item_size = ext4_get16(sb, inode_size);
-       uint32_t inode_table_bytes;
-
-       /* Check for last block group */
-       if (block_group < block_group_count - 1) {
-               inode_table_bytes = inodes_per_group * inode_table_item_size;
-       } else {
-               /* Last block group could be smaller */
-               uint32_t inodes_count_total = ext4_get32(sb, inodes_count);
-
-               inode_table_bytes =
-                   (inodes_count_total -
-                    ((block_group_count - 1) * inodes_per_group)) *
-                   inode_table_item_size;
-       }
-
-       uint32_t inode_table_blocks = inode_table_bytes / block_size;
-
-       if (inode_table_bytes % block_size)
-               inode_table_blocks++;
-
-       *goal = inode_table_first_block + inode_table_blocks;
-
-       return ext4_fs_put_block_group_ref(&bg_ref);
-}
-
-int ext4_balloc_alloc_block(struct ext4_inode_ref *inode_ref, uint32_t *fblock)
-{
-       uint32_t allocated_block = 0;
-       uint32_t bitmap_block_addr;
+       ext4_fsblk_t allocated_block = 0;
+       ext4_fsblk_t bitmap_block_addr;
        uint32_t rel_block_idx = 0;
-       uint32_t free_blocks;
-       uint32_t goal;
+       uint64_t free_blocks;
        struct ext4_block bitmap_block;
-
-       int rc = ext4_balloc_find_goal(inode_ref, &goal);
-       if (rc != EOK) {
-               /* no goal found => partition is full */
-               return rc;
-       }
+       int rc;
 
        struct ext4_sblock *sb = &inode_ref->fs->sb;
 
@@ -352,7 +316,7 @@ int ext4_balloc_alloc_block(struct ext4_inode_ref *inode_ref, uint32_t *fblock)
        }
 
        /* Compute indexes */
-       uint32_t first_in_group = ext4_balloc_get_block_of_bgid(sb, bg_ref.index);
+       ext4_fsblk_t first_in_group = ext4_balloc_get_block_of_bgid(sb, bg_ref.index);
 
        uint32_t first_in_group_index =
            ext4_fs_baddr2_index_in_group(sb, first_in_group);
@@ -373,6 +337,8 @@ int ext4_balloc_alloc_block(struct ext4_inode_ref *inode_ref, uint32_t *fblock)
        /* Check if goal is free */
        if (ext4_bmap_is_bit_clr(bitmap_block.data, index_in_group)) {
                ext4_bmap_bit_set(bitmap_block.data, index_in_group);
+               ext4_balloc_set_bitmap_csum(sb, bg_ref.block_group,
+                                           bitmap_block.data);
                bitmap_block.dirty = true;
                rc = ext4_block_set(inode_ref->fs->bdev, &bitmap_block);
                if (rc != EOK) {
@@ -398,6 +364,8 @@ int ext4_balloc_alloc_block(struct ext4_inode_ref *inode_ref, uint32_t *fblock)
                if (ext4_bmap_is_bit_clr(bitmap_block.data, tmp_idx)) {
                        ext4_bmap_bit_set(bitmap_block.data, tmp_idx);
 
+                       ext4_balloc_set_bitmap_csum(sb, bg_ref.block_group,
+                                                   bitmap_block.data);
                        bitmap_block.dirty = true;
                        rc = ext4_block_set(inode_ref->fs->bdev, &bitmap_block);
                        if (rc != EOK)
@@ -415,6 +383,8 @@ int ext4_balloc_alloc_block(struct ext4_inode_ref *inode_ref, uint32_t *fblock)
                                    blocks_in_group, &rel_block_idx);
        if (rc == EOK) {
                ext4_bmap_bit_set(bitmap_block.data, rel_block_idx);
+               ext4_balloc_set_bitmap_csum(sb, bg_ref.block_group,
+                                           bitmap_block.data);
                bitmap_block.dirty = true;
                rc = ext4_block_set(inode_ref->fs->bdev, &bitmap_block);
                if (rc != EOK)
@@ -488,6 +458,8 @@ goal_failed:
 
                        ext4_bmap_bit_set(bitmap_block.data, rel_block_idx);
 
+                       ext4_balloc_set_bitmap_csum(sb, bg_ref.block_group,
+                                                   bitmap_block.data);
                        bitmap_block.dirty = true;
                        rc = ext4_block_set(inode_ref->fs->bdev, &bitmap_block);
                        if (rc != EOK) {
@@ -538,7 +510,7 @@ success:
        inode_ref->dirty = true;
 
        /* Update block group free blocks count */
-       uint32_t bg_free_blocks =
+       uint64_t bg_free_blocks =
            ext4_bg_get_free_blocks_count(bg_ref.block_group, sb);
        bg_free_blocks--;
        ext4_bg_set_free_blocks_count(bg_ref.block_group, sb, bg_free_blocks);
@@ -552,7 +524,7 @@ success:
 }
 
 int ext4_balloc_try_alloc_block(struct ext4_inode_ref *inode_ref,
-                               uint32_t baddr, bool *free)
+                               ext4_fsblk_t baddr, bool *free)
 {
        int rc;
 
@@ -570,7 +542,7 @@ int ext4_balloc_try_alloc_block(struct ext4_inode_ref *inode_ref,
                return rc;
 
        /* Load block with bitmap */
-       uint32_t bitmap_block_addr =
+       ext4_fsblk_t bitmap_block_addr =
            ext4_bg_get_block_bitmap(bg_ref.block_group, sb);
 
        struct ext4_block bitmap_block;
@@ -587,6 +559,8 @@ int ext4_balloc_try_alloc_block(struct ext4_inode_ref *inode_ref,
        /* Allocate block if possible */
        if (*free) {
                ext4_bmap_bit_set(bitmap_block.data, index_in_group);
+               ext4_balloc_set_bitmap_csum(sb, bg_ref.block_group,
+                                           bitmap_block.data);
                bitmap_block.dirty = true;
        }