- change UDF_SB_ALLOC_PARTMAPS macro to udf_sb_alloc_partition_maps inline 
function
- convert kmalloc + memset to kzalloc
- check if kzalloc failed (partially)

Signed-off-by: Marcin Slusarz <[EMAIL PROTECTED]>
CC: Ben Fennema <[EMAIL PROTECTED]>
CC: Jan Kara <[EMAIL PROTECTED]>
---
 fs/udf/super.c  |    6 ++++--
 fs/udf/udf_sb.h |   27 +++++++++++++++------------
 2 files changed, 19 insertions(+), 14 deletions(-)

diff --git a/fs/udf/super.c b/fs/udf/super.c
index dcab9e8..4aef612 100644
--- a/fs/udf/super.c
+++ b/fs/udf/super.c
@@ -1013,7 +1013,9 @@ static int udf_load_logicalvol(struct super_block *sb, 
struct buffer_head *bh,
 
        lvd = (struct logicalVolDesc *)bh->b_data;
 
-       UDF_SB_ALLOC_PARTMAPS(sb, le32_to_cpu(lvd->numPartitionMaps));
+       i = udf_sb_alloc_partition_maps(sb, le32_to_cpu(lvd->numPartitionMaps));
+       if (i != 0)
+               return i;
 
        for (i = 0, offset = 0;
             i < udf_sb_num_parts(sb) && offset < 
le32_to_cpu(lvd->mapTableLength);
@@ -1214,7 +1216,7 @@ static int udf_process_sequence(struct super_block *sb, 
long block,
                        if (i == VDS_POS_PRIMARY_VOL_DESC) {
                                udf_load_pvoldesc(sb, bh);
                        } else if (i == VDS_POS_LOGICAL_VOL_DESC) {
-                               udf_load_logicalvol(sb, bh, fileset);
+                               udf_load_logicalvol(sb, bh, fileset); /* TODO: 
check return value */
                        } else if (i == VDS_POS_PARTITION_DESC) {
                                struct buffer_head *bh2 = NULL;
                                if (udf_load_partdesc(sb, bh)) {
diff --git a/fs/udf/udf_sb.h b/fs/udf/udf_sb.h
index fa09d50..565c9ef 100644
--- a/fs/udf/udf_sb.h
+++ b/fs/udf/udf_sb.h
@@ -1,6 +1,8 @@
 #ifndef __LINUX_UDF_SB_H
 #define __LINUX_UDF_SB_H
 
+#include <linux/errno.h>
+
 /* Since UDF 2.01 is ISO 13346 based... */
 #define UDF_SUPER_MAGIC                        0x15013346
 
@@ -46,6 +48,19 @@ static inline struct udf_part_map *udf_sb_partmaps(struct 
super_block *sb)
        return udf_sb(sb)->s_partmaps;
 }
 
+static inline int __must_check udf_sb_alloc_partition_maps(struct super_block 
*sb, __u32 count)
+{
+       struct udf_sb_info *info = udf_sb(sb);
+       info->s_partmaps = kzalloc(sizeof(struct udf_part_map) * count, 
GFP_KERNEL);
+       if (info->s_partmaps != NULL)
+               info->s_partitions = count;
+       else {
+               info->s_partitions = 0;
+               udf_error(sb, __FUNCTION__, "Unable to allocate space for %d 
partition maps", count);
+       }
+       return info->s_partmaps != NULL ? 0 : -ENOMEM;
+}
+
 static inline void udf_sb_free(struct super_block *sb)
 {
        struct udf_sb_info *sb_info = udf_sb(sb);
@@ -123,18 +138,6 @@ static inline struct inode *udf_sb_vat_inode(struct 
super_block *sb)
        return udf_sb(sb)->s_vat_inode;
 }
 
-#define UDF_SB_ALLOC_PARTMAPS(X,Y)\
-{\
-       udf_sb(X)->s_partmaps = kmalloc(sizeof(struct udf_part_map) * Y, 
GFP_KERNEL);\
-       if (udf_sb_partmaps(X) != NULL) {\
-               udf_sb(X)->s_partitions = Y;\
-               memset(udf_sb_partmaps(X), 0x00, sizeof(struct udf_part_map) * 
Y);\
-       } else {\
-               udf_sb(X)->s_partitions = 0;\
-               udf_error(X, __FUNCTION__, "Unable to allocate space for %d 
partition maps", Y);\
-       }\
-}
-
 #define UDF_SB_ALLOC_BITMAP(X,Y,Z)\
 {\
        int nr_groups = ((UDF_SB_PARTLEN((X),(Y)) + (sizeof(struct 
spaceBitmapDesc) << 3) +\
-- 
1.5.3.4

--
To unsubscribe from this list: send the line "unsubscribe linux-kernel" in
the body of a message to [EMAIL PROTECTED]
More majordomo info at  http://vger.kernel.org/majordomo-info.html
Please read the FAQ at  http://www.tux.org/lkml/

Reply via email to