shithub: scc

ref: 1c58d7b01da37935bf4a4d6c6144b0ab03d5df77
dir: /src/libmach/coff32.c/

#include <assert.h>
#include <ctype.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>

#include <scc/coff32/filehdr.h>
#include <scc/coff32/scnhdr.h>
#include <scc/coff32/syms.h>
#include <scc/mach.h>

#include "libmach.h"

typedef struct coff32 Coff32;

struct arch {
	char *name;
	unsigned char magic[2];
	int type;
};

struct coff32 {
	FILHDR hdr;
	SCNHDR *scns;
	SYMENT *ents;
	char *strtbl;
	unsigned long strsiz;
};

static struct arch archs[] = {
	"coff32-i386", "\x4c\x01", OBJ(COFF32, ARCH386, LITTLE_ENDIAN),
	"coff32-z80", "\x5a\x80", OBJ(COFF32, ARCHZ80, LITTLE_ENDIAN),
	NULL,
};

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_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, "ll", buf, &ent->n_zeroes, &ent->n_offset);
}

int
coff32probe(unsigned char *buf, char **name)
{
	struct arch *ap;

	for (ap = archs; ap->name; ap++) {
		if (ap->magic[0] == buf[0] && ap->magic[1] == buf[1]) {
			if (name)
				*name = ap->name;
			return ap->type;
		}
	}
	return -1;
}

int
coff32open(FILE *fp, int type, Obj *obj)
{
	int order;
	long i, siz;
	FILHDR *hdr;
	SCNHDR *scn = NULL;
	SYMENT *ent = NULL;
	char *str = NULL;
	struct coff32 *coff = NULL;
	unsigned char buf[100];

	assert(FILHSZ < sizeof(buf));
	assert(SCNHSZ < sizeof(buf));
	assert(SYMESZ < sizeof(buf));

	coff = calloc(1, sizeof(*coff));
	if (!coff)
		goto error;
	hdr = &coff->hdr;

	order = ORDER(type);
	if (fgetpos(fp, &obj->pos))
		goto error;
	if (fread(buf, FILHSZ, 1, fp) != 1)
		goto error;
	unpack_hdr(order, buf, hdr);

	if (hdr->f_nscns > 0) {
		scn = calloc(hdr->f_nscns, sizeof(*scn));
		if (!scn)
			goto error;
		coff->scns = scn;
	}
	if (hdr->f_nsyms > 0) {
		ent = calloc(hdr->f_nsyms, sizeof(*ent));
		if (!ent)
			goto error;
		coff->ents = ent;
	}

	if (fseek(fp, hdr->f_opthdr, SEEK_CUR) < 0)
		goto error;
	for (i = 0; i < hdr->f_nscns; i++) {
		if (fread(buf, SCNHSZ, 1, fp) < 0)
			goto error;
		unpack_scn(order, buf, &scn[i]);
	}

	if (fsetpos(fp, &obj->pos))
		return -1;
	if (fseek(fp, hdr->f_symptr, SEEK_CUR) < 0)
		return -1;
	for (i = 0; i < hdr->f_nsyms; i++) {
		if (fread(buf, SYMESZ, 1, fp) != 1)
			goto error;
		unpack_ent(order, buf, &ent[i]);
	}

	if (fread(buf, 4, 1, fp) != 1)
		goto error;
	unpack(order, buf, "l", &siz);
	siz -= 4;
	if (siz > 0) {
		if (siz > SIZE_MAX)
			goto error;
		str = malloc(siz);
		if (!str)
			goto error;
		if (fread(str, siz, 1, fp) != 1)
			goto error;
		coff->strtbl = str;
		coff->strsiz = siz;
	}
	obj->data = coff;

	return 0;

error:
	free(str);
	free(scn);
	free(ent);
	free(coff);

	return -1;
}

static char *
symname(Coff32 *coff, SYMENT *ent)
{
	long off;

	if (ent->n_zeroes != 0)
		return ent->n_name;

	off = ent->n_offset;
	if (off >= coff->strsiz)
		return NULL;
	return &coff->strtbl[off];
}

static int
typeof(Coff32 *coff, SYMENT *ent)
{
	int c;
	SCNHDR *scn;
	long flags;

	switch (ent->n_scnum) {
	case N_DEBUG:
		c = 'N';
		break;
	case N_ABS:
		c = 'a';
		break;
	case N_UNDEF:
		c = (ent->n_value != 0) ? 'C' : 'U';
		break;
	default:
		if (ent->n_scnum > coff->hdr.f_nscns)
			return -1;
		scn = &coff->scns[ent->n_scnum-1];
		flags = scn->s_flags;
		if (flags & STYP_TEXT)
			c = 't';
		else if (flags & STYP_DATA)
			c = 'd';
		else if (flags & STYP_BSS)
			c = 'b';
		else
			c = '?';
		break;
	}

	if (ent->n_sclass == C_EXT)
		c = toupper(c);

	return c;

}

int
coff32read(Obj *obj, Symbol *sym)
{
	int t;
	char *s;
	SYMENT *ent;
	Coff32 *coff = obj->data;

	if (obj->cursym >= coff->hdr.f_nsyms)
		return 0;
	ent = &coff->ents[obj->cursym];

	if ((t = typeof(coff, ent)) < 0)
		return -1;

	if ((s = symname(coff, ent)) == NULL)
		return -1;

	obj->cursym += ent->n_numaux + 1;
	sym->type = t;
	sym->name = s;
	sym->value = ent->n_value;
	sym->size = (sym->type == 'C') ? ent->n_value : 0;
	return 1;
}

void
coff32close(Obj *obj)
{
	struct coff32 *coff = obj->data;

	free(coff->scns);
	free(obj->data);
}