diff options
Diffstat (limited to 'heimdall/source/InterfaceManager.cpp')
| -rw-r--r-- | heimdall/source/InterfaceManager.cpp | 265 |
1 files changed, 265 insertions, 0 deletions
diff --git a/heimdall/source/InterfaceManager.cpp b/heimdall/source/InterfaceManager.cpp new file mode 100644 index 0000000..23791bc --- /dev/null +++ b/heimdall/source/InterfaceManager.cpp @@ -0,0 +1,265 @@ +/* Copyright (c) 2010 Benjamin Dobell, Glass Echidna
+
+ Permission is hereby granted, free of charge, to any person obtaining a copy
+ of this software and associated documentation files (the "Software"), to deal
+ in the Software without restriction, including without limitation the rights
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ copies of the Software, and to permit persons to whom the Software is
+ furnished to do so, subject to the following conditions:
+
+ The above copyright notice and this permission notice shall be included in
+ all copies or substantial portions of the Software.
+
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ THE SOFTWARE.*/
+
+// C/C++ Standard Library
+#include <cstdarg>
+#include <stdio.h>
+
+// Heimdall
+#include "Heimdall.h"
+#include "InterfaceManager.h"
+
+using namespace std;
+using namespace Heimdall;
+
+string InterfaceManager::actionNames[kActionCount] = { "flash", "close-pc-screen", "dump", "help" };
+
+string InterfaceManager::flashArgumentNames[kFlashArgCount * 2] = {
+ // --- Long Names ---
+ "-repartition",
+
+ "-pit", "-factoryfs", "-cache", "-dbdata", "-primary-boot", "-secondary-boot", "-secondary-boot-backup", "-param", "-kernel", "-recovery", "-efs", "-modem",
+
+ // --- Short Names ---
+ "r",
+
+ "pit", "fs", "cache", "db", "boot", "sbl", "sbl2", "param", "z", "rec", "efs", "m"
+};
+
+string InterfaceManager::dumpArgumentNames[kDumpArgCount * 2] = {
+ // --- Long Names ---
+ "-chip-type", "-chip-id", "-output",
+
+ // --- Short Names ---
+ "type", "id", "out"
+};
+
+string InterfaceManager::commonArgumentNames[kCommonArgCount * 2] = {
+ // --- Long Names ---
+ "-verbose",
+
+ "-delay",
+
+ // --- Short Names ---
+ "v",
+
+ "d"
+};
+
+// argumentNames[kActionCount][] stores common arguments.
+string *InterfaceManager::actionArgumentNames[kActionCount + 1] = {
+ // kActionFlash
+ flashArgumentNames,
+
+ // kActionClosePcScreen
+ nullptr,
+
+ // kActionDump
+ dumpArgumentNames,
+
+ // kActionHelp
+ nullptr,
+
+ // Common (kActionCount)
+ commonArgumentNames
+};
+
+int InterfaceManager::actionArgumentCounts[kActionCount + 1] = {
+ kFlashArgCount, 0, kDumpArgCount, 0, kCommonArgCount
+};
+
+int InterfaceManager::actionValuelessArgumentCounts[kActionCount + 1] = {
+ kFlashArgPit, 0, kDumpArgChipType, 0, kCommonArgDelay
+};
+
+const char *InterfaceManager::usage = "Usage: heimdall <action> <arguments> [--verbose] [--delay <ms>]\n\
+\n\
+action: flash\n\
+arguments:\n\
+ --repartition --pit <filename> --factoryfs <filename>\n\
+ --cache <filename> --dbdata <filename> --primary-boot <filename>\n\
+ --secondary-boot <filename> --param <filename> --kernel <filename>\n\
+ --modem <filename>\n\
+ or:\n\
+ [--pit <filename>] [--factoryfs <filename>] [--cache <filename>]\n\
+ [--dbdata <filename>] [--primary-boot <filename>]\n\
+ [--secondary-boot <filename>] [--secondary-boot-backup <filename>]\n\
+ [--param <filename>] [--kernel <filename>] [--recovery <filename>]\n\
+ [--efs <filename>] [--modem <filename>]\n\
+description: Flashes firmware files to your phone.\n\
+\n\
+action: close-pc-screen\n\
+description: Attempts to get rid off the \"connect phone to PC\" screen.\n\
+\n\
+action: dump\n\
+arguments: --chip-type <NAND | RAM> --chip-id <integer> --output <filename>\n\
+description: Attempts to dump data from the phone corresponding to the\n\
+ specified chip type and chip ID.\n\
+NOTE: Galaxy S phones don't appear to properly support this functionality.\n\
+\n\
+action: help\n\
+description: Display this dialogue.\n";
+
+bool InterfaceManager::GetArguments(int argc, char **argv, map<string, string>& argumentMap, int *actionIndex)
+{
+ if (argc < 2)
+ {
+ Print(usage);
+ return (false);
+ }
+
+ const char *actionName = argv[1];
+ *actionIndex = -1;
+
+ for (int i = 0; i < kActionCount; i++)
+ {
+ if (actionNames[i] == actionName)
+ {
+ *actionIndex = i;
+ break;
+ }
+ }
+
+ if (*actionIndex < 0)
+ {
+ Print("Unknown action \"%s\"\n\n", actionName);
+ Print(usage);
+ return (false);
+ }
+
+ int actionArgumentCount = actionArgumentCounts[*actionIndex];
+ int commonArgumentCount = actionArgumentCounts[kActionCount];
+
+ int actionValuelessArgumentCount = actionValuelessArgumentCounts[*actionIndex];
+ int commonValuelessArgumentCount = actionValuelessArgumentCounts[kActionCount];
+
+ string *argumentNames = actionArgumentNames[*actionIndex];
+ string *commonArgumentNames = actionArgumentNames[kActionCount];
+
+ for (int argIndex = 2; argIndex < argc; argIndex++)
+ {
+ if (*(argv[argIndex]) != '-')
+ {
+ Print(usage);
+ return (false);
+ }
+
+ string argumentName = (char *)(argv[argIndex] + 1);
+
+ // Check if the argument is a valid valueless argument
+ bool valid = false;
+
+ for (int i = 0; i < actionValuelessArgumentCount; i++)
+ {
+ if (argumentName == argumentNames[i] || argumentName == argumentNames[actionArgumentCount + i])
+ {
+ argumentName = argumentNames[i];
+ valid = true;
+ break;
+ }
+ }
+
+ if (!valid)
+ {
+ // Check if it's a common valueless argument
+ for (int i = 0; i < commonValuelessArgumentCount; i++)
+ {
+ if (argumentName == commonArgumentNames[i] || argumentName == commonArgumentNames[commonArgumentCount + i])
+ {
+ argumentName = commonArgumentNames[i];
+ valid = true;
+ break;
+ }
+ }
+ }
+
+ if (valid)
+ {
+ // The argument is valueless
+ argumentMap.insert(pair<string, string>(argumentName, ""));
+ continue;
+ }
+
+ // Check if the argument is a valid regular argument
+ for (int i = actionValuelessArgumentCount; i < actionArgumentCount; i++)
+ {
+ if (argumentName == argumentNames[i] || argumentName == argumentNames[actionArgumentCount + i])
+ {
+ argumentName = argumentNames[i];
+ valid = true;
+ break;
+ }
+ }
+
+ if (!valid)
+ {
+ // Check if it's a common regular argument
+ for (int i = commonValuelessArgumentCount; i < commonArgumentCount; i++)
+ {
+ if (argumentName == commonArgumentNames[i] || argumentName == commonArgumentNames[commonArgumentCount + i])
+ {
+ argumentName = commonArgumentNames[i];
+ valid = true;
+ break;
+ }
+ }
+ }
+
+ if (!valid)
+ {
+ PrintError("\"%s\" is not a valid argument\n", argumentName.c_str());
+ return (false);
+ }
+
+ argIndex++;
+
+ if (argIndex >= argc)
+ {
+ PrintError("\"%s\" is missing a value\n", argumentName.c_str());
+ return (false);
+ }
+
+ argumentMap.insert(pair<string, string>(argumentName, argv[argIndex]));
+ }
+
+ return (true);
+}
+
+void InterfaceManager::Print(const char *format, ...)
+{
+ va_list args;
+
+ va_start(args, format);
+ vfprintf(stdout, format, args);
+ va_end(args);
+
+ fflush(stdout); // Make sure output isn't buffered.
+}
+
+void InterfaceManager::PrintError(const char *format, ...)
+{
+ va_list args;
+
+ va_start(args, format);
+ vfprintf(stderr, format, args);
+ va_end(args);
+
+ fflush(stderr); // Make sure output isn't buffered.
+}
|