ref: 433b9cd26eb5d27977eb8cde6bb1c019606e0ffa
dir: /src/libmach/coff32/coff32read.c/
#include <assert.h> #include <ctype.h> #include <stdint.h> #include <stdio.h> #include <stdlib.h> #include <string.h> #include <scc/mach.h> #include "../libmach.h" #include "coff32.h" static void unpack_hdr(int order, unsigned char *buf, FILHDR *hdr) { int n; n = unpack(order, buf, "sslllss", &hdr->f_magic, &hdr->f_nscns, &hdr->f_timdat, &hdr->f_symptr, &hdr->f_nsyms, &hdr->f_opthdr, &hdr->f_flags); assert(n == FILHSZ); } static void unpack_line(int order, unsigned char *buf, LINENO *lp) { int n; n = unpack(order, buf, "lls", &lp->l_symndx, &lp->l_paddr, &lp->l_lnno); assert(n == LINESZ); } static void unpack_scn(int order, unsigned char *buf, SCNHDR *scn) { int n; n = unpack(order, buf, "'8llllllssl", scn->s_name, &scn->s_paddr, &scn->s_vaddr, &scn->s_size, &scn->s_scnptr, &scn->s_relptr, &scn->s_lnnoptr, &scn->s_nrelloc, &scn->s_nlnno, &scn->s_flags); assert(n == SCNHSZ); } static void unpack_ent(int order, unsigned char *buf, SYMENT *ent) { int n; char *s; n = unpack(order, buf, "'8lsscc", ent->n_name, &ent->n_value, &ent->n_scnum, &ent->n_type, &ent->n_sclass, &ent->n_numaux); assert(n == SYMESZ); s = ent->n_name; if (!s[0] && !s[1] && !s[2] && !s[3]) unpack(order, buf, "ll", &ent->n_zeroes, &ent->n_offset); } static void unpack_reloc(int order, unsigned char *buf, RELOC *rel) { int n; n = unpack(order, buf, "lls", &rel->r_vaddr, &rel->r_symndx, &rel->r_type); assert(n == RELSZ); } static void unpack_aout(int order, unsigned char *buf, AOUTHDR *aout) { int n; n = unpack(order, buf, "ssllllll", &aout->magic, &aout->vstamp, &aout->tsize, &aout->dsize, &aout->bsize, &aout->entry, &aout->text_start, &aout->data_start); assert(n == AOUTSZ); } static int readhdr(Obj *obj, FILE *fp) { FILHDR *hdr; struct coff32 *coff; unsigned char buf[FILHSZ]; coff = obj->data; hdr = &coff->hdr; if (fread(buf, FILHSZ, 1, fp) != 1) return 0; unpack_hdr(ORDER(obj->type), buf, hdr); return 1; } static int readstr(Obj *obj, FILE *fp) { FILHDR *hdr; struct coff32 *coff; long siz; char *str; unsigned char buf[10]; coff = obj->data; hdr = &coff->hdr; coff->strsiz = 0; coff->strtbl = NULL; if (hdr->f_nsyms == 0) return 1; if (fread(buf, 4, 1, fp) != 1) return 0; unpack(ORDER(obj->type), buf, "l", &siz); if (siz < 4 || siz > SIZE_MAX) return 0; if (siz == 4) return 1; if ((str = malloc(siz)) == NULL) return 0; coff->strtbl = str; coff->strsiz = siz; return fread(str+4, siz-4, 1, fp) == 1; } static int readreloc(Obj *obj, FILE *fp) { int i; long j; RELOC **rels, *rp; SCNHDR *scn; FILHDR *hdr; struct coff32 *coff; unsigned char buf[RELSZ]; coff = obj->data; hdr = &coff->hdr; if (hdr->f_nscns == 0) return 1; rels = calloc(hdr->f_nscns, sizeof(*rels)); if (!rels) return 0; coff->rels = rels; for (i = 0; i < hdr->f_nscns; i++) { scn = &coff->scns[i]; if (scn->s_nrelloc == 0) continue; if (!objpos(obj, fp, scn->s_relptr)) return 0; rp = calloc(scn->s_nrelloc, sizeof(RELOC)); if (!rp) return 0; rels[i] = rp; for (j = 0; j < scn->s_nrelloc; j++) { if (fread(buf, RELSZ, 1, fp) != 1) return 0; unpack_reloc(ORDER(obj->type), buf, &rp[i]); if (rp[i].r_symndx >= hdr->f_nsyms) return 0; } } return 1; } static int readents(Obj *obj, FILE *fp) { FILHDR *hdr; struct coff32 *coff; SYMENT *ent; long i; unsigned char buf[SYMESZ]; coff = obj->data; hdr = &coff->hdr; if (hdr->f_nsyms == 0) return 1; ent = calloc(hdr->f_nsyms, sizeof(*ent)); if (!ent) return 0; coff->ents = ent; if (!objpos(obj, fp, hdr->f_symptr)) return 0; for (i = 0; i < hdr->f_nsyms; i++) { if (fread(buf, SYMESZ, 1, fp) != 1) return 0; unpack_ent(ORDER(obj->type), buf, &ent[i]); if (ent->n_scnum > hdr->f_nscns) return 0; } return 1; } static int readscns(Obj *obj, FILE *fp) { FILHDR *hdr; struct coff32 *coff; SCNHDR *scn; long i; unsigned char buf[SCNHSZ]; coff = obj->data; hdr = &coff->hdr; if (hdr->f_nscns == 0) return 1; scn = calloc(hdr->f_nscns, sizeof(*scn)); if (!scn) return 0; coff->scns = scn; for (i = 0; i < hdr->f_nscns; i++) { if (fread(buf, SCNHSZ, 1, fp) < 0) return 0; unpack_scn(ORDER(obj->type), buf, &scn[i]); } return 1; } static int readlines(Obj *obj, FILE *fp) { int i,j; LINENO **lines, *lp; FILHDR *hdr; SCNHDR *scn; struct coff32 *coff; unsigned char buf[LINESZ]; coff = obj->data; hdr = &coff->hdr; if (hdr->f_nscns == 0) return 1; lines = calloc(hdr->f_nscns, sizeof(lp)); if (!lines) return 0; coff->lines = lines; for (i = 0; i < hdr->f_nscns; i++) { scn = &coff->scns[i]; if (scn->s_nlnno == 0) continue; lp = calloc(sizeof(*lp), scn->s_nlnno); if (!lp) return 0; lines[i] = lp; for (j = 0; j < scn->s_nlnno; j++) { if (!objpos(obj, fp, scn->s_lnnoptr)) return 0; if (fread(buf, LINESZ, 1, fp) == 1) return 0; unpack_line(ORDER(obj->type), buf, &lp[j]); if (lp[i].l_symndx >= hdr->f_nsyms) return 0; } } return 1; } static int readaout(Obj *obj, FILE *fp) { struct coff32 *coff = obj->data; FILHDR *hdr = &coff->hdr; unsigned char buf[AOUTSZ]; if (hdr->f_opthdr == 0) return 1; if (fread(buf, AOUTSZ, 1, fp) != 1) return 0; unpack_aout(ORDER(obj->type), buf, &coff->aout); return 1; } int coff32read(Obj *obj, FILE *fp) { long i; struct coff32 *coff = obj->data; FILHDR *hdr = &coff->hdr; if (!readhdr(obj, fp)) return -1; if (!readaout(obj, fp)) return -1; if (!readscns(obj, fp)) return -1; if (!readents(obj, fp)) return -1; if (!readstr(obj, fp)) return -1; if (!readreloc(obj, fp)) return -1; if (!readlines(obj, fp)) return -1; for (i = 0; i < hdr->f_nsyms; i++) { SYMENT *ent = &coff->ents[i]; if (ent->n_zeroes != 0 && ent->n_offset > coff->strsiz) return -1; } return 0; }