Index: lld/ELF/MapFile.cpp =================================================================== --- lld/ELF/MapFile.cpp +++ lld/ELF/MapFile.cpp @@ -35,7 +35,7 @@ using namespace lld; using namespace lld::elf; -typedef DenseMap> SymbolMapTy; +typedef DenseMap> SymbolMapTy; // Print out the first three columns of a line. static void writeHeader(raw_ostream &OS, uint64_t Addr, uint64_t Size, @@ -47,30 +47,41 @@ static std::string indent(int Depth) { return std::string(Depth * 8, ' '); } // Returns a list of all symbols that we want to print out. -static std::vector getSymbols() { - std::vector V; +static std::vector getSymbols() { + std::vector V; + for (InputFile *File : ObjectFiles) for (Symbol *B : File->getSymbols()) if (auto *DR = dyn_cast(B)) if (DR->File == File && !DR->isSection() && DR->Section && DR->Section->Live) V.push_back(DR); + + for (Symbol *B : Symtab->getSymbols()) + if (auto *Sym = dyn_cast(B)) + if (Sym->CopyRelSec) + V.push_back(Sym); + return V; } // Returns a map from sections to their symbols. -static SymbolMapTy getSectionSyms(ArrayRef Syms) { +static SymbolMapTy getSectionSyms(ArrayRef Syms) { SymbolMapTy Ret; - for (Defined *S : Syms) - Ret[S->Section].push_back(S); + for (Symbol *S : Syms) { + if (auto *Sym = dyn_cast(S)) + Ret[Sym->Section].push_back(Sym); + else + Ret[cast(S)->CopyRelSec].push_back(S); + } // Sort symbols by address. We want to print out symbols in the // order in the output file rather than the order they appeared // in the input files. for (auto &It : Ret) { - SmallVectorImpl &V = It.second; + SmallVectorImpl &V = It.second; std::sort(V.begin(), V.end(), - [](Defined *A, Defined *B) { return A->getVA() < B->getVA(); }); + [](Symbol *A, Symbol *B) { return A->getVA() < B->getVA(); }); } return Ret; } @@ -78,8 +89,8 @@ // Construct a map from symbols to their stringified representations. // Demangling symbols (which is what toString() does) is slow, so // we do that in batch using parallel-for. -static DenseMap -getSymbolStrings(ArrayRef Syms) { +static DenseMap +getSymbolStrings(ArrayRef Syms) { std::vector Str(Syms.size()); parallelForEachN(0, Syms.size(), [&](size_t I) { raw_string_ostream OS(Str[I]); @@ -87,7 +98,7 @@ OS << indent(2) << toString(*Syms[I]); }); - DenseMap Ret; + DenseMap Ret; for (size_t I = 0, E = Syms.size(); I < E; ++I) Ret[Syms[I]] = std::move(Str[I]); return Ret; @@ -106,9 +117,9 @@ } // Collect symbol info that we want to print out. - std::vector Syms = getSymbols(); + std::vector Syms = getSymbols(); SymbolMapTy SectionSyms = getSectionSyms(Syms); - DenseMap SymStr = getSymbolStrings(Syms); + DenseMap SymStr = getSymbolStrings(Syms); // Print out the header line. int W = Config->Is64 ? 16 : 8; @@ -129,7 +140,7 @@ writeHeader(OS, OSec->Addr + IS->OutSecOff, IS->getSize(), IS->Alignment); OS << indent(1) << toString(IS) << '\n'; - for (Defined *Sym : SectionSyms[IS]) + for (Symbol *Sym : SectionSyms[IS]) OS << SymStr[Sym] << '\n'; } } Index: lld/test/ELF/Inputs/map-file5.s =================================================================== --- /dev/null +++ lld/test/ELF/Inputs/map-file5.s @@ -0,0 +1,4 @@ +.global copy +.type copy, @object +copy: +.size copy, 32 Index: lld/test/ELF/map-file.s =================================================================== --- lld/test/ELF/map-file.s +++ lld/test/ELF/map-file.s @@ -6,14 +6,19 @@ // RUN: llvm-mc -filetype=obj -triple=x86_64-pc-linux %p/Inputs/map-file4.s -o %t4.o // RUN: rm -f %t4.a // RUN: llvm-ar rc %t4.a %t4.o -// RUN: ld.lld %t1.o %t2.o %t3.o %t4.a -o %t -M | FileCheck -strict-whitespace %s -// RUN: ld.lld %t1.o %t2.o %t3.o %t4.a -o %t -print-map | FileCheck -strict-whitespace %s -// RUN: ld.lld %t1.o %t2.o %t3.o %t4.a -o %t -Map=%t.map + +// RUN: llvm-mc -filetype=obj -triple=x86_64-pc-linux %p/Inputs/map-file5.s -o %t5.o +// RUN: ld.lld -o %t5.so -shared %t5.o + +// RUN: ld.lld %t1.o %t2.o %t3.o %t4.a %t5.so -o %t -M | FileCheck -strict-whitespace %s +// RUN: ld.lld %t1.o %t2.o %t3.o %t4.a %t5.so -o %t -print-map | FileCheck -strict-whitespace %s +// RUN: ld.lld %t1.o %t2.o %t3.o %t4.a %t5.so -o %t -Map=%t.map // RUN: FileCheck -strict-whitespace %s < %t.map .global _start _start: call baz + .quad copy .global _Z1fi _Z1fi: .cfi_startproc @@ -30,34 +35,49 @@ labs = 0x1AB5 // CHECK: Address Size Align Out In Symbol -// CHECK-NEXT: 0000000000200158 0000000000000030 8 .eh_frame -// CHECK-NEXT: 0000000000200158 0000000000000030 8 :(.eh_frame) -// CHECK-NEXT: 0000000000201000 0000000000000015 4 .text -// CHECK-NEXT: 0000000000201000 000000000000000e 4 {{.*}}{{/|\\}}map-file.s.tmp1.o:(.text) +// CHECK-NEXT: 00000000002001c8 0000000000000030 8 .dynsym +// CHECK-NEXT: 00000000002001c8 0000000000000030 8 :(.dynsym) +// CHECK-NEXT: 00000000002001f8 0000000000000020 8 .gnu.hash +// CHECK-NEXT: 00000000002001f8 0000000000000020 8 :(.gnu.hash) +// CHECK-NEXT: 0000000000200218 0000000000000018 4 .hash +// CHECK-NEXT: 0000000000200218 0000000000000018 4 :(.hash) +// CHECK-NEXT: 0000000000200230 000000000000003a 1 .dynstr +// CHECK-NEXT: 0000000000200230 000000000000003a 1 :(.dynstr) +// CHECK-NEXT: 0000000000200270 0000000000000018 8 .rela.dyn +// CHECK-NEXT: 0000000000200270 0000000000000018 8 :(.rela.dyn) +// CHECK-NEXT: 0000000000200288 0000000000000030 8 .eh_frame +// CHECK-NEXT: 0000000000200288 0000000000000030 8 :(.eh_frame) +// CHECK-NEXT: 0000000000201000 000000000000001d 4 .text +// CHECK-NEXT: 0000000000201000 0000000000000016 4 {{.*}}{{/|\\}}map-file.s.tmp1.o:(.text) // CHECK-NEXT: 0000000000201000 0000000000000000 0 _start -// CHECK-NEXT: 0000000000201005 0000000000000000 0 f(int) -// CHECK-NEXT: 000000000020100e 0000000000000000 0 local -// CHECK-NEXT: 0000000000201010 0000000000000002 4 {{.*}}{{/|\\}}map-file.s.tmp2.o:(.text) -// CHECK-NEXT: 0000000000201010 0000000000000000 0 foo -// CHECK-NEXT: 0000000000201011 0000000000000000 0 bar -// CHECK-NEXT: 0000000000201012 0000000000000000 1 {{.*}}{{/|\\}}map-file.s.tmp2.o:(.text.zed) -// CHECK-NEXT: 0000000000201012 0000000000000000 0 zed -// CHECK-NEXT: 0000000000201014 0000000000000000 4 {{.*}}{{/|\\}}map-file.s.tmp3.o:(.text) -// CHECK-NEXT: 0000000000201014 0000000000000000 0 bah -// CHECK-NEXT: 0000000000201014 0000000000000001 4 {{.*}}{{/|\\}}map-file.s.tmp4.a(map-file.s.tmp4.o):(.text) -// CHECK-NEXT: 0000000000201014 0000000000000000 0 baz -// CHECK-NEXT: 0000000000202000 0000000000000004 16 .bss -// CHECK-NEXT: 0000000000202000 0000000000000004 16 {{.*}}{{/|\\}}map-file.s.tmp1.o:(COMMON) -// CHECK-NEXT: 0000000000202000 0000000000000004 0 common +// CHECK-NEXT: 000000000020100d 0000000000000000 0 f(int) +// CHECK-NEXT: 0000000000201016 0000000000000000 0 local +// CHECK-NEXT: 0000000000201018 0000000000000002 4 {{.*}}{{/|\\}}map-file.s.tmp2.o:(.text) +// CHECK-NEXT: 0000000000201018 0000000000000000 0 foo +// CHECK-NEXT: 0000000000201019 0000000000000000 0 bar +// CHECK-NEXT: 000000000020101a 0000000000000000 1 {{.*}}{{/|\\}}map-file.s.tmp2.o:(.text.zed) +// CHECK-NEXT: 000000000020101a 0000000000000000 0 zed +// CHECK-NEXT: 000000000020101c 0000000000000000 4 {{.*}}{{/|\\}}map-file.s.tmp3.o:(.text) +// CHECK-NEXT: 000000000020101c 0000000000000000 0 bah +// CHECK-NEXT: 000000000020101c 0000000000000001 4 {{.*}}{{/|\\}}map-file.s.tmp4.a(map-file.s.tmp4.o):(.text) +// CHECK-NEXT: 000000000020101c 0000000000000000 0 baz +// CHECK-NEXT: 0000000000202000 00000000000000c0 8 .dynamic +// CHECK-NEXT: 0000000000202000 00000000000000c0 8 :(.dynamic) +// CHECK-NEXT: 00000000002020c0 0000000000000020 4 .bss.rel.ro +// CHECK-NEXT: 00000000002020c0 0000000000000020 4 :(.bss.rel.ro) +// CHECK-NEXT: 00000000002020c0 0000000000000020 0 copy +// CHECK-NEXT: 0000000000203000 0000000000000004 16 .bss +// CHECK-NEXT: 0000000000203000 0000000000000004 16 {{.*}}{{/|\\}}map-file.s.tmp1.o:(COMMON) +// CHECK-NEXT: 0000000000203000 0000000000000004 0 common // CHECK-NEXT: 0000000000000000 0000000000000008 1 .comment // CHECK-NEXT: 0000000000000000 0000000000000008 1 :(.comment) -// CHECK-NEXT: 0000000000000000 0000000000000120 8 .symtab -// CHECK-NEXT: 0000000000000000 0000000000000120 8 :(.symtab) -// CHECK-NEXT: 0000000000000000 0000000000000039 1 .shstrtab -// CHECK-NEXT: 0000000000000000 0000000000000039 1 :(.shstrtab) -// CHECK-NEXT: 0000000000000000 0000000000000038 1 .strtab -// CHECK-NEXT: 0000000000000000 0000000000000038 1 :(.strtab) +// CHECK-NEXT: 0000000000000000 0000000000000150 8 .symtab +// CHECK-NEXT: 0000000000000000 0000000000000150 8 :(.symtab) +// CHECK-NEXT: 0000000000000000 0000000000000078 1 .shstrtab +// CHECK-NEXT: 0000000000000000 0000000000000078 1 :(.shstrtab) +// CHECK-NEXT: 0000000000000000 0000000000000046 1 .strtab +// CHECK-NEXT: 0000000000000000 0000000000000046 1 :(.strtab) -// RUN: not ld.lld %t1.o %t2.o %t3.o %t4.a -o %t -Map=/ 2>&1 \ +// RUN: not ld.lld %t1.o %t2.o %t3.o %t4.a %t5.so -o %t -Map=/ 2>&1 \ // RUN: | FileCheck -check-prefix=FAIL %s // FAIL: cannot open map file /