mirror of
https://github.com/ventoy/Ventoy.git
synced 2025-08-28 08:21:14 +00:00
1. Optimization for WIMBOOT mode.
2. Add WIMBOOT for UEFI mode.
This commit is contained in:
longpanda
157
wimboot/wimboot-2.7.3/src/efiboot.c
Normal file
157
wimboot/wimboot-2.7.3/src/efiboot.c
Normal file
@@ -0,0 +1,157 @@
|
||||
/*
|
||||
* 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 boot manager invocation
|
||||
*
|
||||
*/
|
||||
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#include "wimboot.h"
|
||||
#include "cmdline.h"
|
||||
#include "vdisk.h"
|
||||
#include "pause.h"
|
||||
#include "efi.h"
|
||||
#include "efi/Protocol/GraphicsOutput.h"
|
||||
#include "efipath.h"
|
||||
#include "efiboot.h"
|
||||
|
||||
/** Original OpenProtocol() method */
|
||||
static EFI_OPEN_PROTOCOL orig_open_protocol;
|
||||
|
||||
/**
|
||||
* Intercept OpenProtocol()
|
||||
*
|
||||
* @v handle EFI handle
|
||||
* @v protocol Protocol GUID
|
||||
* @v interface Opened interface
|
||||
* @v agent_handle Agent handle
|
||||
* @v controller_handle Controller handle
|
||||
* @v attributes Attributes
|
||||
* @ret efirc EFI status code
|
||||
*/
|
||||
static EFI_STATUS EFIAPI
|
||||
efi_open_protocol_wrapper ( EFI_HANDLE handle, EFI_GUID *protocol,
|
||||
VOID **interface, EFI_HANDLE agent_handle,
|
||||
EFI_HANDLE controller_handle, UINT32 attributes ) {
|
||||
static unsigned int count;
|
||||
EFI_STATUS efirc;
|
||||
|
||||
/* Open the protocol */
|
||||
if ( ( efirc = orig_open_protocol ( handle, protocol, interface,
|
||||
agent_handle, controller_handle,
|
||||
attributes ) ) != 0 ) {
|
||||
return efirc;
|
||||
}
|
||||
|
||||
/* Block first attempt by bootmgfw.efi to open
|
||||
* EFI_GRAPHICS_OUTPUT_PROTOCOL. This forces error messages
|
||||
* to be displayed in text mode (thereby avoiding the totally
|
||||
* blank error screen if the fonts are missing). We must
|
||||
* allow subsequent attempts to succeed, otherwise the OS will
|
||||
* fail to boot.
|
||||
*/
|
||||
if ( ( memcmp ( protocol, &efi_graphics_output_protocol_guid,
|
||||
sizeof ( *protocol ) ) == 0 ) &&
|
||||
( count++ == 0 ) && ( ! cmdline_gui ) ) {
|
||||
DBG ( "Forcing text mode output\n" );
|
||||
return EFI_INVALID_PARAMETER;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
/**
|
||||
* Boot from EFI device
|
||||
*
|
||||
* @v file Virtual file
|
||||
* @v path Device path
|
||||
* @v device Device handle
|
||||
*/
|
||||
void efi_boot ( struct vdisk_file *file, EFI_DEVICE_PATH_PROTOCOL *path,
|
||||
EFI_HANDLE device ) {
|
||||
EFI_BOOT_SERVICES *bs = efi_systab->BootServices;
|
||||
union {
|
||||
EFI_LOADED_IMAGE_PROTOCOL *image;
|
||||
void *intf;
|
||||
} loaded;
|
||||
EFI_PHYSICAL_ADDRESS phys;
|
||||
void *data;
|
||||
unsigned int pages;
|
||||
EFI_HANDLE handle;
|
||||
EFI_STATUS efirc;
|
||||
|
||||
/* Allocate memory */
|
||||
pages = ( ( file->len + PAGE_SIZE - 1 ) / PAGE_SIZE );
|
||||
if ( ( efirc = bs->AllocatePages ( AllocateAnyPages,
|
||||
EfiBootServicesData, pages,
|
||||
&phys ) ) != 0 ) {
|
||||
die ( "Could not allocate %d pages: %#lx\n",
|
||||
pages, ( ( unsigned long ) efirc ) );
|
||||
}
|
||||
data = ( ( void * ) ( intptr_t ) phys );
|
||||
|
||||
|
||||
/* Read image */
|
||||
file->read ( file, data, 0, file->len );
|
||||
DBG ( "Read %s\n", file->name );
|
||||
|
||||
/* Load image */
|
||||
if ( ( efirc = bs->LoadImage ( FALSE, efi_image_handle, path, data,
|
||||
file->len, &handle ) ) != 0 ) {
|
||||
die ( "Could not load %s: %#lx\n",
|
||||
file->name, ( ( unsigned long ) efirc ) );
|
||||
}
|
||||
DBG ( "Loaded %s\n", file->name );
|
||||
|
||||
/* Get loaded image protocol */
|
||||
if ( ( efirc = bs->OpenProtocol ( handle,
|
||||
&efi_loaded_image_protocol_guid,
|
||||
&loaded.intf, efi_image_handle, NULL,
|
||||
EFI_OPEN_PROTOCOL_GET_PROTOCOL ))!=0){
|
||||
die ( "Could not get loaded image protocol for %s: %#lx\n",
|
||||
file->name, ( ( unsigned long ) efirc ) );
|
||||
}
|
||||
|
||||
/* Force correct device handle */
|
||||
if ( loaded.image->DeviceHandle != device ) {
|
||||
DBG ( "Forcing correct DeviceHandle (%p->%p)\n",
|
||||
loaded.image->DeviceHandle, device );
|
||||
loaded.image->DeviceHandle = device;
|
||||
}
|
||||
|
||||
/* Intercept calls to OpenProtocol() */
|
||||
orig_open_protocol =
|
||||
loaded.image->SystemTable->BootServices->OpenProtocol;
|
||||
loaded.image->SystemTable->BootServices->OpenProtocol =
|
||||
efi_open_protocol_wrapper;
|
||||
|
||||
/* Start image */
|
||||
if ( cmdline_pause )
|
||||
pause();
|
||||
if ( ( efirc = bs->StartImage ( handle, NULL, NULL ) ) != 0 ) {
|
||||
die ( "Could not start %s: %#lx\n",
|
||||
file->name, ( ( unsigned long ) efirc ) );
|
||||
}
|
||||
|
||||
die ( "%s returned\n", file->name );
|
||||
}
|
||||
Reference in New Issue
Block a user