sebaubuntu_libs: libandroid: partitions: Import from aospdtgen

Change-Id: Ia79b408e40946227f644fabb9562c34b163c9a4b

Author: SebaUbuntu

sebaubuntu_libs/libandroid/partitions/__init__.py

@@ -0,0 +1,6 @@
+#
+# Copyright (C) 2022 Sebastiano Barezzi
+#
+# SPDX-License-Identifier: Apache-2.0
+#
+"""AOSP partitions library."""

sebaubuntu_libs/libandroid/partitions/partition.py

@@ -0,0 +1,77 @@
+#
+# Copyright (C) 2022 Sebastiano Barezzi
+#
+# SPDX-License-Identifier: Apache-2.0
+#
+
+from __future__ import annotations
+from pathlib import Path
+from typing import List
+
+from sebaubuntu_libs.libandroid.fstab import Fstab, FstabEntry
+from sebaubuntu_libs.libandroid.partitions.partition_model import PartitionModel
+from sebaubuntu_libs.libandroid.props import BuildProp
+from sebaubuntu_libs.libandroid.vintf.manifest import Manifest
+from sebaubuntu_libs.libpath import is_relative_to
+from sebaubuntu_libs.libreorder import strcoll_files_key
+
+BUILD_PROP_LOCATION = ["build.prop", "etc/build.prop"]
+DEFAULT_PROP_LOCATION = ["default.prop", "etc/default.prop"]
+
+def get_dir(path: Path):
+	dir = {}
+	for i in path.iterdir():
+		dir[i.name] = i if i.is_file() else get_dir(i)
+	return dir
+
+class AndroidPartition:
+	def __init__(self, model: PartitionModel, real_path: Path, dump_path: Path):
+		self.model = model
+		self.real_path = real_path
+		self.dump_path = dump_path
+
+		self.files: List[Path] = []
+		self.fstab_entry: FstabEntry = None
+
+		self.build_prop = BuildProp()
+		for possible_paths in BUILD_PROP_LOCATION + DEFAULT_PROP_LOCATION:
+			build_prop_path = self.real_path / possible_paths
+			if not build_prop_path.is_file():
+				continue
+
+			self.build_prop.import_props(build_prop_path)
+
+		self.manifest = Manifest()
+		for possible_paths in ["etc/vintf/manifest.xml", "manifest.xml"]:
+			manifest_path = self.real_path / possible_paths
+			if not manifest_path.is_file():
+				continue
+
+			self.manifest.import_file(manifest_path)
+
+	def get_relative_path(self):
+		return self.real_path.relative_to(self.dump_path)
+
+	def fill_files(self, files: List[Path]):
+		for file in files:
+			if not is_relative_to(file, self.real_path):
+				continue
+
+			self.files.append(file)
+
+	def fill_fstab_entry(self, fstab: Fstab):
+		for mount_point in self.model.mount_points:
+			self.fstab_entry = fstab.get_partition_by_mount_point(mount_point)
+			if self.fstab_entry is not None:
+				return
+
+	def get_files(self):
+		"""Returns the ordered list of files."""
+		self.files.sort(key=strcoll_files_key)
+		return self.files
+
+	def get_formatted_file(self, file: Path):
+		return self.model.proprietary_files_prefix / file.relative_to(self.real_path)
+
+	def get_formatted_files(self):
+		return [self.get_formatted_file(file) for file in self.get_files()]

sebaubuntu_libs/libandroid/partitions/partition_model.py

