diff --git a/tests/data/__init__.py b/tests/data/__init__.py index fa8722f..e9af56d 100644 --- a/tests/data/__init__.py +++ b/tests/data/__init__.py @@ -11,7 +11,13 @@ if system == 'Windows': elif system == 'Darwin': target_program = os.path.join(os.path.dirname(__file__), "unixvictim-macos") else: - arch = "x86_64" if sys.maxsize > 2**32 else "x86" + machine = platform.machine() + if machine == 'aarch64': + arch = 'arm64' + elif machine.startswith('arm'): + arch = 'armhf' + else: + arch = 'x86_64' if sys.maxsize > 2**32 else 'x86' target_program = os.path.join(os.path.dirname(__file__), "unixvictim-linux-" + arch) diff --git a/tests/data/unixvictim-linux-arm64 b/tests/data/unixvictim-linux-arm64 new file mode 100755 index 0000000..caa2dc4 Binary files /dev/null and b/tests/data/unixvictim-linux-arm64 differ diff --git a/tests/data/unixvictim-linux-armhf b/tests/data/unixvictim-linux-armhf new file mode 100755 index 0000000..c0536a9 Binary files /dev/null and b/tests/data/unixvictim-linux-armhf differ