mirror of
https://github.com/Fledge68/WiiFlow_Lite.git
synced 2024-11-27 13:44:15 +01:00
219 lines
10 KiB
C++
219 lines
10 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
|
|
***************************************************************************/
|
|
#ifndef PARTITION_HANDLE_H
|
|
#define PARTITION_HANDLE_H
|
|
|
|
#include <gccore.h>
|
|
#include "safe_vector.hpp"
|
|
#include "libwbfs/libwbfs.h"
|
|
#include <string>
|
|
|
|
#define MAX_PARTITIONS 32 /* Maximum number of partitions that can be found */
|
|
#define MAX_MOUNTS 10 /* Maximum number of mounts available at one time */
|
|
#define MAX_SYMLINK_DEPTH 10 /* Maximum search depth when resolving symbolic links */
|
|
|
|
#define MBR_SIGNATURE 0x55AA
|
|
#define EBR_SIGNATURE MBR_SIGNATURE
|
|
#define VBR_SIGNATURE MBR_SIGNATURE
|
|
#define GPT_SIGNATURE "EFI PART"
|
|
|
|
#define PARTITION_BOOTABLE 0x80 /* Bootable (active) */
|
|
#define PARTITION_NONBOOTABLE 0x00 /* Non-bootable */
|
|
|
|
#define BYTES_PER_SECTOR 512 /* Default in libogc */
|
|
|
|
enum SIG_OFFSETS {
|
|
BPB_NTFS_ADDR = 0x3,
|
|
BPB_FAT16_ADDR = 0x36,
|
|
BPB_EXT2_ADDR = 0x38,
|
|
BPB_FAT32_ADDR = 0x52,
|
|
};
|
|
|
|
static const char FAT_SIGNATURE[3] = {'F', 'A', 'T'};
|
|
static const char NTFS_SIGNATURE[4] = {'N', 'T', 'F', 'S'};
|
|
static const char WBFS_SIGNATURE[4] = {'W', 'B', 'F', 'S'};
|
|
static const char EXT_SIGNATURE[2] = {0x53, 0xEF};
|
|
|
|
typedef struct _PARTITION_RECORD {
|
|
u8 status; /* Partition status; see above */
|
|
u8 chs_start[3]; /* Cylinder-head-sector address to first block of partition */
|
|
u8 type; /* Partition type; see above */
|
|
u8 chs_end[3]; /* Cylinder-head-sector address to last block of partition */
|
|
u32 lba_start; /* Local block address to first sector of partition */
|
|
u32 block_count; /* Number of blocks in partition */
|
|
} __attribute__((__packed__)) PARTITION_RECORD;
|
|
|
|
typedef struct _MASTER_BOOT_RECORD {
|
|
u8 code_area[440]; /* Code area; normally empty */
|
|
u8 disk_guid[4]; /* Disk signature (optional) */
|
|
u8 reserved[2]; /* Usually zeroed */
|
|
PARTITION_RECORD partitions[4]; /* 4 primary partitions */
|
|
u16 signature; /* MBR signature; 0xAA55 */
|
|
} __attribute__((__packed__)) MASTER_BOOT_RECORD;
|
|
|
|
typedef struct _EXTENDED_BOOT_RECORD {
|
|
u8 code_area[446]; /* Code area; normally empty */
|
|
PARTITION_RECORD partition; /* Primary partition */
|
|
PARTITION_RECORD next_ebr; /* Next extended boot record in the chain */
|
|
u8 reserved[32]; /* Normally empty */
|
|
u16 signature; /* EBR signature; 0xAA55 */
|
|
} __attribute__((__packed__)) EXTENDED_BOOT_RECORD;
|
|
|
|
typedef struct _GUID_PARTITION_ENTRY
|
|
{
|
|
u8 Type_GUID[16]; /* Partition type GUID */
|
|
u8 Unique_GUID[16]; /* Unique partition GUID */
|
|
u64 First_LBA; /* First LBA (little-endian) */
|
|
u64 Last_LBA; /* Last LBA (inclusive, usually odd) */
|
|
u64 Attributes; /* GUID Attribute flags (e.g. bit 60 denotes read-only) */
|
|
char Name[72]; /* Partition name (36 UTF-16LE code units) */
|
|
} __attribute__((__packed__)) GUID_PARTITION_ENTRY;
|
|
|
|
typedef struct _GPT_PARTITION_TABLE {
|
|
char magic[8]; /* "EFI PART" */
|
|
u32 Revision;
|
|
u32 Header_Size; /* Size of this header */
|
|
u32 CheckSum;
|
|
u32 Reserved; /* Must be 0 */
|
|
u64 Header_LBA; /* Location of this header, always 1 in primary copy */
|
|
u64 Backup_Header_LBA; /* Location of backup header, always max lba - 1 */
|
|
u64 First_Usable_LBA; /* Primary GPT partition table's last LBA + 1 */
|
|
u64 Last_Usable_LBA; /* Secondary GPT partition table's first LBA - 1 */
|
|
u8 GUID[16]; /* Disk GUID (also referred as UUID on UNIXes) */
|
|
u64 Table_LBA; /* Always 2 in primary copy, or Header_LBA + 1. Secondary copy is Backup_Header_LBA - 1 */
|
|
u32 Num_Entries; /* Number of entries in the partition info array */
|
|
u32 Entry_Size; /* Size of each array entry, usually 128 */
|
|
u32 Entries_CheckSum; /* CRC32 of partition array */
|
|
u8 Zeroes[420]; /* Pad to a total 512 byte LBA or sizeof 1 LBA */
|
|
GUID_PARTITION_ENTRY partitions[128]; /* Max 128 Partition Entries */
|
|
} __attribute__((__packed__)) GPT_PARTITION_TABLE;
|
|
|
|
typedef struct _PartitionFS {
|
|
const char * FSName;
|
|
u32 LBA_Start;
|
|
u32 SecCount;
|
|
bool Bootable;
|
|
u8 PartitionType;
|
|
u8 PartitionNum;
|
|
u32 EBR_Sector;
|
|
wbfs_t *wbfshandle;
|
|
} PartitionFS;
|
|
|
|
typedef struct _BIOS_PARAMETER_BLOCK {
|
|
u16 Bytes_Per_Sector; /* Size of a sector in bytes. */
|
|
u8 Sectors_Per_Cluster; /* Size of a cluster in sectors. */
|
|
u16 ReservedSsectors; /* zero on ntfs */
|
|
u8 Fats; /* zero on ntfs */
|
|
u16 Root_Entries; /* zero on ntfs */
|
|
u16 FatSectors; /* Number of sectors in volume. (FAT) Zero on ntfs */
|
|
u8 Media_Type; /* 0xf8 = hard disk */
|
|
u16 Sectors_Per_Fat; /* zero on ntfs*/
|
|
u16 Sectors_Per_Track; /* Required to boot Windows. (0x0d) */
|
|
u16 Heads; /* Required to boot Windows. (0x0f) */
|
|
u32 Hidden_Sectors; /* Offset to the start of the partition (0x11) in sectors. */
|
|
u32 Large_Sectors; /* Number of sectors in volume if Sectors is 0 (FAT) Zero on ntfs (0x15) */
|
|
} __attribute__((__packed__)) BIOS_PARAMETER_BLOCK; /* 25 (0x19) bytes */
|
|
|
|
typedef struct _VOLUME_BOOT_RECORD {
|
|
u8 Jump[3]; /* Irrelevant (jump to boot up code).*/
|
|
char Name[8]; /* Magic "NTFS ". */
|
|
BIOS_PARAMETER_BLOCK bpb; /* See BIOS_PARAMETER_BLOCK. (0x0b) */
|
|
u8 Drive_Type; /* 0x00 floppy, 0x80 hard disk */
|
|
u8 Current_Head; /* zero on ntfs */
|
|
u8 Extended_Boot_Signature; /* 0x80 on ntfs (Doesnt show this in M$ docs)*/
|
|
u8 Reserved0; /* zero on ntfs */
|
|
s64 Number_of_Sectors; /* Number of sectors in volume. (NTFS)(0x28)*/
|
|
s64 MFT; /* Cluster location of mft data. */
|
|
s64 MFT_Mirror; /* Cluster location of copy of mft. */
|
|
s8 Clusters_Per_MFT; /* Mft record size in clusters. */
|
|
u8 Reserved1[3]; /* zero */
|
|
s8 Clusters_Per_Index; /* Index block size in clusters. */
|
|
u8 Reserved2[3]; /* zero */
|
|
u64 Volume_Serial_Number; /* Irrelevant (serial number). */
|
|
u8 Checksum[4]; /* Boot sector checksum. */
|
|
u8 Bootstrap[426]; /* Irrelevant (boot up code). (0x54) */
|
|
u16 Signature; /* End of bootsector magic. LE 0xaa55 */
|
|
} __attribute__((__packed__)) VOLUME_BOOT_RECORD; /* 512 (0x200) bytes */
|
|
|
|
class PartitionHandle
|
|
{
|
|
public:
|
|
//! Constructor reads the MBR and all EBRs and lists up the Partitions
|
|
PartitionHandle(const DISC_INTERFACE *discio);
|
|
//! Destructor unmounts drives
|
|
~PartitionHandle();
|
|
//! Is Drive inserted
|
|
bool IsInserted() { if(!interface) return false; else return interface->isInserted(); };
|
|
//! Is the partition Mounted
|
|
bool IsMounted(int pos);
|
|
//! Mount a specific Partition
|
|
bool Mount(int pos, const char * name);
|
|
//! UnMount a specific Partition
|
|
void UnMount(int pos);
|
|
//! UnMount all Partition
|
|
void UnMountAll() { for(u32 i = 0; i < PartitionList.size(); ++i) UnMount(i); };
|
|
//! Get the Mountname
|
|
const char * MountName(int pos) { if(pos < 0 || pos >= (int) MountNameList.size() || !MountNameList[pos].size()) return NULL; else return MountNameList[pos].c_str(); };
|
|
//! Get the Name of the FileSystem e.g. "FAT32"
|
|
const char * GetFSName(int pos) { if(valid(pos)) return PartitionList[pos].FSName; else return NULL; };
|
|
//! Get the LBA where the partition is located
|
|
u32 GetLBAStart(int pos) { if(valid(pos)) return PartitionList[pos].LBA_Start; else return 0; };
|
|
//! Get the partition size in sectors of this partition
|
|
u32 GetSecCount(int pos) { if(valid(pos)) return PartitionList[pos].SecCount; else return 0; };
|
|
//! Check if the partition is Active or NonBootable
|
|
bool IsActive(int pos) { if(valid(pos)) return PartitionList[pos].Bootable; else return false; };
|
|
//! Get the partition type
|
|
int GetPartitionType(int pos) { if(valid(pos)) return PartitionList[pos].PartitionType; else return -1; };
|
|
//! Get the entrie number in MBR of this partition
|
|
int GetPartitionNum(int pos) { if(valid(pos)) return PartitionList[pos].PartitionNum; else return -1; };
|
|
//! Get the EBR sector where this partition is described
|
|
int GetEBRSector(int pos) { if(valid(pos)) return PartitionList[pos].EBR_Sector; else return 0; };
|
|
//! Get the count of found partitions
|
|
int GetPartitionCount() { return PartitionList.size(); };
|
|
//! Get the partition size in bytes
|
|
u64 GetSize(int pos) { if(valid(pos)) return (u64) PartitionList[pos].SecCount*BYTES_PER_SECTOR; else return 0; };
|
|
//! Get the wbfs mount handle
|
|
wbfs_t * GetWbfsHandle(int pos) { if(valid(pos)) return PartitionList[pos].wbfshandle; else return 0; };
|
|
//! Set the wbfs mount handle
|
|
bool SetWbfsHandle(int pos, wbfs_t * wbfshandle) { if(valid(pos)) {PartitionList[pos].wbfshandle = wbfshandle; return true;} else return false; };
|
|
//! Get the whole partition record struct
|
|
PartitionFS * GetPartitionRecord(int pos) { if(valid(pos)) return &PartitionList[pos]; else return NULL; };
|
|
//! Get the disc interface of this handle
|
|
const DISC_INTERFACE * GetDiscInterface() { return interface; };
|
|
protected:
|
|
bool valid(int pos) { return (pos >= 0 && pos < (int) PartitionList.size()); }
|
|
bool CheckRAW(VOLUME_BOOT_RECORD * vbr);
|
|
int FindPartitions();
|
|
void CheckEBR(u8 PartNum, sec_t ebr_lba);
|
|
bool CheckGPT(void);
|
|
|
|
const DISC_INTERFACE *interface;
|
|
safe_vector<PartitionFS> PartitionList;
|
|
safe_vector<std::string> MountNameList;
|
|
};
|
|
|
|
#endif
|