1
0
mirror of synced 2024-12-05 04:27:56 +01:00
ImHex/plugins/builtin/source/content/pl_builtin_functions.cpp

95 lines
3.7 KiB
C++
Raw Normal View History

#include <hex/api/imhex_api.hpp>
#include <hex/api/content_registry.hpp>
#include <hex/providers/provider.hpp>
#include <hex/helpers/http_requests.hpp>
2021-08-29 22:15:18 +02:00
#include <pl/core/token.hpp>
#include <pl/core/evaluator.hpp>
2022-09-30 12:49:49 +02:00
#include <llvm/Demangle/Demangle.h>
#include <pl/patterns/pattern.hpp>
namespace hex::plugin::builtin {
void registerPatternLanguageFunctions() {
using namespace pl::core;
using FunctionParameterCount = pl::api::FunctionParameterCount;
{
2023-11-10 20:47:08 +01:00
const pl::api::Namespace nsHexCore = { "builtin", "hex", "core" };
/* get_selection() */
2022-09-30 12:49:49 +02:00
ContentRegistry::PatternLanguage::addFunction(nsHexCore, "get_selection", FunctionParameterCount::none(), [](Evaluator *, auto) -> std::optional<Token::Literal> {
if (!ImHexApi::HexEditor::isSelectionValid())
return std::numeric_limits<u128>::max();
auto selection = ImHexApi::HexEditor::getSelection();
return u128(u128(selection->getStartAddress()) << 64 | u128(selection->getSize()));
});
/* add_virtual_file(path, pattern) */
ContentRegistry::PatternLanguage::addFunction(nsHexCore, "add_virtual_file", FunctionParameterCount::exactly(2), [](Evaluator *, auto params) -> std::optional<Token::Literal> {
auto path = params[0].toString(false);
auto pattern = params[1].toPattern();
Region region = Region::Invalid();
if (pattern->getSection() == pl::ptrn::Pattern::MainSectionId)
region = Region(pattern->getOffset(), pattern->getSize());
ImHexApi::HexEditor::addVirtualFile(path, pattern->getBytes(), region);
return std::nullopt;
});
}
{
2023-11-10 20:47:08 +01:00
const pl::api::Namespace nsHexPrv = { "builtin", "hex", "prv" };
/* get_information() */
ContentRegistry::PatternLanguage::addFunction(nsHexPrv, "get_information", FunctionParameterCount::between(1, 2), [](Evaluator *, auto params) -> std::optional<Token::Literal> {
std::string category = params[0].toString(false);
std::string argument = params.size() == 2 ? params[1].toString(false) : "";
if (!ImHexApi::Provider::isValid())
return u128(0);
auto provider = ImHexApi::Provider::get();
if (!provider->isAvailable())
return u128(0);
return std::visit(
[](auto &&value) -> Token::Literal {
return value;
},
provider->queryInformation(category, argument)
);
});
}
2022-09-30 12:49:49 +02:00
{
2023-11-10 20:47:08 +01:00
const pl::api::Namespace nsHexDec = { "builtin", "hex", "dec" };
2022-09-30 12:49:49 +02:00
/* demangle(mangled_string) */
ContentRegistry::PatternLanguage::addFunction(nsHexDec, "demangle", FunctionParameterCount::exactly(1), [](Evaluator *, auto params) -> std::optional<Token::Literal> {
2023-01-23 13:53:29 +01:00
const auto mangledString = params[0].toString(false);
2022-09-30 12:49:49 +02:00
return llvm::demangle(mangledString);
});
}
2021-09-25 16:24:08 +02:00
{
2023-11-10 20:47:08 +01:00
const pl::api::Namespace nsHexHttp = { "builtin", "hex", "http" };
2021-09-25 16:24:08 +02:00
/* get(url) */
ContentRegistry::PatternLanguage::addDangerousFunction(nsHexHttp, "get", FunctionParameterCount::exactly(1), [](Evaluator *, auto params) -> std::optional<Token::Literal> {
2023-01-23 13:53:29 +01:00
const auto url = params[0].toString(false);
2021-09-25 16:24:08 +02:00
hex::HttpRequest request("GET", url);
return request.execute().get().getData();
});
2021-09-25 16:24:08 +02:00
}
}
}