diff options
Diffstat (limited to 'lineformatter.cpp')
| -rw-r--r-- | lineformatter.cpp | 225 |
1 files changed, 0 insertions, 225 deletions
diff --git a/lineformatter.cpp b/lineformatter.cpp deleted file mode 100644 index 71da6b6e..00000000 --- a/lineformatter.cpp +++ /dev/null @@ -1,225 +0,0 @@ -// Copyright (c) 2024 Vector 35 Inc -// -// 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. - -#include "binaryninjaapi.h" - -using namespace BinaryNinja; -using namespace std; - - -LineFormatterSettings LineFormatterSettings::GetDefault(DisassemblySettings* settings, HighLevelILFunction* func) -{ - BNLineFormatterSettings* apiObj = - BNGetDefaultLineFormatterSettings(settings ? settings->GetObject() : nullptr, func->GetObject()); - LineFormatterSettings result = FromAPIObject(apiObj); - BNFreeLineFormatterSettings(apiObj); - return result; -} - - -LineFormatterSettings LineFormatterSettings::GetLanguageRepresentationSettings( - DisassemblySettings* settings, LanguageRepresentationFunction* func) -{ - BNLineFormatterSettings* apiObj = - BNGetLanguageRepresentationLineFormatterSettings(settings ? settings->GetObject() : nullptr, func->GetObject()); - LineFormatterSettings result = FromAPIObject(apiObj); - BNFreeLineFormatterSettings(apiObj); - return result; -} - - -LineFormatterSettings LineFormatterSettings::FromAPIObject(const BNLineFormatterSettings* settings) -{ - LineFormatterSettings result; - result.highLevelIL = new HighLevelILFunction(BNNewHighLevelILFunctionReference(settings->highLevelIL)); - result.desiredLineLength = settings->desiredLineLength; - result.minimumContentLength = settings->minimumContentLength; - result.tabWidth = settings->tabWidth; - result.languageName = settings->languageName; - result.commentStartString = settings->commentStartString; - result.commentEndString = settings->commentEndString; - result.annotationStartString = settings->annotationStartString; - result.annotationEndString = settings->annotationEndString; - return result; -} - - -BNLineFormatterSettings LineFormatterSettings::ToAPIObject() const -{ - BNLineFormatterSettings result; - result.highLevelIL = highLevelIL->GetObject(); - result.desiredLineLength = desiredLineLength; - result.minimumContentLength = minimumContentLength; - result.tabWidth = tabWidth; - result.languageName = (char*)languageName.c_str(); - result.commentStartString = (char*)commentStartString.c_str(); - result.commentEndString = (char*)commentEndString.c_str(); - result.annotationStartString = (char*)annotationStartString.c_str(); - result.annotationEndString = (char*)annotationEndString.c_str(); - return result; -} - - -LineFormatter::LineFormatter(const string& name) : m_nameForRegister(name) {} - - -LineFormatter::LineFormatter(BNLineFormatter* formatter) -{ - m_object = formatter; -} - - -void LineFormatter::Register(LineFormatter* formatter) -{ - BNCustomLineFormatter cb; - cb.context = formatter; - cb.formatLines = FormatLinesCallback; - cb.freeLines = FreeLinesCallback; - - formatter->AddRefForRegistration(); - formatter->m_object = BNRegisterLineFormatter(formatter->m_nameForRegister.c_str(), &cb); -} - - -BNDisassemblyTextLine* LineFormatter::FormatLinesCallback(void* ctxt, BNDisassemblyTextLine* inLines, size_t inCount, - const BNLineFormatterSettings* settings, size_t* outCount) -{ - LineFormatter* formatter = (LineFormatter*)ctxt; - - vector<DisassemblyTextLine> input; - input.reserve(inCount); - for (size_t i = 0; i < inCount; i++) - { - DisassemblyTextLine line; - line.addr = inLines[i].addr; - line.instrIndex = inLines[i].instrIndex; - line.highlight = inLines[i].highlight; - line.tokens = InstructionTextToken::ConvertInstructionTextTokenList(inLines[i].tokens, inLines[i].count); - line.tags = Tag::ConvertTagList(inLines[i].tags, inLines[i].tagCount); - input.push_back(line); - } - - vector<DisassemblyTextLine> outLines = - formatter->FormatLines(input, LineFormatterSettings::FromAPIObject(settings)); - - *outCount = outLines.size(); - BNDisassemblyTextLine* buf = new BNDisassemblyTextLine[outLines.size()]; - for (size_t i = 0; i < outLines.size(); i++) - { - const DisassemblyTextLine& line = outLines[i]; - buf[i].addr = line.addr; - buf[i].instrIndex = line.instrIndex; - buf[i].highlight = line.highlight; - buf[i].tokens = InstructionTextToken::CreateInstructionTextTokenList(line.tokens); - buf[i].count = line.tokens.size(); - buf[i].tags = Tag::CreateTagList(line.tags, &(buf[i].tagCount)); - } - - return buf; -} - - -void LineFormatter::FreeLinesCallback(void*, BNDisassemblyTextLine* lines, size_t count) -{ - for (size_t i = 0; i < count; i++) - { - InstructionTextToken::FreeInstructionTextTokenList(lines[i].tokens, lines[i].count); - Tag::FreeTagList(lines[i].tags, lines[i].tagCount); - } - delete[] lines; -} - - -vector<Ref<LineFormatter>> LineFormatter::GetList() -{ - size_t count; - BNLineFormatter** list = BNGetLineFormatterList(&count); - vector<Ref<LineFormatter>> result; - for (size_t i = 0; i < count; i++) - result.push_back(new CoreLineFormatter(list[i])); - BNFreeLineFormatterList(list); - return result; -} - - -Ref<LineFormatter> LineFormatter::GetByName(const string& name) -{ - BNLineFormatter* result = BNGetLineFormatterByName(name.c_str()); - if (!result) - return nullptr; - return new CoreLineFormatter(result); -} - - -Ref<LineFormatter> LineFormatter::GetDefault() -{ - BNLineFormatter* result = BNGetDefaultLineFormatter(); - if (!result) - return nullptr; - return new CoreLineFormatter(result); -} - - -CoreLineFormatter::CoreLineFormatter(BNLineFormatter* formatter) : LineFormatter(formatter) {} - - -vector<DisassemblyTextLine> CoreLineFormatter::FormatLines( - const vector<DisassemblyTextLine>& lines, const LineFormatterSettings& settings) -{ - size_t inCount = lines.size(); - BNDisassemblyTextLine* inLines = new BNDisassemblyTextLine[lines.size()]; - for (size_t i = 0; i < lines.size(); i++) - { - const DisassemblyTextLine& line = lines[i]; - inLines[i].addr = line.addr; - inLines[i].instrIndex = line.instrIndex; - inLines[i].highlight = line.highlight; - inLines[i].tokens = InstructionTextToken::CreateInstructionTextTokenList(line.tokens); - inLines[i].count = line.tokens.size(); - inLines[i].tags = Tag::CreateTagList(line.tags, &(inLines[i].tagCount)); - } - - size_t outCount = 0; - BNLineFormatterSettings apiSettings = settings.ToAPIObject(); - BNDisassemblyTextLine* outLines = BNFormatLines(m_object, inLines, inCount, &apiSettings, &outCount); - - for (size_t i = 0; i < inCount; i++) - { - InstructionTextToken::FreeInstructionTextTokenList(inLines[i].tokens, inLines[i].count); - Tag::FreeTagList(inLines[i].tags, inLines[i].tagCount); - } - delete[] inLines; - - vector<DisassemblyTextLine> result; - result.reserve(outCount); - for (size_t i = 0; i < outCount; i++) - { - DisassemblyTextLine line; - line.addr = outLines[i].addr; - line.instrIndex = outLines[i].instrIndex; - line.highlight = outLines[i].highlight; - line.tokens = InstructionTextToken::ConvertInstructionTextTokenList(outLines[i].tokens, outLines[i].count); - line.tags = Tag::ConvertTagList(outLines[i].tags, outLines[i].tagCount); - result.push_back(line); - } - - BNFreeDisassemblyTextLines(outLines, outCount); - return result; -} |