CfgUSBLoader/plugins/mighty/source/codes/codes.c

/*
config_bytes[2] hooktypes
0 no
1 VBI
2 KPAD read
3 Joypad Hook
4 GXDraw Hook
5 GXFlush Hook
6 OSSleepThread Hook
7 AXNextFrame Hook
8 Default

config_bytes[4] ocarina
0 no
1 yes

config_bytes[5] paused start
0 no
1 yes


config_bytes[7] debugger
0 no
1 yes
*/

#include <gccore.h>
#include <string.h>
#include <fat.h>
#include <ogc/usbstorage.h>
#include <sdcard/wiisd_io.h>
#include <stdio.h>
#include <malloc.h>
#include <sys/unistd.h>



#include "codehandler.h"
#include "codehandlerslota.h"
#include "codehandleronly.h"
#include "multidol.h"
#include "patchcode.h"

#include "tools.h"
#include "config.h"
#include "codes.h"

static const u8 *codelistend = (u8 *) 0x80003000;

void *codelist;

DISC_INTERFACE storage;

void storage_shutdown(){
	//if (ocarinaoption < 4 || wbfsdevice != 1)
	{
		storage.shutdown();
	}
}


s32 load_codes(char *filename, u32 maxsize, u8 *buffer){
	char text[4];	
	
	if (ocarinaoption == 1){
		text[0] = 'S';
		text[1] = 'D';
		text[2] = 0;
		text[3] = 0;
		storage = __io_wiisd;
	}else{
		text[0] = 'U';
		text[1] = 'S';
		text[2] = 'B';
		text[3] = 0;
		storage = __io_usbstorage;
	}
	
	FILE *fp;
	u32 filesize;
	u32 ret;
	char buf[128];
	
	ret = storage.startup();
	if (ret < 0) 
	{
		printf("%s Error\n", text);
		//write_font(185, 346, "%s Error", text);
		return ret;
	}
	ret = fatMountSimple("fat", &storage);

	if (ret < 0) 
	{
		storage_shutdown();
		printf("FAT Error\n");
		//write_font(185, 346, "FAT Error");
		return ret;
	}

	fflush(stdout);
	
	sprintf(buf, "fat:/usb-loader/codes/%s.gct", filename);
	printf("Ocaraina trying to open file %s\n", buf);
	fp = fopen(buf, "rb");

	if (!fp){
		printf("Failed to open %s\n", buf);
		sprintf(buf, "fat:/apps/USBLoader/codes/%s.gct", filename);
		fp = fopen(buf, "rb");
	}

	if (!fp){
		printf("Failed to open %s\n", buf);
		sprintf(buf, "fat:/codes/%s.gct", filename);
		fp = fopen(buf, "rb");
	}

	if (!fp){
		printf("Failed to open %s\n", buf);
		sprintf(buf, "fat:/config/mighty/cheats/%s.gct", filename);
		fp = fopen(buf, "rb");
	}
	
	if (!fp){
		fatUnmount("fat");
		storage_shutdown();
		printf("Failed to open %s\n", buf);
		printf("No %s codes found\n", text);
		sleep(3);
		//write_font(185, 346, "No %s codes found", text);
		return -1;
	}

	printf("Ocaraina using %s\n", buf);

	fseek(fp, 0, SEEK_END);
	filesize = ftell(fp);
	fseek(fp, 0, SEEK_SET);
	
	if (filesize > maxsize){
		fatUnmount("fat");
		storage_shutdown();
		printf("Too many codes\n");
		//write_font(185, 346, ""Too many codes");
		return -1;
	}	
	
	ret = fread(buffer, 1, filesize, fp);
	if(ret != filesize)
	{	
		fclose(fp);
		fatUnmount("fat");
		storage_shutdown();
		printf("%s Code Error\n", text);
		//write_font(185, 346, "%s Code Error", text);
		return -1;
	}

	fclose(fp);
	
	fatUnmount("fat");
	storage_shutdown();
	
	return 0;
}



