shithub: doll

ref: 8403ad31262165522d19e4189292c7c3fdbba45e
dir: /main.c/

#include <u.h>
#include <libc.h>
#include <bio.h>
#include <mach.h>

int keepgoing;
char *out = "out.dol";

typedef struct Dolhdr Dolhdr;
	
struct Dolhdr
{
	u32int txtoff[7];
	u32int datoff[11];
	u32int txtaddr[7];
	u32int dataddr[11];
	u32int txtsz[7];
	u32int datsz[11];
	u32int bssaddr;
	u32int bsssz;
	u32int entry;
	uchar padding[26];
};

enum
{
	BIG, LITTLE
};

int
curendianness(void)
{
	union
	{
		int i;
		uchar c;
	} v;

	v.i = 0xFF;
	return v.c == 0xFF;
}

void
bigstore(char *dest, void* src, usize len)
{
	char *s = src;

	if(curendianness() == BIG)
		memcpy(dest, src, len);
	else
		for(int c = 0; c < len; c++)
			dest[c] = s[len-c];
}

#pragma varargck argpos error 1
void
error(char *fmt, ...)
{
	va_list args;
	char *str;

	va_start(args, fmt);
	str = vsmprint(fmt, args);
	fprint(2, "%s: %s\n", argv0, str);
	if(!keepgoing)
		exits(str);
	free(str);
	va_end(args);
}

void
usage(void)
{
	fprint(2, "usage: %s [-k] [-o out] exec\n", argv0);
	exits("usage");
}

void
main(int argc, char **argv)
{
	Dolhdr dol;
	Fhdr fhdr;
	int ofd, ifd;
	char buf[2048];

	ARGBEGIN {
	case 'o':
		out = EARGF(usage());
		break;
	case 'k':
		keepgoing++;
		break;
	default:
		usage();
	} ARGEND;

	if(argc != 1)
		usage();

	if((ifd = open(*argv, OREAD)) < 0)
		sysfatal("open: %r");

	if((ofd = create(out, OWRITE|OTRUNC, 0755)) < 0)
		sysfatal("create: %r");

	crackhdr(ifd, &fhdr);

	if(fhdr.type != FPOWER && fhdr.type != FPOWERB)
		error("not a PowerPC executable (%d)", fhdr.type);

	machbytype(fhdr.type);

	dol.txtaddr[0] = beswal(fhdr.txtaddr);
	dol.txtoff[0] = beswal(fhdr.txtoff);
	dol.txtsz[0] = beswal(fhdr.txtsz);
	if(fhdr.txtaddr < 0x80003F00 || fhdr.txtaddr > 0x81330000)
		error("text outside standard executable area (0x%08lX)", (ulong)fhdr.txtaddr);

	dol.dataddr[0] = beswal(fhdr.dataddr);
	dol.datoff[0] = beswal(fhdr.datoff);
	dol.datsz[0] = beswal(fhdr.datsz);

	dol.bssaddr = fhdr.dataddr + fhdr.datsz;	/* FIXME? */
	dol.bsssz = fhdr.bsssz;

	dol.entry = fhdr.entry;

	/* TODO validate addresses */

	if(write(ofd, &dol, sizeof(dol)) != sizeof(dol))
		sysfatal("write: %r");

	seek(ifd, fhdr.hdrsz, 0);
	while(read(ifd, buf, sizeof(buf)) > 0)
		if(write(ofd, buf, sizeof(buf)) != sizeof(buf))
			sysfatal("write: %r");

	close(ofd);
	close(ifd);
	exits(0);
}