#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <malloc.h>
#include <ogcsys.h>
#include <wiilight.h>
#include <unistd.h>
#include <runtimeiospatch.h>

#include "sys.h"
#include "fat.h"
#include "nand.h"
#include "restart.h"
#include "title.h"
#include "usbstorage.h"
#include "utils.h"
#include "video.h"
#include "wad.h"
#include "wpad.h"
#include <ogc/pad.h>
#include "globals.h"



/* FAT device list  */
//static fatDevice fdevList[] = {
fatDevice fdevList[] = {
	{ "sd",		"SD-Karte",						&__io_wiisd },
	{ "usb",	"USB-Speichermedium",			&__io_usbstorage },
	{ "usb2",	"USB2.0-Speichermedium",		&__io_wiiums },
	{ "gcsda",	"SD Gecko (Slot A)",			&__io_gcsda },
	{ "gcsdb",	"SD Gecko (Slot B)",			&__io_gcsdb },
	//{ "smb",	"SMB share",					NULL },
};

/* NAND device list */
//static nandDevice ndevList[] = {
nandDevice ndevList[] = {
	{ "Deaktiviert",				0,	0x00,	0x00 },
	{ "SD/SDHC Karte",				1,	0xF0,	0xF1 },
	{ "USB2.0-Speichermedium",		2,	0xF2,	0xF3 },
};

/* FAT device */
static fatDevice  *fdev = NULL;
static nandDevice *ndev = NULL;

// wiiNinja: Define a buffer holding the previous path names as user
// traverses the directory tree. Max of 10 levels is define at this point
static u8 gDirLevel = 0;
static char gDirList [MAX_DIR_LEVELS][MAX_FILE_PATH_LEN];
static s32  gSeleted[MAX_DIR_LEVELS];
static s32  gStart[MAX_DIR_LEVELS];

/* Macros */
#define NB_FAT_DEVICES		(sizeof(fdevList) / sizeof(fatDevice))
#define NB_NAND_DEVICES		(sizeof(ndevList) / sizeof(nandDevice))

// Local prototypes: wiiNinja
void WaitPrompt (char *prompt);
int PushCurrentDir(char *dirStr, s32 Selected, s32 Start);
char *PopCurrentDir(s32 *Selected, s32 *Start);
bool IsListFull (void);
char *PeekCurrentDir (void);
u32 WaitButtons(void);
u32 Pad_GetButtons(void);
void WiiLightControl (int state);

s32 __Menu_IsGreater(const void *p1, const void *p2)
{
	u32 n1 = *(u32 *)p1;
	u32 n2 = *(u32 *)p2;

	/* Equal */
	if (n1 == n2)
		return 0;

	return (n1 > n2) ? 1 : -1;
}


s32 __Menu_EntryCmp(const void *p1, const void *p2)
{
	fatFile *f1 = (fatFile *)p1;
	fatFile *f2 = (fatFile *)p2;

	/* Compare entries */ // wiiNinja: Include directory
    if ((f1->entry.d_type==DT_DIR) && !(f2->entry.d_type==DT_DIR))
        return (-1);
    else if (!(f1->entry.d_type==DT_DIR) && (f2->entry.d_type==DT_DIR))
        return (1);
    else
        return strcmp(f1->filename, f2->filename);
}

char gFileName[MAX_FILE_PATH_LEN];
s32 __Menu_RetrieveList(char *inPath, fatFile **outbuf, u32 *outlen)
{
	fatFile  *buffer = NULL;
	DIR *dir = NULL;
	
	struct dirent *entry;

	//struct stat filestat;

	//char dirpath[256], filename[768];
	u32  cnt = 0;

	/* Generate dirpath */
	//sprintf(dirpath, "%s:" WAD_DIRECTORY, fdev->mount);

	/* Open directory */
	dir = opendir(inPath);
	if (!dir)
		return -1;
		
	while ((entry = readdir(dir))) {
		cnt++;
	}

	/* Count entries */
	/*for (cnt = 0; !dirnext(dir, gFileName, &filestat);) {
		// if (!(filestat.st_mode & S_IFDIR)) // wiiNinja
			cnt++;
	}*/

	if (cnt > 0) {
		/* Allocate memory */
		buffer = malloc(sizeof(fatFile) * cnt);
		if (!buffer) {
			closedir(dir);
			return -2;
		}

		/* Reset directory */
		//dirreset(dir);
		closedir(dir);
		dir = opendir(inPath);

		/* Get entries */
		for (cnt = 0; (entry = readdir(dir));)
		{
			bool addFlag = false;

			if (entry->d_type==DT_DIR) 
            {
                // Add only the item ".." which is the previous directory
                // AND if we're not at the root directory
                if ((strcmp (entry->d_name, "..") == 0) && (gDirLevel > 1))
                    addFlag = true;
                else if (strcmp (entry->d_name, ".") != 0)
                    addFlag = true;
            }
            else
			{
				if(strlen(entry->d_name)>4)
				{
					if(!stricmp(entry->d_name+strlen(entry->d_name)-4, ".wad"))
						addFlag = true;
				}
			}

            if (addFlag == true)
            {
				fatFile *file = &buffer[cnt++];

				/* File name */
				strcpy(file->filename, entry->d_name);

				/* File stats */
				file->entry = *entry;
			}
		}

		/* Sort list */
		qsort(buffer, cnt, sizeof(fatFile), __Menu_EntryCmp);
	}

	/* Close directory */
	closedir(dir);

	/* Set values */
	*outbuf = buffer;
	*outlen = cnt;

	return 0;
}


