def test_get_layout(self, disk_number):
from infi.diskmanagement.ioctl.structures import DRIVE_LAYOUT_INFORMATION_EX
with self._ioctl_disk_get_drive_layout_ex as mock:
with open("drive_layout_{}".format(disk_number), 'rb') as fd:
mock.return_value = DRIVE_LAYOUT_INFORMATION_EX.create_from_string(fd.read())
io = Disk(1)
layout = io._get_layout()
self.assertEqual(len(layout.PartitionEntry), EXPECTED_NUMBERS_OF_PARTITIONS[disk_number])
def test_get_partitions(self, disk_number):
from infi.diskmanagement.ioctl.structures import DRIVE_LAYOUT_INFORMATION_EX
with self._ioctl_disk_get_drive_layout_ex as mock:
with open("drive_layout_{}".format(disk_number), 'rb') as fd:
mock.return_value = DRIVE_LAYOUT_INFORMATION_EX.create_from_string(fd.read())
io = Disk(1)
partitions = io.get_partitions()
for partition in partitions:
self.assertIn(partition._struct.PartitionStyle, [0, 1, ])
def investigate(self, disk_number=0):
from infi.diskmanagement.ioctl.structures import DRIVE_LAYOUT_INFORMATION_EX
with self._ioctl_disk_get_drive_layout_ex as mock:
with open("drive_layout_{}".format(disk_number), 'rb') as fd:
string = fd.read()
mock.return_value = DRIVE_LAYOUT_INFORMATION_EX.create_from_string(string)
io = Disk(1)
layout = io._get_layout()
self.assertEqual(layout.PartitionStyle, 0) # :0
self.assertEqual(layout.PartitionCount, 4) # :4
self.assertEqual(layout.union.Signature, 0xf35f23ba) # :8
partition = layout.PartitionEntry[0]
self.assertEqual(partition.PartitionStyle, 0) # :48
# FIXME
# For some reason, the offset of StartingOffset is before the 32nd byte
# Maybe a bug in instruct?
self.assertEqual(partition.StartingOffset.QuadPart, 0x100000) # :56
def setUp(self):
unittest.TestCase.setUp(self)
self._disk = Disk(7)
