mirror of
https://github.com/ipxe/ipxe
synced 2025-12-29 19:38:37 +03:00
[efi] Add efirom utility and .efirom image format
This commit is contained in:
Michael Brown
321
src/util/efirom.c
Normal file
321
src/util/efirom.c
Normal file
@@ -0,0 +1,321 @@
|
||||
/*
|
||||
* Copyright (C) 2009 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 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., 675 Mass Ave, Cambridge, MA 02139, USA.
|
||||
*/
|
||||
|
||||
#include <stdint.h>
|
||||
#include <stddef.h>
|
||||
#include <stdlib.h>
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#include <unistd.h>
|
||||
#include <errno.h>
|
||||
#include <assert.h>
|
||||
#include <getopt.h>
|
||||
#include <gpxe/efi/efi.h>
|
||||
#include <gpxe/efi/IndustryStandard/PeImage.h>
|
||||
#include <gpxe/efi/IndustryStandard/Pci22.h>
|
||||
|
||||
#define eprintf(...) fprintf ( stderr, __VA_ARGS__ )
|
||||
|
||||
/** Command-line options */
|
||||
struct options {
|
||||
uint16_t vendor;
|
||||
uint16_t device;
|
||||
};
|
||||
|
||||
/**
|
||||
* Allocate memory
|
||||
*
|
||||
* @v len Length of memory to allocate
|
||||
* @ret ptr Pointer to allocated memory
|
||||
*/
|
||||
static void * xmalloc ( size_t len ) {
|
||||
void *ptr;
|
||||
|
||||
ptr = malloc ( len );
|
||||
if ( ! ptr ) {
|
||||
eprintf ( "Could not allocate %zd bytes\n", len );
|
||||
exit ( 1 );
|
||||
}
|
||||
|
||||
return ptr;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get file size
|
||||
*
|
||||
* @v file File
|
||||
* @v len File size
|
||||
*/
|
||||
static size_t file_size ( FILE *file ) {
|
||||
ssize_t len;
|
||||
|
||||
if ( fseek ( file, 0, SEEK_END ) != 0 ) {
|
||||
eprintf ( "Could not seek: %s\n", strerror ( errno ) );
|
||||
exit ( 1 );
|
||||
}
|
||||
len = ftell ( file );
|
||||
if ( len < 0 ) {
|
||||
eprintf ( "Could not determine file size: %s\n",
|
||||
strerror ( errno ) );
|
||||
exit ( 1 );
|
||||
}
|
||||
return len;
|
||||
}
|
||||
|
||||
/**
|
||||
* Copy file
|
||||
*
|
||||
* @v in Input file
|
||||
* @v out Output file
|
||||
* @v len Length to copy
|
||||
*/
|
||||
static void file_copy ( FILE *in, FILE *out, size_t len ) {
|
||||
char buf[4096];
|
||||
size_t frag_len;
|
||||
|
||||
while ( len ) {
|
||||
frag_len = len;
|
||||
if ( frag_len > sizeof ( buf ) )
|
||||
frag_len = sizeof ( buf );
|
||||
if ( fread ( buf, frag_len, 1, in ) != 1 ) {
|
||||
eprintf ( "Could not read: %s\n",
|
||||
strerror ( errno ) );
|
||||
exit ( 1 );
|
||||
}
|
||||
if ( fwrite ( buf, frag_len, 1, out ) != 1 ) {
|
||||
eprintf ( "Could not write: %s\n",
|
||||
strerror ( errno ) );
|
||||
exit ( 1 );
|
||||
}
|
||||
len -= frag_len;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Read information from PE headers
|
||||
*
|
||||
* @v pe PE file
|
||||
* @ret machine Machine type
|
||||
* @ret subsystem EFI subsystem
|
||||
*/
|
||||
static void read_pe_info ( FILE *pe, uint16_t *machine,
|
||||
uint16_t *subsystem ) {
|
||||
EFI_IMAGE_DOS_HEADER dos;
|
||||
union {
|
||||
EFI_IMAGE_NT_HEADERS32 nt32;
|
||||
EFI_IMAGE_NT_HEADERS64 nt64;
|
||||
} nt;
|
||||
|
||||
/* Read DOS header */
|
||||
if ( fseek ( pe, 0, SEEK_SET ) != 0 ) {
|
||||
eprintf ( "Could not seek: %s\n", strerror ( errno ) );
|
||||
exit ( 1 );
|
||||
}
|
||||
if ( fread ( &dos, sizeof ( dos ), 1, pe ) != 1 ) {
|
||||
eprintf ( "Could not read: %s\n", strerror ( errno ) );
|
||||
exit ( 1 );
|
||||
}
|
||||
|
||||
/* Read NT header */
|
||||
if ( fseek ( pe, dos.e_lfanew, SEEK_SET ) != 0 ) {
|
||||
eprintf ( "Could not seek: %s\n", strerror ( errno ) );
|
||||
exit ( 1 );
|
||||
}
|
||||
if ( fread ( &nt, sizeof ( nt ), 1, pe ) != 1 ) {
|
||||
eprintf ( "Could not read: %s\n", strerror ( errno ) );
|
||||
exit ( 1 );
|
||||
}
|
||||
|
||||
/* Locate NT header */
|
||||
*machine = nt.nt32.FileHeader.Machine;
|
||||
switch ( *machine ) {
|
||||
case EFI_IMAGE_MACHINE_IA32:
|
||||
*subsystem = nt.nt32.OptionalHeader.Subsystem;
|
||||
break;
|
||||
case EFI_IMAGE_MACHINE_X64:
|
||||
*subsystem = nt.nt64.OptionalHeader.Subsystem;
|
||||
break;
|
||||
default:
|
||||
eprintf ( "Unrecognised machine type %04x\n", *machine );
|
||||
exit ( 1 );
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Convert EFI image to ROM image
|
||||
*
|
||||
* @v pe EFI file
|
||||
* @v rom ROM file
|
||||
*/
|
||||
static void make_efi_rom ( FILE *pe, FILE *rom, struct options *opts ) {
|
||||
struct {
|
||||
EFI_PCI_EXPANSION_ROM_HEADER rom;
|
||||
PCI_DATA_STRUCTURE pci __attribute__ (( aligned ( 4 ) ));
|
||||
} headers;
|
||||
size_t pe_size;
|
||||
size_t rom_size;
|
||||
unsigned int rom_size_sectors;
|
||||
|
||||
/* Determine output file size */
|
||||
pe_size = file_size ( pe );
|
||||
rom_size = ( pe_size + sizeof ( headers ) );
|
||||
rom_size_sectors = ( ( rom_size + 511 ) / 512 );
|
||||
|
||||
/* Construct ROM header */
|
||||
memset ( &headers, 0, sizeof ( headers ) );
|
||||
headers.rom.Signature = PCI_EXPANSION_ROM_HEADER_SIGNATURE;
|
||||
headers.rom.InitializationSize = rom_size_sectors;
|
||||
headers.rom.EfiSignature = EFI_PCI_EXPANSION_ROM_HEADER_EFISIGNATURE;
|
||||
read_pe_info ( pe, &headers.rom.EfiMachineType,
|
||||
&headers.rom.EfiSubsystem );
|
||||
headers.rom.EfiImageHeaderOffset = sizeof ( headers );
|
||||
headers.rom.PcirOffset =
|
||||
offsetof ( typeof ( headers ), pci );
|
||||
headers.pci.Signature = PCI_DATA_STRUCTURE_SIGNATURE;
|
||||
headers.pci.VendorId = opts->vendor;
|
||||
headers.pci.DeviceId = opts->device;
|
||||
headers.pci.Length = sizeof ( headers.pci );
|
||||
headers.pci.ClassCode[0] = PCI_CLASS_NETWORK;
|
||||
headers.pci.ImageLength = rom_size_sectors;
|
||||
headers.pci.CodeType = 0x03; /* No constant in EFI headers? */
|
||||
headers.pci.Indicator = 0x80; /* No constant in EFI headers? */
|
||||
|
||||
/* Write out ROM header */
|
||||
if ( fwrite ( &headers, sizeof ( headers ), 1, rom ) != 1 ) {
|
||||
eprintf ( "Could not write headers: %s\n",
|
||||
strerror ( errno ) );
|
||||
exit ( 1 );
|
||||
}
|
||||
|
||||
/* Write out payload */
|
||||
if ( fseek ( pe, 0, SEEK_SET ) != 0 ) {
|
||||
eprintf ( "Could not seek: %s\n", strerror ( errno ) );
|
||||
exit ( 1 );
|
||||
}
|
||||
file_copy ( pe, rom, pe_size );
|
||||
|
||||
/* Round up to 512-byte boundary */
|
||||
if ( ftruncate ( fileno ( rom ), ( rom_size_sectors * 512 ) ) != 0 ) {
|
||||
eprintf ( "Could not set length: %s\n", strerror ( errno ) );
|
||||
exit ( 1 );
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Print help
|
||||
*
|
||||
* @v program_name Program name
|
||||
*/
|
||||
static void print_help ( const char *program_name ) {
|
||||
eprintf ( "Syntax: %s [--vendor=VVVV] [--device=DDDD] "
|
||||
"infile outfile\n", program_name );
|
||||
}
|
||||
|
||||
/**
|
||||
* Parse command-line options
|
||||
*
|
||||
* @v argc Argument count
|
||||
* @v argv Argument list
|
||||
* @v opts Options structure to populate
|
||||
*/
|
||||
static int parse_options ( const int argc, char **argv,
|
||||
struct options *opts ) {
|
||||
char *end;
|
||||
int c;
|
||||
|
||||
while (1) {
|
||||
int option_index = 0;
|
||||
static struct option long_options[] = {
|
||||
{ "vendor", required_argument, NULL, 'v' },
|
||||
{ "device", required_argument, NULL, 'd' },
|
||||
{ "help", 0, NULL, 'h' },
|
||||
{ 0, 0, 0, 0 }
|
||||
};
|
||||
|
||||
if ( ( c = getopt_long ( argc, argv, "v:d:h",
|
||||
long_options,
|
||||
&option_index ) ) == -1 ) {
|
||||
break;
|
||||
}
|
||||
|
||||
switch ( c ) {
|
||||
case 'v':
|
||||
opts->vendor = strtoul ( optarg, &end, 16 );
|
||||
if ( *end ) {
|
||||
eprintf ( "Invalid vendor \"%s\"\n", optarg );
|
||||
exit ( 2 );
|
||||
}
|
||||
break;
|
||||
case 'd':
|
||||
opts->device = strtoul ( optarg, &end, 16 );
|
||||
if ( *end ) {
|
||||
eprintf ( "Invalid device \"%s\"\n", optarg );
|
||||
exit ( 2 );
|
||||
}
|
||||
break;
|
||||
case 'h':
|
||||
print_help ( argv[0] );
|
||||
exit ( 0 );
|
||||
case '?':
|
||||
default:
|
||||
exit ( 2 );
|
||||
}
|
||||
}
|
||||
return optind;
|
||||
}
|
||||
|
||||
int main ( int argc, char **argv ) {
|
||||
struct options opts = {
|
||||
};
|
||||
unsigned int infile_index;
|
||||
const char *infile_name;
|
||||
const char *outfile_name;
|
||||
FILE *infile;
|
||||
FILE *outfile;
|
||||
|
||||
/* Parse command-line arguments */
|
||||
infile_index = parse_options ( argc, argv, &opts );
|
||||
if ( argc != ( infile_index + 2 ) ) {
|
||||
print_help ( argv[0] );
|
||||
exit ( 2 );
|
||||
}
|
||||
infile_name = argv[infile_index];
|
||||
outfile_name = argv[infile_index + 1];
|
||||
|
||||
/* Open input and output files */
|
||||
infile = fopen ( infile_name, "r" );
|
||||
if ( ! infile ) {
|
||||
eprintf ( "Could not open %s for reading: %s\n",
|
||||
infile_name, strerror ( errno ) );
|
||||
exit ( 1 );
|
||||
}
|
||||
outfile = fopen ( outfile_name, "w" );
|
||||
if ( ! outfile ) {
|
||||
eprintf ( "Could not open %s for writing: %s\n",
|
||||
outfile_name, strerror ( errno ) );
|
||||
exit ( 1 );
|
||||
}
|
||||
|
||||
/* Convert file */
|
||||
make_efi_rom ( infile, outfile, &opts );
|
||||
|
||||
fclose ( outfile );
|
||||
fclose ( infile );
|
||||
|
||||
return 0;
|
||||
}
|
||||
Reference in New Issue
Block a user