[HAL]

- Return the correct value if the PCI slot number is invalid
- Use the bus number from the PCI BIOS instead of doing a manual probe if we can because it is much more accurate (our probing code doesn't detect buses without devices present)
- Don't probe for devices at invalid locations on PCI type 2 buses
- Check for a valid bus number so we don't return the wrong value

svn path=/trunk/; revision=47162

reactos/hal/halx86/generic/bus/pcibus.c

@@ -360,15 +360,19 @@ HalpGetPCIData(IN PBUS_HANDLER BusHandler,
         (1 == BusHandler->BusNumber && 0 != Slot.u.bits.DeviceNumber))
     {
         DPRINT("Blacklisted PCI slot\n");
-        if (0 == Offset && 2 <= Length)
+        if (0 == Offset && sizeof(USHORT) <= Length)
         {
             *(PUSHORT)Buffer = PCI_INVALID_VENDORID;
-            return 2;
+            return sizeof(USHORT);
         }
         return 0;
     }
 #endif
 
+    /* Make sure the bus number is in our range of good bus numbers */
+    if (BusHandler->BusNumber > HalpMaxPciBus || BusHandler->BusNumber < HalpMinPciBus)
+        return 0;
+
     /* Normalize the length */
     if (Length > sizeof(PCI_COMMON_CONFIG)) Length = sizeof(PCI_COMMON_CONFIG);
 
@@ -390,9 +394,15 @@ HalpGetPCIData(IN PBUS_HANDLER BusHandler,
         /* Validate the vendor ID */
         if (PciConfig->VendorID == PCI_INVALID_VENDORID)
         {
-            /* It's invalid, but we want to return this much */
-            PciConfig->VendorID = PCI_INVALID_VENDORID;
-            Len = sizeof(USHORT);
+            /* It's invalid, but we can copy PCI_INVALID_VENDORID */
+            if (Offset == 0 && Length >= sizeof(USHORT))
+            {
+                *(PUSHORT)Buffer = PCI_INVALID_VENDORID;
+                return sizeof(USHORT);
+            }
+
+            /* We can't copy PCI_INVALID_VENDORID so just return 0 */
+            return 0;
         }
 
         /* Now check if there's space left */
@@ -455,6 +465,10 @@ HalpSetPCIData(IN PBUS_HANDLER BusHandler,
     }
 #endif
 
+    /* Make sure this bus number is in our range of good bus numbers */
+    if (BusHandler->BusNumber > HalpMaxPciBus || BusHandler->BusNumber < HalpMinPciBus)
+        return 0;
+
     /* Normalize the length */
     if (Length > sizeof(PCI_COMMON_CONFIG)) Length = sizeof(PCI_COMMON_CONFIG);
 
@@ -700,7 +714,6 @@ HaliPciInterfaceReadConfig(IN PBUS_HANDLER RootBusHandler,
                            IN ULONG Length)
 {
     BUS_HANDLER BusHandler;
-    PPCI_COMMON_CONFIG PciData = (PPCI_COMMON_CONFIG)Buffer;
 
     /* Setup fake PCI Bus handler */
     RtlCopyMemory(&BusHandler, &HalpFakePciBusHandler, sizeof(BUS_HANDLER));
@@ -709,21 +722,6 @@ HaliPciInterfaceReadConfig(IN PBUS_HANDLER RootBusHandler,
     /* Read configuration data */
     HalpReadPCIConfig(&BusHandler, SlotNumber, Buffer, Offset, Length);
 
-    /* Check if caller only wanted at least Vendor ID */
-    if (Length >= 2)
-    {
-        /* Validate it */
-        if (PciData->VendorID != PCI_INVALID_VENDORID)
-        {
-            /* Check if this is the new maximum bus number */
-            if (HalpMaxPciBus < BusHandler.BusNumber)
-            {
-                /* Set it */
-                HalpMaxPciBus = BusHandler.BusNumber;
-            }
-        }
-    }
-
     /* Return length */
     return Length;
 }
@@ -970,6 +968,7 @@ HalpInitializePciStubs(VOID)
     ULONG i;
     PCI_SLOT_NUMBER j;
     ULONG VendorId = 0;
+    ULONG MaxPciBusNumber;
 
     /* Query registry information */
     PciRegistryInfo = HalpQueryPciRegistryInfo();
@@ -977,11 +976,19 @@ HalpInitializePciStubs(VOID)
     {
         /* Assume type 1 */
         PciType = 1;
+
+        /* Force a manual bus scan later */
+        MaxPciBusNumber = MAXULONG;
     }
     else
     {
-        /* Get the type and free the info structure */
+        /* Get the PCI type */
         PciType = PciRegistryInfo->HardwareMechanism & 0xF;
+
+        /* Get MaxPciBusNumber and make it 0-based */
+        MaxPciBusNumber = PciRegistryInfo->NoBuses - 1;
+
+        /* Free the info structure */
         ExFreePool(PciRegistryInfo);
     }
 
@@ -1007,7 +1014,7 @@ HalpInitializePciStubs(VOID)
         /* Type 2 PCI Bus */
         case 2:
 
-            /* Copy the Type 1 handler data */
+            /* Copy the Type 2 handler data */
             RtlCopyMemory(&PCIConfigHandler,
                           &PCIConfigHandlerType2,
                           sizeof (PCIConfigHandler));
@@ -1027,31 +1034,44 @@ HalpInitializePciStubs(VOID)
             DbgPrint("HAL: Unknown PCI type\n");
     }
 
-    /* Loop all possible buses */
-    for (i = 0; i < 256; i++)
+    /* Run a forced bus scan if needed */
+    if (MaxPciBusNumber == MAXULONG)
     {
-        /* Loop all devices */
-        for (j.u.AsULONG = 0; j.u.AsULONG < 32; j.u.AsULONG++)
+        /* Initialize the max bus number to 0xFF */
+        HalpMaxPciBus = 0xFF;
+
+        /* Initialize the counter */
+        MaxPciBusNumber = 0;
+
+        /* Loop all possible buses */
+        for (i = 0; i < HalpMaxPciBus; i++)
         {
-            /* Query the interface */
-            if (HaliPciInterfaceReadConfig(NULL,
-                                           i,
-                                           j,
-                                           &VendorId,
-                                           0,
-                                           sizeof(ULONG)))
+            /* Loop all devices */
+            for (j.u.AsULONG = 0; j.u.AsULONG < BusData->MaxDevice; j.u.AsULONG++)
             {
-                /* Validate the vendor ID */
-                if ((USHORT)VendorId != PCI_INVALID_VENDORID)
+                /* Query the interface */
+                if (HaliPciInterfaceReadConfig(NULL,
+                                               i,
+                                               j,
+                                               &VendorId,
+                                               0,
+                                               sizeof(ULONG)))
                 {
-                    /* Set this as the maximum ID */
-                    HalpMaxPciBus = i;
-                    break;
+                    /* Validate the vendor ID */
+                    if ((USHORT)VendorId != PCI_INVALID_VENDORID)
+                    {
+                        /* Set this as the maximum ID */
+                        MaxPciBusNumber = i;
+                        break;
+                    }
                 }
             }
         }
     }
 
+    /* Set the real max bus number */
+    HalpMaxPciBus = MaxPciBusNumber;
+
     /* We're done */
     HalpPCIConfigInitialized = TRUE;
 }