From 19d42b4d5ce3802e59560791aa1caab19f2219b0 Mon Sep 17 00:00:00 2001 From: Brian Potchik Date: Sat, 21 Sep 2019 22:02:39 -0400 Subject: Add unit tests for address rebasing. --- suite/testcommon.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'suite/testcommon.py') diff --git a/suite/testcommon.py b/suite/testcommon.py index 782f5e8d..6670d67c 100644 --- a/suite/testcommon.py +++ b/suite/testcommon.py @@ -98,9 +98,12 @@ class BinaryViewTestBuilder(Builder): - Function doc string used as 'on error' message - Should return: list of strings """ - def __init__(self, filename): + def __init__(self, filename, imageBase=None): self.filename = os.path.join(os.path.dirname(__file__), filename) - self.bv = BinaryViewType.get_view_of_file(self.filename) + if imageBase is None: + self.bv = BinaryViewType.get_view_of_file(self.filename) + else: + self.bv = BinaryViewType.get_view_of_file_with_options(self.filename, options={'loader.imageBase' : imageBase}) if self.bv is None: print("%s is not an executable format" % filename) return -- cgit v1.3.1