summaryrefslogtreecommitdiff
path: root/python/function.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/function.py')
-rw-r--r--python/function.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/python/function.py b/python/function.py
index c860fced..9eac5728 100644
--- a/python/function.py
+++ b/python/function.py
@@ -769,8 +769,10 @@ class Function(object):
try:
if i < 0:
i = count.value + i
- if i < 0 or i >= count.value:
+ if i < -len(self.basic_blocks) or i >= count.value:
raise IndexError("index out of range")
+ if i < 0:
+ i = len(self.basic_blocks) + i
block = binaryninja.basicblock.BasicBlock(core.BNNewBasicBlockReference(blocks[i]), self._view)
return block
finally: