X-Git-Url: https://glassweightruler.freedombox.rocks/gitweb/Ventoy.git/blobdiff_plain/fe0dda690426b8e1b73c79b0dff46e1459ea40ec..84c500666addc8378fb03c3844bf6d25faf54a87:/GRUB2/grub-2.04/grub-core/ventoy/ventoy_windows.c

diff --git a/GRUB2/grub-2.04/grub-core/ventoy/ventoy_windows.c b/GRUB2/grub-2.04/grub-core/ventoy/ventoy_windows.c
index e2b4eed..529d308 100644
--- a/GRUB2/grub-2.04/grub-core/ventoy/ventoy_windows.c
+++ b/GRUB2/grub-2.04/grub-core/ventoy/ventoy_windows.c
@@ -401,7 +401,7 @@ static int ventoy_update_all_hash(void *meta_data, wim_directory_entry *dir)
         return 0;
     }
 
-    if (dir->len == 0)
+    if (dir->len < sizeof(wim_directory_entry))
     {
         return 0;
     }
@@ -420,7 +420,7 @@ static int ventoy_update_all_hash(void *meta_data, wim_directory_entry *dir)
         }
     
         dir = (wim_directory_entry *)((char *)dir + dir->len);
-    } while (dir->len);
+    } while (dir->len >= sizeof(wim_directory_entry));
 
     return 0;
 }
@@ -957,4 +957,3 @@ grub_err_t ventoy_cmd_windows_chain_data(grub_extcmd_context_t ctxt, int argc, c
     VENTOY_CMD_RETURN(GRUB_ERR_NONE);
 }
 
-