X-Git-Url: https://glassweightruler.freedombox.rocks/gitweb/Ventoy.git/blobdiff_plain/c72f96312c1caabafff6f424aac70493d99faa38..790fa744f85c7dff5fe6f1905cd4268980c60dda:/GRUB2/grub-2.04/grub-core/normal/menu.c

diff --git a/GRUB2/grub-2.04/grub-core/normal/menu.c b/GRUB2/grub-2.04/grub-core/normal/menu.c
index a523181..4df03bf 100644
--- a/GRUB2/grub-2.04/grub-core/normal/menu.c
+++ b/GRUB2/grub-2.04/grub-core/normal/menu.c
@@ -32,6 +32,7 @@
 #include <grub/script_sh.h>
 #include <grub/gfxterm.h>
 #include <grub/dl.h>
+#include <grub/env.h>
 
 /* Time to delay after displaying an error message about a default/fallback
    entry failing to boot.  */
@@ -786,6 +787,36 @@ run_menu (grub_menu_t menu, int nested, int *auto_boot)
 		}
 	      goto refresh;
 
+        case GRUB_TERM_KEY_F1:
+            menu_fini ();
+            if (grub_env_get("VTOY_MEM_DISK")) {
+                grub_env_unset("VTOY_MEM_DISK");
+            }else {
+                grub_env_set("VTOY_MEM_DISK", grub_env_get("VTOY_MEM_DISK_STR"));
+            }
+            grub_env_set("VTOY_MENU_REFRESH", "1");
+            goto refresh;
+            
+        case GRUB_TERM_KEY_F3:
+            menu_fini ();
+            if (grub_env_get("VTOY_ISO_RAW")) {
+                grub_env_unset("VTOY_ISO_RAW");
+            }else {
+                grub_env_set("VTOY_ISO_RAW", grub_env_get("VTOY_ISO_RAW_STR"));
+            }
+            grub_env_set("VTOY_MENU_REFRESH", "1");
+            goto refresh;
+            
+        case GRUB_TERM_KEY_F4:
+            menu_fini ();
+            if (grub_env_get("VTOY_ISO_UEFI_DRV")) {
+                grub_env_unset("VTOY_ISO_UEFI_DRV");
+            }else {
+                grub_env_set("VTOY_ISO_UEFI_DRV", grub_env_get("VTOY_ISO_UEFI_DRV_STR"));
+            }
+            grub_env_set("VTOY_MENU_REFRESH", "1");
+            goto refresh;
+
 	    default:
 	      {
 		int entry;