initial commit

EDK2/edk2-edk2-stable201911/MdeModulePkg/Application/Ventoy/Ventoy.h

@@ -0,0 +1,233 @@
+/******************************************************************************
+ * Ventoy.h
+ *
+ * Copyright (c) 2020, longpanda <admin@ventoy.net>
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License as
+ * published by the Free Software Foundation; either version 3 of the
+ * License, or (at your option) any later version.
+ * 
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * General Public License for more details.
+ * 
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, see <http://www.gnu.org/licenses/>.
+ *
+ */
+ 
+#ifndef __VENTOY_H__
+#define __VENTOY_H__
+
+#define COMPILE_ASSERT(expr)  extern char __compile_assert[(expr) ? 1 : -1]
+
+#define VENTOY_GUID { 0x77772020, 0x2e77, 0x6576, { 0x6e, 0x74, 0x6f, 0x79, 0x2e, 0x6e, 0x65, 0x74 }}
+
+#pragma pack(1)
+
+typedef struct ventoy_guid
+{
+    UINT32   data1;
+    UINT16   data2;
+    UINT16   data3;
+    UINT8    data4[8];
+}ventoy_guid;
+
+typedef struct ventoy_image_disk_region
+{
+    UINT32   image_sector_count; /* image sectors contained in this region */
+    UINT32   image_start_sector; /* image sector start */
+    UINT64   disk_start_sector;  /* disk sector start */
+}ventoy_image_disk_region;
+
+typedef struct ventoy_image_location
+{
+    ventoy_guid  guid;
+    
+    /* image sector size, currently this value is always 2048 */
+    UINT32   image_sector_size;
+
+    /* disk sector size, normally the value is 512 */
+    UINT32   disk_sector_size;
+
+    UINT32   region_count;
+    
+    /*
+     * disk region data
+     * If the image file has more than one fragments in disk, 
+     * there will be more than one region data here.
+     *
+     */
+    ventoy_image_disk_region regions[1];
+
+    /* ventoy_image_disk_region regions[2~region_count-1] */
+}ventoy_image_location;
+
+typedef struct ventoy_os_param
+{
+    ventoy_guid    guid;                  // VENTOY_GUID
+    UINT8   chksum;                // checksum
+
+    UINT8   vtoy_disk_guid[16];
+    UINT64  vtoy_disk_size;       // disk size in bytes
+    UINT16  vtoy_disk_part_id;    // begin with 1
+    UINT16  vtoy_disk_part_type;  // 0:exfat   1:ntfs  other: reserved
+    char    vtoy_img_path[384];   // It seems to be enough, utf-8 format
+    UINT64  vtoy_img_size;        // image file size in bytes
+
+    /* 
+     * Ventoy will write a copy of ventoy_image_location data into runtime memory
+     * this is the physically address and length of that memory.
+     * Address 0 means no such data exist.
+     * Address will be aligned by 4KB.
+     *
+     */
+    UINT64  vtoy_img_location_addr;
+    UINT32  vtoy_img_location_len;
+    
+    UINT64  vtoy_reserved[4];     // Internal use by ventoy
+
+    UINT8   reserved[31];
+}ventoy_os_param;
+
+#pragma pack()
+
+// compile assert to check that size of ventoy_os_param must be 512
+COMPILE_ASSERT(sizeof(ventoy_os_param) == 512);
+
+
+
+#pragma pack(4)
+
+typedef struct ventoy_chain_head
+{
+    ventoy_os_param os_param;
+
+    UINT32 disk_drive;
+    UINT32 drive_map;
+    UINT32 disk_sector_size;
+
+    UINT64 real_img_size_in_bytes;
+    UINT64 virt_img_size_in_bytes;
+    UINT32 boot_catalog;
+    UINT8  boot_catalog_sector[2048];
+    
+    UINT32 img_chunk_offset;
+    UINT32 img_chunk_num;
+
+    UINT32 override_chunk_offset;
+    UINT32 override_chunk_num;
+
+    UINT32 virt_chunk_offset;
+    UINT32 virt_chunk_num;
+}ventoy_chain_head;
+
+
+typedef struct ventoy_img_chunk
+{
+    UINT32 img_start_sector; //2KB
+    UINT32 img_end_sector;
+
+    UINT64 disk_start_sector; // in disk_sector_size
+    UINT64 disk_end_sector;
+}ventoy_img_chunk;
+
+
+typedef struct ventoy_override_chunk
+{
+    UINT64 img_offset;
+    UINT32 override_size;
+    UINT8  override_data[512];
+}ventoy_override_chunk;
+
+typedef struct ventoy_virt_chunk
+{
+    UINT32 mem_sector_start;
+    UINT32 mem_sector_end;
+    UINT32 mem_sector_offset;
+    UINT32 remap_sector_start;
+    UINT32 remap_sector_end;
+    UINT32 org_sector_start;
+}ventoy_virt_chunk;
+
+
+#pragma pack()
+
+
+#define VTOY_BLOCK_DEVICE_PATH_GUID					\
+	{ 0x37b87ac6, 0xc180, 0x4583, { 0xa7, 0x05, 0x41, 0x4d, 0xa8, 0xf7, 0x7e, 0xd2 }}
+
+#define VTOY_BLOCK_DEVICE_PATH_NAME  L"ventoy"
+
+#if   defined (MDE_CPU_IA32)
+  #define VENTOY_UEFI_DESC   L"IA32 UEFI"
+#elif defined (MDE_CPU_X64)
+  #define VENTOY_UEFI_DESC   L"X64 UEFI"
+#elif defined (MDE_CPU_EBC)
+#elif defined (MDE_CPU_ARM)
+  #define VENTOY_UEFI_DESC   L"ARM UEFI"
+#elif defined (MDE_CPU_AARCH64)
+  #define VENTOY_UEFI_DESC   L"ARM64 UEFI"
+#else
+  #error Unknown Processor Type
+#endif
+
+typedef struct ventoy_sector_flag
+{
+    UINT8 flag; // 0:init   1:mem  2:remap
+    UINT64 remap_lba;    
+}ventoy_sector_flag;
+
+
+typedef struct vtoy_block_data 
+{
+	EFI_HANDLE Handle;
+	EFI_BLOCK_IO_MEDIA Media;       /* Media descriptor */
+	EFI_BLOCK_IO_PROTOCOL BlockIo;	/* Block I/O protocol */
+
+    UINTN DevicePathCompareLen;
+	EFI_DEVICE_PATH_PROTOCOL *Path;	/* Device path protocol */
+
+    EFI_HANDLE RawBlockIoHandle;
+    EFI_BLOCK_IO_PROTOCOL *pRawBlockIo;
+    EFI_DEVICE_PATH_PROTOCOL *pDiskDevPath;
+
+    /* ventoy disk part2 ESP */
+    EFI_HANDLE DiskFsHandle;
+    EFI_SIMPLE_FILE_SYSTEM_PROTOCOL *pDiskFs;
+    EFI_DEVICE_PATH_PROTOCOL *pDiskFsDevPath;
+
+    EFI_HANDLE IsoDriverImage;
+}vtoy_block_data;
+
+#define ISO9660_EFI_DRIVER_PATH  L"\\ventoy\\iso9660_x64.efi"
+
+#define debug(expr, ...) if (gDebugPrint) VtoyDebug("[VTOY] "expr"\r\n", ##__VA_ARGS__)
+#define sleep(sec) gBS->Stall(1000000 * (sec))
+
+#define ventoy_debug_pause() \
+if (gDebugPrint) \
+{ \
+    UINTN __Index = 0; \
+    gST->ConOut->OutputString(gST->ConOut, L"[VTOY] ###### Press Enter to continue... ######\r\n");\
+    gST->ConIn->Reset(gST->ConIn, FALSE); \
+    gBS->WaitForEvent(1, &gST->ConIn->WaitForKey, &__Index);\
+}
+
+typedef const char * (*grub_env_get_pf)(const char *name);
+
+#pragma pack(1)
+typedef struct ventoy_grub_param
+{
+    grub_env_get_pf grub_env_get;
+}ventoy_grub_param;
+#pragma pack()
+
+
+extern BOOLEAN gDebugPrint;
+VOID EFIAPI VtoyDebug(IN CONST CHAR8  *Format, ...);
+
+#endif
+

