//functions for manipulating the HBC stub by giantpune

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


#include "lstub.h"
#include "filelist.h"
#include "gecko.h"

#include "wad/nandtitle.h"

static char* determineStubTIDLocation()
{
    u32 *stubID = ( u32* )0x80001818;

    //HBC stub 1.0.6 and lower, and stub.bin
    if ( stubID[ 0 ] == 0x480004c1 && stubID[ 1 ] == 0x480004f5 )
        return ( char * )0x800024C6;

    //HBC stub changed @ version 1.0.7.  this file was last updated for HBC 1.0.8
    else if ( stubID[ 0 ] == 0x48000859 && stubID[ 1 ] == 0x4800088d )
        return ( char * )0x8000286A;

    //hexdump( stubID, 0x20 );
    return NULL;

}

s32 Set_Stub( u64 reqID )
{
    if ( titles.IndexOf( reqID ) < 0 )
        return WII_EINSTALL;

    char *stub = determineStubTIDLocation();
    if ( !stub )
        return -68;

    stub[0] = TITLE_7( reqID );
    stub[1] = TITLE_6( reqID );
    stub[8] = TITLE_5( reqID );
    stub[9] = TITLE_4( reqID );
    stub[4] = TITLE_3( reqID );
    stub[5] = TITLE_2( reqID );
    stub[12] = TITLE_1( reqID );
    stub[13] = ( ( u8 )( reqID ) );

    DCFlushRange( stub, 0x10 );

    return 1;

}

s32 Set_Stub_Split( u32 type, const char* reqID )
{
    char tmp[4];
    u32 lower;
    sprintf( tmp, "%c%c%c%c", reqID[0], reqID[1], reqID[2], reqID[3] );
    memcpy( &lower, tmp, 4 );
    u64 reqID64 = TITLE_ID( type, lower );
    return Set_Stub( reqID64 );

}

void loadStub()
{
    char *stubLoc = ( char * )0x80001800;
    memcpy( stubLoc, stub_bin, stub_bin_size );
    DCFlushRange( stubLoc, stub_bin_size );
}

u64 getStubDest()
{
    if ( !hbcStubAvailable() )
        return 0;

    char ret[ 8 ];
    u64 retu = 0;

    char *stub = determineStubTIDLocation();
    if ( !stub )
        return 0;


    ret[0] = stub[0];
    ret[1] = stub[1];
    ret[2] = stub[8];
    ret[3] = stub[9];
    ret[4] = stub[4];
    ret[5] = stub[5];
    ret[6] = stub[12];
    ret[7] = stub[13];

    memcpy( &retu, ret, 8 );

    return retu;
}

u8 hbcStubAvailable()
{
    char * sig = ( char * )0x80001804;
    return (
               sig[0] == 'S' &&
               sig[1] == 'T' &&
               sig[2] == 'U' &&
               sig[3] == 'B' &&
               sig[4] == 'H' &&
               sig[5] == 'A' &&
               sig[6] == 'X' &&
               sig[7] == 'X' ) ? 1 : 0;
}