From ce335c283f87297855bd9360e7063f3b8ef34afc Mon Sep 17 00:00:00 2001
From: Peter LaFosse <peter@vector35.com>
Date: Thu, 23 Jun 2022 13:57:56 -0400
Subject: Add some helper routines for getting the IL of callsites

---
 python/function.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

(limited to 'python/function.py')

diff --git a/python/function.py b/python/function.py
index 4d859683..c43a5868 100644
--- a/python/function.py
+++ b/python/function.py
@@ -3120,7 +3120,7 @@ class Function:
 		``callers`` returns a list of functions that call this function
 		Does not point to the actual address where the call occurs, just the start of the function that contains the call.
 
-		:return: List of start addresess for Functions that call this function
+		:return: List of start addresses for Functions that call this function
 		:rtype: list(int)
 		"""
 		functions = []
@@ -3129,6 +3129,17 @@ class Function:
 				functions.append(ref.function)
 		return functions
 
+	@property
+	def caller_sites(self) -> Generator['ReferenceSource', None, None]:
+		"""
+		``caller_sites`` returns a list of ReferenceSource objects corresponding to the addresses
+		in functions which reference this function
+
+		:return: List of start ReferenceSource objects of the call sites to this function
+		:rtype: list(int)
+		"""
+		return self.view.get_code_refs(self.start)
+
 	@property
 	def workflow(self):
 		handle = core.BNGetWorkflowForFunction(self.handle)
-- 
cgit v1.3.1