ref: 1da7f45ae0e42aff174c695d948449bf22bf4b33
dir: /ld/main.c/
static char sccsid[] = "@(#) ./ld/main.c"; #include <ctype.h> #include <errno.h> #include <limits.h> #include <stdio.h> #include <stdlib.h> #include <string.h> #include "../inc/scc.h" #include "../inc/ar.h" #include "../inc/syslibs.h" #include "ld.h" char *output = "a.out", *entry, *datasiz; int pass; int sflag; /* discard all the symbols */ int xflag; /* discard local symbols */ int Xflag; /* discard locals starting with 'L' */ int rflag; /* preserve relocation bits */ int dflag; /* define common even with rflag */ int gflag; /* preserve debug symbols */ void outmem(void) { fputs("ld: out of memory\n", stderr); exit(EXIT_FAILURE); } static int object(char *fname, char *member, FILE *fp) { extern Fmt *formats[]; Fmt **p, *fmt; Obj *obj; void (*fun)(Obj *obj); for (p = formats; *p; ++p) { fmt = *p; obj = (*fmt->probe)(fname, member, fp); if (obj) break; } if (*p == NULL) return 0; (*obj->fmt->pass1)(obj); return 1; } static char * getfname(struct ar_hdr *hdr, char *dst) { char *p; int i; memcpy(dst, hdr->ar_name, SARNAM); dst[SARNAM] = '\0'; for (i = SARNAM-1; i >= 0; i--) { if (dst[i] != ' ' && dst[i] != '/') break; dst[i] = '\0'; } return dst; } static void ar(char *fname, FILE *fp) { struct ar_hdr hdr; long pos, siz; char member[SARNAM+1]; if (fseek(fp, SARMAG, SEEK_SET) == EOF) goto file_error; while (fread(&hdr, sizeof(hdr), 1, fp) == 1) { pos = ftell(fp); if (strncmp(hdr.ar_fmag, ARFMAG, sizeof(hdr.ar_fmag))) goto corrupted; siz = 0; sscanf(hdr.ar_size, "%10ld", &siz); if (siz == 0) goto corrupted; if (siz & 1) siz++; if (pos == -1 || pos > LONG_MAX - siz) die("ld: %s: overflow in size of archive", fname); pos += siz; getfname(&hdr, member); object(fname, member, fp); if (fseek(fp, pos, SEEK_SET) == EOF) goto file_error; } if (ferror(fp)) goto file_error; return; corrupted: die("ld: %s: corrupted archive", fname); file_error: die("ld: %s: %s", fname, strerror(errno)); } static int archive(char *fname, FILE *fp) { char magic[SARMAG]; fpos_t pos; fgetpos(fp, &pos); fread(magic, SARMAG, 1, fp); fsetpos(fp, &pos); if (ferror(fp)) die("ld: %s: %s", fname, strerror(errno)); if (strncmp(magic, ARMAG, SARMAG) != 0) return 0; ar(fname, fp); return 1; } static void process(char *fname) { FILE *fp; if ((fp = fopen(fname, "rb")) == NULL) die("ld: %s: %s", fname, strerror(errno)); if (!object(fname, NULL, fp) && !archive(fname, fp)) die("ld: %s: File format not recognized", fname); if (ferror(fp)) die("ld: %s: %s", fname, strerror(errno)); fclose(fp); } static void pass1(int argc, char *argv[]) { pass = 1; while (*argv) process(*argv++); } static void pass2(int argc, char *argv[]) { pass = 2; } static void usage(void) { fputs("usage: ld [options] file ...\n", stderr); exit(1); } static void Lpath(char *path) { char **bp; for (bp = syslibs; bp < &syslibs[MAX_LIB_PATHS] && *bp; ++bp) ; if (bp == &syslibs[MAX_LIB_PATHS]) { fputs("ld: too many -L options\n", stderr); exit(1); } *bp = path; } int main(int argc, char *argv[]) { char *cp, **p; for (--argc; *++argv; --argc) { if (argv[0][0] != '-' || argv[0][1] == 'l') break; if (argv[0][1] == '-') { --argc, ++argv; break; } for (cp = &argv[0][1]; *cp; ++cp) { switch (*cp) { case 's': sflag = 1; break; case 'x': xflag = 1; break; case 'X': Xflag = 1; break; case 'r': rflag = 1; break; case 'd': dflag = 1; break; case 'i': case 'n': /* TODO */ break; case 'L': if (argc == 0) goto usage; ++argv, --argc; Lpath(*argv); break; case 'u': if (argc == 0) goto usage; ++argv, --argc; lookup(*argv); break; case 'o': if (argc == 0) goto usage; ++argv, --argc; output = *argv; break; case 'e': if (argc == 0) goto usage; ++argv, --argc; entry = *argv; break; case 'D': if (argc == 0) goto usage; ++argv, --argc; datasiz = *argv; break; default: usage: usage(); } } } if (argc < 0) usage(); pass1(argc, argv); pass2(argc, argv); return 0; }