@@ -0,0 +1,80 @@
+#
+# Copyright (C) 2022 Sebastiano Barezzi
+#
+# SPDX-License-Identifier: Apache-2.0
+#
+
+from __future__ import annotations
+from pathlib import Path
+from typing import List
+
+(
+	BOOTLOADER,
+	SSI,
+	TREBLE,
+	DATA,
+) = range(4)
+
+class _PartitionModel:
+	ALL: List[_PartitionModel] = []
+
+	def __init__(self,
+	             name: str,
+	             group: int,
+				 mount_points: List[str] = None,
+				 proprietary_files_prefix: Path = None,
+	            ):
+		self.name = name
+		self.group = group
+		self.mount_points = mount_points
+		self.proprietary_files_prefix = proprietary_files_prefix
+
+		if self.mount_points is None:
+			self.mount_points = [f"/{self.name}"]
+
+		if self.proprietary_files_prefix is None:
+			self.proprietary_files_prefix = Path(self.name)
+
+		_PartitionModel.ALL.append(self)
+
+	@classmethod
+	def from_name(cls, name: str):
+		for model in cls.ALL:
+			if model.name == name:
+				return model
+
+		return None
+
+	@classmethod
+	def from_group(cls, group: int):
+		return [model for model in cls.ALL if model.group == group]
+
+	@classmethod
+	def from_mount_point(cls, mount_point: str):
+		for model in cls.ALL:
+			if mount_point in model.mount_points:
+				return model
+
+		return None
+
+class PartitionModel(_PartitionModel):
+	BOOT = _PartitionModel("boot", BOOTLOADER)
+	DTBO = _PartitionModel("dtbo", BOOTLOADER)
+	RECOVERY = _PartitionModel("recovery", BOOTLOADER)
+	MISC = _PartitionModel("misc", BOOTLOADER)
+	VBMETA = _PartitionModel("vbmeta", BOOTLOADER)
+	VBMETA_SYSTEM = _PartitionModel("vbmeta_system", BOOTLOADER)
+	VBMETA_VENDOR = _PartitionModel("vbmeta_vendor", BOOTLOADER)
+
+	SYSTEM = _PartitionModel("system", SSI, ["/system", "/"], Path(""))
+	PRODUCT = _PartitionModel("product", SSI)
+	SYSTEM_EXT = _PartitionModel("system_ext", SSI)
+
+	VENDOR = _PartitionModel("vendor", TREBLE)
+	ODM = _PartitionModel("odm", TREBLE)
+	ODM_DLKM = _PartitionModel("odm_dlkm", TREBLE)
+	VENDOR_DLKM = _PartitionModel("vendor_dlkm", TREBLE)
+
+	USERDATA = _PartitionModel("data", DATA)
+	CACHE = _PartitionModel("cache", DATA)
+	METADATA = _PartitionModel("metadata", DATA)

sebaubuntu_libs/libandroid/partitions/partitions.py

@@ -0,0 +1,75 @@
+#
+# Copyright (C) 2022 Sebastiano Barezzi
+#
+# SPDX-License-Identifier: Apache-2.0
+#
+
+from pathlib import Path
+from typing import Dict
+
+from sebaubuntu_libs.libandroid.partitions.partition import AndroidPartition, BUILD_PROP_LOCATION
+from sebaubuntu_libs.libandroid.partitions.partition_model import SSI, TREBLE, PartitionModel
+
+class Partitions:
+	def __init__(self, dump_path: Path):
+		self.dump_path = dump_path
+
+		self.partitions: Dict[PartitionModel, AndroidPartition] = {}
+
+		# Search for system
+		for system in [self.dump_path / "system", self.dump_path / "system/system"]:
+			for build_prop_location in BUILD_PROP_LOCATION:
+				if not (system / build_prop_location).is_file():
+					continue
+
+				self.partitions[PartitionModel.SYSTEM] = AndroidPartition(PartitionModel.SYSTEM, system, self.dump_path)
+
+		assert PartitionModel.SYSTEM in self.partitions
+		self.system = self.partitions[PartitionModel.SYSTEM]
+
+		# Search for vendor
+		for vendor in [self.partitions[PartitionModel.SYSTEM].real_path / "vendor", self.dump_path / "vendor"]:
+			for build_prop_location in BUILD_PROP_LOCATION:
+				if not (vendor / build_prop_location).is_file():
+					continue
+
+				self.partitions[PartitionModel.VENDOR] = AndroidPartition(PartitionModel.VENDOR, vendor, self.dump_path)
+
+		assert PartitionModel.VENDOR in self.partitions
+		self.vendor = self.partitions[PartitionModel.VENDOR]
+
+		# Search for the other partitions
+		for model in [model for model in PartitionModel.from_group(SSI) if not (model is PartitionModel.SYSTEM)]:
+			self._search_for_partition(model)
+
+		for model in [model for model in PartitionModel.from_group(TREBLE) if not (model is PartitionModel.VENDOR)]:
+			self._search_for_partition(model)
+
+	def get_partition(self, model: PartitionModel):
+		if not model:
+			return None
+
+		if model in self.partitions:
+			return self.partitions[model]
+
+		return None
+
+	def get_partition_by_name(self, name: str):
+		return self.get_partition(PartitionModel.from_name(name))
+
+	def get_all_partitions(self):
+		return self.partitions.values()
+
+	def _search_for_partition(self, model: PartitionModel):
+		possible_locations = [
+			self.partitions[PartitionModel.SYSTEM].real_path / model.name,
+			self.partitions[PartitionModel.VENDOR].real_path / model.name,
+			self.dump_path / model.name
+		]
+
+		for location in possible_locations:
+			for build_prop_location in BUILD_PROP_LOCATION:
+				if not (location / build_prop_location).is_file():
+					continue
+
+				self.partitions[model] = AndroidPartition(model, location, self.dump_path)