blob: 3a102aec6e0d6c7ef954adca6a308e5debf3c1b9 [file] [log] [blame]
srs5694221e0872009-08-29 15:00:31 -04001/* 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
srs5694e7b4ff92009-08-18 13:16:10 -04004#include <stdint.h>
5#include <unistd.h>
6#include <stdlib.h>
srs5694978041c2009-09-21 20:51:47 -04007#include <string.h>
8
9#ifndef __GPTSUPPORT
10#define __GPTSUPPORT
srs5694e7b4ff92009-08-18 13:16:10 -040011
srs5694221e0872009-08-29 15:00:31 -040012#if defined (__FreeBSD__) || defined (__APPLE__)
srs5694e7b4ff92009-08-18 13:16:10 -040013// Darwin (Mac OS) only: disk IOCTLs are different, and there is no lseek64
srs5694da79b852009-08-18 13:29:54 -040014// 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).
srs5694e7b4ff92009-08-18 13:16:10 -040017#include <sys/disk.h>
18#define lseek64 lseek
srs5694546a9c72010-01-26 16:00:26 -050019#endif
srs5694e7b4ff92009-08-18 13:16:10 -040020
21// Linux only....
srs5694546a9c72010-01-26 16:00:26 -050022#ifdef __linux__
srs5694e7b4ff92009-08-18 13:16:10 -040023#include <linux/fs.h>
24#endif
25
srs5694546a9c72010-01-26 16:00:26 -050026/* #ifdef __FreeBSD__
srs5694978041c2009-09-21 20:51:47 -040027#define fstat64 fstat
28#define stat64 stat
srs5694546a9c72010-01-26 16:00:26 -050029#endif */
srs5694e7b4ff92009-08-18 13:16:10 -040030
31// Set this as a default
32#define SECTOR_SIZE UINT32_C(512)
33
srs5694221e0872009-08-29 15:00:31 -040034// 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
srs5694978041c2009-09-21 20:51:47 -040047#define HEADER_SIZE UINT32_C(92)
srs5694221e0872009-08-29 15:00:31 -040048#define GPT_RESERVED 420
49#define NAME_SIZE 72
50
srs5694e7b4ff92009-08-18 13:16:10 -040051using namespace std;
52
53// a GUID
54struct GUIDData {
55 uint64_t data1;
56 uint64_t data2;
57}; // struct GUIDData
58
srs5694546a9c72010-01-26 16:00:26 -050059static char theFile[255];
60
srs5694e7b4ff92009-08-18 13:16:10 -040061int GetNumber(int low, int high, int def, const char prompt[]);
62char GetYN(void);
srs5694e4ac11e2009-08-31 10:13:04 -040063uint64_t GetSectorNum(uint64_t low, uint64_t high, uint64_t def, char prompt[]);
srs5694e7b4ff92009-08-18 13:16:10 -040064char* BytesToSI(uint64_t size, char theValue[]);
srs5694e7b4ff92009-08-18 13:16:10 -040065char* GUIDToStr(struct GUIDData theGUID, char* theString);
66GUIDData GetGUID(void);
srs56942a9f5da2009-08-26 00:48:01 -040067int IsLittleEndian(void); // Returns 1 if CPU is little-endian, 0 if it's big-endian
srs5694221e0872009-08-29 15:00:31 -040068void ReverseBytes(void* theValue, int numBytes); // Reverses byte-order of theValue
srs5694e7b4ff92009-08-18 13:16:10 -040069uint64_t PowerOf2(int value);
srs5694e7b4ff92009-08-18 13:16:10 -040070
71#endif