def main():
if len(sys.argv) < 2:
print("Please provide a path to a disk")
sys.exit(1)
disk = sys.argv[1]
with open(disk, "rb") as b:
data = b.read(Mbr.SIZE)
mbr = Mbr(data)
print("Disk Signature {}".format(to_hex(mbr.disk_signature)))
if mbr.is_protected():
print("MBR is protected")
if mbr.is_gpt():
print("GUID Partition Table is used")
print("Null {}".format(to_hex(mbr.null)))
print("Boot Signature {}".format(to_hex(mbr.boot_signature)))
print()
for i, partition in enumerate(mbr.partitions):
print("Partition {}\n{}\n".format(i, partition))
if mbr.is_gpt():
gpt_data = b.read(Gpt.SIZE)
gpt = Gpt(gpt_data)
print("GPT Signature {}".format(to_hex(gpt.signature)))
print("GPT Revision {}".format(to_hex(gpt.revision)))
print("GPT Header Size {}".format(gpt.header_size))
print("GPT Disk UUID {}".format(gpt.guid))
print("GPT Partition Entries {}".format(gpt.nr_entries))
print("GPT Entry Size {}".format(gpt.entry_size))
print("GPT Current LBA {}".format(gpt.current_lba))
print("GPT Backup LBA {}".format(gpt.backup_lba))
print("GPT First LBA {}".format(gpt.first_lba))
print("GPT Last LBA {}".format(gpt.last_lba))
print("GPT Starting LBA {}".format(gpt.start_lba))
print()
for i in range(0, 5):
entry_data = b.read(GptPartition.DEFAULT_SIZE)
entry = GptPartition(entry_data)
print("GPT {} Type GUID {}".format(i, entry.type_guid))
print("GPT {} Type {}".format(
i, entry.get_partition_type()))
print("GPT {} Unique GUID {}".format(i, entry.unique_guid))
print("GPT {} Name {}".format(i, entry.name))
print("GPT {} First LBA {}".format(i, entry.first_lba))
print("GPT {} Last LBA {}".format(i, entry.last_lba))
print("GPT {} Attribute Flags {}".format(
i, to_hex(entry.attribute_flags)))
print()
def __str__(self):
return "Data {}\nStatus {} is bootable {}\nType {} " \
"{}".format(to_hex(self.data), to_hex(self.status),
self.is_bootable(), to_hex(self.type),
self.get_partition_type())