srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 1 | /* gpt.h -- GPT and data structure definitions, types, and |
| 2 | functions */ |
| 3 | |
srs5694 | 64cbd17 | 2011-03-01 22:03:54 -0500 | [diff] [blame] | 4 | /* This program is copyright (c) 2009-2011 by Roderick W. Smith. It is distributed |
srs5694 | 221e087 | 2009-08-29 15:00:31 -0400 | [diff] [blame] | 5 | under the terms of the GNU GPL version 2, as detailed in the COPYING file. */ |
| 6 | |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 7 | #include <stdint.h> |
| 8 | #include <sys/types.h> |
srs5694 | 546a9c7 | 2010-01-26 16:00:26 -0500 | [diff] [blame] | 9 | #include "gptpart.h" |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 10 | #include "support.h" |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 11 | #include "mbr.h" |
srs5694 | 221e087 | 2009-08-29 15:00:31 -0400 | [diff] [blame] | 12 | #include "bsd.h" |
| 13 | #include "gptpart.h" |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 14 | |
| 15 | #ifndef __GPTSTRUCTS |
| 16 | #define __GPTSTRUCTS |
| 17 | |
srs5694 | a8582cf | 2010-03-19 14:21:59 -0400 | [diff] [blame] | 18 | // Default values for sector alignment |
| 19 | #define DEFAULT_ALIGNMENT 2048 |
srs5694 | 8a4ddfc | 2010-03-21 19:05:49 -0400 | [diff] [blame] | 20 | #define MAX_ALIGNMENT 65536 |
srs5694 | 0873e9d | 2010-10-07 13:00:45 -0400 | [diff] [blame] | 21 | #define MIN_AF_ALIGNMENT 8 |
srs5694 | a8582cf | 2010-03-19 14:21:59 -0400 | [diff] [blame] | 22 | |
srs5694 | 64cbd17 | 2011-03-01 22:03:54 -0500 | [diff] [blame] | 23 | // Below constant corresponds to a ~279GiB (300GB) disk, since the |
| 24 | // smallest Advanced Format drive I know of is 320GB in size |
| 25 | #define SMALLEST_ADVANCED_FORMAT UINT64_C(585937500) |
srs5694 | a8582cf | 2010-03-19 14:21:59 -0400 | [diff] [blame] | 26 | |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 27 | using namespace std; |
| 28 | |
| 29 | /**************************************** |
| 30 | * * |
| 31 | * GPTData class and related structures * |
| 32 | * * |
| 33 | ****************************************/ |
| 34 | |
| 35 | // Validity state of GPT data |
| 36 | enum GPTValidity {gpt_valid, gpt_corrupt, gpt_invalid}; |
| 37 | |
| 38 | // Which set of partition data to use |
srs5694 | 3c0af38 | 2010-01-15 19:19:18 -0500 | [diff] [blame] | 39 | enum WhichToUse {use_gpt, use_mbr, use_bsd, use_new, use_abort}; |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 40 | |
| 41 | // Header (first 512 bytes) of GPT table |
srs5694 | ba00fed | 2010-01-12 18:18:36 -0500 | [diff] [blame] | 42 | #pragma pack(1) |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 43 | struct GPTHeader { |
| 44 | uint64_t signature; |
| 45 | uint32_t revision; |
| 46 | uint32_t headerSize; |
| 47 | uint32_t headerCRC; |
| 48 | uint32_t reserved; |
| 49 | uint64_t currentLBA; |
| 50 | uint64_t backupLBA; |
| 51 | uint64_t firstUsableLBA; |
| 52 | uint64_t lastUsableLBA; |
srs5694 | 6699b01 | 2010-02-04 00:55:30 -0500 | [diff] [blame] | 53 | GUIDData diskGUID; |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 54 | uint64_t partitionEntriesLBA; |
| 55 | uint32_t numParts; |
| 56 | uint32_t sizeOfPartitionEntries; |
| 57 | uint32_t partitionEntriesCRC; |
| 58 | unsigned char reserved2[GPT_RESERVED]; |
| 59 | }; // struct GPTHeader |
| 60 | |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 61 | // Data in GPT format |
| 62 | class GPTData { |
| 63 | protected: |
| 64 | struct GPTHeader mainHeader; |
srs5694 | ba00fed | 2010-01-12 18:18:36 -0500 | [diff] [blame] | 65 | GPTPart *partitions; |
srs5694 | 815fb65 | 2011-03-18 12:35:56 -0400 | [diff] [blame] | 66 | uint32_t numParts; // # of partitions the table can hold |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 67 | struct GPTHeader secondHeader; |
| 68 | MBRData protectiveMBR; |
srs5694 | fed16d0 | 2010-01-27 23:03:40 -0500 | [diff] [blame] | 69 | string device; // device filename |
srs5694 | 546a9c7 | 2010-01-26 16:00:26 -0500 | [diff] [blame] | 70 | DiskIO myDisk; |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 71 | uint32_t blockSize; // device block size |
| 72 | uint64_t diskSize; // size of device, in blocks |
| 73 | GPTValidity state; // is GPT valid? |
srs5694 | 5d58fe0 | 2010-01-03 20:57:08 -0500 | [diff] [blame] | 74 | int justLooking; // Set to 1 if program launched with "-l" or if read-only |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 75 | int mainCrcOk; |
| 76 | int secondCrcOk; |
| 77 | int mainPartsCrcOk; |
| 78 | int secondPartsCrcOk; |
srs5694 | 221e087 | 2009-08-29 15:00:31 -0400 | [diff] [blame] | 79 | int apmFound; // set to 1 if APM detected |
| 80 | int bsdFound; // set to 1 if BSD disklabel detected in MBR |
srs5694 | 0873e9d | 2010-10-07 13:00:45 -0400 | [diff] [blame] | 81 | uint32_t sectorAlignment; // Start partitions at multiples of sectorAlignment |
srs5694 | ba00fed | 2010-01-12 18:18:36 -0500 | [diff] [blame] | 82 | int beQuiet; |
| 83 | WhichToUse whichWasUsed; |
srs5694 | cb76c67 | 2010-02-11 22:22:22 -0500 | [diff] [blame] | 84 | |
| 85 | int LoadHeader(struct GPTHeader *header, DiskIO & disk, uint64_t sector, int *crcOk); |
| 86 | int LoadPartitionTable(const struct GPTHeader & header, DiskIO & disk, uint64_t sector = 0); |
| 87 | int CheckTable(struct GPTHeader *header); |
| 88 | int SaveHeader(struct GPTHeader *header, DiskIO & disk, uint64_t sector); |
| 89 | int SavePartitionTable(DiskIO & disk, uint64_t sector); |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 90 | public: |
srs5694 | e4ac11e | 2009-08-31 10:13:04 -0400 | [diff] [blame] | 91 | // Basic necessary functions.... |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 92 | GPTData(void); |
srs5694 | fed16d0 | 2010-01-27 23:03:40 -0500 | [diff] [blame] | 93 | GPTData(string deviceFilename); |
srs5694 | 08bb0da | 2010-02-19 17:19:55 -0500 | [diff] [blame] | 94 | virtual ~GPTData(void); |
srs5694 | 64cbd17 | 2011-03-01 22:03:54 -0500 | [diff] [blame] | 95 | GPTData & operator=(const GPTData & orig); |
srs5694 | e4ac11e | 2009-08-31 10:13:04 -0400 | [diff] [blame] | 96 | |
| 97 | // Verify (or update) data integrity |
| 98 | int Verify(void); |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 99 | int CheckGPTSize(void); |
srs5694 | e4ac11e | 2009-08-31 10:13:04 -0400 | [diff] [blame] | 100 | int CheckHeaderValidity(void); |
srs5694 | d1b11e8 | 2011-09-18 21:12:28 -0400 | [diff] [blame] | 101 | int CheckHeaderCRC(struct GPTHeader* header, int warn = 0); |
srs5694 | e4ac11e | 2009-08-31 10:13:04 -0400 | [diff] [blame] | 102 | void RecomputeCRCs(void); |
| 103 | void RebuildMainHeader(void); |
| 104 | void RebuildSecondHeader(void); |
srs5694 | bf8950c | 2011-03-12 01:23:12 -0500 | [diff] [blame] | 105 | int VerifyMBR(void) {return protectiveMBR.FindOverlaps();} |
srs5694 | e4ac11e | 2009-08-31 10:13:04 -0400 | [diff] [blame] | 106 | int FindHybridMismatches(void); |
| 107 | int FindOverlaps(void); |
srs5694 | 55d9261 | 2010-03-07 22:16:07 -0500 | [diff] [blame] | 108 | int FindInsanePartitions(void); |
srs5694 | e4ac11e | 2009-08-31 10:13:04 -0400 | [diff] [blame] | 109 | |
| 110 | // Load or save data from/to disk |
srs5694 | bf8950c | 2011-03-12 01:23:12 -0500 | [diff] [blame] | 111 | int SetDisk(const string & deviceFilename); |
| 112 | DiskIO* GetDisk(void) {return &myDisk;} |
srs5694 | 0a69731 | 2010-01-28 21:10:52 -0500 | [diff] [blame] | 113 | int LoadMBR(const string & f) {return protectiveMBR.ReadMBRData(f);} |
srs5694 | 08bb0da | 2010-02-19 17:19:55 -0500 | [diff] [blame] | 114 | int WriteProtectiveMBR(void) {return protectiveMBR.WriteMBRData(&myDisk);} |
srs5694 | 546a9c7 | 2010-01-26 16:00:26 -0500 | [diff] [blame] | 115 | void PartitionScan(void); |
srs5694 | 0a69731 | 2010-01-28 21:10:52 -0500 | [diff] [blame] | 116 | int LoadPartitions(const string & deviceFilename); |
srs5694 | 546a9c7 | 2010-01-26 16:00:26 -0500 | [diff] [blame] | 117 | int ForceLoadGPTData(void); |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 118 | int LoadMainTable(void); |
srs5694 | 546a9c7 | 2010-01-26 16:00:26 -0500 | [diff] [blame] | 119 | int LoadSecondTableAsMain(void); |
srs5694 | 64cbd17 | 2011-03-01 22:03:54 -0500 | [diff] [blame] | 120 | int SaveGPTData(int quiet = 0); |
srs5694 | 0a69731 | 2010-01-28 21:10:52 -0500 | [diff] [blame] | 121 | int SaveGPTBackup(const string & filename); |
| 122 | int LoadGPTBackup(const string & filename); |
srs5694 | 08bb0da | 2010-02-19 17:19:55 -0500 | [diff] [blame] | 123 | int SaveMBR(void); |
| 124 | int DestroyGPT(void); |
| 125 | int DestroyMBR(void); |
srs5694 | e4ac11e | 2009-08-31 10:13:04 -0400 | [diff] [blame] | 126 | |
| 127 | // Display data.... |
| 128 | void ShowAPMState(void); |
| 129 | void ShowGPTState(void); |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 130 | void DisplayGPTData(void); |
| 131 | void DisplayMBRData(void) {protectiveMBR.DisplayMBRData();} |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 132 | void ShowPartDetails(uint32_t partNum); |
srs5694 | e4ac11e | 2009-08-31 10:13:04 -0400 | [diff] [blame] | 133 | |
srs5694 | 08bb0da | 2010-02-19 17:19:55 -0500 | [diff] [blame] | 134 | // Convert between GPT and other formats |
| 135 | virtual WhichToUse UseWhichPartitions(void); |
| 136 | void XFormPartitions(void); |
| 137 | virtual int XFormDisklabel(uint32_t partNum); |
| 138 | int XFormDisklabel(BSDData* disklabel); |
srs5694 | 978041c | 2009-09-21 20:51:47 -0400 | [diff] [blame] | 139 | int OnePartToMBR(uint32_t gptPart, int mbrPart); // add one partition to MBR. Returns 1 if successful |
srs5694 | e4ac11e | 2009-08-31 10:13:04 -0400 | [diff] [blame] | 140 | |
| 141 | // Adjust GPT structures WITHOUT user interaction... |
srs5694 | 706e512 | 2012-01-21 13:47:24 -0500 | [diff] [blame] | 142 | int SetGPTSize(uint32_t numEntries, int fillGPTSectors = 1); |
srs5694 | e4ac11e | 2009-08-31 10:13:04 -0400 | [diff] [blame] | 143 | void BlankPartitions(void); |
srs5694 | ba00fed | 2010-01-12 18:18:36 -0500 | [diff] [blame] | 144 | int DeletePartition(uint32_t partNum); |
srs5694 | e321d44 | 2010-01-29 17:44:04 -0500 | [diff] [blame] | 145 | uint32_t CreatePartition(uint32_t partNum, uint64_t startSector, uint64_t endSector); |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 146 | void SortGPT(void); |
srs5694 | 08bb0da | 2010-02-19 17:19:55 -0500 | [diff] [blame] | 147 | int SwapPartitions(uint32_t partNum1, uint32_t partNum2); |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 148 | int ClearGPTData(void); |
srs5694 | 247657a | 2009-11-26 18:36:12 -0500 | [diff] [blame] | 149 | void MoveSecondHeaderToEnd(); |
srs5694 | 699941e | 2011-03-21 21:33:57 -0400 | [diff] [blame] | 150 | int SetName(uint32_t partNum, const UnicodeString & theName); |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 151 | void SetDiskGUID(GUIDData newGUID); |
| 152 | int SetPartitionGUID(uint32_t pn, GUIDData theGUID); |
srs5694 | 9ba5421 | 2010-05-18 23:24:02 -0400 | [diff] [blame] | 153 | void RandomizeGUIDs(void); |
srs5694 | 327129e | 2010-09-22 01:07:31 -0400 | [diff] [blame] | 154 | int ChangePartType(uint32_t pn, PartType theGUID); |
srs5694 | e4ac11e | 2009-08-31 10:13:04 -0400 | [diff] [blame] | 155 | void MakeProtectiveMBR(void) {protectiveMBR.MakeProtectiveMBR();} |
srs5694 | 9ba5421 | 2010-05-18 23:24:02 -0400 | [diff] [blame] | 156 | void RecomputeCHS(void); |
srs5694 | 1d1448a | 2009-12-31 21:20:19 -0500 | [diff] [blame] | 157 | int Align(uint64_t* sector); |
srs5694 | bf8950c | 2011-03-12 01:23:12 -0500 | [diff] [blame] | 158 | void SetProtectiveMBR(BasicMBRData & newMBR) {protectiveMBR = newMBR;} |
srs5694 | 00b6d7a | 2011-06-26 22:40:06 -0400 | [diff] [blame] | 159 | |
| 160 | // Return data about the GPT structures.... |
| 161 | WhichToUse GetState(void) {return whichWasUsed;} |
srs5694 | e4ac11e | 2009-08-31 10:13:04 -0400 | [diff] [blame] | 162 | int GetPartRange(uint32_t* low, uint32_t* high); |
srs5694 | 08bb0da | 2010-02-19 17:19:55 -0500 | [diff] [blame] | 163 | int FindFirstFreePart(void); |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 164 | uint32_t GetNumParts(void) {return mainHeader.numParts;} |
| 165 | uint64_t GetMainHeaderLBA(void) {return mainHeader.currentLBA;} |
| 166 | uint64_t GetSecondHeaderLBA(void) {return secondHeader.currentLBA;} |
| 167 | uint64_t GetMainPartsLBA(void) {return mainHeader.partitionEntriesLBA;} |
| 168 | uint64_t GetSecondPartsLBA(void) {return secondHeader.partitionEntriesLBA;} |
srs5694 | 55d9261 | 2010-03-07 22:16:07 -0500 | [diff] [blame] | 169 | uint64_t GetFirstUsableLBA(void) {return mainHeader.firstUsableLBA;} |
| 170 | uint64_t GetLastUsableLBA(void) {return mainHeader.lastUsableLBA;} |
srs5694 | 978041c | 2009-09-21 20:51:47 -0400 | [diff] [blame] | 171 | uint32_t CountParts(void); |
srs5694 | 9ddc14b | 2010-08-22 22:44:42 -0400 | [diff] [blame] | 172 | bool ValidPartNum (const uint32_t partNum); |
srs5694 | 5a08175 | 2010-09-24 20:39:41 -0400 | [diff] [blame] | 173 | const GPTPart & operator[](uint32_t partNum) const; |
| 174 | const GUIDData & GetDiskGUID(void) const; |
srs5694 | df9d363 | 2011-01-08 18:33:24 -0500 | [diff] [blame] | 175 | uint32_t GetBlockSize(void) {return blockSize;} |
srs5694 | e4ac11e | 2009-08-31 10:13:04 -0400 | [diff] [blame] | 176 | |
| 177 | // Find information about free space |
| 178 | uint64_t FindFirstAvailable(uint64_t start = 0); |
| 179 | uint64_t FindFirstInLargest(void); |
srs5694 | cb76c67 | 2010-02-11 22:22:22 -0500 | [diff] [blame] | 180 | uint64_t FindLastAvailable(); |
srs5694 | e4ac11e | 2009-08-31 10:13:04 -0400 | [diff] [blame] | 181 | uint64_t FindLastInFree(uint64_t start); |
srs5694 | e321d44 | 2010-01-29 17:44:04 -0500 | [diff] [blame] | 182 | uint64_t FindFreeBlocks(uint32_t *numSegments, uint64_t *largestSegment); |
srs5694 | 55d9261 | 2010-03-07 22:16:07 -0500 | [diff] [blame] | 183 | int IsFree(uint64_t sector, uint32_t *partNum = NULL); |
srs5694 | ba00fed | 2010-01-12 18:18:36 -0500 | [diff] [blame] | 184 | int IsFreePartNum(uint32_t partNum); |
srs5694 | 815fb65 | 2011-03-18 12:35:56 -0400 | [diff] [blame] | 185 | int IsUsedPartNum(uint32_t partNum); |
| 186 | |
srs5694 | ba00fed | 2010-01-12 18:18:36 -0500 | [diff] [blame] | 187 | // Change how functions work, or return information on same |
srs5694 | a8582cf | 2010-03-19 14:21:59 -0400 | [diff] [blame] | 188 | void SetAlignment(uint32_t n); |
| 189 | uint32_t ComputeAlignment(void); // Set alignment based on current partitions |
| 190 | uint32_t GetAlignment(void) {return sectorAlignment;} |
srs5694 | ba00fed | 2010-01-12 18:18:36 -0500 | [diff] [blame] | 191 | void JustLooking(int i = 1) {justLooking = i;} |
| 192 | void BeQuiet(int i = 1) {beQuiet = i;} |
| 193 | WhichToUse WhichWasUsed(void) {return whichWasUsed;} |
srs5694 | e4ac11e | 2009-08-31 10:13:04 -0400 | [diff] [blame] | 194 | |
| 195 | // Endianness functions |
srs5694 | 0a69731 | 2010-01-28 21:10:52 -0500 | [diff] [blame] | 196 | void ReverseHeaderBytes(struct GPTHeader* header); |
srs5694 | e4ac11e | 2009-08-31 10:13:04 -0400 | [diff] [blame] | 197 | void ReversePartitionBytes(); // for endianness |
srs5694 | 9ddc14b | 2010-08-22 22:44:42 -0400 | [diff] [blame] | 198 | |
| 199 | // Attributes functions |
| 200 | int ManageAttributes(int partNum, const string & command, const string & bits); |
| 201 | void ShowAttributes(const uint32_t partNum); |
| 202 | void GetAttribute(const uint32_t partNum, const string& attributeBits); |
| 203 | |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 204 | }; // class GPTData |
| 205 | |
| 206 | // Function prototypes.... |
srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 207 | int SizesOK(void); |
| 208 | |
| 209 | #endif |