blob: d6e4f92cc8168159b2f7e0d5bd17efe26dd8f9e2 [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>
7
srs5694221e0872009-08-29 15:00:31 -04008#if defined (__FreeBSD__) || defined (__APPLE__)
srs5694e7b4ff92009-08-18 13:16:10 -04009// Darwin (Mac OS) only: disk IOCTLs are different, and there is no lseek64
srs5694da79b852009-08-18 13:29:54 -040010// This used to use __DARWIN_UNIX03 rather than __APPLE__, but __APPLE__
11// is more general. If the code fails to work on older versions of OS X/
12// Darwin, this may need to be changed back (and in various .cc files).
srs5694e7b4ff92009-08-18 13:16:10 -040013#include <sys/disk.h>
14#define lseek64 lseek
15#else
16
17// Linux only....
18#include <linux/fs.h>
19#endif
20
srs5694e35eb1b2009-09-14 00:29:34 -040021#include <string.h>
srs5694e7b4ff92009-08-18 13:16:10 -040022
23#ifndef __GPTSUPPORT
24#define __GPTSUPPORT
25
26// Set this as a default
27#define SECTOR_SIZE UINT32_C(512)
28
srs5694221e0872009-08-29 15:00:31 -040029// Signatures for Apple (APM) disks, multiplied by 0x100000000
30#define APM_SIGNATURE1 UINT64_C(0x00004D5000000000)
31#define APM_SIGNATURE2 UINT64_C(0x0000535400000000)
32
33/**************************
34 * Some GPT constants.... *
35 **************************/
36
37#define GPT_SIGNATURE UINT64_C(0x5452415020494645)
38
39// Number and size of GPT entries...
40#define NUM_GPT_ENTRIES 128
41#define GPT_SIZE 128
42#define HEADER_SIZE 92
43#define GPT_RESERVED 420
44#define NAME_SIZE 72
45
srs5694e7b4ff92009-08-18 13:16:10 -040046using namespace std;
47
48// a GUID
49struct GUIDData {
50 uint64_t data1;
51 uint64_t data2;
52}; // struct GUIDData
53
54int GetNumber(int low, int high, int def, const char prompt[]);
55char GetYN(void);
srs5694e4ac11e2009-08-31 10:13:04 -040056uint64_t GetSectorNum(uint64_t low, uint64_t high, uint64_t def, char prompt[]);
srs5694e7b4ff92009-08-18 13:16:10 -040057char* BytesToSI(uint64_t size, char theValue[]);
58int GetBlockSize(int fd);
59char* GUIDToStr(struct GUIDData theGUID, char* theString);
60GUIDData GetGUID(void);
srs56942a9f5da2009-08-26 00:48:01 -040061int IsLittleEndian(void); // Returns 1 if CPU is little-endian, 0 if it's big-endian
srs5694221e0872009-08-29 15:00:31 -040062void ReverseBytes(void* theValue, int numBytes); // Reverses byte-order of theValue
srs5694e7b4ff92009-08-18 13:16:10 -040063uint64_t PowerOf2(int value);
srs5694e4ac11e2009-08-31 10:13:04 -040064int OpenForWrite(char* deviceFilename);
srs5694e35eb1b2009-09-14 00:29:34 -040065void DiskSync(int fd); // resync disk caches to use new partitions
srs5694e7b4ff92009-08-18 13:16:10 -040066
67uint64_t disksize(int fd, int* err);
68
69#endif