[FREELDR] Improve disk detection

Author: iLauncherDev

boot/freeldr/freeldr/arch/uefi/uefidisk.c

@@ -431,30 +431,24 @@ UefiSetupBlockDevices(VOID)
         }
         else if (handles[i] == PublicBootHandle)
         {
-            ULONG increment = 0;
+            ULONG increment = i;
             ULONG i;
 
-            /* 3) Grab the offset into the array of handles and decrement per volume (valid partition) */
-            for (increment = OffsetToBoot; increment > 0; increment--)
+            GlobalSystemTable->BootServices->HandleProtocol(handles[increment], &bioGuid, (void**)&bio);
+            if (bio->Media->LogicalPartition == FALSE)
             {
-                GlobalSystemTable->BootServices->HandleProtocol(handles[increment], &bioGuid, (void**)&bio);
-                if (bio->Media->LogicalPartition == FALSE)
-                {
-                    TRACE("Found root at increment %u\n", increment);
-                    UefiBootRootIdentifier = increment;
+                TRACE("Found root at increment %u\n", increment);
+                UefiBootRootIdentifier = increment;
 
-                    for (i = 0; i <= PcBiosDiskCount; ++i)
+                for (i = 0; i <= PcBiosDiskCount; ++i)
+                {
+                    /* Now only of the root drive number is equal to this drive we found above */
+                    if (InternalUefiDisk[i].UefiRootNumber == UefiBootRootIdentifier)
                     {
-                        /* Now only of the root drive number is equal to this drive we found above */
-                        if (InternalUefiDisk[i].UefiRootNumber == UefiBootRootIdentifier)
-                        {
-                            InternalUefiDisk[i].IsThisTheBootDrive = TRUE;
-                            PublicBootArcDisk = i;
-                            TRACE("Found Boot drive\n");
-                        }
+                        InternalUefiDisk[i].IsThisTheBootDrive = TRUE;
+                        PublicBootArcDisk = i;
+                        TRACE("Found Boot drive\n");
                     }
-
-                    break;
                 }
             }
         }