Auto Injector V2

#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

int filesize( FILE * fp){
	int sz = 0;
	fseek(fp, 0L, SEEK_END);
	sz = ftell(fp);
	fseek(fp, 0L, SEEK_SET);
	return sz;
}

int GetFileNameDeb(FILE * fn, char ** filnamPtr){
	int i = 0, nstart = 0, nsize = 0, next = 0, estart = 0, esize = 0;
	char c = '0';
	int tabarr[4] = {0,0,0,0};
	char * filnam;

	if(*filnamPtr != NULL){free(*filnamPtr);}

	if(ftell(fn) > 0x143C0){return -1;}

	while(c != '\n'){
		fread(&c, 1, 1, fn);
		if((c == ' ') && (i < 4)){
			tabarr[i] = ftell(fn);
			i++;
		}
	}

	next = ftell(fn);
	nstart = tabarr[0];
	nsize = tabarr[1]-tabarr[0]-1;
	estart = tabarr[2];
	esize = tabarr[3]-tabarr[2]-1;

	*filnamPtr = (char *) malloc(nsize+esize+1);

	filnam = *filnamPtr;

	fseek(fn, nstart, SEEK_SET);
	fread(filnam, 1, nsize, fn);

	fseek(fn, estart, SEEK_SET);
	fread(&(filnam[nsize]), 1, esize, fn);

	filnam[nsize+esize] = '\0';

	fseek(fn, next, SEEK_SET);

	if(filnam[nsize+esize-1] == 'e'){return 1;} else {return 0;}
}

int FindTable(FILE * rom){

	int base = 0x7000, i=0, dummy = 0;
	int check[5] = {0,0,0,0,0};
	int littab[5] = {0, 0x60100000, 0, 0, 0x60100000};

	while(i < 5){

		base += 0x10;

		fseek(rom, base, SEEK_SET);
		dummy = fread(check,4,5,rom);

		for(i=0;i<5;i++){
			if(check[i] != littab[i]){break;}
		}

		if(base > 0x13000){
			base = 0;
			break;
		}
	}

	return base;
}

int SceneFix(FILE * scene, int RoomNo, int Offset1, int Offset2){

	int scenebase = 0, numrooms = 0, dummy = 0, OffsetTable = 0;

	if(scene == NULL){return -1;}

	fseek(scene, 0x09, SEEK_SET);
	dummy = fread(&numrooms, 1, 1, scene);

	fseek(scene, 0x0E, SEEK_SET);
	dummy = fread(&scenebase, 2, 1, scene);

	scenebase = _byteswap_ushort(scenebase);

	fseek(scene, scenebase + 8*RoomNo, SEEK_SET);
	dummy = fread(&OffsetTable, 4, 1, scene);

	if(_byteswap_ulong(OffsetTable) != Offset1){
		printf("Unexpected end of scene\n\r");
		return 1;
	}

	fseek(scene, scenebase + 8*RoomNo + 4, SEEK_SET);
	dummy = fwrite(&Offset2, 4, 1, scene);

	if(RoomNo < numrooms-1){return 0;}

	return 1;
}

void SceneInsert(FILE * scene, FILE * rom, int size, int offset){
	int dummy = 0;
	char * buffer = NULL;

	fseek(rom, offset, SEEK_SET);
	fseek(scene, 0, SEEK_SET);

	buffer = (char *) malloc(size);

	dummy = fread(buffer, 1, size, scene);

	dummy = fwrite(buffer, 1, size, rom);

	free(buffer);
}

