Fix a regression with header rebuilding

0cyn · Nov 10, 2023 · aaf144c · aaf144c
1 parent 6a98d4d
commit aaf144c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/ktool/macho.py b/src/ktool/macho.py
@@ -572,7 +572,7 @@ def insert_load_command(self, load_command, index=-1, suffix=None):
                 for i in range(command.nsects):
                     sects.append(Section(None, Struct.create_with_bytes(struct_class, sect_data[i * struct_class.SIZE:(
                                                                                                                                   i + 1) * struct_class.SIZE],
-                                                                        "little")))
+                                                                        "little"), 8 if self.is64 else 4))
                 seg = SegmentLoadCommand.from_values(isinstance(command, segment_command_64), command.segname,
                                                      command.vmaddr, command.vmsize, command.fileoff, command.filesize,
                                                      command.maxprot, command.initprot, command.flags, sects)
@@ -640,7 +640,7 @@ def remove_load_command(self, index):
                 for i in range(command.nsects):
                     sects.append(Section(None, Struct.create_with_bytes(struct_class, sect_data[i * struct_class.SIZE:(
                                                                                                                                   i + 1) * struct_class.SIZE],
-                                                                        "little")))
+                                                                        "little"), 8 if self.is64 else 4))
                 seg = SegmentLoadCommand.from_values(isinstance(command, segment_command_64), command.segname,
                                                      command.vmaddr, command.vmsize, command.fileoff, command.filesize,
                                                      command.maxprot, command.initprot, command.flags, sects)
@@ -708,7 +708,7 @@ def replace_load_command(self, load_command, index=-1, suffix=None):
                 for i in range(command.nsects):
                     sects.append(Section(None, Struct.create_with_bytes(struct_class, sect_data[i * struct_class.SIZE:(
                                                                                                                                   i + 1) * struct_class.SIZE],
-                                                                        "little")))
+                                                                        "little"), 8 if self.is64 else 4))
                 seg = SegmentLoadCommand.from_values(isinstance(command, segment_command_64), command.segname,
                                                      command.vmaddr, command.vmsize, command.fileoff, command.filesize,
                                                      command.maxprot, command.initprot, command.flags, sects)