EDK2/edk2-edk2-stable201911/MdeModulePkg/Application/Ventoy/Ventoy.inf

@@ -0,0 +1,49 @@
+#************************************************************************************
+# Copyright (c) 2020, longpanda <admin@ventoy.net>
+# 
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of the GNU General Public License as
+# published by the Free Software Foundation; either version 3 of the
+# License, or (at your option) any later version.
+# 
+# This program is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# General Public License for more details.
+# 
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, see <http://www.gnu.org/licenses/>.
+# 
+#************************************************************************************
+
+[Defines]
+  INF_VERSION                    = 0x00010005
+  BASE_NAME                      = Ventoy
+  FILE_GUID                      = 1c3a0915-09dc-49c2-873d-0aaaa7733299
+  MODULE_TYPE                    = UEFI_APPLICATION
+  VERSION_STRING                 = 1.0
+  ENTRY_POINT                    = VentoyEfiMain
+
+
+[Sources]
+  Ventoy.h
+  Ventoy.c
+
+[Packages]
+  MdePkg/MdePkg.dec
+  MdeModulePkg/MdeModulePkg.dec
+  ShellPkg/ShellPkg.dec
+
+[LibraryClasses]
+  UefiApplicationEntryPoint
+  UefiLib
+  DebugLib
+
+[Guids]
+  gShellVariableGuid
+  
+[Protocols]
+  gEfiLoadedImageProtocolGuid
+  gEfiBlockIoProtocolGuid
+  gEfiDevicePathProtocolGuid
+  gEfiSimpleFileSystemProtocolGuid