mirror of https://github.com/ventoy/Ventoy.git
308 lines
8.5 KiB
C
308 lines
8.5 KiB
C
/*
|
|
* Copyright (C) 2014 Michael Brown <mbrown@fensystems.co.uk>.
|
|
*
|
|
* 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 2 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, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
|
|
* 02110-1301, USA.
|
|
*/
|
|
|
|
/**
|
|
* @file
|
|
*
|
|
* EFI block device
|
|
*
|
|
*/
|
|
|
|
#include <stddef.h>
|
|
#include <stdint.h>
|
|
#include <stdio.h>
|
|
#include "wimboot.h"
|
|
#include "vdisk.h"
|
|
#include "efi.h"
|
|
#include "efipath.h"
|
|
#include "efiblock.h"
|
|
|
|
/** A block I/O device */
|
|
struct efi_block {
|
|
/** EFI block I/O protocol */
|
|
EFI_BLOCK_IO_PROTOCOL block;
|
|
/** Device path */
|
|
EFI_DEVICE_PATH_PROTOCOL *path;
|
|
/** Starting LBA */
|
|
uint64_t lba;
|
|
/** Name */
|
|
const char *name;
|
|
};
|
|
|
|
/**
|
|
* Reset block I/O protocol
|
|
*
|
|
* @v this Block I/O protocol
|
|
* @v extended Perform extended verification
|
|
* @ret efirc EFI status code
|
|
*/
|
|
static EFI_STATUS EFIAPI
|
|
efi_reset_blocks ( EFI_BLOCK_IO_PROTOCOL *this, BOOLEAN extended ) {
|
|
struct efi_block *block =
|
|
container_of ( this, struct efi_block, block );
|
|
void *retaddr = __builtin_return_address ( 0 );
|
|
|
|
DBG2 ( "EFI %s %sreset -> %p\n",
|
|
block->name, ( extended ? "extended " : "" ), retaddr );
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* Read blocks
|
|
*
|
|
* @v this Block I/O protocol
|
|
* @v media Media ID
|
|
* @v lba Starting LBA
|
|
* @v len Length of data
|
|
* @v data Data buffer
|
|
* @ret efirc EFI status code
|
|
*/
|
|
static EFI_STATUS EFIAPI
|
|
efi_read_blocks ( EFI_BLOCK_IO_PROTOCOL *this, UINT32 media, EFI_LBA lba,
|
|
UINTN len, VOID *data ) {
|
|
struct efi_block *block =
|
|
container_of ( this, struct efi_block, block );
|
|
void *retaddr = __builtin_return_address ( 0 );
|
|
|
|
DBG2 ( "EFI %s read media %08x LBA %#llx to %p+%zx -> %p\n",
|
|
block->name, media, lba, data, ( ( size_t ) len ), retaddr );
|
|
vdisk_read ( ( lba + block->lba ), ( len / VDISK_SECTOR_SIZE ), data );
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* Write blocks
|
|
*
|
|
* @v this Block I/O protocol
|
|
* @v media Media ID
|
|
* @v lba Starting LBA
|
|
* @v len Length of data
|
|
* @v data Data buffer
|
|
* @ret efirc EFI status code
|
|
*/
|
|
static EFI_STATUS EFIAPI
|
|
efi_write_blocks ( EFI_BLOCK_IO_PROTOCOL *this __unused,
|
|
UINT32 media __unused, EFI_LBA lba __unused,
|
|
UINTN len __unused, VOID *data __unused ) {
|
|
struct efi_block *block =
|
|
container_of ( this, struct efi_block, block );
|
|
void *retaddr = __builtin_return_address ( 0 );
|
|
|
|
DBG2 ( "EFI %s write media %08x LBA %#llx from %p+%zx -> %p\n",
|
|
block->name, media, lba, data, ( ( size_t ) len ), retaddr );
|
|
return EFI_WRITE_PROTECTED;
|
|
}
|
|
|
|
/**
|
|
* Flush block operations
|
|
*
|
|
* @v this Block I/O protocol
|
|
* @ret efirc EFI status code
|
|
*/
|
|
static EFI_STATUS EFIAPI
|
|
efi_flush_blocks ( EFI_BLOCK_IO_PROTOCOL *this ) {
|
|
struct efi_block *block =
|
|
container_of ( this, struct efi_block, block );
|
|
void *retaddr = __builtin_return_address ( 0 );
|
|
|
|
DBG2 ( "EFI %s flush -> %p\n", block->name, retaddr );
|
|
return 0;
|
|
}
|
|
|
|
/** GUID used in vendor device path */
|
|
#define EFIBLOCK_GUID { \
|
|
0x1322d197, 0x15dc, 0x4a45, \
|
|
{ 0xa6, 0xa4, 0xfa, 0x57, 0x05, 0x4e, 0xa6, 0x14 } \
|
|
}
|
|
|
|
/**
|
|
* Initialise vendor device path
|
|
*
|
|
* @v name Variable name
|
|
*/
|
|
#define EFIBLOCK_DEVPATH_VENDOR_INIT( name ) { \
|
|
.Header = EFI_DEVPATH_INIT ( name, HARDWARE_DEVICE_PATH, \
|
|
HW_VENDOR_DP ), \
|
|
.Guid = EFIBLOCK_GUID, \
|
|
}
|
|
|
|
/**
|
|
* Initialise ATA device path
|
|
*
|
|
* @v name Variable name
|
|
*/
|
|
#define EFIBLOCK_DEVPATH_ATA_INIT( name ) { \
|
|
.Header = EFI_DEVPATH_INIT ( name, MESSAGING_DEVICE_PATH, \
|
|
MSG_ATAPI_DP ), \
|
|
.PrimarySecondary = 0, \
|
|
.SlaveMaster = 0, \
|
|
.Lun = 0, \
|
|
}
|
|
|
|
/**
|
|
* Initialise hard disk device path
|
|
*
|
|
* @v name Variable name
|
|
*/
|
|
#define EFIBLOCK_DEVPATH_HD_INIT( name ) { \
|
|
.Header = EFI_DEVPATH_INIT ( name, MEDIA_DEVICE_PATH, \
|
|
MEDIA_HARDDRIVE_DP ), \
|
|
.PartitionNumber = 1, \
|
|
.PartitionStart = VDISK_PARTITION_LBA, \
|
|
.PartitionSize = VDISK_PARTITION_COUNT, \
|
|
.Signature[0] = ( ( VDISK_MBR_SIGNATURE >> 0 ) & 0xff ), \
|
|
.Signature[1] = ( ( VDISK_MBR_SIGNATURE >> 8 ) & 0xff ), \
|
|
.Signature[2] = ( ( VDISK_MBR_SIGNATURE >> 16 ) & 0xff ), \
|
|
.Signature[3] = ( ( VDISK_MBR_SIGNATURE >> 24 ) & 0xff ), \
|
|
.MBRType = MBR_TYPE_PCAT, \
|
|
.SignatureType = SIGNATURE_TYPE_MBR, \
|
|
}
|
|
|
|
/** Virtual disk media */
|
|
static EFI_BLOCK_IO_MEDIA efi_vdisk_media = {
|
|
.MediaId = VDISK_MBR_SIGNATURE,
|
|
.MediaPresent = TRUE,
|
|
.LogicalPartition = FALSE,
|
|
.ReadOnly = TRUE,
|
|
.BlockSize = VDISK_SECTOR_SIZE,
|
|
.LastBlock = ( VDISK_COUNT - 1 ),
|
|
};
|
|
|
|
/** Virtual disk device path */
|
|
static struct {
|
|
VENDOR_DEVICE_PATH vendor;
|
|
ATAPI_DEVICE_PATH ata;
|
|
EFI_DEVICE_PATH_PROTOCOL end;
|
|
} __attribute__ (( packed )) efi_vdisk_path = {
|
|
.vendor = EFIBLOCK_DEVPATH_VENDOR_INIT ( efi_vdisk_path.vendor ),
|
|
.ata = EFIBLOCK_DEVPATH_ATA_INIT ( efi_vdisk_path.ata ),
|
|
.end = EFI_DEVPATH_END_INIT ( efi_vdisk_path.end ),
|
|
};
|
|
|
|
/** Virtual disk device */
|
|
static struct efi_block efi_vdisk = {
|
|
.block = {
|
|
.Revision = EFI_BLOCK_IO_PROTOCOL_REVISION,
|
|
.Media = &efi_vdisk_media,
|
|
.Reset = efi_reset_blocks,
|
|
.ReadBlocks = efi_read_blocks,
|
|
.WriteBlocks = efi_write_blocks,
|
|
.FlushBlocks = efi_flush_blocks,
|
|
},
|
|
.path = &efi_vdisk_path.vendor.Header,
|
|
.lba = 0,
|
|
.name = "vdisk",
|
|
};
|
|
|
|
/** Virtual partition media */
|
|
static EFI_BLOCK_IO_MEDIA efi_vpartition_media = {
|
|
.MediaId = VDISK_MBR_SIGNATURE,
|
|
.MediaPresent = TRUE,
|
|
.LogicalPartition = TRUE,
|
|
.ReadOnly = TRUE,
|
|
.BlockSize = VDISK_SECTOR_SIZE,
|
|
.LastBlock = ( VDISK_PARTITION_COUNT - 1 ),
|
|
};
|
|
|
|
/** Virtual partition device path */
|
|
static struct {
|
|
VENDOR_DEVICE_PATH vendor;
|
|
ATAPI_DEVICE_PATH ata;
|
|
HARDDRIVE_DEVICE_PATH hd;
|
|
EFI_DEVICE_PATH_PROTOCOL end;
|
|
} __attribute__ (( packed )) efi_vpartition_path = {
|
|
.vendor = EFIBLOCK_DEVPATH_VENDOR_INIT ( efi_vpartition_path.vendor ),
|
|
.ata = EFIBLOCK_DEVPATH_ATA_INIT ( efi_vpartition_path.ata ),
|
|
.hd = EFIBLOCK_DEVPATH_HD_INIT ( efi_vpartition_path.hd ),
|
|
.end = EFI_DEVPATH_END_INIT ( efi_vpartition_path.end ),
|
|
};
|
|
|
|
/** Virtual partition device */
|
|
static struct efi_block efi_vpartition = {
|
|
.block = {
|
|
.Revision = EFI_BLOCK_IO_PROTOCOL_REVISION,
|
|
.Media = &efi_vpartition_media,
|
|
.Reset = efi_reset_blocks,
|
|
.ReadBlocks = efi_read_blocks,
|
|
.WriteBlocks = efi_write_blocks,
|
|
.FlushBlocks = efi_flush_blocks,
|
|
},
|
|
.path = &efi_vpartition_path.vendor.Header,
|
|
.lba = VDISK_PARTITION_LBA,
|
|
.name = "vpartition",
|
|
};
|
|
|
|
/**
|
|
* Install block I/O protocols
|
|
*
|
|
* @ret vdisk New virtual disk handle
|
|
* @ret vpartition New virtual partition handle
|
|
*/
|
|
void efi_install ( EFI_HANDLE *vdisk, EFI_HANDLE *vpartition ) {
|
|
EFI_BOOT_SERVICES *bs = efi_systab->BootServices;
|
|
EFI_STATUS efirc;
|
|
|
|
/* Install virtual disk */
|
|
if ( ( efirc = bs->InstallMultipleProtocolInterfaces (
|
|
vdisk,
|
|
&efi_block_io_protocol_guid, &efi_vdisk.block,
|
|
&efi_device_path_protocol_guid, efi_vdisk.path,
|
|
NULL ) ) != 0 ) {
|
|
die ( "Could not install disk block I/O protocols: %#lx\n",
|
|
( ( unsigned long ) efirc ) );
|
|
}
|
|
|
|
/* Install virtual partition */
|
|
if ( ( efirc = bs->InstallMultipleProtocolInterfaces (
|
|
vpartition,
|
|
&efi_block_io_protocol_guid, &efi_vpartition.block,
|
|
&efi_device_path_protocol_guid, efi_vpartition.path,
|
|
NULL ) ) != 0 ) {
|
|
die ( "Could not install partition block I/O protocols: %#lx\n",
|
|
( ( unsigned long ) efirc ) );
|
|
}
|
|
}
|
|
|
|
/** Boot image path */
|
|
static struct {
|
|
VENDOR_DEVICE_PATH vendor;
|
|
ATAPI_DEVICE_PATH ata;
|
|
HARDDRIVE_DEVICE_PATH hd;
|
|
struct {
|
|
EFI_DEVICE_PATH header;
|
|
CHAR16 name[ sizeof ( EFI_REMOVABLE_MEDIA_FILE_NAME ) /
|
|
sizeof ( CHAR16 ) ];
|
|
} __attribute__ (( packed )) file;
|
|
EFI_DEVICE_PATH_PROTOCOL end;
|
|
} __attribute__ (( packed )) efi_bootmgfw_path = {
|
|
.vendor = EFIBLOCK_DEVPATH_VENDOR_INIT ( efi_bootmgfw_path.vendor ),
|
|
.ata = EFIBLOCK_DEVPATH_ATA_INIT ( efi_bootmgfw_path.ata ),
|
|
.hd = EFIBLOCK_DEVPATH_HD_INIT ( efi_bootmgfw_path.hd ),
|
|
.file = {
|
|
.header = EFI_DEVPATH_INIT ( efi_bootmgfw_path.file,
|
|
MEDIA_DEVICE_PATH,
|
|
MEDIA_FILEPATH_DP ),
|
|
.name = EFI_REMOVABLE_MEDIA_FILE_NAME,
|
|
},
|
|
.end = EFI_DEVPATH_END_INIT ( efi_bootmgfw_path.end ),
|
|
};
|
|
|
|
/** Boot image path */
|
|
EFI_DEVICE_PATH_PROTOCOL *bootmgfw_path = &efi_bootmgfw_path.vendor.Header;
|