blob: 79e62a9fd7353950eec2af70888891efb6ad9eac [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>
srs5694fed16d02010-01-27 23:03:40 -05007#include <string>
srs5694978041c2009-09-21 20:51:47 -04008
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
srs5694fed16d02010-01-27 23:03:40 -050038// Maximum line length ignored on some input functions
39#define MAX_IGNORED 999
40
srs5694221e0872009-08-29 15:00:31 -040041/**************************
42 * Some GPT constants.... *
43 **************************/
44
45#define GPT_SIGNATURE UINT64_C(0x5452415020494645)
46
47// Number and size of GPT entries...
48#define NUM_GPT_ENTRIES 128
49#define GPT_SIZE 128
srs5694978041c2009-09-21 20:51:47 -040050#define HEADER_SIZE UINT32_C(92)
srs5694221e0872009-08-29 15:00:31 -040051#define GPT_RESERVED 420
52#define NAME_SIZE 72
53
srs5694e7b4ff92009-08-18 13:16:10 -040054using namespace std;
55
56// a GUID
57struct GUIDData {
58 uint64_t data1;
59 uint64_t data2;
60}; // struct GUIDData
61
srs5694546a9c72010-01-26 16:00:26 -050062static char theFile[255];
63
srs5694fed16d02010-01-27 23:03:40 -050064int GetNumber(int low, int high, int def, const string & prompt);
srs5694e7b4ff92009-08-18 13:16:10 -040065char GetYN(void);
srs5694fed16d02010-01-27 23:03:40 -050066uint64_t GetSectorNum(uint64_t low, uint64_t high, uint64_t def, const string & prompt);
67string BytesToSI(uint64_t size);
68string GUIDToStr(struct GUIDData theGUID);
srs5694e7b4ff92009-08-18 13:16:10 -040069GUIDData GetGUID(void);
srs56942a9f5da2009-08-26 00:48:01 -040070int IsLittleEndian(void); // Returns 1 if CPU is little-endian, 0 if it's big-endian
srs5694221e0872009-08-29 15:00:31 -040071void ReverseBytes(void* theValue, int numBytes); // Reverses byte-order of theValue
srs5694e7b4ff92009-08-18 13:16:10 -040072uint64_t PowerOf2(int value);
srs5694e7b4ff92009-08-18 13:16:10 -040073
74#endif