[mromprefix] Report a dummy size at offset 0x02 of .mrom payload
[ipxe.git] / src / util / efifatbin.c
1 /*
2 * Copyright (C) 2014 Michael Brown <mbrown@fensystems.co.uk>.
3 *
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public License as
6 * published by the Free Software Foundation; either version 2 of the
7 * License, or any later version.
8 *
9 * This program is distributed in the hope that it will be useful, but
10 * WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
17 * 02110-1301, USA.
18 */
19
20 #include <stdint.h>
21 #include <stddef.h>
22 #include <stdlib.h>
23 #include <stdio.h>
24 #include <string.h>
25 #include <sys/stat.h>
26 #include <unistd.h>
27 #include <errno.h>
28 #include <assert.h>
29 #include <getopt.h>
30 #include <ipxe/efi/efi.h>
31 #include <ipxe/efi/IndustryStandard/PeImage.h>
32
33 #define eprintf(...) fprintf ( stderr, __VA_ARGS__ )
34
35 /** Command-line options */
36 struct options {
37 };
38
39 /** EFI fat binary file header */
40 struct efifatbin_file_header {
41 /** Signature */
42 uint32_t signature;
43 /** Count */
44 uint32_t count;
45 } __attribute__ (( packed ));
46
47 /** EFI fat binary signature */
48 #define EFIFATBIN_SIGNATURE 0x0ef1fab9
49
50 /** EFI fat binary image header */
51 struct efifatbin_image_header {
52 /** Flags */
53 uint64_t flags;
54 /** Offset */
55 uint32_t offset;
56 /** Length */
57 uint32_t len;
58 /** Padding */
59 uint32_t pad;
60 } __attribute__ (( packed ));
61
62 /** EFI fat binary default flags */
63 #define EFIFATBIN_FLAGS 0x0000000300000007ULL
64
65 /** EFI fat binary 64-bit flag */
66 #define EFIFATBIN_64BIT 0x0000000001000000ULL
67
68 /**
69 * Allocate memory
70 *
71 * @v len Length of memory to allocate
72 * @ret ptr Pointer to allocated memory
73 */
74 static void * xmalloc ( size_t len ) {
75 void *ptr;
76
77 ptr = malloc ( len );
78 if ( ! ptr ) {
79 eprintf ( "Could not allocate %zd bytes\n", len );
80 exit ( 1 );
81 }
82
83 return ptr;
84 }
85
86 /**
87 * Generate EFI fat binary
88 *
89 * @v count Number of input files
90 * @v infile_names Input filenames
91 * @v outfile_name Output filename
92 */
93 static void make_efifatbin ( unsigned int count, char **infile_names,
94 const char *outfile_name ) {
95 FILE *infile[count];
96 FILE *outfile;
97 struct stat stat[count];
98 void *buf[count];
99 struct efifatbin_file_header file_header;
100 struct efifatbin_image_header header[count];
101 size_t offset;
102 EFI_IMAGE_DOS_HEADER *dos;
103 union {
104 EFI_IMAGE_NT_HEADERS32 nt32;
105 EFI_IMAGE_NT_HEADERS64 nt64;
106 } *nt;
107 unsigned int i;
108
109 /* Generate file header */
110 file_header.signature = EFIFATBIN_SIGNATURE;
111 file_header.count = count;
112 offset = ( sizeof ( file_header ) + sizeof ( header ) );
113
114 /* Process input files */
115 for ( i = 0 ; i < count ; i++ ) {
116
117 /* Open input file */
118 infile[i] = fopen ( infile_names[i], "r" );
119 if ( ! infile[i] ) {
120 eprintf ( "Could not open %s for reading: %s\n",
121 infile_names[i], strerror ( errno ) );
122 exit ( 1 );
123 }
124
125 /* Determine PE file size */
126 if ( fstat ( fileno ( infile[i] ), &stat[i] ) != 0 ) {
127 eprintf ( "Could not stat %s: %s\n",
128 infile_names[i], strerror ( errno ) );
129 exit ( 1 );
130 }
131
132 /* Allocate buffer and read in PE file */
133 buf[i] = xmalloc ( stat[i].st_size );
134 if ( fread ( buf[i], stat[i].st_size, 1, infile[i] ) != 1 ) {
135 eprintf ( "Could not read %s: %s\n",
136 infile_names[i], strerror ( errno ) );
137 exit ( 1 );
138 }
139
140 /* Close input file */
141 fclose ( infile[i] );
142
143 /* Generate image header */
144 header[i].flags = EFIFATBIN_FLAGS;
145 header[i].offset = offset;
146 header[i].len = stat[i].st_size;
147 header[i].pad = 0;
148
149 /* Determine architecture */
150 dos = buf[i];
151 nt = ( buf[i] + dos->e_lfanew );
152 if ( nt->nt32.FileHeader.Machine == EFI_IMAGE_MACHINE_X64 )
153 header[i].flags |= EFIFATBIN_64BIT;
154
155 /* Allow space for this image */
156 offset += stat[i].st_size;
157 }
158
159 /* Open output file */
160 outfile = fopen ( outfile_name, "w" );
161 if ( ! outfile ) {
162 eprintf ( "Could not open %s for writing: %s\n",
163 outfile_name, strerror ( errno ) );
164 exit ( 1 );
165 }
166
167 /* Write fat binary header */
168 if ( fwrite ( &file_header, sizeof ( file_header ), 1, outfile ) != 1 ){
169 eprintf ( "Could not write %s: %s\n",
170 outfile_name, strerror ( errno ) );
171 exit ( 1 );
172 }
173 for ( i = 0 ; i < count ; i++ ) {
174 if ( fwrite ( &header[i], sizeof ( header[i] ), 1,
175 outfile ) != 1 ) {
176 eprintf ( "Could not write %s: %s\n",
177 outfile_name, strerror ( errno ) );
178 exit ( 1 );
179 }
180 }
181
182 /* Write images */
183 for ( i = 0 ; i < count ; i++ ) {
184 if ( fwrite ( buf[i], stat[i].st_size, 1, outfile ) != 1 ) {
185 eprintf ( "Could not write %s: %s\n",
186 outfile_name, strerror ( errno ) );
187 exit ( 1 );
188 }
189 }
190
191 /* Close output file */
192 fclose ( outfile );
193 }
194
195 /**
196 * Print help
197 *
198 * @v program_name Program name
199 */
200 static void print_help ( const char *program_name ) {
201 eprintf ( "Syntax: %s infile [infile...] outfile\n", program_name );
202 }
203
204 /**
205 * Parse command-line options
206 *
207 * @v argc Argument count
208 * @v argv Argument list
209 * @v opts Options structure to populate
210 */
211 static int parse_options ( const int argc, char **argv,
212 struct options *opts __attribute__ (( unused )) ) {
213 int c;
214
215 while (1) {
216 int option_index = 0;
217 static struct option long_options[] = {
218 { "help", 0, NULL, 'h' },
219 { 0, 0, 0, 0 }
220 };
221
222 if ( ( c = getopt_long ( argc, argv, "h",
223 long_options,
224 &option_index ) ) == -1 ) {
225 break;
226 }
227
228 switch ( c ) {
229 case 'h':
230 print_help ( argv[0] );
231 exit ( 0 );
232 case '?':
233 default:
234 exit ( 2 );
235 }
236 }
237 return optind;
238 }
239
240 int main ( int argc, char **argv ) {
241 struct options opts;
242 int infile_index;
243 int outfile_index;
244 int count;
245
246 /* Parse command-line arguments */
247 memset ( &opts, 0, sizeof ( opts ) );
248 infile_index = parse_options ( argc, argv, &opts );
249 outfile_index = ( argc - 1 );
250 count = ( outfile_index - infile_index );
251 if ( count <= 0 ) {
252 print_help ( argv[0] );
253 exit ( 2 );
254 }
255
256 /* Generate fat binary */
257 make_efifatbin ( count, &argv[infile_index], argv[outfile_index] );
258
259 return 0;
260 }