diff --git a/extract-partitions.py b/extract-partitions.py index 328f2b5..64fb3de 100755 --- a/extract-partitions.py +++ b/extract-partitions.py @@ -26,7 +26,7 @@ def dump_partitions(comm, disk_fd, outdir, max_size): diskinfo = partitions.get_partitions(comm, disk_fd) for part in diskinfo.gpt.partitions: part_offset = part.first_lba * partitions.BLOCK_SIZE - part_size = (part.last_lba - part.first_lba) * partitions.BLOCK_SIZE + part_size = (part.last_lba - (part.first_lba - 1)) * partitions.BLOCK_SIZE part_name = part.name part_label = "/dev/mmcblk0p%i" % part.index if max_size and part_size > max_size: diff --git a/partitions.py b/partitions.py index 7f333c2..190512f 100755 --- a/partitions.py +++ b/partitions.py @@ -259,7 +259,7 @@ def main(): _logger.debug("%s", info) part_offset = part.first_lba * BLOCK_SIZE - part_size = (part.last_lba - part.first_lba) * BLOCK_SIZE + part_size = (part.last_lba - (part.first_lba - 1)) * BLOCK_SIZE _logger.debug("Opened fd %d for disk", disk_fd) if args.dump: