[UNIATA] Fix for uninitialized Lun pointer
[reactos.git] / drivers / storage / ide / uniata / id_init.cpp
index 6268c96..b9beb1b 100644 (file)
@@ -1,6 +1,6 @@
 /*++
 
-Copyright (c) 2004-2012 Alexandr A. Telyatnikov (Alter)
+Copyright (c) 2004-2018 Alexandr A. Telyatnikov (Alter)
 
 Module Name:
     id_init.cpp
@@ -38,12 +38,16 @@ Revision History:
         VIA, nVidia
     added support for non-standard layout of registers
     added SATA support
+    added AHCI support
+
+Licence:
+    GPLv2
 
 --*/
 
 #include "stdafx.h"
 
-static BUSMASTER_CONTROLLER_INFORMATION const AtiSouthAdapters[] = {
+static BUSMASTER_CONTROLLER_INFORMATION_BASE const AtiSouthAdapters[] = {
     PCI_DEV_HW_SPEC_BM( 4385, 1002, 0x00, ATA_MODE_NOT_SPEC, "ATI South", 0 ),
     PCI_DEV_HW_SPEC_BM( ffff, ffff, 0xff, BMLIST_TERMINATOR, NULL      , BMLIST_TERMINATOR )
     };
@@ -66,9 +70,12 @@ UniataChipDetectChannels(
     //ULONG RevID    =  deviceExtension->RevID;
     ULONG ChipType = deviceExtension->HwFlags & CHIPTYPE_MASK;
     ULONG ChipFlags= deviceExtension->HwFlags & CHIPFLAG_MASK;
+    ULONG i,n;
 
     KdPrint2((PRINT_PREFIX "UniataChipDetectChannels:\n" ));
 
+    deviceExtension->AHCI_PI_mask = 0;
+
     if(ChipFlags & (UNIATA_SATA | UNIATA_AHCI)) {
         if(!deviceExtension->NumberChannels) {
             KdPrint2((PRINT_PREFIX "uninitialized SATA/AHCI port number -> 1\n"));
@@ -88,6 +95,21 @@ UniataChipDetectChannels(
         KdPrint2((PRINT_PREFIX "MasterDev -> 1 chan\n"));
         deviceExtension->NumberChannels = 1;
     }
+    for(n=0; n<deviceExtension->NumberChannels; n++) {
+        if(AtapiRegCheckDevValue(deviceExtension, n, DEVNUM_NOT_SPECIFIED, L"Exclude", 0)) {
+            KdPrint2((PRINT_PREFIX "Channel %d excluded\n", n));
+            deviceExtension->AHCI_PI_mask &= ~((ULONG)1 << n);
+        } else {
+            deviceExtension->AHCI_PI_mask |= ((ULONG)1 << n);
+        }
+    }
+    KdPrint2((PRINT_PREFIX "PortMask %#x\n", deviceExtension->AHCI_PI_mask));
+    deviceExtension->AHCI_PI_mask = 
+        AtapiRegCheckDevValue(deviceExtension, CHAN_NOT_SPECIFIED, DEVNUM_NOT_SPECIFIED, L"PortMask", (ULONG)0xffffffff >> (32-deviceExtension->NumberChannels) );
+    KdPrint2((PRINT_PREFIX "Force PortMask %#x\n", deviceExtension->AHCI_PI_mask));
+
+    for(i=deviceExtension->AHCI_PI_mask, n=0; i; n++, i=i>>1);
+    KdPrint2((PRINT_PREFIX "mask -> %d chans\n", n));
 
     switch(VendorID) {
     case ATA_ACER_LABS_ID:
@@ -147,7 +169,7 @@ UniataChipDetectChannels(
         case ATA_ATI_IXP700: {
             UCHAR satacfg = 0;
             PCI_SLOT_NUMBER slotData;
-            ULONG i, slotNumber;
+            ULONG j, slotNumber;
                  
             KdPrint2((PRINT_PREFIX "  IXP700\n"));
             /*
@@ -155,8 +177,8 @@ UniataChipDetectChannels(
              * emulated with two SATA ports and appears on this device.
              * This mode can only be detected via SMB controller.
              */
-            i = AtapiFindListedDev((BUSMASTER_CONTROLLER_INFORMATION*)&AtiSouthAdapters[0], -1, HwDeviceExtension, SystemIoBusNumber, PCISLOTNUM_NOT_SPECIFIED, &slotData);
-            if(i != BMLIST_TERMINATOR) {
+            j = AtapiFindListedDev((BUSMASTER_CONTROLLER_INFORMATION_BASE*)&AtiSouthAdapters[0], -1, HwDeviceExtension, SystemIoBusNumber, PCISLOTNUM_NOT_SPECIFIED, &slotData);
+            if(j != BMLIST_TERMINATOR) {
                 slotNumber = slotData.u.AsULONG;
 
                 GetPciConfig1(0xad, satacfg);
@@ -257,20 +279,57 @@ UniataChipDetectChannels(
         }
         break;
 #endif // this code is removed from newer FreeBSD
+#if 0
     case ATA_JMICRON_ID:
         /* New JMicron PATA controllers */
         if(deviceExtension->DevID == ATA_JMB361 ||
            deviceExtension->DevID == ATA_JMB363 ||
+           deviceExtension->DevID == ATA_JMB365 ||
+           deviceExtension->DevID == ATA_JMB366 ||
            deviceExtension->DevID == ATA_JMB368) { 
-            if(BMList[deviceExtension->DevIndex].channel) {
-                KdPrint2((PRINT_PREFIX "New JMicron has no 2nd chan\n"));
+
+            ULONG tmp32, port_mask;
+
+            port_mask = BMList[deviceExtension->DevIndex].channel;
+
+            GetPciConfig4(0x40, tmp32);
+
+            deviceExtension->NumberChannels = 2;
+            //KdPrint2((PRINT_PREFIX "New JMicron PATA 1 chan\n"));
+        }
+        break;
+#endif // this code is unnecessary since port mapping is implemented
+    case ATA_CYRIX_ID:
+        if(ChipType == CYRIX_OLD) {
+            UCHAR tmp8;
+            ULONG slotNumber;
+            slotNumber = deviceExtension->slotNumber;
+            KdPrint2((PRINT_PREFIX "Cyrix slot %#x\n", slotNumber));
+            GetPciConfig1(0x60, tmp8);
+            if(tmp8 & (1 << BMList[deviceExtension->DevIndex].channel)) {
+                KdPrint2((PRINT_PREFIX "Old Cyrix chan %d ok\n", BMList[deviceExtension->DevIndex].channel));
+            } else {
+                KdPrint2((PRINT_PREFIX "Old Cyrix no chan %d\n", BMList[deviceExtension->DevIndex].channel));
                 return FALSE;
             }
-            deviceExtension->NumberChannels = 1;
-            KdPrint2((PRINT_PREFIX "New JMicron PATA 1 chan\n"));
         }
         break;
     } // end switch(VendorID)
+
+    i = AtapiRegCheckDevValue(deviceExtension, CHAN_NOT_SPECIFIED, DEVNUM_NOT_SPECIFIED, L"NumberChannels", n);
+    if(!i) {
+        i = n;
+    }
+    KdPrint2((PRINT_PREFIX "reg -> %d chans\n", n));
+
+    deviceExtension->NumberChannels = min(i, deviceExtension->NumberChannels);
+    if(!deviceExtension->NumberChannels) {
+        KdPrint2((PRINT_PREFIX "all channels blocked\n", n));
+        return FALSE;
+    }
+    deviceExtension->AHCI_PI_mask &= (ULONG)0xffffffff >> (32-deviceExtension->NumberChannels);
+    KdPrint2((PRINT_PREFIX "Final PortMask %#x\n", deviceExtension->AHCI_PI_mask));
+
     return TRUE;
 
 } // end UniataChipDetectChannels()
@@ -292,23 +351,23 @@ UniataChipDetect(
     ULONG DeviceID = (deviceExtension->DevID >> 16) & 0xffff;
     ULONG RevID    =  deviceExtension->RevID;
     ULONG i, c;
-    BUSMASTER_CONTROLLER_INFORMATION* DevTypeInfo;
+    BUSMASTER_CONTROLLER_INFORMATION_BASE* DevTypeInfo;
     PHW_CHANNEL chan;
     ULONG ChipType;
     ULONG ChipFlags;
     ULONG tmp32;
     UCHAR tmp8;
-    ULONG BaseMemAddress;
-    ULONG BaseIoAddress1;
-    ULONG BaseIoAddress2;
-    ULONG BaseIoAddressBM;
+    ULONG_PTR BaseMemAddress;
+    ULONG_PTR BaseIoAddress1;
+    ULONG_PTR BaseIoAddress2;
+    ULONG_PTR BaseIoAddressBM;
     BOOLEAN MemIo = FALSE;
     BOOLEAN IsPata = FALSE;
 
     KdPrint2((PRINT_PREFIX "UniataChipDetect:\n" ));
     KdPrint2((PRINT_PREFIX "HwFlags: %#x\n", deviceExtension->HwFlags));
 
-    i = Ata_is_dev_listed((PBUSMASTER_CONTROLLER_INFORMATION)&BusMasterAdapters[0], VendorID, 0xffff, 0, NUM_BUSMASTER_ADAPTERS);
+    i = Ata_is_dev_listed((PBUSMASTER_CONTROLLER_INFORMATION_BASE)&BusMasterAdapters[0], VendorID, 0xffff, 0, NUM_BUSMASTER_ADAPTERS);
 
     c = AtapiRegCheckDevValue(deviceExtension, CHAN_NOT_SPECIFIED, DEVNUM_NOT_SPECIFIED, L"ForceSimplex", 0);
     if(c) {
@@ -326,7 +385,7 @@ UniataChipDetect(
 
     KdPrint2((PRINT_PREFIX "i: %#x\n", i));
     if(i != BMLIST_TERMINATOR) {
-        DevTypeInfo = (PBUSMASTER_CONTROLLER_INFORMATION)&BusMasterAdapters[i];
+        DevTypeInfo = (PBUSMASTER_CONTROLLER_INFORMATION_BASE)&BusMasterAdapters[i];
     } else {
 unknown_dev:
         if(Ata_is_ahci_dev(pciData)) {
@@ -349,12 +408,10 @@ unknown_dev:
         if(!UniataAllocateLunExt(deviceExtension, UNIATA_ALLOCATE_NEW_LUNS)) {
             return STATUS_UNSUCCESSFUL;
         }
-        // DEBUG, we shall return success when AHCI is completly supported
-        //return STATUS_NOT_FOUND;
         return STATUS_SUCCESS;
     }
 
-    static BUSMASTER_CONTROLLER_INFORMATION const SiSAdapters[] = {
+    static BUSMASTER_CONTROLLER_INFORMATION_BASE const SiSAdapters[] = {
         PCI_DEV_HW_SPEC_BM( 1183, 1039, 0x00, ATA_SA150, "SiS 1183 IDE" , SIS133NEW),
         PCI_DEV_HW_SPEC_BM( 1182, 1039, 0x00, ATA_SA150, "SiS 1182" , SISSATA   | UNIATA_SATA),
         PCI_DEV_HW_SPEC_BM( 0183, 1039, 0x00, ATA_SA150, "SiS 183 RAID"  , SISSATA   | UNIATA_SATA),
@@ -395,7 +452,7 @@ unknown_dev:
         PCI_DEV_HW_SPEC_BM( ffff, ffff, 0xff, BMLIST_TERMINATOR       , NULL       , BMLIST_TERMINATOR )
         };
 
-    static BUSMASTER_CONTROLLER_INFORMATION const ViaAdapters[] = {
+    static BUSMASTER_CONTROLLER_INFORMATION_BASE const ViaAdapters[] = {
         PCI_DEV_HW_SPEC_BM( 0586, 1106, 0x41, ATA_UDMA2, "VIA 82C586B", VIA33  | 0x00   ),
         PCI_DEV_HW_SPEC_BM( 0586, 1106, 0x40, ATA_UDMA2, "VIA 82C586B", VIA33  | VIAPRQ ),
         PCI_DEV_HW_SPEC_BM( 0586, 1106, 0x02, ATA_UDMA2, "VIA 82C586B", VIA33  | 0x00   ),
@@ -417,14 +474,14 @@ unknown_dev:
         PCI_DEV_HW_SPEC_BM( 5372, 1106, 0x00, ATA_UDMA6, "VIA 8237"   , VIA133 | 0x00 ),
         PCI_DEV_HW_SPEC_BM( 7372, 1106, 0x00, ATA_UDMA6, "VIA 8237"   , VIA133 | 0x00 ),
         PCI_DEV_HW_SPEC_BM( 3349, 1106, 0x00, ATA_UDMA6, "VIA 8251"   , VIA133 | 0x00 ),
-        PCI_DEV_HW_SPEC_BM( 8324, 1106, 0x00, ATA_SA150, "VIA CX700"  , VIA133 | VIASATA),
-        PCI_DEV_HW_SPEC_BM( 8353, 1106, 0x00, ATA_SA150, "VIA VX800"  , VIA133 | VIASATA),
+        PCI_DEV_HW_SPEC_BM( 8324, 1106, 0x00, ATA_SA150, "VIA CX700"  , VIANEW | VIASATA),
+        PCI_DEV_HW_SPEC_BM( 8353, 1106, 0x00, ATA_SA150, "VIA VX800"  , VIANEW | VIASATA),
         PCI_DEV_HW_SPEC_BM( 8409, 1106, 0x00, ATA_UDMA6, "VIA VX855"  , VIA133 | 0x00 ),
-        PCI_DEV_HW_SPEC_BM( 8410, 1106, 0x00, ATA_SA300, "VIA VX900"  , VIA133 | VIASATA),
+        PCI_DEV_HW_SPEC_BM( 8410, 1106, 0x00, ATA_SA300, "VIA VX900"  , VIANEW | VIASATA),
         PCI_DEV_HW_SPEC_BM( ffff, ffff, 0xff, BMLIST_TERMINATOR       , NULL         , BMLIST_TERMINATOR )
         };
 
-    static BUSMASTER_CONTROLLER_INFORMATION const ViaSouthAdapters[] = {
+    static BUSMASTER_CONTROLLER_INFORMATION_BASE const ViaSouthAdapters[] = {
         PCI_DEV_HW_SPEC_BM( 3112, 1106, 0x00, ATA_MODE_NOT_SPEC, "VIA 8361", VIASOUTH ),
         PCI_DEV_HW_SPEC_BM( 0305, 1106, 0x00, ATA_MODE_NOT_SPEC, "VIA 8363", VIASOUTH ),
         PCI_DEV_HW_SPEC_BM( 0391, 1106, 0x00, ATA_MODE_NOT_SPEC, "VIA 8371", VIASOUTH ),
@@ -442,7 +499,7 @@ unknown_dev:
            Then perform bus scan to find SIS bridge and decide what to do with controller
          */
         KdPrint2((PRINT_PREFIX "ATA_SIS_ID\n"));
-        DevTypeInfo = (BUSMASTER_CONTROLLER_INFORMATION*)&SiSAdapters[0];
+        DevTypeInfo = (BUSMASTER_CONTROLLER_INFORMATION_BASE*)&SiSAdapters[0];
         i = AtapiFindListedDev(DevTypeInfo, -1, HwDeviceExtension, SystemIoBusNumber, PCISLOTNUM_NOT_SPECIFIED, NULL);
         if(i != BMLIST_TERMINATOR) {
             deviceExtension->FullDevName = SiSAdapters[i].FullDevName;
@@ -463,7 +520,7 @@ unknown_dev:
         KdPrint2((PRINT_PREFIX "Via-old-style %x\n", deviceExtension->DevID));
         // Traditionally, chips have same DeviceId, we can distinguish between them
         // only by ISA Bridge DeviceId
-        DevTypeInfo = (BUSMASTER_CONTROLLER_INFORMATION*)&ViaSouthAdapters[0];
+        DevTypeInfo = (BUSMASTER_CONTROLLER_INFORMATION_BASE*)&ViaSouthAdapters[0];
         i = AtapiFindListedDev(DevTypeInfo, -1, HwDeviceExtension, SystemIoBusNumber,
                                PCISLOTNUM_NOT_SPECIFIED/*slotNumber*/, NULL);
 /*        if(i == BMLIST_TERMINATOR) {
@@ -473,7 +530,7 @@ unknown_dev:
             KdPrint2((PRINT_PREFIX "VIASOUTH\n"));
             deviceExtension->HwFlags |= VIASOUTH;
         }
-        DevTypeInfo = (BUSMASTER_CONTROLLER_INFORMATION*)&ViaAdapters[0];
+        DevTypeInfo = (BUSMASTER_CONTROLLER_INFORMATION_BASE*)&ViaAdapters[0];
         i = AtapiFindListedDev(DevTypeInfo, -1, HwDeviceExtension, SystemIoBusNumber,
                                PCISLOTNUM_NOT_SPECIFIED/*slotNumber*/, NULL);
         if(i != BMLIST_TERMINATOR) {
@@ -522,7 +579,7 @@ unknown_dev:
                 ScsiPortFreeDeviceBase(HwDeviceExtension,
                                        deviceExtension->BaseIoAddressBM_0);
 
-            deviceExtension->BaseIoAddressBM_0 = 0;
+            UniataInitIoResEx(&deviceExtension->BaseIoAddressBM_0, 0, FALSE, FALSE);
             deviceExtension->BusMaster = DMA_MODE_NONE;
             deviceExtension->MaxTransferMode = ATA_PIO4;
             break;
@@ -573,6 +630,7 @@ for_ugly_chips:
     }
 
     if(deviceExtension->MaxTransferMode >= ATA_SA150) {
+        KdPrint2((PRINT_PREFIX "setting UNIATA_SATA flag\n"));
         deviceExtension->HwFlags |= UNIATA_SATA;
     }
 
@@ -599,15 +657,24 @@ for_ugly_chips:
                 deviceExtension->HwFlags &= ~UNIATA_AHCI;
             }
             break;
+        default:
+            if(!ScsiPortConvertPhysicalAddressToUlong((*ConfigInfo->AccessRanges)[5].RangeStart)) {
+                KdPrint2((PRINT_PREFIX "No BAR5, try BM\n"));
+                ChipFlags &= ~UNIATA_AHCI;
+                deviceExtension->HwFlags &= ~UNIATA_AHCI;
+            }
+            break;
         }
     }
 
     if(ChipFlags & UNIATA_AHCI) {
+
         deviceExtension->NumberChannels = 0;
         if(!UniataAhciDetect(HwDeviceExtension, pciData, ConfigInfo)) {
             KdPrint2((PRINT_PREFIX "  AHCI detect failed\n"));
             return STATUS_UNSUCCESSFUL;
         }
+
     } else
     if(!UniataChipDetectChannels(HwDeviceExtension, pciData, DeviceNumber, ConfigInfo)) {
         return STATUS_UNSUCCESSFUL;
@@ -633,13 +700,13 @@ for_ugly_chips:
                 chan = &deviceExtension->chan[c];
 
                 for (i=0; i<=IDX_IO1_SZ; i++) {
-                    chan->RegTranslation[IDX_IO1+i].Addr           = BaseIoAddress1  + i + (unit10 ? 8 : 0);
+                    UniataInitIoRes(chan, IDX_IO1+i, BaseIoAddress1 + i + (unit10 ? 8 : 0), FALSE, FALSE);
                 }
-                chan->RegTranslation[IDX_IO2_AltStatus].Addr       = BaseIoAddress2  + 2 + (unit10 ? 4 : 0);
+                UniataInitIoRes(chan, IDX_IO2_AltStatus, BaseIoAddress2  + 2 + (unit10 ? 4 : 0), FALSE, FALSE);
                 UniataInitSyncBaseIO(chan);
 
                 for (i=0; i<=IDX_BM_IO_SZ; i++) {
-                    chan->RegTranslation[IDX_BM_IO+i].Addr         = BaseIoAddressBM + i + (c * sizeof(IDE_BUSMASTER_REGISTERS));
+                    UniataInitIoRes(chan, IDX_BM_IO+i, BaseIoAddressBM + i + (c * sizeof(IDE_BUSMASTER_REGISTERS)), FALSE, FALSE);
                 }
 
                 // SATA not supported yet
@@ -665,17 +732,13 @@ for_ugly_chips:
                 KdPrint2((PRINT_PREFIX "MemIo\n"));
                 MemIo = TRUE;
             }
-            deviceExtension->BaseIoAddressSATA_0.Addr  = BaseMemAddress;
-            deviceExtension->BaseIoAddressSATA_0.MemIo = MemIo;
+            UniataInitIoResEx(&deviceExtension->BaseIoAddressSATA_0, BaseMemAddress, MemIo, FALSE);
             for(c=0; c<deviceExtension->NumberChannels; c++) {
                 chan = &deviceExtension->chan[c];
 
-                chan->RegTranslation[IDX_SATA_SStatus].Addr   = BaseMemAddress +     (c << 6);
-                chan->RegTranslation[IDX_SATA_SStatus].MemIo  = MemIo;
-                chan->RegTranslation[IDX_SATA_SError].Addr    = BaseMemAddress + 4 + (c << 6);
-                chan->RegTranslation[IDX_SATA_SError].MemIo   = MemIo;
-                chan->RegTranslation[IDX_SATA_SControl].Addr  = BaseMemAddress + 8 + (c << 6);
-                chan->RegTranslation[IDX_SATA_SControl].MemIo = MemIo;
+                UniataInitIoRes(chan, IDX_SATA_SStatus,  BaseMemAddress +     (c << 6), MemIo, FALSE);
+                UniataInitIoRes(chan, IDX_SATA_SError,   BaseMemAddress + 4 + (c << 6), MemIo, FALSE);
+                UniataInitIoRes(chan, IDX_SATA_SControl, BaseMemAddress + 8 + (c << 6), MemIo, FALSE);
 
                 chan->ChannelCtrlFlags |= CTRFLAGS_NO_SLAVE;
             }
@@ -702,8 +765,7 @@ for_ugly_chips:
             KdPrint2((PRINT_PREFIX "MemIo\n"));
             MemIo = TRUE;
         }
-        deviceExtension->BaseIoAddressSATA_0.Addr  = BaseMemAddress;
-        deviceExtension->BaseIoAddressSATA_0.MemIo = MemIo;
+        UniataInitIoResEx(&deviceExtension->BaseIoAddressSATA_0, BaseMemAddress, MemIo, FALSE);
 
         /* BAR3 -> res2 */
         BaseMemAddress = AtapiGetIoRange(HwDeviceExtension, ConfigInfo, pciData, SystemIoBusNumber,
@@ -716,8 +778,7 @@ for_ugly_chips:
             KdPrint2((PRINT_PREFIX "MemIo\n"));
             MemIo = TRUE;
         }
-        deviceExtension->BaseIoAddressBM_0.Addr  = BaseMemAddress;
-        deviceExtension->BaseIoAddressBM_0.MemIo = MemIo;
+        UniataInitIoResEx(&deviceExtension->BaseIoAddressBM_0, BaseMemAddress, MemIo, FALSE);
 
         if(!(ChipFlags & UNIATA_SATA)) {
             UCHAR reg48;
@@ -740,31 +801,23 @@ for_ugly_chips:
             offs7 = c << 7;
 
             for (i=0; i<=IDX_IO1_SZ; i++) {
-                chan->RegTranslation[IDX_IO1+i].Addr           = BaseMemAddress + 0x200 + (i << 2) + offs7;
-                chan->RegTranslation[IDX_IO1+i].MemIo          = MemIo;
+                UniataInitIoRes(chan, IDX_IO1+i,          BaseMemAddress + 0x200 + (i << 2) + offs8, MemIo, FALSE);
             }
-            chan->RegTranslation[IDX_IO2_AltStatus].Addr       = BaseMemAddress + 0x238 + offs7;
-            chan->RegTranslation[IDX_IO2_AltStatus].MemIo      = MemIo;
+            UniataInitIoRes(chan, IDX_IO2_AltStatus,      BaseMemAddress + 0x238 + offs7, MemIo, FALSE);
 
             UniataInitSyncBaseIO(chan);
 
-            chan->RegTranslation[IDX_BM_Command].Addr          = BaseMemAddress + 0x260 + offs7;
-            chan->RegTranslation[IDX_BM_Command].MemIo         = MemIo;
-            chan->RegTranslation[IDX_BM_PRD_Table].Addr        = BaseMemAddress + 0x244 + offs7;
-            chan->RegTranslation[IDX_BM_PRD_Table].MemIo       = MemIo;
-            chan->RegTranslation[IDX_BM_DeviceSpecific0].Addr  = BaseMemAddress + (c << 2);
-            chan->RegTranslation[IDX_BM_DeviceSpecific0].MemIo = MemIo;
+            UniataInitIoRes(chan, IDX_BM_Command,         BaseMemAddress + 0x260 + offs7, MemIo, FALSE);
+            UniataInitIoRes(chan, IDX_BM_PRD_Table,       BaseMemAddress + 0x244 + offs7, MemIo, FALSE);
+            UniataInitIoRes(chan, IDX_BM_DeviceSpecific0, BaseMemAddress + (c << 2),      MemIo, FALSE);
 
             if((ChipFlags & PRSATA) ||
                ((ChipFlags & PRCMBO) && c<2)) {
                 KdPrint2((PRINT_PREFIX "Promise SATA\n"));
 
-                chan->RegTranslation[IDX_SATA_SStatus].Addr        = BaseMemAddress + 0x400 + offs7;
-                chan->RegTranslation[IDX_SATA_SStatus].MemIo       = MemIo;
-                chan->RegTranslation[IDX_SATA_SError].Addr         = BaseMemAddress + 0x404 + offs7;
-                chan->RegTranslation[IDX_SATA_SError].MemIo        = MemIo;
-                chan->RegTranslation[IDX_SATA_SControl].Addr       = BaseMemAddress + 0x408 + offs7;
-                chan->RegTranslation[IDX_SATA_SControl].MemIo      = MemIo;
+                UniataInitIoRes(chan, IDX_SATA_SStatus,  BaseMemAddress + 0x400 + offs7, MemIo, FALSE);
+                UniataInitIoRes(chan, IDX_SATA_SError,   BaseMemAddress + 0x404 + offs7, MemIo, FALSE);
+                UniataInitIoRes(chan, IDX_SATA_SControl, BaseMemAddress + 0x408 + offs7, MemIo, FALSE);
 
                 chan->ChannelCtrlFlags |= CTRFLAGS_NO_SLAVE;
             } else {
@@ -815,8 +868,7 @@ for_ugly_chips:
             KdPrint2((PRINT_PREFIX "MemIo\n"));
             MemIo = TRUE;
         }
-        deviceExtension->BaseIoAddressSATA_0.Addr  = BaseMemAddress;
-        deviceExtension->BaseIoAddressSATA_0.MemIo = MemIo;
+        UniataInitIoResEx(&deviceExtension->BaseIoAddressSATA_0, BaseMemAddress, MemIo, FALSE);
 
         for(c=0; c<deviceExtension->NumberChannels; c++) {
             ULONG unit01 = (c & 1);
@@ -826,25 +878,16 @@ for_ugly_chips:
 
             if(deviceExtension->AltRegMap) {
                 for (i=0; i<=IDX_IO1_SZ; i++) {
-                    chan->RegTranslation[IDX_IO1+i].Addr           = BaseMemAddress + 0x80 + i + (unit01 << 6) + (unit10 << 8);
-                    chan->RegTranslation[IDX_IO1+i].MemIo          = MemIo;
+                    UniataInitIoRes(chan, IDX_IO1+i, BaseMemAddress + 0x80 + i + (unit01 << 6) + (unit10 << 8), MemIo, FALSE);
                 }
-                chan->RegTranslation[IDX_IO2_AltStatus].Addr       = BaseMemAddress + 0x8a + (unit01 << 6) + (unit10 << 8);
-                chan->RegTranslation[IDX_IO2_AltStatus].MemIo      = MemIo;
+                UniataInitIoRes(chan, IDX_IO2_AltStatus, BaseMemAddress + 0x8a + (unit01 << 6) + (unit10 << 8), MemIo, FALSE);
                 UniataInitSyncBaseIO(chan);
 
-                chan->RegTranslation[IDX_BM_Command].Addr          = BaseMemAddress + 0x00 + (unit01 << 3) + (unit10 << 8);
-                chan->RegTranslation[IDX_BM_Command].MemIo         = MemIo;
-                chan->RegTranslation[IDX_BM_Status].Addr           = BaseMemAddress + 0x02 + (unit01 << 3) + (unit10 << 8);
-                chan->RegTranslation[IDX_BM_Status].MemIo          = MemIo;
-                chan->RegTranslation[IDX_BM_PRD_Table].Addr        = BaseMemAddress + 0x04 + (unit01 << 3) + (unit10 << 8);
-                chan->RegTranslation[IDX_BM_PRD_Table].MemIo       = MemIo;
-                //chan->RegTranslation[IDX_BM_DeviceSpecific0].Addr  = BaseMemAddress + 0xa1 + (unit01 << 6) + (unit10 << 8);
-                //chan->RegTranslation[IDX_BM_DeviceSpecific0].MemIo = MemIo;
-                chan->RegTranslation[IDX_BM_DeviceSpecific0].Addr  = BaseMemAddress + 0x10 + (unit01 << 3) + (unit10 << 8);
-                chan->RegTranslation[IDX_BM_DeviceSpecific0].MemIo = MemIo;
-                chan->RegTranslation[IDX_BM_DeviceSpecific1].Addr  = BaseMemAddress + 0x40 + (unit01 << 2) + (unit10 << 8);
-                chan->RegTranslation[IDX_BM_DeviceSpecific1].MemIo = MemIo;
+                UniataInitIoRes(chan, IDX_BM_Command,   BaseMemAddress + 0x00 + (unit01 << 3) + (unit10 << 8), MemIo, FALSE);
+                UniataInitIoRes(chan, IDX_BM_Status,    BaseMemAddress + 0x02 + (unit01 << 3) + (unit10 << 8), MemIo, FALSE);
+                UniataInitIoRes(chan, IDX_BM_PRD_Table, BaseMemAddress + 0x04 + (unit01 << 3) + (unit10 << 8), MemIo, FALSE);
+                UniataInitIoRes(chan, IDX_BM_DeviceSpecific0, BaseMemAddress + 0x10 + (unit01 << 3) + (unit10 << 8), MemIo, FALSE);
+                UniataInitIoRes(chan, IDX_BM_DeviceSpecific1, BaseMemAddress + 0x40 + (unit01 << 2) + (unit10 << 8), MemIo, FALSE);
             }
 
             if(chan->MaxTransferMode < ATA_SA150) {
@@ -852,12 +895,9 @@ for_ugly_chips:
                 KdPrint2((PRINT_PREFIX "No SATA regs for PATA part\n"));
             } else
             if(ChipFlags & UNIATA_SATA) {
-                chan->RegTranslation[IDX_SATA_SStatus].Addr        = BaseMemAddress + 0x104 + (unit01 << 7) + (unit10 << 8);
-                chan->RegTranslation[IDX_SATA_SStatus].MemIo       = MemIo;
-                chan->RegTranslation[IDX_SATA_SError].Addr         = BaseMemAddress + 0x108 + (unit01 << 7) + (unit10 << 8);
-                chan->RegTranslation[IDX_SATA_SError].MemIo        = MemIo;
-                chan->RegTranslation[IDX_SATA_SControl].Addr       = BaseMemAddress + 0x100 + (unit01 << 7) + (unit10 << 8);
-                chan->RegTranslation[IDX_SATA_SControl].MemIo      = MemIo;
+                UniataInitIoRes(chan, IDX_SATA_SStatus,  BaseMemAddress + 0x104 + (unit01 << 7) + (unit10 << 8), MemIo, FALSE);
+                UniataInitIoRes(chan, IDX_SATA_SError,   BaseMemAddress + 0x108 + (unit01 << 2) + (unit10 << 8), MemIo, FALSE);
+                UniataInitIoRes(chan, IDX_SATA_SControl, BaseMemAddress + 0x100 + (unit01 << 2) + (unit10 << 8), MemIo, FALSE);
 
                 chan->ChannelCtrlFlags |= CTRFLAGS_NO_SLAVE;
             }
@@ -886,34 +926,25 @@ for_ugly_chips:
             KdPrint2((PRINT_PREFIX "MemIo\n"));
             MemIo = TRUE;
         }
-        deviceExtension->BaseIoAddressSATA_0.Addr  = BaseMemAddress;
-        deviceExtension->BaseIoAddressSATA_0.MemIo = MemIo;
+        UniataInitIoResEx(&deviceExtension->BaseIoAddressSATA_0, BaseMemAddress, MemIo, FALSE);
 
         for(c=0; c<deviceExtension->NumberChannels; c++) {
             ULONG offs = c*0x100;
 
             chan = &deviceExtension->chan[c];
             for (i=0; i<=IDX_IO1_SZ; i++) {
-                chan->RegTranslation[IDX_IO1+i].Addr           = BaseMemAddress + offs + i*4;
-                chan->RegTranslation[IDX_IO1+i].MemIo          = MemIo;
+                UniataInitIoRes(chan, IDX_IO1+i, BaseMemAddress + offs + i*4, MemIo, FALSE);
             }
-            chan->RegTranslation[IDX_IO2_AltStatus].Addr       = BaseMemAddress + offs + 0x20;
-            chan->RegTranslation[IDX_IO2_AltStatus].MemIo      = MemIo;
+            UniataInitIoRes(chan, IDX_IO2_AltStatus, BaseMemAddress + offs + 0x20, MemIo, FALSE);
             UniataInitSyncBaseIO(chan);
 
-            chan->RegTranslation[IDX_BM_Command].Addr          = BaseMemAddress + offs + 0x30;
-            chan->RegTranslation[IDX_BM_Command].MemIo         = MemIo;
-            chan->RegTranslation[IDX_BM_Status].Addr           = BaseMemAddress + offs + 0x32;
-            chan->RegTranslation[IDX_BM_Status].MemIo          = MemIo;
-            chan->RegTranslation[IDX_BM_PRD_Table].Addr        = BaseMemAddress + offs + 0x34;
-            chan->RegTranslation[IDX_BM_PRD_Table].MemIo       = MemIo;
+            UniataInitIoRes(chan, IDX_BM_Command,   BaseMemAddress + offs + 0x30, MemIo, FALSE);
+            UniataInitIoRes(chan, IDX_BM_Status,    BaseMemAddress + offs + 0x32, MemIo, FALSE);
+            UniataInitIoRes(chan, IDX_BM_PRD_Table, BaseMemAddress + offs + 0x34, MemIo, FALSE);
 
-            chan->RegTranslation[IDX_SATA_SStatus].Addr        = BaseMemAddress + offs + 0x40;
-            chan->RegTranslation[IDX_SATA_SStatus].MemIo       = MemIo;               
-            chan->RegTranslation[IDX_SATA_SError].Addr         = BaseMemAddress + offs + 0x44;
-            chan->RegTranslation[IDX_SATA_SError].MemIo        = MemIo;               
-            chan->RegTranslation[IDX_SATA_SControl].Addr       = BaseMemAddress + offs + 0x48;
-            chan->RegTranslation[IDX_SATA_SControl].MemIo      = MemIo;
+            UniataInitIoRes(chan, IDX_SATA_SStatus,  BaseMemAddress + offs + 0x40, MemIo, FALSE);
+            UniataInitIoRes(chan, IDX_SATA_SError,   BaseMemAddress + offs + 0x44, MemIo, FALSE);
+            UniataInitIoRes(chan, IDX_SATA_SControl, BaseMemAddress + offs + 0x48, MemIo, FALSE);
 
             chan->ChannelCtrlFlags |= CTRFLAGS_NO_SLAVE;
         }
@@ -941,7 +972,7 @@ for_ugly_chips:
             // Restore device ID
             ChangePciConfig1(0x57, (a | 0x80));
         } else {
-            static BUSMASTER_CONTROLLER_INFORMATION const SiSSouthAdapters[] = {
+            static BUSMASTER_CONTROLLER_INFORMATION_BASE const SiSSouthAdapters[] = {
                 PCI_DEV_HW_SPEC_BM( 0008, 1039, 0x10, ATA_MODE_NOT_SPEC, "SiS 961", 0 ),
 //                PCI_DEV_HW_SPEC_BM( 0008, 1039, 0x00, ATA_MODE_NOT_SPEC, "SiS 961", 0 ),
                 PCI_DEV_HW_SPEC_BM( ffff, ffff, 0xff, ATA_MODE_NOT_SPEC, NULL     , -1 )
@@ -951,16 +982,19 @@ for_ugly_chips:
             ChangePciConfig1(0x4a, (a | 0x10));
             if(tmp32 == ATA_SIS5513 ||
                tmp32 == ATA_SIS5517) {
-                i = AtapiFindListedDev((BUSMASTER_CONTROLLER_INFORMATION*)&SiSSouthAdapters[0],
+                i = AtapiFindListedDev((BUSMASTER_CONTROLLER_INFORMATION_BASE*)&SiSSouthAdapters[0],
                      -1, HwDeviceExtension, SystemIoBusNumber, PCISLOTNUM_NOT_SPECIFIED, NULL); 
                 if(i != BMLIST_TERMINATOR) {
+                    KdPrint2((PRINT_PREFIX "SIS South\n"));
                     deviceExtension->HwFlags = (deviceExtension->HwFlags & ~CHIPTYPE_MASK) | SIS133OLD;
                     deviceExtension->MaxTransferMode = ATA_UDMA6;
                     //deviceExtension->MaxTransferMode = SiSSouthAdapters[i].MaxTransferMode;
                     if(SiSSouthAdapters[i].RaidFlags & UNIATA_SATA) {
+                        KdPrint2((PRINT_PREFIX "SIS South SATA\n"));
                         deviceExtension->HwFlags |= UNIATA_SATA;
                         if(SiSSouthAdapters[i].nDeviceId == 0x1182 ||
                            SiSSouthAdapters[i].nDeviceId == 0x1183) {
+                            KdPrint2((PRINT_PREFIX "SIS_182\n"));
                             SIS_182 = TRUE;
                         }
                     }
@@ -990,19 +1024,15 @@ for_ugly_chips:
                         KdPrint2((PRINT_PREFIX "MemIo\n"));
                         MemIo = TRUE;
                     }
-                    deviceExtension->BaseIoAddressSATA_0.Addr  = BaseMemAddress;
-                    deviceExtension->BaseIoAddressSATA_0.MemIo = MemIo;
+                    UniataInitIoResEx(&deviceExtension->BaseIoAddressSATA_0, BaseMemAddress, MemIo, FALSE);
 
                     for(c=0; c<deviceExtension->NumberChannels; c++) {
                         ULONG offs = c << (SIS_182 ? 5 : 6);
 
                         chan = &deviceExtension->chan[c];
-                        chan->RegTranslation[IDX_SATA_SStatus].Addr        = BaseMemAddress + 0 + offs;
-                        chan->RegTranslation[IDX_SATA_SStatus].MemIo       = MemIo;
-                        chan->RegTranslation[IDX_SATA_SError].Addr         = BaseMemAddress + 4 + offs;
-                        chan->RegTranslation[IDX_SATA_SError].MemIo        = MemIo;
-                        chan->RegTranslation[IDX_SATA_SControl].Addr       = BaseMemAddress + 8 + offs;
-                        chan->RegTranslation[IDX_SATA_SControl].MemIo      = MemIo;
+                        UniataInitIoRes(chan, IDX_SATA_SStatus,  BaseMemAddress + 0 + offs, MemIo, FALSE);
+                        UniataInitIoRes(chan, IDX_SATA_SError,   BaseMemAddress + 4 + offs, MemIo, FALSE);
+                        UniataInitIoRes(chan, IDX_SATA_SControl, BaseMemAddress + 8 + offs, MemIo, FALSE);
 
                         chan->ChannelCtrlFlags |= CTRFLAGS_NO_SLAVE;
                     }
@@ -1023,7 +1053,7 @@ for_ugly_chips:
         if(ChipFlags & UNIATA_SATA) {
 
             ULONG IoSize = 0;
-            ULONG BaseMemAddress = 0;
+            BaseMemAddress = 0;
 
             switch(DeviceID) {
             case 0x3149: // VIA 6420
@@ -1043,8 +1073,7 @@ for_ugly_chips:
                     KdPrint2((PRINT_PREFIX "MemIo\n"));
                     MemIo = TRUE;
                 }
-                deviceExtension->BaseIoAddressSATA_0.Addr  = BaseMemAddress;
-                deviceExtension->BaseIoAddressSATA_0.MemIo = MemIo;
+                UniataInitIoResEx(&deviceExtension->BaseIoAddressSATA_0, BaseMemAddress, MemIo, FALSE);
             }
             if(/*deviceExtension->*/BaseMemAddress) {
                 KdPrint2((PRINT_PREFIX "UniataChipDetect: BAR5 %x\n", /*deviceExtension->*/BaseMemAddress));
@@ -1065,13 +1094,13 @@ for_ugly_chips:
                         BaseIo = AtapiGetIoRange(HwDeviceExtension, ConfigInfo, pciData, SystemIoBusNumber, c, 0, /*0x80*/ sizeof(IDE_REGISTERS_1) + sizeof(IDE_REGISTERS_2)*2);
 
                         for (i=0; i<=IDX_IO1_SZ; i++) {
-                            chan->RegTranslation[IDX_IO1+i].Addr           = BaseIo + i;
+                            UniataInitIoRes(chan, IDX_IO1+i, BaseIo + i, FALSE, FALSE);
                         }
-                        chan->RegTranslation[IDX_IO2_AltStatus].Addr       = BaseIo + sizeof(IDE_REGISTERS_1) + 2;
+                        UniataInitIoRes(chan, IDX_IO2_AltStatus, BaseIo + sizeof(IDE_REGISTERS_1) + 2, FALSE, FALSE);
                         UniataInitSyncBaseIO(chan);
 
                         for (i=0; i<=IDX_BM_IO_SZ; i++) {
-                            chan->RegTranslation[IDX_BM_IO+i].Addr         = BaseIoAddressBM_0 + sizeof(IDE_BUSMASTER_REGISTERS)*c + i;
+                            UniataInitIoRes(chan, IDX_BM_IO+i, BaseIoAddressBM_0 + sizeof(IDE_BUSMASTER_REGISTERS)*c + i, FALSE, FALSE);
                         }
 
                     }
@@ -1081,17 +1110,13 @@ for_ugly_chips:
                     if((ChipFlags & VIABAR) && (c==2)) {
                         // Do not setup SATA registers for PATA part
                         for (i=0; i<=IDX_SATA_IO_SZ; i++) {
-                            chan->RegTranslation[IDX_SATA_IO+i].Addr = 0;
-                            chan->RegTranslation[IDX_SATA_IO+i].MemIo = 0;
+                            UniataInitIoRes(chan, IDX_SATA_IO+i, 0, FALSE, FALSE);
                         }
                         break;
                     }
-                    chan->RegTranslation[IDX_SATA_SStatus].Addr        = BaseMemAddress + (c * IoSize);
-                    chan->RegTranslation[IDX_SATA_SStatus].MemIo       = MemIo;
-                    chan->RegTranslation[IDX_SATA_SError].Addr         = BaseMemAddress + 4 + (c * IoSize);
-                    chan->RegTranslation[IDX_SATA_SError].MemIo        = MemIo;
-                    chan->RegTranslation[IDX_SATA_SControl].Addr       = BaseMemAddress + 8 + (c * IoSize);
-                    chan->RegTranslation[IDX_SATA_SControl].MemIo      = MemIo;
+                    UniataInitIoRes(chan, IDX_SATA_SStatus,  BaseMemAddress + (c * IoSize), MemIo, FALSE);
+                    UniataInitIoRes(chan, IDX_SATA_SError,   BaseMemAddress + 4 + (c * IoSize), MemIo, FALSE);
+                    UniataInitIoRes(chan, IDX_SATA_SControl, BaseMemAddress + 8 + (c * IoSize), MemIo, FALSE);
 
                     chan->ChannelCtrlFlags |= CTRFLAGS_NO_SLAVE;
                 }
@@ -1120,49 +1145,31 @@ for_ugly_chips:
                 MemIo = TRUE;
             }
             deviceExtension->AltRegMap = TRUE; // inform generic resource allocator
-            deviceExtension->BaseIoAddressSATA_0.Addr  = BaseMemAddress;
-            deviceExtension->BaseIoAddressSATA_0.MemIo = MemIo;
+            UniataInitIoResEx(&deviceExtension->BaseIoAddressSATA_0, BaseMemAddress, MemIo, FALSE);
 
             for(c=0; c<deviceExtension->NumberChannels; c++) {
                 ULONG offs = 0x200 + c*0x200;
 
                 chan = &deviceExtension->chan[c];
                 for (i=0; i<=IDX_IO1_SZ; i++) {
-                    chan->RegTranslation[IDX_IO1+i].MemIo          = MemIo;
-                    chan->RegTranslation[IDX_IO1_o+i].MemIo        = MemIo;
+                    UniataInitIoRes(chan, IDX_BM_IO+i, BaseMemAddress + i*4 + offs, MemIo, FALSE);
                 }
 
-                chan->RegTranslation[IDX_IO1_i_Data        ].Addr       = BaseMemAddress + 0x00 + offs;
-                chan->RegTranslation[IDX_IO1_i_Error       ].Addr       = BaseMemAddress + 0x04 + offs;
-                chan->RegTranslation[IDX_IO1_i_BlockCount  ].Addr       = BaseMemAddress + 0x08 + offs;
-                chan->RegTranslation[IDX_IO1_i_BlockNumber ].Addr       = BaseMemAddress + 0x0c + offs;
-                chan->RegTranslation[IDX_IO1_i_CylinderLow ].Addr       = BaseMemAddress + 0x10 + offs;
-                chan->RegTranslation[IDX_IO1_i_CylinderHigh].Addr       = BaseMemAddress + 0x14 + offs;
-                chan->RegTranslation[IDX_IO1_i_DriveSelect ].Addr       = BaseMemAddress + 0x18 + offs;
-                chan->RegTranslation[IDX_IO1_i_Status      ].Addr       = BaseMemAddress + 0x1c + offs;
-
                 UniataInitSyncBaseIO(chan);
 
-                chan->RegTranslation[IDX_IO1_o_Command     ].Addr       = BaseMemAddress + 0x1d + offs;
-                chan->RegTranslation[IDX_IO1_o_Feature     ].Addr       = BaseMemAddress + 0x06 + offs;
-                chan->RegTranslation[IDX_IO2_o_Control     ].Addr       = BaseMemAddress + 0x29 + offs;
+                UniataInitIoRes(chan, IDX_IO1_o_Command, BaseMemAddress + 0x1d + offs, MemIo, FALSE);
+                UniataInitIoRes(chan, IDX_IO1_o_Feature, BaseMemAddress + 0x06 + offs, MemIo, FALSE);
+                UniataInitIoRes(chan, IDX_IO2_o_Control, BaseMemAddress + 0x29 + offs, MemIo, FALSE);
 
-                chan->RegTranslation[IDX_IO2_AltStatus].Addr       = BaseMemAddress + 0x28 + offs;
-                chan->RegTranslation[IDX_IO2_AltStatus].MemIo      = MemIo;
+                UniataInitIoRes(chan, IDX_IO2_AltStatus, BaseMemAddress + 0x28 + offs, MemIo, FALSE);
 
-                chan->RegTranslation[IDX_BM_Command].Addr          = BaseMemAddress + offs + 0x70;
-                chan->RegTranslation[IDX_BM_Command].MemIo         = MemIo;
-                chan->RegTranslation[IDX_BM_Status].Addr           = BaseMemAddress + offs + 0x72;
-                chan->RegTranslation[IDX_BM_Status].MemIo          = MemIo;
-                chan->RegTranslation[IDX_BM_PRD_Table].Addr        = BaseMemAddress + offs + 0x74;
-                chan->RegTranslation[IDX_BM_PRD_Table].MemIo       = MemIo;
+                UniataInitIoRes(chan, IDX_BM_Command,   BaseMemAddress + 0x70 + offs, MemIo, FALSE);
+                UniataInitIoRes(chan, IDX_BM_Status,    BaseMemAddress + 0x72 + offs, MemIo, FALSE);
+                UniataInitIoRes(chan, IDX_BM_PRD_Table, BaseMemAddress + 0x74 + offs, MemIo, FALSE);
 
-                chan->RegTranslation[IDX_SATA_SStatus].Addr        = BaseMemAddress + 0x100 + offs;
-                chan->RegTranslation[IDX_SATA_SStatus].MemIo       = MemIo;
-                chan->RegTranslation[IDX_SATA_SError].Addr         = BaseMemAddress + 0x104 + offs;
-                chan->RegTranslation[IDX_SATA_SError].MemIo        = MemIo;
-                chan->RegTranslation[IDX_SATA_SControl].Addr       = BaseMemAddress + 0x108 + offs;
-                chan->RegTranslation[IDX_SATA_SControl].MemIo      = MemIo;
+                UniataInitIoRes(chan, IDX_SATA_SStatus,  BaseMemAddress + 0x100 + offs, MemIo, FALSE);
+                UniataInitIoRes(chan, IDX_SATA_SError,   BaseMemAddress + 0x104 + offs, MemIo, FALSE);
+                UniataInitIoRes(chan, IDX_SATA_SControl, BaseMemAddress + 0x108 + offs, MemIo, FALSE);
 
                 chan->ChannelCtrlFlags |= CTRFLAGS_NO_SLAVE;
             }
@@ -1170,22 +1177,63 @@ for_ugly_chips:
             break;
         }
         if(deviceExtension->MaxTransferMode >= ATA_SA150) {
+
+            BOOLEAN OrigAHCI = FALSE;
+
             GetPciConfig1(0x90, tmp8);
             KdPrint2((PRINT_PREFIX "Intel chip config: %x\n", tmp8));
             /* SATA parts can be either compat or AHCI */
+            MemIo = FALSE;
             if(ChipFlags & UNIATA_AHCI) {
-
+                OrigAHCI = TRUE;
                 if(tmp8 & 0xc0) {
                     //KdPrint2((PRINT_PREFIX "AHCI not supported yet\n"));
                     //return FALSE;
                     KdPrint2((PRINT_PREFIX "try run AHCI\n"));
-                    break;
+                    if(ScsiPortConvertPhysicalAddressToUlong((*ConfigInfo->AccessRanges)[5].RangeStart)) {
+                        break;
+                    }
+                    KdPrint2((PRINT_PREFIX "No BAR5, try BM\n"));
+                    deviceExtension->HwFlags &= ~UNIATA_AHCI;
+                }
+                BaseIoAddressBM = AtapiGetIoRange(HwDeviceExtension, ConfigInfo, pciData, SystemIoBusNumber,
+                                        4, 0, sizeof(IDE_BUSMASTER_REGISTERS));
+                if(BaseIoAddressBM) {
+                    KdPrint2((PRINT_PREFIX "Intel BM check at %x\n", BaseIoAddressBM));
+                    /* check if we really have valid BM registers */
+                    if((*ConfigInfo->AccessRanges)[4].RangeInMemory) {
+                        KdPrint2((PRINT_PREFIX "MemIo[4]\n"));
+                        MemIo = TRUE;
+                    }
+                    UniataInitIoResEx(&deviceExtension->BaseIoAddressBM_0, BaseIoAddressBM, MemIo, FALSE);
+
+                    tmp8 = AtapiReadPortEx1(NULL, (ULONGIO_PTR)(&deviceExtension->BaseIoAddressBM_0),IDX_BM_Status);
+                    KdPrint2((PRINT_PREFIX "BM status: %x\n", tmp8));
+                    /* cleanup */
+                    ScsiPortFreeDeviceBase(HwDeviceExtension, (PCHAR)(ULONG_PTR)BaseIoAddressBM);
+                    UniataInitIoResEx(&deviceExtension->BaseIoAddressBM_0, 0, 0, FALSE);
+
+                    if(tmp8 == 0xff) {
+                        KdPrint2((PRINT_PREFIX "invalid BM status, keep AHCI mode\n"));
+                        break;
+                    }
+                }
+                KdPrint2((PRINT_PREFIX "Compatible mode, reallocate LUNs\n"));
+                deviceExtension->NumberLuns = 2; // we may be in Legacy mode
+                if(!UniataAllocateLunExt(deviceExtension, 2)) {
+                    KdPrint2((PRINT_PREFIX "can't re-allocate Luns\n"));
+                    return STATUS_UNSUCCESSFUL;
                 }
-                KdPrint2((PRINT_PREFIX "Compatible mode\n"));
             }
             deviceExtension->HwFlags &= ~UNIATA_AHCI;
 
+            MemIo = FALSE;
             /* if BAR(5) is IO it should point to SATA interface registers */
+            if(OrigAHCI) {
+                /* Skip BAR(5) in compatible mode */
+                KdPrint2((PRINT_PREFIX "Ignore BAR5 on compatible\n"));
+                BaseMemAddress = 0;
+            } else
             if(deviceExtension->DevID == 0x28288086 &&
                 pciData->u.type0.SubVendorID == 0x106b) {
                 /* Skip BAR(5) on ICH8M Apples, system locks up on access. */
@@ -1195,15 +1243,15 @@ for_ugly_chips:
                 BaseMemAddress = AtapiGetIoRange(HwDeviceExtension, ConfigInfo, pciData, SystemIoBusNumber,
                                     5, 0, 0x10);
                 if(BaseMemAddress && (*ConfigInfo->AccessRanges)[5].RangeInMemory) {
-                    KdPrint2((PRINT_PREFIX "MemIo\n"));
+                    KdPrint2((PRINT_PREFIX "MemIo[5]\n"));
                     MemIo = TRUE;
                 }
             }
-            deviceExtension->BaseIoAddressSATA_0.Addr  = BaseMemAddress;
-            deviceExtension->BaseIoAddressSATA_0.MemIo = MemIo;
+            UniataInitIoResEx(&deviceExtension->BaseIoAddressSATA_0, BaseMemAddress, MemIo, FALSE);
 
             for(c=0; c<deviceExtension->NumberChannels; c++) {
                 chan = &deviceExtension->chan[c];
+                AtapiSetupLunPtrs(chan, deviceExtension, c);
                 IsPata = FALSE;
                 if(ChipFlags & ICH5) {
                     KdPrint2((PRINT_PREFIX "ICH5\n"));
@@ -1249,10 +1297,8 @@ for_ugly_chips:
 
                     if(!(ChipFlags & ICH7) && BaseMemAddress) {
                         KdPrint2((PRINT_PREFIX "BaseMemAddress[5] -> indexed\n"));
-                        chan->RegTranslation[IDX_INDEXED_ADDR].Addr        = BaseMemAddress + 0;
-                        chan->RegTranslation[IDX_INDEXED_ADDR].MemIo       = MemIo;
-                        chan->RegTranslation[IDX_INDEXED_DATA].Addr        = BaseMemAddress + 4;
-                        chan->RegTranslation[IDX_INDEXED_DATA].MemIo       = MemIo;
+                        UniataInitIoRes(chan, IDX_INDEXED_ADDR,   BaseMemAddress + 0, MemIo, FALSE);
+                        UniataInitIoRes(chan, IDX_INDEXED_DATA,   BaseMemAddress + 4, MemIo, FALSE);
                     }
                     if((ChipFlags & ICH5) || BaseMemAddress) {
 
@@ -1265,12 +1311,9 @@ for_ugly_chips:
                         if(ChipFlags & ICH7) {
                             KdPrint2((PRINT_PREFIX "ICH7 way\n"));
                         }
-                        chan->RegTranslation[IDX_SATA_SStatus].Addr        = 0x200*c + 0; // this is fake non-zero value
-                        chan->RegTranslation[IDX_SATA_SStatus].Proc        = 1;
-                        chan->RegTranslation[IDX_SATA_SError].Addr         = 0x200*c + 2; // this is fake non-zero value
-                        chan->RegTranslation[IDX_SATA_SError].Proc         = 1;
-                        chan->RegTranslation[IDX_SATA_SControl].Addr       = 0x200*c + 1; // this is fake non-zero value
-                        chan->RegTranslation[IDX_SATA_SControl].Proc       = 1;
+                        UniataInitIoRes(chan, IDX_SATA_SStatus,  0x200*c + 0, FALSE, TRUE); // this is fake non-zero value
+                        UniataInitIoRes(chan, IDX_SATA_SError,   0x200*c + 2, FALSE, TRUE);
+                        UniataInitIoRes(chan, IDX_SATA_SControl, 0x200*c + 1, FALSE, TRUE);
                     }
                 }
 
@@ -1288,7 +1331,7 @@ for_ugly_chips:
         }
         break;
     case ATA_JMICRON_ID:
-        /* New JMicron PATA controllers */
+        /* New JMicron PATA/SATA controllers */
         GetPciConfig1(0xdf, tmp8);
         if(tmp8 & 0x40) {
             KdPrint(("  Check JMicron AHCI\n"));
@@ -1296,13 +1339,16 @@ for_ugly_chips:
                 ChipFlags |= UNIATA_AHCI;
                 deviceExtension->HwFlags |= UNIATA_AHCI;
             } else {
-                KdPrint(("  JMicron PATA\n"));
+                KdPrint(("  JMicron PATA/SATA\n"));
             }
         } else {
+#if 0 // do not touch, see Linux sources
             /* set controller configuration to a combined setup we support */
             SetPciConfig4(0x40, 0x80c0a131);
             SetPciConfig4(0x80, 0x01200000);
-            //KdPrint(("  JMicron Combined (not supported yet)\n"));
+#endif
+            //GetPciConfig1(0x40, tmp32);
+            KdPrint(("  JMicron Combined\n"));
             //return STATUS_NOT_FOUND;
         }
         break;
@@ -1502,8 +1548,19 @@ hpt_cable80(
     UCHAR reg, val, res;
     PCI_SLOT_NUMBER slotData;
 
+    PHW_CHANNEL chan;
+    ULONG  c; // logical channel (for Compatible Mode controllers)
+
+    c = channel - deviceExtension->Channel; // logical channel (for Compatible Mode controllers)
+    chan = &deviceExtension->chan[c];
+
     slotData.u.AsULONG = deviceExtension->slotNumber;
 
+    if(deviceExtension->HwFlags & UNIATA_NO80CHK) {
+        KdPrint2((PRINT_PREFIX "UNIATA_NO80CHK\n"));
+        return TRUE;
+    }
+
     if(ChipType == HPT374 && slotData.u.bits.FunctionNumber == 1)  {
         reg = channel ? 0x57 : 0x53;
         GetPciConfig1(reg, val);
@@ -1517,6 +1574,10 @@ hpt_cable80(
     GetPciConfig1(0x5a, res);
     res = res & (channel ? 0x01 : 0x02);
     SetPciConfig1(reg, val);
+    if(chan->Force80pin) {
+        KdPrint2((PRINT_PREFIX "Force80pin\n"));
+        res = 0;
+    }
     KdPrint2((PRINT_PREFIX "hpt_cable80(%d) = %d\n", channel, !res));
     return !res;
 } // end hpt_cable80()
@@ -1598,6 +1659,11 @@ generic_cable80(
     c = channel - deviceExtension->Channel; // logical channel (for Compatible Mode controllers)
     chan = &deviceExtension->chan[c];
 
+    if(chan->Force80pin) {
+        KdPrint2((PRINT_PREFIX "Force80pin\n"));
+        return TRUE;
+    }
+
     GetPciConfig1(pci_reg, tmp8);
     if(!(tmp8 & (1 << (channel << bit_offs)))) {
         chan->MaxTransferMode = min(deviceExtension->MaxTransferMode, ATA_UDMA2);
@@ -1703,7 +1769,10 @@ UniAtaReadLunConfig(
     if(tmp32) {
         LunExt->DeviceFlags |= DFLAGS_HIDDEN;
     }
-
+    tmp32 = AtapiRegCheckDevValue(deviceExtension, channel, DeviceNumber, L"Exclude", 0);
+    if(tmp32) {
+        LunExt->DeviceFlags |= DFLAGS_HIDDEN;
+    }
 
     return;
 } // end UniAtaReadLunConfig()
@@ -1753,6 +1822,13 @@ AtapiReadChipConfig(
                 KdPrint2((PRINT_PREFIX "MaxTransferMode (overriden): %#x\n", chan->MaxTransferMode));
                 chan->MaxTransferMode = tmp32;
             }
+            tmp32 = AtapiRegCheckDevValue(deviceExtension, c, DEVNUM_NOT_SPECIFIED, L"Force80pin", FALSE);
+            chan->Force80pin = tmp32 ? TRUE : FALSE;
+            if(chan->Force80pin) {
+                KdPrint2((PRINT_PREFIX "Force80pin on chip\n"));
+                deviceExtension->HwFlags |= UNIATA_NO80CHK;
+            }
+
             //UniAtaReadLunConfig(deviceExtension, c, 0);
             //UniAtaReadLunConfig(deviceExtension, c, 1);
         }
@@ -1784,6 +1860,12 @@ AtapiReadChipConfig(
         tmp32 = AtapiRegCheckDevValue(deviceExtension, c, DEVNUM_NOT_SPECIFIED, L"ReorderEnable", TRUE);
         chan->UseReorder = tmp32 ? TRUE : FALSE;
 
+        tmp32 = AtapiRegCheckDevValue(deviceExtension, c, DEVNUM_NOT_SPECIFIED, L"Force80pin", FALSE);
+        chan->Force80pin = tmp32 ? TRUE : FALSE;
+        if(chan->Force80pin) {
+            KdPrint2((PRINT_PREFIX "Force80pin on channel\n"));
+        }
+
         for(i=0; i<deviceExtension->NumberLuns; i++) {
             UniAtaReadLunConfig(deviceExtension, channel, i);
         }
@@ -1807,13 +1889,13 @@ AtapiChipInit(
     ULONG DeviceID = (deviceExtension->DevID >> 16) & 0xffff;
     ULONG RevID    =  deviceExtension->RevID;
 //    ULONG i;
-//    BUSMASTER_CONTROLLER_INFORMATION* DevTypeInfo;
+//    BUSMASTER_CONTROLLER_INFORMATION_BASE* DevTypeInfo;
     ULONG ChipType  = deviceExtension->HwFlags & CHIPTYPE_MASK;
     ULONG ChipFlags = deviceExtension->HwFlags & CHIPFLAG_MASK;
     PHW_CHANNEL chan;
     UCHAR  tmp8;
     USHORT tmp16;
-    //ULONG  tmp32;
+    ULONG  tmp32;
     ULONG  c; // logical channel (for Compatible Mode controllers)
     BOOLEAN CheckCable = FALSE;
     BOOLEAN GlobalInit = FALSE;
@@ -1982,12 +2064,13 @@ AtapiChipInit(
     case ATA_INTEL_ID: {
         BOOLEAN IsPata;
         USHORT reg54;
-        UCHAR tmp8;
         if(ChipFlags & UNIATA_SATA) {
 
             KdPrint2((PRINT_PREFIX "Intel SATA\n"));
             if(ChipFlags & UNIATA_AHCI) {
                 KdPrint2((PRINT_PREFIX "Do nothing for AHCI\n"));
+                /* enable PCI interrupt */
+                ChangePciConfig2(offsetof(PCI_COMMON_CONFIG, Command), (a & ~0x0400));
                 break;
             }
             if(c == CHAN_NOT_SPECIFIED) {
@@ -2105,6 +2188,12 @@ AtapiChipInit(
             chan = &deviceExtension->chan[c];
             GetPciConfig2(0x54, reg54);
             KdPrint2((PRINT_PREFIX " intel 80-pin check (reg54=%x)\n", reg54));
+            if(deviceExtension->HwFlags & UNIATA_NO80CHK) {
+                KdPrint2((PRINT_PREFIX " No check (administrative)\n"));
+                if(chan->Force80pin) {
+                    KdPrint2((PRINT_PREFIX "Force80pin\n"));
+                }
+            } else
             if(reg54 == 0x0000 || reg54 == 0xffff) {
                 KdPrint2((PRINT_PREFIX " check failed (not supported)\n"));
             } else
@@ -2124,13 +2213,29 @@ AtapiChipInit(
                 AtapiStallExecution(10);
                 KdPrint2((PRINT_PREFIX "BaseIoAddressSATA_0=%x\n", deviceExtension->BaseIoAddressSATA_0.Addr));
                 if(ChipFlags & NVQ) {
+                    KdPrint2((PRINT_PREFIX "Disable NCQ\n"));
+                    tmp32 = AtapiReadPortEx4(NULL, (ULONGIO_PTR)(&deviceExtension->BaseIoAddressSATA_0),0x0400);
+                    KdPrint2((PRINT_PREFIX "MODE=%#x\n", tmp32));
+                    if(tmp32 & ~0xfffffff9) {
+                        AtapiWritePortEx4(NULL, (ULONGIO_PTR)(&deviceExtension->BaseIoAddressSATA_0),0x0400, 
+                             tmp32 & 0xfffffff9);
+                    }
+                    ChipFlags &= ~NVQ;
+                    deviceExtension->HwFlags = ChipFlags;
+                }
+                if(ChipFlags & NVQ) {
+                    /* disable  ECO 398 */
+                    ChangePciConfig1(0x7f, (a & ~(1 << 7)));
+
+                    KdPrint2((PRINT_PREFIX "Enable NCQ\n"));
+                    /* enable NCQ support */
+                    AtapiWritePortEx4(NULL, (ULONGIO_PTR)(&deviceExtension->BaseIoAddressSATA_0),0x0400, 
+                         tmp32 | ~0x00000006);
+
                     /* clear interrupt status */
                     AtapiWritePortEx4(NULL, (ULONGIO_PTR)(&deviceExtension->BaseIoAddressSATA_0),offs, 0x00ff00ff);
                     /* enable device and PHY state change interrupts */
                     AtapiWritePortEx4(NULL, (ULONGIO_PTR)(&deviceExtension->BaseIoAddressSATA_0),offs+4, 0x000d000d);
-                    /* disable NCQ support */
-                    AtapiWritePortEx4(NULL, (ULONGIO_PTR)(&deviceExtension->BaseIoAddressSATA_0),0x0400, 
-                        AtapiReadPortEx4(NULL, (ULONGIO_PTR)(&deviceExtension->BaseIoAddressSATA_0),0x0400) & 0xfffffff9);
                 } else {
                     /* clear interrupt status */
                     AtapiWritePortEx1(NULL, (ULONGIO_PTR)(&deviceExtension->BaseIoAddressSATA_0),offs, 0xff);
@@ -2394,7 +2499,6 @@ AtapiChipInit(
             // do nothing for SATA
         } else
         if(ChipType == SIS133NEW) {
-            USHORT tmp16;
             // check 80-pin cable
             if(c == CHAN_NOT_SPECIFIED) {
                 // do nothing
@@ -2529,6 +2633,98 @@ AtapiChipInit(
             }
         } else
         if(ChipType == ITE_133_NEW) {
+        }
+        break;
+    case ATA_CYRIX_ID:
+        KdPrint2((PRINT_PREFIX "Cyrix\n"));
+        if(ChipType == CYRIX_OLD) {
+            if(c == CHAN_NOT_SPECIFIED) {
+                GetPciConfig1(0x60, tmp8);
+                if(!(tmp8 & 0x40)) {
+                    KdPrint2((PRINT_PREFIX "Enable DMA\n"));
+                    tmp8 |= 0x40;
+                    SetPciConfig1(0x60, tmp8);
+                }
+            }
+        }
+        break;
+    case ATA_JMICRON_ID:
+        /* New JMicron PATA controllers */
+        if(deviceExtension->DevID == ATA_JMB361 ||
+           deviceExtension->DevID == ATA_JMB363 ||
+           deviceExtension->DevID == ATA_JMB365 ||
+           deviceExtension->DevID == ATA_JMB366 ||
+           deviceExtension->DevID == ATA_JMB368) { 
+            KdPrint2((PRINT_PREFIX "JMicron\n"));
+
+            ULONG c_swp = 0;
+            ULONG reg40, reg80;
+
+            GetPciConfig4(0x40, reg40);
+            KdPrint2((PRINT_PREFIX "reg 40: %x\n", reg40));
+
+            c_swp = (reg40 & (1<<22)) ? 1 : 0; // 1=swap, 0=keep
+            KdPrint2((PRINT_PREFIX "c_swp: %x\n", c_swp));
+
+            GetPciConfig4(0x80, reg80);
+            KdPrint2((PRINT_PREFIX "reg 80: %x\n", reg80));
+
+            if(c == CHAN_NOT_SPECIFIED) {
+                UCHAR P1mode;
+
+                P1mode = (reg80 & (1<<24)) ? ATA_UDMA6 : ATA_SA300;
+                KdPrint2((PRINT_PREFIX "p1 mode: %x\n", P1mode));
+
+                if(reg40 & (1 << 23)) {
+                    KdPrint2((PRINT_PREFIX "SATA+PATA0\n"));
+                    deviceExtension->chan[0 ^ c_swp].MaxTransferMode = P1mode;
+                    deviceExtension->chan[1 ^ c_swp].MaxTransferMode = ATA_UDMA6;
+                    deviceExtension->chan[1 ^ c_swp].ChannelCtrlFlags |= CTRFLAGS_NO_SLAVE;
+
+                } else {
+                    KdPrint2((PRINT_PREFIX "SATA+SATA\n"));
+                    deviceExtension->chan[0 ^ c_swp].MaxTransferMode = P1mode;
+                    //deviceExtension->chan[0 ^ c_swp].ChannelCtrlFlags |= CTRFLAGS_NO_SLAVE;
+                    deviceExtension->chan[1 ^ c_swp].MaxTransferMode = ATA_SA300;
+                    deviceExtension->chan[1 ^ c_swp].ChannelCtrlFlags |= CTRFLAGS_NO_SLAVE;
+                }
+
+            } else {
+              /*
+                deviceExtension->chan[0 ^ c_swp].lun[0]->SATA_lun_map =
+                deviceExtension->chan[0 ^ c_swp].lun[0]->SATA_lun_map = 0;
+                deviceExtension->chan[1 ^ c_swp].lun[0]->SATA_lun_map =
+                deviceExtension->chan[1 ^ c_swp].lun[0]->SATA_lun_map = 1;
+                */
+                KdPrint2((PRINT_PREFIX "chan %d\n", c));
+                chan = &deviceExtension->chan[c];
+
+                UCHAR ph_channel = (UCHAR)(c ^ c_swp);
+                //c_swp = chan->lun[0]->SATA_lun_map;
+                if(chan->MaxTransferMode >= ATA_SA150) {
+                    KdPrint2((PRINT_PREFIX "SATA, map -> %x\n", ph_channel));
+                } else {
+                    KdPrint2((PRINT_PREFIX "PATA, map -> %x\n", ph_channel));
+                    if(!ph_channel) {
+                        if(!(reg40 & (1<<5))) {
+                            KdPrint2((PRINT_PREFIX "disabled\n", ph_channel));
+                        } else
+                        if(!(reg40 & (1<<3))) {
+                            KdPrint2((PRINT_PREFIX "40-pin\n"));
+                            chan->MaxTransferMode = min(deviceExtension->MaxTransferMode, ATA_UDMA2);
+                        }
+                    } else {
+                        if(!(reg80 & (1<<21))) {
+                            KdPrint2((PRINT_PREFIX "disabled\n", ph_channel));
+                        } else
+                        if(!(reg80 & (1<<19))) {
+                            KdPrint2((PRINT_PREFIX "40-pin\n"));
+                            chan->MaxTransferMode = min(deviceExtension->MaxTransferMode, ATA_UDMA2);
+                        }
+                    }
+                }
+            }
+
         }
         break;
     default:
@@ -2544,7 +2740,8 @@ AtapiChipInit(
 
     // In all places separate channels are inited after common controller init
     // The only exception is probe. But there we may need info about 40/80 pin and MaxTransferRate
-    if(CheckCable && !(ChipFlags & (UNIATA_NO80CHK | UNIATA_SATA))) {
+    // Do not check UNIATA_SATA here since we may have controller with mixed ports
+    if(CheckCable && !(ChipFlags & (UNIATA_NO80CHK/* | UNIATA_SATA*/))) {
         for(c=0; c<deviceExtension->NumberChannels; c++) {
             AtapiChipInit(HwDeviceExtension, DeviceNumber, c);
         }
@@ -2571,8 +2768,7 @@ UniataInitMapBM(
     for(c=0; c<deviceExtension->NumberChannels; c++) {
         chan = &deviceExtension->chan[c];
         for (i=0; i<IDX_BM_IO_SZ; i++) {
-            chan->RegTranslation[IDX_BM_IO+i].Addr  = BaseIoAddressBM_0 ? ((ULONGIO_PTR)BaseIoAddressBM_0 + i) : 0;
-            chan->RegTranslation[IDX_BM_IO+i].MemIo = MemIo;
+            UniataInitIoRes(chan, IDX_BM_IO+i, BaseIoAddressBM_0 ? ((ULONGIO_PTR)BaseIoAddressBM_0 + i) : 0, MemIo, FALSE);
         }
         if(BaseIoAddressBM_0) {
             BaseIoAddressBM_0++;
@@ -2592,12 +2788,10 @@ UniataInitMapBase(
     ULONG i;
 
     for (i=0; i<IDX_IO1_SZ; i++) {
-        chan->RegTranslation[IDX_IO1+i].Addr = BaseIoAddress1 ? ((ULONGIO_PTR)BaseIoAddress1 + i) : 0;
-        chan->RegTranslation[IDX_IO1+i].MemIo = FALSE;
+        UniataInitIoRes(chan, IDX_IO1+i, BaseIoAddress1 ? ((ULONGIO_PTR)BaseIoAddress1 + i) : 0, FALSE, FALSE);
     }
     for (i=0; i<IDX_IO2_SZ; i++) {
-        chan->RegTranslation[IDX_IO2+i].Addr = BaseIoAddress2 ? ((ULONGIO_PTR)BaseIoAddress2 + i) : 0;
-        chan->RegTranslation[IDX_IO2+i].MemIo = FALSE;
+        UniataInitIoRes(chan, IDX_IO2+i, BaseIoAddress2 ? ((ULONGIO_PTR)BaseIoAddress2 + i) : 0, FALSE, FALSE);
     }
     UniataInitSyncBaseIO(chan);
     return;
@@ -2614,6 +2808,39 @@ UniataInitSyncBaseIO(
     return;
 } // end UniataInitSyncBaseIO()
 
+VOID
+UniataInitIoRes(
+    IN PHW_CHANNEL chan,
+    IN ULONG idx,
+    IN ULONG addr,
+    IN BOOLEAN MemIo,
+    IN BOOLEAN Proc
+    )
+{
+    if(!addr) {
+        MemIo = Proc = FALSE;
+    }
+    chan->RegTranslation[idx].Addr  = addr;
+    chan->RegTranslation[idx].MemIo = MemIo;
+    chan->RegTranslation[idx].Proc  = Proc;
+} // end UniataInitIoRes()
+
+VOID
+UniataInitIoResEx(
+    IN PIORES IoRes,
+    IN ULONG addr,
+    IN BOOLEAN MemIo,
+    IN BOOLEAN Proc
+    )
+{
+    if(!addr) {
+        MemIo = Proc = FALSE;
+    }
+    IoRes->Addr  = addr;
+    IoRes->MemIo = MemIo;
+    IoRes->Proc  = Proc;
+} // end UniataInitIoResEx()
+
 VOID
 NTAPI
 AtapiSetupLunPtrs(
@@ -2641,6 +2868,7 @@ AtapiSetupLunPtrs(
     }
     chan->AltRegMap       = deviceExtension->AltRegMap;
     chan->NextDpcChan     = -1;
+    chan->last_devsel     = -1;
     for(i=0; i<deviceExtension->NumberLuns; i++) {
         chan->lun[i]->DeviceExtension = deviceExtension;
         chan->lun[i]->chan            = chan;