2 * Copyright (c) 1998 Robert Nordier
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
17 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS
18 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
19 * OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT
20 * OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
21 * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
22 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
23 * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
24 * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28 static const char rcsid[] =
32 #include <sys/param.h>
33 #include <sys/endian.h>
37 /* XXX make this work as an i386/amd64 cross-tool */
38 #include <machine/exec.h>
42 #include <netinet/in.h>
57 #define BTX_PATH "/sys/boot/i386/btx"
59 #define I_LDR 0 /* BTX loader */
60 #define I_BTX 1 /* BTX kernel */
61 #define I_CLNT 2 /* Client program */
63 #define F_BIN 0 /* Binary */
64 #define F_AOUT 1 /* ZMAGIC a.out */
65 #define F_ELF 2 /* 32-bit ELF */
66 #define F_CNT 3 /* Number of formats */
68 #define IMPURE 1 /* Writable text */
69 #define MAXU32 0xffffffff /* Maximum unsigned 32-bit quantity */
72 uint32_t fmt; /* Format */
73 uint32_t flags; /* Bit flags */
74 uint32_t size; /* Size of file */
75 uint32_t text; /* Size of text segment */
76 uint32_t data; /* Size of data segment */
77 uint32_t bss; /* Size of bss segment */
78 uint32_t org; /* Program origin */
79 uint32_t entry; /* Program entry point */
82 static const char *const fmtlist[] = {"bin", "aout", "elf"};
84 static const char binfo[] =
85 "kernel: ver=%u.%02u size=%x load=%x entry=%x map=%uM "
87 static const char cinfo[] =
88 "client: fmt=%s size=%x text=%x data=%x bss=%x entry=%x\n";
89 static const char oinfo[] =
90 "output: fmt=%s size=%x text=%x data=%x org=%x entry=%x\n";
92 static const char *lname =
93 BTX_PATH "/btxldr/btxldr"; /* BTX loader */
94 static const char *bname =
95 BTX_PATH "/btx/btx"; /* BTX kernel */
96 static const char *oname =
97 "a.out"; /* Output filename */
99 static int ppage = -1; /* First page present */
100 static int wpage = -1; /* First page writable */
102 static unsigned int format; /* Output format */
104 static uint32_t centry; /* Client entry address */
105 static uint32_t lentry; /* Loader entry address */
107 static int Eflag; /* Client entry option */
109 static int quiet; /* Inhibit warnings */
110 static int verbose; /* Display information */
112 static const char *tname; /* Temporary output file */
113 static const char *fname; /* Current input file */
115 static void cleanup(void);
116 static void btxld(const char *);
117 static void getbtx(int, struct btx_hdr *);
118 static void gethdr(int, struct hdr *);
119 static void puthdr(int, struct hdr *);
120 static void copy(int, int, size_t, off_t);
121 static size_t readx(int, void *, size_t, off_t);
122 static void writex(int, const void *, size_t);
123 static void seekx(int, off_t);
124 static unsigned int optfmt(const char *);
125 static uint32_t optaddr(const char *);
126 static int optpage(const char *, int);
127 static void Warn(const char *, const char *, ...);
128 static void usage(void);
131 * A link editor for BTX clients.
134 main(int argc, char *argv[])
138 while ((c = getopt(argc, argv, "qvb:E:e:f:l:o:P:W:")) != -1)
150 centry = optaddr(optarg);
154 lentry = optaddr(optarg);
157 format = optfmt(optarg);
166 ppage = optpage(optarg, 1);
169 wpage = optpage(optarg, BTX_MAXCWR);
184 * Clean up after errors.
194 * Read the input files; write the output file; display information.
197 btxld(const char *iname)
199 char name[FILENAME_MAX];
200 struct btx_hdr btx, btxle;
201 struct hdr ihdr, ohdr;
202 unsigned int ldr_size, cwr;
207 for (i = I_LDR; i <= I_CLNT; i++) {
208 fname = i == I_LDR ? lname : i == I_BTX ? bname : iname;
209 if ((fdi[i] = open(fname, O_RDONLY)) == -1)
213 gethdr(fdi[i], &ihdr);
214 if (ihdr.fmt != F_BIN)
215 Warn(fname, "Loader format is %s; processing as %s",
216 fmtlist[ihdr.fmt], fmtlist[F_BIN]);
217 ldr_size = ihdr.size;
220 getbtx(fdi[i], &btx);
223 gethdr(fdi[i], &ihdr);
224 if (ihdr.org && ihdr.org != BTX_PGSIZE)
226 "Client origin is 0x%x; expecting 0 or 0x%x",
227 ihdr.org, BTX_PGSIZE);
230 memset(&ohdr, 0, sizeof(ohdr));
232 ohdr.text = ldr_size;
233 ohdr.data = btx.btx_textsz + ihdr.size;
237 if (wpage > 0 || (wpage == -1 && !(ihdr.flags & IMPURE))) {
241 cwr = howmany(ihdr.text, BTX_PGSIZE);
242 if (cwr > BTX_MAXCWR)
246 if (ppage > 0 || (ppage && wpage && ihdr.org >= BTX_PGSIZE)) {
247 btx.btx_flags |= BTX_MAPONE;
251 btx.btx_pgctl -= cwr;
252 btx.btx_entry = Eflag ? centry : ihdr.entry;
253 if ((size_t)snprintf(name, sizeof(name), "%s.tmp", oname) >= sizeof(name))
254 errx(2, "%s: Filename too long", oname);
255 if ((fdo = open(name, O_CREAT | O_TRUNC | O_WRONLY, 0666)) == -1)
257 if (!(tname = strdup(name)))
260 for (i = I_LDR; i <= I_CLNT; i++) {
261 fname = i == I_LDR ? lname : i == I_BTX ? bname : iname;
264 copy(fdi[i], fdo, ldr_size, 0);
265 seekx(fdo, ohdr.size += ohdr.text);
269 btxle.btx_pgctl = htole16(btxle.btx_pgctl);
270 btxle.btx_textsz = htole16(btxle.btx_textsz);
271 btxle.btx_entry = htole32(btxle.btx_entry);
272 writex(fdo, &btxle, sizeof(btxle));
273 copy(fdi[i], fdo, btx.btx_textsz - sizeof(btx),
277 copy(fdi[i], fdo, ihdr.size, 0);
278 if (ftruncate(fdo, ohdr.size += ohdr.data))
286 if (rename(tname, oname))
287 err(2, "%s: Can't rename to %s", tname, oname);
290 printf(binfo, btx.btx_majver, btx.btx_minver, btx.btx_textsz,
291 BTX_ORIGIN(btx), BTX_ENTRY(btx), BTX_MAPPED(btx) *
292 BTX_PGSIZE / 0x100000, !!(btx.btx_flags & BTX_MAPONE),
293 BTX_MAPPED(btx) - btx.btx_pgctl - BTX_PGBASE /
294 BTX_PGSIZE - BTX_MAPPED(btx) * 4 / BTX_PGSIZE);
295 printf(cinfo, fmtlist[ihdr.fmt], ihdr.size, ihdr.text,
296 ihdr.data, ihdr.bss, ihdr.entry);
297 printf(oinfo, fmtlist[ohdr.fmt], ohdr.size, ohdr.text,
298 ohdr.data, ohdr.org, ohdr.entry);
303 * Read BTX file header.
306 getbtx(int fd, struct btx_hdr * btx)
308 if (readx(fd, btx, sizeof(*btx), 0) != sizeof(*btx) ||
309 btx->btx_magic[0] != BTX_MAG0 ||
310 btx->btx_magic[1] != BTX_MAG1 ||
311 btx->btx_magic[2] != BTX_MAG2)
312 errx(1, "%s: Not a BTX kernel", fname);
313 btx->btx_pgctl = le16toh(btx->btx_pgctl);
314 btx->btx_textsz = le16toh(btx->btx_textsz);
315 btx->btx_entry = le32toh(btx->btx_entry);
319 * Get file size and read a.out or ELF header.
322 gethdr(int fd, struct hdr *hdr)
325 const struct exec *ex;
326 const Elf32_Ehdr *ee;
327 const Elf32_Phdr *ep;
329 unsigned int fmt, x, n, i;
331 memset(hdr, 0, sizeof(*hdr));
334 if (sb.st_size > MAXU32)
335 errx(1, "%s: Too big", fname);
336 hdr->size = sb.st_size;
339 if ((p = mmap(NULL, hdr->size, PROT_READ, MAP_SHARED, fd,
342 for (fmt = F_CNT - 1; !hdr->fmt && fmt; fmt--)
346 if (hdr->size >= sizeof(struct exec) && !N_BADMAG(*ex)) {
349 if (x == OMAGIC || x == NMAGIC) {
351 Warn(fname, "Treating %s NMAGIC as OMAGIC",
353 hdr->flags |= IMPURE;
355 hdr->text = le32toh(ex->a_text);
356 hdr->data = le32toh(ex->a_data);
357 hdr->bss = le32toh(ex->a_bss);
358 hdr->entry = le32toh(ex->a_entry);
359 if (le32toh(ex->a_entry) >= BTX_PGSIZE)
360 hdr->org = BTX_PGSIZE;
365 if (hdr->size >= sizeof(Elf32_Ehdr) && IS_ELF(*ee)) {
367 for (n = i = 0; i < le16toh(ee->e_phnum); i++) {
368 ep = (void *)((uint8_t *)p + le32toh(ee->e_phoff) +
369 le16toh(ee->e_phentsize) * i);
370 if (le32toh(ep->p_type) == PT_LOAD)
373 hdr->text = le32toh(ep->p_filesz);
374 hdr->org = le32toh(ep->p_paddr);
375 if (le32toh(ep->p_flags) & PF_W)
376 hdr->flags |= IMPURE;
379 hdr->data = le32toh(ep->p_filesz);
380 hdr->bss = le32toh(ep->p_memsz) -
381 le32toh(ep->p_filesz);
385 "Ignoring extra %s PT_LOAD segments",
389 hdr->entry = le32toh(ee->e_entry);
392 if (munmap(p, hdr->size))
397 * Write a.out or ELF header.
400 puthdr(int fd, struct hdr *hdr)
407 memset(&ex, 0, sizeof(ex));
408 N_SETMAGIC(ex, ZMAGIC, MID_I386, 0);
409 hdr->text = N_ALIGN(ex, hdr->text);
410 ex.a_text = htole32(hdr->text);
411 hdr->data = N_ALIGN(ex, hdr->data);
412 ex.a_data = htole32(hdr->data);
413 ex.a_entry = htole32(hdr->entry);
414 writex(fd, &ex, sizeof(ex));
415 hdr->size = N_ALIGN(ex, sizeof(ex));
416 seekx(fd, hdr->size);
420 eh.e.e_entry = htole32(hdr->entry);
421 eh.p[0].p_vaddr = eh.p[0].p_paddr = htole32(hdr->org);
422 eh.p[0].p_filesz = eh.p[0].p_memsz = htole32(hdr->text);
423 eh.p[1].p_offset = htole32(le32toh(eh.p[0].p_offset) +
424 le32toh(eh.p[0].p_filesz));
425 eh.p[1].p_vaddr = eh.p[1].p_paddr =
426 htole32(roundup2(le32toh(eh.p[0].p_paddr) + le32toh(eh.p[0].p_memsz),
428 eh.p[1].p_filesz = eh.p[1].p_memsz = htole32(hdr->data);
429 eh.sh[2].sh_addr = eh.p[0].p_vaddr;
430 eh.sh[2].sh_offset = eh.p[0].p_offset;
431 eh.sh[2].sh_size = eh.p[0].p_filesz;
432 eh.sh[3].sh_addr = eh.p[1].p_vaddr;
433 eh.sh[3].sh_offset = eh.p[1].p_offset;
434 eh.sh[3].sh_size = eh.p[1].p_filesz;
435 writex(fd, &eh, sizeof(eh));
436 hdr->size = sizeof(eh);
441 * Safe copy from input file to output file.
444 copy(int fdi, int fdo, size_t nbyte, off_t offset)
450 if ((n = sizeof(buf)) > nbyte)
452 if (readx(fdi, buf, n, offset) != n)
453 errx(2, "%s: Short read", fname);
461 * Safe read from input file.
464 readx(int fd, void *buf, size_t nbyte, off_t offset)
468 if (offset != -1 && lseek(fd, offset, SEEK_SET) != offset)
470 if ((n = read(fd, buf, nbyte)) == -1)
476 * Safe write to output file.
479 writex(int fd, const void *buf, size_t nbyte)
483 if ((n = write(fd, buf, nbyte)) == -1)
485 if ((size_t)n != nbyte)
486 errx(2, "%s: Short write", tname);
490 * Safe seek in output file.
493 seekx(int fd, off_t offset)
495 if (lseek(fd, offset, SEEK_SET) != offset)
500 * Convert an option argument to a format code.
503 optfmt(const char *arg)
507 for (i = 0; i < F_CNT && strcmp(arg, fmtlist[i]); i++);
509 errx(1, "%s: Unknown format", arg);
514 * Convert an option argument to an address.
517 optaddr(const char *arg)
523 x = strtoul(arg, &s, 0);
524 if (errno || !*arg || *s || x > MAXU32)
525 errx(1, "%s: Illegal address", arg);
530 * Convert an option argument to a page number.
533 optpage(const char *arg, int hi)
539 x = strtol(arg, &s, 0);
540 if (errno || !*arg || *s || x < 0 || x > hi)
541 errx(1, "%s: Illegal page number", arg);
549 Warn(const char *locus, const char *fmt, ...)
555 asprintf(&s, "%s: Warning: %s", locus, fmt);
564 * Display usage information.
569 fprintf(stderr, "%s\n%s\n",
570 "usage: btxld [-qv] [-b file] [-E address] [-e address] [-f format]",
571 " [-l file] [-o filename] [-P page] [-W page] file");