MachO::Binary: treat thread local zerofill as global zerofill

Co-authored-by: Richard Dzenis <[email protected]>
lief-project · Dec 10, 2024 · 544ab8d · 544ab8d
1 parent 441bb46
commit 544ab8d
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/MachO/Binary.cpp b/src/MachO/Binary.cpp
@@ -924,7 +924,8 @@ ok_error_t Binary::shift(size_t value) {
 
       for (const std::unique_ptr<Section>& section : segment->sections_) {
         if (section->virtual_address() >= loadcommands_end_va ||
-            section->type() == Section::TYPE::ZEROFILL)
+            section->type() == Section::TYPE::ZEROFILL ||
+            section->type() == Section::TYPE::THREAD_LOCAL_ZEROFILL)
         {
           section->virtual_address(section->virtual_address() + value);
         }
@@ -1223,9 +1224,9 @@ bool Binary::extend_segment(const SegmentCommand& segment, size_t size) {
         section->offset(section->offset() + size_aligned);
         section->virtual_address(section->virtual_address() + size_aligned);
       }
-
-      if (section->type() == Section::TYPE::ZEROFILL &&
-          section->virtual_address() > last_va)
+      else if (section->virtual_address() > last_va &&
+          (section->type() == Section::TYPE::ZEROFILL ||
+           section->type() == Section::TYPE::THREAD_LOCAL_ZEROFILL))
       {
         section->virtual_address(section->virtual_address() + size_aligned);
       }