WiiFlow_Lite/source/homebrew/homebrew.cpp
fix94.1 cc7c5a9917 -changed homebrew loading yet again, not as safe as previous
way but maybe working better for bigger homebrew
2012-07-10 00:18:37 +00:00

159 lines
3.3 KiB
C++

#include <gccore.h>
#include <ogcsys.h>
#include <stdio.h>
#include <string.h>
#include <ogc/machine/processor.h>
#include <vector>
#include <string>
#include "smartptr.hpp"
#include "gecko.h"
#include "mem2.hpp"
#define EXECUTE_ADDR ((u8 *)0x92000000)
#define BOOTER_ADDR ((u8 *)0x93000000)
#define ARGS_ADDR ((u8 *)0x93200000)
using namespace std;
extern const u8 app_booter_bin[];
extern const u32 app_booter_bin_size;
extern const u8 stub_bin[];
extern const u32 stub_bin_size;
typedef void (*entrypoint) (void);
extern "C" { void __exception_closeall(); }
u32 buffer_size = 0;
static vector<string> Arguments;
static u32 stubtitlepositions[8] = { 0x80001bf2, 0x80001bf3, 0x80001c06, 0x80001c07,
0x80001bfa, 0x80001bfb, 0x80001c0a, 0x80001c0b };
bool IsDollZ (u8 *buff)
{
u8 dollz_stamp[] = {0x3C};
int dollz_offs = 0x100;
int ret = memcmp (&buff[dollz_offs], dollz_stamp, sizeof(dollz_stamp));
if (ret == 0)
return true;
return false;
}
void AddBootArgument(const char * argv)
{
string arg(argv);
Arguments.push_back(arg);
}
int LoadHomebrew(const char *filepath)
{
if(!filepath)
return -1;
FILE *file = fopen(filepath ,"rb");
if(!file)
return -2;
fseek(file, 0, SEEK_END);
u32 filesize = ftell(file);
rewind(file);
buffer_size = filesize;
fread(EXECUTE_ADDR, 1, buffer_size, file);
DCFlushRange(EXECUTE_ADDR, buffer_size);
fclose(file);
return 1;
}
static int SetupARGV(struct __argv * args)
{
if(!args)
return -1;
bzero(args, sizeof(struct __argv));
args->argvMagic = ARGV_MAGIC;
u32 argc = 0;
u32 position = 0;
u32 stringlength = 1;
/** Append Arguments **/
for(u32 i = 0; i < Arguments.size(); i++)
stringlength += Arguments[i].size()+1;
args->length = stringlength;
//! Put the argument into mem2 too, to avoid overwriting it
args->commandLine = (char *) ARGS_ADDR + sizeof(struct __argv);
/** Append Arguments **/
for(u32 i = 0; i < Arguments.size(); i++)
{
strcpy(&args->commandLine[position], Arguments[i].c_str());
position += Arguments[i].size() + 1;
argc++;
}
args->argc = argc;
args->commandLine[args->length - 1] = '\0';
args->argv = &args->commandLine;
args->endARGV = args->argv + 1;
Arguments.clear();
return 0;
}
static void writeStub(u64 chan_title)
{
u8 i;
u32 digit;
char title[2][9];
snprintf(title[0], sizeof(title[0]), "%08x", TITLE_UPPER(chan_title));
snprintf(title[1], sizeof(title[1]), "%08x", TITLE_LOWER(chan_title));
/* Clear potential homebrew channel stub */
memset((void*)0x80001800, 0, 0x1800);
/* Copy our own stub into memory */
memcpy((void*)0x80001800, stub_bin, stub_bin_size);
/* Write in the Title ID we got */
for(i = 0; i < 4; i++)
{
sscanf(&title[0][i*2], "%02x", &digit);
//gprintf("%x\n", digit);
*(vu8*)stubtitlepositions[i] = digit;
sscanf(&title[1][i*2], "%02x", &digit);
//gprintf("%x\n", digit);
*(vu8*)stubtitlepositions[i+4] = digit;
}
DCFlushRange((void*)0x80001800, stub_bin_size);
}
int BootHomebrew(u64 chan_title)
{
writeStub(chan_title);
struct __argv args;
if (!IsDollZ(EXECUTE_ADDR))
SetupARGV(&args);
memcpy(BOOTER_ADDR, app_booter_bin, app_booter_bin_size);
DCFlushRange(BOOTER_ADDR, app_booter_bin_size);
entrypoint entry = (entrypoint)BOOTER_ADDR;
memmove(ARGS_ADDR, &args, sizeof(args));
DCFlushRange(ARGS_ADDR, sizeof(args) + args.length);
SYS_ResetSystem(SYS_SHUTDOWN, 0, 0);
entry();
return 0;
}