usbloadergx/source/usbloader/wbfs.cpp

#include <ogcsys.h>
#include <unistd.h>
#include <time.h>

#include "usbloader/usbstorage2.h"
#include "fatmounter.h"
#include "wbfs.h"
#include "usbloader/wbfs/wbfs_base.h"
#include "usbloader/wbfs/wbfs_wbfs.h"
#include "usbloader/wbfs/wbfs_fat.h"
#include "usbloader/wbfs/wbfs_ntfs.h"

#include "usbloader/partition_usbloader.h"
#include "usbloader/GameList.h"
#include "gecko.h"

Wbfs *current = NULL;
#define DEBUG_WBFS

/* WBFS device */
s32 wbfsDev = WBFS_MIN_DEVICE;

// partition
char wbfs_fs_drive[16];
int wbfs_part_fs = PART_FS_WBFS;
u32 wbfs_part_idx = 0;
u32 wbfs_part_lba = 0;

wbfs_disc_t* WBFS_OpenDisc(u8 *discid)
{
    return current->OpenDisc(discid);
}

void WBFS_CloseDisc(wbfs_disc_t *disc)
{
    current->CloseDisc(disc);
}

wbfs_t *GetHddInfo(void)
{
    return current->GetHddInfo();
}

s32 WBFS_Init(u32 device)
{
    return Wbfs::Init(device);
}

s32 WBFS_Open(void)
{
    WBFS_Close();

    current = new Wbfs_Wbfs(WBFS_DEVICE_USB, 0, 0); // Fix me!

    wbfs_part_fs = wbfs_part_idx = wbfs_part_lba = 0;
    wbfs_part_idx = 1;

    return current->Open();
}

s32 WBFS_OpenPart(u32 part_fs, u32 part_idx, u32 part_lba, u32 part_size, char *partition)
{
    // close
    WBFS_Close();

    if (part_fs == PART_FS_FAT)
    {
        current = new Wbfs_Fat(wbfsDev, part_lba, part_size);
        strcpy(wbfs_fs_drive, "USB:");
#ifdef DEBUG_WBFS
        gprintf("\n\tCreated WBFS_Fat instance at lba: %d of size %d", part_lba, part_size);
#endif
    }
    else if (part_fs == PART_FS_NTFS)
    {
        current = new Wbfs_Ntfs(wbfsDev, part_lba, part_size);
        strcpy(wbfs_fs_drive, "NTFS:");
#ifdef DEBUG_WBFS
        gprintf("\n\tCreated WBFS_Ntfs instance at lba: %d of size %d", part_lba, part_size);
#endif
    }
    else
    {
        current = new Wbfs_Wbfs(wbfsDev, part_lba, part_size);
#ifdef DEBUG_WBFS
        gprintf("\n\tCreated WBFS_Wbfs instance at lba: %d of size %d", part_lba, part_size);
#endif
    }
    if (current->Open())
    {
        delete current;
        current = NULL;
        return -1;
    }

    // success
    wbfs_part_fs = part_fs;
    wbfs_part_idx = part_idx;
    wbfs_part_lba = part_lba;

    const char *fs = "WBFS";
    if (wbfs_part_fs == PART_FS_FAT) fs = "FAT";
    if (wbfs_part_fs == PART_FS_NTFS) fs = "NTFS";
    sprintf(partition, "%s%d", fs, wbfs_part_idx);
    return 0;
}

