mirror of
https://github.com/Lekensteyn/lglaf.git
synced 2024-11-14 11:25:29 -05:00
6def07a75f
IRC excerpt: [12-20 15:20] <steadfasterX> i have a partition: end=378879 start=377856 [12-20 15:21] <steadfasterX> so the calcluation would be 378879-377856 [12-20 15:21] <steadfasterX> which is 1023 [12-20 15:21] <steadfasterX> which can never be correct [12-20 15:22] <steadfasterX> this is bc in the calculation it will need to include the start (or end) sector itself [12-20 15:23] <steadfasterX> when i dump without that +1 i miss exact 512 bytes so 1 sector
70 lines
2.9 KiB
Python
Executable file
70 lines
2.9 KiB
Python
Executable file
#!/usr/bin/env python
|
|
#
|
|
# Dump partitions to file.
|
|
#
|
|
# Copyright (C) 2015 Peter Wu <peter@lekensteyn.nl>
|
|
# Licensed under the MIT license <http://opensource.org/licenses/MIT>.
|
|
|
|
from contextlib import closing
|
|
import argparse, logging, os, struct
|
|
import lglaf, partitions
|
|
|
|
_logger = logging.getLogger("extract-partitions")
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("-d", "--outdir", default=".",
|
|
help="Output directory for disk images.")
|
|
# Do not dump partitions larger than this size
|
|
# (userdata 11728 MiB, system 2064 MiB, cache 608 MiB, cust 256 MiB)
|
|
parser.add_argument("--max-size", metavar="kbytes", type=int, default=65535,
|
|
help="Maximum partition size to dump (in KiB) or 0 to dump all (default %(default)d)")
|
|
parser.add_argument("--debug", action='store_true', help="Enable debug messages")
|
|
parser.add_argument("--skip-hello", action="store_true",
|
|
help="Immediately send commands, skip HELO message")
|
|
|
|
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 - 1)) * partitions.BLOCK_SIZE
|
|
part_name = part.name
|
|
part_label = "/dev/mmcblk0p%i" % part.index
|
|
if max_size and part_size > max_size:
|
|
_logger.info("Ignoring large partition %s (%s) of size %dK",
|
|
part_label, part_name, part_size / 1024)
|
|
continue
|
|
out_path = os.path.join(outdir, "%s.bin" % part_name)
|
|
try:
|
|
current_size = os.path.getsize(out_path)
|
|
if current_size > part_size:
|
|
_logger.warn("%s: unexpected size %dK, larger than %dK",
|
|
out_path, current_size / 1024, part_size / 1024)
|
|
continue
|
|
elif current_size == part_size:
|
|
_logger.info("Skipping partition %s (%s), already found at %s",
|
|
part_label, part_name, out_path)
|
|
continue
|
|
except OSError: pass
|
|
_logger.info("Dumping partition %s (%s) to %s (%d bytes)",
|
|
part_label, part_name, out_path, part_size)
|
|
partitions.dump_partition(comm, disk_fd, out_path, part_offset, part_size)
|
|
|
|
def main():
|
|
args = parser.parse_args()
|
|
logging.basicConfig(format='%(asctime)s %(name)s: %(levelname)s: %(message)s',
|
|
level=logging.DEBUG if args.debug else logging.INFO)
|
|
|
|
try: os.makedirs(args.outdir)
|
|
except OSError: pass
|
|
|
|
comm = lglaf.autodetect_device()
|
|
with closing(comm):
|
|
if not args.skip_hello:
|
|
lglaf.try_hello(comm)
|
|
|
|
with partitions.laf_open_disk(comm) as disk_fd:
|
|
_logger.debug("Opened fd %d for disk", disk_fd)
|
|
dump_partitions(comm, disk_fd, args.outdir, args.max_size * 1024)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|