Annotation of src/usr.bin/strip/strip.c, Revision 1.18
1.17 espie 1: /* $OpenBSD: strip.c,v 1.16 2002/02/16 21:27:53 millert Exp $ */
1.2 deraadt 2:
1.1 deraadt 3: /*
4: * Copyright (c) 1988 Regents of the University of California.
5: * All rights reserved.
6: *
7: * Redistribution and use in source and binary forms, with or without
8: * modification, are permitted provided that the following conditions
9: * are met:
10: * 1. Redistributions of source code must retain the above copyright
11: * notice, this list of conditions and the following disclaimer.
12: * 2. Redistributions in binary form must reproduce the above copyright
13: * notice, this list of conditions and the following disclaimer in the
14: * documentation and/or other materials provided with the distribution.
15: * 3. All advertising materials mentioning features or use of this software
16: * must display the following acknowledgement:
17: * This product includes software developed by the University of
18: * California, Berkeley and its contributors.
19: * 4. Neither the name of the University nor the names of its contributors
20: * may be used to endorse or promote products derived from this software
21: * without specific prior written permission.
22: *
23: * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
24: * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25: * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26: * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
27: * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28: * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29: * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30: * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31: * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32: * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33: * SUCH DAMAGE.
34: */
35:
36: #ifndef lint
37: char copyright[] =
38: "@(#) Copyright (c) 1988 Regents of the University of California.\n\
39: All rights reserved.\n";
40: #endif /* not lint */
41:
42: #ifndef lint
43: /*static char sccsid[] = "from: @(#)strip.c 5.8 (Berkeley) 11/6/91";*/
1.17 espie 44: static char rcsid[] = "$OpenBSD: strip.c,v 1.16 2002/02/16 21:27:53 millert Exp $";
1.1 deraadt 45: #endif /* not lint */
46:
1.4 deraadt 47: #include <sys/param.h>
1.1 deraadt 48: #include <sys/types.h>
49: #include <sys/stat.h>
50: #include <sys/mman.h>
51: #include <fcntl.h>
52: #include <errno.h>
53: #include <a.out.h>
54: #include <unistd.h>
55: #include <stdio.h>
56: #include <stdlib.h>
57: #include <string.h>
1.7 mickey 58: #include <err.h>
1.11 espie 59: #include <ranlib.h>
60: #include "byte.c"
1.10 niklas 61:
62: #ifdef MID_MACHINE_OVERRIDE
63: #undef MID_MACHINE
64: #define MID_MACHINE MID_MACHINE_OVERRIDE
65: #endif
1.1 deraadt 66:
67: typedef struct exec EXEC;
68: typedef struct nlist NLIST;
69:
70: #define strx n_un.n_strx
71:
1.17 espie 72: int s_stab(const char *, int, EXEC *, struct stat *, off_t *);
73: int s_sym(const char *, int, EXEC *, struct stat *, off_t *);
1.16 millert 74: void usage(void);
1.1 deraadt 75:
76: int xflag = 0;
77:
1.9 deraadt 78: int
1.1 deraadt 79: main(argc, argv)
80: int argc;
81: char *argv[];
82: {
1.15 mpech 83: int fd;
1.1 deraadt 84: EXEC *ep;
85: struct stat sb;
1.17 espie 86: int (*sfcn)(const char *, int, EXEC *, struct stat *, off_t *);
1.1 deraadt 87: int ch, errors;
88: char *fn;
1.17 espie 89: off_t newsize;
1.1 deraadt 90:
91: sfcn = s_sym;
1.6 millert 92: while ((ch = getopt(argc, argv, "dx")) != -1)
1.1 deraadt 93: switch(ch) {
94: case 'x':
95: xflag = 1;
96: /*FALLTHROUGH*/
97: case 'd':
98: sfcn = s_stab;
99: break;
100: case '?':
101: default:
102: usage();
103: }
104: argc -= optind;
105: argv += optind;
106:
107: errors = 0;
1.8 mickey 108: #define ERROR(x) errors |= 1; warnx("%s: %s", fn, strerror(x)); continue;
1.9 deraadt 109: while ((fn = *argv++)) {
1.1 deraadt 110: if ((fd = open(fn, O_RDWR)) < 0) {
111: ERROR(errno);
112: }
113: if (fstat(fd, &sb)) {
114: (void)close(fd);
115: ERROR(errno);
116: }
117: if (sb.st_size < sizeof(EXEC)) {
118: (void)close(fd);
119: ERROR(EFTYPE);
120: }
121: if ((ep = (EXEC *)mmap(NULL, sb.st_size, PROT_READ|PROT_WRITE,
1.13 art 122: MAP_SHARED, fd, (off_t)0)) == MAP_FAILED) {
1.1 deraadt 123: (void)close(fd);
124: ERROR(errno);
125: }
1.11 espie 126: if (BAD_OBJECT(*ep)) {
1.1 deraadt 127: munmap((caddr_t)ep, sb.st_size);
128: (void)close(fd);
129: ERROR(EFTYPE);
130: }
1.11 espie 131: /* since we're dealing with an mmap there, we have to convert once
132: for dealing with data in memory, and a second time for out
133: */
134: fix_header_order(ep);
1.17 espie 135: newsize = 0;
136: errors |= sfcn(fn, fd, ep, &sb, &newsize);
1.11 espie 137: fix_header_order(ep);
1.1 deraadt 138: munmap((caddr_t)ep, sb.st_size);
1.17 espie 139: if (newsize && ftruncate(fd, newsize)) {
140: warn("%s", fn);
141: errors = 1;
142: }
1.1 deraadt 143: if (close(fd)) {
144: ERROR(errno);
145: }
146: }
147: #undef ERROR
148: exit(errors);
149: }
150:
151: int
1.17 espie 152: s_sym(fn, fd, ep, sp, sz)
1.1 deraadt 153: const char *fn;
154: int fd;
1.15 mpech 155: EXEC *ep;
1.1 deraadt 156: struct stat *sp;
1.17 espie 157: off_t *sz;
1.1 deraadt 158: {
1.15 mpech 159: char *neweof;
1.14 pvalchev 160: #if 0
1.15 mpech 161: char *mineof;
1.14 pvalchev 162: #endif
1.1 deraadt 163: int zmagic;
164:
165: zmagic = ep->a_data &&
166: (N_GETMAGIC(*ep) == ZMAGIC || N_GETMAGIC(*ep) == QMAGIC);
167:
168: /*
169: * If no symbols or data/text relocation info and
170: * the file data segment size is already minimized, quit.
171: */
172: if (!ep->a_syms && !ep->a_trsize && !ep->a_drsize) {
173: #if 0
174: if (!zmagic)
175: return 0;
176: if (sp->st_size < N_TRELOFF(*ep))
177: #endif
178: return 0;
179: }
180:
181: /*
182: * New file size is the header plus text and data segments; OMAGIC
183: * and NMAGIC formats have the text/data immediately following the
184: * header. ZMAGIC format wastes the rest of of header page.
185: */
186: neweof = (char *)ep + N_TRELOFF(*ep);
187:
188: #if 0
189: /*
190: * Unfortunately, this can't work correctly without changing the way
191: * the loader works. We could cap it at one page, or even fiddle with
192: * a_data and a_bss, but this only works for CLBYTES == NBPG. If we
193: * are on a system where, e.g., CLBYTES is 8k and NBPG is 4k, and we
194: * happen to remove 4.5k, we will lose. And we really don't want to
195: * fiddle with pages, because that breaks binary compatibility. Lose.
196: */
197:
198: if (zmagic) {
199: /*
200: * Get rid of unneeded zeroes at the end of the data segment
201: * to reduce the file size even more.
202: */
203: mineof = (char *)ep + N_DATOFF(*ep);
204: while (neweof > mineof && neweof[-1] == '\0')
205: neweof--;
206: }
207: #endif
208:
209: /* Set symbol size and relocation info values to 0. */
210: ep->a_syms = ep->a_trsize = ep->a_drsize = 0;
211:
212: /* Truncate the file. */
1.17 espie 213: *sz = neweof - (char *)ep;
1.1 deraadt 214:
215: return 0;
216: }
217:
218: int
1.17 espie 219: s_stab(fn, fd, ep, sp, sz)
1.1 deraadt 220: const char *fn;
221: int fd;
222: EXEC *ep;
223: struct stat *sp;
1.17 espie 224: off_t *sz;
1.1 deraadt 225: {
1.15 mpech 226: int cnt, len;
1.18 ! espie 227: char *nstr, *nstrbase=0, *used=0, *p, *strbase;
1.15 mpech 228: NLIST *sym, *nsym;
1.11 espie 229: u_long allocsize;
230: int mid;
1.1 deraadt 231: NLIST *symbase;
1.18 ! espie 232: unsigned int *mapping=0;
! 233: int error=1;
! 234: unsigned int nsyms;
! 235: struct relocation_info *reloc_base;
! 236: unsigned int i, j;
1.1 deraadt 237:
238: /* Quit if no symbols. */
239: if (ep->a_syms == 0)
240: return 0;
241:
242: if (N_SYMOFF(*ep) >= sp->st_size) {
1.7 mickey 243: warnx("%s: bad symbol table", fn);
1.1 deraadt 244: return 1;
245: }
246:
1.11 espie 247: mid = N_GETMID(*ep);
248:
1.1 deraadt 249: /*
250: * Initialize old and new symbol pointers. They both point to the
251: * beginning of the symbol table in memory, since we're deleting
252: * entries.
253: */
254: sym = nsym = symbase = (NLIST *)((char *)ep + N_SYMOFF(*ep));
255:
256: /*
257: * Allocate space for the new string table, initialize old and
258: * new string pointers. Handle the extra long at the beginning
259: * of the string table.
260: */
261: strbase = (char *)ep + N_STROFF(*ep);
1.11 espie 262: allocsize = fix_long_order(*(u_long *)strbase, mid);
263: if ((nstrbase = malloc((u_int) allocsize)) == NULL) {
1.7 mickey 264: warnx("%s", strerror(ENOMEM));
1.18 ! espie 265: goto end;
1.1 deraadt 266: }
267: nstr = nstrbase + sizeof(u_long);
268:
1.18 ! espie 269: /* okay, so we also need to keep symbol numbers for relocations. */
! 270: nsyms = ep->a_syms/ sizeof(NLIST);
! 271: used = calloc(nsyms, 1);
! 272: if (!used) {
! 273: warnx("%s", strerror(ENOMEM));
! 274: goto end;
! 275: }
! 276: mapping = malloc(nsyms * sizeof(unsigned int));
! 277: if (!mapping) {
! 278: warnx("%s", strerror(ENOMEM));
! 279: goto end;
! 280: }
! 281:
! 282: if ((ep->a_trsize || ep->a_drsize) && byte_sex(mid) != BYTE_ORDER) {
! 283: warnx("%s: cross-stripping not supported", fn);
! 284: goto end;
! 285: }
! 286:
! 287: /* first check the relocations for used symbols, and mark them */
! 288: /* text */
! 289: reloc_base = (struct relocation_info *) ((char *)ep + N_TRELOFF(*ep));
! 290: if (N_TRELOFF(*ep) + ep->a_trsize > sp->st_size) {
! 291: warnx("%s: bad text relocation", fn);
! 292: goto end;
! 293: }
! 294: for (i = 0; i < ep->a_trsize / sizeof(struct relocation_info); i++) {
! 295: if (!reloc_base[i].r_extern)
! 296: continue;
! 297: if (reloc_base[i].r_symbolnum > nsyms) {
! 298: warnx("%s: bad symbol number in text relocation", fn);
! 299: goto end;
! 300: }
! 301: used[reloc_base[i].r_symbolnum] = 1;
! 302: }
! 303: /* data */
! 304: reloc_base = (struct relocation_info *) ((char *)ep + N_DRELOFF(*ep));
! 305: if (N_DRELOFF(*ep) + ep->a_drsize > sp->st_size) {
! 306: warnx("%s: bad data relocation", fn);
! 307: goto end;
! 308: }
! 309: for (i = 0; i < ep->a_drsize / sizeof(struct relocation_info); i++) {
! 310: if (!reloc_base[i].r_extern)
! 311: continue;
! 312: if (reloc_base[i].r_symbolnum > nsyms) {
! 313: warnx("%s: bad symbol number in data relocation", fn);
! 314: goto end;
! 315: }
! 316: used[reloc_base[i].r_symbolnum] = 1;
! 317: }
! 318:
1.1 deraadt 319: /*
320: * Read through the symbol table. For each non-debugging symbol,
321: * copy it and save its string in the new string table. Keep
322: * track of the number of symbols.
323: */
1.18 ! espie 324: for (cnt = nsyms, i = 0, j = 0; cnt--; ++sym, ++i) {
1.11 espie 325: fix_nlist_order(sym, mid);
1.1 deraadt 326: if (!(sym->n_type & N_STAB) && sym->strx) {
327: *nsym = *sym;
328: nsym->strx = nstr - nstrbase;
329: p = strbase + sym->strx;
1.18 ! espie 330: if (xflag && !used[i] &&
1.1 deraadt 331: (!(sym->n_type & N_EXT) ||
332: (sym->n_type & ~N_EXT) == N_FN ||
333: strcmp(p, "gcc_compiled.") == 0 ||
334: strcmp(p, "gcc2_compiled.") == 0 ||
335: strncmp(p, "___gnu_compiled_", 16) == 0)) {
336: continue;
337: }
338: len = strlen(p) + 1;
1.18 ! espie 339: mapping[i] = j++;
1.17 espie 340: if (N_STROFF(*ep) + sym->strx + len > sp->st_size) {
341: warnx("%s: bad symbol table", fn);
1.18 ! espie 342: goto end;
1.17 espie 343: }
1.1 deraadt 344: bcopy(p, nstr, len);
345: nstr += len;
1.11 espie 346: fix_nlist_order(nsym++, mid);
1.1 deraadt 347: }
1.11 espie 348: }
1.1 deraadt 349:
1.18 ! espie 350: /* renumber symbol relocations */
! 351: /* text */
! 352: reloc_base = (struct relocation_info *) ((char *)ep + N_TRELOFF(*ep));
! 353: for (i = 0; i < ep->a_trsize / sizeof(struct relocation_info); i++) {
! 354: if (!reloc_base[i].r_extern)
! 355: continue;
! 356: reloc_base[i].r_symbolnum = mapping[reloc_base[i].r_symbolnum];
! 357: }
! 358: /* data */
! 359: reloc_base = (struct relocation_info *) ((char *)ep + N_DRELOFF(*ep));
! 360: for (i = 0; i < ep->a_drsize / sizeof(struct relocation_info); i++) {
! 361: if (!reloc_base[i].r_extern)
! 362: continue;
! 363: reloc_base[i].r_symbolnum = mapping[reloc_base[i].r_symbolnum];
! 364: }
! 365:
1.1 deraadt 366: /* Fill in new symbol table size. */
367: ep->a_syms = (nsym - symbase) * sizeof(NLIST);
368:
369: /* Fill in the new size of the string table. */
1.11 espie 370: len = nstr - nstrbase;
371: *(u_long *)nstrbase = fix_long_order(len, mid);
1.1 deraadt 372:
373: /*
374: * Copy the new string table into place. Nsym should be pointing
375: * at the address past the last symbol entry.
376: */
377: bcopy(nstrbase, (void *)nsym, len);
1.18 ! espie 378: error = 0;
! 379: end:
1.1 deraadt 380: free(nstrbase);
1.18 ! espie 381: free(used);
! 382: free(mapping);
1.1 deraadt 383:
384: /* Truncate to the current length. */
1.17 espie 385: *sz = (char *)nsym + len - (char *)ep;
1.1 deraadt 386:
1.18 ! espie 387: return error;
1.1 deraadt 388: }
389:
390: void
391: usage()
392: {
393: (void)fprintf(stderr, "usage: strip [-dx] file ...\n");
394: exit(1);
395: }
396: