From 05b8dae707da62437abf262fe8d1dd2b74a5f98f Mon Sep 17 00:00:00 2001
From: Rusty Wagner <rusty@vector35.com>
Date: Wed, 6 Jul 2016 22:58:22 -0400
Subject: Add API to get disassembly text for a basic block outside of a graph

---
 binaryninjaapi.h | 39 ++++++++++++++++++++++++++++-----------
 1 file changed, 28 insertions(+), 11 deletions(-)

(limited to 'binaryninjaapi.h')

diff --git a/binaryninjaapi.h b/binaryninjaapi.h
index 3f0b1f4b..9e8f516c 100644
--- a/binaryninjaapi.h
+++ b/binaryninjaapi.h
@@ -1380,6 +1380,28 @@ namespace BinaryNinja
 		void AddMemberWithValue(const std::string& name, uint64_t value);
 	};
 
+	class DisassemblySettings: public CoreRefCountObject<BNDisassemblySettings,
+		BNNewDisassemblySettingsReference, BNFreeDisassemblySettings>
+	{
+	public:
+		DisassemblySettings();
+		DisassemblySettings(BNDisassemblySettings* settings);
+
+		bool IsOptionSet(BNDisassemblyOption option) const;
+		void SetOption(BNDisassemblyOption option, bool state = true);
+
+		size_t GetWidth() const;
+		void SetWidth(size_t width);
+		size_t GetMaximumSymbolWidth() const;
+		void SetMaximumSymbolWidth(size_t width);
+	};
+
+	struct DisassemblyTextLine
+	{
+		uint64_t addr;
+		std::vector<InstructionTextToken> tokens;
+	};
+
 	class Function;
 
 	struct BasicBlockEdge
@@ -1407,6 +1429,8 @@ namespace BinaryNinja
 		void MarkRecentUse();
 
 		std::vector<std::vector<InstructionTextToken>> GetAnnotations();
+
+		std::vector<DisassemblyTextLine> GetDisassemblyText(DisassemblySettings* settings);
 	};
 
 	struct StackVariable
@@ -1527,12 +1551,6 @@ namespace BinaryNinja
 		std::vector<std::vector<InstructionTextToken>> GetBlockAnnotations(Architecture* arch, uint64_t addr);
 	};
 
-	struct FunctionGraphTextLine
-	{
-		uint64_t addr;
-		std::vector<InstructionTextToken> tokens;
-	};
-
 	struct FunctionGraphEdge
 	{
 		BNBranchType type;
@@ -1555,7 +1573,7 @@ namespace BinaryNinja
 		int GetWidth() const;
 		int GetHeight() const;
 
-		std::vector<FunctionGraphTextLine> GetLines() const;
+		std::vector<DisassemblyTextLine> GetLines() const;
 		std::vector<FunctionGraphEdge> GetOutgoingEdges() const;
 	};
 
@@ -1578,8 +1596,7 @@ namespace BinaryNinja
 		int GetVerticalBlockMargin() const;
 		void SetBlockMargins(int horiz, int vert);
 
-		size_t GetMaximumSymbolWidth() const;
-		void SetMaximumSymbolWidth(size_t width);
+		Ref<DisassemblySettings> GetSettings();
 
 		void StartLayout(BNFunctionGraphType = NormalFunctionGraph);
 		bool IsLayoutComplete();
@@ -1592,8 +1609,8 @@ namespace BinaryNinja
 		int GetHeight() const;
 		std::vector<Ref<FunctionGraphBlock>> GetBlocksInRegion(int left, int top, int right, int bottom);
 
-		bool IsOptionSet(BNFunctionGraphOption option) const;
-		void SetOption(BNFunctionGraphOption option, bool state = true);
+		bool IsOptionSet(BNDisassemblyOption option) const;
+		void SetOption(BNDisassemblyOption option, bool state = true);
 	};
 
 	struct LowLevelILLabel: public BNLowLevelILLabel
-- 
cgit v1.3.1