usbloadergx/source/menu/menu_check.cpp

#include <dirent.h>
#include <unistd.h>

#include "menus.h"
#include "wpad.h"
#include "fatmounter.h"
#include "usbloader/wbfs.h"
#include "xml/xml.h"

extern int load_from_fs;
extern char game_partition[6];
extern char headlessID[8];

/****************************************************************************
 * MenuCheck
 ***************************************************************************/
int MenuCheck()
{
    gprintf("\nMenuCheck()");
    int menu = MENU_NONE;
    int i = 0;
    int choice;
    s32 ret2, wbfsinit;
    OptionList options;
    options.length = i;

    VIDEO_WaitVSync();

    wbfsinit = WBFS_Init(WBFS_DEVICE_USB);
    if (wbfsinit < 0)
    {
        ret2 = WindowPrompt(tr( "No USB Device found." ), tr( "Do you want to retry for 30 secs?" ), "cIOS249",
                "cIOS222", tr( "Back to Wii Menu" ));
        SDCard_deInit();
        USBDevice_deInit();
        WPAD_Flush(0);
        WPAD_Disconnect(0);
        WPAD_Shutdown();
        if (ret2 == 1)
        {
            Settings.cios = 249;
        }
        else if (ret2 == 2)
        {
            Settings.cios = 222;
        }
        else
        {
            Sys_LoadMenu();
        }
        ret2 = DiscWait(tr( "No USB Device" ), tr( "Waiting for USB Device" ), 0, 0, 1);
        //reinitialize SD and USB
        Wpad_Init();
        WPAD_SetDataFormat(WPAD_CHAN_ALL, WPAD_FMT_BTNS_ACC_IR);
        WPAD_SetVRes(WPAD_CHAN_ALL, screenwidth, screenheight);
        if (ret2 < 0)
        {
            WindowPrompt(tr( "Error !" ), tr( "USB Device not found" ), tr( "OK" ));
            Sys_LoadMenu();
        }
    }

    ret2 = -1;
    memset(game_partition, 0, 6);
    load_from_fs = -1;

    extern PartList partitions;
    // Added for slow HDD
    for (int runs = 0; runs < 10; runs++)
    {
        if (Partition_GetList(WBFS_DEVICE_USB, &partitions) != 0)
        {
            sleep(1);
            continue;
        }

        if (Settings.partition != -1 && partitions.num > Settings.partition)
        {
            PartInfo pinfo = partitions.pinfo[Settings.partition];
            if (!WBFS_OpenPart(pinfo.part_fs, pinfo.index, partitions.pentry[Settings.partition].sector,
                    partitions.pentry[Settings.partition].size, (char *) &game_partition))
            {
                ret2 = 0;
                load_from_fs = pinfo.part_fs;
                break;
            }
        }

        if (partitions.wbfs_n != 0)
        {
            ret2 = WBFS_Open();
            for (int p = 0; p < partitions.num; p++)
            {
                if (partitions.pinfo[p].fs_type == FS_TYPE_WBFS)
                {
                    Settings.partition = p;
                    load_from_fs = PART_FS_WBFS;
                    break;
                }
            }
        }
        else if (Sys_IsHermes() && (partitions.fat_n != 0 || partitions.ntfs_n != 0))
        {
            // Loop through FAT/NTFS partitions, and find the first partition with games on it (if there is one)
            u32 count;

            for (int i = 0; i < partitions.num; i++)
            {
                if (partitions.pinfo[i].fs_type == FS_TYPE_FAT32 || partitions.pinfo[i].fs_type == FS_TYPE_NTFS)
                {

                    if (!WBFS_OpenPart(partitions.pinfo[i].part_fs, partitions.pinfo[i].index,
                            partitions.pentry[i].sector, partitions.pentry[i].size, (char *) &game_partition))
                    {
                        // Get the game count...
                        WBFS_GetCount(&count);

                        if (count > 0)
                        {
                            load_from_fs = partitions.pinfo[i].part_fs;
                            Settings.partition = i;
                            break;
                        }
                        else
                        {
                            WBFS_Close();
                        }
                    }
                }
            }
        }

        if ((ret2 >= 0 || load_from_fs != PART_FS_WBFS) && isInserted(bootDevice))
        {
            Settings.Save();
            break;
        }
        sleep(1);
    }

    if (ret2 < 0 && load_from_fs != PART_FS_WBFS)
    {
        choice = WindowPrompt(tr( "No WBFS or FAT/NTFS partition found" ),
                tr( "You need to select or format a partition" ), tr( "Select" ), tr( "Format" ), tr( "Return" ));
        if (choice == 0)
        {
            Sys_LoadMenu();
        }
        else
        {
            load_from_fs = choice == 1 ? PART_FS_FAT : PART_FS_WBFS;
            menu = MENU_FORMAT;
        }
    }

    ret2 = Disc_Init();
    if (ret2 < 0)
    {
        WindowPrompt(tr( "Error !" ), tr( "Could not initialize DIP module!" ), tr( "OK" ));
        Sys_LoadMenu();
    }

    if (shutdown == 1) Sys_Shutdown();
    if (reset == 1) Sys_Reboot();

    if (wbfsinit < 0)
    {
        sleep(1);
    }

    // open database if needed, load titles if needed
    if (isInserted(bootDevice)) OpenXMLDatabase(Settings.titlestxt_path, Settings.db_language, Settings.db_JPtoEN,
            true, Settings.titlesOverride == 1 ? true : false, true);

    // titles.txt loaded after database to override database titles with custom titles
    //snprintf(pathname, sizeof(pathname), "%stitles.txt", Settings.titlestxt_path);
    //cfg_parsefile(pathname, &title_set);

    //Spieleliste laden
    //__Menu_GetEntries(0);//no point getting the gamelist here

    if (strcmp(headlessID, "") != 0) menu = MENU_EXIT;

    if (menu == MENU_NONE) menu = MENU_DISCLIST;

    //for HDDs with issues
    if (wbfsinit < 0)
    {
        sleep(1);
        USBDevice_Init();
        SDCard_Init();
    }

    return menu;
}