void Menu_SelectIOS(void)
{
	u8 *iosVersion = NULL;
	u32 iosCnt;
	u8 tmpVersion;

	u32 cnt;
	s32 ret, selected = 0;
	bool found = false;

	/* Get IOS versions */
	ret = Title_GetIOSVersions(&iosVersion, &iosCnt);
	if (ret < 0)
		return;

	/* Sort list */
	qsort(iosVersion, iosCnt, sizeof(u8), __Menu_IsGreater);

	if (gConfig.cIOSVersion < 0)
		tmpVersion = CIOS_VERSION;
	else
	{
		tmpVersion = (u8)gConfig.cIOSVersion;
		// For debugging only
		//printf ("User pre-selected cIOS: %i\n", tmpVersion);
		//WaitButtons();
	}

	/* Set default version */
	for (cnt = 0; cnt < iosCnt; cnt++) {
		u8 version = iosVersion[cnt];

		/* Custom IOS available */
		//if (version == CIOS_VERSION)
		if (version == tmpVersion)
		{
			selected = cnt;
			found = true;
			break;
		}

		/* Current IOS */
		if (version == IOS_GetVersion())
			selected = cnt;
	}

	/* Ask user for IOS version */
	if ((gConfig.cIOSVersion < 0) || (found == false))
	{
		for (;;)
		{
			/* Clear console */
			Con_Clear();

			printf("\t>> W%chle das zu benutzende IOS: < IOS%d >\n\n", 132, iosVersion[selected]);

			printf("\t   Dr%ccke LINKS/RECHTS um die IOS Version zu wechseln.\n\n",129);

			printf("\t   Dr%ccke (A) zum Fortfahren.\n",129);
			printf("\t   Dr%ccke (HOME) zum Beenden.\n",129);

			u32 buttons = WaitButtons();

			/* LEFT/RIGHT buttons */
			if (buttons & WPAD_BUTTON_LEFT) {
				if ((--selected) <= -1)
					selected = (iosCnt - 1);
			}
			if (buttons & WPAD_BUTTON_RIGHT) {
				if ((++selected) >= iosCnt)
					selected = 0;
			}

			/* HOME button */
			if (buttons & WPAD_BUTTON_HOME)
				Restart();

			/* A button */
			if (buttons & WPAD_BUTTON_A)
				break;
		}
	}


	u8 version = iosVersion[selected];

	if (IOS_GetVersion() != version) {
		/* Shutdown subsystems */
		Wpad_Disconnect();
		//mload_close();

		/* Load IOS */

		if(!loadIOS(version)) Wpad_Init(), Menu_SelectIOS();

		/* Initialize subsystems */
		Wpad_Init();
	}
}

void Menu_FatDevice(void)
{
	s32 ret, selected = 0;

	/* Unmount FAT device */
	//if (fdev)
		//Fat_Unmount(fdev);
	//if (((fdevList[selected].mount[0] == 's') && (ndev->name[0] == 'S')))
		//selected++;

	/* Select source device */
	if (gConfig.fatDeviceIndex < 0)
	{
		for (;;) {
			/* Clear console */
			Con_Clear();

			/* Selected device */
			fdev = &fdevList[selected];

			printf("\t>> W%chle den WAD-Speicherort: < %s >\n\n", 132, fdev->name);

			printf("\t   Dr%ccke LINKS/RECHTS um das Ger%ct zu wechseln.\n\n",129,132);

			printf("\t   Dr%ccke (A) zum Fortfahren.\n",129);
			printf("\t   Dr%ccke (HOME) zum Beenden.\n\n",129);

			u32 buttons = WaitButtons();

			/* LEFT/RIGHT buttons */
			if (buttons & WPAD_BUTTON_LEFT) {
				if ((--selected) <= -1)
					selected = (NB_FAT_DEVICES - 1);
				if ((fdevList[selected].mount[0] == 's') && (ndev->name[0] == 'S'))
					selected--;
				if ((fdevList[selected].mount[0] == 'u' && fdevList[selected].mount[3] == '2') && (ndev->name[0] == 'U'))
					selected--;
				if ((selected) <= -1)
					selected = (NB_FAT_DEVICES - 1);
			}
			if (buttons & WPAD_BUTTON_RIGHT) {
				if ((++selected) >= NB_FAT_DEVICES)
					selected = 0;
				if ((fdevList[selected].mount[0] == 's') && (ndev->name[0] == 'S'))
					selected++;
				if ((fdevList[selected].mount[0] == 'u' && fdevList[selected].mount[3] == '2') && (ndev->name[0] == 'U'))
					selected++;
			}

			/* HOME button */
			if (buttons & WPAD_BUTTON_HOME)
				Restart();

			/* A button */
			if (buttons & WPAD_BUTTON_A)
				break;
		}
	}
	else
	{
		sleep(5);
		fdev = &fdevList[gConfig.fatDeviceIndex];
	}

	printf("[+] Lade Ger%ct %s, bitte warten...",132, fdev->name);
	fflush(stdout);

	/* Mount FAT device */

	ret = Fat_Mount(fdev);
	if (ret < 0) {
		printf("\n    FEHLER! (ret = %d)\n", ret);
		goto err;
	} else
		printf("\n    OK!\n");

	return;

err:

    if(gConfig.fatDeviceIndex >= 0) gConfig.fatDeviceIndex = -1;
	WiiLightControl (WII_LIGHT_OFF);
	printf("\n");
	printf("    Dr%ccke eine Taste zum Fortfahren...\n",129);

	WaitButtons();

	/* Prompt menu again */
	Menu_FatDevice();
}

