1 //===- MapFile.cpp --------------------------------------------------------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 // 9 // This file implements the -map option. It shows lists in order and 10 // hierarchically the outputFile, arch, input files, output sections and 11 // symbol: 12 // 13 // # Path: test 14 // # Arch: x86_84 15 // # Object files: 16 // [ 0] linker synthesized 17 // [ 1] a.o 18 // # Sections: 19 // # Address Size Segment Section 20 // 0x1000005C0 0x0000004C __TEXT __text 21 // # Symbols: 22 // # Address File Name 23 // 0x1000005C0 [ 1] _main 24 // 25 //===----------------------------------------------------------------------===// 26 27 #include "MapFile.h" 28 #include "Config.h" 29 #include "InputFiles.h" 30 #include "InputSection.h" 31 #include "OutputSection.h" 32 #include "OutputSegment.h" 33 #include "Symbols.h" 34 #include "Target.h" 35 #include "llvm/Support/Parallel.h" 36 #include "llvm/Support/TimeProfiler.h" 37 38 using namespace llvm; 39 using namespace llvm::sys; 40 using namespace lld; 41 using namespace lld::macho; 42 43 using SymbolMapTy = DenseMap<const InputSection *, SmallVector<Defined *, 4>>; 44 45 // Returns a map from sections to their symbols. 46 static SymbolMapTy getSectionSyms(ArrayRef<Defined *> syms) { 47 SymbolMapTy ret; 48 for (Defined *dr : syms) 49 ret[dr->isec].push_back(dr); 50 51 // Sort symbols by address. We want to print out symbols in the order they 52 // appear in the output file rather than the order they appeared in the input 53 // files. 54 for (auto &it : ret) 55 llvm::stable_sort(it.second, [](Defined *a, Defined *b) { 56 return a->getVA() < b->getVA(); 57 }); 58 return ret; 59 } 60 61 // Returns a list of all symbols that we want to print out. 62 static std::vector<Defined *> getSymbols() { 63 std::vector<Defined *> v; 64 for (InputFile *file : inputFiles) 65 if (isa<ObjFile>(file)) 66 for (Symbol *sym : file->symbols) 67 if (auto *d = dyn_cast_or_null<Defined>(sym)) 68 if (d->isLive() && d->isec && d->getFile() == file) 69 v.push_back(d); 70 return v; 71 } 72 73 // Construct a map from symbols to their stringified representations. 74 // Demangling symbols (which is what toString() does) is slow, so 75 // we do that in batch using parallel-for. 76 static DenseMap<Symbol *, std::string> 77 getSymbolStrings(ArrayRef<Defined *> syms) { 78 std::vector<std::string> str(syms.size()); 79 parallelForEachN(0, syms.size(), [&](size_t i) { 80 raw_string_ostream os(str[i]); 81 os << toString(*syms[i]); 82 }); 83 84 DenseMap<Symbol *, std::string> ret; 85 for (size_t i = 0, e = syms.size(); i < e; ++i) 86 ret[syms[i]] = std::move(str[i]); 87 return ret; 88 } 89 90 void macho::writeMapFile() { 91 if (config->mapFile.empty()) 92 return; 93 94 TimeTraceScope timeScope("Write map file"); 95 96 // Open a map file for writing. 97 std::error_code ec; 98 raw_fd_ostream os(config->mapFile, ec, sys::fs::OF_None); 99 if (ec) { 100 error("cannot open " + config->mapFile + ": " + ec.message()); 101 return; 102 } 103 104 // Dump output path. 105 os << format("# Path: %s\n", config->outputFile.str().c_str()); 106 107 // Dump output architecture. 108 os << format("# Arch: %s\n", 109 getArchitectureName(config->arch()).str().c_str()); 110 111 // Dump table of object files. 112 os << "# Object files:\n"; 113 os << format("[%3u] %s\n", 0, (const char *)"linker synthesized"); 114 uint32_t fileIndex = 1; 115 DenseMap<lld::macho::InputFile *, uint32_t> readerToFileOrdinal; 116 for (InputFile *file : inputFiles) { 117 if (isa<ObjFile>(file)) { 118 os << format("[%3u] %s\n", fileIndex, file->getName().str().c_str()); 119 readerToFileOrdinal[file] = fileIndex++; 120 } 121 } 122 123 // Collect symbol info that we want to print out. 124 std::vector<Defined *> syms = getSymbols(); 125 SymbolMapTy sectionSyms = getSectionSyms(syms); 126 DenseMap<Symbol *, std::string> symStr = getSymbolStrings(syms); 127 128 // Dump table of sections 129 os << "# Sections:\n"; 130 os << "# Address\tSize \tSegment\tSection\n"; 131 for (OutputSegment *seg : outputSegments) 132 for (OutputSection *osec : seg->getSections()) { 133 if (osec->isHidden()) 134 continue; 135 136 os << format("0x%08llX\t0x%08llX\t%s\t%s\n", osec->addr, osec->getSize(), 137 seg->name.str().c_str(), osec->name.str().c_str()); 138 } 139 140 // Dump table of symbols 141 os << "# Symbols:\n"; 142 os << "# Address\t File Name\n"; 143 for (InputSection *isec : inputSections) { 144 auto symsIt = sectionSyms.find(isec); 145 if (symsIt == sectionSyms.end()) 146 continue; 147 for (Symbol *sym : symsIt->second) { 148 os << format("0x%08llX\t[%3u] %s\n", sym->getVA(), 149 readerToFileOrdinal[sym->getFile()], symStr[sym].c_str()); 150 } 151 } 152 153 // TODO: when we implement -dead_strip, we should dump dead stripped symbols 154 } 155