Add Bootsplash image support

igelfs/models/boot_registry.py

@@ -40,7 +40,12 @@ def _flag_bits(self) -> tuple[str, str, str]:
 
     @property
     def _flag_values(self) -> tuple[int, int, int]:
-        """Return tuple of integers for flag values."""
+        """
+        Return tuple of integers for flag values.
+
+        Tuple consists of next block index, next block present, key length
+        as integers.
+        """
         return tuple(map(partial(int, base=2), self._flag_bits))
 
     @property

igelfs/models/bootsplash.py

@@ -1,10 +1,14 @@
 """Data models for bootsplash structures."""
 
+import io
 from dataclasses import dataclass
 from typing import ClassVar
 
+from PIL import Image
+
 from igelfs.constants import BOOTSPLASH_MAGIC
-from igelfs.models.base import BaseDataModel
+from igelfs.models.base import BaseDataGroup, BaseDataModel
+from igelfs.models.collections import DataModelCollection
 
 
 @dataclass
@@ -35,3 +39,38 @@ class Bootsplash(BaseDataModel):
     offset: int
     length: int
     ident: bytes
+
+
+@dataclass
+class BootsplashExtent(BaseDataGroup):
+    """
+    Dataclass to handle data of a bootsplash partition extent.
+
+    Extent is not a fixed size so cannot use mapping of attribute to sizes.
+    """
+
+    header: BootsplashHeader
+    splashes: DataModelCollection[Bootsplash]
+    data: bytes
+
+    @classmethod
+    def from_bytes(cls: type["BootsplashExtent"], data: bytes) -> "BootsplashExtent":
+        """Return bootsplash extent model from bytes."""
+        header, data = BootsplashHeader.from_bytes_with_remaining(data)
+        splashes = DataModelCollection()
+        for _ in range(header.num_splashs):
+            splash, data = Bootsplash.from_bytes_with_remaining(data)
+            splashes.append(splash)
+        return cls(header=header, splashes=splashes, data=data)
+
+    def _get_image_data(self) -> list[bytes]:
+        """Return list of bytes for images."""
+        offset = self.header.get_actual_size() + self.splashes.get_actual_size()
+        return [
+            self.data[splash.offset - offset : splash.offset - offset + splash.length]
+            for splash in self.splashes
+        ]
+
+    def get_images(self) -> list[Image.Image]:
+        """Return list of image instances."""
+        return [Image.open(io.BytesIO(image)) for image in self._get_image_data()]

pyproject.toml

@@ -14,7 +14,10 @@ authors = [
 maintainers = [
   {name = "Zack Didcott"}
 ]
-dependencies = ["rsa"]
+dependencies = [
+  "pillow",
+  "rsa",
+]
 requires-python = ">= 3.10"
 
 [project.optional-dependencies]

requirements.txt

@@ -1,7 +1,9 @@
 iniconfig==2.0.0
 packaging==24.1
+pillow==11.0.0
 pluggy==1.5.0
 pyasn1==0.6.1
 pyparted==3.12.0
 pytest==8.3.3
+python-magic==0.4.27
 rsa==4.9