udf: convert macros related to bitmaps to functions

convert UDF_SB_ALLOC_BITMAP macro to udf_sb_alloc_bitmap function
convert UDF_SB_FREE_BITMAP macro to udf_sb_free_bitmap function

Signed-off-by: Marcin Slusarz <marcin.slusarz@gmail.com>
Cc: Ben Fennema <bfennema@falcon.csc.calpoly.edu>
Acked-by: Jan Kara <jack@suse.cz>
Cc: Christoph Hellwig <hch@infradead.org>
Signed-off-by: Andrew Morton <akpm@linux-foundation.org>
Signed-off-by: Linus Torvalds <torvalds@linux-foundation.org>

1 files changed, 55 insertions, 6 deletions

diff --git a/fs/udf/super.c b/fs/udf/super.c
index 913ece8eec61..86aa2238bc7b 100644
--- a/fs/udf/super.c
+++ b/fs/udf/super.c
@@ -937,6 +937,39 @@ static void udf_load_fileset(struct super_block *sb, struct buffer_head *bh,
                   root->logicalBlockNum, root->partitionReferenceNum);
 }
 
+static struct udf_bitmap *udf_sb_alloc_bitmap(struct super_block *sb, u32 index)
+{
+        struct udf_part_map *map = &UDF_SB(sb)->s_partmaps[index];
+        struct udf_bitmap *bitmap;
+        int nr_groups;
+        int size;
+
+        /* TODO: move calculating of nr_groups into helper function */
+        nr_groups = (map->s_partition_len +
+                        (sizeof(struct spaceBitmapDesc) << 3) +
+                        (sb->s_blocksize * 8) - 1) /
+                        (sb->s_blocksize * 8);
+        size = sizeof(struct udf_bitmap) +
+                (sizeof(struct buffer_head *) * nr_groups);
+
+        if (size <= PAGE_SIZE)
+                bitmap = kmalloc(size, GFP_KERNEL);
+        else
+                bitmap = vmalloc(size); /* TODO: get rid of vmalloc */
+
+        if (bitmap == NULL) {
+                udf_error(sb, __FUNCTION__,
+                          "Unable to allocate space for bitmap "
+                          "and %d buffer_head pointers", nr_groups);
+                return NULL;
+        }
+
+        memset(bitmap, 0x00, size);
+        bitmap->s_block_bitmap = (struct buffer_head **)(bitmap + 1);
+        bitmap->s_nr_groups = nr_groups;
+        return bitmap;
+}
+
 static int udf_load_partdesc(struct super_block *sb, struct buffer_head *bh)
 {
         struct partitionDesc *p;
@@ -987,7 +1020,7 @@ static int udf_load_partdesc(struct super_block *sb, struct buffer_head *bh)
                                                   i, map->s_uspace.s_table->i_ino);
                                 }
                                 if (phd->unallocSpaceBitmap.extLength) {
-                                        UDF_SB_ALLOC_BITMAP(sb, i, s_uspace);
+                                        map->s_uspace.s_bitmap = udf_sb_alloc_bitmap(sb, i);
                                         if (map->s_uspace.s_bitmap != NULL) {
                                                 map->s_uspace.s_bitmap->s_extLength =
                                                         le32_to_cpu(phd->unallocSpaceBitmap.extLength);
@@ -1017,7 +1050,7 @@ static int udf_load_partdesc(struct super_block *sb, struct buffer_head *bh)
                                                   i, map->s_fspace.s_table->i_ino);
                                 }
                                 if (phd->freedSpaceBitmap.extLength) {
-                                        UDF_SB_ALLOC_BITMAP(sb, i, s_fspace);
+                                        map->s_fspace.s_bitmap = udf_sb_alloc_bitmap(sb, i);
                                         if (map->s_fspace.s_bitmap != NULL) {
                                                 map->s_fspace.s_bitmap->s_extLength =
                                                         le32_to_cpu(phd->freedSpaceBitmap.extLength);
@@ -1504,6 +1537,22 @@ static void udf_close_lvid(struct super_block *sb)
         }
 }
 
+static void udf_sb_free_bitmap(struct udf_bitmap *bitmap)
+{
+        int i;
+        int nr_groups = bitmap->s_nr_groups;
+        int size = sizeof(struct udf_bitmap) + (sizeof(struct buffer_head *) * nr_groups);
+
+        for (i = 0; i < nr_groups; i++)
+                if (bitmap->s_block_bitmap[i])
+                        brelse(bitmap->s_block_bitmap[i]);
+
+        if (size <= PAGE_SIZE)
+                kfree(bitmap);
+        else
+                vfree(bitmap);
+}
+
 /*
  * udf_read_super
  *
@@ -1689,9 +1738,9 @@ error_out:
                 if (map->s_partition_flags & UDF_PART_FLAG_FREED_TABLE)
                         iput(map->s_fspace.s_table);
                 if (map->s_partition_flags & UDF_PART_FLAG_UNALLOC_BITMAP)
-                        UDF_SB_FREE_BITMAP(sb, sbi->s_partition, s_uspace);
+                        udf_sb_free_bitmap(map->s_uspace.s_bitmap);
                 if (map->s_partition_flags & UDF_PART_FLAG_FREED_BITMAP)
-                        UDF_SB_FREE_BITMAP(sb, sbi->s_partition, s_fspace);
+                        udf_sb_free_bitmap(map->s_fspace.s_bitmap);
                 if (map->s_partition_type == UDF_SPARABLE_MAP15)
                         for (i = 0; i < 4; i++)
                                 brelse(map->s_type_specific.s_sparing.s_spar_map[i]);
@@ -1767,9 +1816,9 @@ static void udf_put_super(struct super_block *sb)
                 if (map->s_partition_flags & UDF_PART_FLAG_FREED_TABLE)
                         iput(map->s_fspace.s_table);
                 if (map->s_partition_flags & UDF_PART_FLAG_UNALLOC_BITMAP)
-                        UDF_SB_FREE_BITMAP(sb, sbi->s_partition, s_uspace);
+                        udf_sb_free_bitmap(map->s_uspace.s_bitmap);
                 if (map->s_partition_flags & UDF_PART_FLAG_FREED_BITMAP)
-                        UDF_SB_FREE_BITMAP(sb, sbi->s_partition, s_fspace);
+                        udf_sb_free_bitmap(map->s_fspace.s_bitmap);
                 if (map->s_partition_type == UDF_SPARABLE_MAP15)
                         for (i = 0; i < 4; i++)
                                 brelse(map->s_type_specific.s_sparing.s_spar_map[i]);