/**************************************************************************** * Copyright (C) 2010 by Dimok * (C) 2012 by FIX94 * * 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. ***************************************************************************/ #ifndef PARTITION_HANDLE_H #define PARTITION_HANDLE_H #include #include "libwbfs/libwbfs.h" #include #include using namespace std; #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 /* for WiiU modified drives to ignore formatting it (thx jayjay123) */ #define MBR_SIGNATURE_MOD 0x55AB #define EBR_SIGNATURE_MOD MBR_SIGNATURE_MOD #define PARTITION_BOOTABLE 0x80 /* Bootable (active) */ #define PARTITION_NONBOOTABLE 0x00 /* Non-bootable */ #define PARTITION_TYPE_GPT 0xEE /* Indicates that a GPT header is available */ #define GUID_SYSTEM_PARTITION 0x0000000000000001LL /* System partition (disk partitioning utilities must reserve the partition as is) */ #define GUID_READ_ONLY_PARTITION 0x0800000000000000LL /* Read-only partition */ #define GUID_HIDDEN_PARTITION 0x2000000000000000LL /* Hidden partition */ #define GUID_NO_AUTOMOUNT_PARTITION 0x4000000000000000LL /* Do not automount (e.g., do not assign drive letter) */ #define BYTES_PER_SECTOR 512 /* Default in libogc */ #define MAX_BYTES_PER_SECTOR 4096 /* Max bytes per sector */ 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[446]; /* Code area; normally empty */ 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_PART_ENTRY { u8 part_type_guid[16]; /* Partition type GUID */ u8 uniq_part_guid[16]; /* Unique partition GUID */ u64 part_first_lba; /* First LBA (little-endian) */ u64 part_last_lba; /* Last LBA (inclusive, usually odd) */ u64 attribute_flags; /* GUID Attribute flags (e.g. bit 60 denotes read-only) */ char partition_name[72]; /* Partition name (36 UTF-16LE code units) */ } ATTRIBUTE_PACKED GUID_PART_ENTRY; typedef struct _GPT_HEADER { char magic[8]; /* "EFI PART" */ u32 revision; /* For version 1.0 */ u32 header_size; /* Header size in bytes */ u32 checksum; /* CRC32 of header (0 to header size), with this field zeroed during calculation */ u32 reserved; /* must be 0 */ u64 header_lba; /* Current LBA (location of this header copy) */ u64 backup_lba; /* Backup LBA (location of the other header copy) */ u64 first_part_lba; /* First usable LBA for partitions (primary partition table last LBA + 1) */ u64 last_part_lba; /* Last usable LBA (secondary partition table first LBA - 1) */ u8 disk_guid[16]; /* Disk GUID (also referred as UUID on UNIXes) */ u64 part_table_lba; /* Partition entries starting LBA (always 2 in primary copy) */ u32 part_entries; /* Number of partition entries */ u32 part_entry_size; /* Size of a partition entry (usually 128) */ u32 part_entry_checksum; /* CRC32 of partition array */ u8 zeros[420]; } ATTRIBUTE_PACKED GPT_HEADER; typedef struct _PartitionFS { const char *FSName; u64 LBA_Start; u64 SecCount; bool Bootable; u8 PartitionType; u8 PartitionNum; wbfs_t *wbfshandle; } ATTRIBUTE_PACKED PartitionFS; class PartitionHandle { public: void Init(); //! Read the MBR and all EBRs and lists up the Partitions void SetDevice(const DISC_INTERFACE *discio); //! Unmount drives void Cleanup(); //! 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, bool forceFAT = false); //! 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 ""; 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 ""; }; //! 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 count of found partitions int GetPartitionCount() const { 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 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; }; //! Get the wbfs mount handle wbfs_t *GetWbfsHandle(int pos); //! Set the wbfs mount handle bool SetWbfsHandle(int pos, wbfs_t *wbfshandle); protected: bool valid(int pos) { return (pos >= 0 && pos < (int)PartitionList.size()); }; void AddPartition(const char *name, u64 lba_start, u64 sec_count, bool bootable, u8 part_type, u8 part_num); bool IsExisting(u64 lba); s8 FindPartitions(); void CheckEBR(u8 PartNum, sec_t ebr_lba); s8 CheckGPT(); const DISC_INTERFACE *interface; vector PartitionList; vector MountNameList; }; #endif