diff --git a/ofrak_core/ofrak/core/binwalk.py b/ofrak_core/ofrak/core/binwalk.py index 6a87d3723..f8e621ba4 100644 --- a/ofrak_core/ofrak/core/binwalk.py +++ b/ofrak_core/ofrak/core/binwalk.py @@ -31,6 +31,8 @@ def __init__(self): "binwalk", "https://github.com/ReFirmLabs/binwalk", install_check_arg="", + apt_package="binwalk", + brew_package="binwalk", ) async def is_tool_installed(self) -> bool: diff --git a/ofrak_core/ofrak/core/squashfs.py b/ofrak_core/ofrak/core/squashfs.py index 10bdf645a..832029d04 100644 --- a/ofrak_core/ofrak/core/squashfs.py +++ b/ofrak_core/ofrak/core/squashfs.py @@ -18,13 +18,23 @@ LOGGER = logging.getLogger(__name__) MKSQUASHFS = ComponentExternalTool( - "mksquashfs", "https://github.com/plougher/squashfs-tools", "-help" + "mksquashfs", + "https://github.com/plougher/squashfs-tools", + "-help", + apt_package="squashfs-tools", + brew_package="squashfs", ) class _UnsquashfsV45Tool(ComponentExternalTool): def __init__(self): - super().__init__("unsquashfs", "https://github.com/plougher/squashfs-tools", "") + super().__init__( + "unsquashfs", + "https://github.com/plougher/squashfs-tools", + "", + apt_package="squashfs-tools", + brew_package="squashfs", + ) async def is_tool_installed(self) -> bool: try: