Set partition array size to 128

swysocki · Nov 26, 2021 · ed111ad · ed111ad
1 parent e308574
commit ed111ad
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/pygpt_disk/table.py b/pygpt_disk/table.py
@@ -36,7 +36,7 @@ def __init__(self, disk: Disk) -> None:
         self._partition_last_lba = Table.HeaderEntry(48, 8, int(self.disk.sectors - 34))
         self._disk_guid = Table.HeaderEntry(56, 16, uuid.uuid4().bytes_le)
         self._partition_array_start = Table.HeaderEntry(72, 8, 2)
-        self._partition_array_length = Table.HeaderEntry(80, 4, 0)
+        self._partition_array_length = Table.HeaderEntry(80, 4, 128)
         self._partition_entry_size = Table.HeaderEntry(84, 4, 128)
         self._partition_array_crc = Table.HeaderEntry(88, 4, 0)
 
@@ -120,5 +120,9 @@ def _checksum_header(self, offset: int):
         self._header_crc.content = binascii.crc32(raw_header)
         self._write_section(self._header_crc, offset)
 
-    def update_header(self):
-        pass
+
+if __name__ == "__main__":
+    disk = Disk(8 * 1024 * 1024, "/tmp/testgpt.img")
+    t = Table(disk)
+    t.write()
+    disk.write()
diff --git a/tests/test_table.py b/tests/test_table.py
@@ -53,7 +53,7 @@ def read_header(primary=True):
         # partition array start LBA
         assert t.disk.buffer.read(8) == (2).to_bytes(8, "little")
         # partition array length
-        assert t.disk.buffer.read(4) == (0).to_bytes(4, "little")
+        assert t.disk.buffer.read(4) == (128).to_bytes(4, "little")
         # partition entry length
         assert t.disk.buffer.read(4) == (128).to_bytes(4, "little")
         # partition array crc