//---------------------------------------------------------------------------------
void load_handler()
//---------------------------------------------------------------------------------
{
	if (hooktypeoption != 0x00)
	{
		if (debuggeroption == 0x01)
		{
			switch(gecko_channel)
			{
				case 0: // Slot A
					memset((void*)0x80001800,0,codehandlerslota_size);
					memcpy((void*)0x80001800,codehandlerslota,codehandlerslota_size);
					if (pausedstartoption == 0x01)
						*(u32*)0x80002798 = 1;
					memcpy((void*)0x80001CDE, &codelist, 2);
					memcpy((void*)0x80001CE2, ((u8*) &codelist) + 2, 2);
					memcpy((void*)0x80001F5A, &codelist, 2);
					memcpy((void*)0x80001F5E, ((u8*) &codelist) + 2, 2);
					DCFlushRange((void*)0x80001800,codehandlerslota_size);
					break;
					
				case 1:	// slot B
					memset((void*)0x80001800,0,codehandler_size);
					memcpy((void*)0x80001800,codehandler,codehandler_size);
					if (pausedstartoption == 0x01)
						*(u32*)0x80002798 = 1;
					memcpy((void*)0x80001CDE, &codelist, 2);
					memcpy((void*)0x80001CE2, ((u8*) &codelist) + 2, 2);
					memcpy((void*)0x80001F5A, &codelist, 2);
					memcpy((void*)0x80001F5E, ((u8*) &codelist) + 2, 2);
					DCFlushRange((void*)0x80001800,codehandler_size);
					break;
					
				case 2:
					memset((void*)0x80001800,0,codehandler_size);
					memcpy((void*)0x80001800,codehandler,codehandler_size);
					if (pausedstartoption == 0x01)
						*(u32*)0x80002798 = 1;
					memcpy((void*)0x80001CDE, &codelist, 2);
					memcpy((void*)0x80001CE2, ((u8*) &codelist) + 2, 2);
					memcpy((void*)0x80001F5A, &codelist, 2);
					memcpy((void*)0x80001F5E, ((u8*) &codelist) + 2, 2);
					DCFlushRange((void*)0x80001800,codehandler_size);
					break;
			}
		}
		else
		{
			memset((void*)0x80001800,0,codehandleronly_size);
			memcpy((void*)0x80001800,codehandleronly,codehandleronly_size);
			memcpy((void*)0x80001906, &codelist, 2);
			memcpy((void*)0x8000190A, ((u8*) &codelist) + 2, 2);
			DCFlushRange((void*)0x80001800,codehandleronly_size);
		}
		// Load multidol handler
		memset((void*)0x80001000,0,multidol_size);
		memcpy((void*)0x80001000,multidol,multidol_size); 
		DCFlushRange((void*)0x80001000,multidol_size);
		switch(hooktypeoption)
		{
			case 0x01:
				memcpy((void*)0x8000119C,viwiihooks,12);
				memcpy((void*)0x80001198,viwiihooks+3,4);
				break;
			case 0x02:
				memcpy((void*)0x8000119C,kpadhooks,12);
				memcpy((void*)0x80001198,kpadhooks+3,4);
				break;
			case 0x03:
				memcpy((void*)0x8000119C,joypadhooks,12);
				memcpy((void*)0x80001198,joypadhooks+3,4);
				break;
			case 0x04:
				memcpy((void*)0x8000119C,gxdrawhooks,12);
				memcpy((void*)0x80001198,gxdrawhooks+3,4);
				break;
			case 0x05:
				memcpy((void*)0x8000119C,gxflushhooks,12);
				memcpy((void*)0x80001198,gxflushhooks+3,4);
				break;
			case 0x06:
				memcpy((void*)0x8000119C,ossleepthreadhooks,12);
				memcpy((void*)0x80001198,ossleepthreadhooks+3,4);
				break;
			case 0x07:
				memcpy((void*)0x8000119C,axnextframehooks,12);
				memcpy((void*)0x80001198,axnextframehooks+3,4);
				break;
			case 0x08:
				//if (customhooksize == 16)
				//{
				//	memcpy((void*)0x8000119C,customhook,12);
				//	memcpy((void*)0x80001198,customhook+3,4);
				//}
				break;
			case 0x09:
				//memcpy((void*)0x8000119C,wpadbuttonsdownhooks,12);
				//memcpy((void*)0x80001198,wpadbuttonsdownhooks+3,4);
				break;
			case 0x0A:
				//memcpy((void*)0x8000119C,wpadbuttonsdown2hooks,12);
				//memcpy((void*)0x80001198,wpadbuttonsdown2hooks+3,4);
				break;
		}
		DCFlushRange((void*)0x80001198,16);
	}
}


void do_codes(u64 titleid)
{
	s32 ret = 0;
	char gameidbuffer[8];
	memset(gameidbuffer, 0, 8);
	gameidbuffer[0] = (titleid & 0xff000000) >> 24;
	gameidbuffer[1] = (titleid & 0x00ff0000) >> 16;
	gameidbuffer[2] = (titleid & 0x0000ff00) >> 8;
	gameidbuffer[3] = titleid & 0x000000ff;

	if (debuggeroption == 0x00)
		codelist = (u8 *) 0x800022A8;
	else
		codelist = (u8 *) 0x800028B8;

	load_handler();
	memcpy((void *)0x80001800, gameidbuffer, 6);

	if (ocarinaoption != 0)
	{
		memset(codelist, 0, (u32)codelistend - (u32)codelist);
		
		ret = load_codes(gameidbuffer, (u32)codelistend - (u32)codelist, codelist);
		if (ret >= 0)
		{
			printf("Codes found. Applying\n");
			//write_font(185, 346, "Codes found. Applying");
		}
		sleep(3);
	}

	DCFlushRange((void*)0x80000000, 0x3f00);
}