1 files changed, 17 insertions, 12 deletions

diff --git a/drivers/ide/pci/cs5520.c b/drivers/ide/pci/cs5520.c
index d1a91bcb5b29..78058ca2ce76 100644
--- a/drivers/ide/pci/cs5520.c
+++ b/drivers/ide/pci/cs5520.c
@@ -71,7 +71,6 @@ static void cs5520_set_pio_mode(ide_drive_t *drive, const u8 pio)
         ide_hwif_t *hwif = HWIF(drive);
         struct pci_dev *pdev = hwif->pci_dev;
         int controller = drive->dn > 1 ? 1 : 0;
-        u8 reg;
 
         /* FIXME: if DMA = 1 do we need to set the DMA bit here ? */
 
@@ -91,11 +90,6 @@ static void cs5520_set_pio_mode(ide_drive_t *drive, const u8 pio)
         pci_write_config_byte(pdev, 0x66 + 4*controller + (drive->dn&1),
                 (cs5520_pio_clocks[pio].recovery << 4) |
                 (cs5520_pio_clocks[pio].assert));
-                
-        /* Set the DMA enable/disable flag */
-        reg = inb(hwif->dma_base + 0x02 + 8*controller);
-        reg |= 1<<((drive->dn&1)+5);
-        outb(reg, hwif->dma_base + 0x02 + 8*controller);
 }
 
 static void cs5520_set_dma_mode(ide_drive_t *drive, const u8 speed)
@@ -109,13 +103,23 @@ static void cs5520_set_dma_mode(ide_drive_t *drive, const u8 speed)
  *      We wrap the DMA activate to set the vdma flag. This is needed
  *      so that the IDE DMA layer issues PIO not DMA commands over the
  *      DMA channel
+ *
+ *      ATAPI is harder so disable it for now using IDE_HFLAG_NO_ATAPI_DMA
  */
- 
-static int cs5520_dma_on(ide_drive_t *drive)
+
+static void cs5520_dma_host_on(ide_drive_t *drive)
 {
-        /* ATAPI is harder so leave it for now */
-        drive->vdma = 1;
-        return 0;
+        if (drive->using_dma)
+                drive->vdma = 1;
+
+        ide_dma_host_on(drive);
+}
+
+static void cs5520_dma_host_off(ide_drive_t *drive)
+{
+        drive->vdma = 0;
+
+        ide_dma_host_off(drive);
 }
 
 static void __devinit init_hwif_cs5520(ide_hwif_t *hwif)
@@ -126,7 +130,8 @@ static void __devinit init_hwif_cs5520(ide_hwif_t *hwif)
         if (hwif->dma_base == 0)
                 return;
 
-        hwif->ide_dma_on = &cs5520_dma_on;
+        hwif->dma_host_on  = &cs5520_dma_host_on;
+        hwif->dma_host_off = &cs5520_dma_host_off;
 }
 
 #define DECLARE_CS_DEV(name_str)                                \