int main(){
	int fileindex = 0;
	int offset1 = 0;
	int offset2 = 0;
	int injectsize = 0;
	int injnum = 0;
	int crccheck = 0;
	int excess = 0;
	int tablebase = 0;
	int dummy = 0;
	int status = 0;
	int roomnum = 0;
	int scenstat = 0;
	int scenoff = 0;
	int scensize = 0;
	char * buffer = NULL;
	char * filename = NULL;
	FILE * currentscene = NULL;
	FILE * rom = NULL;
	FILE * injection = NULL;
	FILE * fn = NULL;

	rom = fopen("ZELOOTMA.z64","rb+");
	fn = fopen("DebugList.txt", "rb");

	if((rom == NULL)||(fn == NULL)){
		printf("Failed to open ROM or DebugList.txt\n\r");
		printf("Make sure your ROM is named ZELOOTMA.z64\n\r");
		goto end;
	}

	fseek(fn, 0, SEEK_SET);

	printf("ROM and DebugList opened.\n\r");

	tablebase = FindTable(rom);

	if(tablebase == 0){
		printf("File table not found.");
		fclose(fn);
		fclose(rom);
		goto end;
	}

	printf("File table found at %X\n\r", tablebase);

	for(fileindex = 0 ; fileindex < 1532; fileindex++){

		status = GetFileNameDeb(fn, &filename);
		scenstat = 0;

		if(status == -1){
			printf("Ran out of DebugList\n\r");
			goto end;
		}

		injection = fopen(filename, "rb+");

		if(injection == NULL){/*printf("Failed to open file %s %d",filename,fileindex);*/ continue;}

		printf("%s  ", filename);

		fseek(rom, tablebase + 16*fileindex, SEEK_SET);

		printf("Table Slot %X\n\r", tablebase + 16*fileindex);

		dummy = fread(&offset1, 4, 1, rom);

		offset1 = _byteswap_ulong(offset1);

		fseek(rom, 12, SEEK_CUR);

		dummy = fread(&offset2, 4, 1, rom);

		offset2 = _byteswap_ulong(offset2);

		printf("Start %X End %X\n\r", offset1, offset2);

		injectsize = filesize(injection);

		excess = injectsize - (offset2 - offset1);

		if( excess > 0 ){
			printf("%s is %X bytes too large to inject.\n\r", filename, excess);
			continue;
		}

		offset2 = _byteswap_ulong(offset2 + excess);

		printf("New End %X\n\r",_byteswap_ulong(offset2));

		fseek(rom, tablebase + fileindex*16 + 4, SEEK_SET);

		fwrite(&offset2, 4, 1, rom);

		fseek(rom, offset1, SEEK_SET);

		buffer = (char *) malloc(injectsize);

		dummy = fread(buffer, 1, injectsize, injection);

		dummy = fwrite(buffer, 1, injectsize, rom);

		printf("%X bytes transferred\n\r\n\r", dummy);

		free(buffer);

		injnum++;

		if(status != 1){dummy = fclose(injection);}else{printf("Scene not closed.\n\r\n\r");}

		if(status == 1){
			if(currentscene != NULL){
				SceneInsert(currentscene, rom, scensize, scenoff);

				dummy = fclose(currentscene);
				currentscene = NULL;

				printf("Previous scene closed.\n\r\n\r");
			}

			currentscene = injection;
			scensize = injectsize;
			scenoff = offset1;
			roomnum = -1;
		}

		if((status != 1)&&(currentscene != NULL)){scenstat = SceneFix(currentscene, roomnum, offset1, offset2);}

		roomnum++;

		if(((scenstat == 1)||(roomnum > 32)) && (currentscene != NULL)){
			SceneInsert(currentscene, rom, scensize, scenoff);

			dummy = fclose(currentscene);
			currentscene = NULL;

			printf("Scene closed.\n\r\n\r");
		}
	}

	if(injection != NULL){dummy = fclose(injection);}
	if(currentscene != NULL){
		SceneInsert(currentscene, rom, scensize, scenoff);
		dummy = fclose(currentscene);
	}
	dummy = fclose(fn);
	dummy = fclose(rom);

	printf("Injected %d files.  Now fix the CRC and you're done.\n\r", injnum);

	end:

	printf("AutoInjector program by petrie911.\n\r");
	system("pause");
}