diff --git a/scripts/testbed/run_rule_on_testbed.py b/scripts/testbed/run_rule_on_testbed.py index aa78a830..61b10305 100644 --- a/scripts/testbed/run_rule_on_testbed.py +++ b/scripts/testbed/run_rule_on_testbed.py @@ -18,9 +18,12 @@ import argparse import capa.main import capa.rules import capa.features.freeze -from scripts.testbed import FNAMES_EXTENSION, FREEZE_EXTENSION +from utils import FNAMES_EXTENSION, FREEZE_EXTENSION from start_ida_export_fimages import export_fimages +# enable import of code in the same directory +sys.path.append(os.path.dirname(__file__)) + logger = logging.getLogger(__name__) # sorry globals... diff --git a/scripts/testbed/start_ida_dump_fnames.py b/scripts/testbed/start_ida_dump_fnames.py index b701d509..bea8c892 100644 --- a/scripts/testbed/start_ida_dump_fnames.py +++ b/scripts/testbed/start_ida_dump_fnames.py @@ -17,7 +17,10 @@ import subprocess import argparse -from scripts.testbed import FNAMES_EXTENSION +from utils import FNAMES_EXTENSION + +# enable import of code in the same directory +sys.path.append(os.path.dirname(__file__)) IDA32_PATH = "C:\\Program Files\\IDA Pro 7.3\\ida.exe" IDA64_PATH = "C:\\Program Files\\IDA Pro 7.3\\ida64.exe" diff --git a/scripts/testbed/__init__.py b/scripts/testbed/utils.py similarity index 100% rename from scripts/testbed/__init__.py rename to scripts/testbed/utils.py