void Menu_NandDevice(void)
{
	s32 ret, selected = 0;

	/* Disable NAND emulator */
	if (ndev) {
		Nand_Unmount(ndev);
		Nand_Disable();
	}

	/* Select source device */
	if (gConfig.nandDeviceIndex < 0)
	{
		for (;;) {
			/* Clear console */
			Con_Clear();

			/* Selected device */
			ndev = &ndevList[selected];

			printf("\t>> W%chle NAND-Emulation: < %s >\n\n",132, ndev->name);

			printf("\t   Dr%ccke LINKS/RECHTS um das Ger%ct zu wechseln.\n\n",129,132);

			printf("\t   Dr%ccke (A) zum Fortfahren.\n",129);
			printf("\t   Dr%ccke (HOME) zum Beenden.\n\n",129);

			u32 buttons = WaitButtons();

			/* LEFT/RIGHT buttons */
			if (buttons & WPAD_BUTTON_LEFT) {
				if ((--selected) <= -1)
					selected = (NB_NAND_DEVICES - 1);
			}
			if (buttons & WPAD_BUTTON_RIGHT) {
				if ((++selected) >= NB_NAND_DEVICES)
					selected = 0;
			}

			/* HOME button */
			if (buttons & WPAD_BUTTON_HOME)
				Restart();

			/* A button */
			if (buttons & WPAD_BUTTON_A)
				break;
		}
	}
	else
	{
		ndev = &ndevList[gConfig.nandDeviceIndex];
	}

	/* No NAND device */
	if (!ndev->mode)
		return;

	printf("[+] Aktiviere NAND-Emulation...");
	fflush(stdout);

	/* Mount NAND device */
	ret = Nand_Mount(ndev);
	if (ret < 0) {
		printf("\n    FEHLER! (ret = %d)\n", ret);
		goto err;
	}

	/* Enable NAND emulator */
	ret = Nand_Enable(ndev);
	if (ret < 0) {
		printf("\n    FEHLER! (ret = %d)\n", ret);
		goto err;
	} else
		printf("\n    OK!\n");

	return;

err:
	printf("\n");
	printf("    Dr%ccke eine Taste zum Fortfahren...\n",129);

	WaitButtons();

	/* Prompt menu again */
	Menu_NandDevice();
}

char gTmpFilePath[MAX_FILE_PATH_LEN];
/* Install and/or Uninstall multiple WADs - Leathl */
int Menu_BatchProcessWads(fatFile *files, int fileCount, char *inFilePath, int installCnt, int uninstallCnt)
{
	int count;

	for (;;)
	{
		Con_Clear();

		if ((installCnt > 0) & (uninstallCnt == 0)) {
			printf("[+] %d Datei%s ausgew%chlt zum Installieren.\n\n", installCnt, (installCnt == 1) ? "" : "en", 132);
			printf("    Wollen Sie Fortfahren?\n");
		}
		else if ((installCnt == 0) & (uninstallCnt > 0)) {
			printf("[+] %d Datei%s ausgew%chlt zum Deinstallieren.\n\n", uninstallCnt, (uninstallCnt == 1) ? "" : "en", 132);
			printf("    Wollen Sie Fortfahren?\n");
		}
		else {
			printf("[+] %d Datei%s ausgew%chlt zum Installieren.\n", installCnt, (installCnt == 1) ? "" : "en", 132);
			printf("    %d Datei%s ausgew%chlt zum Deinstallieren.\n\n", uninstallCnt, (uninstallCnt == 1) ? "" : "en", 132);
			printf("    Wollen Sie Fortfahren?\n");
		}

		printf("\n    Dr%ccke (A) zum Fortfahren.\n",129);
		printf("    Dr%ccke [B] f%cr den Dateibrowser.\n",129,129);

		u32 buttons = WaitButtons();

		if (buttons & WPAD_BUTTON_A)
			break;

		if (buttons & WPAD_BUTTON_B)
			return 0;
	}

	WiiLightControl (WII_LIGHT_ON);
	int errors = 0;
	int success = 0;
	s32 ret;

	for (count = 0; count < fileCount; count++)
	{
		fatFile *thisFile = &files[count];

		if ((thisFile->install == 1) | (thisFile->install == 2)) {
			int mode = thisFile->install;
			Con_Clear();
			printf("[+] %cffne \"%s\", bitte warten...",153, thisFile->filename);

			sprintf(gTmpFilePath, "%s/%s", inFilePath, thisFile->filename);

			FILE *fp = fopen(gTmpFilePath, "rb");
			if (!fp)
			{
				printf("\n    FEHLER!\n");
				errors += 1;
				continue;
			}

			printf("[+] %s WAD, bitte warten...\n", (mode == 2) ? "Deinstalliere" : "Installiere");
			if (mode == 2)
				ret = Wad_Uninstall(fp);
			else
				ret = Wad_Install(fp, thisFile);

			if (ret < 0) errors += 1;
			else success += 1;

			thisFile->installstate = ret;

			if (fp)
				fclose(fp);
		}
	}

	WiiLightControl (WII_LIGHT_OFF);

	printf("\n");
	printf("    %d Dateien installiert und %d fehlgeschlagen...\n", success, errors);

	if (errors > 0)
	{
		printf("\n    Einige Installationen fehlgeschlagen");
		printf("\n    Dr%ccke (A) um diese aufzulisten.\n", 129);
		printf("    Dr%ccke [B] zum %cberspringen.\n", 129, 154);

		u32 buttons = WaitButtons();

		if ((buttons & WPAD_BUTTON_A))
		{
			Con_Clear();

			int i=0;
			for (count = 0; count < fileCount; count++)
			{
				fatFile *thisFile = &files[count];

				if (thisFile->installstate <0)
				{
					char str[41];
					strncpy(str, thisFile->filename, 40); //Only 40 chars to fit the screen
					str[40]=0;
					i++;
					if(thisFile->installstate == -999) printf("    %s BRICK BLOCKED\n", str);
					else if(thisFile->installstate == -998) printf("    %s %cbersprungen\n", str, 129);
					else if(thisFile->installstate == -106) printf("    %s nicht installiert?\n", str);
					else if(thisFile->installstate == -1036 ) printf("    %s benötigtes IOS fehlt\n", str);
					else if(thisFile->installstate == -4100 ) printf("    %s kein trucha bug?\n", str);
					else printf("    %s Fehler %d\n", str, thisFile->installstate);
					if( i == 17 )
					{
						printf("\n    Dr%ccke eine Taste zum Fortfahren...\n",129);
						WaitButtons();
						i = 0;
					}
				}
			}
		}
	}
	printf("\n    Dr%ccke eine Taste zum Fortfahren...\n",129);
	WaitButtons();

	return 1;
}

