1 files changed, 225 insertions, 0 deletions
diff --git a/lineformatter.cpp b/lineformatter.cpp
new file mode 100644
index 00000000..71da6b6e
--- /dev/null
+++ b/lineformatter.cpp
@@ -0,0 +1,225 @@
+// 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;
+}