fixparts.cc

   1 // fixparts
   2 // Program to fix certain types of damaged Master Boot Record (MBR) partition
   3 // tables
   4 //
   5 // Copyright 2011 by Roderick W. Smith
   6 //
   7 // This program is distributed under the terms of the GNU GPL, as described
   8 // in the COPYING file.
   9 //
  10 // Based on C++ classes originally created for GPT fdisk (gdisk and sgdisk)
  11 // programs
  12
  13 #include <stdio.h>
  14 #include <string.h>
  15 #include <string>
  16 #include <iostream>
  17 #include <sstream>
  18 #include "basicmbr.h"
  19 #include "support.h"
  20
  21 using namespace std;
  22
  23 void DoMBR(BasicMBRData & mbrTable);
  24
  25 int main(int argc, char* argv[]) {
  26    BasicMBRData mbrTable;
  27    string device;
  28
  29    cout << "FixParts " << GPTFDISK_VERSION << "\n";
  30
  31    switch (argc) {
  32       case 1:
  33          cout << "Type device filename, or press <Enter> to exit: ";
  34          device = ReadString();
  35          if (device.length() == 0)
  36             exit(0);
  37          break;
  38       case 2:
  39          device = argv[1];
  40          break;
  41       default:
  42          cerr << "Usage: " << argv[0] << " device_filename\n";
  43          exit(1);
  44    } // switch
  45
  46    cout << "\nLoading MBR data from " << device << "\n";
  47    if (!mbrTable.ReadMBRData(device)) {
  48       cerr << "\nUnable to read MBR data from '" << device << "'! Exiting!\n\n";
  49       exit(1);
  50    } // if
  51
  52    // This switch() statement weeds out disks with GPT signatures and non-MBR
  53    // disks so we don't accidentally damage them....
  54    switch(mbrTable.GetValidity()) {
  55       case hybrid: case gpt:
  56          cerr << "\nThis disk appears to be a GPT disk. Use GNU Parted or GPT fdisk on it!\n";
  57          cerr << "Exiting!\n\n";
  58          exit(1);
  59          break;
  60       case invalid:
  61          cerr << "\nCannot find valid MBR data on '" << device << "'! Exiting!\n\n";
  62          exit(1);
  63          break;
  64       case mbr:
  65          DoMBR(mbrTable);
  66          break;
  67       default:
  68          cerr << "\nCannot determine the validity of the disk on '" << device
  69               << "'! Exiting!\n\n";
  70          exit(1);
  71          break;
  72    } // switch()
  73    return 0;
  74 } // main()
  75
  76 // Do the bulk of the processing on actual MBR disks. First checks for old
  77 // GPT data (note this is different from the earlier check; this one only
  78 // looks for the GPT signatures in the main and backup GPT area, not for
  79 // a protective partition in the MBR, which we know is NOT present, since
  80 // if it were, this function would NOT be called!) and offers to destroy
  81 // it, if found; then makes sure the partitions are in a consistent and
  82 // legal state; then presents the MBR menu and, if it returns a "1" value
  83 // (meaning the user opted to write changes), writes the table to disk.
  84 void DoMBR(BasicMBRData & mbrTable) {
  85    int doItAgain;
  86
  87    if (mbrTable.CheckForGPT() > 0) {
  88       cout << "\nNOTICE: GPT signatures detected on the disk, but no 0xEE protective "
  89            << "partition!\nThe GPT signatures are probably left over from a previous "
  90            << "partition table.\nDo you want to delete them (if you answer 'Y', this "
  91            << "will happen\nimmediately)? ";
  92       if (GetYN() == 'Y') {
  93          cout << "Erasing GPT data!\n";
  94          if (mbrTable.BlankGPTData() != 1)
  95             cerr << "GPT signature erasure failed!\n";
  96       } // if
  97    } // if
  98
  99    mbrTable.MakeItLegal();
 100    do {
 101       doItAgain = 0;
 102       if (mbrTable.DoMenu() > 0) {
 103          cout << "\nFinal checks complete. About to write MBR data. THIS WILL OVERWRITE "
 104               << "EXISTING\nPARTITIONS!!\n\nDo you want to proceed? ";
 105          if (GetYN() == 'Y') {
 106             mbrTable.WriteMBRData();
 107             mbrTable.DiskSync();
 108             doItAgain = 0;
 109          } else {
 110             doItAgain = 1;
 111          } // else
 112       } // if
 113    } while (doItAgain);
 114 } // DoMBR()