Index: ELF/LinkerScript.cpp =================================================================== --- ELF/LinkerScript.cpp +++ ELF/LinkerScript.cpp @@ -20,6 +20,7 @@ #include "OutputSections.h" #include "ScriptParser.h" #include "SymbolTable.h" +#include "Target.h" #include "llvm/ADT/StringSwitch.h" #include "llvm/Support/ELF.h" #include "llvm/Support/FileSystem.h" @@ -222,6 +223,7 @@ // Assign addresses as instructed by linker script SECTIONS sub-commands. Dot = Out::ElfHeader->getSize() + Out::ProgramHeaders->getSize(); + uintX_t MinVA = std::numeric_limits::max(); uintX_t ThreadBssOffset = 0; for (SectionsCommand &Cmd : Opt.Commands) { @@ -248,11 +250,19 @@ if (Sec->getFlags() & SHF_ALLOC) { Dot = alignTo(Dot, Sec->getAlignment()); Sec->setVA(Dot); + MinVA = std::min(MinVA, Dot); Dot += Sec->getSize(); continue; } } } + + // ELF and Program headers need to be right before the first section in memory. + // Set their addresses accordingly. + MinVA = alignDown(MinVA - Out::ElfHeader->getSize() - + Out::ProgramHeaders->getSize(), Target->PageSize); + Out::ElfHeader->setVA(MinVA); + Out::ProgramHeaders->setVA(Out::ElfHeader->getSize() + MinVA); } template Index: ELF/Writer.cpp =================================================================== --- ELF/Writer.cpp +++ ELF/Writer.cpp @@ -1132,10 +1132,8 @@ template void Writer::fixHeaders() { uintX_t BaseVA = ScriptConfig->DoLayout ? 0 : Target->getVAStart(); Out::ElfHeader->setVA(BaseVA); - Out::ElfHeader->setFileOffset(0); uintX_t Off = Out::ElfHeader->getSize(); Out::ProgramHeaders->setVA(Off + BaseVA); - Out::ProgramHeaders->setFileOffset(Off); } // Assign VAs (addresses at run-time) to output sections. @@ -1183,19 +1181,25 @@ // Assign file offsets to output sections. template void Writer::assignFileOffsets() { - uintX_t Off = - Out::ElfHeader->getSize() + Out::ProgramHeaders->getSize(); + uintX_t Off = 0; - for (OutputSectionBase *Sec : OutputSections) { + auto Set = [&](OutputSectionBase *Sec) { if (Sec->getType() == SHT_NOBITS) { + if (Sec == OutputSections.front()) + Off = getFileAlignment(Off, Sec); Sec->setFileOffset(Off); - continue; + return; } - Off = getFileAlignment(Off, Sec); Sec->setFileOffset(Off); Off += Sec->getSize(); - } + }; + + Set(Out::ElfHeader); + Set(Out::ProgramHeaders); + for (OutputSectionBase *Sec : OutputSections) + Set(Sec); + SectionHeaderOff = alignTo(Off, sizeof(uintX_t)); FileSize = SectionHeaderOff + (OutputSections.size() + 1) * sizeof(Elf_Shdr); } Index: test/ELF/linkerscript-phdr-check.s =================================================================== --- test/ELF/linkerscript-phdr-check.s +++ test/ELF/linkerscript-phdr-check.s @@ -0,0 +1,16 @@ +# REQUIRES: x86 +# RUN: llvm-mc -filetype=obj -triple=x86_64-unknown-linux %s -o %t + +# RUN: echo "SECTIONS { . = 0x10000000; .text : {*(.text.*)} }" > %t.script +# RUN: ld.lld -o %t1 --script %t.script %t +# RUN: llvm-readobj -program-headers %t1 | FileCheck %s +# CHECK: ProgramHeaders [ +# CHECK-NEXT: ProgramHeader { +# CHECK-NEXT: Type: PT_PHDR (0x6) +# CHECK-NEXT: Offset: 0x40 +# CHECK-NEXT: VirtualAddress: 0xFFFF040 + +.global _start +_start: + nop +