1 files changed, 42 insertions, 44 deletions

diff --git a/drivers/firewire/fw-ohci.c b/drivers/firewire/fw-ohci.c
index 5833ce1b040b..1f5c70461b8b 100644
--- a/drivers/firewire/fw-ohci.c
+++ b/drivers/firewire/fw-ohci.c
@@ -1715,44 +1715,13 @@ static int software_reset(struct fw_ohci *ohci)
         return -EBUSY;
 }
 
-enum {
-        CLEANUP_SELF_ID,
-        CLEANUP_REGISTERS,
-        CLEANUP_IOMEM,
-        CLEANUP_DISABLE,
-        CLEANUP_PUT_CARD,
-};
-
-static int cleanup(struct fw_ohci *ohci, int stage, int code)
-{
-        struct pci_dev *dev = to_pci_dev(ohci->card.device);
-
-        switch (stage) {
-        case CLEANUP_SELF_ID:
-                dma_free_coherent(ohci->card.device, SELF_ID_BUF_SIZE,
-                                  ohci->self_id_cpu, ohci->self_id_bus);
-        case CLEANUP_REGISTERS:
-                kfree(ohci->it_context_list);
-                kfree(ohci->ir_context_list);
-                pci_iounmap(dev, ohci->registers);
-        case CLEANUP_IOMEM:
-                pci_release_region(dev, 0);
-        case CLEANUP_DISABLE:
-                pci_disable_device(dev);
-        case CLEANUP_PUT_CARD:
-                fw_card_put(&ohci->card);
-        }
-
-        return code;
-}
-
 static int __devinit
 pci_probe(struct pci_dev *dev, const struct pci_device_id *ent)
 {
         struct fw_ohci *ohci;
         u32 bus_options, max_receive, link_speed;
         u64 guid;
-        int error_code;
+        int err;
         size_t size;
 
         ohci = kzalloc(sizeof(*ohci), GFP_KERNEL);
@@ -1763,9 +1732,10 @@ pci_probe(struct pci_dev *dev, const struct pci_device_id *ent)
 
         fw_card_initialize(&ohci->card, &ohci_driver, &dev->dev);
 
-        if (pci_enable_device(dev)) {
+        err = pci_enable_device(dev);
+        if (err) {
                 fw_error("Failed to enable OHCI hardware.\n");
-                return cleanup(ohci, CLEANUP_PUT_CARD, -ENODEV);
+                goto fail_put_card;
         }
 
         pci_set_master(dev);
@@ -1777,20 +1747,23 @@ pci_probe(struct pci_dev *dev, const struct pci_device_id *ent)
         tasklet_init(&ohci->bus_reset_tasklet,
                      bus_reset_tasklet, (unsigned long)ohci);
 
-        if (pci_request_region(dev, 0, ohci_driver_name)) {
+        err = pci_request_region(dev, 0, ohci_driver_name);
+        if (err) {
                 fw_error("MMIO resource unavailable\n");
-                return cleanup(ohci, CLEANUP_DISABLE, -EBUSY);
+                goto fail_disable;
         }
 
         ohci->registers = pci_iomap(dev, 0, OHCI1394_REGISTER_SIZE);
         if (ohci->registers == NULL) {
                 fw_error("Failed to remap registers\n");
-                return cleanup(ohci, CLEANUP_IOMEM, -ENXIO);
+                err = -ENXIO;
+                goto fail_iomem;
         }
 
         if (software_reset(ohci)) {
                 fw_error("Failed to reset ohci card.\n");
-                return cleanup(ohci, CLEANUP_REGISTERS, -EBUSY);
+                err = -EBUSY;
+                goto fail_registers;
         }
 
         /*
@@ -1845,7 +1818,8 @@ pci_probe(struct pci_dev *dev, const struct pci_device_id *ent)
 
         if (ohci->it_context_list == NULL || ohci->ir_context_list == NULL) {
                 fw_error("Out of memory for it/ir contexts.\n");
-                return cleanup(ohci, CLEANUP_REGISTERS, -ENOMEM);
+                err = -ENOMEM;
+                goto fail_registers;
         }
 
         /* self-id dma buffer allocation */
@@ -1855,7 +1829,8 @@ pci_probe(struct pci_dev *dev, const struct pci_device_id *ent)
                                                GFP_KERNEL);
         if (ohci->self_id_cpu == NULL) {
                 fw_error("Out of memory for self ID buffer.\n");
-                return cleanup(ohci, CLEANUP_REGISTERS, -ENOMEM);
+                err = -ENOMEM;
+                goto fail_registers;
         }
 
         reg_write(ohci, OHCI1394_SelfIDBuffer, ohci->self_id_bus);
@@ -1876,15 +1851,31 @@ pci_probe(struct pci_dev *dev, const struct pci_device_id *ent)
         guid = ((u64) reg_read(ohci, OHCI1394_GUIDHi) << 32) |
                 reg_read(ohci, OHCI1394_GUIDLo);
 
-        error_code = fw_card_add(&ohci->card, max_receive, link_speed, guid);
-        if (error_code < 0)
-                return cleanup(ohci, CLEANUP_SELF_ID, error_code);
+        err = fw_card_add(&ohci->card, max_receive, link_speed, guid);
+        if (err < 0)
+                goto fail_self_id;
 
         ohci->version = reg_read(ohci, OHCI1394_Version) & 0x00ff00ff;
         fw_notify("Added fw-ohci device %s, OHCI version %x.%x\n",
                   dev->dev.bus_id, ohci->version >> 16, ohci->version & 0xff);
 
         return 0;
+
+ fail_self_id:
+        dma_free_coherent(ohci->card.device, SELF_ID_BUF_SIZE,
+                          ohci->self_id_cpu, ohci->self_id_bus);
+ fail_registers:
+        kfree(ohci->it_context_list);
+        kfree(ohci->ir_context_list);
+        pci_iounmap(dev, ohci->registers);
+ fail_iomem:
+        pci_release_region(dev, 0);
+ fail_disable:
+        pci_disable_device(dev);
+ fail_put_card:
+        fw_card_put(&ohci->card);
+
+        return err;
 }
 
 static void pci_remove(struct pci_dev *dev)
@@ -1903,7 +1894,14 @@ static void pci_remove(struct pci_dev *dev)
 
         software_reset(ohci);
         free_irq(dev->irq, ohci);
-        cleanup(ohci, CLEANUP_SELF_ID, 0);
+        dma_free_coherent(ohci->card.device, SELF_ID_BUF_SIZE,
+                          ohci->self_id_cpu, ohci->self_id_bus);
+        kfree(ohci->it_context_list);
+        kfree(ohci->ir_context_list);
+        pci_iounmap(dev, ohci->registers);
+        pci_release_region(dev, 0);
+        pci_disable_device(dev);
+        fw_card_put(&ohci->card);
 
         fw_notify("Removed fw-ohci device.\n");
 }