blob: 0de365fed140434cbef4cadb5a9a2d5325a94976 [file] [log] [blame]
/*
MBRPart class, part of GPT fdisk program family.
Copyright (C) 2011 Roderick W. Smith
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License along
with this program; if not, write to the Free Software Foundation, Inc.,
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
#ifndef MBRPART_H
#define MBRPART_H
#include <stdint.h>
#define MAX_HEADS 255 /* numbered 0 - 254 */
#define MAX_SECSPERTRACK 63 /* numbered 1 - 63 */
#define MAX_CYLINDERS 1024 /* numbered 0 - 1023 */
#define NONE 0 /* don't include partition when writing */
#define PRIMARY 1 /* write partition as primary */
#define LOGICAL 2 /* write partition as logical */
#define EBR 4 /* sector is used as an EBR or MBR */
#define INVALID 8 /* sector number is too large for disk */
using namespace std;
// Data for a single MBR partition record
// Note that firstSector and lastSector are in CHS addressing, which
// splits the bits up in a weird way.
// On read or write of MBR entries, firstLBA is an absolute disk sector.
// On read of logical entries, it's relative to the EBR record for that
// partition. When writing EBR records, it's relative to the extended
// partition's start.
#pragma pack(1)
struct MBRRecord {
uint8_t status;
uint8_t firstSector[3];
uint8_t partitionType;
uint8_t lastSector[3];
uint32_t firstLBA; // see above
uint32_t lengthLBA;
}; // struct MBRRecord
#pragma pack ()
class MBRPart {
protected:
uint8_t status;
uint8_t firstSector[3];
uint8_t partitionType;
uint8_t lastSector[3];
uint32_t firstLBA; // see above
uint32_t lengthLBA;
int includeAs; // PRIMARY, LOGICAL, or NONE
int canBeLogical;
int canBePrimary;
static uint32_t numHeads;
static uint32_t numSecspTrack;
static uint64_t diskSize;
static uint32_t blockSize;
static int numInstances;
public:
MBRPart();
MBRPart(const MBRPart& other);
virtual ~MBRPart();
virtual MBRPart& operator=(const MBRPart& orig);
virtual MBRPart& operator=(const struct MBRRecord& orig);
bool operator<(const MBRPart &other) const;
// Set information on partitions or disks...
void SetGeometry(uint32_t heads, uint32_t sectors, uint64_t ds, uint32_t bs);
void Empty(void);
void SetStartLBA(uint64_t s);
void SetLengthLBA(uint64_t l);
void SetLocation(uint64_t start, uint64_t length);
int SetType(uint8_t typeCode, int isExtended = 0);
void SetStatus(uint8_t s) {status = s;}
void SetInclusion(int status = PRIMARY) {includeAs = status;}
void SetCanBeLogical(int c) {canBeLogical = c;}
void SetCanBePrimary(int c) {canBePrimary = c;}
void StoreInStruct(struct MBRRecord *theStruct);
// Get information on partitions or disk....
uint8_t GetType(void) {return partitionType;}
uint8_t GetStatus(void) {return status;}
uint64_t GetStartLBA(void) {return firstLBA;}
uint64_t GetLengthLBA(void) {return lengthLBA;}
uint64_t GetLastLBA(void) const;
uint8_t GetInclusion(void) {return includeAs;}
int CanBeLogical(void) {return canBeLogical;}
int CanBePrimary(void) {return canBePrimary;}
int DoTheyOverlap (const MBRPart& other);
// Adjust information on partitions or disks...
int RecomputeCHS(void);
int LBAtoCHS(uint32_t lba, uint8_t * chs);
void ReverseByteOrder(void);
// User I/O...
void ShowData(int isGpt);
}; // MBRPart
#endif // MBRPART_H