/*
    "$Id: gbafix.c,v 1.2 2008-07-30 17:12:51 wntrmute Exp $"

    DevkitPro GBA ROM fix utility

    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Library General Public
    License as published by the Free Software Foundation; either
    version 2 of the License, or (at your option) any later version.

    This library is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
    Library General Public License for more details.

    You should have received a copy of the GNU Library General Public
    License along with this library; if not, write to the Free Software
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
    USA.

    Please report all bugs and problems through the bug tracker at
    "http://sourceforge.net/tracker/?group_id=114505&atid=668551".

    "$Header: /lvm/shared/ds/ds/cvs/devkitpro-cvsbackup/tools/gba/gbatools/gbafix.c,v 1.2 2008-07-30 17:12:51 wntrmute Exp $"

*/
//---------------------------------------------------------------------------------
// gbafix.c
//---------------------------------------------------------------------------------
/*
    Gameboy Advance ROM fixer (by Dark Fader / BlackThunder / WinterMute / Sierraffinity)
    Validates header of GBA roms.

    History
    -------
    v1.07 - added support for ELF input, (PikalaxALT)
    v1.06 - added output silencing, (Sierraffinity)
    v1.05 - added debug offset argument, (Sierraffinity)
    v1.04 - converted to plain C, (WinterMute)
    v1.03 - header.fixed, header.device_type
    v1.02 - redefined the options (rgbfix style), checksum=0
    v1.01 - fix in parameters
    v1.00 - logo, complement
*/

#pragma pack(1)

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

#define VER        "1.07"
#define ARGV    argv[arg]
#define VALUE    (ARGV+2)
#define NUMBER    strtoul(VALUE, NULL, 0)

typedef struct
{
    uint32_t    start_code;            // B instruction
    uint8_t        logo[0xA0-0x04];    // logo data
    uint8_t        title[0xC];            // game title name
    uint32_t    game_code;            //
    uint16_t    maker_code;            //
    uint8_t        fixed;                // 0x96
    uint8_t        unit_code;            // 0x00
    uint8_t        device_type;        // 0x00
    uint8_t        unused[7];            //
    uint8_t        game_version;        // 0x00
    uint8_t        complement;            // 800000A0..800000BC
    uint16_t    checksum;            // 0x0000
} Header;


Header header;

unsigned short checksum_without_header = 0;

const Header good_header =
{
    // start_code
    0xEA00002E,
    // logo
    { 0x24,0xFF,0xAE,0x51,0x69,0x9A,0xA2,0x21,0x3D,0x84,0x82,0x0A,0x84,0xE4,0x09,0xAD,
    0x11,0x24,0x8B,0x98,0xC0,0x81,0x7F,0x21,0xA3,0x52,0xBE,0x19,0x93,0x09,0xCE,0x20,
    0x10,0x46,0x4A,0x4A,0xF8,0x27,0x31,0xEC,0x58,0xC7,0xE8,0x33,0x82,0xE3,0xCE,0xBF,
    0x85,0xF4,0xDF,0x94,0xCE,0x4B,0x09,0xC1,0x94,0x56,0x8A,0xC0,0x13,0x72,0xA7,0xFC,
    0x9F,0x84,0x4D,0x73,0xA3,0xCA,0x9A,0x61,0x58,0x97,0xA3,0x27,0xFC,0x03,0x98,0x76,
    0x23,0x1D,0xC7,0x61,0x03,0x04,0xAE,0x56,0xBF,0x38,0x84,0x00,0x40,0xA7,0x0E,0xFD,
    0xFF,0x52,0xFE,0x03,0x6F,0x95,0x30,0xF1,0x97,0xFB,0xC0,0x85,0x60,0xD6,0x80,0x25,
    0xA9,0x63,0xBE,0x03,0x01,0x4E,0x38,0xE2,0xF9,0xA2,0x34,0xFF,0xBB,0x3E,0x03,0x44,
    0x78,0x00,0x90,0xCB,0x88,0x11,0x3A,0x94,0x65,0xC0,0x7C,0x63,0x87,0xF0,0x3C,0xAF,
    0xD6,0x25,0xE4,0x8B,0x38,0x0A,0xAC,0x72,0x21,0xD4,0xF8,0x07 } ,
    // title
    { 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00 },
    // game code
    0x00000000,
    // maker code
    0x3130,
    // fixed
    0x96,
    // unit_code
    0x00,
    // device type
    0x00,
    // unused
    { 0x00,0x00,0x00,0x00,0x00,0x00,0x00 },
    // game version
    0x00,
    // complement
    0x00,
    // checksum
    0x0000
};

//---------------------------------------------------------------------------------
char HeaderComplement()
/*---------------------------------------------------------------------------------
    Calculate Header complement check
---------------------------------------------------------------------------------*/
{
    int n;
    char c = 0;
    char *p = (char *)&header + 0xA0;
    for (n=0; n<0xBD-0xA0; n++)
    {
        c += *p++;
    }
    return -(0x19+c);
}


