usbloadergx/source/fatmounter.c

#include <string.h>
#include <ogc/lwp_watchdog.h>
#include <ogc/mutex.h>
#include <ogc/system.h>
#include <ogc/usbstorage.h>
#include <sdcard/wiisd_io.h>
#include <locale.h>

#include "usbloader/usbstorage2.h"
#include "usbloader/sdhc.h"
#include "usbloader/wbfs.h"
#include "libntfs/ntfs.h"
#include "libfat/fat.h"
#include "gecko.h"

//these are the only stable and speed is good
#define CACHE 32
#define SECTORS 64
#define SECTORS_SD 32

#define MOUNT_NONE 0
#define MOUNT_SD   1
#define MOUNT_SDHC 2

#define DEBUG_FAT

/* Disc interfaces */
extern const DISC_INTERFACE __io_sdhc;

void _FAT_mem_init();
extern sec_t _FAT_startSector;

extern s32 wbfsDev;

int fat_sd_mount = MOUNT_NONE;
sec_t fat_sd_sec = 0; // u32

int fat_usb_mount = 0;
sec_t fat_usb_sec = 0;

int fat_wbfs_mount = 0;
sec_t fat_wbfs_sec = 0;

int fs_ntfs_mount = 0;
sec_t fs_ntfs_sec = 0;

int USBDevice_Init()
{
#ifdef DEBUG_FAT
    gprintf("USBDevice_Init()");
#endif
    //closing all open Files write back the cache and then shutdown em!
    fatUnmount("USB:/");
    //right now mounts first FAT-partition

    //try first mount with cIOS
    //    if (!fatMount("USB", &__io_wiiums, 0, CACHE, SECTORS)) {
    //      //try now mount with libogc
    if (!fatMount("USB", &__io_usbstorage2, 0, CACHE, SECTORS))
    {
#ifdef DEBUG_FAT
        gprintf(":-1\n");
#endif
        return -1;
    }
    //  }

    fat_usb_mount = 1;
    fat_usb_sec = _FAT_startSector;
#ifdef DEBUG_FAT
    gprintf(":0\n");
#endif
    return 0;
}

void USBDevice_deInit()
{
#ifdef DEBUG_FAT
    gprintf("USBDevice_deInit(): ");
#endif
    //closing all open Files write back the cache and then shutdown em!
    fatUnmount("USB:/");

    fat_usb_mount = 0;
    fat_usb_sec = 0;
#ifdef DEBUG_FAT
    gprintf("ok\n");
#endif
}

int WBFSDevice_Init(u32 sector)
{
    //closing all open Files write back the cache and then shutdown em!
    fatUnmount("WBFS:/");
    //right now mounts first FAT-partition

    //try first mount with cIOS
    //    if (!fatMount("WBFS", &__io_wiiums, 0, CACHE, SECTORS)) {
    //try now mount with libogc
    if (!fatMount("WBFS", &__io_usbstorage2, 0, CACHE, SECTORS))
    {
        return -1;
    }
    //  }

    fat_wbfs_mount = 1;
    fat_wbfs_sec = _FAT_startSector;
    if (sector && fat_wbfs_sec != sector)
    {
        // This is an error situation...actually, but is ignored in Config loader also
        // Should ask Oggzee about it...
    }
    return 0;
}

void WBFSDevice_deInit()
{
    //closing all open Files write back the cache and then shutdown em!
    fatUnmount("WBFS:/");

    fat_wbfs_mount = 0;
    fat_wbfs_sec = 0;
}

int isInserted(const char *path)
{
    if (!strncmp(path, "USB:", 4)) return 1;

    return __io_sdhc.isInserted() || __io_wiisd.isInserted();
}

