1 // 2 // C++ Interface: gptpart 3 // 4 // Description: Class to implement a single GPT partition 5 // 6 // 7 // Author: Rod Smith <rodsmith@rodsbooks.com>, (C) 2009 8 // 9 // Copyright: See COPYING file that comes with this distribution 10 // 11 // 12 // This program is copyright (c) 2009 by Roderick W. Smith. It is distributed 13 // under the terms of the GNU GPL version 2, as detailed in the COPYING file. 14 15 #ifndef __GPTPART_H 16 #define __GPTPART_H 17 18 #include <stdint.h> 19 #include <string> 20 #include <sys/types.h> 21 #include "support.h" 22 #include "parttypes.h" 23 #include "guid.h" 24 #include "attributes.h" 25 26 using namespace std; 27 28 // Values returned by GPTPart::IsSizedForMBR() 29 #define MBR_SIZED_GOOD 0 /* Whole partition under 2^32 sectors */ 30 #define MBR_SIZED_IFFY 1 /* Partition starts under 2^32 & is less than 2^32, but ends over 2^32 */ 31 #define MBR_SIZED_BAD 2 /* Partition starts over 2^32, is bigger than 2^32, or otherwise bad */ 32 33 /**************************************** 34 * * 35 * GPTPart class and related structures * 36 * * 37 ****************************************/ 38 39 class GPTPart { 40 protected: 41 // Caution: The non-static data in GPTPart is precisely the right size 42 // to enable easy loading of the data directly from disk. If any 43 // non-static variables are added to the below, the data size will 44 // change and the program will stop working. This can be corrected by 45 // adjusting the data-load operation in GPTData::LoadMainTable() and 46 // GPTData::LoadSecondTableAsMain() and then removing the GPTPart 47 // size check in SizesOK() (in gpt.cc file). 48 PartType partitionType; 49 GUIDData uniqueGUID; 50 uint64_t firstLBA; 51 uint64_t lastLBA; 52 Attributes attributes; 53 uint16_t name[NAME_SIZE]; 54 public: 55 GPTPart(void); 56 ~GPTPart(void); 57 58 // Simple data retrieval: GetType(void)59 PartType & GetType(void) {return partitionType;} 60 uint16_t GetHexType(void) const; 61 string GetTypeName(void); 62 UnicodeString GetUTypeName(void); GetUniqueGUID(void)63 const GUIDData GetUniqueGUID(void) const {return uniqueGUID;} GetFirstLBA(void)64 uint64_t GetFirstLBA(void) const {return firstLBA;} GetLastLBA(void)65 uint64_t GetLastLBA(void) const {return lastLBA;} 66 uint64_t GetLengthLBA(void) const; GetAttributes(void)67 Attributes GetAttributes(void) {return attributes;} ShowAttributes(uint32_t partNum)68 void ShowAttributes(uint32_t partNum) {attributes.ShowAttributes(partNum);} 69 UnicodeString GetDescription(void); 70 int IsUsed(void); 71 int IsSizedForMBR(void); 72 73 // Simple data assignment: 74 void SetType(PartType t); SetType(uint16_t hex)75 void SetType(uint16_t hex) {partitionType = hex;} SetUniqueGUID(GUIDData u)76 void SetUniqueGUID(GUIDData u) {uniqueGUID = u;} RandomizeUniqueGUID(void)77 void RandomizeUniqueGUID(void) {uniqueGUID.Randomize();} SetFirstLBA(uint64_t f)78 void SetFirstLBA(uint64_t f) {firstLBA = f;} SetLastLBA(uint64_t l)79 void SetLastLBA(uint64_t l) {lastLBA = l;} SetAttributes(uint64_t a)80 void SetAttributes(uint64_t a) {attributes = a;} SetAttributes(void)81 void SetAttributes(void) {attributes.ChangeAttributes();} 82 void SetName(const string & theName); 83 #ifdef USE_UTF16 84 void SetName(const UnicodeString & theName); 85 #endif 86 void SetDefaultDescription(void); 87 88 // Additional functions 89 GPTPart & operator=(const GPTPart & orig); 90 bool operator<(const GPTPart &other) const; 91 void ShowSummary(int partNum, uint32_t blockSize); // display summary information (1-line) 92 void ShowDetails(uint32_t blockSize); // display detailed information (multi-line) 93 void BlankPartition(void); // empty partition of data 94 int DoTheyOverlap(const GPTPart & other); // returns 1 if there's overlap 95 void ReversePartBytes(void); // reverse byte order of all integer fields 96 97 // Functions requiring user interaction 98 void ChangeType(void); // Change the type code 99 }; // struct GPTPart 100 101 #endif 102