mirror of
https://github.com/Fledge68/WiiFlow_Lite.git
synced 2024-11-27 21:54:15 +01:00
332 lines
7.6 KiB
C++
332 lines
7.6 KiB
C++
/****************************************************************************
|
|
* Copyright (C) 2010
|
|
* by Dimok
|
|
*
|
|
* This software is provided 'as-is', without any express or implied
|
|
* warranty. In no event will the authors be held liable for any
|
|
* damages arising from the use of this software.
|
|
*
|
|
* Permission is granted to anyone to use this software for any
|
|
* purpose, including commercial applications, and to alter it and
|
|
* redistribute it freely, subject to the following restrictions:
|
|
*
|
|
* 1. The origin of this software must not be misrepresented; you
|
|
* must not claim that you wrote the original software. If you use
|
|
* this software in a product, an acknowledgment in the product
|
|
* documentation would be appreciated but is not required.
|
|
*
|
|
* 2. Altered source versions must be plainly marked as such, and
|
|
* must not be misrepresented as being the original software.
|
|
*
|
|
* 3. This notice may not be removed or altered from any source
|
|
* distribution.
|
|
*
|
|
* for WiiXplorer 2010
|
|
***************************************************************************/
|
|
#include <malloc.h>
|
|
#include <unistd.h>
|
|
#include <string.h>
|
|
#include <ogc/mutex.h>
|
|
#include <ogc/system.h>
|
|
#include <sdcard/gcsd.h>
|
|
|
|
#include "DeviceHandler.hpp"
|
|
#include "wbfs.h"
|
|
#include "usbstorage.h"
|
|
|
|
extern const DISC_INTERFACE __io_sdhc;
|
|
|
|
DeviceHandler * DeviceHandler::instance = NULL;
|
|
unsigned int DeviceHandler::watchdog_timeout = 10;
|
|
|
|
DeviceHandler::~DeviceHandler()
|
|
{
|
|
UnMountAll();
|
|
}
|
|
|
|
DeviceHandler * DeviceHandler::Instance()
|
|
{
|
|
if (instance == NULL)
|
|
instance = new DeviceHandler();
|
|
|
|
return instance;
|
|
}
|
|
|
|
void DeviceHandler::DestroyInstance()
|
|
{
|
|
if(instance) delete instance;
|
|
instance = NULL;
|
|
}
|
|
|
|
bool DeviceHandler::MountAll()
|
|
{
|
|
bool result = false;
|
|
|
|
for(u32 i = SD; i <= USB8; i++)
|
|
if(Mount(i)) result = true;
|
|
|
|
return result;
|
|
}
|
|
|
|
void DeviceHandler::UnMountAll()
|
|
{
|
|
for(u32 i = SD; i <= GCSDB; i++)
|
|
UnMount(i);
|
|
|
|
if(sd) delete sd;
|
|
if(usb) delete usb;
|
|
if(gca) delete gca;
|
|
if(gcb) delete gca;
|
|
|
|
sd = NULL;
|
|
usb = NULL;
|
|
gca = NULL;
|
|
gcb = NULL;
|
|
}
|
|
|
|
bool DeviceHandler::Mount(int dev)
|
|
{
|
|
if(dev == SD)
|
|
return MountSD();
|
|
else if(dev >= USB1 && dev <= USB8)
|
|
return MountUSB(dev-USB1);
|
|
else if(dev == GCSDA)
|
|
return MountGCA();
|
|
else if(dev == GCSDB)
|
|
return MountGCB();
|
|
|
|
return false;
|
|
}
|
|
|
|
bool DeviceHandler::IsInserted(int dev)
|
|
{
|
|
if(dev == SD)
|
|
return SD_Inserted() && sd->IsMounted(0);
|
|
else if(dev >= USB1 && dev <= USB8)
|
|
return USB_Inserted() && usb->IsMounted(dev-USB1);
|
|
else if(dev == GCSDA)
|
|
return GCA_Inserted() && gca->IsMounted(0);
|
|
else if(dev == GCSDB)
|
|
return GCB_Inserted() && gcb->IsMounted(0);
|
|
|
|
return false;
|
|
}
|
|
|
|
void DeviceHandler::UnMount(int dev)
|
|
{
|
|
if(dev == SD)
|
|
UnMountSD();
|
|
else if(dev >= USB1 && dev <= USB8)
|
|
UnMountUSB(dev-USB1);
|
|
else if(dev == GCSDA)
|
|
UnMountGCA();
|
|
else if(dev == GCSDB)
|
|
UnMountGCB();
|
|
}
|
|
|
|
bool DeviceHandler::MountSD()
|
|
{
|
|
if(!sd)
|
|
{
|
|
sd = new PartitionHandle(&__io_sdhc);
|
|
if(sd->GetPartitionCount() < 1)
|
|
{
|
|
delete sd;
|
|
sd = NULL;
|
|
return false;
|
|
}
|
|
}
|
|
|
|
//! Mount only one SD Partition
|
|
return sd->Mount(0, DeviceName[SD]);
|
|
}
|
|
|
|
bool DeviceHandler::MountUSB(int pos)
|
|
{
|
|
if(!usb) usb = new PartitionHandle(&__io_usbstorage);
|
|
|
|
if(usb->GetPartitionCount() < 1)
|
|
{
|
|
delete usb;
|
|
usb = NULL;
|
|
return false;
|
|
}
|
|
|
|
// Set the watchdog
|
|
InternalSetWatchdog(watchdog_timeout);
|
|
|
|
if(pos >= usb->GetPartitionCount())
|
|
return false;
|
|
|
|
return usb->Mount(pos, DeviceName[USB1+pos]);
|
|
}
|
|
|
|
bool DeviceHandler::MountAllUSB()
|
|
{
|
|
if(!usb) usb = new PartitionHandle(&__io_usbstorage);
|
|
|
|
bool result = false;
|
|
|
|
for(int i = 0; i < usb->GetPartitionCount(); i++)
|
|
if(MountUSB(i))
|
|
result = true;
|
|
|
|
return result;
|
|
}
|
|
|
|
bool DeviceHandler::MountGCA()
|
|
{
|
|
if(!gca) gca = new PartitionHandle(&__io_gcsda);
|
|
|
|
if(gca->GetPartitionCount() < 1)
|
|
{
|
|
delete gca;
|
|
gca = NULL;
|
|
return false;
|
|
}
|
|
|
|
//! Mount only one Partition
|
|
return gca->Mount(0, DeviceName[GCSDA]);
|
|
}
|
|
|
|
bool DeviceHandler::MountGCB()
|
|
{
|
|
if(!gcb) gcb = new PartitionHandle(&__io_gcsdb);
|
|
|
|
if(gcb->GetPartitionCount() < 1)
|
|
{
|
|
delete gcb;
|
|
gcb = NULL;
|
|
return false;
|
|
}
|
|
|
|
//! Mount only one Partition
|
|
return gcb->Mount(0, DeviceName[GCSDB]);;
|
|
}
|
|
|
|
void DeviceHandler::UnMountUSB(int pos)
|
|
{
|
|
if(!usb) return;
|
|
|
|
if(pos >= usb->GetPartitionCount())
|
|
return;
|
|
|
|
usb->UnMount(pos);
|
|
}
|
|
|
|
void DeviceHandler::UnMountAllUSB()
|
|
{
|
|
if(!usb) return;
|
|
|
|
for(int i = 0; i < usb->GetPartitionCount(); i++)
|
|
usb->UnMount(i);
|
|
|
|
delete usb;
|
|
usb = NULL;
|
|
}
|
|
|
|
bool DeviceHandler::InternalSetWatchdog(unsigned int timeout)
|
|
{
|
|
if (Instance()->USB_Inserted())
|
|
return USBStorage_SetWatchdog(timeout) == 0;
|
|
|
|
return false;
|
|
}
|
|
|
|
bool DeviceHandler::SetWatchdog(unsigned int timeout)
|
|
{
|
|
watchdog_timeout = timeout;
|
|
return InternalSetWatchdog(timeout);
|
|
}
|
|
|
|
int DeviceHandler::PathToDriveType(const char * path)
|
|
{
|
|
if(!path) return -1;
|
|
|
|
for(int i = SD; i <= GCSDB; i++)
|
|
if(strncmp(path, DeviceName[i], strlen(DeviceName[i])) == 0)
|
|
return i;
|
|
|
|
return -1;
|
|
}
|
|
|
|
const char * DeviceHandler::GetFSName(int dev)
|
|
{
|
|
if(dev == SD && DeviceHandler::instance->sd)
|
|
return DeviceHandler::instance->sd->GetFSName(0);
|
|
else if(dev >= USB1 && dev <= USB8 && DeviceHandler::instance->usb)
|
|
return DeviceHandler::instance->usb->GetFSName(dev-USB1);
|
|
else if(dev == GCSDA && DeviceHandler::instance->gca)
|
|
return DeviceHandler::instance->gca->GetFSName(0);
|
|
else if(dev == GCSDB && DeviceHandler::instance->gcb)
|
|
return DeviceHandler::instance->gcb->GetFSName(0);
|
|
|
|
return NULL;
|
|
}
|
|
|
|
int DeviceHandler::GetFSType(int dev)
|
|
{
|
|
const char *name = GetFSName(dev);
|
|
if(!name) return -1;
|
|
|
|
if (strncasecmp(name, "WBFS", 4) == 0)
|
|
return PART_FS_WBFS;
|
|
else if (strncasecmp(name, "FAT", 3) == 0)
|
|
return PART_FS_FAT;
|
|
else if (strncasecmp(name, "NTFS", 4) == 0)
|
|
return PART_FS_NTFS;
|
|
else if (strncasecmp(name, "LINUX", 5) == 0)
|
|
return PART_FS_EXT;
|
|
|
|
return -1;
|
|
}
|
|
|
|
s16 DeviceHandler::GetMountedCount(int dev)
|
|
{
|
|
if(dev == SD && DeviceHandler::instance->sd && IsInserted(SD))
|
|
return 1;
|
|
else if(dev >= USB1 && dev <= USB8 && DeviceHandler::instance->usb)
|
|
for(int i = 0; i < usb->GetPartitionCount(); i++)
|
|
{
|
|
if(!IsInserted(i)) return i;
|
|
}
|
|
else if(dev == GCSDA && DeviceHandler::instance->gca && IsInserted(GCSDA))
|
|
return 1;
|
|
else if(dev == GCSDB && DeviceHandler::instance->gcb && IsInserted(GCSDB))
|
|
return 1;
|
|
|
|
return -1;
|
|
}
|
|
|
|
wbfs_t * DeviceHandler::GetWbfsHandle(int dev)
|
|
{
|
|
if(dev == SD && DeviceHandler::instance->sd)
|
|
return DeviceHandler::instance->sd->GetWbfsHandle(0);
|
|
else if(dev >= USB1 && dev <= USB8 && DeviceHandler::instance->usb)
|
|
return DeviceHandler::instance->usb->GetWbfsHandle(dev-USB1);
|
|
else if(dev == GCSDA && DeviceHandler::instance->gca)
|
|
return DeviceHandler::instance->gca->GetWbfsHandle(0);
|
|
else if(dev == GCSDB && DeviceHandler::instance->gcb)
|
|
return DeviceHandler::instance->gcb->GetWbfsHandle(0);
|
|
|
|
return NULL;
|
|
}
|
|
|
|
s32 DeviceHandler::Open_WBFS(int dev)
|
|
{
|
|
u32 part_lba;
|
|
u32 part_fs = GetFSType(dev);
|
|
char *partition = (char *)DeviceName[dev];
|
|
|
|
if(dev == SD && IsInserted(dev))
|
|
part_lba = Instance()->sd->GetLBAStart(dev);
|
|
else if(dev >= USB1 && dev <= USB8 && IsInserted(dev))
|
|
part_lba = Instance()->usb->GetLBAStart(dev - USB1);
|
|
else if(dev == GCSDA && IsInserted(dev))
|
|
part_lba = Instance()->gca->GetLBAStart(dev);
|
|
else if(dev == GCSDB && IsInserted(dev))
|
|
part_lba = Instance()->gcb->GetLBAStart(dev);
|
|
else return -1;
|
|
|
|
return WBFS_Init(GetWbfsHandle(dev), part_fs, part_lba, partition, dev);
|
|
} |