From d05eab2be16d355b950a8790dd0abb65c3a1d3e9 Mon Sep 17 00:00:00 2001
From: longpanda <admin@ventoy.net>
Date: Sat, 8 May 2021 15:33:11 +0800
Subject: [PATCH] support boot Windows VHD(x) at local disk

---
 .../grub-2.04/grub-core/ventoy/ventoy_vhd.c   | 36 +++++++++++++++++--
 1 file changed, 34 insertions(+), 2 deletions(-)

diff --git a/GRUB2/MOD_SRC/grub-2.04/grub-core/ventoy/ventoy_vhd.c b/GRUB2/MOD_SRC/grub-2.04/grub-core/ventoy/ventoy_vhd.c
index a5203e2f..502a248b 100644
--- a/GRUB2/MOD_SRC/grub-2.04/grub-core/ventoy/ventoy_vhd.c
+++ b/GRUB2/MOD_SRC/grub-2.04/grub-core/ventoy/ventoy_vhd.c
@@ -150,9 +150,12 @@ static int ventoy_vhd_patch_path(char *vhdpath, ventoy_patch_vhd *patch1, ventoy
 
 static int ventoy_vhd_read_parttbl(const char *filename, ventoy_gpt_info *gpt, int *index)
 {
+    int i;
     int ret = 1;
+    grub_uint64_t start;
     grub_file_t file = NULL;
     grub_disk_t disk = NULL;
+    grub_uint8_t zeroguid[16] = {0};
 
     file = grub_file_open(filename, VENTOY_FILE_TYPE);
     if (!file)
@@ -165,10 +168,39 @@ static int ventoy_vhd_read_parttbl(const char *filename, ventoy_gpt_info *gpt, i
     {
         goto end;
     }
-    
-    *index = file->device->disk->partition->index;
+
     grub_disk_read(disk, 0, 0, sizeof(ventoy_gpt_info), gpt);
 
+    start = file->device->disk->partition->start;
+
+    if (grub_memcmp(gpt->Head.Signature, "EFI PART", 8) == 0)
+    {
+        debug("GPT part start: %llu\n", (ulonglong)start);
+        for (i = 0; i < 128; i++)
+        {
+            if (grub_memcmp(gpt->PartTbl[i].PartGuid, zeroguid, 16))
+            {
+                if (start == gpt->PartTbl[i].StartLBA)
+                {
+                    *index = i;
+                    break;
+                }
+            }
+        }
+    }
+    else
+    {
+        debug("MBR part start: %llu\n", (ulonglong)start);
+        for (i = 0; i < 4; i++)
+        {
+            if ((grub_uint32_t)start == gpt->MBR.PartTbl[i].StartSectorId)
+            {
+                *index = i;
+                break;
+            }
+        }
+    }    
+
     ret = 0;
 
 end: