Adding API to retrieve basic block annotations

1 files changed, 25 insertions, 0 deletions

diff --git a/function.cpp b/function.cpp
index 7cc30351..70cbfe2a 100644
--- a/function.cpp
+++ b/function.cpp
@@ -458,3 +458,28 @@ vector<IndirectBranchInfo> Function::GetIndirectBranchesAt(Architecture* arch, u
 	BNFreeIndirectBranchList(branches);
 	return result;
 }
+
+
+vector<vector<InstructionTextToken>> Function::GetBlockAnnotations(Architecture* arch, uint64_t addr)
+{
+	size_t count;
+	BNInstructionTextLine* lines = BNGetFunctionBlockAnnotations(m_object, arch->GetObject(), addr, &count);
+
+	vector<vector<InstructionTextToken>> result;
+	for (size_t i = 0; i < count; i++)
+	{
+		vector<InstructionTextToken> line;
+		for (size_t j = 0; j < lines[i].count; j++)
+		{
+			InstructionTextToken token;
+			token.type = lines[i].tokens[j].type;
+			token.text = lines[i].tokens[j].text;
+			token.value = lines[i].tokens[j].value;
+			line.push_back(token);
+		}
+		result.push_back(line);
+	}
+
+	BNFreeInstructionTextLines(lines, count);
+	return result;
+}