/* File Operations - Leathl */
int Menu_FileOperations(fatFile *file, char *inFilePath)
{
	//f32 filesize = (file->filestat.st_size / MB_SIZE);

	for (;;)
	{
		Con_Clear();

		printf("[+] WAD-Dateiname : %s\n",          file->filename);
		//printf("    WAD-Dateigr%csse : %.2f MB\n\n",148, filesize);


		printf("[+] W%chle Aktion: < %s WAD >\n\n", 132, "L%csche"); //There's yet nothing else than delete

		printf("    Dr%ccke LINKS/RECHTS um die Aktion zu wechseln.\n\n",129);

		printf("    Dr%ccke (A) zum Fortfahren.\n",129);
		printf("    Dr%ccke [B] f%cr den Dateibrowser.\n",129,129);

		u32 buttons = WaitButtons();

		/* A button */
		if (buttons & WPAD_BUTTON_A)
			break;

		/* B button */
		if (buttons & WPAD_BUTTON_B)
			return 0;
	}

	Con_Clear();

	printf("[+] L%cschen \"%s\", bitte warten...\n", 148, file->filename);

	sprintf(gTmpFilePath, "%s/%s", inFilePath, file->filename);
	int error = remove(gTmpFilePath);
	if (error != 0)
		printf("    FEHLER!");
	else
		printf("    Erfolgreich gel%cscht!", 148);

	printf("\n");
	printf("    Dr%ccke eine Taste zum Fortfahren...\n", 129);

	WaitButtons();

	return !error;
}

void Menu_WadEdit(fatFile *file)
{
	/******************************************************************************************/
	// ios liste
	/******************************************************************************************/
	u8 *iosVersion = NULL;
	u32 iosCnt;

	u32 cnt;
	s32 ret, ios_selected = 0;

	/* Get IOS versions */
	ret = Title_GetIOSVersions(&iosVersion, &iosCnt);
	if (ret < 0)
		return;

	/* Sort list */
	qsort(iosVersion, iosCnt, sizeof(u8), __Menu_IsGreater);

	/* Set default version */
	for (cnt = 0; cnt < iosCnt; cnt++)
	{
		u8 version = iosVersion[cnt];

		/* Channel IOS */
		if (version == file->new_ios)
		{
			ios_selected = cnt;
			break;
		}		

		/* Current IOS */
		if (version == IOS_GetVersion())
			ios_selected = cnt;
	}
	/******************************************************************************************/

	int selected = 0;
	bool ahbprot = file->new_ahbprot;
	bool pass = file->new_pass;
	char ios_text[30];
	
	while(1)
	{
		/* Clear console */
		Con_Clear();
		
		if(iosVersion[ios_selected] == 58)
			sprintf(ios_text, "(USB2, empfohlen)");
		else
			ios_text[0] = '\0';
		
		printf("[+] Erweiterte Einstellungen\n\n");
		
		printf("%2s  Start-IOS des Kanals:   %i    %s\n", (selected == 0) ? ">>" : "  ",  iosVersion[ios_selected], ios_text);
		printf("%2s  AHBPROT aktivieren:     %s\n", (selected == 1) ? ">>" : "  ",  (!ahbprot) ? "Nein" : "Ja");
		printf("%2s  Altersbeschr%cnkung:     %s\n\n", (selected == 2) ? ">>" : "  ", 132, (!pass) ? "Nein" : "Ja");
		
		printf("    Dr%ccke LINKS/RECHTS um die Aktion zu wechseln.\n\n",129);
		
		Con_FgColor(1, 1);
		printf("    ACHTUNG: Bei Kan%clen bzw Anwendungen mit IOS-RELOAD,\n             geht der AHBPROT-Flag verloren.\n\n", 132);
		
		Con_FgColor(7, 1);
		printf("    Dr%ccke (A) zum Speichern.\n",129);
		printf("    Dr%ccke [B] zum Abbrechen.\n\n",129);
		
		u32 buttons = Wpad_WaitButtons();

		/* UP button */
		if (buttons & WPAD_BUTTON_UP && selected > 0)
			selected--;

		/* DOWN button */
		if (buttons & WPAD_BUTTON_DOWN && selected < 2)
			selected++;

		/* LEFT/RIGHT buttons */
		if (buttons & (WPAD_BUTTON_LEFT | WPAD_BUTTON_RIGHT))
		{
			if(selected == 0)
			{
				if(buttons & WPAD_BUTTON_LEFT && --ios_selected <= -1)
					ios_selected = (iosCnt - 1);
						
				else if(buttons & WPAD_BUTTON_RIGHT && ++ios_selected >= iosCnt)
					ios_selected = 0;
			}
			
			else if(selected == 1)
				ahbprot ^= 1;
			
			else if(selected == 2)
				pass ^= 1;
		}
		
		/* A button */
		if (buttons & WPAD_BUTTON_A)
			break;

		/* B button */
		if (buttons & WPAD_BUTTON_B)
			return;
	}

	printf("    gespeichert\n\n");
	file->new_ios = iosVersion[ios_selected];
	file->new_ahbprot = ahbprot;
	file->new_pass = pass;

	/* Wait for button */
	printf("    Dr%ccke eine Taste zum Fortfahren...\n",129);
	Wpad_WaitButtons();
}

