diff options
Diffstat (limited to 'stringrecognizer.cpp')
| -rw-r--r-- | stringrecognizer.cpp | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/stringrecognizer.cpp b/stringrecognizer.cpp index dc6ac2f8..47e0ca5e 100644 --- a/stringrecognizer.cpp +++ b/stringrecognizer.cpp @@ -136,6 +136,12 @@ std::optional<DerivedString> StringRecognizer::RecognizeImport(const HighLevelIL } +std::optional<DerivedString> StringRecognizer::RecognizeConstantData(const HighLevelILInstruction&) +{ + return std::nullopt; +} + + void StringRecognizer::Register(StringRecognizer* recognizer) { BNCustomStringRecognizer callbacks; @@ -145,6 +151,7 @@ void StringRecognizer::Register(StringRecognizer* recognizer) callbacks.recognizeConstantPointer = RecognizeConstantPointerCallback; callbacks.recognizeExternPointer = RecognizeExternPointerCallback; callbacks.recognizeImport = RecognizeImportCallback; + callbacks.recognizeConstantData = RecognizeConstantDataCallback; recognizer->AddRefForRegistration(); recognizer->m_object = BNRegisterStringRecognizer(recognizer->m_nameForRegister.c_str(), &callbacks); @@ -220,6 +227,20 @@ bool StringRecognizer::RecognizeImportCallback( } +bool StringRecognizer::RecognizeConstantDataCallback( + void* ctxt, BNHighLevelILFunction* hlil, size_t expr, BNDerivedString* result) +{ + StringRecognizer* recognizer = (StringRecognizer*)ctxt; + Ref<HighLevelILFunction> hlilObj = new HighLevelILFunction(BNNewHighLevelILFunctionReference(hlil)); + HighLevelILInstruction instr = hlilObj->GetExpr(expr); + auto str = recognizer->RecognizeConstantData(instr); + if (!str.has_value()) + return false; + *result = str->ToAPIObject(true); + return true; +} + + Ref<StringRecognizer> StringRecognizer::GetByName(const std::string& name) { BNStringRecognizer* recognizer = BNGetStringRecognizerByName(name.c_str()); @@ -298,3 +319,13 @@ std::optional<DerivedString> CoreStringRecognizer::RecognizeImport( return std::nullopt; return DerivedString::FromAPIObject(&str, true); } + + +std::optional<DerivedString> CoreStringRecognizer::RecognizeConstantData( + const HighLevelILInstruction& instr) +{ + BNDerivedString str; + if (!BNStringRecognizerRecognizeConstantData(m_object, instr.function->GetObject(), instr.exprIndex, &str)) + return std::nullopt; + return DerivedString::FromAPIObject(&str, true); +} |