diff --git a/payload_dumper/dumper.py b/payload_dumper/dumper.py index b4258e6..435eae2 100755 --- a/payload_dumper/dumper.py +++ b/payload_dumper/dumper.py @@ -55,12 +55,19 @@ def run(self): partitions = [] for image in self.images.split(","): image = image.strip() + found = False for dam_part in self.dam.partitions: if dam_part.partition_name == image: partitions.append(dam_part) + found = True break + if not found: print("Partition %s not found in image" % image) + if len(partitions) == 0: + print("Not operating on any partitions") + return 0 + partitions_with_ops = [] for partition in partitions: operations = [] diff --git a/pyproject.toml b/pyproject.toml index dda5e1e..369872d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "payload_dumper" -version = "0.2.0" +version = "0.2.1" description = "Dump partitions from Android's payload.bin" authors = ["Rasmus Moorats "] readme = "README.md"