1e7b4ff9317fc4e551cf974684eaa88697de5a28srs/* gpt.h -- GPT and data structure definitions, types, and 2e7b4ff9317fc4e551cf974684eaa88697de5a28srs functions */ 3e7b4ff9317fc4e551cf974684eaa88697de5a28srs 464cbd171067eb34054741bfcd73f0b91d727a371srs/* This program is copyright (c) 2009-2011 by Roderick W. Smith. It is distributed 5221e08768de7fe42ba533ca22baf671420569c07srs under the terms of the GNU GPL version 2, as detailed in the COPYING file. */ 6221e08768de7fe42ba533ca22baf671420569c07srs 7e7b4ff9317fc4e551cf974684eaa88697de5a28srs#include <stdint.h> 8e7b4ff9317fc4e551cf974684eaa88697de5a28srs#include <sys/types.h> 9546a9c7c369df465021feecb20f6a8f81b6df6bcsrs#include "gptpart.h" 10e7b4ff9317fc4e551cf974684eaa88697de5a28srs#include "support.h" 11e7b4ff9317fc4e551cf974684eaa88697de5a28srs#include "mbr.h" 12221e08768de7fe42ba533ca22baf671420569c07srs#include "bsd.h" 13221e08768de7fe42ba533ca22baf671420569c07srs#include "gptpart.h" 14e7b4ff9317fc4e551cf974684eaa88697de5a28srs 15e7b4ff9317fc4e551cf974684eaa88697de5a28srs#ifndef __GPTSTRUCTS 16e7b4ff9317fc4e551cf974684eaa88697de5a28srs#define __GPTSTRUCTS 17e7b4ff9317fc4e551cf974684eaa88697de5a28srs 18a8582cfe6c1aa5e5f80458ac72d881a04ae0ba44srs// Default values for sector alignment 19a8582cfe6c1aa5e5f80458ac72d881a04ae0ba44srs#define DEFAULT_ALIGNMENT 2048 208a4ddfc919d5569c68489cf53d9cf5abc94c410csrs#define MAX_ALIGNMENT 65536 210873e9d0e9345a2c4418b4718db525c9f1111c83srs#define MIN_AF_ALIGNMENT 8 22a8582cfe6c1aa5e5f80458ac72d881a04ae0ba44srs 2364cbd171067eb34054741bfcd73f0b91d727a371srs// Below constant corresponds to a ~279GiB (300GB) disk, since the 2464cbd171067eb34054741bfcd73f0b91d727a371srs// smallest Advanced Format drive I know of is 320GB in size 2564cbd171067eb34054741bfcd73f0b91d727a371srs#define SMALLEST_ADVANCED_FORMAT UINT64_C(585937500) 26a8582cfe6c1aa5e5f80458ac72d881a04ae0ba44srs 27e7b4ff9317fc4e551cf974684eaa88697de5a28srsusing namespace std; 28e7b4ff9317fc4e551cf974684eaa88697de5a28srs 29e7b4ff9317fc4e551cf974684eaa88697de5a28srs/**************************************** 30e7b4ff9317fc4e551cf974684eaa88697de5a28srs * * 31e7b4ff9317fc4e551cf974684eaa88697de5a28srs * GPTData class and related structures * 32e7b4ff9317fc4e551cf974684eaa88697de5a28srs * * 33e7b4ff9317fc4e551cf974684eaa88697de5a28srs ****************************************/ 34e7b4ff9317fc4e551cf974684eaa88697de5a28srs 35e7b4ff9317fc4e551cf974684eaa88697de5a28srs// Validity state of GPT data 36e7b4ff9317fc4e551cf974684eaa88697de5a28srsenum GPTValidity {gpt_valid, gpt_corrupt, gpt_invalid}; 37e7b4ff9317fc4e551cf974684eaa88697de5a28srs 38e7b4ff9317fc4e551cf974684eaa88697de5a28srs// Which set of partition data to use 393c0af38237d0f40aaea8233a5cbfdd030a77817dsrsenum WhichToUse {use_gpt, use_mbr, use_bsd, use_new, use_abort}; 40e7b4ff9317fc4e551cf974684eaa88697de5a28srs 41e7b4ff9317fc4e551cf974684eaa88697de5a28srs// Header (first 512 bytes) of GPT table 42ba00fed2efd6c0cba60da9afb0ce3dff84fc69f9srs#pragma pack(1) 43e7b4ff9317fc4e551cf974684eaa88697de5a28srsstruct GPTHeader { 44e7b4ff9317fc4e551cf974684eaa88697de5a28srs uint64_t signature; 45e7b4ff9317fc4e551cf974684eaa88697de5a28srs uint32_t revision; 46e7b4ff9317fc4e551cf974684eaa88697de5a28srs uint32_t headerSize; 47e7b4ff9317fc4e551cf974684eaa88697de5a28srs uint32_t headerCRC; 48e7b4ff9317fc4e551cf974684eaa88697de5a28srs uint32_t reserved; 49e7b4ff9317fc4e551cf974684eaa88697de5a28srs uint64_t currentLBA; 50e7b4ff9317fc4e551cf974684eaa88697de5a28srs uint64_t backupLBA; 51e7b4ff9317fc4e551cf974684eaa88697de5a28srs uint64_t firstUsableLBA; 52e7b4ff9317fc4e551cf974684eaa88697de5a28srs uint64_t lastUsableLBA; 536699b01eda84d24bfaf80ad725304fef2b0e1b2asrs GUIDData diskGUID; 54e7b4ff9317fc4e551cf974684eaa88697de5a28srs uint64_t partitionEntriesLBA; 55e7b4ff9317fc4e551cf974684eaa88697de5a28srs uint32_t numParts; 56e7b4ff9317fc4e551cf974684eaa88697de5a28srs uint32_t sizeOfPartitionEntries; 57e7b4ff9317fc4e551cf974684eaa88697de5a28srs uint32_t partitionEntriesCRC; 58e7b4ff9317fc4e551cf974684eaa88697de5a28srs unsigned char reserved2[GPT_RESERVED]; 59e7b4ff9317fc4e551cf974684eaa88697de5a28srs}; // struct GPTHeader 60e7b4ff9317fc4e551cf974684eaa88697de5a28srs 61e7b4ff9317fc4e551cf974684eaa88697de5a28srs// Data in GPT format 62e7b4ff9317fc4e551cf974684eaa88697de5a28srsclass GPTData { 63e7b4ff9317fc4e551cf974684eaa88697de5a28srsprotected: 64e7b4ff9317fc4e551cf974684eaa88697de5a28srs struct GPTHeader mainHeader; 65ba00fed2efd6c0cba60da9afb0ce3dff84fc69f9srs GPTPart *partitions; 66815fb65195106b8afe1b8dfec5dae605dbd7ccbesrs uint32_t numParts; // # of partitions the table can hold 67e7b4ff9317fc4e551cf974684eaa88697de5a28srs struct GPTHeader secondHeader; 68e7b4ff9317fc4e551cf974684eaa88697de5a28srs MBRData protectiveMBR; 69fed16d043a14e8b86c97a6413aec7281fefcbcb5srs string device; // device filename 70546a9c7c369df465021feecb20f6a8f81b6df6bcsrs DiskIO myDisk; 71e7b4ff9317fc4e551cf974684eaa88697de5a28srs uint32_t blockSize; // device block size 72e7b4ff9317fc4e551cf974684eaa88697de5a28srs uint64_t diskSize; // size of device, in blocks 73e7b4ff9317fc4e551cf974684eaa88697de5a28srs GPTValidity state; // is GPT valid? 745d58fe0ea12c9c727c8a970c8e1ac08ea7fbe05fsrs int justLooking; // Set to 1 if program launched with "-l" or if read-only 75e7b4ff9317fc4e551cf974684eaa88697de5a28srs int mainCrcOk; 76e7b4ff9317fc4e551cf974684eaa88697de5a28srs int secondCrcOk; 77e7b4ff9317fc4e551cf974684eaa88697de5a28srs int mainPartsCrcOk; 78e7b4ff9317fc4e551cf974684eaa88697de5a28srs int secondPartsCrcOk; 79221e08768de7fe42ba533ca22baf671420569c07srs int apmFound; // set to 1 if APM detected 80221e08768de7fe42ba533ca22baf671420569c07srs int bsdFound; // set to 1 if BSD disklabel detected in MBR 810873e9d0e9345a2c4418b4718db525c9f1111c83srs uint32_t sectorAlignment; // Start partitions at multiples of sectorAlignment 82ba00fed2efd6c0cba60da9afb0ce3dff84fc69f9srs int beQuiet; 83ba00fed2efd6c0cba60da9afb0ce3dff84fc69f9srs WhichToUse whichWasUsed; 84cb76c673eeb84344887715d36d44b799042be5a5srs 85cb76c673eeb84344887715d36d44b799042be5a5srs int LoadHeader(struct GPTHeader *header, DiskIO & disk, uint64_t sector, int *crcOk); 86cb76c673eeb84344887715d36d44b799042be5a5srs int LoadPartitionTable(const struct GPTHeader & header, DiskIO & disk, uint64_t sector = 0); 87cb76c673eeb84344887715d36d44b799042be5a5srs int CheckTable(struct GPTHeader *header); 88cb76c673eeb84344887715d36d44b799042be5a5srs int SaveHeader(struct GPTHeader *header, DiskIO & disk, uint64_t sector); 89cb76c673eeb84344887715d36d44b799042be5a5srs int SavePartitionTable(DiskIO & disk, uint64_t sector); 90e7b4ff9317fc4e551cf974684eaa88697de5a28srspublic: 91e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs // Basic necessary functions.... 92e7b4ff9317fc4e551cf974684eaa88697de5a28srs GPTData(void); 93fed16d043a14e8b86c97a6413aec7281fefcbcb5srs GPTData(string deviceFilename); 9408bb0da07953af605b4918e268272de15ac151aasrs virtual ~GPTData(void); 9564cbd171067eb34054741bfcd73f0b91d727a371srs GPTData & operator=(const GPTData & orig); 96e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs 97e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs // Verify (or update) data integrity 98e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs int Verify(void); 99e7b4ff9317fc4e551cf974684eaa88697de5a28srs int CheckGPTSize(void); 100e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs int CheckHeaderValidity(void); 101d1b11e8305621d73ff675af940e7f1f28b639b0dsrs int CheckHeaderCRC(struct GPTHeader* header, int warn = 0); 102e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs void RecomputeCRCs(void); 103e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs void RebuildMainHeader(void); 104e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs void RebuildSecondHeader(void); 105bf8950cad0285ee6ab8a896e8d0a30c5fb62c7afsrs int VerifyMBR(void) {return protectiveMBR.FindOverlaps();} 106e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs int FindHybridMismatches(void); 107e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs int FindOverlaps(void); 10855d926192adc984462509b2966e23bc0d1129bbdsrs int FindInsanePartitions(void); 109e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs 110e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs // Load or save data from/to disk 111bf8950cad0285ee6ab8a896e8d0a30c5fb62c7afsrs int SetDisk(const string & deviceFilename); 112bf8950cad0285ee6ab8a896e8d0a30c5fb62c7afsrs DiskIO* GetDisk(void) {return &myDisk;} 1130a6973119c9e9984ad47a6da3231e8d16f996c5csrs int LoadMBR(const string & f) {return protectiveMBR.ReadMBRData(f);} 11408bb0da07953af605b4918e268272de15ac151aasrs int WriteProtectiveMBR(void) {return protectiveMBR.WriteMBRData(&myDisk);} 115546a9c7c369df465021feecb20f6a8f81b6df6bcsrs void PartitionScan(void); 1160a6973119c9e9984ad47a6da3231e8d16f996c5csrs int LoadPartitions(const string & deviceFilename); 117546a9c7c369df465021feecb20f6a8f81b6df6bcsrs int ForceLoadGPTData(void); 118e7b4ff9317fc4e551cf974684eaa88697de5a28srs int LoadMainTable(void); 119546a9c7c369df465021feecb20f6a8f81b6df6bcsrs int LoadSecondTableAsMain(void); 12064cbd171067eb34054741bfcd73f0b91d727a371srs int SaveGPTData(int quiet = 0); 1210a6973119c9e9984ad47a6da3231e8d16f996c5csrs int SaveGPTBackup(const string & filename); 1220a6973119c9e9984ad47a6da3231e8d16f996c5csrs int LoadGPTBackup(const string & filename); 12308bb0da07953af605b4918e268272de15ac151aasrs int SaveMBR(void); 12408bb0da07953af605b4918e268272de15ac151aasrs int DestroyGPT(void); 12508bb0da07953af605b4918e268272de15ac151aasrs int DestroyMBR(void); 126e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs 127e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs // Display data.... 128e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs void ShowAPMState(void); 129e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs void ShowGPTState(void); 130e7b4ff9317fc4e551cf974684eaa88697de5a28srs void DisplayGPTData(void); 131e7b4ff9317fc4e551cf974684eaa88697de5a28srs void DisplayMBRData(void) {protectiveMBR.DisplayMBRData();} 132d761ff5a2aac9d9b6bd0bc8236419b1cf0128c86Jeff Sharkey void ShowPartDetails(uint32_t partNum); 133e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs 13408bb0da07953af605b4918e268272de15ac151aasrs // Convert between GPT and other formats 13508bb0da07953af605b4918e268272de15ac151aasrs virtual WhichToUse UseWhichPartitions(void); 13608bb0da07953af605b4918e268272de15ac151aasrs void XFormPartitions(void); 1371eea9b0b51367472ce12efb59b0018c0ac96a463Roderick W. Smith int XFormDisklabel(uint32_t partNum); 13808bb0da07953af605b4918e268272de15ac151aasrs int XFormDisklabel(BSDData* disklabel); 139978041ca613dcb881763b36cf53639d924e52a56srs int OnePartToMBR(uint32_t gptPart, int mbrPart); // add one partition to MBR. Returns 1 if successful 140e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs 141e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs // Adjust GPT structures WITHOUT user interaction... 142706e51217a531c46afc743b556e10fd5c0585fcfsrs int SetGPTSize(uint32_t numEntries, int fillGPTSectors = 1); 143e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs void BlankPartitions(void); 144ba00fed2efd6c0cba60da9afb0ce3dff84fc69f9srs int DeletePartition(uint32_t partNum); 145e321d444dcca514cf6b53459e388ddcbaab6176csrs uint32_t CreatePartition(uint32_t partNum, uint64_t startSector, uint64_t endSector); 146e7b4ff9317fc4e551cf974684eaa88697de5a28srs void SortGPT(void); 14708bb0da07953af605b4918e268272de15ac151aasrs int SwapPartitions(uint32_t partNum1, uint32_t partNum2); 148e7b4ff9317fc4e551cf974684eaa88697de5a28srs int ClearGPTData(void); 149247657a5acbb7eb21c336ba84a68b801b7c19be0srs void MoveSecondHeaderToEnd(); 150699941e25a1fcf0beec124203747c8ed20842989srs int SetName(uint32_t partNum, const UnicodeString & theName); 151e7b4ff9317fc4e551cf974684eaa88697de5a28srs void SetDiskGUID(GUIDData newGUID); 152e7b4ff9317fc4e551cf974684eaa88697de5a28srs int SetPartitionGUID(uint32_t pn, GUIDData theGUID); 1539ba5421f920e192dbb808d30aa6d34849938bab4srs void RandomizeGUIDs(void); 154327129e9331f888a8fc08d688dcb0a739a3c17besrs int ChangePartType(uint32_t pn, PartType theGUID); 155e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs void MakeProtectiveMBR(void) {protectiveMBR.MakeProtectiveMBR();} 1569ba5421f920e192dbb808d30aa6d34849938bab4srs void RecomputeCHS(void); 1571d1448a82d62ad32a8d597ed9ade46b4f37d8eb5srs int Align(uint64_t* sector); 158bf8950cad0285ee6ab8a896e8d0a30c5fb62c7afsrs void SetProtectiveMBR(BasicMBRData & newMBR) {protectiveMBR = newMBR;} 15900b6d7a4604e759eb3c92b3ecea608d6fe024b81srs 16000b6d7a4604e759eb3c92b3ecea608d6fe024b81srs // Return data about the GPT structures.... 16100b6d7a4604e759eb3c92b3ecea608d6fe024b81srs WhichToUse GetState(void) {return whichWasUsed;} 162e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs int GetPartRange(uint32_t* low, uint32_t* high); 16308bb0da07953af605b4918e268272de15ac151aasrs int FindFirstFreePart(void); 164e7b4ff9317fc4e551cf974684eaa88697de5a28srs uint32_t GetNumParts(void) {return mainHeader.numParts;} 165e7b4ff9317fc4e551cf974684eaa88697de5a28srs uint64_t GetMainHeaderLBA(void) {return mainHeader.currentLBA;} 166e7b4ff9317fc4e551cf974684eaa88697de5a28srs uint64_t GetSecondHeaderLBA(void) {return secondHeader.currentLBA;} 167e7b4ff9317fc4e551cf974684eaa88697de5a28srs uint64_t GetMainPartsLBA(void) {return mainHeader.partitionEntriesLBA;} 168e7b4ff9317fc4e551cf974684eaa88697de5a28srs uint64_t GetSecondPartsLBA(void) {return secondHeader.partitionEntriesLBA;} 16955d926192adc984462509b2966e23bc0d1129bbdsrs uint64_t GetFirstUsableLBA(void) {return mainHeader.firstUsableLBA;} 17055d926192adc984462509b2966e23bc0d1129bbdsrs uint64_t GetLastUsableLBA(void) {return mainHeader.lastUsableLBA;} 171978041ca613dcb881763b36cf53639d924e52a56srs uint32_t CountParts(void); 1729ddc14bb9b154518e2b8384d3f4571cf657c7920srs bool ValidPartNum (const uint32_t partNum); 1735a081757ea2e32a491349544fea92826ccf739f6srs const GPTPart & operator[](uint32_t partNum) const; 1745a081757ea2e32a491349544fea92826ccf739f6srs const GUIDData & GetDiskGUID(void) const; 175df9d363d341a0ffdd05250fd4ffb842f59815690srs uint32_t GetBlockSize(void) {return blockSize;} 176e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs 177e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs // Find information about free space 178e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs uint64_t FindFirstAvailable(uint64_t start = 0); 179e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs uint64_t FindFirstInLargest(void); 180cb76c673eeb84344887715d36d44b799042be5a5srs uint64_t FindLastAvailable(); 181e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs uint64_t FindLastInFree(uint64_t start); 182e321d444dcca514cf6b53459e388ddcbaab6176csrs uint64_t FindFreeBlocks(uint32_t *numSegments, uint64_t *largestSegment); 18355d926192adc984462509b2966e23bc0d1129bbdsrs int IsFree(uint64_t sector, uint32_t *partNum = NULL); 184ba00fed2efd6c0cba60da9afb0ce3dff84fc69f9srs int IsFreePartNum(uint32_t partNum); 185815fb65195106b8afe1b8dfec5dae605dbd7ccbesrs int IsUsedPartNum(uint32_t partNum); 186d8eed4629449a325999808a0170dbda53bd4a6dfsrs 187ba00fed2efd6c0cba60da9afb0ce3dff84fc69f9srs // Change how functions work, or return information on same 188a8582cfe6c1aa5e5f80458ac72d881a04ae0ba44srs void SetAlignment(uint32_t n); 189a8582cfe6c1aa5e5f80458ac72d881a04ae0ba44srs uint32_t ComputeAlignment(void); // Set alignment based on current partitions 190a8582cfe6c1aa5e5f80458ac72d881a04ae0ba44srs uint32_t GetAlignment(void) {return sectorAlignment;} 191ba00fed2efd6c0cba60da9afb0ce3dff84fc69f9srs void JustLooking(int i = 1) {justLooking = i;} 192ba00fed2efd6c0cba60da9afb0ce3dff84fc69f9srs void BeQuiet(int i = 1) {beQuiet = i;} 193ba00fed2efd6c0cba60da9afb0ce3dff84fc69f9srs WhichToUse WhichWasUsed(void) {return whichWasUsed;} 194e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs 195e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs // Endianness functions 1960a6973119c9e9984ad47a6da3231e8d16f996c5csrs void ReverseHeaderBytes(struct GPTHeader* header); 197e4ac11ebee0ad586a538f49b9c0e32ad19cae665srs void ReversePartitionBytes(); // for endianness 1989ddc14bb9b154518e2b8384d3f4571cf657c7920srs 1999ddc14bb9b154518e2b8384d3f4571cf657c7920srs // Attributes functions 2009ddc14bb9b154518e2b8384d3f4571cf657c7920srs int ManageAttributes(int partNum, const string & command, const string & bits); 2019ddc14bb9b154518e2b8384d3f4571cf657c7920srs void ShowAttributes(const uint32_t partNum); 2029ddc14bb9b154518e2b8384d3f4571cf657c7920srs void GetAttribute(const uint32_t partNum, const string& attributeBits); 2039ddc14bb9b154518e2b8384d3f4571cf657c7920srs 204e7b4ff9317fc4e551cf974684eaa88697de5a28srs}; // class GPTData 205e7b4ff9317fc4e551cf974684eaa88697de5a28srs 206e7b4ff9317fc4e551cf974684eaa88697de5a28srs// Function prototypes.... 207e7b4ff9317fc4e551cf974684eaa88697de5a28srsint SizesOK(void); 208e7b4ff9317fc4e551cf974684eaa88697de5a28srs 209e7b4ff9317fc4e551cf974684eaa88697de5a28srs#endif 210