| srs5694 | 221e087 | 2009-08-29 15:00:31 -0400 | [diff] [blame] | 1 | /* This program is copyright (c) 2009 by Roderick W. Smith. It is distributed |
| 2 | under the terms of the GNU GPL version 2, as detailed in the COPYING file. */ |
| 3 | |
| srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 4 | #include <stdint.h> |
| 5 | #include <unistd.h> |
| 6 | #include <stdlib.h> |
| srs5694 | 978041c | 2009-09-21 20:51:47 -0400 | [diff] [blame] | 7 | #include <string.h> |
| 8 | |
| 9 | #ifndef __GPTSUPPORT |
| 10 | #define __GPTSUPPORT |
| srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 11 | |
| srs5694 | 221e087 | 2009-08-29 15:00:31 -0400 | [diff] [blame] | 12 | #if defined (__FreeBSD__) || defined (__APPLE__) |
| srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 13 | // Darwin (Mac OS) only: disk IOCTLs are different, and there is no lseek64 |
| srs5694 | da79b85 | 2009-08-18 13:29:54 -0400 | [diff] [blame] | 14 | // This used to use __DARWIN_UNIX03 rather than __APPLE__, but __APPLE__ |
| 15 | // is more general. If the code fails to work on older versions of OS X/ |
| 16 | // Darwin, this may need to be changed back (and in various .cc files). |
| srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 17 | #include <sys/disk.h> |
| 18 | #define lseek64 lseek |
| srs5694 | 546a9c7 | 2010-01-26 16:00:26 -0500 | [diff] [blame^] | 19 | #endif |
| srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 20 | |
| 21 | // Linux only.... |
| srs5694 | 546a9c7 | 2010-01-26 16:00:26 -0500 | [diff] [blame^] | 22 | #ifdef __linux__ |
| srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 23 | #include <linux/fs.h> |
| 24 | #endif |
| 25 | |
| srs5694 | 546a9c7 | 2010-01-26 16:00:26 -0500 | [diff] [blame^] | 26 | /* #ifdef __FreeBSD__ |
| srs5694 | 978041c | 2009-09-21 20:51:47 -0400 | [diff] [blame] | 27 | #define fstat64 fstat |
| 28 | #define stat64 stat |
| srs5694 | 546a9c7 | 2010-01-26 16:00:26 -0500 | [diff] [blame^] | 29 | #endif */ |
| srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 30 | |
| 31 | // Set this as a default |
| 32 | #define SECTOR_SIZE UINT32_C(512) |
| 33 | |
| srs5694 | 221e087 | 2009-08-29 15:00:31 -0400 | [diff] [blame] | 34 | // Signatures for Apple (APM) disks, multiplied by 0x100000000 |
| 35 | #define APM_SIGNATURE1 UINT64_C(0x00004D5000000000) |
| 36 | #define APM_SIGNATURE2 UINT64_C(0x0000535400000000) |
| 37 | |
| 38 | /************************** |
| 39 | * Some GPT constants.... * |
| 40 | **************************/ |
| 41 | |
| 42 | #define GPT_SIGNATURE UINT64_C(0x5452415020494645) |
| 43 | |
| 44 | // Number and size of GPT entries... |
| 45 | #define NUM_GPT_ENTRIES 128 |
| 46 | #define GPT_SIZE 128 |
| srs5694 | 978041c | 2009-09-21 20:51:47 -0400 | [diff] [blame] | 47 | #define HEADER_SIZE UINT32_C(92) |
| srs5694 | 221e087 | 2009-08-29 15:00:31 -0400 | [diff] [blame] | 48 | #define GPT_RESERVED 420 |
| 49 | #define NAME_SIZE 72 |
| 50 | |
| srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 51 | using namespace std; |
| 52 | |
| 53 | // a GUID |
| 54 | struct GUIDData { |
| 55 | uint64_t data1; |
| 56 | uint64_t data2; |
| 57 | }; // struct GUIDData |
| 58 | |
| srs5694 | 546a9c7 | 2010-01-26 16:00:26 -0500 | [diff] [blame^] | 59 | static char theFile[255]; |
| 60 | |
| srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 61 | int GetNumber(int low, int high, int def, const char prompt[]); |
| 62 | char GetYN(void); |
| srs5694 | e4ac11e | 2009-08-31 10:13:04 -0400 | [diff] [blame] | 63 | uint64_t GetSectorNum(uint64_t low, uint64_t high, uint64_t def, char prompt[]); |
| srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 64 | char* BytesToSI(uint64_t size, char theValue[]); |
| srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 65 | char* GUIDToStr(struct GUIDData theGUID, char* theString); |
| 66 | GUIDData GetGUID(void); |
| srs5694 | 2a9f5da | 2009-08-26 00:48:01 -0400 | [diff] [blame] | 67 | int IsLittleEndian(void); // Returns 1 if CPU is little-endian, 0 if it's big-endian |
| srs5694 | 221e087 | 2009-08-29 15:00:31 -0400 | [diff] [blame] | 68 | void ReverseBytes(void* theValue, int numBytes); // Reverses byte-order of theValue |
| srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 69 | uint64_t PowerOf2(int value); |
| srs5694 | e7b4ff9 | 2009-08-18 13:16:10 -0400 | [diff] [blame] | 70 | |
| 71 | #endif |