mkboot.c revision 1.5 1 /* $NetBSD: mkboot.c,v 1.5 2024/10/31 01:21:12 gutteridge Exp $ */
2
3 /* $OpenBSD: mkboot.c,v 1.9 2001/05/17 00:57:55 pvalchev Exp $ */
4
5 /*
6 * Copyright (c) 1990, 1993
7 * The Regents of the University of California. All rights reserved.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions
11 * are met:
12 * 1. Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
14 * 2. Redistributions in binary form must reproduce the above copyright
15 * notice, this list of conditions and the following disclaimer in the
16 * documentation and/or other materials provided with the distribution.
17 * 3. Neither the name of the University nor the names of its contributors
18 * may be used to endorse or promote products derived from this software
19 * without specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31 * SUCH DAMAGE.
32 *
33 * @(#)mkboot.c 8.1 (Berkeley) 7/15/93
34 */
35
36 #if 0
37 #ifndef lint
38 static char copyright[] =
39 "@(#) Copyright (c) 1990, 1993\n\
40 The Regents of the University of California. All rights reserved.\n";
41 #endif /* not lint */
42
43 #ifndef lint
44 static char rcsid[] = "$OpenBSD: mkboot.c,v 1.9 2001/05/17 00:57:55 pvalchev Exp $";
45 #endif /* not lint */
46 #endif
47
48 #if HAVE_NBTOOL_CONFIG_H
49 #include "nbtool_config.h"
50 #include "../../sys/sys/bootblock.h"
51 #else
52 #include <sys/bootblock.h>
53 #endif
54
55 #include <sys/param.h>
56 #include <sys/endian.h>
57 #include <sys/file.h>
58 #include <sys/stat.h>
59 #include <string.h>
60 #include <stdlib.h>
61 #include <unistd.h>
62 #include <time.h>
63 #include <err.h>
64
65 /* BFD ELF headers */
66 #include <elf/common.h>
67 #include <elf/external.h>
68
69 #define IS_ELF(ehdr) ((ehdr).e_ident[EI_MAG0] == ELFMAG0 && \
70 (ehdr).e_ident[EI_MAG1] == ELFMAG1 && \
71 (ehdr).e_ident[EI_MAG2] == ELFMAG2 && \
72 (ehdr).e_ident[EI_MAG3] == ELFMAG3)
73
74 /*
75 * Macros to get values from multi-byte ELF header fields. These assume
76 * a big-endian image.
77 */
78 #define ELFGET16(x) (((x)[0] << 8) | (x)[1])
79
80 #define ELFGET32(x) (((x)[0] << 24) | ((x)[1] << 16) | \
81 ((x)[2] << 8) | (x)[3])
82
83 /*
84 * Header prepended to each a.out file.
85 */
86 struct exec {
87 u_long a_midmag; /* htobe32(flags<<26 | mid<<16 | magic) */
88 u_long a_text; /* text segment size */
89 u_long a_data; /* initialized data size */
90 u_long a_bss; /* uninitialized data size */
91 u_long a_syms; /* symbol table size */
92 u_long a_entry; /* entry point */
93 u_long a_trsize; /* text relocation size */
94 u_long a_drsize; /* data relocation size */
95 };
96
97 /* a_magic */
98 #define OMAGIC 0407 /* old impure format */
99 #define NMAGIC 0410 /* read-only text */
100 #define ZMAGIC 0413 /* demand load format */
101 #define QMAGIC 0314 /* "compact" demand load format; deprecated */
102
103 #define N_GETMAGIC(ex) \
104 ((((ex).a_midmag)&0xffff0000) ? \
105 (be32toh((uint32_t)((ex).a_midmag))&0xffff) : ((ex).a_midmag))
106
107 #include <stdio.h>
108 #include <ctype.h>
109
110 int putfile(char *, int);
111 void __dead usage(void);
112 void bcddate(char *, char *);
113 char *lifname(char *);
114 int cksum(int, int *, int);
115
116 char *to_file;
117 int loadpoint, verbose;
118 u_long entry;
119 time_t repro_epoch = 0;
120
121 /*
122 * Old Format:
123 * sector 0: LIF volume header (40 bytes)
124 * sector 1: <unused>
125 * sector 2: LIF directory (8 x 32 == 256 bytes)
126 * sector 3-: LIF file 0, LIF file 1, etc.
127 * where sectors are 256 bytes.
128 *
129 * New Format:
130 * sector 0: LIF volume header (40 bytes)
131 * sector 1: <unused>
132 * sector 2: LIF directory (8 x 32 == 256 bytes)
133 * sector 3: <unused>
134 * sector 4-31: disklabel (~300 bytes right now)
135 * sector 32-: LIF file 0, LIF file 1, etc.
136 */
137 int
138 main(int argc, char **argv)
139 {
140 int to, n, pos, c;
141 char buf[HPPA_LIF_FILESTART];
142 struct hppa_lifvol *lifv = (struct hppa_lifvol *)buf;
143 struct hppa_lifdir *lifd = (struct hppa_lifdir *)(buf + HPPA_LIF_DIRSTART);
144
145 while ((c = getopt(argc, argv, "l:t:v")) != -1) {
146 switch (c) {
147 case 'l':
148 loadpoint = strtol(optarg, NULL, 0);
149 break;
150 case 't':
151 repro_epoch = atoll(optarg);
152 break;
153 case 'v':
154 verbose++;
155 break;
156 default:
157 usage();
158 }
159 }
160 if (argc - optind < 2)
161 usage();
162 else if (argc - optind > 8)
163 errx(1, "too many boot programs (max 8 supported)");
164
165 to_file = argv[--argc];
166 if ((to = open(to_file, O_RDWR | O_TRUNC | O_CREAT, 0644)) < 0)
167 err(1, "%s: open", to_file);
168
169 memset(buf, 0, sizeof(buf));
170
171 /* record volume info */
172 lifv->vol_id = htobe16(HPPA_LIF_VOL_ID);
173 strncpy(lifv->vol_label, "MKBOOT", 6);
174 lifv->vol_addr = htobe32(hppa_btolifs(HPPA_LIF_DIRSTART));
175 lifv->vol_oct = htobe16(HPPA_LIF_VOL_OCT);
176 lifv->vol_dirsize = htobe32(hppa_btolifs(HPPA_LIF_DIRSIZE));
177 lifv->vol_version = htobe16(1);
178 lifv->vol_number = htobe32(1);
179 lifv->vol_lastvol = htobe32(1);
180 lifv->vol_length = HPPA_LIF_FILESTART; /* ... so far. */
181 bcddate(to_file, lifv->vol_toc);
182 lifv->ipl_addr = htobe32(HPPA_LIF_FILESTART);
183
184 argv += optind;
185 argc -= optind;
186 optind = 0;
187 for (pos = HPPA_LIF_FILESTART; optind < argc; optind++) {
188
189 /* output bootfile */
190 if (lseek(to, pos, SEEK_SET) < 0)
191 err(1, "%s: lseek", to_file);
192 lifd[optind].dir_addr = htobe32(hppa_btolifs(pos));
193 n = hppa_btolifs(putfile(argv[optind], to));
194 if (lifv->ipl_entry == 0) {
195 lifv->ipl_entry = htobe32(loadpoint + entry);
196 lifv->ipl_size = htobe32(hppa_lifstob(n));
197 lifd[optind].dir_type = htobe16(HPPA_LIF_DIR_ISL);
198 lifd[optind].dir_implement = 0;
199 } else {
200 lifd[optind].dir_type = htobe16(HPPA_LIF_DIR_TYPE);
201 lifd[optind].dir_implement = htobe32(loadpoint + entry);
202 }
203
204 memcpy(lifd[optind].dir_name, lifname(argv[optind]),
205 sizeof(lifd[optind].dir_name));
206 lifd[optind].dir_length = htobe32(n);
207 bcddate(argv[optind], lifd[optind].dir_toc);
208 lifd[optind].dir_flag = htobe16(HPPA_LIF_DIR_FLAG);
209
210 lifv->vol_length += n;
211 pos += hppa_lifstob(n);
212 }
213
214 /* terminate the directory */
215 lifd[optind].dir_type = htobe16(0xffff);
216
217 /* byte-swap the length now that we're done computing it */
218 lifv->vol_length = htobe32(lifv->vol_length);
219
220 /* output volume/directory header info */
221 if (lseek(to, HPPA_LIF_VOLSTART, SEEK_SET) < 0)
222 err(1, "%s: lseek", to_file);
223 if (write(to, buf, sizeof(buf)) != sizeof(buf))
224 err(1, "%s: write LIF volume", to_file);
225 lseek(to, 0, SEEK_END);
226
227 if (close(to) < 0)
228 err(1, "%s", to_file);
229
230 return(0);
231 }
232
233 int
234 putfile(char *from_file, int to)
235 {
236 struct exec ex;
237 char buf[2048];
238 int n, total;
239 int from, check_sum = 0;
240 struct hppa_lifload load;
241 Elf32_External_Ehdr elf_header;
242 Elf32_External_Phdr *elf_segments = NULL;
243 int i, header_count, memory_needed, elf_load_image_segment;
244
245 if ((from = open(from_file, O_RDONLY)) < 0)
246 err(1, "%s", from_file);
247
248 n = read(from, &ex, sizeof(ex));
249 if (n != sizeof(ex))
250 err(1, "%s: reading file header", from_file);
251
252 entry = ex.a_entry;
253 if (N_GETMAGIC(ex) == OMAGIC || N_GETMAGIC(ex) == NMAGIC)
254 entry += sizeof(ex);
255 else if (IS_ELF(*(Elf32_External_Ehdr *)&ex)) {
256
257 if (lseek(from, 0, SEEK_SET) < 0)
258 err(1, "lseek");
259 n = read(from, &elf_header, sizeof (elf_header));
260 if (n != sizeof (elf_header))
261 err(1, "%s: reading ELF header", from_file);
262 header_count = ELFGET16(elf_header.e_phnum);
263 memory_needed = header_count * sizeof (Elf32_External_Phdr);
264 elf_segments = malloc(memory_needed);
265 if (elf_segments == NULL)
266 err(1, "malloc");
267 if (lseek(from, ELFGET32(elf_header.e_phoff), SEEK_SET) < 0)
268 err(1, "lseek");
269 n = read(from, elf_segments, memory_needed);
270 if (n != memory_needed)
271 err(1, "%s: reading ELF segments", from_file);
272 elf_load_image_segment = -1;
273 for (i = 0; i < header_count; i++) {
274 if (ELFGET32(elf_segments[i].p_filesz) &&
275 ELFGET32(elf_segments[i].p_flags) & PF_X) {
276 if (elf_load_image_segment != -1)
277 errx(1, "%s: more than one ELF program "
278 "segment", from_file);
279 elf_load_image_segment = i;
280 }
281 }
282 if (elf_load_image_segment == -1)
283 errx(1, "%s: no suitable ELF program segment",
284 from_file);
285 entry = ELFGET32(elf_header.e_entry) +
286 ELFGET32(elf_segments[elf_load_image_segment].p_offset) -
287 ELFGET32(elf_segments[elf_load_image_segment].p_vaddr);
288 } else if (*(uint8_t *)&ex == 0x1f && ((uint8_t *)&ex)[1] == 0x8b) {
289 entry = 0;
290 } else
291 errx(1, "%s: bad magic number", from_file);
292
293 entry += sizeof(load);
294 lseek(to, sizeof(load), SEEK_CUR);
295
296 total = 0;
297 n = sizeof(buf) - sizeof(load);
298 /* copy the whole file */
299 for (lseek(from, 0, SEEK_SET); ; n = sizeof(buf)) {
300 memset(buf, 0, sizeof(buf));
301 if ((n = read(from, buf, n)) < 0)
302 err(1, "%s", from_file);
303 else if (n == 0)
304 break;
305
306 if (write(to, buf, n) != n)
307 err(1, "%s", to_file);
308
309 total += n;
310 check_sum = cksum(check_sum, (int *)buf, n);
311 }
312
313 /* load header */
314 load.address = htobe32(loadpoint + sizeof(load));
315 load.count = htobe32(4 + total);
316 check_sum = cksum(check_sum, (int *)&load, sizeof(load));
317
318 if (verbose)
319 warnx("wrote %d bytes of file \'%s\'", total, from_file);
320
321 total += sizeof(load);
322 /* insert the header */
323 lseek(to, -total, SEEK_CUR);
324 if (write(to, &load, sizeof(load)) != sizeof(load))
325 err(1, "%s", to_file);
326 lseek(to, total - sizeof(load), SEEK_CUR);
327
328 memset(buf, 0, sizeof(buf));
329 /* pad to int */
330 n = sizeof(int) - total % sizeof(int);
331 if (total % sizeof(int)) {
332 if (write(to, buf, n) != n)
333 err(1, "%s", to_file);
334 else
335 total += n;
336 }
337
338 /* pad to the blocksize */
339 n = sizeof(buf) - total % sizeof(buf);
340
341 if (n < sizeof(int)) {
342 n += sizeof(buf);
343 total += sizeof(buf);
344 } else
345 total += n;
346
347 /* TODO should pad here to the 65k boundary for tape boot */
348
349 if (verbose)
350 warnx("checksum is 0x%08x", -check_sum);
351
352 check_sum = htobe32(-check_sum);
353 if (write(to, &check_sum, sizeof(int)) != sizeof(int))
354 err(1, "%s", to_file);
355
356 n -= sizeof(int);
357
358 if (write(to, buf, n) != n)
359 err(1, "%s", to_file);
360
361 if (close(from) < 0)
362 err(1, "%s", from_file);
363
364 free(elf_segments);
365 return total;
366 }
367
368 int
369 cksum(int ck, int *p, int size)
370 {
371 /* we assume size is int-aligned */
372 for (size = (size + sizeof(int) - 1) / sizeof(int); size--; p++ )
373 ck += be32toh(*p);
374
375 return ck;
376 }
377
378 void __dead
379 usage(void)
380 {
381 fprintf(stderr,
382 "Usage: %s [-v] [-l <loadpoint>] [-t <timestamp>] prog1 {progN} outfile\n",
383 getprogname());
384 exit(1);
385 }
386
387 char *
388 lifname(char *str)
389 {
390 static char lname[10] = "XXXXXXXXXX";
391 char *cp;
392 int i;
393
394 cp = strrchr(str, '/');
395 if (cp != NULL) {
396 str = cp + 1;
397 }
398 for (i = 0; i < 9; i++) {
399 if (islower(*str))
400 lname[i] = toupper(*str);
401 else if (isalnum(*str) || *str == '_')
402 lname[i] = *str;
403 else
404 break;
405 str++;
406 }
407 for ( ; i < 10; i++)
408 lname[i] = ' ';
409 return(lname);
410 }
411
412
413 void
414 bcddate(char *file, char *toc)
415 {
416 struct stat statb;
417 struct tm *tm;
418
419 if (repro_epoch)
420 tm = gmtime(&repro_epoch);
421 else {
422 stat(file, &statb);
423 tm = localtime(&statb.st_ctime);
424 }
425 tm->tm_year %= 100;
426 *toc = (tm->tm_year / 10) << 4;
427 *toc++ |= tm->tm_year % 10;
428 *toc = ((tm->tm_mon+1) / 10) << 4;
429 *toc++ |= (tm->tm_mon+1) % 10;
430 *toc = (tm->tm_mday / 10) << 4;
431 *toc++ |= tm->tm_mday % 10;
432 *toc = (tm->tm_hour / 10) << 4;
433 *toc++ |= tm->tm_hour % 10;
434 *toc = (tm->tm_min / 10) << 4;
435 *toc++ |= tm->tm_min % 10;
436 *toc = (tm->tm_sec / 10) << 4;
437 *toc |= tm->tm_sec % 10;
438 }
439