void Menu_WadManage(fatFile *file, char *inFilePath)
{
	FILE *fp  = NULL;

	//char filepath[128];
	//f32  filesize = 0;
	u32  mode = 0;

	/* File size in megabytes */
	//filesize = (file->filestat.st_size / MB_SIZE);
	
	for (;;) {
		/* Clear console */
		Con_Clear();

		printf("[+] WAD-Dateiname : %s\n",          file->filename);
		//printf("    WAD-Dateigr%csse : %.2f MB\n\n",148, filesize);
		

		printf("[+] W%chle Aktion: < %s WAD >\n\n",132, (!mode) ? "Installiere" : "Deinstalliere");

		printf("    Dr%ccke LINKS/RECHTS um die Aktion zu wechseln.\n\n",129);

		printf("    Dr%ccke (A) zum Fortfahren.\n",129);
		printf("    Dr%ccke [B] f%cr den Dateibrowser.\n",129,129);
		if(file->new_ios)
			printf("    Dr%ccke (1) f%cr erweiterte Einstellungen.",129,129);

		u32 buttons = WaitButtons();

		/* LEFT/RIGHT buttons */
		if (buttons & (WPAD_BUTTON_LEFT | WPAD_BUTTON_RIGHT))
			mode ^= 1;

		/* A button */
		if (buttons & WPAD_BUTTON_A)
			break;

		/* B button */
		if (buttons & WPAD_BUTTON_B)
			return;

		/* 1 button */
		if (buttons & WPAD_BUTTON_1 && file->new_ios)
			Menu_WadEdit(file);
	}

	/* Clear console */
	Con_Clear();

	printf("[+] %cffne \"%s\", bitte warten...",153, file->filename);
	fflush(stdout);

	/* Generate filepath */
	// sprintf(filepath, "%s:" WAD_DIRECTORY "/%s", fdev->mount, file->filename);
	sprintf(gTmpFilePath, "%s/%s", inFilePath, file->filename); // wiiNinja

	/* Open WAD */
	fp = fopen(gTmpFilePath, "rb");
	if (!fp)
	{
		printf(" FEHLER!\n");
		goto out;
	}
	else
		printf(" OK!\n\n");

	printf("[+] %s WAD, bitte warten...\n", (!mode) ? "Installiere" : "Deinstalliere");

	/* Do install/uninstall */
	WiiLightControl (WII_LIGHT_ON);
	if (!mode)
		Wad_Install(fp, file);
	else
		Wad_Uninstall(fp);
	WiiLightControl (WII_LIGHT_OFF);

out:
	/* Close file */
	if (fp)
		fclose(fp);

	printf("\n");
	printf("    Dr%ccke eine Taste zum Fortfahren...\n",129);

	/* Wait for button */
	WaitButtons();
}

