Home | History | Annotate | Download | only in gptfdisk 
      1 // sgdisk.cc
      2 // Command-line-based version of gdisk. This program is named after sfdisk,
      3 // and it can serve a similar role (easily scripted, etc.), but it's used
      4 // strictly via command-line arguments, and it doesn't bear much resemblance
      5 // to sfdisk in actual use.
      6 //
      7 // by Rod Smith, project began February 2009; sgdisk begun January 2010.
      8 
      9 /* This program is copyright (c) 2009-2011 by Roderick W. Smith. It is distributed
     10   under the terms of the GNU GPL version 2, as detailed in the COPYING file. */
     11 
     12 #include <iostream>
     13 #include <fstream>
     14 #include <string.h>
     15 #include <string>
     16 #include <iostream>
     17 #include <sstream>
     18 #include <errno.h>
     19 #include "gptcl.h"
     20 #include <fcntl.h>
     21 #include <unistd.h>
     22 
     23 using namespace std;
     24 
     25 #define MAX_OPTIONS 50
     26 
     27 /*
     28  * Dump partition details in a machine readable format:
     29  *
     30  * DISK [mbr|gpt] [guid]
     31  * PART [n] [type] [guid]
     32  */
     33 static int android_dump(char* device) {
     34     BasicMBRData mbrData;
     35     GPTData gptData;
     36     GPTPart partData;
     37     int numParts = 0;
     38     stringstream res;
     39 
     40     /* Silence noisy underlying library */
     41     int stdout = dup(STDOUT_FILENO);
     42     int silence = open("/dev/null", 0);
     43     dup2(silence, STDOUT_FILENO);
     44     dup2(silence, STDERR_FILENO);
     45 
     46     if (!mbrData.ReadMBRData((string) device)) {
     47         cerr << "Failed to read MBR" << endl;
     48         return 8;
     49     }
     50 
     51     switch (mbrData.GetValidity()) {
     52     case mbr:
     53         res << "DISK mbr" << endl;
     54         for (int i = 0; i < MAX_MBR_PARTS; i++) {
     55             if (mbrData.GetLength(i) > 0) {
     56                 res << "PART " << (i + 1) << " " << hex
     57                         << (int) mbrData.GetType(i) << dec << endl;
     58             }
     59         }
     60         break;
     61     case gpt:
     62         gptData.JustLooking();
     63         if (!gptData.LoadPartitions((string) device)) {
     64             cerr << "Failed to read GPT" << endl;
     65             return 9;
     66         }
     67 
     68         res << "DISK gpt " << gptData.GetDiskGUID() << endl;
     69         numParts = gptData.GetNumParts();
     70         for (int i = 0; i < numParts; i++) {
     71             partData = gptData[i];
     72             if (partData.GetFirstLBA() > 0) {
     73                 res << "PART " << (i + 1) << " " << partData.GetType() << " "
     74                         << partData.GetUniqueGUID() << " "
     75                         << partData.GetDescription() << endl;
     76             }
     77         }
     78         break;
     79     default:
     80         cerr << "Unknown partition table" << endl;
     81         return 10;
     82     }
     83 
     84     /* Write our actual output */
     85     string resString = res.str();
     86     write(stdout, resString.c_str(), resString.length());
     87     return 0;
     88 }
     89 
     90 int main(int argc, char *argv[]) {
     91     for (int i = 0; i < argc; i++) {
     92         if (!strcmp("--android-dump", argv[i])) {
     93             return android_dump(argv[i + 1]);
     94         }
     95     }
     96 
     97     GPTDataCL theGPT;
     98     return theGPT.DoOptions(argc, argv);
     99 }
    100