//functions for manipulating the HBC stub by giantpune

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


#include "lstub.h"
#include "stub_bin.h"
#include "gecko.h"

#define TITLE_1(x)		((u8)((x) >> 8))
#define TITLE_2(x)		((u8)((x) >> 16))
#define TITLE_3(x)		((u8)((x) >> 24))
#define TITLE_4(x)		((u8)((x) >> 32))
#define TITLE_5(x)		((u8)((x) >> 40))
#define TITLE_6(x)		((u8)((x) >> 48))
#define TITLE_7(x)		((u8)((x) >> 56))
#define TITLE_ID(x,y)	(((u64)(x) << 32) | (y))

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)
{
	u32 tmdsize;
	u64 tid = 0;
	u64 *list;
	u32 titlecount;
	s32 ret;
	u32 i;

	ret = ES_GetNumTitles(&titlecount);
	if(ret < 0)
		return WII_EINTERNAL;

	list = memalign(32, titlecount * sizeof(u64) + 32);

	ret = ES_GetTitles(list, titlecount);
	if(ret < 0) {
		free(list);
		return WII_EINTERNAL;
	}
	
	for(i=0; i<titlecount; i++) {
		if (list[i]==reqID)
		{
			tid = list[i];
			break;
		}
	}
	free(list);

	if(!tid)
		return -69;//id to stub is not installed

	if(ES_GetStoredTMDSize(tid, &tmdsize) < 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;
}