static u8 sdIsInited = 0;
int SDCard_Init()
{
#ifdef DEBUG_FAT
    gprintf("SDCard_Init()");
#endif
    //closing all open Files write back the cache and then shutdown em!
    fatUnmount("SD:/");
    //right now mounts first FAT-partition
    if (fatMount("SD", &__io_wiisd, 0, CACHE, SECTORS))
    {
        fat_sd_mount = MOUNT_SD;
        fat_sd_sec = _FAT_startSector;
        sdIsInited = 1;
#ifdef DEBUG_FAT
        gprintf(":1\n");
#endif
        return 1;
    }
    else if (fatMount("SD", &__io_sdhc, 0, CACHE, SDHC_SECTOR_SIZE))
    {
        fat_sd_mount = MOUNT_SDHC;
        fat_sd_sec = _FAT_startSector;
        sdIsInited = 1;
#ifdef DEBUG_FAT
        gprintf(":1\n");
#endif
        return 1;
    }
#ifdef DEBUG_FAT
    gprintf(":-1\n");
#endif
    return -1;
}

void SDCard_deInit()
{
#ifdef DEBUG_FAT
    gprintf("SDCard_deInit( %d ): ", sdIsInited);
#endif
    //closing all open Files write back the cache and then shutdown em!
    if (sdIsInited)
    {
        fatUnmount("SD:/");

        fat_sd_mount = MOUNT_NONE;
        fat_sd_sec = 0;
        sdIsInited = 0;
    }
#ifdef DEBUG_FAT
    gprintf("ok\n");
#endif
}

void ntfsInit();

s32 MountNTFS(u32 sector)
{
    s32 ret;

    if (fs_ntfs_mount) return 0;
    //printf("mounting NTFS\n");
    //Wpad_WaitButtons();
    _FAT_mem_init();

    ntfsInit(); // Call ntfs init here, to prevent locale resets

    // ntfsInit resets locale settings
    // which breaks unicode in console
    // so we change it back to C-UTF-8
    setlocale(LC_CTYPE, "C-UTF-8");
    setlocale(LC_MESSAGES, "C-UTF-8");

    if (wbfsDev == WBFS_DEVICE_USB)
    {
        /* Initialize WBFS interface */
        //      if (!__io_wiiums.startup()) {
        ret = __io_usbstorage2.startup();
        if (!ret)
        {
            return -1;
        }
        //      }
        /* Mount device */
        //      if (!ntfsMount("NTFS", &__io_wiiums, sector, CACHE, SECTORS, NTFS_SHOW_HIDDEN_FILES | NTFS_RECOVER)) {
        ret = ntfsMount("NTFS", &__io_usbstorage2, sector, CACHE, SECTORS, NTFS_SHOW_HIDDEN_FILES | NTFS_RECOVER);
        if (!ret)
        {
            return -2;
        }
        //      }
    }
    else if (wbfsDev == WBFS_DEVICE_SDHC)
    {
        if (sdhc_mode_sd == 0)
        {
            ret = ntfsMount("NTFS", &__io_sdhc, 0, CACHE, SECTORS, NTFS_SHOW_HIDDEN_FILES | NTFS_RECOVER);
        }
        else
        {
            ret = ntfsMount("NTFS", &__io_sdhc, 0, CACHE, SECTORS_SD, NTFS_SHOW_HIDDEN_FILES | NTFS_RECOVER);
        }
        if (!ret)
        {
            return -5;
        }
    }

    fs_ntfs_mount = 1;
    fs_ntfs_sec = sector; //_FAT_startSector;

    return 0;
}

s32 UnmountNTFS(void)
{
    /* Unmount device */
    ntfsUnmount("NTFS:/", true);

    fs_ntfs_mount = 0;
    fs_ntfs_sec = 0;

    return 0;
}

void _FAT_mem_init()
{
}

void* _FAT_mem_allocate(size_t size)
{
    return malloc(size);
}

void* _FAT_mem_align(size_t size)
{
    return memalign(32, size);
}

void _FAT_mem_free(void *mem)
{
    free(mem);
}