AutoInjector Source

#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 main(){
        int fileindex = 0;
        int offset1 = 0;
        int offset2 = 0;
        int injectsize = 0;
        int injnum = 0;
        int excess = 0;
        int tablebase = 0;
        int dummy = 0;
        int status = 0;
        char * buffer = NULL;
        char * filename = 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, 0x509C, SEEK_SET);

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

        tablebase = 0x12F70;

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

                status = GetFileNameDeb(fn, &filename);

                if(status == -1){
                        printf("Ran out of fn");
                        break;
                }

                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++;

                dummy = fclose(injection);
        }

        dummy = fclose(fn);
        dummy = fclose(rom);

end:
        printf("Injected %d files.  Now fix the CRC and you're done.\n\r", injnum);
        printf("AutoInjector program by petrie911.");
}