You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
969 lines
35 KiB
969 lines
35 KiB
/*
|
|
Copyright (C) 2010-2018 <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.
|
|
|
|
*/
|
|
|
|
/* This class implements an interactive text-mode interface atop the
|
|
GPTData class */
|
|
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
#include <stdint.h>
|
|
#include <limits.h>
|
|
#include <iostream>
|
|
#include <fstream>
|
|
#include <sstream>
|
|
#include <cstdio>
|
|
#include "attributes.h"
|
|
#include "gpttext.h"
|
|
#include "support.h"
|
|
|
|
using namespace std;
|
|
|
|
/********************************************
|
|
* *
|
|
* GPTDataText class and related structures *
|
|
* *
|
|
********************************************/
|
|
|
|
GPTDataTextUI::GPTDataTextUI(void) : GPTData() {
|
|
} // default constructor
|
|
|
|
GPTDataTextUI::GPTDataTextUI(string filename) : GPTData(filename) {
|
|
} // constructor passing filename
|
|
|
|
GPTDataTextUI::~GPTDataTextUI(void) {
|
|
} // default destructor
|
|
|
|
/*********************************************************************
|
|
* *
|
|
* The following functions are extended (interactive) versions of *
|
|
* simpler functions in the base class.... *
|
|
* *
|
|
*********************************************************************/
|
|
|
|
// Overridden function; calls base-class function and then makes
|
|
// additional queries of the user, if the base-class function can't
|
|
// decide what to do.
|
|
WhichToUse GPTDataTextUI::UseWhichPartitions(void) {
|
|
WhichToUse which;
|
|
MBRValidity mbrState;
|
|
int answer;
|
|
|
|
which = GPTData::UseWhichPartitions();
|
|
if ((which != use_abort) || beQuiet)
|
|
return which;
|
|
|
|
// If we get past here, it means that the non-interactive tests were
|
|
// inconclusive, so we must ask the user which table to use....
|
|
mbrState = protectiveMBR.GetValidity();
|
|
|
|
if ((state == gpt_valid) && (mbrState == mbr)) {
|
|
cout << "Found valid MBR and GPT. Which do you want to use?\n";
|
|
answer = GetNumber(1, 3, 2, " 1 - MBR\n 2 - GPT\n 3 - Create blank GPT\n\nYour answer: ");
|
|
if (answer == 1) {
|
|
which = use_mbr;
|
|
} else if (answer == 2) {
|
|
which = use_gpt;
|
|
cout << "Using GPT and creating fresh protective MBR.\n";
|
|
} else which = use_new;
|
|
} // if
|
|
|
|
// Nasty decisions here -- GPT is present, but corrupt (bad CRCs or other
|
|
// problems)
|
|
if (state == gpt_corrupt) {
|
|
if ((mbrState == mbr) || (mbrState == hybrid)) {
|
|
cout << "Found valid MBR and corrupt GPT. Which do you want to use? (Using the\n"
|
|
<< "GPT MAY permit recovery of GPT data.)\n";
|
|
answer = GetNumber(1, 3, 2, " 1 - MBR\n 2 - GPT\n 3 - Create blank GPT\n\nYour answer: ");
|
|
if (answer == 1) {
|
|
which = use_mbr;
|
|
} else if (answer == 2) {
|
|
which = use_gpt;
|
|
} else which = use_new;
|
|
} else if (mbrState == invalid) {
|
|
cout << "Found invalid MBR and corrupt GPT. What do you want to do? (Using the\n"
|
|
<< "GPT MAY permit recovery of GPT data.)\n";
|
|
answer = GetNumber(1, 2, 1, " 1 - Use current GPT\n 2 - Create blank GPT\n\nYour answer: ");
|
|
if (answer == 1) {
|
|
which = use_gpt;
|
|
} else which = use_new;
|
|
} // if/else/else
|
|
} // if (corrupt GPT)
|
|
|
|
return which;
|
|
} // UseWhichPartitions()
|
|
|
|
// Ask the user for a partition number; and prompt for verification
|
|
// if the requested partition isn't of a known BSD type.
|
|
// Lets the base-class function do the work, and returns its value (the
|
|
// number of converted partitions).
|
|
int GPTDataTextUI::XFormDisklabel(void) {
|
|
uint32_t partNum;
|
|
uint16_t hexCode;
|
|
int goOn = 1, numDone = 0;
|
|
BSDData disklabel;
|
|
|
|
partNum = GetPartNum();
|
|
|
|
// Now see if the specified partition has a BSD type code....
|
|
hexCode = partitions[partNum].GetHexType();
|
|
if ((hexCode != 0xa500) && (hexCode != 0xa900)) {
|
|
cout << "Specified partition doesn't have a disklabel partition type "
|
|
<< "code.\nContinue anyway? ";
|
|
goOn = (GetYN() == 'Y');
|
|
} // if
|
|
|
|
if (goOn)
|
|
numDone = GPTData::XFormDisklabel(partNum);
|
|
|
|
return numDone;
|
|
} // GPTDataTextUI::XFormDisklabel(void)
|
|
|
|
|
|
/*********************************************************************
|
|
* *
|
|
* Begin functions that obtain information from the users, and often *
|
|
* do something with that information (call other functions) *
|
|
* *
|
|
*********************************************************************/
|
|
|
|
// Prompts user for partition number and returns the result. Returns "0"
|
|
// (the first partition) if none are currently defined.
|
|
uint32_t GPTDataTextUI::GetPartNum(void) {
|
|
uint32_t partNum;
|
|
uint32_t low, high;
|
|
ostringstream prompt;
|
|
|
|
if (GetPartRange(&low, &high) > 0) {
|
|
prompt << "Partition number (" << low + 1 << "-" << high + 1 << "): ";
|
|
partNum = GetNumber(low + 1, high + 1, low, prompt.str());
|
|
} else partNum = 1;
|
|
return (partNum - 1);
|
|
} // GPTDataTextUI::GetPartNum()
|
|
|
|
// What it says: Resize the partition table. (Default is 128 entries.)
|
|
void GPTDataTextUI::ResizePartitionTable(void) {
|
|
int newSize;
|
|
ostringstream prompt;
|
|
uint32_t curLow, curHigh;
|
|
|
|
cout << "Current partition table size is " << numParts << ".\n";
|
|
GetPartRange(&curLow, &curHigh);
|
|
curHigh++; // since GetPartRange() returns numbers starting from 0...
|
|
// There's no point in having fewer than four partitions....
|
|
if (curHigh < (blockSize / GPT_SIZE))
|
|
curHigh = blockSize / GPT_SIZE;
|
|
prompt << "Enter new size (" << curHigh << " up, default " << NUM_GPT_ENTRIES << "): ";
|
|
newSize = GetNumber(4, 65535, 128, prompt.str());
|
|
if (newSize < 128) {
|
|
cout << "Caution: The partition table size should officially be 16KB or larger,\n"
|
|
<< "which works out to 128 entries. In practice, smaller tables seem to\n"
|
|
<< "work with most OSes, but this practice is risky. I'm proceeding with\n"
|
|
<< "the resize, but you may want to reconsider this action and undo it.\n\n";
|
|
} // if
|
|
SetGPTSize(newSize);
|
|
} // GPTDataTextUI::ResizePartitionTable()
|
|
|
|
// Move the main partition table (to enable some SoC boot loaders to place
|
|
// code at sector 2, for instance).
|
|
void GPTDataTextUI::MoveMainTable(void) {
|
|
uint64_t newStart, pteSize = GetTableSizeInSectors();
|
|
uint64_t maxValue = FindFirstUsedLBA() - pteSize;
|
|
ostringstream prompt;
|
|
|
|
cout << "Currently, main partition table begins at sector " << mainHeader.partitionEntriesLBA
|
|
<< " and ends at sector " << mainHeader.partitionEntriesLBA + pteSize - 1 << "\n";
|
|
prompt << "Enter new starting location (2 to " << maxValue << "; default is 2; 1 to abort): ";
|
|
newStart = GetNumber(1, maxValue, 2, prompt.str());
|
|
if (newStart != 1) {
|
|
GPTData::MoveMainTable(newStart);
|
|
} else {
|
|
cout << "Aborting change!\n";
|
|
} // if
|
|
} // GPTDataTextUI::MoveMainTable()
|
|
|
|
// Interactively create a partition
|
|
void GPTDataTextUI::CreatePartition(void) {
|
|
uint64_t firstBlock, firstInLargest, lastBlock, sector, origSector;
|
|
uint32_t firstFreePart = 0;
|
|
ostringstream prompt1, prompt2, prompt3;
|
|
int partNum;
|
|
|
|
// Find first free partition...
|
|
while (partitions[firstFreePart].GetFirstLBA() != 0) {
|
|
firstFreePart++;
|
|
} // while
|
|
|
|
if (((firstBlock = FindFirstAvailable()) != 0) &&
|
|
(firstFreePart < numParts)) {
|
|
lastBlock = FindLastAvailable();
|
|
firstInLargest = FindFirstInLargest();
|
|
Align(&firstInLargest);
|
|
|
|
// Get partition number....
|
|
prompt1 << "Partition number (" << firstFreePart + 1 << "-" << numParts
|
|
<< ", default " << firstFreePart + 1 << "): ";
|
|
do {
|
|
partNum = GetNumber(firstFreePart + 1, numParts,
|
|
firstFreePart + 1, prompt1.str()) - 1;
|
|
if (partitions[partNum].GetFirstLBA() != 0)
|
|
cout << "partition " << partNum + 1 << " is in use.\n";
|
|
} while (partitions[partNum].GetFirstLBA() != 0);
|
|
|
|
// Get first block for new partition...
|
|
prompt2 << "First sector (" << firstBlock << "-" << lastBlock << ", default = "
|
|
<< firstInLargest << ") or {+-}size{KMGTP}: ";
|
|
do {
|
|
sector = GetSectorNum(firstBlock, lastBlock, firstInLargest, blockSize, prompt2.str());
|
|
} while (IsFree(sector) == 0);
|
|
origSector = sector;
|
|
if (Align(§or)) {
|
|
cout << "Information: Moved requested sector from " << origSector << " to "
|
|
<< sector << " in\norder to align on " << sectorAlignment
|
|
<< "-sector boundaries.\n";
|
|
if (!beQuiet)
|
|
cout << "Use 'l' on the experts' menu to adjust alignment\n";
|
|
} // if
|
|
// Align(§or); // Align sector to correct multiple
|
|
firstBlock = sector;
|
|
|
|
// Get last block for new partitions...
|
|
lastBlock = FindLastInFree(firstBlock);
|
|
prompt3 << "Last sector (" << firstBlock << "-" << lastBlock << ", default = "
|
|
<< lastBlock << ") or {+-}size{KMGTP}: ";
|
|
do {
|
|
sector = GetSectorNum(firstBlock, lastBlock, lastBlock, blockSize, prompt3.str());
|
|
} while (IsFree(sector) == 0);
|
|
lastBlock = sector;
|
|
|
|
firstFreePart = GPTData::CreatePartition(partNum, firstBlock, lastBlock);
|
|
partitions[partNum].ChangeType();
|
|
partitions[partNum].SetDefaultDescription();
|
|
} else {
|
|
if (firstFreePart >= numParts)
|
|
cout << "No table partition entries left\n";
|
|
else
|
|
cout << "No free sectors available\n";
|
|
} // if/else
|
|
} // GPTDataTextUI::CreatePartition()
|
|
|
|
// Interactively delete a partition (duh!)
|
|
void GPTDataTextUI::DeletePartition(void) {
|
|
int partNum;
|
|
uint32_t low, high;
|
|
ostringstream prompt;
|
|
|
|
if (GetPartRange(&low, &high) > 0) {
|
|
prompt << "Partition number (" << low + 1 << "-" << high + 1 << "): ";
|
|
partNum = GetNumber(low + 1, high + 1, low, prompt.str());
|
|
GPTData::DeletePartition(partNum - 1);
|
|
} else {
|
|
cout << "No partitions\n";
|
|
} // if/else
|
|
} // GPTDataTextUI::DeletePartition()
|
|
|
|
// Prompt user for a partition number, then change its type code
|
|
void GPTDataTextUI::ChangePartType(void) {
|
|
int partNum;
|
|
uint32_t low, high;
|
|
|
|
if (GetPartRange(&low, &high) > 0) {
|
|
partNum = GetPartNum();
|
|
partitions[partNum].ChangeType();
|
|
} else {
|
|
cout << "No partitions\n";
|
|
} // if/else
|
|
} // GPTDataTextUI::ChangePartType()
|
|
|
|
// Prompt user for a partition number, then change its unique
|
|
// GUID.
|
|
void GPTDataTextUI::ChangeUniqueGuid(void) {
|
|
int partNum;
|
|
uint32_t low, high;
|
|
string guidStr;
|
|
|
|
if (GetPartRange(&low, &high) > 0) {
|
|
partNum = GetPartNum();
|
|
cout << "Enter the partition's new unique GUID ('R' to randomize): ";
|
|
guidStr = ReadString();
|
|
if ((guidStr.length() >= 32) || (guidStr[0] == 'R') || (guidStr[0] == 'r')) {
|
|
SetPartitionGUID(partNum, (GUIDData) guidStr);
|
|
cout << "New GUID is " << partitions[partNum].GetUniqueGUID() << "\n";
|
|
} else {
|
|
cout << "GUID is too short!\n";
|
|
} // if/else
|
|
} else
|
|
cout << "No partitions\n";
|
|
} // GPTDataTextUI::ChangeUniqueGuid()
|
|
|
|
// Partition attributes seem to be rarely used, but I want a way to
|
|
// adjust them for completeness....
|
|
void GPTDataTextUI::SetAttributes(uint32_t partNum) {
|
|
partitions[partNum].SetAttributes();
|
|
} // GPTDataTextUI::SetAttributes()
|
|
|
|
// Prompts the user for a partition name and sets the partition's
|
|
// name. Returns 1 on success, 0 on failure (invalid partition
|
|
// number). (Note that the function skips prompting when an
|
|
// invalid partition number is detected.)
|
|
int GPTDataTextUI::SetName(uint32_t partNum) {
|
|
UnicodeString theName = "";
|
|
int retval = 1;
|
|
|
|
if (IsUsedPartNum(partNum)) {
|
|
cout << "Enter name: ";
|
|
#ifdef USE_UTF16
|
|
theName = ReadUString();
|
|
#else
|
|
theName = ReadString();
|
|
#endif
|
|
partitions[partNum].SetName(theName);
|
|
} else {
|
|
cerr << "Invalid partition number (" << partNum << ")\n";
|
|
retval = 0;
|
|
} // if/else
|
|
|
|
return retval;
|
|
} // GPTDataTextUI::SetName()
|
|
|
|
// Ask user for two partition numbers and swap them in the table. Note that
|
|
// this just reorders table entries; it doesn't adjust partition layout on
|
|
// the disk.
|
|
// Returns 1 if successful, 0 if not. (If user enters identical numbers, it
|
|
// counts as successful.)
|
|
int GPTDataTextUI::SwapPartitions(void) {
|
|
int partNum1, partNum2, didIt = 0;
|
|
uint32_t low, high;
|
|
ostringstream prompt;
|
|
GPTPart temp;
|
|
|
|
if (GetPartRange(&low, &high) > 0) {
|
|
partNum1 = GetPartNum();
|
|
if (high >= numParts - 1)
|
|
high = 0;
|
|
prompt << "New partition number (1-" << numParts
|
|
<< ", default " << high + 2 << "): ";
|
|
partNum2 = GetNumber(1, numParts, high + 2, prompt.str()) - 1;
|
|
didIt = GPTData::SwapPartitions(partNum1, partNum2);
|
|
} else {
|
|
cout << "No partitions\n";
|
|
} // if/else
|
|
return didIt;
|
|
} // GPTDataTextUI::SwapPartitionNumbers()
|
|
|
|
// This function destroys the on-disk GPT structures. Returns 1 if the user
|
|
// confirms destruction, 0 if the user aborts or if there's a disk error.
|
|
int GPTDataTextUI::DestroyGPTwPrompt(void) {
|
|
int allOK = 1;
|
|
|
|
if ((apmFound) || (bsdFound)) {
|
|
cout << "WARNING: APM or BSD disklabel structures detected! This operation could\n"
|
|
<< "damage any APM or BSD partitions on this disk!\n";
|
|
} // if APM or BSD
|
|
cout << "\a\aAbout to wipe out GPT on " << device << ". Proceed? ";
|
|
if (GetYN() == 'Y') {
|
|
if (DestroyGPT()) {
|
|
// Note on below: Touch the MBR only if the user wants it completely
|
|
// blanked out. Version 0.4.2 deleted the 0xEE partition and re-wrote
|
|
// the MBR, but this could wipe out a valid MBR that the program
|
|
// had subsequently discarded (say, if it conflicted with older GPT
|
|
// structures).
|
|
cout << "Blank out MBR? ";
|
|
if (GetYN() == 'Y') {
|
|
DestroyMBR();
|
|
} else {
|
|
cout << "MBR is unchanged. You may need to delete an EFI GPT (0xEE) partition\n"
|
|
<< "with fdisk or another tool.\n";
|
|
} // if/else
|
|
} else allOK = 0; // if GPT structures destroyed
|
|
} else allOK = 0; // if user confirms destruction
|
|
return (allOK);
|
|
} // GPTDataTextUI::DestroyGPTwPrompt()
|
|
|
|
// Get partition number from user and then call ShowPartDetails(partNum)
|
|
// to show its detailed information
|
|
void GPTDataTextUI::ShowDetails(void) {
|
|
int partNum;
|
|
uint32_t low, high;
|
|
|
|
if (GetPartRange(&low, &high) > 0) {
|
|
partNum = GetPartNum();
|
|
ShowPartDetails(partNum);
|
|
} else {
|
|
cout << "No partitions\n";
|
|
} // if/else
|
|
} // GPTDataTextUI::ShowDetails()
|
|
|
|
// Create a hybrid MBR -- an ugly, funky thing that helps GPT work with
|
|
// OSes that don't understand GPT.
|
|
void GPTDataTextUI::MakeHybrid(void) {
|
|
uint32_t partNums[3] = {0, 0, 0};
|
|
string line;
|
|
int numPartsToCvt = 0, numConverted = 0, i, j, mbrNum = 0;
|
|
unsigned int hexCode = 0;
|
|
MBRPart hybridPart;
|
|
MBRData hybridMBR;
|
|
char eeFirst = 'Y'; // Whether EFI GPT (0xEE) partition comes first in table
|
|
|
|
cout << "\nWARNING! Hybrid MBRs are flaky and dangerous! If you decide not to use one,\n"
|
|
<< "just hit the Enter key at the below prompt and your MBR partition table will\n"
|
|
<< "be untouched.\n\n\a";
|
|
|
|
// Use a local MBR structure, copying from protectiveMBR to keep its
|
|
// boot loader code intact....
|
|
hybridMBR = protectiveMBR;
|
|
hybridMBR.EmptyMBR(0);
|
|
|
|
// Now get the numbers of up to three partitions to add to the
|
|
// hybrid MBR....
|
|
cout << "Type from one to three GPT partition numbers, separated by spaces, to be\n"
|
|
<< "added to the hybrid MBR, in sequence: ";
|
|
line = ReadString();
|
|
istringstream inLine(line);
|
|
do {
|
|
inLine >> partNums[numPartsToCvt];
|
|
if (partNums[numPartsToCvt] > 0)
|
|
numPartsToCvt++;
|
|
} while (!inLine.eof() && (numPartsToCvt < 3));
|
|
|
|
if (numPartsToCvt > 0) {
|
|
cout << "Place EFI GPT (0xEE) partition first in MBR (good for GRUB)? ";
|
|
eeFirst = GetYN();
|
|
} // if
|
|
|
|
for (i = 0; i < numPartsToCvt; i++) {
|
|
j = partNums[i] - 1;
|
|
if (partitions[j].IsUsed() && (partitions[j].IsSizedForMBR() != MBR_SIZED_BAD)) {
|
|
mbrNum = i + (eeFirst == 'Y');
|
|
cout << "\nCreating entry for GPT partition #" << j + 1
|
|
<< " (MBR partition #" << mbrNum + 1 << ")\n";
|
|
hybridPart.SetType(GetMBRTypeCode(partitions[j].GetHexType() / 256));
|
|
hybridPart.SetLocation(partitions[j].GetFirstLBA(), partitions[j].GetLengthLBA());
|
|
hybridPart.SetInclusion(PRIMARY);
|
|
cout << "Set the bootable flag? ";
|
|
if (GetYN() == 'Y')
|
|
hybridPart.SetStatus(0x80);
|
|
else
|
|
hybridPart.SetStatus(0x00);
|
|
hybridPart.SetInclusion(PRIMARY);
|
|
if (partitions[j].IsSizedForMBR() == MBR_SIZED_IFFY)
|
|
WarnAboutIffyMBRPart(j + 1);
|
|
numConverted++;
|
|
} else {
|
|
cerr << "\nGPT partition #" << j + 1 << " does not exist or is too big; skipping.\n";
|
|
} // if/else
|
|
hybridMBR.AddPart(mbrNum, hybridPart);
|
|
} // for
|
|
|
|
if (numConverted > 0) { // User opted to create a hybrid MBR....
|
|
// Create EFI protective partition that covers the start of the disk.
|
|
// If this location (covering the main GPT data structures) is omitted,
|
|
// Linux won't find any partitions on the disk.
|
|
hybridPart.SetLocation(1, hybridMBR.FindLastInFree(1));
|
|
hybridPart.SetStatus(0);
|
|
hybridPart.SetType(0xEE);
|
|
hybridPart.SetInclusion(PRIMARY);
|
|
// newNote firstLBA and lastLBA are computed later...
|
|
if (eeFirst == 'Y') {
|
|
hybridMBR.AddPart(0, hybridPart);
|
|
} else {
|
|
hybridMBR.AddPart(numConverted, hybridPart);
|
|
} // else
|
|
hybridMBR.SetHybrid();
|
|
|
|
// ... and for good measure, if there are any partition spaces left,
|
|
// optionally create another protective EFI partition to cover as much
|
|
// space as possible....
|
|
if (hybridMBR.CountParts() < 4) { // unused entry....
|
|
cout << "\nUnused partition space(s) found. Use one to protect more partitions? ";
|
|
if (GetYN() == 'Y') {
|
|
cout << "Note: Default is 0xEE, but this may confuse Mac OS X.\n";
|
|
// Comment on above: Mac OS treats disks with more than one
|
|
// 0xEE MBR partition as MBR disks, not as GPT disks.
|
|
hexCode = GetMBRTypeCode(0xEE);
|
|
hybridMBR.MakeBiggestPart(3, hexCode);
|
|
} // if (GetYN() == 'Y')
|
|
} // if unused entry
|
|
protectiveMBR = hybridMBR;
|
|
} else {
|
|
cout << "\nNo partitions converted; original protective/hybrid MBR is unmodified!\n";
|
|
} // if/else (numConverted > 0)
|
|
} // GPTDataTextUI::MakeHybrid()
|
|
|
|
// Convert the GPT to MBR form, storing partitions in the protectiveMBR
|
|
// variable. This function is necessarily limited; it may not be able to
|
|
// convert all partitions, depending on the disk size and available space
|
|
// before each partition (one free sector is required to create a logical
|
|
// partition, which are necessary to convert more than four partitions).
|
|
// Returns the number of converted partitions; if this value
|
|
// is over 0, the calling function should call DestroyGPT() to destroy
|
|
// the GPT data, call SaveMBR() to save the MBR, and then exit.
|
|
int GPTDataTextUI::XFormToMBR(void) {
|
|
uint32_t i;
|
|
|
|
protectiveMBR.EmptyMBR(0);
|
|
for (i = 0; i < numParts; i++) {
|
|
if (partitions[i].IsUsed()) {
|
|
if (partitions[i].IsSizedForMBR() == MBR_SIZED_IFFY)
|
|
WarnAboutIffyMBRPart(i + 1);
|
|
// Note: MakePart() checks for oversized partitions, so don't
|
|
// bother checking other IsSizedForMBR() return values....
|
|
protectiveMBR.MakePart(i, partitions[i].GetFirstLBA(),
|
|
partitions[i].GetLengthLBA(),
|
|
partitions[i].GetHexType() / 0x0100, 0);
|
|
} // if
|
|
} // for
|
|
protectiveMBR.MakeItLegal();
|
|
return protectiveMBR.DoMenu();
|
|
} // GPTDataTextUI::XFormToMBR()
|
|
|
|
/******************************************************
|
|
* *
|
|
* Display informational messages for the user.... *
|
|
* *
|
|
******************************************************/
|
|
|
|
// Although an MBR partition that begins below sector 2^32 and is less than 2^32 sectors in
|
|
// length is technically legal even if it ends above the 2^32-sector mark, such a partition
|
|
// tends to confuse a lot of OSes, so warn the user about such partitions. This function is
|
|
// called by XFormToMBR() and MakeHybrid(); it's a separate function just to consolidate the
|
|
// lengthy message in one place.
|
|
void GPTDataTextUI::WarnAboutIffyMBRPart(int partNum) {
|
|
cout << "\a\nWarning! GPT partition " << partNum << " ends after the 2^32 sector mark! The partition\n"
|
|
<< "begins before this point, and is smaller than 2^32 sectors. This is technically\n"
|
|
<< "legal, but will confuse some OSes. The partition IS being added to the MBR, but\n"
|
|
<< "if your OS misbehaves or can't see the partition, the partition may simply be\n"
|
|
<< "unusable in that OS and may need to be resized or omitted from the MBR.\n\n";
|
|
} // GPTDataTextUI::WarnAboutIffyMBRPart()
|
|
|
|
/*********************************************************************
|
|
* *
|
|
* The following functions provide the main menus for the gdisk *
|
|
* program.... *
|
|
* *
|
|
*********************************************************************/
|
|
|
|
// Accept a command and execute it. Returns only when the user
|
|
// wants to exit (such as after a 'w' or 'q' command).
|
|
void GPTDataTextUI::MainMenu(string filename) {
|
|
int goOn = 1;
|
|
PartType typeHelper;
|
|
uint32_t temp1, temp2;
|
|
|
|
do {
|
|
cout << "\nCommand (? for help): ";
|
|
switch (ReadString()[0]) {
|
|
case '\0':
|
|
goOn = cin.good();
|
|
break;
|
|
case 'b': case 'B':
|
|
cout << "Enter backup filename to save: ";
|
|
SaveGPTBackup(ReadString());
|
|
break;
|
|
case 'c': case 'C':
|
|
if (GetPartRange(&temp1, &temp2) > 0)
|
|
SetName(GetPartNum());
|
|
else
|
|
cout << "No partitions\n";
|
|
break;
|
|
case 'd': case 'D':
|
|
DeletePartition();
|
|
break;
|
|
case 'i': case 'I':
|
|
ShowDetails();
|
|
break;
|
|
case 'l': case 'L':
|
|
typeHelper.ShowAllTypes();
|
|
break;
|
|
case 'n': case 'N':
|
|
CreatePartition();
|
|
break;
|
|
case 'o': case 'O':
|
|
cout << "This option deletes all partitions and creates a new protective MBR.\n"
|
|
<< "Proceed? ";
|
|
if (GetYN() == 'Y') {
|
|
ClearGPTData();
|
|
MakeProtectiveMBR();
|
|
} // if
|
|
break;
|
|
case 'p': case 'P':
|
|
DisplayGPTData();
|
|
break;
|
|
case 'q': case 'Q':
|
|
goOn = 0;
|
|
break;
|
|
case 'r': case 'R':
|
|
RecoveryMenu(filename);
|
|
goOn = 0;
|
|
break;
|
|
case 's': case 'S':
|
|
SortGPT();
|
|
cout << "You may need to edit /etc/fstab and/or your boot loader configuration!\n";
|
|
break;
|
|
case 't': case 'T':
|
|
ChangePartType();
|
|
break;
|
|
case 'v': case 'V':
|
|
Verify();
|
|
break;
|
|
case 'w': case 'W':
|
|
if (SaveGPTData() == 1)
|
|
goOn = 0;
|
|
break;
|
|
case 'x': case 'X':
|
|
ExpertsMenu(filename);
|
|
goOn = 0;
|
|
break;
|
|
default:
|
|
ShowCommands();
|
|
break;
|
|
} // switch
|
|
} while (goOn);
|
|
} // GPTDataTextUI::MainMenu()
|
|
|
|
void GPTDataTextUI::ShowCommands(void) {
|
|
cout << "b\tback up GPT data to a file\n";
|
|
cout << "c\tchange a partition's name\n";
|
|
cout << "d\tdelete a partition\n";
|
|
cout << "i\tshow detailed information on a partition\n";
|
|
cout << "l\tlist known partition types\n";
|
|
cout << "n\tadd a new partition\n";
|
|
cout << "o\tcreate a new empty GUID partition table (GPT)\n";
|
|
cout << "p\tprint the partition table\n";
|
|
cout << "q\tquit without saving changes\n";
|
|
cout << "r\trecovery and transformation options (experts only)\n";
|
|
cout << "s\tsort partitions\n";
|
|
cout << "t\tchange a partition's type code\n";
|
|
cout << "v\tverify disk\n";
|
|
cout << "w\twrite table to disk and exit\n";
|
|
cout << "x\textra functionality (experts only)\n";
|
|
cout << "?\tprint this menu\n";
|
|
} // GPTDataTextUI::ShowCommands()
|
|
|
|
// Accept a recovery & transformation menu command. Returns only when the user
|
|
// issues an exit command, such as 'w' or 'q'.
|
|
void GPTDataTextUI::RecoveryMenu(string filename) {
|
|
uint32_t numParts;
|
|
int goOn = 1, temp1;
|
|
|
|
do {
|
|
cout << "\nRecovery/transformation command (? for help): ";
|
|
switch (ReadString()[0]) {
|
|
case '\0':
|
|
goOn = cin.good();
|
|
break;
|
|
case 'b': case 'B':
|
|
RebuildMainHeader();
|
|
break;
|
|
case 'c': case 'C':
|
|
cout << "Warning! This will probably do weird things if you've converted an MBR to\n"
|
|
<< "GPT form and haven't yet saved the GPT! Proceed? ";
|
|
if (GetYN() == 'Y')
|
|
LoadSecondTableAsMain();
|
|
break;
|
|
case 'd': case 'D':
|
|
RebuildSecondHeader();
|
|
break;
|
|
case 'e': case 'E':
|
|
cout << "Warning! This will probably do weird things if you've converted an MBR to\n"
|
|
<< "GPT form and haven't yet saved the GPT! Proceed? ";
|
|
if (GetYN() == 'Y')
|
|
LoadMainTable();
|
|
break;
|
|
case 'f': case 'F':
|
|
cout << "Warning! This will destroy the currently defined partitions! Proceed? ";
|
|
if (GetYN() == 'Y') {
|
|
if (LoadMBR(filename) == 1) { // successful load
|
|
XFormPartitions();
|
|
} else {
|
|
cout << "Problem loading MBR! GPT is untouched; regenerating protective MBR!\n";
|
|
MakeProtectiveMBR();
|
|
} // if/else
|
|
} // if
|
|
break;
|
|
case 'g': case 'G':
|
|
numParts = GetNumParts();
|
|
temp1 = XFormToMBR();
|
|
if (temp1 > 0)
|
|
cout << "\nConverted " << temp1 << " partitions. Finalize and exit? ";
|
|
if ((temp1 > 0) && (GetYN() == 'Y')) {
|
|
if ((DestroyGPT() > 0) && (SaveMBR())) {
|
|
goOn = 0;
|
|
} // if
|
|
} else {
|
|
MakeProtectiveMBR();
|
|
SetGPTSize(numParts, 0);
|
|
cout << "Note: New protective MBR created\n\n";
|
|
} // if/else
|
|
break;
|
|
case 'h': case 'H':
|
|
MakeHybrid();
|
|
break;
|
|
case 'i': case 'I':
|
|
ShowDetails();
|
|
break;
|
|
case 'l': case 'L':
|
|
cout << "Enter backup filename to load: ";
|
|
LoadGPTBackup(ReadString());
|
|
break;
|
|
case 'm': case 'M':
|
|
MainMenu(filename);
|
|
goOn = 0;
|
|
break;
|
|
case 'o': case 'O':
|
|
DisplayMBRData();
|
|
break;
|
|
case 'p': case 'P':
|
|
DisplayGPTData();
|
|
break;
|
|
case 'q': case 'Q':
|
|
goOn = 0;
|
|
break;
|
|
case 't': case 'T':
|
|
XFormDisklabel();
|
|
break;
|
|
case 'v': case 'V':
|
|
Verify();
|
|
break;
|
|
case 'w': case 'W':
|
|
if (SaveGPTData() == 1) {
|
|
goOn = 0;
|
|
} // if
|
|
break;
|
|
case 'x': case 'X':
|
|
ExpertsMenu(filename);
|
|
goOn = 0;
|
|
break;
|
|
default:
|
|
ShowRecoveryCommands();
|
|
break;
|
|
} // switch
|
|
} while (goOn);
|
|
} // GPTDataTextUI::RecoveryMenu()
|
|
|
|
void GPTDataTextUI::ShowRecoveryCommands(void) {
|
|
cout << "b\tuse backup GPT header (rebuilding main)\n";
|
|
cout << "c\tload backup partition table from disk (rebuilding main)\n";
|
|
cout << "d\tuse main GPT header (rebuilding backup)\n";
|
|
cout << "e\tload main partition table from disk (rebuilding backup)\n";
|
|
cout << "f\tload MBR and build fresh GPT from it\n";
|
|
cout << "g\tconvert GPT into MBR and exit\n";
|
|
cout << "h\tmake hybrid MBR\n";
|
|
cout << "i\tshow detailed information on a partition\n";
|
|
cout << "l\tload partition data from a backup file\n";
|
|
cout << "m\treturn to main menu\n";
|
|
cout << "o\tprint protective MBR data\n";
|
|
cout << "p\tprint the partition table\n";
|
|
cout << "q\tquit without saving changes\n";
|
|
cout << "t\ttransform BSD disklabel partition\n";
|
|
cout << "v\tverify disk\n";
|
|
cout << "w\twrite table to disk and exit\n";
|
|
cout << "x\textra functionality (experts only)\n";
|
|
cout << "?\tprint this menu\n";
|
|
} // GPTDataTextUI::ShowRecoveryCommands()
|
|
|
|
// Accept an experts' menu command. Returns only after the user
|
|
// selects an exit command, such as 'w' or 'q'.
|
|
void GPTDataTextUI::ExpertsMenu(string filename) {
|
|
GPTData secondDevice;
|
|
uint32_t temp1, temp2;
|
|
int goOn = 1;
|
|
string guidStr, device;
|
|
GUIDData aGUID;
|
|
ostringstream prompt;
|
|
|
|
do {
|
|
cout << "\nExpert command (? for help): ";
|
|
switch (ReadString()[0]) {
|
|
case '\0':
|
|
goOn = cin.good();
|
|
break;
|
|
case 'a': case 'A':
|
|
if (GetPartRange(&temp1, &temp2) > 0)
|
|
SetAttributes(GetPartNum());
|
|
else
|
|
cout << "No partitions\n";
|
|
break;
|
|
case 'c': case 'C':
|
|
ChangeUniqueGuid();
|
|
break;
|
|
case 'd': case 'D':
|
|
cout << "Partitions will begin on " << GetAlignment()
|
|
<< "-sector boundaries.\n";
|
|
break;
|
|
case 'e': case 'E':
|
|
cout << "Relocating backup data structures to the end of the disk\n";
|
|
MoveSecondHeaderToEnd();
|
|
break;
|
|
case 'f': case 'F':
|
|
RandomizeGUIDs();
|
|
break;
|
|
case 'g': case 'G':
|
|
cout << "Enter the disk's unique GUID ('R' to randomize): ";
|
|
guidStr = ReadString();
|
|
if ((guidStr.length() >= 32) || (guidStr[0] == 'R') || (guidStr[0] == 'r')) {
|
|
SetDiskGUID((GUIDData) guidStr);
|
|
cout << "The new disk GUID is " << GetDiskGUID() << "\n";
|
|
} else {
|
|
cout << "GUID is too short!\n";
|
|
} // if/else
|
|
break;
|
|
case 'h': case 'H':
|
|
RecomputeCHS();
|
|
break;
|
|
case 'i': case 'I':
|
|
ShowDetails();
|
|
break;
|
|
case 'j': case 'J':
|
|
MoveMainTable();
|
|
break;
|
|
case 'l': case 'L':
|
|
prompt.seekp(0);
|
|
prompt << "Enter the sector alignment value (1-" << MAX_ALIGNMENT << ", default = "
|
|
<< DEFAULT_ALIGNMENT << "): ";
|
|
temp1 = GetNumber(1, MAX_ALIGNMENT, DEFAULT_ALIGNMENT, prompt.str());
|
|
SetAlignment(temp1);
|
|
break;
|
|
case 'm': case 'M':
|
|
MainMenu(filename);
|
|
goOn = 0;
|
|
break;
|
|
case 'n': case 'N':
|
|
MakeProtectiveMBR();
|
|
break;
|
|
case 'o': case 'O':
|
|
DisplayMBRData();
|
|
break;
|
|
case 'p': case 'P':
|
|
DisplayGPTData();
|
|
break;
|
|
case 'q': case 'Q':
|
|
goOn = 0;
|
|
break;
|
|
case 'r': case 'R':
|
|
RecoveryMenu(filename);
|
|
goOn = 0;
|
|
break;
|
|
case 's': case 'S':
|
|
ResizePartitionTable();
|
|
break;
|
|
case 't': case 'T':
|
|
SwapPartitions();
|
|
break;
|
|
case 'u': case 'U':
|
|
cout << "Type device filename, or press <Enter> to exit: ";
|
|
device = ReadString();
|
|
if (device.length() > 0) {
|
|
secondDevice = *this;
|
|
secondDevice.SetDisk(device);
|
|
secondDevice.SaveGPTData(0);
|
|
} // if
|
|
break;
|
|
case 'v': case 'V':
|
|
Verify();
|
|
break;
|
|
case 'w': case 'W':
|
|
if (SaveGPTData() == 1) {
|
|
goOn = 0;
|
|
} // if
|
|
break;
|
|
case 'z': case 'Z':
|
|
if (DestroyGPTwPrompt() == 1) {
|
|
goOn = 0;
|
|
}
|
|
break;
|
|
default:
|
|
ShowExpertCommands();
|
|
break;
|
|
} // switch
|
|
} while (goOn);
|
|
} // GPTDataTextUI::ExpertsMenu()
|
|
|
|
void GPTDataTextUI::ShowExpertCommands(void) {
|
|
cout << "a\tset attributes\n";
|
|
cout << "c\tchange partition GUID\n";
|
|
cout << "d\tdisplay the sector alignment value\n";
|
|
cout << "e\trelocate backup data structures to the end of the disk\n";
|
|
cout << "f\trandomize disk and partition unique GUIDs\n";
|
|
cout << "g\tchange disk GUID\n";
|
|
cout << "h\trecompute CHS values in protective/hybrid MBR\n";
|
|
cout << "i\tshow detailed information on a partition\n";
|
|
cout << "j\tmove the main partition table\n";
|
|
cout << "l\tset the sector alignment value\n";
|
|
cout << "m\treturn to main menu\n";
|
|
cout << "n\tcreate a new protective MBR\n";
|
|
cout << "o\tprint protective MBR data\n";
|
|
cout << "p\tprint the partition table\n";
|
|
cout << "q\tquit without saving changes\n";
|
|
cout << "r\trecovery and transformation options (experts only)\n";
|
|
cout << "s\tresize partition table\n";
|
|
cout << "t\ttranspose two partition table entries\n";
|
|
cout << "u\treplicate partition table on new device\n";
|
|
cout << "v\tverify disk\n";
|
|
cout << "w\twrite table to disk and exit\n";
|
|
cout << "z\tzap (destroy) GPT data structures and exit\n";
|
|
cout << "?\tprint this menu\n";
|
|
} // GPTDataTextUI::ShowExpertCommands()
|
|
|
|
|
|
|
|
/********************************
|
|
* *
|
|
* Non-class support functions. *
|
|
* *
|
|
********************************/
|
|
|
|
// GetMBRTypeCode() doesn't really belong in the class, since it's MBR-
|
|
// specific, but it's also user I/O-related, so I want to keep it in
|
|
// this file....
|
|
|
|
// Get an MBR type code from the user and return it
|
|
int GetMBRTypeCode(int defType) {
|
|
string line;
|
|
int typeCode;
|
|
|
|
cout.setf(ios::uppercase);
|
|
cout.fill('0');
|
|
do {
|
|
cout << "Enter an MBR hex code (default " << hex;
|
|
cout.width(2);
|
|
cout << defType << "): " << dec;
|
|
line = ReadString();
|
|
if (line[0] == '\0')
|
|
typeCode = defType;
|
|
else
|
|
typeCode = StrToHex(line, 0);
|
|
} while ((typeCode <= 0) || (typeCode > 255));
|
|
cout.fill(' ');
|
|
return typeCode;
|
|
} // GetMBRTypeCode
|
|
|
|
#ifdef USE_UTF16
|
|
// Note: ReadUString() is here rather than in support.cc so that the ICU
|
|
// libraries need not be linked to fixparts.
|
|
|
|
// Reads a Unicode string from stdin, returning it as an ICU-style string.
|
|
// Note that the returned string will NOT include the carriage return
|
|
// entered by the user. Relies on the ICU constructor from a string
|
|
// encoded in the current codepage to work.
|
|
UnicodeString ReadUString(void) {
|
|
return ReadString().c_str();
|
|
} // ReadUString()
|
|
#endif
|
|
|