s32 WBFS_OpenNamed(char *partition)
{
    u32 i;
    u32 part_fs = PART_FS_WBFS;
    u32 part_idx = 0;
    u32 part_lba = 0;
    s32 ret = 0;
    PartList plist;

    // close
    WBFS_Close();

    // parse partition option
    if (strncasecmp(partition, "WBFS", 4) == 0)
    {
        i = atoi(partition + 4);
        if (i < 1 || i > 4) goto err;
        part_fs = PART_FS_WBFS;
        part_idx = i;
    }
    else if (strncasecmp(partition, "FAT", 3) == 0)
    {
        if (wbfsDev != WBFS_DEVICE_USB) goto err;
        i = atoi(partition + 3);
        if (i < 1 || i > 9) goto err;
        part_fs = PART_FS_FAT;
        part_idx = i;
    }
    else if (strncasecmp(partition, "NTFS", 4) == 0)
    {
        i = atoi(partition + 4);
        if (i < 1 || i > 9) goto err;
        part_fs = PART_FS_NTFS;
        part_idx = i;
    }
    else
    {
        goto err;
    }

    // Get partition entries
    ret = Partition_GetList(wbfsDev, &plist);
    if (ret || plist.num == 0) return -1;

    if (part_fs == PART_FS_WBFS)
    {
        if (part_idx > plist.wbfs_n) goto err;
        for (i = 0; i < plist.num; i++)
        {
            if (plist.pinfo[i].wbfs_i == part_idx) break;
        }
    }
    else if (part_fs == PART_FS_FAT)
    {
        if (part_idx > plist.fat_n) goto err;
        for (i = 0; i < plist.num; i++)
        {
            if (plist.pinfo[i].fat_i == part_idx) break;
        }
    }
    else if (part_fs == PART_FS_NTFS)
    {
        if (part_idx > plist.ntfs_n) goto err;
        for (i = 0; i < plist.num; i++)
        {
            if (plist.pinfo[i].ntfs_i == part_idx) break;
        }
    }
    if (i >= plist.num) goto err;
    // set partition lba sector
    part_lba = plist.pentry[i].sector;

    if (WBFS_OpenPart(part_fs, part_idx, part_lba, plist.pentry[i].size, partition))
    {
        goto err;
    }
    // success
    return 0;
    err: return -1;
}

s32 WBFS_OpenLBA(u32 lba, u32 size)
{
    Wbfs *part = new Wbfs_Wbfs(wbfsDev, lba, size);
    if (part->Open() != 0)
    {
        delete part;
        return -1;
    }

    WBFS_Close();
    current = part;
    return 0;
}

bool WBFS_Close(void)
{
    if (current != NULL)
    {
        current->Close();
        delete current;
        current = NULL;
    }

    wbfs_part_fs = 0;
    wbfs_part_idx = 0;
    wbfs_part_lba = 0;
    wbfs_fs_drive[0] = '\0';

    gameList.clear();

    return 0;
}

bool WBFS_Mounted()
{
    return (current != NULL && current->Mounted());
}

s32 WBFS_Format(u32 lba, u32 size)
{
    return current->Format();
}

s32 WBFS_GetCount(u32 *count)
{
    return current->GetCount(count);
}

s32 WBFS_GetHeaders(struct discHdr *outbuf, u32 cnt, u32 len)
{
    return current->GetHeaders(outbuf, cnt, len);
}

s32 WBFS_CheckGame(u8 *discid)
{
    return current->CheckGame(discid);
}

s32 WBFS_AddGame(void)
{
    s32 retval = current->AddGame();
    if (retval == 0) gameList.clear();

    return retval;
}

s32 WBFS_RemoveGame(u8 *discid)
{
    s32 retval = current->RemoveGame(discid);
    if (retval == 0) gameList.clear();

    return retval;
}

s32 WBFS_GameSize(u8 *discid, f32 *size)
{
    return current->GameSize(discid, size);
}

s32 WBFS_DiskSpace(f32 *used, f32 *free)
{
    return current->DiskSpace(used, free);
}

s32 WBFS_RenameGame(u8 *discid, const void *newname)
{
    s32 retval = current->RenameGame(discid, newname);
    if (retval == 0) gameList.clear();

    return retval;
}

s32 WBFS_ReIDGame(u8 *discid, const void *newID)
{
    s32 retval = current->ReIDGame(discid, newID);
    if (retval == 0) gameList.clear();

    return retval;
}

f32 WBFS_EstimeGameSize(void)
{
    return current->EstimateGameSize();
}

int WBFS_GetFragList(u8 *id)
{
    return current->GetFragList(id);
}

bool WBFS_ShowFreeSpace(void)
{
    return current->ShowFreeSpace();
}

int MountWBFS()
{
    int ret = -1;
    time_t currTime = time(0);

    while (time(0) - currTime < 30)
    {

        USBDevice_deInit();
        USBStorage2_Deinit();
        USBDevice_Init();
        ret = WBFS_Init(WBFS_DEVICE_USB);
        printf("%i...", int(time(0) - currTime));
        if (ret < 0)
            sleep(1);
        else break;
    }

    printf("\n");

    return ret;
}