usbloadergx/source/main.cpp

/*
 *      loadMii loader v0.3
 *      main.c
 *      http://code.google.com/p/loadmii
 *
 *      Copyright 2009 The Lemon Man
 *      Thanks to luccax, Wiipower, Aurelio and crediar
 *      GX added by dimok (thanks to Tantric)
 *      usbGecko powered by Nuke
 *
 *      This program is free software; you can redistribute it and/or modify
 *      it under the terms of the GNU General Public License as published by
 *      the Free Software Foundation; either version 2 of the License, or
 *      (at your option) any later version.
 *
 *      This program is distributed in the hope that it will be useful,
 *      but WITHOUT ANY WARRANTY; without even the implied warranty of
 *      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *      GNU General Public License for more details.
 *
 *      You should have received a copy of the GNU General Public License
 *      along with this program; if not, write to the Free Software
 *      Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 *      MA 02110-1301, USA.
 */

#include <gccore.h>
#include <wiiuse/wpad.h>
#include <fat.h>
#include <sdcard/wiisd_io.h>
#include <malloc.h>
#include <stdio.h>
#include <stdlib.h>
#include <ogc/machine/processor.h>

#include "pngu/pngu.h"
#include "video.h"
#include "filelist.h"
#include "dolloader.h"
#include "elfloader.h"
#include "cfg.h"


PNGUPROP imgProp;
IMGCTX ctx;


u8 * GetImageData(void) {

	u8 * data = NULL;

	int ret;

	ctx = PNGU_SelectImageFromBuffer(background_png);
	if (!ctx)
		return NULL;

	ret = PNGU_GetImageProperties(ctx, &imgProp);
	if (ret != PNGU_OK)
		return NULL;

    int len = imgProp.imgWidth * imgProp.imgHeight * 4;

    if(len%32) len += (32-len%32);
    data = (u8 *)memalign (32, len);
    ret = PNGU_DecodeTo4x4RGBA8 (ctx, imgProp.imgWidth, imgProp.imgHeight, data, 255);
	DCFlushRange(data, len);

	PNGU_ReleaseImageContext(ctx);

    return data;
}

void Background_Show(int x, int y, int z, u8 * data, int angle, int scaleX, int scaleY, int alpha)
{
	/* Draw image */
	Menu_DrawImg(x, y, z, imgProp.imgWidth, imgProp.imgHeight, data, angle, scaleX, scaleY, alpha);
}


int main(int argc, char **argv) {

	u32 cookie;
	FILE *exeFile = NULL;
	void *exeBuffer          = (void *)EXECUTABLE_MEM_ADDR;
	int exeSize              = 0;
	u32 exeEntryPointAddress = 0;
	entrypoint exeEntryPoint;


    /* int videomod */
    InitVideo();

    /* get imagedata */
    u8 * imgdata = GetImageData();

    /* fadein of image */
    for(int i = 0; i < 255; i = i+10) {
    if(i>255) i = 255;
	Background_Show(0, 0, 0, imgdata, 0, 1, 1, i);
    Menu_Render();
    }
    /* check devices */
    __io_wiisd.startup();
	fatMount("SD", &__io_wiisd, 0, 32, 128);

    char cfgpath[40];
    sprintf(cfgpath, "SD:/config/GXGlobal.cfg");

	if(!cfg_parsefile(cfgpath, &cfg_set)) {
        /* Open dol File and check exist */
        exeFile = fopen ("SD:/apps/usbloader_gx/boot.dol" ,"rb");
        if (exeFile==NULL) {
            fclose(exeFile);
            exeFile = fopen ("SD:/apps/usbloader_gx/boot.elf" ,"rb");
            if (exeFile==NULL) {
            fclose(exeFile);
            SYS_ResetSystem(SYS_RETURNTOMENU, 0, 0);
            }
        }
	} else {
        sprintf(cfgpath, "%sboot.dol", update_path);
        /* Open dol File and check exist */
        exeFile = fopen (cfgpath, "rb");
        if (exeFile==NULL) {
            fclose(exeFile);
            sprintf(cfgpath, "%sboot.elf", update_path);
            exeFile = fopen (cfgpath,"rb");
            if (exeFile==NULL) {
            fclose(exeFile);
            SYS_ResetSystem(SYS_RETURNTOMENU, 0, 0);
            }
        }
	}

	fseek (exeFile, 0, SEEK_END);
	exeSize = ftell(exeFile);
	fseek (exeFile, 0, SEEK_SET);
	if(fread (exeBuffer, 1, exeSize, exeFile) != (unsigned int) exeSize) {
		printf("Can't open DOL File...\n");
		SYS_ResetSystem(SYS_RETURNTOMENU, 0, 0);;
	}

	fclose (exeFile);

    /* load entry point */
	struct __argv args[10];

	int ret = valid_elf_image(exeBuffer);
	if (ret == 1) {
        exeEntryPointAddress = load_elf_image(exeBuffer);
	} else {
        exeEntryPointAddress = load_dol_image(exeBuffer, args);
	}

    /* fadeout of image */
    for(int i = 255; i > 1; i = i-7) {
        if(i < 0) i = 0;
        Background_Show(0, 0, 0, imgdata, 0, 1, 1, i);
        Menu_Render();
	}

	fatUnmount("SD");
    __io_wiisd.shutdown();
    StopGX();

	if (exeEntryPointAddress == 0) {
		printf("EntryPointAddress failed...\n");
		SYS_ResetSystem(SYS_RETURNTOMENU, 0, 0);;
	}

	exeEntryPoint = (entrypoint) exeEntryPointAddress;

    /* cleaning up and load dol */
	SYS_ResetSystem(SYS_SHUTDOWN, 0, 0);
	_CPU_ISR_Disable (cookie);
	__exception_closeall ();
	exeEntryPoint ();
	_CPU_ISR_Restore (cookie);
	return 0;

}