[mach-o] extract atom alignment information from mach-o files

llvm-svn: 222201

1 files changed, 10 insertions, 3 deletions

diff --git a/lld/lib/ReaderWriter/MachO/File.h b/lld/lib/ReaderWriter/MachO/File.h
index 666bab6b111..831de2451fa 100644
--- a/lld/lib/ReaderWriter/MachO/File.h
+++ b/lld/lib/ReaderWriter/MachO/File.h
@@ -39,9 +39,11 @@ public:
       name = name.copy(_allocator);
       content = content.copy(_allocator);
     }
+    DefinedAtom::Alignment align(inSection->alignment,
+                                 sectionOffset % (1 << inSection->alignment));
     MachODefinedAtom *atom =
         new (_allocator) MachODefinedAtom(*this, name, scope, type, merge,
-                                          thumb, noDeadStrip, content);
+                                          thumb, noDeadStrip, content, align);
     addAtomForSection(inSection, atom, sectionOffset);
   }
 
@@ -59,11 +61,13 @@ public:
       content = content.copy(_allocator);
       sectionName = sectionName.copy(_allocator);
     }
+    DefinedAtom::Alignment align(inSection->alignment,
+                                 sectionOffset % (1 << inSection->alignment));
     MachODefinedCustomSectionAtom *atom =
         new (_allocator) MachODefinedCustomSectionAtom(*this, name, scope, type,
                                                         merge, thumb,
                                                         noDeadStrip, content,
-                                                        sectionName);
+                                                        sectionName, align);
     addAtomForSection(inSection, atom, sectionOffset);
   }
 
@@ -75,8 +79,11 @@ public:
       // Make a copy of the atom's name and content that is owned by this file.
       name = name.copy(_allocator);
     }
+    DefinedAtom::Alignment align(inSection->alignment,
+                                 sectionOffset % (1 << inSection->alignment));
     MachODefinedAtom *atom =
-       new (_allocator) MachODefinedAtom(*this, name, scope, size, noDeadStrip);
+       new (_allocator) MachODefinedAtom(*this, name, scope, size, noDeadStrip,
+                                         align);
     addAtomForSection(inSection, atom, sectionOffset);
   }