tmp fix for p_align issue

Author: DennyDai

src/patcherex2/components/allocation_managers/allocation_manager.py

@@ -154,7 +154,7 @@ def _find_in_mapped_blocks(
     def _create_new_mapped_block(
         self, size: int, flag=MemoryFlag.RWX, align=0x1
     ) -> bool:
-        # map 0x1000 bytes # TODO: currently we won't use available file/mem blocks, instead we create new one at the end of the file
+        # TODO: currently we won't use available file/mem blocks, instead we create new one at the end of the file
         file_addr = None
         mem_addr = None
         for block in self.blocks[FileBlock]:
@@ -163,8 +163,19 @@ def _create_new_mapped_block(
                 block.addr += 0x2000
         for block in self.blocks[MemoryBlock]:
             if block.size == -1:
-                # mem_addr % 0x1000 should equal to file_addr % 0x1000 TODO
-                mem_addr = block.addr + (file_addr % 0x1000)
+                # NOTE: mem_addr % p_align should equal to file_addr % p_align
+                # Check `man elf` and search for `p_align` for more information
+                # FIXME: shouldn't do any assumption on component type, reimpl in a better way
+                # FIXME: even worse, importing ELF will cause circular import
+                # TODO: consider merge allocation_manager and binfmt_tool into one component
+                if self.p.binfmt_tool.__class__.__name__ == "ELF":
+                    max_seg_align = max(
+                        [segment["p_align"] for segment in self.p.binfmt_tool._segments]
+                        + [0]
+                    )
+                    mem_addr = block.addr + (file_addr % max_seg_align)
+                else:
+                    mem_addr = block.addr + (file_addr % 0x1000)
                 block.addr = mem_addr + 0x2000
         if file_addr and mem_addr:
             self.add_block(