1*479ab7f0SSascha Wildner /*
2*479ab7f0SSascha Wildner * Copyright (c) 1998 Robert Nordier
3*479ab7f0SSascha Wildner * All rights reserved.
4*479ab7f0SSascha Wildner *
5*479ab7f0SSascha Wildner * Redistribution and use in source and binary forms, with or without
6*479ab7f0SSascha Wildner * modification, are permitted provided that the following conditions
7*479ab7f0SSascha Wildner * are met:
8*479ab7f0SSascha Wildner * 1. Redistributions of source code must retain the above copyright
9*479ab7f0SSascha Wildner * notice, this list of conditions and the following disclaimer.
10*479ab7f0SSascha Wildner * 2. Redistributions in binary form must reproduce the above copyright
11*479ab7f0SSascha Wildner * notice, this list of conditions and the following disclaimer in the
12*479ab7f0SSascha Wildner * documentation and/or other materials provided with the distribution.
13*479ab7f0SSascha Wildner *
14*479ab7f0SSascha Wildner * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS``AS IS'' AND
15*479ab7f0SSascha Wildner * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16*479ab7f0SSascha Wildner * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
17*479ab7f0SSascha Wildner * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS
18*479ab7f0SSascha Wildner * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
19*479ab7f0SSascha Wildner * OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT
20*479ab7f0SSascha Wildner * OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
21*479ab7f0SSascha Wildner * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
22*479ab7f0SSascha Wildner * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
23*479ab7f0SSascha Wildner * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
24*479ab7f0SSascha Wildner * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25*479ab7f0SSascha Wildner *
26*479ab7f0SSascha Wildner * $FreeBSD: src/usr.sbin/btxld/btxld.c,v 1.4 2000/01/04 14:10:36 marcel Exp $
27*479ab7f0SSascha Wildner */
28*479ab7f0SSascha Wildner
29*479ab7f0SSascha Wildner #define AOUT_H_FORCE32
30*479ab7f0SSascha Wildner #include <sys/param.h>
31*479ab7f0SSascha Wildner #include <sys/stat.h>
32*479ab7f0SSascha Wildner #include <sys/mman.h>
33*479ab7f0SSascha Wildner
34*479ab7f0SSascha Wildner #include <err.h>
35*479ab7f0SSascha Wildner #include <errno.h>
36*479ab7f0SSascha Wildner #include <fcntl.h>
37*479ab7f0SSascha Wildner #include <stdarg.h>
38*479ab7f0SSascha Wildner #include <stdio.h>
39*479ab7f0SSascha Wildner #include <stdlib.h>
40*479ab7f0SSascha Wildner #include <string.h>
41*479ab7f0SSascha Wildner #include <unistd.h>
42*479ab7f0SSascha Wildner
43*479ab7f0SSascha Wildner #include <a.out.h>
44*479ab7f0SSascha Wildner
45*479ab7f0SSascha Wildner #include "btx.h"
46*479ab7f0SSascha Wildner #include "elfh.h"
47*479ab7f0SSascha Wildner
48*479ab7f0SSascha Wildner #define BTX_PATH "/stand/boot/i386/btx"
49*479ab7f0SSascha Wildner
50*479ab7f0SSascha Wildner #define I_LDR 0 /* BTX loader */
51*479ab7f0SSascha Wildner #define I_BTX 1 /* BTX kernel */
52*479ab7f0SSascha Wildner #define I_CLNT 2 /* Client program */
53*479ab7f0SSascha Wildner
54*479ab7f0SSascha Wildner #define F_BIN 0 /* Binary */
55*479ab7f0SSascha Wildner #define F_AOUT 1 /* ZMAGIC a.out */
56*479ab7f0SSascha Wildner #define F_ELF 2 /* 32-bit ELF */
57*479ab7f0SSascha Wildner #define F_CNT 3 /* Number of formats */
58*479ab7f0SSascha Wildner
59*479ab7f0SSascha Wildner #define IMPURE 1 /* Writable text */
60*479ab7f0SSascha Wildner #define MAXU32 0xffffffff /* Maximum unsigned 32-bit quantity */
61*479ab7f0SSascha Wildner
62*479ab7f0SSascha Wildner struct hdr {
63*479ab7f0SSascha Wildner uint32_t fmt; /* Format */
64*479ab7f0SSascha Wildner uint32_t flags; /* Bit flags */
65*479ab7f0SSascha Wildner uint32_t size; /* Size of file */
66*479ab7f0SSascha Wildner uint32_t text; /* Size of text segment */
67*479ab7f0SSascha Wildner uint32_t data; /* Size of data segment */
68*479ab7f0SSascha Wildner uint32_t bss; /* Size of bss segment */
69*479ab7f0SSascha Wildner uint32_t org; /* Program origin */
70*479ab7f0SSascha Wildner uint32_t entry; /* Program entry point */
71*479ab7f0SSascha Wildner };
72*479ab7f0SSascha Wildner
73*479ab7f0SSascha Wildner static const char *const fmtlist[] = {"bin", "aout", "elf"};
74*479ab7f0SSascha Wildner
75*479ab7f0SSascha Wildner static const char binfo[] =
76*479ab7f0SSascha Wildner "kernel: ver=%u.%02u size=%x load=%x entry=%x map=%uM "
77*479ab7f0SSascha Wildner "pgctl=%x:%x\n";
78*479ab7f0SSascha Wildner static const char cinfo[] =
79*479ab7f0SSascha Wildner "client: fmt=%s size=%x text=%x data=%x bss=%x entry=%x\n";
80*479ab7f0SSascha Wildner static const char oinfo[] =
81*479ab7f0SSascha Wildner "output: fmt=%s size=%x text=%x data=%x org=%x entry=%x\n";
82*479ab7f0SSascha Wildner
83*479ab7f0SSascha Wildner static const char *lname =
84*479ab7f0SSascha Wildner BTX_PATH "/btxldr/btxldr"; /* BTX loader */
85*479ab7f0SSascha Wildner static const char *bname =
86*479ab7f0SSascha Wildner BTX_PATH "/btx/btx"; /* BTX kernel */
87*479ab7f0SSascha Wildner static const char *oname =
88*479ab7f0SSascha Wildner "a.out"; /* Output filename */
89*479ab7f0SSascha Wildner
90*479ab7f0SSascha Wildner static int ppage = -1; /* First page present */
91*479ab7f0SSascha Wildner static int wpage = -1; /* First page writable */
92*479ab7f0SSascha Wildner
93*479ab7f0SSascha Wildner static unsigned int format; /* Output format */
94*479ab7f0SSascha Wildner
95*479ab7f0SSascha Wildner static uint32_t centry; /* Client entry address */
96*479ab7f0SSascha Wildner static uint32_t lentry; /* Loader entry address */
97*479ab7f0SSascha Wildner
98*479ab7f0SSascha Wildner static int Eflag; /* Client entry option */
99*479ab7f0SSascha Wildner
100*479ab7f0SSascha Wildner static int quiet; /* Inhibit warnings */
101*479ab7f0SSascha Wildner static int verbose; /* Display information */
102*479ab7f0SSascha Wildner
103*479ab7f0SSascha Wildner static const char *tname; /* Temporary output file */
104*479ab7f0SSascha Wildner static const char *fname; /* Current input file */
105*479ab7f0SSascha Wildner
106*479ab7f0SSascha Wildner static void cleanup(void);
107*479ab7f0SSascha Wildner static void btxld(const char *);
108*479ab7f0SSascha Wildner static void getbtx(int, struct btx_hdr *);
109*479ab7f0SSascha Wildner static void gethdr(int, struct hdr *);
110*479ab7f0SSascha Wildner static void puthdr(int, struct hdr *);
111*479ab7f0SSascha Wildner static void copy(int, int, size_t, off_t);
112*479ab7f0SSascha Wildner static size_t readx(int, void *, size_t, off_t);
113*479ab7f0SSascha Wildner static void writex(int, const void *, size_t);
114*479ab7f0SSascha Wildner static void seekx(int, off_t);
115*479ab7f0SSascha Wildner static unsigned int optfmt(const char *);
116*479ab7f0SSascha Wildner static uint32_t optaddr(const char *);
117*479ab7f0SSascha Wildner static int optpage(const char *, int);
118*479ab7f0SSascha Wildner static void Warn(const char *, const char *, ...) __printflike(2, 3);
119*479ab7f0SSascha Wildner static void usage(void);
120*479ab7f0SSascha Wildner
121*479ab7f0SSascha Wildner /*
122*479ab7f0SSascha Wildner * A link editor for BTX clients.
123*479ab7f0SSascha Wildner */
124*479ab7f0SSascha Wildner int
main(int argc,char * argv[])125*479ab7f0SSascha Wildner main(int argc, char *argv[])
126*479ab7f0SSascha Wildner {
127*479ab7f0SSascha Wildner int c;
128*479ab7f0SSascha Wildner
129*479ab7f0SSascha Wildner while ((c = getopt(argc, argv, "qvb:E:e:f:l:o:P:W:")) != -1)
130*479ab7f0SSascha Wildner switch (c) {
131*479ab7f0SSascha Wildner case 'q':
132*479ab7f0SSascha Wildner quiet = 1;
133*479ab7f0SSascha Wildner break;
134*479ab7f0SSascha Wildner case 'v':
135*479ab7f0SSascha Wildner verbose = 1;
136*479ab7f0SSascha Wildner break;
137*479ab7f0SSascha Wildner case 'b':
138*479ab7f0SSascha Wildner bname = optarg;
139*479ab7f0SSascha Wildner break;
140*479ab7f0SSascha Wildner case 'E':
141*479ab7f0SSascha Wildner centry = optaddr(optarg);
142*479ab7f0SSascha Wildner Eflag = 1;
143*479ab7f0SSascha Wildner break;
144*479ab7f0SSascha Wildner case 'e':
145*479ab7f0SSascha Wildner lentry = optaddr(optarg);
146*479ab7f0SSascha Wildner break;
147*479ab7f0SSascha Wildner case 'f':
148*479ab7f0SSascha Wildner format = optfmt(optarg);
149*479ab7f0SSascha Wildner break;
150*479ab7f0SSascha Wildner case 'l':
151*479ab7f0SSascha Wildner lname = optarg;
152*479ab7f0SSascha Wildner break;
153*479ab7f0SSascha Wildner case 'o':
154*479ab7f0SSascha Wildner oname = optarg;
155*479ab7f0SSascha Wildner break;
156*479ab7f0SSascha Wildner case 'P':
157*479ab7f0SSascha Wildner ppage = optpage(optarg, 1);
158*479ab7f0SSascha Wildner break;
159*479ab7f0SSascha Wildner case 'W':
160*479ab7f0SSascha Wildner wpage = optpage(optarg, BTX_MAXCWR);
161*479ab7f0SSascha Wildner break;
162*479ab7f0SSascha Wildner default:
163*479ab7f0SSascha Wildner usage();
164*479ab7f0SSascha Wildner }
165*479ab7f0SSascha Wildner argc -= optind;
166*479ab7f0SSascha Wildner argv += optind;
167*479ab7f0SSascha Wildner if (argc != 1)
168*479ab7f0SSascha Wildner usage();
169*479ab7f0SSascha Wildner atexit(cleanup);
170*479ab7f0SSascha Wildner btxld(*argv);
171*479ab7f0SSascha Wildner return 0;
172*479ab7f0SSascha Wildner }
173*479ab7f0SSascha Wildner
174*479ab7f0SSascha Wildner /*
175*479ab7f0SSascha Wildner * Clean up after errors.
176*479ab7f0SSascha Wildner */
177*479ab7f0SSascha Wildner static void
cleanup(void)178*479ab7f0SSascha Wildner cleanup(void)
179*479ab7f0SSascha Wildner {
180*479ab7f0SSascha Wildner if (tname)
181*479ab7f0SSascha Wildner remove(tname);
182*479ab7f0SSascha Wildner }
183*479ab7f0SSascha Wildner
184*479ab7f0SSascha Wildner /*
185*479ab7f0SSascha Wildner * Read the input files; write the output file; display information.
186*479ab7f0SSascha Wildner */
187*479ab7f0SSascha Wildner static void
btxld(const char * iname)188*479ab7f0SSascha Wildner btxld(const char *iname)
189*479ab7f0SSascha Wildner {
190*479ab7f0SSascha Wildner char name[FILENAME_MAX];
191*479ab7f0SSascha Wildner struct btx_hdr btx;
192*479ab7f0SSascha Wildner struct hdr ihdr, ohdr;
193*479ab7f0SSascha Wildner unsigned int ldr_size, cwr;
194*479ab7f0SSascha Wildner int fdi[3], fdo, i;
195*479ab7f0SSascha Wildner
196*479ab7f0SSascha Wildner ldr_size = 0;
197*479ab7f0SSascha Wildner
198*479ab7f0SSascha Wildner for (i = I_LDR; i <= I_CLNT; i++) {
199*479ab7f0SSascha Wildner fname = i == I_LDR ? lname : i == I_BTX ? bname : iname;
200*479ab7f0SSascha Wildner if ((fdi[i] = open(fname, O_RDONLY)) == -1)
201*479ab7f0SSascha Wildner err(2, "%s", fname);
202*479ab7f0SSascha Wildner switch (i) {
203*479ab7f0SSascha Wildner case I_LDR:
204*479ab7f0SSascha Wildner gethdr(fdi[i], &ihdr);
205*479ab7f0SSascha Wildner if (ihdr.fmt != F_BIN)
206*479ab7f0SSascha Wildner Warn(fname, "Loader format is %s; processing as %s",
207*479ab7f0SSascha Wildner fmtlist[ihdr.fmt], fmtlist[F_BIN]);
208*479ab7f0SSascha Wildner ldr_size = ihdr.size;
209*479ab7f0SSascha Wildner break;
210*479ab7f0SSascha Wildner case I_BTX:
211*479ab7f0SSascha Wildner getbtx(fdi[i], &btx);
212*479ab7f0SSascha Wildner break;
213*479ab7f0SSascha Wildner case I_CLNT:
214*479ab7f0SSascha Wildner gethdr(fdi[i], &ihdr);
215*479ab7f0SSascha Wildner if (ihdr.org && ihdr.org != BTX_PGSIZE)
216*479ab7f0SSascha Wildner Warn(fname,
217*479ab7f0SSascha Wildner "Client origin is 0x%x; expecting 0 or 0x%x",
218*479ab7f0SSascha Wildner ihdr.org, BTX_PGSIZE);
219*479ab7f0SSascha Wildner }
220*479ab7f0SSascha Wildner }
221*479ab7f0SSascha Wildner memset(&ohdr, 0, sizeof(ohdr));
222*479ab7f0SSascha Wildner ohdr.fmt = format;
223*479ab7f0SSascha Wildner ohdr.text = ldr_size;
224*479ab7f0SSascha Wildner ohdr.data = btx.btx_textsz + ihdr.size;
225*479ab7f0SSascha Wildner ohdr.org = lentry;
226*479ab7f0SSascha Wildner ohdr.entry = lentry;
227*479ab7f0SSascha Wildner cwr = 0;
228*479ab7f0SSascha Wildner if (wpage > 0 || (wpage == -1 && !(ihdr.flags & IMPURE))) {
229*479ab7f0SSascha Wildner if (wpage > 0)
230*479ab7f0SSascha Wildner cwr = wpage;
231*479ab7f0SSascha Wildner else {
232*479ab7f0SSascha Wildner cwr = howmany(ihdr.text, BTX_PGSIZE);
233*479ab7f0SSascha Wildner if (cwr > BTX_MAXCWR)
234*479ab7f0SSascha Wildner cwr = BTX_MAXCWR;
235*479ab7f0SSascha Wildner }
236*479ab7f0SSascha Wildner }
237*479ab7f0SSascha Wildner if (ppage > 0 || (ppage && wpage && ihdr.org >= BTX_PGSIZE)) {
238*479ab7f0SSascha Wildner btx.btx_flags |= BTX_MAPONE;
239*479ab7f0SSascha Wildner if (!cwr)
240*479ab7f0SSascha Wildner cwr++;
241*479ab7f0SSascha Wildner }
242*479ab7f0SSascha Wildner btx.btx_pgctl -= cwr;
243*479ab7f0SSascha Wildner btx.btx_entry = Eflag ? centry : ihdr.entry;
244*479ab7f0SSascha Wildner if ((size_t)snprintf(name, sizeof(name), "%s.tmp", oname) >= sizeof(name))
245*479ab7f0SSascha Wildner errx(2, "%s: Filename too long", oname);
246*479ab7f0SSascha Wildner if ((fdo = open(name, O_CREAT | O_TRUNC | O_WRONLY, 0666)) == -1)
247*479ab7f0SSascha Wildner err(2, "%s", name);
248*479ab7f0SSascha Wildner if (!(tname = strdup(name)))
249*479ab7f0SSascha Wildner err(2, NULL);
250*479ab7f0SSascha Wildner puthdr(fdo, &ohdr);
251*479ab7f0SSascha Wildner for (i = I_LDR; i <= I_CLNT; i++) {
252*479ab7f0SSascha Wildner fname = i == I_LDR ? lname : i == I_BTX ? bname : iname;
253*479ab7f0SSascha Wildner switch (i) {
254*479ab7f0SSascha Wildner case I_LDR:
255*479ab7f0SSascha Wildner copy(fdi[i], fdo, ldr_size, 0);
256*479ab7f0SSascha Wildner seekx(fdo, ohdr.size += ohdr.text);
257*479ab7f0SSascha Wildner break;
258*479ab7f0SSascha Wildner case I_BTX:
259*479ab7f0SSascha Wildner writex(fdo, &btx, sizeof(btx));
260*479ab7f0SSascha Wildner copy(fdi[i], fdo, btx.btx_textsz - sizeof(btx),
261*479ab7f0SSascha Wildner sizeof(btx));
262*479ab7f0SSascha Wildner break;
263*479ab7f0SSascha Wildner case I_CLNT:
264*479ab7f0SSascha Wildner copy(fdi[i], fdo, ihdr.size, 0);
265*479ab7f0SSascha Wildner if (ftruncate(fdo, ohdr.size += ohdr.data))
266*479ab7f0SSascha Wildner err(2, "%s", tname);
267*479ab7f0SSascha Wildner }
268*479ab7f0SSascha Wildner if (close(fdi[i]))
269*479ab7f0SSascha Wildner err(2, "%s", fname);
270*479ab7f0SSascha Wildner }
271*479ab7f0SSascha Wildner if (close(fdo))
272*479ab7f0SSascha Wildner err(2, "%s", tname);
273*479ab7f0SSascha Wildner if (rename(tname, oname))
274*479ab7f0SSascha Wildner err(2, "%s: Can't rename to %s", tname, oname);
275*479ab7f0SSascha Wildner tname = NULL;
276*479ab7f0SSascha Wildner if (verbose) {
277*479ab7f0SSascha Wildner printf(binfo, btx.btx_majver, btx.btx_minver, btx.btx_textsz,
278*479ab7f0SSascha Wildner BTX_ORIGIN(btx), BTX_ENTRY(btx), BTX_MAPPED(btx) *
279*479ab7f0SSascha Wildner BTX_PGSIZE / 0x100000, !!(btx.btx_flags & BTX_MAPONE),
280*479ab7f0SSascha Wildner BTX_MAPPED(btx) - btx.btx_pgctl - BTX_PGBASE /
281*479ab7f0SSascha Wildner BTX_PGSIZE - BTX_MAPPED(btx) * 4 / BTX_PGSIZE);
282*479ab7f0SSascha Wildner printf(cinfo, fmtlist[ihdr.fmt], ihdr.size, ihdr.text,
283*479ab7f0SSascha Wildner ihdr.data, ihdr.bss, ihdr.entry);
284*479ab7f0SSascha Wildner printf(oinfo, fmtlist[ohdr.fmt], ohdr.size, ohdr.text,
285*479ab7f0SSascha Wildner ohdr.data, ohdr.org, ohdr.entry);
286*479ab7f0SSascha Wildner }
287*479ab7f0SSascha Wildner }
288*479ab7f0SSascha Wildner
289*479ab7f0SSascha Wildner /*
290*479ab7f0SSascha Wildner * Read BTX file header.
291*479ab7f0SSascha Wildner */
292*479ab7f0SSascha Wildner static void
getbtx(int fd,struct btx_hdr * btx)293*479ab7f0SSascha Wildner getbtx(int fd, struct btx_hdr * btx)
294*479ab7f0SSascha Wildner {
295*479ab7f0SSascha Wildner if (readx(fd, btx, sizeof(*btx), 0) != sizeof(*btx) ||
296*479ab7f0SSascha Wildner btx->btx_magic[0] != BTX_MAG0 ||
297*479ab7f0SSascha Wildner btx->btx_magic[1] != BTX_MAG1 ||
298*479ab7f0SSascha Wildner btx->btx_magic[2] != BTX_MAG2)
299*479ab7f0SSascha Wildner errx(1, "%s: Not a BTX kernel", fname);
300*479ab7f0SSascha Wildner }
301*479ab7f0SSascha Wildner
302*479ab7f0SSascha Wildner /*
303*479ab7f0SSascha Wildner * Get file size and read a.out or ELF header.
304*479ab7f0SSascha Wildner */
305*479ab7f0SSascha Wildner static void
gethdr(int fd,struct hdr * hdr)306*479ab7f0SSascha Wildner gethdr(int fd, struct hdr *hdr)
307*479ab7f0SSascha Wildner {
308*479ab7f0SSascha Wildner struct stat sb;
309*479ab7f0SSascha Wildner const struct exec *ex;
310*479ab7f0SSascha Wildner const Elf32_Ehdr *ee;
311*479ab7f0SSascha Wildner const Elf32_Phdr *ep;
312*479ab7f0SSascha Wildner void *p;
313*479ab7f0SSascha Wildner unsigned int fmt, x, n, i;
314*479ab7f0SSascha Wildner
315*479ab7f0SSascha Wildner memset(hdr, 0, sizeof(*hdr));
316*479ab7f0SSascha Wildner if (fstat(fd, &sb))
317*479ab7f0SSascha Wildner err(2, "%s", fname);
318*479ab7f0SSascha Wildner if (sb.st_size > MAXU32)
319*479ab7f0SSascha Wildner errx(1, "%s: Too big", fname);
320*479ab7f0SSascha Wildner hdr->size = sb.st_size;
321*479ab7f0SSascha Wildner if ((p = mmap(NULL, hdr->size, PROT_READ, MAP_SHARED, fd,
322*479ab7f0SSascha Wildner 0)) == MAP_FAILED)
323*479ab7f0SSascha Wildner err(2, "%s", fname);
324*479ab7f0SSascha Wildner for (fmt = F_CNT - 1; !hdr->fmt && fmt; fmt--)
325*479ab7f0SSascha Wildner switch (fmt) {
326*479ab7f0SSascha Wildner case F_AOUT:
327*479ab7f0SSascha Wildner ex = p;
328*479ab7f0SSascha Wildner if (hdr->size >= sizeof(struct exec) && !N_BADMAG(*ex)) {
329*479ab7f0SSascha Wildner hdr->fmt = fmt;
330*479ab7f0SSascha Wildner x = N_GETMAGIC(*ex);
331*479ab7f0SSascha Wildner if (x == OMAGIC || x == NMAGIC) {
332*479ab7f0SSascha Wildner if (x == NMAGIC)
333*479ab7f0SSascha Wildner Warn(fname, "Treating %s NMAGIC as OMAGIC",
334*479ab7f0SSascha Wildner fmtlist[fmt]);
335*479ab7f0SSascha Wildner hdr->flags |= IMPURE;
336*479ab7f0SSascha Wildner }
337*479ab7f0SSascha Wildner hdr->text = ex->a_text;
338*479ab7f0SSascha Wildner hdr->data = ex->a_data;
339*479ab7f0SSascha Wildner hdr->bss = ex->a_bss;
340*479ab7f0SSascha Wildner hdr->entry = ex->a_entry;
341*479ab7f0SSascha Wildner if (ex->a_entry >= BTX_PGSIZE)
342*479ab7f0SSascha Wildner hdr->org = BTX_PGSIZE;
343*479ab7f0SSascha Wildner }
344*479ab7f0SSascha Wildner break;
345*479ab7f0SSascha Wildner case F_ELF:
346*479ab7f0SSascha Wildner ee = p;
347*479ab7f0SSascha Wildner if (hdr->size >= sizeof(Elf32_Ehdr) && IS_ELF(*ee)) {
348*479ab7f0SSascha Wildner hdr->fmt = fmt;
349*479ab7f0SSascha Wildner for (n = i = 0; i < ee->e_phnum; i++) {
350*479ab7f0SSascha Wildner ep = (void *)((uint8_t *)p + ee->e_phoff +
351*479ab7f0SSascha Wildner ee->e_phentsize * i);
352*479ab7f0SSascha Wildner if (ep->p_type == PT_LOAD)
353*479ab7f0SSascha Wildner switch (n++) {
354*479ab7f0SSascha Wildner case 0:
355*479ab7f0SSascha Wildner hdr->text = ep->p_filesz;
356*479ab7f0SSascha Wildner hdr->org = ep->p_paddr;
357*479ab7f0SSascha Wildner if (ep->p_flags & PF_W)
358*479ab7f0SSascha Wildner hdr->flags |= IMPURE;
359*479ab7f0SSascha Wildner break;
360*479ab7f0SSascha Wildner case 1:
361*479ab7f0SSascha Wildner hdr->data = ep->p_filesz;
362*479ab7f0SSascha Wildner hdr->bss = ep->p_memsz - ep->p_filesz;
363*479ab7f0SSascha Wildner break;
364*479ab7f0SSascha Wildner case 2:
365*479ab7f0SSascha Wildner Warn(fname,
366*479ab7f0SSascha Wildner "Ignoring extra %s PT_LOAD segments",
367*479ab7f0SSascha Wildner fmtlist[fmt]);
368*479ab7f0SSascha Wildner }
369*479ab7f0SSascha Wildner }
370*479ab7f0SSascha Wildner hdr->entry = ee->e_entry;
371*479ab7f0SSascha Wildner }
372*479ab7f0SSascha Wildner }
373*479ab7f0SSascha Wildner if (munmap(p, hdr->size))
374*479ab7f0SSascha Wildner err(2, "%s", fname);
375*479ab7f0SSascha Wildner }
376*479ab7f0SSascha Wildner
377*479ab7f0SSascha Wildner /*
378*479ab7f0SSascha Wildner * Write a.out or ELF header.
379*479ab7f0SSascha Wildner */
380*479ab7f0SSascha Wildner static void
puthdr(int fd,struct hdr * hdr)381*479ab7f0SSascha Wildner puthdr(int fd, struct hdr *hdr)
382*479ab7f0SSascha Wildner {
383*479ab7f0SSascha Wildner struct exec ex;
384*479ab7f0SSascha Wildner struct elfh eh;
385*479ab7f0SSascha Wildner
386*479ab7f0SSascha Wildner switch (hdr->fmt) {
387*479ab7f0SSascha Wildner case F_AOUT:
388*479ab7f0SSascha Wildner memset(&ex, 0, sizeof(ex));
389*479ab7f0SSascha Wildner N_SETMAGIC(ex, ZMAGIC, MID_ZERO, 0);
390*479ab7f0SSascha Wildner hdr->text = N_ALIGN(ex, hdr->text);
391*479ab7f0SSascha Wildner ex.a_text = hdr->text;
392*479ab7f0SSascha Wildner hdr->data = N_ALIGN(ex, hdr->data);
393*479ab7f0SSascha Wildner ex.a_data = hdr->data;
394*479ab7f0SSascha Wildner ex.a_entry = hdr->entry;
395*479ab7f0SSascha Wildner writex(fd, &ex, sizeof(ex));
396*479ab7f0SSascha Wildner hdr->size = N_ALIGN(ex, sizeof(ex));
397*479ab7f0SSascha Wildner seekx(fd, hdr->size);
398*479ab7f0SSascha Wildner break;
399*479ab7f0SSascha Wildner case F_ELF:
400*479ab7f0SSascha Wildner eh = elfhdr;
401*479ab7f0SSascha Wildner eh.e.e_entry = hdr->entry;
402*479ab7f0SSascha Wildner eh.p[0].p_vaddr = eh.p[0].p_paddr = hdr->org;
403*479ab7f0SSascha Wildner eh.p[0].p_filesz = eh.p[0].p_memsz = hdr->text;
404*479ab7f0SSascha Wildner eh.p[1].p_offset = eh.p[0].p_offset + eh.p[0].p_filesz;
405*479ab7f0SSascha Wildner eh.p[1].p_vaddr = eh.p[1].p_paddr =
406*479ab7f0SSascha Wildner roundup2(eh.p[0].p_paddr + eh.p[0].p_memsz, 4);
407*479ab7f0SSascha Wildner eh.p[1].p_filesz = eh.p[1].p_memsz = hdr->data;
408*479ab7f0SSascha Wildner eh.sh[2].sh_addr = eh.p[0].p_vaddr;
409*479ab7f0SSascha Wildner eh.sh[2].sh_offset = eh.p[0].p_offset;
410*479ab7f0SSascha Wildner eh.sh[2].sh_size = eh.p[0].p_filesz;
411*479ab7f0SSascha Wildner eh.sh[3].sh_addr = eh.p[1].p_vaddr;
412*479ab7f0SSascha Wildner eh.sh[3].sh_offset = eh.p[1].p_offset;
413*479ab7f0SSascha Wildner eh.sh[3].sh_size = eh.p[1].p_filesz;
414*479ab7f0SSascha Wildner writex(fd, &eh, sizeof(eh));
415*479ab7f0SSascha Wildner hdr->size = sizeof(eh);
416*479ab7f0SSascha Wildner }
417*479ab7f0SSascha Wildner }
418*479ab7f0SSascha Wildner
419*479ab7f0SSascha Wildner /*
420*479ab7f0SSascha Wildner * Safe copy from input file to output file.
421*479ab7f0SSascha Wildner */
422*479ab7f0SSascha Wildner static void
copy(int fdi,int fdo,size_t nbyte,off_t offset)423*479ab7f0SSascha Wildner copy(int fdi, int fdo, size_t nbyte, off_t offset)
424*479ab7f0SSascha Wildner {
425*479ab7f0SSascha Wildner char buf[8192];
426*479ab7f0SSascha Wildner size_t n;
427*479ab7f0SSascha Wildner
428*479ab7f0SSascha Wildner while (nbyte) {
429*479ab7f0SSascha Wildner if ((n = sizeof(buf)) > nbyte)
430*479ab7f0SSascha Wildner n = nbyte;
431*479ab7f0SSascha Wildner if (readx(fdi, buf, n, offset) != n)
432*479ab7f0SSascha Wildner errx(2, "%s: Short read", fname);
433*479ab7f0SSascha Wildner writex(fdo, buf, n);
434*479ab7f0SSascha Wildner nbyte -= n;
435*479ab7f0SSascha Wildner offset = -1;
436*479ab7f0SSascha Wildner }
437*479ab7f0SSascha Wildner }
438*479ab7f0SSascha Wildner
439*479ab7f0SSascha Wildner /*
440*479ab7f0SSascha Wildner * Safe read from input file.
441*479ab7f0SSascha Wildner */
442*479ab7f0SSascha Wildner static size_t
readx(int fd,void * buf,size_t nbyte,off_t offset)443*479ab7f0SSascha Wildner readx(int fd, void *buf, size_t nbyte, off_t offset)
444*479ab7f0SSascha Wildner {
445*479ab7f0SSascha Wildner ssize_t n;
446*479ab7f0SSascha Wildner
447*479ab7f0SSascha Wildner if (offset != -1 && lseek(fd, offset, SEEK_SET) != offset)
448*479ab7f0SSascha Wildner err(2, "%s", fname);
449*479ab7f0SSascha Wildner if ((n = read(fd, buf, nbyte)) == -1)
450*479ab7f0SSascha Wildner err(2, "%s", fname);
451*479ab7f0SSascha Wildner return n;
452*479ab7f0SSascha Wildner }
453*479ab7f0SSascha Wildner
454*479ab7f0SSascha Wildner /*
455*479ab7f0SSascha Wildner * Safe write to output file.
456*479ab7f0SSascha Wildner */
457*479ab7f0SSascha Wildner static void
writex(int fd,const void * buf,size_t nbyte)458*479ab7f0SSascha Wildner writex(int fd, const void *buf, size_t nbyte)
459*479ab7f0SSascha Wildner {
460*479ab7f0SSascha Wildner ssize_t n;
461*479ab7f0SSascha Wildner
462*479ab7f0SSascha Wildner if ((n = write(fd, buf, nbyte)) == -1)
463*479ab7f0SSascha Wildner err(2, "%s", tname);
464*479ab7f0SSascha Wildner if ((size_t)n != nbyte)
465*479ab7f0SSascha Wildner errx(2, "%s: Short write", tname);
466*479ab7f0SSascha Wildner }
467*479ab7f0SSascha Wildner
468*479ab7f0SSascha Wildner /*
469*479ab7f0SSascha Wildner * Safe seek in output file.
470*479ab7f0SSascha Wildner */
471*479ab7f0SSascha Wildner static void
seekx(int fd,off_t offset)472*479ab7f0SSascha Wildner seekx(int fd, off_t offset)
473*479ab7f0SSascha Wildner {
474*479ab7f0SSascha Wildner if (lseek(fd, offset, SEEK_SET) != offset)
475*479ab7f0SSascha Wildner err(2, "%s", tname);
476*479ab7f0SSascha Wildner }
477*479ab7f0SSascha Wildner
478*479ab7f0SSascha Wildner /*
479*479ab7f0SSascha Wildner * Convert an option argument to a format code.
480*479ab7f0SSascha Wildner */
481*479ab7f0SSascha Wildner static unsigned int
optfmt(const char * arg)482*479ab7f0SSascha Wildner optfmt(const char *arg)
483*479ab7f0SSascha Wildner {
484*479ab7f0SSascha Wildner unsigned int i;
485*479ab7f0SSascha Wildner
486*479ab7f0SSascha Wildner for (i = 0; i < F_CNT && strcmp(arg, fmtlist[i]); i++);
487*479ab7f0SSascha Wildner if (i == F_CNT)
488*479ab7f0SSascha Wildner errx(1, "%s: Unknown format", arg);
489*479ab7f0SSascha Wildner return i;
490*479ab7f0SSascha Wildner }
491*479ab7f0SSascha Wildner
492*479ab7f0SSascha Wildner /*
493*479ab7f0SSascha Wildner * Convert an option argument to an address.
494*479ab7f0SSascha Wildner */
495*479ab7f0SSascha Wildner static uint32_t
optaddr(const char * arg)496*479ab7f0SSascha Wildner optaddr(const char *arg)
497*479ab7f0SSascha Wildner {
498*479ab7f0SSascha Wildner char *s;
499*479ab7f0SSascha Wildner unsigned long x;
500*479ab7f0SSascha Wildner
501*479ab7f0SSascha Wildner errno = 0;
502*479ab7f0SSascha Wildner x = strtoul(arg, &s, 0);
503*479ab7f0SSascha Wildner if (errno || !*arg || *s || x > MAXU32)
504*479ab7f0SSascha Wildner errx(1, "%s: Illegal address", arg);
505*479ab7f0SSascha Wildner return x;
506*479ab7f0SSascha Wildner }
507*479ab7f0SSascha Wildner
508*479ab7f0SSascha Wildner /*
509*479ab7f0SSascha Wildner * Convert an option argument to a page number.
510*479ab7f0SSascha Wildner */
511*479ab7f0SSascha Wildner static int
optpage(const char * arg,int hi)512*479ab7f0SSascha Wildner optpage(const char *arg, int hi)
513*479ab7f0SSascha Wildner {
514*479ab7f0SSascha Wildner char *s;
515*479ab7f0SSascha Wildner long x;
516*479ab7f0SSascha Wildner
517*479ab7f0SSascha Wildner errno = 0;
518*479ab7f0SSascha Wildner x = strtol(arg, &s, 0);
519*479ab7f0SSascha Wildner if (errno || !*arg || *s || x < 0 || x > hi)
520*479ab7f0SSascha Wildner errx(1, "%s: Illegal page number", arg);
521*479ab7f0SSascha Wildner return x;
522*479ab7f0SSascha Wildner }
523*479ab7f0SSascha Wildner
524*479ab7f0SSascha Wildner /*
525*479ab7f0SSascha Wildner * Display a warning.
526*479ab7f0SSascha Wildner */
527*479ab7f0SSascha Wildner static void
Warn(const char * locus,const char * fmt,...)528*479ab7f0SSascha Wildner Warn(const char *locus, const char *fmt, ...)
529*479ab7f0SSascha Wildner {
530*479ab7f0SSascha Wildner va_list ap;
531*479ab7f0SSascha Wildner char *s;
532*479ab7f0SSascha Wildner
533*479ab7f0SSascha Wildner if (!quiet) {
534*479ab7f0SSascha Wildner asprintf(&s, "%s: Warning: %s", locus, fmt);
535*479ab7f0SSascha Wildner va_start(ap, fmt);
536*479ab7f0SSascha Wildner vwarnx(s, ap);
537*479ab7f0SSascha Wildner va_end(ap);
538*479ab7f0SSascha Wildner free(s);
539*479ab7f0SSascha Wildner }
540*479ab7f0SSascha Wildner }
541*479ab7f0SSascha Wildner
542*479ab7f0SSascha Wildner /*
543*479ab7f0SSascha Wildner * Display usage information.
544*479ab7f0SSascha Wildner */
545*479ab7f0SSascha Wildner static void
usage(void)546*479ab7f0SSascha Wildner usage(void)
547*479ab7f0SSascha Wildner {
548*479ab7f0SSascha Wildner fprintf(stderr, "%s\n%s\n",
549*479ab7f0SSascha Wildner "usage: btxld [-qv] [-b file] [-E address] [-e address] [-f format]",
550*479ab7f0SSascha Wildner " [-l file] [-o filename] [-P page] [-W page] file");
551*479ab7f0SSascha Wildner exit(1);
552*479ab7f0SSascha Wildner }
553