#include <stdio.h>
#include <ogcsys.h>
#include <string.h>
#include <malloc.h>

#include "patches/patchcode.h"
#include "apploader.h"
#include "wdvd.h"
#include "wpad.h"
#include "disc.h"
#include "alternatedol.h"
#include "fstfile.h"
#include "settings/cfg.h"
#include "gecko.h"
#include "patches/wip.h"
#include "patches/dolpatcher.h"
#include "patches/gamepatches.h"

extern bool geckoinit;

/* Apploader function pointers */
typedef int   ( *app_main )( void **dst, int *size, int *offset );
typedef void  ( *app_init )( void ( *report )( const char *fmt, ... ) );
typedef void *( *app_final )();
typedef void  ( *app_entry )( void ( **init )( void ( *report )( const char *fmt, ... ) ), int ( **main )(), void *( **final )() );

/* Apploader pointers */
static u8 *appldr = ( u8 * )0x81200000;


/* Constants */
#define APPLDR_OFFSET   0x2440

/* Variables */
static u32 buffer[0x20] ATTRIBUTE_ALIGN( 32 );

void gamepatches( u8 * dst, int len, u8 videoSelected, u8 patchcountrystring, u8 vipatch, u8 cheat )
{
    VideoModePatcher( dst, len, videoSelected );

    if ( cheat )
        dogamehooks( dst, len );

    if ( vipatch )
        vidolpatcher( dst, len );

    /*LANGUAGE PATCH - FISHEARS*/
    langpatcher( dst, len );

    /*Thanks to WiiPower*/
    if ( patchcountrystring == 1 )
        PatchCountryStrings( dst, len );

    NSMBPatch( dst, len );

    do_wip_code( ( u8 * ) dst, len );

    if ( fix002 == 2 )
        Anti_002_fix( dst, len );
}

s32 Apploader_Run( entry_point *entry, char * dolpath, u8 cheat, u8 videoSelected, u8 vipatch, u8 patchcountrystring, u8 error002fix, u8 alternatedol, u32 alternatedoloffset, u32 returnTo )
{
    app_entry appldr_entry;
    app_init  appldr_init;
    app_main  appldr_main;
    app_final appldr_final;

    u32 appldr_len;
    s32 ret;
    gprintf( "\nApploader_Run() started\n" );

    //u32 geckoattached = usb_isgeckoalive(EXI_CHANNEL_1);
    //if (geckoattached)usb_flush(EXI_CHANNEL_1);
    geckoinit = InitGecko();

    /* Read apploader header */
    ret = WDVD_Read( buffer, 0x20, APPLDR_OFFSET );
    if ( ret < 0 )
        return ret;

    /* Calculate apploader length */
    appldr_len = buffer[5] + buffer[6];

    /* Read apploader code */
    ret = WDVD_Read( appldr, appldr_len, APPLDR_OFFSET + 0x20 );
    if ( ret < 0 )
        return ret;

    /* Set apploader entry function */
    appldr_entry = ( app_entry )buffer[4];

    /* Call apploader entry */
    appldr_entry( &appldr_init, &appldr_main, &appldr_final );

    /* Initialize apploader */
    appldr_init( gprintf );

    if ( error002fix != 0 )
    {
        /* ERROR 002 fix (thanks to WiiPower for sharing this)*/
        *( u32 * )0x80003188 = *( u32 * )0x80003140;
    }

    u32 dolStart = 0x90000000;
    u32 dolEnd = 0x0;

    for ( ;; )
    {
        void *dst = NULL;
        int len = 0, offset = 0;

        /* Run apploader main function */
        ret = appldr_main( &dst, &len, &offset );
        if ( !ret )
            break;

        /* Read data from DVD */
        WDVD_Read( dst, len, ( u64 )( offset << 2 ) );

        gamepatches( dst, len, videoSelected, patchcountrystring, vipatch, cheat );
        DCFlushRange( dst, len );

	if( (u32)dst < dolStart )dolStart = (u32)dst;
	if( (u32)dst + len > dolEnd ) dolEnd = (u32)dst + len;
    }

    //this patch should be run on the entire dol at 1 time
    if( !alternatedol && returnTo)
    {
	if( PatchReturnTo( (u32*)dolStart, dolEnd - dolStart , returnTo) )
	{
	    DCFlushRange( (u32*)dolStart, dolEnd - dolStart );
	}
    }

    *entry = appldr_final();

    /** Load alternate dol if set **/
    if ( alternatedol == 1 )
    {
        wip_reset_counter();
        void *dolbuffer = NULL;
        int dollen = 0;

        bool dolloaded = Load_Dol( &dolbuffer, &dollen, dolpath, videoSelected, patchcountrystring, vipatch, cheat );
        if ( dolloaded )
        {
            *entry = ( entry_point ) load_dol_image( dolbuffer );
        }

        if ( dolbuffer )
            free( dolbuffer );
    }
    else if ( alternatedol == 2 )
    {
        wip_reset_counter();
        FST_ENTRY *fst = ( FST_ENTRY * ) * ( u32 * )0x80000038;

        *entry = ( entry_point ) Load_Dol_from_disc( fst[alternatedoloffset].fileoffset, videoSelected, patchcountrystring, vipatch, cheat );

        if ( *entry == 0 )
            SYS_ResetSystem( SYS_RETURNTOMENU, 0, 0 );
    }

    return 0;
}