[PATCH] cleanup in drivers/mtd/ftl.c (245-ac16)

From: Rasmus Andersen (rasmus@jaquet.dk)
Date: Fri Jun 22 2001 - 15:29:31 EST

Hi.

The patch below adds one instance of vmalloc return code checking
and a number of error path resource release cleanups in build_maps.
It is against 245-ac16.

(The vmalloc non-check was reported by the Stanford team a
while back.)

--- linux-245-ac16-clean/drivers/mtd/ftl.c Sun May 27 22:15:23 2001
+++ linux-245-ac16/drivers/mtd/ftl.c Fri Jun 22 22:24:09 2001
@@ -313,6 +313,7 @@
     ssize_t retval;
     loff_t offset;
 
+ ret = -1;
     /* Set up erase unit maps */
     part->DataUnits = le16_to_cpu(part->header.NumEraseUnits) -
         part->header.NumTransferUnits;
@@ -324,7 +325,8 @@
     part->XferInfo =
         kmalloc(part->header.NumTransferUnits * sizeof(struct xfer_info_t),
                 GFP_KERNEL);
- if (!part->XferInfo) return -1;
+ if (!part->XferInfo)
+ goto err_free_EUInfo;
 
     xvalid = xtrans = 0;
     for (i = 0; i < le16_to_cpu(part->header.NumEraseUnits); i++) {
@@ -334,8 +336,9 @@
                               (unsigned char *)&header);
         
         if (ret)
- return ret;
+ goto err_free_XferInfo;
 
+ ret = -1;
         /* Is this a transfer partition? */
         hdr_ok = (strcmp(header.DataOrgTuple+3, "FTL100") == 0);
         if (hdr_ok && (le16_to_cpu(header.LogicalEUN) < part->DataUnits) &&
@@ -348,7 +351,7 @@
             if (xtrans == part->header.NumTransferUnits) {
                 printk(KERN_NOTICE "ftl_cs: format error: too many "
                        "transfer units!\n");
- return -1;
+ goto err_free_XferInfo;
             }
             if (hdr_ok && (le16_to_cpu(header.LogicalEUN) == 0xffff)) {
                 part->XferInfo[xtrans].state = XFER_PREPARED;
@@ -369,18 +372,21 @@
         (xvalid+xtrans != le16_to_cpu(header.NumEraseUnits))) {
         printk(KERN_NOTICE "ftl_cs: format error: erase units "
                "don't add up!\n");
- return -1;
+ goto err_free_XferInfo;
     }
     
     /* Set up virtual page map */
     blocks = le32_to_cpu(header.FormattedSize) >> header.BlockSize;
     part->VirtualBlockMap = vmalloc(blocks * sizeof(u_int32_t));
+ if (!part->VirtualBlockMap)
+ goto err_free_XferInfo;
     memset(part->VirtualBlockMap, 0xff, blocks * sizeof(u_int32_t));
     part->BlocksPerUnit = (1 << header.EraseUnitSize) >> header.BlockSize;
 
     part->bam_cache = kmalloc(part->BlocksPerUnit * sizeof(u_int32_t),
                               GFP_KERNEL);
- if (!part->bam_cache) return -1;
+ if (!part->bam_cache)
+ goto err_free_VirtualBlockMap;
 
     part->bam_index = 0xffff;
     part->FreeTotal = 0;
@@ -395,7 +401,7 @@
                               (unsigned char *)part->bam_cache);
         
         if (ret)
- return ret;
+ goto err_free_bam_cache;
 
         for (j = 0; j < part->BlocksPerUnit; j++) {
             if (BLOCK_FREE(le32_to_cpu(part->bam_cache[j]))) {
@@ -411,7 +417,17 @@
     }
     
     return 0;
-
+
+err_free_bam_cache:
+ kfree(part->bam_cache);
+err_free_VirtualBlockMap:
+ vfree(part->VirtualBlockMap);
+err_free_XferInfo:
+ kfree(part->XferInfo);
+err_free_EUInfo:
+ kfree(part->EUNInfo);
+ printk(KERN_ERR "ftl_cs: Out of memory.");
+ return ret;
 } /* build_maps */
 
 /*====================================================================== 

-- 
Regards,
        Rasmus(rasmus@jaquet.dk)

Open Source. Closed Minds. We are Slashdot. 
  --Anonymous Coward
-
To unsubscribe from this list: send the line "unsubscribe linux-kernel" in
the body of a message to majordomo@vger.kernel.org
More majordomo info at  http://vger.kernel.org/majordomo-info.html
Please read the FAQ at  http://www.tux.org/lkml/

This archive was generated by hypermail 2b29 : Sat Jun 23 2001 - 21:00:40 EST