Fix Mach-O issue

lief-project · Jul 27, 2024 · 24d48ad · 24d48ad
1 parent f4bf30d
commit 24d48ad
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/api/python/lief/ELF.pyi b/api/python/lief/ELF.pyi
@@ -1056,8 +1056,8 @@ class DynamicEntry(lief.Object):
         VERNEEDNUM: ClassVar[DynamicEntry.TAG] = ...
         VERSYM: ClassVar[DynamicEntry.TAG] = ...
         X86_64_PLT: ClassVar[DynamicEntry.TAG] = ...
-        X86_64_PLTSZ: ClassVar[DynamicEntry.TAG] = ...
         X86_64_PLTENT: ClassVar[DynamicEntry.TAG] = ...
+        X86_64_PLTSZ: ClassVar[DynamicEntry.TAG] = ...
         __name__: str
         def __init__(self, *args, **kwargs) -> None: ...
         @staticmethod

diff --git a/include/LIEF/MachO/LoadCommand.hpp b/include/LIEF/MachO/LoadCommand.hpp
@@ -27,6 +27,7 @@ namespace LIEF {
 namespace MachO {
 class Builder;
 class BinaryParser;
+class Binary;
 
 namespace details {
 struct load_command;
@@ -36,6 +37,7 @@ struct load_command;
 class LIEF_API LoadCommand : public Object {
   friend class Builder;
   friend class BinaryParser;
+  friend class Binary;
   public:
   using raw_t = std::vector<uint8_t>;
 

diff --git a/src/MachO/Binary.cpp b/src/MachO/Binary.cpp
@@ -1196,6 +1196,7 @@ bool Binary::extend(const LoadCommand& command, uint64_t size) {
   }
 
   cmd->size(size_aligned);
+  cmd->original_data_.resize(size_aligned);
 
   // Update Header
   // =============