[Git][reproducible-builds/diffoscope][issue-333] WIP on DOS/MBR extraction. (Re: reproducible-builds/diffoscope#333)

[Chris Lamb (@lamby)]

Chris Lamb pushed to branch issue-333 at Reproducible Builds / diffoscope


Commits:
81d2d507 by Chris Lamb at 2023-07-20T12:23:26+01:00
WIP on DOS/MBR extraction. (Re: reproducible-builds/diffoscope#333)

- - - - -


3 changed files:

- + diffoscope/comparators/7z.py
- diffoscope/comparators/__init__.py
- diffoscope/external_tools.py


Changes:

=====================================
diffoscope/comparators/7z.py
=====================================
@@ -0,0 +1,93 @@
+#
+# diffoscope: in-depth comparison of files, archives, and directories
+#
+# Copyright © 2023 Chris Lamb <lamby at debian.org>
+#
+# diffoscope is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# diffoscope is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with diffoscope.  If not, see <https://www.gnu.org/licenses/>.
+
+import os
+import re
+import logging
+import subprocess
+
+from diffoscope.tools import tool_required
+from diffoscope.tempfiles import get_temporary_directory
+from diffoscope.difference import Difference
+
+from .utils.archive import Archive
+from .utils.file import File
+from .utils.command import Command, our_check_output
+
+logger = logging.getLogger(__name__)
+
+
+class SevenZList(Command):
+    @tool_required("7z")
+    def cmdline(self):
+        return (
+            "7z",
+            "l",
+            self.path,
+        )
+
+    def filter(self, line):
+        haystack = line.decode("utf-8")
+        if haystack.startswith("Listing archive: ") or haystack.startswith(
+            "Path = "
+        ):
+            return b""
+        return line
+
+
+class SevenZContainer(Archive):
+    @tool_required("7z")
+    def open_archive(self):
+        self._temp_dir = get_temporary_directory(suffix="7z")
+
+        try:
+            our_check_output(
+                (
+                    "7z",
+                    "e",
+                    os.path.abspath(self.source.path),
+                ),
+                cwd=self._temp_dir.name,
+                stderr=subprocess.DEVNULL,
+            )
+        except subprocess.CalledProcessError:
+            return False
+
+        return self
+
+    def close_archive(self):
+        self._temp_dir.cleanup()
+
+    def get_member_names(self):
+        return os.listdir(self._temp_dir.name)
+
+    def extract(self, member_name, dest_dir):
+        return os.path.join(self._temp_dir.name, member_name)
+
+
+class SevenZFile(File):
+    DESCRIPTION = "Filesystem image"
+    FILE_TYPE_RE = re.compile(r"^DOS/MBR boot sector;")
+    CONTAINER_CLASSES = [SevenZContainer]
+
+    def compare_details(self, other, source=None):
+        return [
+            Difference.from_operation(
+                SevenZList, self.path, other.path, source="7z l"
+            )
+        ]


=====================================
diffoscope/comparators/__init__.py
=====================================
@@ -64,6 +64,7 @@ class ComparatorManager:
         ("elf.ElfFile",),
         ("macho.MachoFile",),
         ("fsimage.FsImageFile",),
+        ("7z.SevenZFile",),
         ("llvm.LlvmBitCodeFile",),
         ("sqlite.Sqlite3Database",),
         ("wasm.WasmFile",),


=====================================
diffoscope/external_tools.py
=====================================
@@ -23,6 +23,7 @@ that might resolve to, for example, `/usr/bin/abootimg`.
 """
 
 EXTERNAL_TOOLS = {
+    "7z": {"debian": "p7zip-full"},
     "aapt2": {"debian": "aapt"},
     "abootimg": {"debian": "abootimg", "guix": "abootimg"},
     "androguard": {"debian": "androguard"},



View it on GitLab: https://salsa.debian.org/reproducible-builds/diffoscope/-/commit/81d2d5078dbb859f4cb714021cdf2d180c73a46f

-- 
View it on GitLab: https://salsa.debian.org/reproducible-builds/diffoscope/-/commit/81d2d5078dbb859f4cb714021cdf2d180c73a46f
You're receiving this email because of your account on salsa.debian.org.


-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://lists.reproducible-builds.org/pipermail/rb-commits/attachments/20230720/7b6047cd/attachment.htm>