void Menu_WadList(void)
{
	char str [100];
	fatFile *fileList = NULL;
	u32      fileCnt;
	s32 ret, selected = 0, start = 0;
    char *tmpPath = malloc (MAX_FILE_PATH_LEN);
	int installCnt = 0;
	int uninstallCnt = 0;

	//fatFile *installFiles = malloc(sizeof(fatFile) * 50);
	//int installCount = 0;

    // wiiNinja: check for malloc error
    if (tmpPath == NULL)
    {
		printf(" Fehler beim lesen des Verzeichnisses");
        return;
    }

	printf("[+] Lade Dateiliste...");
	fflush(stdout);

	gDirLevel = 0;

    // wiiNinja: The root is always the primary folder
    // But if the user has a /wad directory, just go there. This makes
    // both sides of the argument win
	sprintf(tmpPath, "%s:" WAD_DIRECTORY, fdev->mount);
    PushCurrentDir(tmpPath,0,0);
	//if (strcmp (WAD_DIRECTORY, WAD_ROOT_DIRECTORY) != 0)
	if (strcmp (WAD_DIRECTORY, gConfig.startupPath) != 0)
	{
        // If the directory can be successfully opened, it must exists
        //DIR_ITER *tmpDirPtr = NULL;
        //tmpDirPtr = diropen(WAD_ROOT_DIRECTORY);
        //if (tmpDirPtr)
        //{
		//	dirclose (tmpDirPtr);

            // Now push the /wad directory as the current operating folder
            //sprintf(tmpPath, "%s:" WAD_ROOT_DIRECTORY, fdev->mount);
			sprintf(tmpPath, "%s:%s", fdev->mount, gConfig.startupPath);
			//printf ("\nThe final startupPath is: %s\n", tmpPath);
			//WaitButtons ();
            PushCurrentDir(tmpPath,0,0); // wiiNinja
        //}
	}

	/* Retrieve filelist */
getList:
    if (fileList)
    {
        free (fileList);
        fileList = NULL;
    }

	ret = __Menu_RetrieveList(tmpPath, &fileList, &fileCnt);
	if (ret < 0) {
		printf(" FEHLER! (ret = %d)\n", ret);
		goto err;
	}

	/* No files */
	if (!fileCnt) {
		printf(" Keine Dateien/Verzeichnisse gefunden!\n");
		goto err;
	}

	/* Set install-values to 0 - Leathl */
	int counter;
	for (counter = 0; counter < fileCnt; counter++) {
		fatFile *file = &fileList[counter];
		file->install = 0;
		
		sprintf(gTmpFilePath, "%s/%s", tmpPath, file->filename);
		if (file->entry.d_type==DT_REG)
			Wad_Read(fopen(gTmpFilePath, "r"), &file->old_ios, &file->old_ahbprot, &file->old_pass, &file->high_id, &file->low_id);
		
		file->new_ios = file->old_ios;
		file->new_ahbprot = file->old_ahbprot;
		file->new_pass = file->old_pass;
	}

	for (;;)
	{
		u32 cnt;
		s32 index;

		/* Clear console */
		Con_Clear();

		/** Print entries **/
		cnt = strlen(tmpPath);
		if(cnt>26)
			index = cnt-26;
		else
			index = 0;

		printf("[+] WAD Datein in [%s]:", tmpPath+index);
		Con_FgColor(2, 1);
		printf("\x1b[0;48H");
		printf("IOS  HW     Pass\n\n");
		Con_FgColor(7, 1);

		/* Print entries */
		for (cnt = start; cnt < fileCnt; cnt++)
		{
			fatFile *file     = &fileList[cnt];
			
			// zeile markieren
			if(cnt == selected)
				Con_BgColor(6, 0);
			else
				Con_BgColor(0, 0);

			/* Entries per page limit */
			if ((cnt - start) >= ENTRIES_PER_PAGE)
				break;

			strncpy(str, file->filename, 40); //Only 40 chars to fit the screen
			str[40]=0;
			
			/* Print filename */
		    if (file->entry.d_type==DT_DIR) // wiiNinja
				printf(" %2s  %-42s (VERZEICHNIS)     \n", (cnt == selected) ? ">>" : "  ", str);
            else if(file->new_ios)
				printf(" %2s%s %-42s %-4i %-6s %-6s\n", (cnt == selected) ? ">>" : "  ", (file->install == 1) ? "+" : ((file->install == 2) ? "-" : " "), str, file->new_ios, (file->new_ahbprot) ? "Ja" : "Nein", (file->new_pass) ? "Ja" : "Nein");
			else
				printf(" %2s%s %-61s\n", (cnt == selected) ? ">>" : "  ", (file->install == 1) ? "+" : ((file->install == 2) ? "-" : " "), str);
		}

		Con_BgColor(0, 0);
		Con_FgColor(2, 1);
		printf("\n    IOS = Kanal-IOS | HW = HW_AHBPROT | Pass = Altersbeschr%cnkung\n", 132);
		Con_FgColor(7, 1);
		printf("    Dr%ccke (A) um das Verzeichnis oder die WAD-Datei zu %cffnen.\n", 129, 148);
		if(gDirLevel>1)
			printf("    Dr%ccke [B] um das Verzeichniss zu verlassen.\n", 129);
		else
			printf("    Dr%ccke [B] um ein anderes Speichermedium zu w%chlen.\n", 129, 132);
		printf("    Dr%ccke (1) f%cr erweiterte Einstellungen.\n", 129, 129);
		printf("    Dr%ccke + und - zum markieren bzw. die Markierung zu entfernen.", 129);

			/** Controls **/
		u32 buttons = WaitButtons();

		/* DPAD buttons */
		if (buttons & (WPAD_BUTTON_UP | WPAD_BUTTON_LEFT))
		{
			selected -= (buttons & WPAD_BUTTON_LEFT) ? ENTRIES_PER_PAGE : 1;

			if (selected <= -1)
				selected = (fileCnt - 1);
		}
		if (buttons & (WPAD_BUTTON_DOWN | WPAD_BUTTON_RIGHT))
		{
			selected += (buttons & WPAD_BUTTON_RIGHT) ? ENTRIES_PER_PAGE : 1;

			if (selected >= fileCnt)
				selected = 0;
		}

		/* HOME button */
		if (buttons & WPAD_BUTTON_HOME)
			Restart();

		/* Plus Button - Leathl */
		if (buttons & WPAD_BUTTON_PLUS)
		{
			if(Wpad_TimeButton())
			{
			  installCnt = 0;
			  int i = 0;
			  while( i < fileCnt)
			  {
			  fatFile *file = &fileList[i];
			  if (((file->entry.d_type==DT_DIR) == false) & (file->install == 0)) {
				  file->install = 1;

				  installCnt += 1;
			  }
			  else if (((file->entry.d_type==DT_DIR) == false) & (file->install == 1)) {
				  file->install = 0;

				  installCnt -= 1;
			  }
			  else if (((file->entry.d_type==DT_DIR) == false) & (file->install == 2)) {
				  file->install = 1;

				  installCnt += 1;
				  uninstallCnt -= 1;
			  }
			  i++;
			  }

			}
			else
			{
			  fatFile *file = &fileList[selected];
			  if (((file->entry.d_type==DT_DIR) == false) & (file->install == 0)) {
				  file->install = 1;

				  installCnt += 1;
			  }
			  else if (((file->entry.d_type==DT_DIR) == false) & (file->install == 1)) {
				  file->install = 0;

				  installCnt -= 1;
			  }
			  else if (((file->entry.d_type==DT_DIR) == false) & (file->install == 2)) {
				  file->install = 1;

				  installCnt += 1;
				  uninstallCnt -= 1;
			  }
			  selected++;

			  if (selected >= fileCnt)
				selected = 0;
			}
		}

		/* Minus Button - Leathl */
		if (buttons & WPAD_BUTTON_MINUS)
		{

			if(Wpad_TimeButton())
			{
			  installCnt = 0;
			  int i = 0;
			  while( i < fileCnt)
			  {
			  fatFile *file = &fileList[i];
			  if (((file->entry.d_type==DT_DIR) == false) & (file->install == 0)) {
				file->install = 2;

				uninstallCnt += 1;
			  }
			  else if (((file->entry.d_type==DT_DIR) == false) & (file->install == 1)) {
				file->install = 2;

				uninstallCnt += 1;
				installCnt -= 1;
			  }
			  else if (((file->entry.d_type==DT_DIR) == false) & (file->install == 2)) {
				file->install = 0;

				uninstallCnt -= 1;
			  }
			  i++;
			  }

			}
			else
			{
			fatFile *file = &fileList[selected];
			if (((file->entry.d_type==DT_DIR) == false) & (file->install == 0)) {
				file->install = 2;

				uninstallCnt += 1;
			}
			else if (((file->entry.d_type==DT_DIR) == false) & (file->install == 1)) {
				file->install = 2;

				uninstallCnt += 1;
				installCnt -= 1;
			}
			else if (((file->entry.d_type==DT_DIR) == false) & (file->install == 2)) {
				file->install = 0;

				uninstallCnt -= 1;
			}
			 selected++;

			  if (selected >= fileCnt)
				selected = 0;
			}
		}

		/* 1 Button - Leathl */
		if (buttons & WPAD_BUTTON_1)
		{
			fatFile *tmpFile = &fileList[selected];
			if(tmpFile->new_ios)
			{
				char *tmpCurPath = PeekCurrentDir ();
				if (tmpCurPath != NULL) {
					Menu_WadEdit(tmpFile);
		//			int res = Menu_FileOperations(tmpFile, tmpCurPath);
		//          if (res != 0)
		//				goto getList;
				}
			}
		}


		/* A button */
		if (buttons & WPAD_BUTTON_A)
		{
				fatFile *tmpFile = &fileList[selected];
				char *tmpCurPath;
				if (tmpFile->entry.d_type==DT_DIR) // wiiNinja
				{
					if (strcmp (tmpFile->filename, "..") == 0)
					{
						selected = 0;
						start = 0;

						// Previous dir
						tmpCurPath = PopCurrentDir(&selected, &start);
						if (tmpCurPath != NULL)
							sprintf(tmpPath, "%s", tmpCurPath);

						installCnt = 0;
						uninstallCnt = 0;

						goto getList;
					}
					else if (IsListFull () == true)
					{
						WaitPrompt ("Maximale Anzahl an Unterordnern erreicht.\n");
					}
					else
					{
						tmpCurPath = PeekCurrentDir ();
						if (tmpCurPath != NULL)
						{
							if(gDirLevel>1)
								sprintf(tmpPath, "%s/%s", tmpCurPath, tmpFile->filename);
							else
								sprintf(tmpPath, "%s%s", tmpCurPath, tmpFile->filename);
						}
						// wiiNinja: Need to PopCurrentDir
						PushCurrentDir (tmpPath, selected, start);
						selected = 0;
						start = 0;

						installCnt = 0;
						uninstallCnt = 0;

						goto getList;
					}
				}
				else
				{
					//If at least one WAD is marked, goto batch screen - Leathl
					if ((installCnt > 0) | (uninstallCnt > 0)) {
						char *thisCurPath = PeekCurrentDir ();
						if (thisCurPath != NULL) {
							int res = Menu_BatchProcessWads(fileList, fileCnt, thisCurPath, installCnt, uninstallCnt);

							if (res == 1) {
								int counter;
								for (counter = 0; counter < fileCnt; counter++) {
									fatFile *temp = &fileList[counter];
									temp->install = 0;
								}

								installCnt = 0;
								uninstallCnt = 0;
							}
						}
					}
					//else use standard wadmanage menu - Leathl
					else {
						tmpCurPath = PeekCurrentDir ();
						if (tmpCurPath != NULL)
							Menu_WadManage(tmpFile, tmpCurPath);
					}
				}
		}

		/* B button */
		if (buttons & WPAD_BUTTON_B)
		{
			if(gDirLevel<=1)
			{
				return;
			}

			char *tmpCurPath;
			selected = 0;
			start = 0;
			// Previous dir
			tmpCurPath = PopCurrentDir(&selected, &start);
			if (tmpCurPath != NULL)
				sprintf(tmpPath, "%s", tmpCurPath);
			goto getList;
			//return;
		}

		/** Scrolling **/
		/* List scrolling */
		index = (selected - start);

		if (index >= ENTRIES_PER_PAGE)
			start += index - (ENTRIES_PER_PAGE - 1);
		if (index <= -1)
			start += index;
	}

err:
	printf("\n");
	printf("    Dr%ccke eine Taste zum Fortfahren...\n",129);

	free (tmpPath);

	/* Wait for button */
	WaitButtons();
}


