Rooba
/
ImHex
mirror of https://github.com/WerWolv/ImHex.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

patterns: If a main function exists, automatically call it

This commit is contained in:
WerWolv 2021-10-08 18:47:05 +02:00
parent 17db605b17
commit a29e3789d2
1 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
plugins/libimhex/source/pattern_language/evaluator.cpp
View File

@ -119,6 +119,7 @@ namespace hex::pl {
try { try {
pushScope(nullptr, patterns); pushScope(nullptr, patterns);
for (auto node : ast) { for (auto node : ast) {
if (dynamic_cast<ASTNodeTypeDecl*>(node)) { if (dynamic_cast<ASTNodeTypeDecl*>(node)) {
;// Don't create patterns from type declarations ;// Don't create patterns from type declarations
@ -130,8 +131,23 @@ namespace hex::pl {
auto newPatterns = node->createPatterns(this); auto newPatterns = node->createPatterns(this);
patterns.insert(patterns.end(), newPatterns.begin(), newPatterns.end()); patterns.insert(patterns.end(), newPatterns.begin(), newPatterns.end());
} }
} }
if (this->m_customFunctions.contains("main")) {
auto mainFunction = this->m_customFunctions["main"];
if (mainFunction.parameterCount > 0)
LogConsole::abortEvaluation("main function may not accept any arguments");
auto result = mainFunction.func(this, { });
if (result) {
auto returnCode = Token::literalToSigned(*result);
if (returnCode != 0)
LogConsole::abortEvaluation(hex::format("non-success value returned from main: {}", returnCode));
}
}
popScope(); popScope();
} catch (const LogConsole::EvaluateError &error) { } catch (const LogConsole::EvaluateError &error) {
this->m_console.log(LogConsole::Level::Error, error.second); this->m_console.log(LogConsole::Level::Error, error.second);