From 9bb1a15b800da0ac5b60c40e3b10321e4efc5a1b Mon Sep 17 00:00:00 2001
From: rollsafe <rollsafe@users.noreply.github.com>
Date: Tue, 30 Jul 2019 21:48:38 -0400
Subject: Cache function callee addresses and create an API to access them

---
 binaryninjaapi.h | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'binaryninjaapi.h')

diff --git a/binaryninjaapi.h b/binaryninjaapi.h
index 1e5ab511..25c32b1c 100644
--- a/binaryninjaapi.h
+++ b/binaryninjaapi.h
@@ -1436,6 +1436,9 @@ namespace BinaryNinja
 		void AddUserDataReference(uint64_t fromAddr, uint64_t toAddr);
 		void RemoveUserDataReference(uint64_t fromAddr, uint64_t toAddr);
 
+		std::vector<uint64_t> GetCallees(ReferenceSource addr);
+		std::vector<ReferenceSource> GetCallers(uint64_t addr);
+
 		Ref<Symbol> GetSymbolByAddress(uint64_t addr, const NameSpace& nameSpace=NameSpace());
 		Ref<Symbol> GetSymbolByRawName(const std::string& name, const NameSpace& nameSpace=NameSpace());
 		std::vector<Ref<Symbol>> GetSymbolsByName(const std::string& name, const NameSpace& nameSpace=NameSpace());
@@ -2618,6 +2621,8 @@ namespace BinaryNinja
 		void SetComment(const std::string& comment);
 		void SetCommentForAddress(uint64_t addr, const std::string& comment);
 
+		std::vector<ReferenceSource> GetCallSites() const;
+
 		void AddUserCodeReference(Architecture* fromArch, uint64_t fromAddr, uint64_t toAddr);
 		void RemoveUserCodeReference(Architecture* fromArch, uint64_t fromAddr, uint64_t toAddr);
 
-- 
cgit v1.3.1