1 files changed, 20 insertions, 20 deletions

diff --git a/drivers/ata/libata-sff.c b/drivers/ata/libata-sff.c
index 08b3a407473e..a620e235ccc2 100644
--- a/drivers/ata/libata-sff.c
+++ b/drivers/ata/libata-sff.c
@@ -946,35 +946,21 @@ int ata_pci_init_one (struct pci_dev *pdev, struct ata_port_info **port_info,
 {
         struct ata_probe_ent *probe_ent = NULL;
         struct ata_port_info *port[2];
-        u8 tmp8, mask;
+        u8 mask;
         unsigned int legacy_mode = 0;
         int disable_dev_on_err = 1;
         int rc;
 
         DPRINTK("ENTER\n");
 
+        BUG_ON(n_ports < 1 || n_ports > 2);
+
         port[0] = port_info[0];
         if (n_ports > 1)
                 port[1] = port_info[1];
         else
                 port[1] = port[0];
 
-        if ((port[0]->flags & ATA_FLAG_NO_LEGACY) == 0
-            && (pdev->class >> 8) == PCI_CLASS_STORAGE_IDE) {
-                /* TODO: What if one channel is in native mode ... */
-                pci_read_config_byte(pdev, PCI_CLASS_PROG, &tmp8);
-                mask = (1 << 2) | (1 << 0);
-                if ((tmp8 & mask) != mask)
-                        legacy_mode = (1 << 3);
-        }
-
-        /* FIXME... */
-        if ((!legacy_mode) && (n_ports > 2)) {
-                printk(KERN_ERR "ata: BUG: native mode, n_ports > 2\n");
-                n_ports = 2;
-                /* For now */
-        }
-
         /* FIXME: Really for ATA it isn't safe because the device may be
            multi-purpose and we want to leave it alone if it was already
            enabled. Secondly for shared use as Arjan says we want refcounting
@@ -987,6 +973,16 @@ int ata_pci_init_one (struct pci_dev *pdev, struct ata_port_info **port_info,
         if (rc)
                 return rc;
 
+        if ((pdev->class >> 8) == PCI_CLASS_STORAGE_IDE) {
+                u8 tmp8;
+
+                /* TODO: What if one channel is in native mode ... */
+                pci_read_config_byte(pdev, PCI_CLASS_PROG, &tmp8);
+                mask = (1 << 2) | (1 << 0);
+                if ((tmp8 & mask) != mask)
+                        legacy_mode = (1 << 3);
+        }
+
         rc = pci_request_regions(pdev, DRV_NAME);
         if (rc) {
                 disable_dev_on_err = 0;
@@ -1039,7 +1035,7 @@ int ata_pci_init_one (struct pci_dev *pdev, struct ata_port_info **port_info,
                 goto err_out_regions;
         }
 
-        /* FIXME: If we get no DMA mask we should fall back to PIO */
+        /* TODO: If we get no DMA mask we should fall back to PIO */
         rc = pci_set_dma_mask(pdev, ATA_DMA_MASK);
         if (rc)
                 goto err_out_regions;
@@ -1062,13 +1058,17 @@ int ata_pci_init_one (struct pci_dev *pdev, struct ata_port_info **port_info,
 
         pci_set_master(pdev);
 
-        /* FIXME: check ata_device_add return */
-        ata_device_add(probe_ent);
+        if (!ata_device_add(probe_ent)) {
+                rc = -ENODEV;
+                goto err_out_ent;
+        }
 
         kfree(probe_ent);
 
         return 0;
 
+err_out_ent:
+        kfree(probe_ent);
 err_out_regions:
         if (legacy_mode & ATA_PORT_PRIMARY)
                 release_region(ATA_PRIMARY_CMD, 8);