//---------------------------------------------------------------------------------
int main(int argc, char *argv[])
//---------------------------------------------------------------------------------
{
    int arg;
    char *argfile = 0;
    FILE *infile;
    int silent = 0;
    int schedule_pad = 0;

    int size,bit;

    // show syntax
    if (argc <= 1)
    {
        printf("GBA ROM fixer v"VER" by Dark Fader / BlackThunder / WinterMute / Sierraffinity \n");
        printf("Syntax: gbafix <rom.gba> [-p] [-t[title]] [-c<game_code>] [-m<maker_code>] [-r<version>] [-d<debug>] [--silent]\n");
        printf("\n");
        printf("parameters:\n");
        printf("    -p              Pad to next exact power of 2. No minimum size!\n");
        printf("    -t[<title>]     Patch title. Stripped filename if none given.\n");
        printf("    -c<game_code>   Patch game code (four characters)\n");
        printf("    -m<maker_code>  Patch maker code (two characters)\n");
        printf("    -r<version>     Patch game version (number)\n");
        printf("    -d<debug>       Enable debugging handler and set debug entry point (0 or 1)\n");
        printf("    --silent           Silence non-error output\n");
        return -1;
    }

    // get filename
    for (arg=1; arg<argc; arg++)
    {
        if (ARGV[0] != '-') { argfile=ARGV; }
        if (strncmp("--silent", &ARGV[0], 7) == 0) { silent = 1; }
    }

    // check filename
    if (!argfile)
    {
        fprintf(stderr, "Filename needed!\n");
        return -1;
    }

    uint32_t sh_offset = 0;

    // read file
    infile = fopen(argfile, "r+b");
    if (!infile) { fprintf(stderr, "Error opening input file!\n"); return -1; }
    fseek(infile, sh_offset, SEEK_SET);
    fread(&header, sizeof(header), 1, infile);

    // elf check
    Elf32_Shdr secHeader;
    if (memcmp(&header, ELFMAG, 4) == 0) {
        Elf32_Ehdr *elfHeader = (Elf32_Ehdr *)&header;
        fseek(infile, elfHeader->e_shoff, SEEK_SET);
        int i;
        for (i = 0; i < elfHeader->e_shnum; i++) {
            fread(&secHeader, sizeof(Elf32_Shdr), 1, infile);
            if (secHeader.sh_type == SHT_PROGBITS && secHeader.sh_addr == elfHeader->e_entry) break;
        }
        if (i == elfHeader->e_shnum) { fprintf(stderr, "Error finding entry point!\n"); return 1; }
        fseek(infile, secHeader.sh_offset, SEEK_SET);
        sh_offset = secHeader.sh_offset;
        fread(&header, sizeof(header), 1, infile);
    }

    // fix some data
    memcpy(header.logo, good_header.logo, sizeof(header.logo));
    memcpy(&header.fixed, &good_header.fixed, sizeof(header.fixed));
    memcpy(&header.device_type, &good_header.device_type, sizeof(header.device_type));

    // parse command line
    for (arg=1; arg<argc; arg++)
    {
        if (ARGV[0] == '-')
        {
            switch (ARGV[1])
            {
                case 'p':    // pad
                {
                    schedule_pad = 1;
                    break;
                }

                case 't':    // title
                {
                    char title[256];
                    memset(title, 0, sizeof(title));
                    if (VALUE[0])
                    {
                        strncpy(title, VALUE, sizeof(header.title));
                    }
                    else
                    {
                        // use filename
                        char s[256], *begin=s, *t; strcpy(s, argfile);
                        t = strrchr(s, '\\'); if (t) begin = t+1;
                        t = strrchr(s, '/'); if (t) begin = t+1;
                        t = strrchr(s, '.'); if (t) *t = 0;
                        strncpy(title, begin, sizeof(header.title));
                        if (!silent) printf("%s\n",begin);
                    }
                    memcpy(header.title, title, sizeof(header.title));    // copy
                    break;
                }

                case 'c':    // game code
                {
                    //if (!VALUE[0]) { fprintf(stderr, "Need value for %s\n", ARGV); break; }
                    //header.game_code = NUMBER;
                    header.game_code = VALUE[0] | VALUE[1]<<8 | VALUE[2]<<16 | VALUE[3]<<24;
                    break;
                }

                case 'm':    // maker code
                {
                    //if (!VALUE[0]) { fprintf(stderr, "Need value for %s\n", ARGV); break; }
                    //header.maker_code = (unsigned short)NUMBER;
                    header.maker_code = VALUE[0] | VALUE[1]<<8;
                    break;
                }

                case 'v':    // ignored, compatability with other gbafix
                {
                    break;
                }

                case 'r':    // version
                {
                    if (!VALUE[0]) { fprintf(stderr, "Need value for %s\n", ARGV); break; }
                    header.game_version = (unsigned char)NUMBER;
                    break;
                }

                case 'd':    // debug
                {
                    if (!VALUE[0]) { fprintf(stderr, "Need value for %s\n", ARGV); break; }
                    header.logo[0x9C-0x04] = 0xA5;    // debug enable
                    header.device_type = (unsigned char)((NUMBER & 1) << 7);    // debug handler entry point
                    break;
                }
                case '-':    // long arguments
                {
                    if (strncmp("silent", &ARGV[2], 6) == 0) { continue; }
                    break;
                }
            default:
                {
                    printf("Invalid option: %s\n", ARGV);
                }
            }
        }
    }

    // update complement check & total checksum
    header.complement = 0;
    header.checksum = 0;    // must be 0
    header.complement = HeaderComplement();
    //header.checksum = checksum_without_header + HeaderChecksum();

    if (schedule_pad) {
        if (sh_offset != 0) {
            fprintf(stderr, "Warning: Cannot safely pad an ELF\n");
        } else {
            fseek(infile, 0, SEEK_END);
            size = ftell(infile);
            for (bit=31; bit>=0; bit--) if (size & (1<<bit)) break;
            if (size != (1<<bit))
            {
                int todo = (1<<(bit+1)) - size;
                while (todo--) fputc(0xFF, infile);
            }
        }
    }

    fseek(infile, sh_offset, SEEK_SET);
    fwrite(&header, sizeof(header), 1, infile);
    fclose(infile);

    if (!silent) printf("ROM fixed!\n");

    return 0;
}