diff options
Diffstat (limited to '')
| -rw-r--r-- | heimdall/source/main.cpp | 778 |
1 files changed, 12 insertions, 766 deletions
diff --git a/heimdall/source/main.cpp b/heimdall/source/main.cpp index db7de1c..41d7a3c 100644 --- a/heimdall/source/main.cpp +++ b/heimdall/source/main.cpp @@ -28,782 +28,28 @@ #include "libpit.h" // Heimdall -#include "BridgeManager.h" -#include "SetupSessionPacket.h" -#include "SetupSessionResponse.h" -#include "EndModemFileTransferPacket.h" -#include "EndPhoneFileTransferPacket.h" +#include "Heimdall.h" +#include "HelpAction.h" #include "Interface.h" using namespace std; using namespace Heimdall; -// Known partitions -enum -{ - kKnownPartitionPit = 0, - kKnownPartitionFactoryFs, - kKnownPartitionCache, - kKnownPartitionDatabaseData, - kKnownPartitionPrimaryBootloader, - kKnownPartitionSecondaryBootloader, - kKnownPartitionSecondaryBootloaderBackup, - kKnownPartitionParam, - kKnownPartitionKernel, - kKnownPartitionRecovery, - kKnownPartitionEfs, - kKnownPartitionModem, - - kKnownPartitionNormalBoot, - kKnownPartitionSystem, - kKnownPartitionUserData, - kKnownPartitionFota, - kKnownPartitionHidden, - kKnownPartitionMovinand, - kKnownPartitionData, - kKnownPartitionUms, - kKnownPartitionEmmc, - - kKnownPartitionCount -}; - -vector<const char *> knownPartitionNames[kKnownPartitionCount]; - -struct PartitionInfo -{ - unsigned int chipIdentifier; - string partitionName; - FILE *file; - - PartitionInfo(unsigned int chipIdentifier, const char *partitionName, FILE *file) - { - this->chipIdentifier = chipIdentifier; - this->partitionName = partitionName; - this->file = file; - } -}; - -void initialiseKnownPartitionNames(void) -{ - knownPartitionNames[kKnownPartitionPit].push_back("PIT"); - knownPartitionNames[kKnownPartitionFactoryFs].push_back("FACTORYFS"); - knownPartitionNames[kKnownPartitionCache].push_back("CACHE"); - knownPartitionNames[kKnownPartitionDatabaseData].push_back("DBDATAFS"); - - knownPartitionNames[kKnownPartitionPrimaryBootloader].push_back("IBL+PBL"); - knownPartitionNames[kKnownPartitionPrimaryBootloader].push_back("BOOT"); - - knownPartitionNames[kKnownPartitionSecondaryBootloader].push_back("SBL"); - knownPartitionNames[kKnownPartitionSecondaryBootloader].push_back("SBL1"); - - knownPartitionNames[kKnownPartitionSecondaryBootloaderBackup].push_back("SBL2"); - knownPartitionNames[kKnownPartitionParam].push_back("PARAM"); - knownPartitionNames[kKnownPartitionKernel].push_back("KERNEL"); - knownPartitionNames[kKnownPartitionRecovery].push_back("RECOVERY"); - knownPartitionNames[kKnownPartitionEfs].push_back("EFS"); - knownPartitionNames[kKnownPartitionModem].push_back("MODEM"); - - knownPartitionNames[kKnownPartitionNormalBoot].push_back("NORMALBOOT"); - knownPartitionNames[kKnownPartitionSystem].push_back("SYSTEM"); - knownPartitionNames[kKnownPartitionUserData].push_back("USERDATA"); - knownPartitionNames[kKnownPartitionFota].push_back("FOTA"); - knownPartitionNames[kKnownPartitionHidden].push_back("HIDDEN"); - knownPartitionNames[kKnownPartitionMovinand].push_back("MOVINAND"); - knownPartitionNames[kKnownPartitionData].push_back("DATAFS"); - knownPartitionNames[kKnownPartitionUms].push_back("UMS.EN"); - knownPartitionNames[kKnownPartitionEmmc].push_back("GANG"); -} - -bool isKnownPartition(const char *partitionName, unsigned int knownPartitionIndex) -{ - for (unsigned int i = 0; i < knownPartitionNames[knownPartitionIndex].size(); i++) - { - if (strcmp(partitionName, knownPartitionNames[knownPartitionIndex][i]) == 0) - return (true); - } - - return (false); -} - -bool isKnownBootPartition(const char *partitionName) -{ - return (isKnownPartition(partitionName, kKnownPartitionPrimaryBootloader) || - isKnownPartition(partitionName, kKnownPartitionSecondaryBootloader) || - isKnownPartition(partitionName, kKnownPartitionSecondaryBootloaderBackup) || - isKnownPartition(partitionName, kKnownPartitionParam) || - isKnownPartition(partitionName, kKnownPartitionNormalBoot)); -} - -bool openFiles(const map<string, string>& argumentMap, map<string, FILE *>& argumentFileMap) -{ - map<string, string>::const_iterator it = argumentMap.begin(); - - for (it = argumentMap.begin(); it != argumentMap.end(); it++) - { - bool isFileArgument = false; - - int partitionIndex = atoi(it->first.substr(it->first.find_first_not_of('-')).c_str()); - - // Was the argument a partition index? - if (partitionIndex > 0 || it->first.compare("-0") == 0) - { - isFileArgument = true; - } - else - { - // The argument wasn't a partition index, check if it's a known partition name. - for (int knownPartition = 0; knownPartition < kKnownPartitionCount; knownPartition++) - { - if (it->first == Interface::actions[Interface::kActionFlash].valueArguments[knownPartition]) - { - isFileArgument = true; - break; - } - } - } - - if (!isFileArgument) - continue; - - pair<string, FILE *> argumentFilePair; - argumentFilePair.first = it->first; - argumentFilePair.second = fopen(it->second.c_str(), "rb"); - - if (!argumentFilePair.second) - { - Interface::PrintError("Failed to open file \"%s\"\n", it->second.c_str()); - return (false); - } - - argumentFileMap.insert(argumentFilePair); - } - - return (true); -} - -bool mapFilesToPartitions(const map<string, FILE *>& argumentFileMap, const PitData *pitData, map<unsigned int, PartitionInfo>& partitionInfoMap) -{ - map<string, FILE *>::const_iterator it = argumentFileMap.begin(); - - for (it = argumentFileMap.begin(); it != argumentFileMap.end(); it++) - { - int partitionIndex = atoi(it->first.substr(it->first.find_first_not_of('-')).c_str()); - - const PitEntry *pitEntry = nullptr; - - // Was the argument a partition index? - if (partitionIndex > 0 || it->first.compare("-0") == 0) - { - pitEntry = pitData->FindEntry(partitionIndex); - } - else - { - // The argument wasn't a partition index, so it must be a known partition name. - int knownPartition; - - for (knownPartition = 0; knownPartition < kKnownPartitionCount; knownPartition++) - { - if (it->first == Interface::actions[Interface::kActionFlash].valueArguments[knownPartition]) - break; - } - - // Check for the partition in the PIT file using all known names. - for (unsigned int i = 0; i < knownPartitionNames[knownPartition].size(); i++) - { - pitEntry = pitData->FindEntry(knownPartitionNames[knownPartition][i]); - - if (pitEntry) - break; - } - - if (!pitEntry && knownPartition == kKnownPartitionPit) - { - // NOTE: We're assuming a PIT file always has chipIdentifier zero. - PartitionInfo partitionInfo(0, knownPartitionNames[kKnownPartitionPit][0], it->second); - partitionInfoMap.insert(pair<unsigned int, PartitionInfo>(0xFFFFFFFF, partitionInfo)); - - return (true); - } - } - - if (!pitEntry) - { - Interface::PrintError("Partition corresponding to %s argument could not be located\n", it->first.c_str()); - return (false); - } - - PartitionInfo partitionInfo(pitEntry->GetChipIdentifier(), pitEntry->GetPartitionName(), it->second); - partitionInfoMap.insert(pair<unsigned int, PartitionInfo>(pitEntry->GetPartitionIdentifier(), partitionInfo)); - } - - return (true); -} - -void closeFiles(map<string, FILE *> argumentfileMap) -{ - for (map<string, FILE *>::iterator it = argumentfileMap.begin(); it != argumentfileMap.end(); it++) - fclose(it->second); - - argumentfileMap.clear(); -} - -int downloadPitFile(BridgeManager *bridgeManager, unsigned char **pitBuffer) -{ - Interface::Print("Downloading device's PIT file...\n"); - - int devicePitFileSize = bridgeManager->ReceivePitFile(pitBuffer); - - if (!*pitBuffer) - { - Interface::PrintError("Failed to download PIT file!\n"); - - return (-1); - } - - Interface::Print("PIT file download sucessful\n\n"); - return (devicePitFileSize); -} - -bool flashFile(BridgeManager *bridgeManager, unsigned int chipIdentifier, unsigned int partitionIndex, const char *partitionName, FILE *file) -{ - // PIT files need to be handled differently, try determine if the partition we're flashing to is a PIT partition. - bool isPit = false; - - for (unsigned int i = 0; i < knownPartitionNames[kKnownPartitionPit].size(); i++) - { - if (strcmp(partitionName, knownPartitionNames[kKnownPartitionPit][i]) == 0) - { - isPit = true; - break; - } - } - - if (isPit) - { - Interface::Print("Uploading %s\n", partitionName); - - if (bridgeManager->SendPitFile(file)) - { - Interface::Print("%s upload successful\n", partitionName); - return (true); - } - else - { - Interface::Print("%s upload failed!\n", partitionName); - return (false); - } - } - else - { - // Modems need to be handled differently, try determine if the partition we're flashing to is a modem partition. - bool isModem = false; - - for (unsigned int i = 0; i < knownPartitionNames[kKnownPartitionModem].size(); i++) - { - if (strcmp(partitionName, knownPartitionNames[kKnownPartitionModem][i]) == 0) - { - isModem = true; - break; - } - } - - if (isModem) - { - Interface::Print("Uploading %s\n", partitionName); - - //if (bridgeManager->SendFile(file, EndPhoneFileTransferPacket::kDestinationPhone, // <-- Kies method. WARNING: Doesn't work on Galaxy Tab! - // EndPhoneFileTransferPacket::kFileModem)) - if (bridgeManager->SendFile(file, EndModemFileTransferPacket::kDestinationModem, chipIdentifier)) // <-- Odin method - { - Interface::Print("%s upload successful\n", partitionName); - return (true); - } - else - { - Interface::Print("%s upload failed!\n", partitionName); - return (false); - } - } - else - { - // We're uploading to a phone partition - Interface::Print("Uploading %s\n", partitionName); - - if (bridgeManager->SendFile(file, EndPhoneFileTransferPacket::kDestinationPhone, chipIdentifier, partitionIndex)) - { - Interface::Print("%s upload successful\n", partitionName); - return (true); - } - else - { - Interface::Print("%s upload failed!\n", partitionName); - return (false); - } - } - } - - return (true); -} - -bool attemptFlash(BridgeManager *bridgeManager, map<string, FILE *> argumentFileMap, bool repartition) -{ - bool success; - - // ------------- SEND TOTAL BYTES TO BE TRANSFERRED ------------- - - int totalBytes = 0; - for (map<string, FILE *>::const_iterator it = argumentFileMap.begin(); it != argumentFileMap.end(); it++) - { - if (repartition || it->first != Interface::GetPitArgument()) - { - fseek(it->second, 0, SEEK_END); - totalBytes += ftell(it->second); - rewind(it->second); - } - } - - SetupSessionPacket *deviceInfoPacket = new SetupSessionPacket(SetupSessionPacket::kTotalBytes, totalBytes); - success = bridgeManager->SendPacket(deviceInfoPacket); - delete deviceInfoPacket; - - if (!success) - { - Interface::PrintError("Failed to send total bytes device info packet!\n"); - return (false); - } - - SetupSessionResponse *deviceInfoResponse = new SetupSessionResponse(); - success = bridgeManager->ReceivePacket(deviceInfoResponse); - int deviceInfoResult = deviceInfoResponse->GetUnknown(); - delete deviceInfoResponse; - - if (!success) - { - Interface::PrintError("Failed to receive device info response!\n"); - return (false); - } - - if (deviceInfoResult != 0) - { - Interface::PrintError("Unexpected device info response!\nExpected: 0\nReceived:%d\n", deviceInfoResult); - return (false); - } - - // ----------------------------------------------------- - - PitData *pitData; - PitData *localPitData = nullptr; - - FILE *localPitFile = nullptr; - - // If a PIT file was passed as an argument then we must unpack it. - map<string, FILE *>::iterator it = argumentFileMap.find(Interface::actions[Interface::kActionFlash].valueArguments[Interface::kFlashValueArgPit]); - - if (it != argumentFileMap.end()) - { - localPitFile = it->second; - - // Load the local pit file into memory. - unsigned char *pitFileBuffer = new unsigned char[4096]; - memset(pitFileBuffer, 0, 4096); - - fseek(localPitFile, 0, SEEK_END); - long localPitFileSize = ftell(localPitFile); - rewind(localPitFile); - - // dataRead is discarded, it's here to remove warnings. - int dataRead = fread(pitFileBuffer, 1, localPitFileSize, localPitFile); - rewind(localPitFile); - - localPitData = new PitData(); - localPitData->Unpack(pitFileBuffer); - - delete [] pitFileBuffer; - } - - if (repartition) - { - // Use the local PIT file data. - pitData = localPitData; - } - else - { - // If we're not repartitioning then we need to retrieve the device's PIT file and unpack it. - unsigned char *pitFileBuffer; - downloadPitFile(bridgeManager, &pitFileBuffer); - - pitData = new PitData(); - pitData->Unpack(pitFileBuffer); - - delete [] pitFileBuffer; - - if (localPitData != nullptr) - { - // The user has specified a PIT without repartitioning, we should verify the local and device PIT data match! - bool pitsMatch = pitData->Matches(localPitData); - delete localPitData; - - if (!pitsMatch) - { - Interface::Print("Local and device PIT files don't match and repartition wasn't specified!\n"); - Interface::PrintError("Flash aborted!\n"); - - delete pitData; - return (false); - } - } - } - - map<unsigned int, PartitionInfo> partitionInfoMap; - - // Map the files being flashed to partitions stored in the PIT file. - if (!mapFilesToPartitions(argumentFileMap, pitData, partitionInfoMap)) - { - delete pitData; - return (false); - } - - delete pitData; - - // If we're repartitioning then we need to flash the PIT file first. - if (repartition) - { - for (map<unsigned int, PartitionInfo>::iterator it = partitionInfoMap.begin(); it != partitionInfoMap.end(); it++) - { - if (it->second.file == localPitFile) - { - PartitionInfo *partitionInfo = &(it->second); - - if (!flashFile(bridgeManager, partitionInfo->chipIdentifier, it->first, partitionInfo->partitionName.c_str(), partitionInfo->file)) - return (false); - - break; - } - } - } - - // Flash partitions not involved in the boot process second. - for (map<unsigned int, PartitionInfo>::iterator it = partitionInfoMap.begin(); it != partitionInfoMap.end(); it++) - { - if (!isKnownPartition(it->second.partitionName.c_str(), kKnownPartitionPit) && !isKnownBootPartition(it->second.partitionName.c_str())) - { - PartitionInfo *partitionInfo = &(it->second); - - if (!flashFile(bridgeManager, partitionInfo->chipIdentifier, it->first, partitionInfo->partitionName.c_str(), partitionInfo->file)) - return (false); - } - } - - // Flash boot partitions last. - for (map<unsigned int, PartitionInfo>::iterator it = partitionInfoMap.begin(); it != partitionInfoMap.end(); it++) - { - if (isKnownBootPartition(it->second.partitionName.c_str())) - { - PartitionInfo *partitionInfo = &(it->second); - - if (!flashFile(bridgeManager, partitionInfo->chipIdentifier, it->first, partitionInfo->partitionName.c_str(), partitionInfo->file)) - return (false); - } - } - - return (true); -} - int main(int argc, char **argv) { - map<string, string> argumentMap; - int actionIndex; - - if (!Interface::GetArguments(argc, argv, argumentMap, &actionIndex)) + if (argc < 2) { - Sleep(250); + Interface::PrintUsage(); return (0); } - initialiseKnownPartitionNames(); - - switch (actionIndex) - { - case Interface::kActionFlash: - if (argumentMap.find(Interface::actions[Interface::kActionFlash].valuelessArguments[Interface::kFlashValuelessArgRepartition]) != argumentMap.end() - && argumentMap.find(Interface::actions[Interface::kActionFlash].valueArguments[Interface::kFlashValueArgPit]) == argumentMap.end()) - { - Interface::Print("If you wish to repartition then a PIT file must be specified.\n\n"); - Interface::PrintUsage(); - return (0); - } - - break; - - case Interface::kActionDownloadPit: - if (argumentMap.find(Interface::actions[Interface::kActionDownloadPit].valueArguments[Interface::kDownloadPitValueArgOutput]) == argumentMap.end()) - { - Interface::Print("Output file was not specified.\n\n"); - Interface::PrintUsage(); - return (0); - } - - break; - - case Interface::kActionDump: - { - if (argumentMap.find(Interface::actions[Interface::kActionDump].valueArguments[Interface::kDumpValueArgOutput]) == argumentMap.end()) - { - Interface::Print("Output file was not specified.\n\n"); - Interface::PrintUsage(); - return (0); - } - - if (argumentMap.find(Interface::actions[Interface::kActionDump].valueArguments[Interface::kDumpValueArgChipType]) == argumentMap.end()) - { - Interface::Print("You must specify a chip type.\n\n"); - Interface::PrintUsage(); - return (0); - } - - string chipType = argumentMap.find(Interface::actions[Interface::kActionDump].valueArguments[Interface::kDumpValueArgChipType])->second; - if (!(chipType == "RAM" || chipType == "ram" || chipType == "NAND" || chipType == "nand")) - { - Interface::Print("Unknown chip type: %s.\n\n", chipType.c_str()); - Interface::PrintUsage(); - return (0); - } - - if (argumentMap.find(Interface::actions[Interface::kActionDump].valueArguments[Interface::kDumpValueArgChipId]) == argumentMap.end()) - { - Interface::Print("You must specify a Chip ID.\n\n"); - Interface::PrintUsage(); - return (0); - } - - int chipId = atoi(argumentMap.find(Interface::actions[Interface::kActionDump].valueArguments[Interface::kDumpValueArgChipId])->second.c_str()); - if (chipId < 0) - { - Interface::Print("Chip ID must be a non-negative integer.\n"); - Interface::PrintUsage(); - return (0); - } - - break; - } - - case Interface::kActionVersion: - Interface::PrintVersion(); - return (0); - - case Interface::kActionHelp: - Interface::PrintUsage(); - return (0); - - case Interface::kActionInfo: - Interface::PrintFullInfo(); - return (0); - } - - bool verbose = argumentMap.find(Interface::commonValuelessArguments[Interface::kCommonValuelessArgVerbose]) != argumentMap.end(); - bool reboot = argumentMap.find(Interface::commonValuelessArguments[Interface::kCommonValuelessArgNoReboot]) == argumentMap.end(); - - Interface::SetStdoutErrors(argumentMap.find(Interface::commonValuelessArguments[Interface::kCommonValuelessArgStdoutErrors]) != argumentMap.end()); - - int communicationDelay = BridgeManager::kCommunicationDelayDefault; + int result = 0; + map<string, Interface::ActionInfo>::const_iterator actionIt = Interface::GetActionMap().find(argv[1]); - if (argumentMap.find(Interface::commonValueArguments[Interface::kCommonValueArgDelay]) != argumentMap.end()) - communicationDelay = atoi(argumentMap.find(Interface::commonValueArguments[Interface::kCommonValueArgDelay])->second.c_str()); - - BridgeManager *bridgeManager = new BridgeManager(verbose, communicationDelay); - - if (actionIndex == Interface::kActionDetect) - { - bool detected = bridgeManager->DetectDevice(); - delete bridgeManager; - - return ((detected) ? 0 : 1); - } - - Interface::PrintReleaseInfo(); - Sleep(1000); - - int initialiseResult = bridgeManager->Initialise(); - - if (initialiseResult != 0) - { - delete bridgeManager; - return ((initialiseResult == BridgeManager::kInitialiseDeviceNotDetected) ? 1 : 0); - } - - bool success; - - switch (actionIndex) - { - case Interface::kActionFlash: - { - map<string, FILE *> argumentFileMap; - - // We open the files before doing anything else to ensure they exist. - if (!openFiles(argumentMap, argumentFileMap)) - { - closeFiles(argumentFileMap); - delete bridgeManager; - - return (0); - } - - if (!bridgeManager->BeginSession()) - { - closeFiles(argumentFileMap); - delete bridgeManager; - - return (-1); - } - - bool repartition = argumentMap.find(Interface::actions[Interface::kActionFlash].valuelessArguments[Interface::kFlashValuelessArgRepartition]) != argumentMap.end(); - success = attemptFlash(bridgeManager, argumentFileMap, repartition); - - success = bridgeManager->EndSession(reboot) && success; - - closeFiles(argumentFileMap); - - break; - } - - case Interface::kActionClosePcScreen: - { - if (!bridgeManager->BeginSession()) - { - delete bridgeManager; - return (-1); - } - - Interface::Print("Attempting to close connect to pc screen...\n"); - - success = bridgeManager->EndSession(reboot); - - if (success) - Interface::Print("Attempt complete\n"); - - break; - } - - case Interface::kActionDownloadPit: - { - map<string, string>::const_iterator it = argumentMap.find(Interface::actions[Interface::kActionDownloadPit].valueArguments[Interface::kDownloadPitValueArgOutput]); - FILE *outputPitFile = fopen(it->second.c_str(), "wb"); - - if (!outputPitFile) - { - delete bridgeManager; - return (0); - } - - if (!bridgeManager->BeginSession()) - { - delete bridgeManager; - fclose(outputPitFile); - return (-1); - } - - unsigned char *pitBuffer; - int fileSize = downloadPitFile(bridgeManager, &pitBuffer); - - if (fileSize > 0) - { - success = fwrite(pitBuffer, 1, fileSize, outputPitFile) == fileSize; - fclose(outputPitFile); - - if (!success) - Interface::PrintError("Failed to write PIT data to output file.\n"); - - success = bridgeManager->EndSession(reboot) && success; - } - else - { - fclose(outputPitFile); - success = false; - bridgeManager->EndSession(reboot); - } - - delete [] pitBuffer; - - break; - } - - case Interface::kActionDump: - { - const char *outputFilename = argumentMap.find(Interface::actions[Interface::kActionDump].valueArguments[Interface::kDumpValueArgOutput])->second.c_str(); - FILE *dumpFile = fopen(outputFilename, "wb"); - if (!dumpFile) - { - Interface::PrintError("Failed to open file \"%s\"\n", outputFilename); - - delete bridgeManager; - return (-1); - } - - int chipType = 0; - string chipTypeName = argumentMap.find(Interface::actions[Interface::kActionDump].valueArguments[Interface::kDumpValueArgChipType])->second; - if (chipTypeName == "NAND" || chipTypeName == "nand") - chipType = 1; - - int chipId = atoi(argumentMap.find(Interface::actions[Interface::kActionDump].valueArguments[Interface::kDumpValueArgChipId])->second.c_str()); - - if (!bridgeManager->BeginSession()) - { - fclose(dumpFile); - - delete bridgeManager; - return (-1); - } - - success = bridgeManager->ReceiveDump(chipType, chipId, dumpFile); - - fclose(dumpFile); - - success = bridgeManager->EndSession(reboot) && success; - - break; - } - - case Interface::kActionPrintPit: - { - if (!bridgeManager->BeginSession()) - { - delete bridgeManager; - return (-1); - } - - unsigned char *devicePit; - - if (downloadPitFile(bridgeManager, &devicePit) < -1) - { - bridgeManager->EndSession(reboot); - - delete bridgeManager; - return (-1); - } - - PitData *pitData = new PitData(); - - if (pitData->Unpack(devicePit)) - { - Interface::PrintPit(pitData); - success = true; - } - else - { - Interface::PrintError("Failed to unpack device's PIT file!\n"); - success = false; - } - - delete [] devicePit; - delete pitData; - - success = bridgeManager->EndSession(reboot) && success; - - break; - } - } - - delete bridgeManager; - - return ((success) ? 0 : -1); + if (actionIt != Interface::GetActionMap().end()) + result = actionIt->second.executeFunction(argc, argv); + else + result = HelpAction::Execute(argc, argv); + + return (result); } |