| 1 | /*␊ |
| 2 | * shell_process.cpp␊ |
| 3 | *␊ |
| 4 | * Created by Rekursor on 1/17/2010.␊ |
| 5 | *␊ |
| 6 | */␊ |
| 7 | ␊ |
| 8 | #include "ShellProcess.h"␊ |
| 9 | #include <string.h>␊ |
| 10 | #include <sys/stat.h>␊ |
| 11 | #include <string_util.h>␊ |
| 12 | ␊ |
| 13 | ␊ |
| 14 | //----------------------------------------------------------------␊ |
| 15 | // portable open process:␊ |
| 16 | FILE * ShellProcess::open(const char *cmd, const char *mode) {␊ |
| 17 | ␉_fpt=::popen(cmd,mode);␊ |
| 18 | ␉return _fpt;␊ |
| 19 | }␊ |
| 20 | ␊ |
| 21 | //----------------------------------------------------------------␊ |
| 22 | int ShellProcess::close() {␊ |
| 23 | ␉int ret = ::pclose(_fpt);␊ |
| 24 | ␉_fpt=NULL;␊ |
| 25 | ␉return ret;␊ |
| 26 | }␊ |
| 27 | ␊ |
| 28 | //----------------------------------------------------------------␊ |
| 29 | void PartitionInfo::removeSpaces(std::string &str)␊ |
| 30 | {␊ |
| 31 | ␉if (!str.size()) return;␊ |
| 32 | ␉int posl=0,posr=0;␊ |
| 33 | ␉for (posl=0; posl<str.size()-1 && str[posl]==' '; posl++);␊ |
| 34 | ␉for (posr=str.size()-1; posr>0 && str[posr]==' '; posr--);␊ |
| 35 | ␉str = (posl>posr) ? "" : str.substr(posl, posr-posl+1);␊ |
| 36 | }␊ |
| 37 | //----------------------------------------------------------------␊ |
| 38 | /**␊ |
| 39 | * ␉return the image index corresponding to a fs:␊ |
| 40 | */␊ |
| 41 | int PartitionInfo::imageIndexFromFs() const␊ |
| 42 | {␊ |
| 43 | ␉if (_fsType.find("Apple")!=std::string::npos ||␊ |
| 44 | ␉␉_fsType.find("HFS")!=std::string::npos)␊ |
| 45 | ␉␉return 0; // Windows␊ |
| 46 | ␉else if (_fsType.find("Windows")!=std::string::npos ||␊ |
| 47 | ␉␉␉ _fsType.find("Microsoft")!=std::string::npos ||␊ |
| 48 | ␉␉␉ _fsType.find("NTFS")!=std::string::npos ||␊ |
| 49 | ␉␉␉ _fsType.find("FAT")!=std::string::npos)␊ |
| 50 | ␉␉return 1; // Windows␊ |
| 51 | ␉else if (_fsType.find("Linux")!=std::string::npos)␊ |
| 52 | ␉␉return 2; // Unknown␊ |
| 53 | ␉return 10; //Unknown␊ |
| 54 | ␉␊ |
| 55 | }␊ |
| 56 | //----------------------------------------------------------------␊ |
| 57 | bool PartitionInfo::fromPartitionHdString(const char * inHdStr)␊ |
| 58 | {␊ |
| 59 | ␉if (!inHdStr || !(*inHdStr) || ␊ |
| 60 | ␉␉strlen(inHdStr)<7 || !strstr(inHdStr,"hd(")) return false;␊ |
| 61 | ␉␊ |
| 62 | ␉// enhance me here: should not assume that we have less than 10 partitions per disk/ disks␊ |
| 63 | ␉_disk = inHdStr[3]-'0'; ␊ |
| 64 | ␉_part = inHdStr[5]-'0';␊ |
| 65 | ␉return true;␊ |
| 66 | }␊ |
| 67 | //----------------------------------------------------------------␊ |
| 68 | const char * PartitionExtractor::checkForRename(const char * label, const char *szHd)␊ |
| 69 | {␊ |
| 70 | ␉const int MAX_ALIAS_SIZE=31;␊ |
| 71 | ␉static char szAlias[MAX_ALIAS_SIZE+1];␊ |
| 72 | ␉char *q=szAlias;␊ |
| 73 | ␉const char* szAliases = _renamedParts.c_str();␊ |
| 74 | ␉␊ |
| 75 | ␉if (!szHd || !*szHd || !szAliases || !*szAliases) return label; // no renaming wanted␊ |
| 76 | ␉␊ |
| 77 | ␉const char * p = strstr(szAliases, szHd);␊ |
| 78 | ␉if(!p || !(*p)) return label; // this volume must not be renamed, or option is malformed␊ |
| 79 | ␉␊ |
| 80 | ␉p+= strlen(szHd); // skip the "hd(n,m) " field␊ |
| 81 | ␉// multiple aliases can be found separated by a semi-column␊ |
| 82 | ␉while(*p && *p != ';' && q<(szAlias+MAX_ALIAS_SIZE)) *q++=*p++;␊ |
| 83 | ␉*q='\0';␊ |
| 84 | ␉␊ |
| 85 | ␉return szAlias;␊ |
| 86 | ␉␊ |
| 87 | }␊ |
| 88 | ␊ |
| 89 | // set to 1 the following line for debugging with a fake diskutil file named /diskutil.txt␊ |
| 90 | // for the debug to be accurate you need to alias (ln -s ...) the user to debug with your real disks␊ |
| 91 | #define DEBUG_DISKUTIL 0␊ |
| 92 | ␊ |
| 93 | //----------------------------------------------------------------␊ |
| 94 | const std::vector<PartitionInfo>& ␊ |
| 95 | PartitionExtractor::extractPartitions(const char* szHide, const char* szRenamed)␊ |
| 96 | {␊ |
| 97 | ␉const char * const diskTag = "/dev/disk";␊ |
| 98 | ␉const char * const nameTag = "NAME";␊ |
| 99 | ␉const char * const sizeTag = "SIZE";␊ |
| 100 | ␊ |
| 101 | ␉PartitionInfo partInfo;␊ |
| 102 | ␊ |
| 103 | ␉char line[1024]="";␊ |
| 104 | ␉char label[32]="", fsType[32]="" ;␊ |
| 105 | ␉size_t len=0;␊ |
| 106 | ␉int disk =0, part=0;␊ |
| 107 | ␉int label_pos=0, size_pos=0;␊ |
| 108 | ␉char * p=0,*q=0;␊ |
| 109 | ␉int skipwhite=0;␊ |
| 110 | ␉␊ |
| 111 | ␉_partList.clear();␊ |
| 112 | ␉if (szHide) hidePartitions(szHide);␊ |
| 113 | ␉if (szRenamed) renamedPartitions(szRenamed);␊ |
| 114 | ␊ |
| 115 | #if DEBUG_DISKUTIL␊ |
| 116 | ␉this->open("cat /diskutil.txt");␊ |
| 117 | #else␊ |
| 118 | ␉this->open("diskutil list");␊ |
| 119 | #endif␊ |
| 120 | ␉␊ |
| 121 | ␉while(get_line(line, sizeof(line)-1))␊ |
| 122 | ␉{␊ |
| 123 | ␉␉// printf("%s\n",line);␊ |
| 124 | ␉␉len = strlen(line);␊ |
| 125 | ␉␉for(skipwhite=0; line[skipwhite]==' ';skipwhite++);␊ |
| 126 | ␉␉␊ |
| 127 | ␉␉const char * sdisk = strstr(line, diskTag);␊ |
| 128 | ␉␉const char * sName = strstr(line, nameTag);␊ |
| 129 | ␉␉const char * sSize = strstr(line, sizeTag);␊ |
| 130 | ␉␉␊ |
| 131 | ␉␉if (sdisk)␊ |
| 132 | ␉␉{␊ |
| 133 | ␉␉␉// extract disk number␊ |
| 134 | ␉␉␉sscanf(&sdisk[strlen(diskTag)],"%d", &disk);␊ |
| 135 | ␉␉␉if (disk>=0 && disk <MAX_HD)␊ |
| 136 | ␉␉␉␉disk = _hdRedirTable[disk];␊ |
| 137 | ␉␉}␊ |
| 138 | ␉␉else if (len && line[skipwhite]=='#' && line[skipwhite+1]==':')␊ |
| 139 | ␉␉{␊ |
| 140 | ␉␉␉label_pos = sName ? (sName - line) : 0;␊ |
| 141 | ␉␉␉size_pos = sSize ? (sSize - line) : 0;␊ |
| 142 | ␉␉}␊ |
| 143 | ␉␉else if (len && line[skipwhite+1]==':' && isdigit(line[skipwhite]))␊ |
| 144 | ␉␉{␊ |
| 145 | ␉␉␉for (p=&line[skipwhite+2], q=fsType; *p && p < &line[label_pos-1]; p++,q++) *q=*p;␊ |
| 146 | ␉␉␉for (p=&line[label_pos], q=label; *p && p < &line[size_pos]; p++,q++) *q=*p;␊ |
| 147 | ␉␉␉␊ |
| 148 | ␉␉␉*q='\0';␊ |
| 149 | ␉␉␉part = line[skipwhite]-'0';␊ |
| 150 | ␉␉␉partInfo.fsType(fsType);␊ |
| 151 | ␉␉␉partInfo.disk(disk);␊ |
| 152 | ␉␉␉partInfo.partition(part);␊ |
| 153 | ␉␉␉partInfo.label(checkForRename(label, partInfo.toHdStr().c_str()));␊ |
| 154 | ␉␉␉size_t len = partInfo.label().length();␊ |
| 155 | ␉␉␉if(len==0){␊ |
| 156 | ␉␉␉␉strncpy(label, "Untitled", sizeof(label));␊ |
| 157 | ␉␉␉␉partInfo.label(label);␊ |
| 158 | ␉␉␉}␊ |
| 159 | ␉␉␉␉␊ |
| 160 | ␉␉␉// filter useless partitions:␊ |
| 161 | ␉␉␉if (partInfo.clabel()[len-1] !='*' && ␊ |
| 162 | ␉␉␉␉partInfo.fsType()!="EFI" &&␊ |
| 163 | ␉␉␉␉partInfo.fsType()!="Apple_partition_scheme" &&␊ |
| 164 | ␉␉␉␉partInfo.fsType()!="Apple_partition_map" &&␊ |
| 165 | ␉␉␉␉partInfo.fsType()!="Apple_Free" ␊ |
| 166 | ␉␉␉␉)␊ |
| 167 | ␉␉␉{␊ |
| 168 | ␉␉␉␉std::string diskLabel(trim(label)); ␊ |
| 169 | ␉␉␉␉bool found=false;␊ |
| 170 | ␉␉␉␉// early bail out if we found what we need: DON't make any assumption ␊ |
| 171 | ␉␉␉␉// that Windows is the name of WIN32 bootable disk dir ...␊ |
| 172 | ␉␉␉␉if (␊ |
| 173 | ␉␉␉␉␉fileExists("/Volumes/" + diskLabel + "/boot") || ␊ |
| 174 | ␉␉␉␉␉fileExists("/Volumes/" + diskLabel + "/System/Library/Extensions") ||␊ |
| 175 | ␉␉␉␉␉fileExists("/Volumes/" + diskLabel + "/ntldr") ||␊ |
| 176 | ␉␉␉␉␉fileExists("/Volumes/" + diskLabel + "/bootmgr") ||␊ |
| 177 | ␉␉␉␉␉fileExists("/Volumes/" + diskLabel + "/Boot") ||␊ |
| 178 | ␉␉␉␉␉fileExists("/Volumes/" + diskLabel + "/pagefile.sys") ||␊ |
| 179 | ␉␉␉␉␉fileExists("/Volumes/" + diskLabel + "/hiberfil.sys")␊ |
| 180 | ␉␉␉␉␉)␊ |
| 181 | ␉␉␉␉␉found=true;␊ |
| 182 | ␉␉␉␉else if ((strstr(fsType,"Linux") ) && !strstr(fsType, "Linux_Swap"))␊ |
| 183 | ␉␉␉␉␉found=true; // Added Linux case␊ |
| 184 | ␉␉␉␉if (found)␊ |
| 185 | ␉␉␉␉{ //check if one of them exists␊ |
| 186 | ␉␉␉␉␉if (partInfo.label().size()==0) partInfo.label("Untitled");␊ |
| 187 | ␉␉␉␉␉if(_hiddenParts.length()==0 || ␊ |
| 188 | ␉␉␉␉␉ _hiddenParts.find(partInfo.toHdStr()) == std::string::npos)␊ |
| 189 | ␉␉␉␉␉␉_partList.push_back(partInfo);␊ |
| 190 | ␉␉␉␉}␊ |
| 191 | ␉␉␉}␊ |
| 192 | ␉␉}␊ |
| 193 | ␉}␊ |
| 194 | ␉close();␊ |
| 195 | ␉␊ |
| 196 | ␉sort(_partList.begin(), _partList.end(), isDiskIndexInf);␊ |
| 197 | ␉return _partList;␊ |
| 198 | }␊ |
| 199 | ␊ |
| 200 | /** ␊ |
| 201 | * Get the index in the internal partlist of the hd(n,m) partition specified by the input string␊ |
| 202 | * return -1 if no match, >=0 if a match happens␊ |
| 203 | */␊ |
| 204 | int PartitionExtractor::getIndexFromHdStringSpec(const char* inHDStr)␊ |
| 205 | {␉␊ |
| 206 | ␉PartitionInfo info;␊ |
| 207 | ␉␊ |
| 208 | ␉if (info.fromPartitionHdString(inHDStr)) // try decode the partition disk and part infos␊ |
| 209 | ␉{␊ |
| 210 | ␉␉for (int i=0; i<_partList.size(); i++)␊ |
| 211 | ␉␉{␊ |
| 212 | ␉␉␉if (_partList[i].disk() == info.disk() && _partList[i].partition() == info.partition())␊ |
| 213 | ␉␉␉␉return i;␊ |
| 214 | ␉␉}␊ |
| 215 | ␉}␊ |
| 216 | ␉␊ |
| 217 | ␉return -1;␊ |
| 218 | }␊ |
| 219 | |