void Menu_Loop(void)
{
	u8 iosVersion;
	
	/* Select IOS menu */
	if (AHBPROT_DISABLED)
	{
		
		u32  mode = 0;
		while(1)
		{
			/* Clear console */
			Con_Clear();
			printf("    HW_AHBPROT Flag vorhanden\n");
			printf("    trotzdem ein anderes IOS verwenden?: < %s >\n\n", (!mode) ? "Nein" : "Ja");
			printf("    Dr%ccke (A) zum Fortfahren.\n", 129);
			printf("    Dr%ccke (HOME) zum Beenden.\n", 129);
			u32 buttons = WaitButtons();
			/* LEFT/RIGHT buttons */
			if (buttons & (WPAD_BUTTON_LEFT | WPAD_BUTTON_RIGHT))
				mode ^= 1;

			/* HOME button */
			if (buttons & WPAD_BUTTON_HOME)
				Restart();

			/* A button */
			if (buttons & WPAD_BUTTON_A)
				break;
		}
		
		if(mode)
			Menu_SelectIOS();
	}
	else
		Menu_SelectIOS();

	/* Retrieve IOS version */
	iosVersion = IOS_GetVersion();

	ndev = &ndevList[0];

	/* NAND device menu */
	if ((iosVersion == CIOS_VERSION || iosVersion == 250) && IOS_GetRevision() >13)
	{
		Menu_NandDevice();
	}
	for (;;) {
		/* FAT device menu */
		Menu_FatDevice();

		/* WAD list menu */
		Menu_WadList();
	}
}

