Add line formatter API and a generic line formatter plugin

2 files changed, 6 insertions, 5 deletions

diff --git a/lang/rust/pseudorust.cpp b/lang/rust/pseudorust.cpp
index 837a7932..a1f7defe 100644
--- a/lang/rust/pseudorust.cpp
+++ b/lang/rust/pseudorust.cpp
@@ -7,9 +7,9 @@ using namespace std;
 using namespace BinaryNinja;
 
 
-PseudoRustFunction::PseudoRustFunction(
-	Architecture* arch, Function* owner, HighLevelILFunction* highLevelILFunction) :
-	LanguageRepresentationFunction(arch, owner, highLevelILFunction), m_highLevelIL(highLevelILFunction)
+PseudoRustFunction::PseudoRustFunction(LanguageRepresentationFunctionType* type, Architecture* arch, Function* owner,
+	HighLevelILFunction* highLevelILFunction) :
+	LanguageRepresentationFunction(type, arch, owner, highLevelILFunction), m_highLevelIL(highLevelILFunction)
 {
 }
 
@@ -2814,7 +2814,7 @@ PseudoRustFunctionType::PseudoRustFunctionType(): LanguageRepresentationFunction
 Ref<LanguageRepresentationFunction> PseudoRustFunctionType::Create(Architecture* arch, Function* owner,
 	HighLevelILFunction* highLevelILFunction)
 {
-	return new PseudoRustFunction(arch, owner, highLevelILFunction);
+	return new PseudoRustFunction(this, arch, owner, highLevelILFunction);
 }
 
 
diff --git a/lang/rust/pseudorust.h b/lang/rust/pseudorust.h
index a2ab9584..96e656c2 100644
--- a/lang/rust/pseudorust.h
+++ b/lang/rust/pseudorust.h
@@ -59,7 +59,8 @@ protected:
 	virtual void EndLines(const BinaryNinja::HighLevelILInstruction& instr, BinaryNinja::HighLevelILTokenEmitter& tokens) override;
 
 public:
-	PseudoRustFunction(BinaryNinja::Architecture* arch, BinaryNinja::Function* owner, BinaryNinja::HighLevelILFunction* highLevelILFunction);
+	PseudoRustFunction(BinaryNinja::LanguageRepresentationFunctionType* type, BinaryNinja::Architecture* arch,
+		BinaryNinja::Function* owner, BinaryNinja::HighLevelILFunction* highLevelILFunction);
 
 	std::string GetAnnotationStartString() const override;
 	std::string GetAnnotationEndString() const override;