352 lines
13 KiB
C++
352 lines
13 KiB
C++
#include "content/command_line_interface.hpp"
|
|
|
|
#include <content/providers/file_provider.hpp>
|
|
#include <hex/api/imhex_api.hpp>
|
|
#include <hex/api/event_manager.hpp>
|
|
|
|
#include <hex/helpers/fmt.hpp>
|
|
|
|
#include <hex/helpers/magic.hpp>
|
|
#include <hex/helpers/crypto.hpp>
|
|
#include <hex/helpers/literals.hpp>
|
|
#include <hex/helpers/utils.hpp>
|
|
#include <romfs/romfs.hpp>
|
|
|
|
#include <hex/api/plugin_manager.hpp>
|
|
#include <hex/api/task_manager.hpp>
|
|
#include <hex/subcommands/subcommands.hpp>
|
|
|
|
#include <wolv/utils/string.hpp>
|
|
#include <wolv/math_eval/math_evaluator.hpp>
|
|
|
|
#include <pl/cli/cli.hpp>
|
|
#include <llvm/Demangle/Demangle.h>
|
|
|
|
namespace hex::plugin::builtin {
|
|
using namespace hex::literals;
|
|
|
|
void handleVersionCommand(const std::vector<std::string> &args) {
|
|
hex::unused(args);
|
|
|
|
hex::log::print(std::string(romfs::get("logo.ans").string()),
|
|
ImHexApi::System::getImHexVersion(),
|
|
ImHexApi::System::getCommitBranch(), ImHexApi::System::getCommitHash(),
|
|
__DATE__, __TIME__,
|
|
ImHexApi::System::isPortableVersion() ? "Portable" : "Installed");
|
|
|
|
std::exit(EXIT_SUCCESS);
|
|
}
|
|
|
|
void handleHelpCommand(const std::vector<std::string> &args) {
|
|
hex::unused(args);
|
|
|
|
hex::log::print(
|
|
"ImHex - A Hex Editor for Reverse Engineers, Programmers and people who value their retinas when working at 3 AM.\n"
|
|
"\n"
|
|
"usage: imhex [subcommand] [options]\n"
|
|
"Available subcommands:\n"
|
|
);
|
|
|
|
size_t longestLongCommand = 0, longestShortCommand = 0;
|
|
for (const auto &plugin : PluginManager::getPlugins()) {
|
|
for (const auto &subCommand : plugin.getSubCommands()) {
|
|
longestLongCommand = std::max(longestLongCommand, subCommand.commandLong.size());
|
|
longestShortCommand = std::max(longestShortCommand, subCommand.commandShort.size());
|
|
}
|
|
}
|
|
|
|
for (const auto &plugin : PluginManager::getPlugins()) {
|
|
for (const auto &subCommand : plugin.getSubCommands()) {
|
|
hex::log::println(" "
|
|
"{}"
|
|
"{: <{}}"
|
|
"{}"
|
|
"{}"
|
|
"{: <{}}"
|
|
"{}",
|
|
subCommand.commandShort.empty() ? " " : "-",
|
|
subCommand.commandShort, longestShortCommand,
|
|
subCommand.commandShort.empty() ? " " : ", ",
|
|
subCommand.commandLong.empty() ? " " : "--",
|
|
subCommand.commandLong, longestLongCommand + 5,
|
|
subCommand.commandDescription
|
|
);
|
|
}
|
|
}
|
|
|
|
std::exit(EXIT_SUCCESS);
|
|
}
|
|
|
|
void handleOpenCommand(const std::vector<std::string> &args) {
|
|
if (args.empty()) {
|
|
hex::log::println("No files provided to open.");
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
|
|
std::vector<std::string> fullPaths;
|
|
bool doubleDashFound = false;
|
|
for (auto &arg : args) {
|
|
|
|
// Skip the first argument named `--`
|
|
if (arg == "--" && !doubleDashFound) {
|
|
doubleDashFound = true;
|
|
} else {
|
|
try {
|
|
auto path = std::filesystem::weakly_canonical(arg);
|
|
fullPaths.push_back(wolv::util::toUTF8String(path));
|
|
} catch (std::exception &e) {
|
|
log::error("Failed to open file '{}'\n {}", arg, e.what());
|
|
}
|
|
}
|
|
}
|
|
|
|
hex::subcommands::forwardSubCommand("open", fullPaths);
|
|
}
|
|
|
|
void handleCalcCommand(const std::vector<std::string> &args) {
|
|
if (args.empty()) {
|
|
hex::log::println("No expression provided!");
|
|
hex::log::println("Example: imhex --calc \"5 * 7\"");
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
|
|
wolv::math_eval::MathEvaluator<long double> evaluator;
|
|
|
|
auto input = hex::format("{}", fmt::join(args, " "));
|
|
auto result = evaluator.evaluate(input);
|
|
|
|
if (!result.has_value())
|
|
hex::log::println("{}\n> '{}'", evaluator.getLastError().value(), input);
|
|
else
|
|
hex::log::println("{}", result.value());
|
|
|
|
std::exit(EXIT_SUCCESS);
|
|
}
|
|
|
|
void handlePluginsCommand(const std::vector<std::string> &args) {
|
|
|
|
if (args.empty()) {
|
|
hex::log::println("Loaded plugins:");
|
|
|
|
for (const auto &plugin : PluginManager::getPlugins()) {
|
|
hex::log::print("- \033[1m{}\033[0m", plugin.getPluginName());
|
|
|
|
hex::log::println(" by {}", plugin.getPluginAuthor());
|
|
|
|
hex::log::println(" \033[2;3m{}\033[0m", plugin.getPluginDescription());
|
|
}
|
|
|
|
std::exit(EXIT_SUCCESS);
|
|
} else {
|
|
for (const auto &arg : args) {
|
|
PluginManager::addLoadPath(reinterpret_cast<const char8_t*>(arg.c_str()));
|
|
}
|
|
}
|
|
}
|
|
|
|
void handleLanguageCommand(const std::vector<std::string> &args) {
|
|
if (args.empty()) {
|
|
hex::log::println("usage: imhex --language <language>");
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
|
|
ImHexApi::System::impl::addInitArgument("language", args[0]);
|
|
}
|
|
|
|
void handleVerboseCommand(const std::vector<std::string> &) {
|
|
hex::log::enableDebugLogging();
|
|
}
|
|
|
|
void handleHashCommand(const std::vector<std::string> &args) {
|
|
if (args.size() != 2) {
|
|
hex::log::println("usage: imhex --hash <algorithm> <file>");
|
|
hex::log::println("Available algorithms: md5, sha1, sha224, sha256, sha384, sha512");
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
|
|
const auto &algorithm = args[0];
|
|
const auto &filePath = std::fs::path(args[1]);
|
|
|
|
wolv::io::File file(filePath, wolv::io::File::Mode::Read);
|
|
if (!file.isValid()) {
|
|
hex::log::println("Failed to open file: {}", wolv::util::toUTF8String(filePath));
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
|
|
constexpr static auto toVector = [](const auto &data) {
|
|
return std::vector<u8>(data.begin(), data.end());
|
|
};
|
|
|
|
std::vector<u8> result;
|
|
if (algorithm == "md5") {
|
|
result = toVector(hex::crypt::md5(file.readVector()));
|
|
} else if (algorithm == "sha1") {
|
|
result = toVector(hex::crypt::sha1(file.readVector()));
|
|
} else if (algorithm == "sha224") {
|
|
result = toVector(hex::crypt::sha224(file.readVector()));
|
|
} else if (algorithm == "sha256") {
|
|
result = toVector(hex::crypt::sha256(file.readVector()));
|
|
} else if (algorithm == "sha384") {
|
|
result = toVector(hex::crypt::sha384(file.readVector()));
|
|
} else if (algorithm == "sha512") {
|
|
result = toVector(hex::crypt::sha512(file.readVector()));
|
|
} else {
|
|
hex::log::println("Unknown algorithm: {}", algorithm);
|
|
hex::log::println("Available algorithms: md5, sha1, sha224, sha256, sha384, sha512");
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
|
|
hex::log::println("{}({}) = {}", algorithm, wolv::util::toUTF8String(filePath.filename()), hex::crypt::encode16(result));
|
|
|
|
std::exit(EXIT_SUCCESS);
|
|
}
|
|
|
|
void handleEncodeCommand(const std::vector<std::string> &args) {
|
|
if (args.size() != 2) {
|
|
hex::log::println("usage: imhex --encode <algorithm> <string>");
|
|
hex::log::println("Available algorithms: base64, hex");
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
|
|
const auto &algorithm = args[0];
|
|
const auto &data = std::vector<u8>(args[1].begin(), args[1].end());
|
|
|
|
std::string result;
|
|
if (algorithm == "base64") {
|
|
auto base64 = hex::crypt::encode64(data);
|
|
result = std::string(base64.begin(), base64.end());
|
|
} else if (algorithm == "hex") {
|
|
result = hex::crypt::encode16(data);
|
|
} else {
|
|
hex::log::println("Unknown algorithm: {}", algorithm);
|
|
hex::log::println("Available algorithms: base64, hex");
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
|
|
hex::log::println("encode_{}({}) = {}", algorithm, args[1], result);
|
|
std::exit(EXIT_SUCCESS);
|
|
}
|
|
|
|
void handleDecodeCommand(const std::vector<std::string> &args) {
|
|
if (args.size() != 2) {
|
|
hex::log::println("usage: imhex --decode <algorithm> <string>");
|
|
hex::log::println("Available algorithms: base64, hex");
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
|
|
const auto &algorithm = args[0];
|
|
const auto &data = std::vector<u8>(args[1].begin(), args[1].end());
|
|
|
|
std::string result;
|
|
if (algorithm == "base64") {
|
|
auto base64 = hex::crypt::decode64(data);
|
|
result = std::string(base64.begin(), base64.end());
|
|
} else if (algorithm == "hex") {
|
|
auto base16 = hex::crypt::decode16(std::string(data.begin(), data.end()));
|
|
result = std::string(base16.begin(), base16.end());
|
|
} else {
|
|
hex::log::println("Unknown algorithm: {}", algorithm);
|
|
hex::log::println("Available algorithms: base64, hex");
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
|
|
hex::log::print("decode_{}({}) = {}", algorithm, args[1], result);
|
|
std::exit(EXIT_SUCCESS);
|
|
}
|
|
|
|
void handleMagicCommand(const std::vector<std::string> &args) {
|
|
if (args.size() != 2) {
|
|
hex::log::println("usage: imhex --magic <operation> <file>");
|
|
hex::log::println("Available operations: mime, desc");
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
|
|
if (!magic::compile()) {
|
|
hex::log::print("Failed to compile magic database!");
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
|
|
const auto &operation = args[0];
|
|
const auto &filePath = std::fs::path(args[1]);
|
|
|
|
wolv::io::File file(filePath, wolv::io::File::Mode::Read);
|
|
if (!file.isValid()) {
|
|
hex::log::println("Failed to open file: {}", wolv::util::toUTF8String(filePath));
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
|
|
auto data = file.readVector(std::min<size_t>(file.getSize(), 100_KiB));
|
|
|
|
if (operation == "mime") {
|
|
auto result = magic::getMIMEType(data);
|
|
hex::log::println("{}", result);
|
|
} else if (operation == "desc") {
|
|
auto result = magic::getDescription(data);
|
|
hex::log::println("{}", result);
|
|
} else {
|
|
hex::log::println("Unknown operation: {}", operation);
|
|
hex::log::println("Available operations: mime, desc");
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
|
|
std::exit(EXIT_SUCCESS);
|
|
}
|
|
|
|
void handlePatternLanguageCommand(const std::vector<std::string> &args) {
|
|
std::vector<std::string> processedArgs = args;
|
|
if (processedArgs.empty()) {
|
|
processedArgs.emplace_back("--help");
|
|
} else {
|
|
for (const auto &path : fs::getDefaultPaths(fs::ImHexPath::PatternsInclude))
|
|
processedArgs.emplace_back(hex::format("--includes={}", wolv::util::toUTF8String(path)));
|
|
}
|
|
|
|
std::exit(pl::cli::executeCommandLineInterface(processedArgs));
|
|
}
|
|
|
|
void handleHexdumpCommand(const std::vector<std::string> &args) {
|
|
if (args.size() < 1 || args.size() > 3) {
|
|
log::println("usage: imhex --hexdump <file> <offset> <size>");
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
|
|
std::fs::path filePath = reinterpret_cast<const char8_t*>(args[0].data());
|
|
|
|
FileProvider provider;
|
|
|
|
provider.setPath(filePath);
|
|
if (!provider.open()) {
|
|
log::println("Failed to open file '{}'", args[0]);
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
|
|
u64 startAddress = args.size() >= 2 ? std::stoull(args[1], nullptr, 0) : 0x00;
|
|
u64 size = args.size() >= 3 ? std::stoull(args[2], nullptr, 0) : provider.getActualSize();
|
|
|
|
size = std::min<u64>(size, provider.getActualSize());
|
|
|
|
log::print("{}", hex::generateHexView(startAddress, size - startAddress, &provider));
|
|
|
|
std::exit(EXIT_SUCCESS);
|
|
}
|
|
|
|
void handleDemangleCommand(const std::vector<std::string> &args) {
|
|
if (args.size() != 1) {
|
|
log::println("usage: imhex --demangle <identifier>");
|
|
std::exit(EXIT_FAILURE);
|
|
}
|
|
|
|
log::println("{}", llvm::demangle(args[0]));
|
|
std::exit(EXIT_SUCCESS);
|
|
}
|
|
|
|
|
|
void registerCommandForwarders() {
|
|
hex::subcommands::registerSubCommand("open", [](const std::vector<std::string> &args){
|
|
for (auto &arg : args) {
|
|
RequestOpenFile::post(arg);
|
|
}
|
|
});
|
|
}
|
|
|
|
} |