blob: 971a0ce062aad13b1f4b8384953af8cb1114c91a [file] [log] [blame]
Linus Torvalds1da177e2005-04-16 15:20:36 -07001/*
Linus Torvalds1da177e2005-04-16 15:20:36 -07002 * Copyright (C) 1991, 1992 Linus Torvalds
3 * Copyright (C) 1997 Martin Mares
H. Peter Anvin4fd06962007-07-11 12:18:56 -07004 * Copyright (C) 2007 H. Peter Anvin
Linus Torvalds1da177e2005-04-16 15:20:36 -07005 */
6
7/*
H. Peter Anvin9aa39092007-07-13 16:28:27 -07008 * This file builds a disk-image from two different files:
Linus Torvalds1da177e2005-04-16 15:20:36 -07009 *
Linus Torvalds1da177e2005-04-16 15:20:36 -070010 * - setup: 8086 machine code, sets up system parm
11 * - system: 80386 code for actual system
12 *
13 * It does some checking that all files are of the correct type, and
14 * just writes the result to stdout, removing headers and padding to
15 * the right amount. It also writes some system data to stderr.
16 */
17
18/*
19 * Changes by tytso to allow root device specification
20 * High loaded stuff by Hans Lermen & Werner Almesberger, Feb. 1996
21 * Cross compiling fixes by Gertjan van Wingerde, July 1996
22 * Rewritten by Martin Mares, April 1997
H. Peter Anvin4fd06962007-07-11 12:18:56 -070023 * Substantially overhauled by H. Peter Anvin, April 2007
Linus Torvalds1da177e2005-04-16 15:20:36 -070024 */
25
26#include <stdio.h>
27#include <string.h>
28#include <stdlib.h>
29#include <stdarg.h>
30#include <sys/types.h>
31#include <sys/stat.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070032#include <unistd.h>
33#include <fcntl.h>
H. Peter Anvin4fd06962007-07-11 12:18:56 -070034#include <sys/mman.h>
Matt Fleming92f42c52012-02-28 13:37:24 +000035#include <tools/le_byteshift.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070036
H. Peter Anvin4fd06962007-07-11 12:18:56 -070037typedef unsigned char u8;
38typedef unsigned short u16;
H. Peter Anvina51f404772012-02-28 23:36:21 -080039typedef unsigned int u32;
Linus Torvalds1da177e2005-04-16 15:20:36 -070040
41#define DEFAULT_MAJOR_ROOT 0
42#define DEFAULT_MINOR_ROOT 0
Matt Fleming92f42c52012-02-28 13:37:24 +000043#define DEFAULT_ROOT_DEV (DEFAULT_MAJOR_ROOT << 8 | DEFAULT_MINOR_ROOT)
Linus Torvalds1da177e2005-04-16 15:20:36 -070044
H. Peter Anvin4fd06962007-07-11 12:18:56 -070045/* Minimal number of setup sectors */
46#define SETUP_SECT_MIN 5
47#define SETUP_SECT_MAX 64
Linus Torvalds1da177e2005-04-16 15:20:36 -070048
H. Peter Anvin4fd06962007-07-11 12:18:56 -070049/* This must be large enough to hold the entire setup */
50u8 buf[SETUP_SECT_MAX*512];
Linus Torvalds1da177e2005-04-16 15:20:36 -070051int is_big_kernel;
52
Jordan Justen743628e2012-06-07 09:05:21 -070053#define PECOFF_RELOC_RESERVE 0x20
54
David Woodhouse99f857d2013-01-10 14:31:59 +000055unsigned long efi_stub_entry;
56unsigned long efi_pe_entry;
57unsigned long startup_64;
58
Ian Campbell7d6e7372008-02-17 20:06:35 +010059/*----------------------------------------------------------------------*/
60
61static const u32 crctab32[] = {
62 0x00000000, 0x77073096, 0xee0e612c, 0x990951ba, 0x076dc419,
63 0x706af48f, 0xe963a535, 0x9e6495a3, 0x0edb8832, 0x79dcb8a4,
64 0xe0d5e91e, 0x97d2d988, 0x09b64c2b, 0x7eb17cbd, 0xe7b82d07,
65 0x90bf1d91, 0x1db71064, 0x6ab020f2, 0xf3b97148, 0x84be41de,
66 0x1adad47d, 0x6ddde4eb, 0xf4d4b551, 0x83d385c7, 0x136c9856,
67 0x646ba8c0, 0xfd62f97a, 0x8a65c9ec, 0x14015c4f, 0x63066cd9,
68 0xfa0f3d63, 0x8d080df5, 0x3b6e20c8, 0x4c69105e, 0xd56041e4,
69 0xa2677172, 0x3c03e4d1, 0x4b04d447, 0xd20d85fd, 0xa50ab56b,
70 0x35b5a8fa, 0x42b2986c, 0xdbbbc9d6, 0xacbcf940, 0x32d86ce3,
71 0x45df5c75, 0xdcd60dcf, 0xabd13d59, 0x26d930ac, 0x51de003a,
72 0xc8d75180, 0xbfd06116, 0x21b4f4b5, 0x56b3c423, 0xcfba9599,
73 0xb8bda50f, 0x2802b89e, 0x5f058808, 0xc60cd9b2, 0xb10be924,
74 0x2f6f7c87, 0x58684c11, 0xc1611dab, 0xb6662d3d, 0x76dc4190,
75 0x01db7106, 0x98d220bc, 0xefd5102a, 0x71b18589, 0x06b6b51f,
76 0x9fbfe4a5, 0xe8b8d433, 0x7807c9a2, 0x0f00f934, 0x9609a88e,
77 0xe10e9818, 0x7f6a0dbb, 0x086d3d2d, 0x91646c97, 0xe6635c01,
78 0x6b6b51f4, 0x1c6c6162, 0x856530d8, 0xf262004e, 0x6c0695ed,
79 0x1b01a57b, 0x8208f4c1, 0xf50fc457, 0x65b0d9c6, 0x12b7e950,
80 0x8bbeb8ea, 0xfcb9887c, 0x62dd1ddf, 0x15da2d49, 0x8cd37cf3,
81 0xfbd44c65, 0x4db26158, 0x3ab551ce, 0xa3bc0074, 0xd4bb30e2,
82 0x4adfa541, 0x3dd895d7, 0xa4d1c46d, 0xd3d6f4fb, 0x4369e96a,
83 0x346ed9fc, 0xad678846, 0xda60b8d0, 0x44042d73, 0x33031de5,
84 0xaa0a4c5f, 0xdd0d7cc9, 0x5005713c, 0x270241aa, 0xbe0b1010,
85 0xc90c2086, 0x5768b525, 0x206f85b3, 0xb966d409, 0xce61e49f,
86 0x5edef90e, 0x29d9c998, 0xb0d09822, 0xc7d7a8b4, 0x59b33d17,
87 0x2eb40d81, 0xb7bd5c3b, 0xc0ba6cad, 0xedb88320, 0x9abfb3b6,
88 0x03b6e20c, 0x74b1d29a, 0xead54739, 0x9dd277af, 0x04db2615,
89 0x73dc1683, 0xe3630b12, 0x94643b84, 0x0d6d6a3e, 0x7a6a5aa8,
90 0xe40ecf0b, 0x9309ff9d, 0x0a00ae27, 0x7d079eb1, 0xf00f9344,
91 0x8708a3d2, 0x1e01f268, 0x6906c2fe, 0xf762575d, 0x806567cb,
92 0x196c3671, 0x6e6b06e7, 0xfed41b76, 0x89d32be0, 0x10da7a5a,
93 0x67dd4acc, 0xf9b9df6f, 0x8ebeeff9, 0x17b7be43, 0x60b08ed5,
94 0xd6d6a3e8, 0xa1d1937e, 0x38d8c2c4, 0x4fdff252, 0xd1bb67f1,
95 0xa6bc5767, 0x3fb506dd, 0x48b2364b, 0xd80d2bda, 0xaf0a1b4c,
96 0x36034af6, 0x41047a60, 0xdf60efc3, 0xa867df55, 0x316e8eef,
97 0x4669be79, 0xcb61b38c, 0xbc66831a, 0x256fd2a0, 0x5268e236,
98 0xcc0c7795, 0xbb0b4703, 0x220216b9, 0x5505262f, 0xc5ba3bbe,
99 0xb2bd0b28, 0x2bb45a92, 0x5cb36a04, 0xc2d7ffa7, 0xb5d0cf31,
100 0x2cd99e8b, 0x5bdeae1d, 0x9b64c2b0, 0xec63f226, 0x756aa39c,
101 0x026d930a, 0x9c0906a9, 0xeb0e363f, 0x72076785, 0x05005713,
102 0x95bf4a82, 0xe2b87a14, 0x7bb12bae, 0x0cb61b38, 0x92d28e9b,
103 0xe5d5be0d, 0x7cdcefb7, 0x0bdbdf21, 0x86d3d2d4, 0xf1d4e242,
104 0x68ddb3f8, 0x1fda836e, 0x81be16cd, 0xf6b9265b, 0x6fb077e1,
105 0x18b74777, 0x88085ae6, 0xff0f6a70, 0x66063bca, 0x11010b5c,
106 0x8f659eff, 0xf862ae69, 0x616bffd3, 0x166ccf45, 0xa00ae278,
107 0xd70dd2ee, 0x4e048354, 0x3903b3c2, 0xa7672661, 0xd06016f7,
108 0x4969474d, 0x3e6e77db, 0xaed16a4a, 0xd9d65adc, 0x40df0b66,
109 0x37d83bf0, 0xa9bcae53, 0xdebb9ec5, 0x47b2cf7f, 0x30b5ffe9,
110 0xbdbdf21c, 0xcabac28a, 0x53b39330, 0x24b4a3a6, 0xbad03605,
111 0xcdd70693, 0x54de5729, 0x23d967bf, 0xb3667a2e, 0xc4614ab8,
112 0x5d681b02, 0x2a6f2b94, 0xb40bbe37, 0xc30c8ea1, 0x5a05df1b,
113 0x2d02ef8d
114};
115
116static u32 partial_crc32_one(u8 c, u32 crc)
117{
118 return crctab32[(crc ^ c) & 0xff] ^ (crc >> 8);
119}
120
121static u32 partial_crc32(const u8 *s, int len, u32 crc)
122{
123 while (len--)
124 crc = partial_crc32_one(*s++, crc);
125 return crc;
126}
127
H. Peter Anvin4fd06962007-07-11 12:18:56 -0700128static void die(const char * str, ...)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700129{
130 va_list args;
131 va_start(args, str);
132 vfprintf(stderr, str, args);
133 fputc('\n', stderr);
134 exit(1);
135}
136
H. Peter Anvin4fd06962007-07-11 12:18:56 -0700137static void usage(void)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700138{
David Woodhouse99f857d2013-01-10 14:31:59 +0000139 die("Usage: build setup system [zoffset.h] [> image]");
Linus Torvalds1da177e2005-04-16 15:20:36 -0700140}
141
Jordan Justen743628e2012-06-07 09:05:21 -0700142#ifdef CONFIG_EFI_STUB
143
Michael Brown819ab992014-07-10 12:26:20 +0100144static void update_pecoff_section_header_fields(char *section_name, u32 vma, u32 size, u32 datasz, u32 offset)
Jordan Justen743628e2012-06-07 09:05:21 -0700145{
146 unsigned int pe_header;
147 unsigned short num_sections;
148 u8 *section;
149
150 pe_header = get_unaligned_le32(&buf[0x3c]);
151 num_sections = get_unaligned_le16(&buf[pe_header + 6]);
152
153#ifdef CONFIG_X86_32
154 section = &buf[pe_header + 0xa8];
155#else
156 section = &buf[pe_header + 0xb8];
157#endif
158
159 while (num_sections > 0) {
160 if (strncmp((char*)section, section_name, 8) == 0) {
161 /* section header size field */
162 put_unaligned_le32(size, section + 0x8);
163
164 /* section header vma field */
Michael Brown819ab992014-07-10 12:26:20 +0100165 put_unaligned_le32(vma, section + 0xc);
Jordan Justen743628e2012-06-07 09:05:21 -0700166
167 /* section header 'size of initialised data' field */
Michael Brown819ab992014-07-10 12:26:20 +0100168 put_unaligned_le32(datasz, section + 0x10);
Jordan Justen743628e2012-06-07 09:05:21 -0700169
170 /* section header 'file offset' field */
171 put_unaligned_le32(offset, section + 0x14);
172
173 break;
174 }
175 section += 0x28;
176 num_sections--;
177 }
178}
179
Michael Brown819ab992014-07-10 12:26:20 +0100180static void update_pecoff_section_header(char *section_name, u32 offset, u32 size)
181{
182 update_pecoff_section_header_fields(section_name, offset, size, size, offset);
183}
184
Jordan Justen743628e2012-06-07 09:05:21 -0700185static void update_pecoff_setup_and_reloc(unsigned int size)
186{
187 u32 setup_offset = 0x200;
188 u32 reloc_offset = size - PECOFF_RELOC_RESERVE;
189 u32 setup_size = reloc_offset - setup_offset;
190
191 update_pecoff_section_header(".setup", setup_offset, setup_size);
192 update_pecoff_section_header(".reloc", reloc_offset, PECOFF_RELOC_RESERVE);
193
194 /*
195 * Modify .reloc section contents with a single entry. The
196 * relocation is applied to offset 10 of the relocation section.
197 */
198 put_unaligned_le32(reloc_offset + 10, &buf[reloc_offset]);
199 put_unaligned_le32(10, &buf[reloc_offset + 4]);
200}
201
202static void update_pecoff_text(unsigned int text_start, unsigned int file_sz)
203{
204 unsigned int pe_header;
205 unsigned int text_sz = file_sz - text_start;
206
207 pe_header = get_unaligned_le32(&buf[0x3c]);
208
Jordan Justen743628e2012-06-07 09:05:21 -0700209 /*
210 * Size of code: Subtract the size of the first sector (512 bytes)
211 * which includes the header.
212 */
213 put_unaligned_le32(file_sz - 512, &buf[pe_header + 0x1c]);
214
Jordan Justen743628e2012-06-07 09:05:21 -0700215 /*
David Woodhouse99f857d2013-01-10 14:31:59 +0000216 * Address of entry point for PE/COFF executable
Jordan Justen743628e2012-06-07 09:05:21 -0700217 */
David Woodhouse99f857d2013-01-10 14:31:59 +0000218 put_unaligned_le32(text_start + efi_pe_entry, &buf[pe_header + 0x28]);
Jordan Justen743628e2012-06-07 09:05:21 -0700219
220 update_pecoff_section_header(".text", text_start, text_sz);
221}
222
Michael Brown819ab992014-07-10 12:26:20 +0100223static void update_pecoff_bss(unsigned int file_sz, unsigned int init_sz)
224{
225 unsigned int pe_header;
226 unsigned int bss_sz = init_sz - file_sz;
227
228 pe_header = get_unaligned_le32(&buf[0x3c]);
229
230 /* Size of uninitialized data */
231 put_unaligned_le32(bss_sz, &buf[pe_header + 0x24]);
232
233 /* Size of image */
234 put_unaligned_le32(init_sz, &buf[pe_header + 0x50]);
235
236 update_pecoff_section_header_fields(".bss", file_sz, bss_sz, 0, 0);
237}
238
Jordan Justen743628e2012-06-07 09:05:21 -0700239#endif /* CONFIG_EFI_STUB */
240
David Woodhouse99f857d2013-01-10 14:31:59 +0000241
242/*
243 * Parse zoffset.h and find the entry points. We could just #include zoffset.h
244 * but that would mean tools/build would have to be rebuilt every time. It's
245 * not as if parsing it is hard...
246 */
247#define PARSE_ZOFS(p, sym) do { \
248 if (!strncmp(p, "#define ZO_" #sym " ", 11+sizeof(#sym))) \
249 sym = strtoul(p + 11 + sizeof(#sym), NULL, 16); \
250} while (0)
251
252static void parse_zoffset(char *fname)
253{
254 FILE *file;
255 char *p;
256 int c;
257
258 file = fopen(fname, "r");
259 if (!file)
260 die("Unable to open `%s': %m", fname);
261 c = fread(buf, 1, sizeof(buf) - 1, file);
262 if (ferror(file))
263 die("read-error on `zoffset.h'");
264 buf[c] = 0;
265
266 p = (char *)buf;
267
268 while (p && *p) {
269 PARSE_ZOFS(p, efi_stub_entry);
270 PARSE_ZOFS(p, efi_pe_entry);
271 PARSE_ZOFS(p, startup_64);
272
273 p = strchr(p, '\n');
274 while (p && (*p == '\r' || *p == '\n'))
275 p++;
276 }
277}
278
Linus Torvalds1da177e2005-04-16 15:20:36 -0700279int main(int argc, char ** argv)
280{
KAMBAROV, ZAURa8f50342005-06-28 20:45:06 -0700281 unsigned int i, sz, setup_sectors;
282 int c;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700283 u32 sys_size;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700284 struct stat sb;
H. Peter Anvin4fd06962007-07-11 12:18:56 -0700285 FILE *file;
286 int fd;
287 void *kernel;
Ian Campbell7d6e7372008-02-17 20:06:35 +0100288 u32 crc = 0xffffffffUL;
Michael Brown819ab992014-07-10 12:26:20 +0100289#ifdef CONFIG_EFI_STUB
290 unsigned int init_sz;
291#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -0700292
David Woodhouse99f857d2013-01-10 14:31:59 +0000293 /* Defaults for old kernel */
294#ifdef CONFIG_X86_32
295 efi_pe_entry = 0x10;
296 efi_stub_entry = 0x30;
297#else
298 efi_pe_entry = 0x210;
299 efi_stub_entry = 0x230;
300 startup_64 = 0x200;
301#endif
302
303 if (argc == 4)
304 parse_zoffset(argv[3]);
305 else if (argc != 3)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700306 usage();
Linus Torvalds1da177e2005-04-16 15:20:36 -0700307
H. Peter Anvin4fd06962007-07-11 12:18:56 -0700308 /* Copy the setup code */
309 file = fopen(argv[1], "r");
310 if (!file)
311 die("Unable to open `%s': %m", argv[1]);
312 c = fread(buf, 1, sizeof(buf), file);
313 if (ferror(file))
314 die("read-error on `setup'");
315 if (c < 1024)
316 die("The setup must be at least 1024 bytes");
Matt Fleming92f42c52012-02-28 13:37:24 +0000317 if (get_unaligned_le16(&buf[510]) != 0xAA55)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700318 die("Boot block hasn't got boot flag (0xAA55)");
H. Peter Anvin4fd06962007-07-11 12:18:56 -0700319 fclose(file);
320
Jordan Justen743628e2012-06-07 09:05:21 -0700321#ifdef CONFIG_EFI_STUB
322 /* Reserve 0x20 bytes for .reloc section */
323 memset(buf+c, 0, PECOFF_RELOC_RESERVE);
324 c += PECOFF_RELOC_RESERVE;
325#endif
326
H. Peter Anvin4fd06962007-07-11 12:18:56 -0700327 /* Pad unused space with zeros */
328 setup_sectors = (c + 511) / 512;
329 if (setup_sectors < SETUP_SECT_MIN)
330 setup_sectors = SETUP_SECT_MIN;
331 i = setup_sectors*512;
332 memset(buf+c, 0, i-c);
333
Jordan Justen743628e2012-06-07 09:05:21 -0700334#ifdef CONFIG_EFI_STUB
335 update_pecoff_setup_and_reloc(i);
336#endif
337
H. Peter Anvin4fd06962007-07-11 12:18:56 -0700338 /* Set the default root device */
Matt Fleming92f42c52012-02-28 13:37:24 +0000339 put_unaligned_le16(DEFAULT_ROOT_DEV, &buf[508]);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700340
H. Peter Anvin4fd06962007-07-11 12:18:56 -0700341 fprintf(stderr, "Setup is %d bytes (padded to %d bytes).\n", c, i);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700342
H. Peter Anvin4fd06962007-07-11 12:18:56 -0700343 /* Open and stat the kernel file */
344 fd = open(argv[2], O_RDONLY);
345 if (fd < 0)
346 die("Unable to open `%s': %m", argv[2]);
347 if (fstat(fd, &sb))
348 die("Unable to stat `%s': %m", argv[2]);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700349 sz = sb.st_size;
H. Peter Anvin4fd06962007-07-11 12:18:56 -0700350 fprintf (stderr, "System is %d kB\n", (sz+1023)/1024);
351 kernel = mmap(NULL, sz, PROT_READ, MAP_SHARED, fd, 0);
352 if (kernel == MAP_FAILED)
353 die("Unable to mmap '%s': %m", argv[2]);
Ian Campbell7d6e7372008-02-17 20:06:35 +0100354 /* Number of 16-byte paragraphs, including space for a 4-byte CRC */
355 sys_size = (sz + 15 + 4) / 16;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700356
H. Peter Anvin4fd06962007-07-11 12:18:56 -0700357 /* Patch the setup code with the appropriate size parameters */
358 buf[0x1f1] = setup_sectors-1;
Matt Fleming92f42c52012-02-28 13:37:24 +0000359 put_unaligned_le32(sys_size, &buf[0x1f4]);
H. Peter Anvin4fd06962007-07-11 12:18:56 -0700360
Matt Fleming291f3632011-12-12 21:27:52 +0000361#ifdef CONFIG_EFI_STUB
Michael Brown819ab992014-07-10 12:26:20 +0100362 update_pecoff_text(setup_sectors * 512, i + (sys_size * 16));
363 init_sz = get_unaligned_le32(&buf[0x260]);
364 update_pecoff_bss(i + (sys_size * 16), init_sz);
David Woodhouse99f857d2013-01-10 14:31:59 +0000365
366#ifdef CONFIG_X86_64 /* Yes, this is really how we defined it :( */
367 efi_stub_entry -= 0x200;
368#endif
369 put_unaligned_le32(efi_stub_entry, &buf[0x264]);
Jordan Justen743628e2012-06-07 09:05:21 -0700370#endif
Matt Fleming291f3632011-12-12 21:27:52 +0000371
Ian Campbell7d6e7372008-02-17 20:06:35 +0100372 crc = partial_crc32(buf, i, crc);
H. Peter Anvin4fd06962007-07-11 12:18:56 -0700373 if (fwrite(buf, 1, i, stdout) != i)
374 die("Writing setup failed");
375
376 /* Copy the kernel code */
Ian Campbell7d6e7372008-02-17 20:06:35 +0100377 crc = partial_crc32(kernel, sz, crc);
H. Peter Anvin4fd06962007-07-11 12:18:56 -0700378 if (fwrite(kernel, 1, sz, stdout) != sz)
379 die("Writing kernel failed");
Ian Campbell7d6e7372008-02-17 20:06:35 +0100380
381 /* Add padding leaving 4 bytes for the checksum */
382 while (sz++ < (sys_size*16) - 4) {
383 crc = partial_crc32_one('\0', crc);
384 if (fwrite("\0", 1, 1, stdout) != 1)
385 die("Writing padding failed");
386 }
387
388 /* Write the CRC */
H. Peter Anvina51f404772012-02-28 23:36:21 -0800389 fprintf(stderr, "CRC %x\n", crc);
390 put_unaligned_le32(crc, buf);
391 if (fwrite(buf, 1, 4, stdout) != 4)
Ian Campbell7d6e7372008-02-17 20:06:35 +0100392 die("Writing CRC failed");
393
Linus Torvalds1da177e2005-04-16 15:20:36 -0700394 close(fd);
395
H. Peter Anvin4fd06962007-07-11 12:18:56 -0700396 /* Everything is OK */
397 return 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700398}