The-Homebrew-Cloud/WiiFlow_Lite
2
0
Fork 0
mirror of https://github.com/Fledge68/WiiFlow_Lite.git synced 2024-11-01 00:55:06 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #350 from kcpants/master

Browse Source 
- Ignore USB storage devices without an MBR or WBFS signature
This commit is contained in:
Fledge68 2023-05-26 16:49:13 -05:00 committed by GitHub
commit 57e7e64b1f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 20 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
source/devicemounter/PartitionHandle.cpp
View File

@ -151,7 +151,10 @@ bool PartitionHandle::Mount(int pos, const char *name, bool forceFAT)
} }
} }
if(!valid(pos)) if(!valid(pos))
{
gprintf("No valid partition position found! Failed to mount any partition.\n");
return false; return false;
}
SetWbfsHandle(pos, NULL); SetWbfsHandle(pos, NULL);
if(strncmp(GetFSName(pos), "FAT", 3) == 0 || strcmp(GetFSName(pos), "GUID-Entry") == 0) if(strncmp(GetFSName(pos), "FAT", 3) == 0 || strcmp(GetFSName(pos), "GUID-Entry") == 0)
@ -195,6 +198,7 @@ bool PartitionHandle::Mount(int pos, const char *name, bool forceFAT)
} }
} }
/* FAIL */ /* FAIL */
gprintf("Exhausted all supported filesystem types! Failed to mount any partition.\n");
MountNameList[pos].clear(); MountNameList[pos].clear();
return false; return false;
} }

56
source/devicemounter/usbstorage_libogc.c
View File

@ -96,32 +96,9 @@ distribution.
#define DEVLIST_MAXSIZE 8 #define DEVLIST_MAXSIZE 8
static inline const char *PartFromType(u8 type) #define MBR_SIGNATURE 0x55AA
{ #define MBR_SIGNATURE_MOD 0x55AB // UStealth modifies the mbr signature to 0x55AB so the WiiU will not attempt to format it; treat 0x55AB as a valid signature
switch (type) #define WBFS_HEADER_MAGIC 0x57424653 // 0x57424653 is the characters "WBFS" in hex
{
case 0x00: return "Unused";
case 0x01: return "FAT12";
case 0x04: return "FAT16";
case 0x05: return "Extended";
case 0x06: return "FAT16";
case 0x07: return "NTFS";
case 0x0b: return "FAT32";
case 0x0c: return "FAT32";
case 0x0e: return "FAT16";
case 0x0f: return "Extended";
case 0x82: return "LxSWP";
case 0x83: return "LINUX";
case 0x8e: return "LxLVM";
case 0xa8: return "OSX";
case 0xab: return "OSXBT";
case 0xaf: return "OSXHF";
case 0xbf: return "WBFS";
case 0xe8: return "LUKS";
default: return "Unknown";
}
}
static heap_cntrl __heap; static heap_cntrl __heap;
static bool __inited = false; static bool __inited = false;
@ -848,6 +825,8 @@ static bool __usbstorage_ogc_IsInserted(void)
u8 device_count; u8 device_count;
u8 i, j; u8 i, j;
u16 vid, pid; u16 vid, pid;
u16 mbrSignature;
u32 wbfsHeaderMagic;
s32 maxLun; s32 maxLun;
s32 retval; s32 retval;
u32 sectorsize, numSectors; u32 sectorsize, numSectors;
@ -928,22 +907,19 @@ static bool __usbstorage_ogc_IsInserted(void)
} }
u8* mbr = (u8*)__lwp_heap_allocate(&__heap, 512); u8* mbr = (u8*)__lwp_heap_allocate(&__heap, 512);
USBStorage_OGC_Read(&__usbfd, j, 0, 1, mbr); if (mbr)
bool readablePartition = false;
for (u8 i = 0; i < 4; i++)
{ {
u8 rawPartitionType = mbr[450 + i * 16]; USBStorage_OGC_Read(&__usbfd, j, 0, 1, mbr);
const char* partitionType = PartFromType(rawPartitionType); mbrSignature = ((u16*)mbr)[255];
if (strcmp(partitionType, "Unknown") != 0) { wbfsHeaderMagic = *((u32*)mbr);
readablePartition = true; __lwp_heap_free(&__heap, mbr);
}
}
__lwp_heap_free(&__heap, mbr);
if (!readablePartition) { if (mbrSignature != MBR_SIGNATURE && mbrSignature != MBR_SIGNATURE_MOD && wbfsHeaderMagic != WBFS_HEADER_MAGIC)
gprintf("USB storage device with vid %lu pid %lu has no readable partitions. Skipping...\n", vid, pid); {
__usbstorage_ogc_reset(&__usbfd); gprintf("USB storage device (vid %lu pid %lu) cannot be identified as MBR, GPT, or WBFS. Skipping...\n", vid, pid);
continue; __usbstorage_ogc_reset(&__usbfd);
continue;
}
} }
__mounted = true; __mounted = true;