Loading...
1// SPDX-License-Identifier: GPL-2.0
2/*
3 * Copyright (C) 1991, 1992 Linus Torvalds
4 * Copyright (C) 1997 Martin Mares
5 * Copyright (C) 2007 H. Peter Anvin
6 */
7
8/*
9 * This file builds a disk-image from three different files:
10 *
11 * - setup: 8086 machine code, sets up system parm
12 * - system: 80386 code for actual system
13 * - zoffset.h: header with ZO_* defines
14 *
15 * It does some checking that all files are of the correct type, and writes
16 * the result to the specified destination, removing headers and padding to
17 * the right amount. It also writes some system data to stdout.
18 */
19
20/*
21 * Changes by tytso to allow root device specification
22 * High loaded stuff by Hans Lermen & Werner Almesberger, Feb. 1996
23 * Cross compiling fixes by Gertjan van Wingerde, July 1996
24 * Rewritten by Martin Mares, April 1997
25 * Substantially overhauled by H. Peter Anvin, April 2007
26 */
27
28#include <stdio.h>
29#include <string.h>
30#include <stdlib.h>
31#include <stdarg.h>
32#include <sys/types.h>
33#include <sys/stat.h>
34#include <unistd.h>
35#include <fcntl.h>
36#include <sys/mman.h>
37#include <tools/le_byteshift.h>
38
39typedef unsigned char u8;
40typedef unsigned short u16;
41typedef unsigned int u32;
42
43#define DEFAULT_MAJOR_ROOT 0
44#define DEFAULT_MINOR_ROOT 0
45#define DEFAULT_ROOT_DEV (DEFAULT_MAJOR_ROOT << 8 | DEFAULT_MINOR_ROOT)
46
47/* Minimal number of setup sectors */
48#define SETUP_SECT_MIN 5
49#define SETUP_SECT_MAX 64
50
51/* This must be large enough to hold the entire setup */
52u8 buf[SETUP_SECT_MAX*512];
53
54#define PECOFF_RELOC_RESERVE 0x20
55
56unsigned long efi32_stub_entry;
57unsigned long efi64_stub_entry;
58unsigned long efi_pe_entry;
59unsigned long startup_64;
60
61/*----------------------------------------------------------------------*/
62
63static const u32 crctab32[] = {
64 0x00000000, 0x77073096, 0xee0e612c, 0x990951ba, 0x076dc419,
65 0x706af48f, 0xe963a535, 0x9e6495a3, 0x0edb8832, 0x79dcb8a4,
66 0xe0d5e91e, 0x97d2d988, 0x09b64c2b, 0x7eb17cbd, 0xe7b82d07,
67 0x90bf1d91, 0x1db71064, 0x6ab020f2, 0xf3b97148, 0x84be41de,
68 0x1adad47d, 0x6ddde4eb, 0xf4d4b551, 0x83d385c7, 0x136c9856,
69 0x646ba8c0, 0xfd62f97a, 0x8a65c9ec, 0x14015c4f, 0x63066cd9,
70 0xfa0f3d63, 0x8d080df5, 0x3b6e20c8, 0x4c69105e, 0xd56041e4,
71 0xa2677172, 0x3c03e4d1, 0x4b04d447, 0xd20d85fd, 0xa50ab56b,
72 0x35b5a8fa, 0x42b2986c, 0xdbbbc9d6, 0xacbcf940, 0x32d86ce3,
73 0x45df5c75, 0xdcd60dcf, 0xabd13d59, 0x26d930ac, 0x51de003a,
74 0xc8d75180, 0xbfd06116, 0x21b4f4b5, 0x56b3c423, 0xcfba9599,
75 0xb8bda50f, 0x2802b89e, 0x5f058808, 0xc60cd9b2, 0xb10be924,
76 0x2f6f7c87, 0x58684c11, 0xc1611dab, 0xb6662d3d, 0x76dc4190,
77 0x01db7106, 0x98d220bc, 0xefd5102a, 0x71b18589, 0x06b6b51f,
78 0x9fbfe4a5, 0xe8b8d433, 0x7807c9a2, 0x0f00f934, 0x9609a88e,
79 0xe10e9818, 0x7f6a0dbb, 0x086d3d2d, 0x91646c97, 0xe6635c01,
80 0x6b6b51f4, 0x1c6c6162, 0x856530d8, 0xf262004e, 0x6c0695ed,
81 0x1b01a57b, 0x8208f4c1, 0xf50fc457, 0x65b0d9c6, 0x12b7e950,
82 0x8bbeb8ea, 0xfcb9887c, 0x62dd1ddf, 0x15da2d49, 0x8cd37cf3,
83 0xfbd44c65, 0x4db26158, 0x3ab551ce, 0xa3bc0074, 0xd4bb30e2,
84 0x4adfa541, 0x3dd895d7, 0xa4d1c46d, 0xd3d6f4fb, 0x4369e96a,
85 0x346ed9fc, 0xad678846, 0xda60b8d0, 0x44042d73, 0x33031de5,
86 0xaa0a4c5f, 0xdd0d7cc9, 0x5005713c, 0x270241aa, 0xbe0b1010,
87 0xc90c2086, 0x5768b525, 0x206f85b3, 0xb966d409, 0xce61e49f,
88 0x5edef90e, 0x29d9c998, 0xb0d09822, 0xc7d7a8b4, 0x59b33d17,
89 0x2eb40d81, 0xb7bd5c3b, 0xc0ba6cad, 0xedb88320, 0x9abfb3b6,
90 0x03b6e20c, 0x74b1d29a, 0xead54739, 0x9dd277af, 0x04db2615,
91 0x73dc1683, 0xe3630b12, 0x94643b84, 0x0d6d6a3e, 0x7a6a5aa8,
92 0xe40ecf0b, 0x9309ff9d, 0x0a00ae27, 0x7d079eb1, 0xf00f9344,
93 0x8708a3d2, 0x1e01f268, 0x6906c2fe, 0xf762575d, 0x806567cb,
94 0x196c3671, 0x6e6b06e7, 0xfed41b76, 0x89d32be0, 0x10da7a5a,
95 0x67dd4acc, 0xf9b9df6f, 0x8ebeeff9, 0x17b7be43, 0x60b08ed5,
96 0xd6d6a3e8, 0xa1d1937e, 0x38d8c2c4, 0x4fdff252, 0xd1bb67f1,
97 0xa6bc5767, 0x3fb506dd, 0x48b2364b, 0xd80d2bda, 0xaf0a1b4c,
98 0x36034af6, 0x41047a60, 0xdf60efc3, 0xa867df55, 0x316e8eef,
99 0x4669be79, 0xcb61b38c, 0xbc66831a, 0x256fd2a0, 0x5268e236,
100 0xcc0c7795, 0xbb0b4703, 0x220216b9, 0x5505262f, 0xc5ba3bbe,
101 0xb2bd0b28, 0x2bb45a92, 0x5cb36a04, 0xc2d7ffa7, 0xb5d0cf31,
102 0x2cd99e8b, 0x5bdeae1d, 0x9b64c2b0, 0xec63f226, 0x756aa39c,
103 0x026d930a, 0x9c0906a9, 0xeb0e363f, 0x72076785, 0x05005713,
104 0x95bf4a82, 0xe2b87a14, 0x7bb12bae, 0x0cb61b38, 0x92d28e9b,
105 0xe5d5be0d, 0x7cdcefb7, 0x0bdbdf21, 0x86d3d2d4, 0xf1d4e242,
106 0x68ddb3f8, 0x1fda836e, 0x81be16cd, 0xf6b9265b, 0x6fb077e1,
107 0x18b74777, 0x88085ae6, 0xff0f6a70, 0x66063bca, 0x11010b5c,
108 0x8f659eff, 0xf862ae69, 0x616bffd3, 0x166ccf45, 0xa00ae278,
109 0xd70dd2ee, 0x4e048354, 0x3903b3c2, 0xa7672661, 0xd06016f7,
110 0x4969474d, 0x3e6e77db, 0xaed16a4a, 0xd9d65adc, 0x40df0b66,
111 0x37d83bf0, 0xa9bcae53, 0xdebb9ec5, 0x47b2cf7f, 0x30b5ffe9,
112 0xbdbdf21c, 0xcabac28a, 0x53b39330, 0x24b4a3a6, 0xbad03605,
113 0xcdd70693, 0x54de5729, 0x23d967bf, 0xb3667a2e, 0xc4614ab8,
114 0x5d681b02, 0x2a6f2b94, 0xb40bbe37, 0xc30c8ea1, 0x5a05df1b,
115 0x2d02ef8d
116};
117
118static u32 partial_crc32_one(u8 c, u32 crc)
119{
120 return crctab32[(crc ^ c) & 0xff] ^ (crc >> 8);
121}
122
123static u32 partial_crc32(const u8 *s, int len, u32 crc)
124{
125 while (len--)
126 crc = partial_crc32_one(*s++, crc);
127 return crc;
128}
129
130static void die(const char * str, ...)
131{
132 va_list args;
133 va_start(args, str);
134 vfprintf(stderr, str, args);
135 va_end(args);
136 fputc('\n', stderr);
137 exit(1);
138}
139
140static void usage(void)
141{
142 die("Usage: build setup system zoffset.h image");
143}
144
145#ifdef CONFIG_EFI_STUB
146
147static void update_pecoff_section_header_fields(char *section_name, u32 vma, u32 size, u32 datasz, u32 offset)
148{
149 unsigned int pe_header;
150 unsigned short num_sections;
151 u8 *section;
152
153 pe_header = get_unaligned_le32(&buf[0x3c]);
154 num_sections = get_unaligned_le16(&buf[pe_header + 6]);
155
156#ifdef CONFIG_X86_32
157 section = &buf[pe_header + 0xa8];
158#else
159 section = &buf[pe_header + 0xb8];
160#endif
161
162 while (num_sections > 0) {
163 if (strncmp((char*)section, section_name, 8) == 0) {
164 /* section header size field */
165 put_unaligned_le32(size, section + 0x8);
166
167 /* section header vma field */
168 put_unaligned_le32(vma, section + 0xc);
169
170 /* section header 'size of initialised data' field */
171 put_unaligned_le32(datasz, section + 0x10);
172
173 /* section header 'file offset' field */
174 put_unaligned_le32(offset, section + 0x14);
175
176 break;
177 }
178 section += 0x28;
179 num_sections--;
180 }
181}
182
183static void update_pecoff_section_header(char *section_name, u32 offset, u32 size)
184{
185 update_pecoff_section_header_fields(section_name, offset, size, size, offset);
186}
187
188static void update_pecoff_setup_and_reloc(unsigned int size)
189{
190 u32 setup_offset = 0x200;
191 u32 reloc_offset = size - PECOFF_RELOC_RESERVE;
192 u32 setup_size = reloc_offset - setup_offset;
193
194 update_pecoff_section_header(".setup", setup_offset, setup_size);
195 update_pecoff_section_header(".reloc", reloc_offset, PECOFF_RELOC_RESERVE);
196
197 /*
198 * Modify .reloc section contents with a single entry. The
199 * relocation is applied to offset 10 of the relocation section.
200 */
201 put_unaligned_le32(reloc_offset + 10, &buf[reloc_offset]);
202 put_unaligned_le32(10, &buf[reloc_offset + 4]);
203}
204
205static void update_pecoff_text(unsigned int text_start, unsigned int file_sz)
206{
207 unsigned int pe_header;
208 unsigned int text_sz = file_sz - text_start;
209
210 pe_header = get_unaligned_le32(&buf[0x3c]);
211
212 /*
213 * Size of code: Subtract the size of the first sector (512 bytes)
214 * which includes the header.
215 */
216 put_unaligned_le32(file_sz - 512, &buf[pe_header + 0x1c]);
217
218 /*
219 * Address of entry point for PE/COFF executable
220 */
221 put_unaligned_le32(text_start + efi_pe_entry, &buf[pe_header + 0x28]);
222
223 update_pecoff_section_header(".text", text_start, text_sz);
224}
225
226static void update_pecoff_bss(unsigned int file_sz, unsigned int init_sz)
227{
228 unsigned int pe_header;
229 unsigned int bss_sz = init_sz - file_sz;
230
231 pe_header = get_unaligned_le32(&buf[0x3c]);
232
233 /* Size of uninitialized data */
234 put_unaligned_le32(bss_sz, &buf[pe_header + 0x24]);
235
236 /* Size of image */
237 put_unaligned_le32(init_sz, &buf[pe_header + 0x50]);
238
239 update_pecoff_section_header_fields(".bss", file_sz, bss_sz, 0, 0);
240}
241
242static int reserve_pecoff_reloc_section(int c)
243{
244 /* Reserve 0x20 bytes for .reloc section */
245 memset(buf+c, 0, PECOFF_RELOC_RESERVE);
246 return PECOFF_RELOC_RESERVE;
247}
248
249static void efi_stub_defaults(void)
250{
251 /* Defaults for old kernel */
252#ifdef CONFIG_X86_32
253 efi_pe_entry = 0x10;
254#else
255 efi_pe_entry = 0x210;
256 startup_64 = 0x200;
257#endif
258}
259
260static void efi_stub_entry_update(void)
261{
262 unsigned long addr = efi32_stub_entry;
263
264#ifdef CONFIG_X86_64
265 /* Yes, this is really how we defined it :( */
266 addr = efi64_stub_entry - 0x200;
267#endif
268
269#ifdef CONFIG_EFI_MIXED
270 if (efi32_stub_entry != addr)
271 die("32-bit and 64-bit EFI entry points do not match\n");
272#endif
273 put_unaligned_le32(addr, &buf[0x264]);
274}
275
276#else
277
278static inline void update_pecoff_setup_and_reloc(unsigned int size) {}
279static inline void update_pecoff_text(unsigned int text_start,
280 unsigned int file_sz) {}
281static inline void update_pecoff_bss(unsigned int file_sz,
282 unsigned int init_sz) {}
283static inline void efi_stub_defaults(void) {}
284static inline void efi_stub_entry_update(void) {}
285
286static inline int reserve_pecoff_reloc_section(int c)
287{
288 return 0;
289}
290#endif /* CONFIG_EFI_STUB */
291
292
293/*
294 * Parse zoffset.h and find the entry points. We could just #include zoffset.h
295 * but that would mean tools/build would have to be rebuilt every time. It's
296 * not as if parsing it is hard...
297 */
298#define PARSE_ZOFS(p, sym) do { \
299 if (!strncmp(p, "#define ZO_" #sym " ", 11+sizeof(#sym))) \
300 sym = strtoul(p + 11 + sizeof(#sym), NULL, 16); \
301} while (0)
302
303static void parse_zoffset(char *fname)
304{
305 FILE *file;
306 char *p;
307 int c;
308
309 file = fopen(fname, "r");
310 if (!file)
311 die("Unable to open `%s': %m", fname);
312 c = fread(buf, 1, sizeof(buf) - 1, file);
313 if (ferror(file))
314 die("read-error on `zoffset.h'");
315 fclose(file);
316 buf[c] = 0;
317
318 p = (char *)buf;
319
320 while (p && *p) {
321 PARSE_ZOFS(p, efi32_stub_entry);
322 PARSE_ZOFS(p, efi64_stub_entry);
323 PARSE_ZOFS(p, efi_pe_entry);
324 PARSE_ZOFS(p, startup_64);
325
326 p = strchr(p, '\n');
327 while (p && (*p == '\r' || *p == '\n'))
328 p++;
329 }
330}
331
332int main(int argc, char ** argv)
333{
334 unsigned int i, sz, setup_sectors, init_sz;
335 int c;
336 u32 sys_size;
337 struct stat sb;
338 FILE *file, *dest;
339 int fd;
340 void *kernel;
341 u32 crc = 0xffffffffUL;
342
343 efi_stub_defaults();
344
345 if (argc != 5)
346 usage();
347 parse_zoffset(argv[3]);
348
349 dest = fopen(argv[4], "w");
350 if (!dest)
351 die("Unable to write `%s': %m", argv[4]);
352
353 /* Copy the setup code */
354 file = fopen(argv[1], "r");
355 if (!file)
356 die("Unable to open `%s': %m", argv[1]);
357 c = fread(buf, 1, sizeof(buf), file);
358 if (ferror(file))
359 die("read-error on `setup'");
360 if (c < 1024)
361 die("The setup must be at least 1024 bytes");
362 if (get_unaligned_le16(&buf[510]) != 0xAA55)
363 die("Boot block hasn't got boot flag (0xAA55)");
364 fclose(file);
365
366 c += reserve_pecoff_reloc_section(c);
367
368 /* Pad unused space with zeros */
369 setup_sectors = (c + 511) / 512;
370 if (setup_sectors < SETUP_SECT_MIN)
371 setup_sectors = SETUP_SECT_MIN;
372 i = setup_sectors*512;
373 memset(buf+c, 0, i-c);
374
375 update_pecoff_setup_and_reloc(i);
376
377 /* Set the default root device */
378 put_unaligned_le16(DEFAULT_ROOT_DEV, &buf[508]);
379
380 printf("Setup is %d bytes (padded to %d bytes).\n", c, i);
381
382 /* Open and stat the kernel file */
383 fd = open(argv[2], O_RDONLY);
384 if (fd < 0)
385 die("Unable to open `%s': %m", argv[2]);
386 if (fstat(fd, &sb))
387 die("Unable to stat `%s': %m", argv[2]);
388 sz = sb.st_size;
389 printf("System is %d kB\n", (sz+1023)/1024);
390 kernel = mmap(NULL, sz, PROT_READ, MAP_SHARED, fd, 0);
391 if (kernel == MAP_FAILED)
392 die("Unable to mmap '%s': %m", argv[2]);
393 /* Number of 16-byte paragraphs, including space for a 4-byte CRC */
394 sys_size = (sz + 15 + 4) / 16;
395#ifdef CONFIG_EFI_STUB
396 /*
397 * COFF requires minimum 32-byte alignment of sections, and
398 * adding a signature is problematic without that alignment.
399 */
400 sys_size = (sys_size + 1) & ~1;
401#endif
402
403 /* Patch the setup code with the appropriate size parameters */
404 buf[0x1f1] = setup_sectors-1;
405 put_unaligned_le32(sys_size, &buf[0x1f4]);
406
407 update_pecoff_text(setup_sectors * 512, i + (sys_size * 16));
408 init_sz = get_unaligned_le32(&buf[0x260]);
409 update_pecoff_bss(i + (sys_size * 16), init_sz);
410
411 efi_stub_entry_update();
412
413 crc = partial_crc32(buf, i, crc);
414 if (fwrite(buf, 1, i, dest) != i)
415 die("Writing setup failed");
416
417 /* Copy the kernel code */
418 crc = partial_crc32(kernel, sz, crc);
419 if (fwrite(kernel, 1, sz, dest) != sz)
420 die("Writing kernel failed");
421
422 /* Add padding leaving 4 bytes for the checksum */
423 while (sz++ < (sys_size*16) - 4) {
424 crc = partial_crc32_one('\0', crc);
425 if (fwrite("\0", 1, 1, dest) != 1)
426 die("Writing padding failed");
427 }
428
429 /* Write the CRC */
430 printf("CRC %x\n", crc);
431 put_unaligned_le32(crc, buf);
432 if (fwrite(buf, 1, 4, dest) != 4)
433 die("Writing CRC failed");
434
435 /* Catch any delayed write failures */
436 if (fclose(dest))
437 die("Writing image failed");
438
439 close(fd);
440
441 /* Everything is OK */
442 return 0;
443}
1// SPDX-License-Identifier: GPL-2.0
2/*
3 * Copyright (C) 1991, 1992 Linus Torvalds
4 * Copyright (C) 1997 Martin Mares
5 * Copyright (C) 2007 H. Peter Anvin
6 */
7
8/*
9 * This file builds a disk-image from three different files:
10 *
11 * - setup: 8086 machine code, sets up system parm
12 * - system: 80386 code for actual system
13 * - zoffset.h: header with ZO_* defines
14 *
15 * It does some checking that all files are of the correct type, and writes
16 * the result to the specified destination, removing headers and padding to
17 * the right amount. It also writes some system data to stdout.
18 */
19
20/*
21 * Changes by tytso to allow root device specification
22 * High loaded stuff by Hans Lermen & Werner Almesberger, Feb. 1996
23 * Cross compiling fixes by Gertjan van Wingerde, July 1996
24 * Rewritten by Martin Mares, April 1997
25 * Substantially overhauled by H. Peter Anvin, April 2007
26 */
27
28#include <stdio.h>
29#include <string.h>
30#include <stdlib.h>
31#include <stdarg.h>
32#include <sys/types.h>
33#include <sys/stat.h>
34#include <unistd.h>
35#include <fcntl.h>
36#include <sys/mman.h>
37#include <tools/le_byteshift.h>
38
39typedef unsigned char u8;
40typedef unsigned short u16;
41typedef unsigned int u32;
42
43/* Minimal number of setup sectors */
44#define SETUP_SECT_MIN 5
45#define SETUP_SECT_MAX 64
46
47/* This must be large enough to hold the entire setup */
48u8 buf[SETUP_SECT_MAX*512];
49
50static unsigned long _edata;
51
52/*----------------------------------------------------------------------*/
53
54static const u32 crctab32[] = {
55 0x00000000, 0x77073096, 0xee0e612c, 0x990951ba, 0x076dc419,
56 0x706af48f, 0xe963a535, 0x9e6495a3, 0x0edb8832, 0x79dcb8a4,
57 0xe0d5e91e, 0x97d2d988, 0x09b64c2b, 0x7eb17cbd, 0xe7b82d07,
58 0x90bf1d91, 0x1db71064, 0x6ab020f2, 0xf3b97148, 0x84be41de,
59 0x1adad47d, 0x6ddde4eb, 0xf4d4b551, 0x83d385c7, 0x136c9856,
60 0x646ba8c0, 0xfd62f97a, 0x8a65c9ec, 0x14015c4f, 0x63066cd9,
61 0xfa0f3d63, 0x8d080df5, 0x3b6e20c8, 0x4c69105e, 0xd56041e4,
62 0xa2677172, 0x3c03e4d1, 0x4b04d447, 0xd20d85fd, 0xa50ab56b,
63 0x35b5a8fa, 0x42b2986c, 0xdbbbc9d6, 0xacbcf940, 0x32d86ce3,
64 0x45df5c75, 0xdcd60dcf, 0xabd13d59, 0x26d930ac, 0x51de003a,
65 0xc8d75180, 0xbfd06116, 0x21b4f4b5, 0x56b3c423, 0xcfba9599,
66 0xb8bda50f, 0x2802b89e, 0x5f058808, 0xc60cd9b2, 0xb10be924,
67 0x2f6f7c87, 0x58684c11, 0xc1611dab, 0xb6662d3d, 0x76dc4190,
68 0x01db7106, 0x98d220bc, 0xefd5102a, 0x71b18589, 0x06b6b51f,
69 0x9fbfe4a5, 0xe8b8d433, 0x7807c9a2, 0x0f00f934, 0x9609a88e,
70 0xe10e9818, 0x7f6a0dbb, 0x086d3d2d, 0x91646c97, 0xe6635c01,
71 0x6b6b51f4, 0x1c6c6162, 0x856530d8, 0xf262004e, 0x6c0695ed,
72 0x1b01a57b, 0x8208f4c1, 0xf50fc457, 0x65b0d9c6, 0x12b7e950,
73 0x8bbeb8ea, 0xfcb9887c, 0x62dd1ddf, 0x15da2d49, 0x8cd37cf3,
74 0xfbd44c65, 0x4db26158, 0x3ab551ce, 0xa3bc0074, 0xd4bb30e2,
75 0x4adfa541, 0x3dd895d7, 0xa4d1c46d, 0xd3d6f4fb, 0x4369e96a,
76 0x346ed9fc, 0xad678846, 0xda60b8d0, 0x44042d73, 0x33031de5,
77 0xaa0a4c5f, 0xdd0d7cc9, 0x5005713c, 0x270241aa, 0xbe0b1010,
78 0xc90c2086, 0x5768b525, 0x206f85b3, 0xb966d409, 0xce61e49f,
79 0x5edef90e, 0x29d9c998, 0xb0d09822, 0xc7d7a8b4, 0x59b33d17,
80 0x2eb40d81, 0xb7bd5c3b, 0xc0ba6cad, 0xedb88320, 0x9abfb3b6,
81 0x03b6e20c, 0x74b1d29a, 0xead54739, 0x9dd277af, 0x04db2615,
82 0x73dc1683, 0xe3630b12, 0x94643b84, 0x0d6d6a3e, 0x7a6a5aa8,
83 0xe40ecf0b, 0x9309ff9d, 0x0a00ae27, 0x7d079eb1, 0xf00f9344,
84 0x8708a3d2, 0x1e01f268, 0x6906c2fe, 0xf762575d, 0x806567cb,
85 0x196c3671, 0x6e6b06e7, 0xfed41b76, 0x89d32be0, 0x10da7a5a,
86 0x67dd4acc, 0xf9b9df6f, 0x8ebeeff9, 0x17b7be43, 0x60b08ed5,
87 0xd6d6a3e8, 0xa1d1937e, 0x38d8c2c4, 0x4fdff252, 0xd1bb67f1,
88 0xa6bc5767, 0x3fb506dd, 0x48b2364b, 0xd80d2bda, 0xaf0a1b4c,
89 0x36034af6, 0x41047a60, 0xdf60efc3, 0xa867df55, 0x316e8eef,
90 0x4669be79, 0xcb61b38c, 0xbc66831a, 0x256fd2a0, 0x5268e236,
91 0xcc0c7795, 0xbb0b4703, 0x220216b9, 0x5505262f, 0xc5ba3bbe,
92 0xb2bd0b28, 0x2bb45a92, 0x5cb36a04, 0xc2d7ffa7, 0xb5d0cf31,
93 0x2cd99e8b, 0x5bdeae1d, 0x9b64c2b0, 0xec63f226, 0x756aa39c,
94 0x026d930a, 0x9c0906a9, 0xeb0e363f, 0x72076785, 0x05005713,
95 0x95bf4a82, 0xe2b87a14, 0x7bb12bae, 0x0cb61b38, 0x92d28e9b,
96 0xe5d5be0d, 0x7cdcefb7, 0x0bdbdf21, 0x86d3d2d4, 0xf1d4e242,
97 0x68ddb3f8, 0x1fda836e, 0x81be16cd, 0xf6b9265b, 0x6fb077e1,
98 0x18b74777, 0x88085ae6, 0xff0f6a70, 0x66063bca, 0x11010b5c,
99 0x8f659eff, 0xf862ae69, 0x616bffd3, 0x166ccf45, 0xa00ae278,
100 0xd70dd2ee, 0x4e048354, 0x3903b3c2, 0xa7672661, 0xd06016f7,
101 0x4969474d, 0x3e6e77db, 0xaed16a4a, 0xd9d65adc, 0x40df0b66,
102 0x37d83bf0, 0xa9bcae53, 0xdebb9ec5, 0x47b2cf7f, 0x30b5ffe9,
103 0xbdbdf21c, 0xcabac28a, 0x53b39330, 0x24b4a3a6, 0xbad03605,
104 0xcdd70693, 0x54de5729, 0x23d967bf, 0xb3667a2e, 0xc4614ab8,
105 0x5d681b02, 0x2a6f2b94, 0xb40bbe37, 0xc30c8ea1, 0x5a05df1b,
106 0x2d02ef8d
107};
108
109static u32 partial_crc32_one(u8 c, u32 crc)
110{
111 return crctab32[(crc ^ c) & 0xff] ^ (crc >> 8);
112}
113
114static u32 partial_crc32(const u8 *s, int len, u32 crc)
115{
116 while (len--)
117 crc = partial_crc32_one(*s++, crc);
118 return crc;
119}
120
121static void die(const char * str, ...)
122{
123 va_list args;
124 va_start(args, str);
125 vfprintf(stderr, str, args);
126 va_end(args);
127 fputc('\n', stderr);
128 exit(1);
129}
130
131static void usage(void)
132{
133 die("Usage: build setup system zoffset.h image");
134}
135
136/*
137 * Parse zoffset.h and find the entry points. We could just #include zoffset.h
138 * but that would mean tools/build would have to be rebuilt every time. It's
139 * not as if parsing it is hard...
140 */
141#define PARSE_ZOFS(p, sym) do { \
142 if (!strncmp(p, "#define ZO_" #sym " ", 11+sizeof(#sym))) \
143 sym = strtoul(p + 11 + sizeof(#sym), NULL, 16); \
144} while (0)
145
146static void parse_zoffset(char *fname)
147{
148 FILE *file;
149 char *p;
150 int c;
151
152 file = fopen(fname, "r");
153 if (!file)
154 die("Unable to open `%s': %m", fname);
155 c = fread(buf, 1, sizeof(buf) - 1, file);
156 if (ferror(file))
157 die("read-error on `zoffset.h'");
158 fclose(file);
159 buf[c] = 0;
160
161 p = (char *)buf;
162
163 while (p && *p) {
164 PARSE_ZOFS(p, _edata);
165
166 p = strchr(p, '\n');
167 while (p && (*p == '\r' || *p == '\n'))
168 p++;
169 }
170}
171
172int main(int argc, char ** argv)
173{
174 unsigned int i, sz, setup_sectors;
175 int c;
176 struct stat sb;
177 FILE *file, *dest;
178 int fd;
179 void *kernel;
180 u32 crc = 0xffffffffUL;
181
182 if (argc != 5)
183 usage();
184 parse_zoffset(argv[3]);
185
186 dest = fopen(argv[4], "w");
187 if (!dest)
188 die("Unable to write `%s': %m", argv[4]);
189
190 /* Copy the setup code */
191 file = fopen(argv[1], "r");
192 if (!file)
193 die("Unable to open `%s': %m", argv[1]);
194 c = fread(buf, 1, sizeof(buf), file);
195 if (ferror(file))
196 die("read-error on `setup'");
197 if (c < 1024)
198 die("The setup must be at least 1024 bytes");
199 if (get_unaligned_le16(&buf[510]) != 0xAA55)
200 die("Boot block hasn't got boot flag (0xAA55)");
201 fclose(file);
202
203 /* Pad unused space with zeros */
204 setup_sectors = (c + 4095) / 4096;
205 setup_sectors *= 8;
206 if (setup_sectors < SETUP_SECT_MIN)
207 setup_sectors = SETUP_SECT_MIN;
208 i = setup_sectors*512;
209 memset(buf+c, 0, i-c);
210
211 /* Open and stat the kernel file */
212 fd = open(argv[2], O_RDONLY);
213 if (fd < 0)
214 die("Unable to open `%s': %m", argv[2]);
215 if (fstat(fd, &sb))
216 die("Unable to stat `%s': %m", argv[2]);
217 if (_edata != sb.st_size)
218 die("Unexpected file size `%s': %u != %u", argv[2], _edata,
219 sb.st_size);
220 sz = _edata - 4;
221 kernel = mmap(NULL, sz, PROT_READ, MAP_SHARED, fd, 0);
222 if (kernel == MAP_FAILED)
223 die("Unable to mmap '%s': %m", argv[2]);
224
225 crc = partial_crc32(buf, i, crc);
226 if (fwrite(buf, 1, i, dest) != i)
227 die("Writing setup failed");
228
229 /* Copy the kernel code */
230 crc = partial_crc32(kernel, sz, crc);
231 if (fwrite(kernel, 1, sz, dest) != sz)
232 die("Writing kernel failed");
233
234 /* Write the CRC */
235 put_unaligned_le32(crc, buf);
236 if (fwrite(buf, 1, 4, dest) != 4)
237 die("Writing CRC failed");
238
239 /* Catch any delayed write failures */
240 if (fclose(dest))
241 die("Writing image failed");
242
243 close(fd);
244
245 /* Everything is OK */
246 return 0;
247}