1 /* 2 * Implementation of GPTData class derivative with curses-based text-mode 3 * interaction 4 * Copyright (C) 2011-2013 Roderick W. Smith 5 * 6 * This program is free software; you can redistribute it and/or modify 7 * it under the terms of the GNU General Public License as published by 8 * the Free Software Foundation; either version 2 of the License, or 9 * (at your option) any later version. 10 * 11 * This program is distributed in the hope that it will be useful, 12 * but WITHOUT ANY WARRANTY; without even the implied warranty of 13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 14 * GNU General Public License for more details. 15 * 16 * You should have received a copy of the GNU General Public License along 17 * with this program; if not, write to the Free Software Foundation, Inc., 18 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. 19 * 20 */ 21 22 #include <iostream> 23 #include <string> 24 #include "gptpart.h" 25 #include "gpt.h" 26 27 #ifndef __GPT_CURSES 28 #define __GPT_CURSES 29 30 using namespace std; 31 32 struct MenuItem { 33 int key; // Keyboard shortcut 34 string name; // Item name; 8 characters 35 string desc; // Description 36 }; 37 38 static struct MenuItem menuMain[] = { 39 { 'a', "Align ", "Set partition alignment policy" }, 40 { 'b', "Backup", "Back up the partition table" }, 41 { 'd', "Delete", "Delete the current partition" }, 42 { 'h', " Help ", "Print help screen" }, 43 { 'i', " Info ", "Display information about the partition" }, 44 { 'l', " Load ", "Load partition table backup from file" }, 45 { 'm', " naMe ", "Change the partition's name" }, 46 { 'n', " New ", "Create new partition from free space" }, 47 { 'q', " Quit ", "Quit program without writing partition table" }, 48 { 't', " Type ", "Change the filesystem type code GUID" }, 49 { 'v', "Verify", "Verify the integrity of the disk's data structures" }, 50 { 'w', "Write ", "Write partition table to disk (this might destroy data)" }, 51 { 0, "", "" } 52 }; 53 54 #define EMPTY_SPACE_OPTIONS "abhlnqvw" 55 #define PARTITION_OPTIONS "abdhilmqtvw" 56 57 // Constants for how to highlight a selected menu item 58 #define USE_CURSES 1 59 #define USE_ARROW 2 60 61 // A "Space" is a partition or an unallocated chunk of disk space, maintained 62 // in a doubly-linked-list data structure to facilitate creating displays of 63 // partitions and unallocated chunks of space on the disk in the main 64 // cgdisk partition list. This list MUST be correctly maintained and in order, 65 // and the numSpaces variable in the main GPTDataCurses class must specify 66 // how many Spaces are in the main linked list of Spaces. 67 struct Space { 68 uint64_t firstLBA; 69 uint64_t lastLBA; 70 GPTPart *origPart; 71 int partNum; 72 Space *nextSpace; 73 Space *prevSpace; 74 }; 75 76 class GPTDataCurses : public GPTData { 77 protected: 78 static int numInstances; 79 GPTPart emptySpace; 80 Space *firstSpace; 81 Space *lastSpace; 82 Space *currentSpace; 83 int currentSpaceNum; 84 string whichOptions; 85 char currentKey; 86 int numSpaces; 87 int displayType; 88 89 // Functions relating to Spaces data structures 90 void EmptySpaces(void); 91 int MakeSpacesFromParts(void); 92 void AddEmptySpace(uint64_t firstLBA, uint64_t lastLBA); 93 int AddEmptySpaces(void); 94 void UnlinkSpace(Space *theSpace); 95 void LinkToEnd(Space *theSpace); 96 void SortSpaces(void); 97 void IdentifySpaces(void); 98 99 // Data display functions 100 Space* ShowSpace(int spaceNum, int lineNum); 101 int DisplayParts(int selected); 102 public: 103 GPTDataCurses(void); 104 ~GPTDataCurses(void); 105 // Functions corresponding to main menu items 106 void DeletePartition(int partNum); 107 void ShowInfo(int partNum); 108 void ChangeName(int partNum); 109 void ChangeType(int partNum); 110 void SetAlignment(void); 111 void Verify(void); 112 void MakeNewPart(void); 113 void SaveData(void); 114 void Backup(void); 115 void LoadBackup(void); 116 void ShowHelp(void); 117 // User input and menuing functions SetDisplayType(int dt)118 void SetDisplayType(int dt) {displayType = dt;} 119 void ChangeSpaceSelection(int delta); 120 void MoveSelection(int delta); 121 void DisplayOptions(char selectedKey); 122 void AcceptInput(); 123 int Dispatch(char operation); 124 void DrawMenu(void); 125 int MainMenu(void); 126 }; // class GPTDataCurses 127 128 // Non-class support functions (mostly to do simple curses stuff).... 129 130 void ClearLine(int lineNum); 131 void ClearBottom(void); 132 void PromptToContinue(void); 133 void Report(string theText); 134 void ShowTypes(void); 135 136 #endif 137