Home | History | Annotate | Line # | Download | only in mkboot
      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