mirror of
https://github.com/ventoy/Ventoy.git
synced 2025-02-05 02:38:10 -05:00
158 lines
4.4 KiB
C
158 lines
4.4 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 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 );
|
|
}
|