usbloadergx/source/patches/wip.c

#include <gccore.h>
#include <string.h>
#include <stdlib.h>
#include <stdio.h>

#include "settings/cfg.h"

u32 doltableoffset[64];
u32 doltablelength[64];
u32 doltableentries;

void wipreset()
{
	doltableentries = 0;
}

void wipregisteroffset(u32 dst, u32 len)
{
	doltableoffset[doltableentries] = dst;
	doltablelength[doltableentries] = len;
	doltableentries++;
}

void patchu8(u32 offset, u8 value)
{
	u32 i = 0;
	u32 tempoffset = 0;

	while ((doltablelength[i] <= offset-tempoffset) && (i+1 < doltableentries))
	{
		tempoffset+=doltablelength[i];
		i++;
	}
	if (offset-tempoffset < doltablelength[i])
	{
		*(u8 *)(offset-tempoffset+doltableoffset[i]) = value;
	}
}

void wipparsebuffer(u8 *buffer, u32 length)
// The buffer needs a 0 at the end to properly terminate the string functions
{
	u32 pos = 0;
	u32 offset;
	char buf[10];
	
	while (pos < length)
	{
		if ( *(char *)(buffer + pos) != '#' && *(char *)(buffer + pos) != ';' && *(char *)(buffer + pos) != 10 && *(char *)(buffer + pos) != 13 && *(char *)(buffer + pos) != 32 && *(char *)(buffer + pos) != 0 )
		{
			memcpy(buf, (char *)(buffer + pos), 8);
			buf[8] = 0;
			offset = strtol(buf,NULL,16);

			pos += (u32)strchr((char *)(buffer + pos), 32)-(u32)(buffer + pos) + 1;
			pos += (u32)strchr((char *)(buffer + pos), 32)-(u32)(buffer + pos) + 1;
			
			while (*(char *)(buffer + pos) != 10 && *(char *)(buffer + pos) != 13 && *(char *)(buffer + pos) != 0)
			{
				memcpy(buf, (char *)(buffer + pos), 2);
				buf[2] = 0;
			
				patchu8(offset, strtol(buf,NULL,16));
				offset++;
				pos +=2;		
			}	
		}
		if (strchr((char *)(buffer + pos), 10) == NULL)
		{
			return;
		} else
		{
			pos += (u32)strchr((char *)(buffer + pos), 10)-(u32)(buffer + pos) + 1;
		}
	}
}

u32 do_wip_code(u8 *gameid)
{
	FILE *fp;
	u32 filesize;
	char filepath[150];
	memset(filepath, 0, 150);
	u8 *wipCode;

	sprintf(filepath, "%s%6s", Settings.WipCodepath, gameid);
	filepath[strlen(Settings.WipCodepath)+6] = '.';
	filepath[strlen(Settings.WipCodepath)+7] = 'w';
	filepath[strlen(Settings.WipCodepath)+8] = 'i';
	filepath[strlen(Settings.WipCodepath)+9] = 'p';

	fp = fopen(filepath, "rb");
	if (!fp) {
		memset(filepath, 0, 150);
		sprintf(filepath, "%s%3s", Settings.WipCodepath, gameid + 1);
		filepath[strlen(Settings.WipCodepath)+3] = '.';
		filepath[strlen(Settings.WipCodepath)+4] = 'w';
		filepath[strlen(Settings.WipCodepath)+5] = 'i';
		filepath[strlen(Settings.WipCodepath)+6] = 'p';
		fp = fopen(filepath, "rb");
		
		if (!fp) {
			return -1;
		}
	}

	if (fp) {
		u32 ret = 0;

		fseek(fp, 0, SEEK_END);
		filesize = ftell(fp);
		
		wipCode = malloc(filesize + 1);
		wipCode[filesize] = 0; // Wip code functions need a 0 termination
		
		fseek(fp, 0, SEEK_SET);
		ret = fread(wipCode, 1, filesize, fp);
		fclose(fp);
		
		if (ret == filesize)
		{
			// Apply wip patch
			wipparsebuffer(wipCode, filesize + 1);
			return 0;
		}
	}
	return -2;
}