// Start of wiiNinja's added routines

int PushCurrentDir (char *dirStr, s32 Selected, s32 Start)
{
    int retval = 0;

    // Store dirStr into the list and increment the gDirLevel
    // WARNING: Make sure dirStr is no larger than MAX_FILE_PATH_LEN
    if (gDirLevel < MAX_DIR_LEVELS)
    {
        strcpy (gDirList [gDirLevel], dirStr);
		gSeleted[gDirLevel]=Selected;
		gStart[gDirLevel]=Start;
        gDirLevel++;
        //if (gDirLevel >= MAX_DIR_LEVELS)
        //    gDirLevel = 0;
    }
    else
        retval = -1;

    return (retval);
}

char *PopCurrentDir(s32 *Selected, s32 *Start)
{
	if (gDirLevel > 1)
        gDirLevel--;
    else
        gDirLevel = 0;

	*Selected = gSeleted[gDirLevel];
	*Start = gStart[gDirLevel];
	return PeekCurrentDir();
}

bool IsListFull (void)
{
    if (gDirLevel < MAX_DIR_LEVELS)
        return (false);
    else
        return (true);
}

char *PeekCurrentDir (void)
{
    // Return the current path
    if (gDirLevel > 0)
        return (gDirList [gDirLevel-1]);
    else
        return (NULL);
}

void WaitPrompt (char *prompt)
{
	printf("\n%s", prompt);
	printf("    Dr%ccke eine Taste zum Fortfahren...\n",129);

	/* Wait for button */
	WaitButtons();
}

u32 Pad_GetButtons(void)
{
	u32 buttons = 0, cnt;

	/* Scan pads */
	PAD_ScanPads();

	/* Get pressed buttons */
	//for (cnt = 0; cnt < MAX_WIIMOTES; cnt++)
	for (cnt = 0; cnt < 4; cnt++)
		buttons |= PAD_ButtonsDown(cnt);

	return buttons;
}


// Routine to wait for a button from either the Wiimote or a gamecube
// controller. The return value will mimic the WPAD buttons to minimize
// the amount of changes to the original code, that is expecting only
// Wiimote button presses. Note that the "HOME" button on the Wiimote
// is mapped to the "SELECT" button on the Gamecube Ctrl. (wiiNinja 5/15/2009)
u32 WaitButtons(void)
{
	u32 buttons = 0;
    u32 buttonsGC = 0;

	/* Wait for button pressing */
	while (!buttons && !buttonsGC)
    {
        // GC buttons
        buttonsGC = Pad_GetButtons ();

        // Wii buttons
		buttons = Wpad_GetButtons();

		VIDEO_WaitVSync();
	}

    if (buttonsGC)
    {
        if(buttonsGC & PAD_BUTTON_A)
        {
            //printf ("Button A on the GC controller\n");
            buttons |= WPAD_BUTTON_A;
        }
        else if(buttonsGC & PAD_BUTTON_B)
        {
            //printf ("Button B on the GC controller\n");
            buttons |= WPAD_BUTTON_B;
        }
        else if(buttonsGC & PAD_BUTTON_LEFT)
        {
            //printf ("Button LEFT on the GC controller\n");
            buttons |= WPAD_BUTTON_LEFT;
        }
        else if(buttonsGC & PAD_BUTTON_RIGHT)
        {
            //printf ("Button RIGHT on the GC controller\n");
            buttons |= WPAD_BUTTON_RIGHT;
        }
        else if(buttonsGC & PAD_BUTTON_DOWN)
        {
            //printf ("Button DOWN on the GC controller\n");
            buttons |= WPAD_BUTTON_DOWN;
        }
        else if(buttonsGC & PAD_BUTTON_UP)
        {
            //printf ("Button UP on the GC controller\n");
            buttons |= WPAD_BUTTON_UP;
        }
        else if(buttonsGC & PAD_BUTTON_START)
        {
            //printf ("Button START on the GC controller\n");
            buttons |= WPAD_BUTTON_HOME;
        }
    }

	return buttons;
} // WaitButtons


void WiiLightControl (int state)
{
	switch (state)
	{
		case WII_LIGHT_ON:
			/* Turn on Wii Light */
			WIILIGHT_SetLevel(255);
			WIILIGHT_TurnOn();
			break;

		case WII_LIGHT_OFF:
		default:
			/* Turn off Wii Light */
			WIILIGHT_SetLevel(0);
			WIILIGHT_TurnOn();
			WIILIGHT_Toggle();
			break;
	}
} // WiiLightControl