mirror of
https://github.com/capstone-engine/llvm-capstone.git
synced 2024-11-23 05:40:09 +00:00
356139bd02
This adds support for generating Chrome-tracing .json profile traces in the LLD COFF driver. Also add the necessary time scopes, so that the profile trace shows in great detail which tasks are executed. As an example, this is what we see when linking a Unreal Engine executable: 
128 lines
4.3 KiB
C++
128 lines
4.3 KiB
C++
//===- LLDMapFile.cpp -----------------------------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file implements the /lldmap option. It shows lists in order and
|
|
// hierarchically the output sections, input sections, input files and
|
|
// symbol:
|
|
//
|
|
// Address Size Align Out File Symbol
|
|
// 00201000 00000015 4 .text
|
|
// 00201000 0000000e 4 test.o:(.text)
|
|
// 0020100e 00000000 0 local
|
|
// 00201005 00000000 0 f(int)
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "LLDMapFile.h"
|
|
#include "COFFLinkerContext.h"
|
|
#include "SymbolTable.h"
|
|
#include "Symbols.h"
|
|
#include "Writer.h"
|
|
#include "lld/Common/ErrorHandler.h"
|
|
#include "llvm/Support/Parallel.h"
|
|
#include "llvm/Support/TimeProfiler.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::object;
|
|
using namespace lld;
|
|
using namespace lld::coff;
|
|
|
|
using SymbolMapTy =
|
|
DenseMap<const SectionChunk *, SmallVector<DefinedRegular *, 4>>;
|
|
|
|
static constexpr char indent8[] = " "; // 8 spaces
|
|
static constexpr char indent16[] = " "; // 16 spaces
|
|
|
|
// Print out the first three columns of a line.
|
|
static void writeHeader(raw_ostream &os, uint64_t addr, uint64_t size,
|
|
uint64_t align) {
|
|
os << format("%08llx %08llx %5lld ", addr, size, align);
|
|
}
|
|
|
|
// Returns a list of all symbols that we want to print out.
|
|
static std::vector<DefinedRegular *> getSymbols(const COFFLinkerContext &ctx) {
|
|
std::vector<DefinedRegular *> v;
|
|
for (ObjFile *file : ctx.objFileInstances)
|
|
for (Symbol *b : file->getSymbols())
|
|
if (auto *sym = dyn_cast_or_null<DefinedRegular>(b))
|
|
if (sym && !sym->getCOFFSymbol().isSectionDefinition())
|
|
v.push_back(sym);
|
|
return v;
|
|
}
|
|
|
|
// Returns a map from sections to their symbols.
|
|
static SymbolMapTy getSectionSyms(ArrayRef<DefinedRegular *> syms) {
|
|
SymbolMapTy ret;
|
|
for (DefinedRegular *s : syms)
|
|
ret[s->getChunk()].push_back(s);
|
|
|
|
// Sort symbols by address.
|
|
for (auto &it : ret) {
|
|
SmallVectorImpl<DefinedRegular *> &v = it.second;
|
|
std::stable_sort(v.begin(), v.end(), [](DefinedRegular *a, DefinedRegular *b) {
|
|
return a->getRVA() < b->getRVA();
|
|
});
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
// Construct a map from symbols to their stringified representations.
|
|
static DenseMap<DefinedRegular *, std::string>
|
|
getSymbolStrings(const COFFLinkerContext &ctx,
|
|
ArrayRef<DefinedRegular *> syms) {
|
|
std::vector<std::string> str(syms.size());
|
|
parallelFor((size_t)0, syms.size(), [&](size_t i) {
|
|
raw_string_ostream os(str[i]);
|
|
writeHeader(os, syms[i]->getRVA(), 0, 0);
|
|
os << indent16 << toString(ctx, *syms[i]);
|
|
});
|
|
|
|
DenseMap<DefinedRegular *, std::string> ret;
|
|
for (size_t i = 0, e = syms.size(); i < e; ++i)
|
|
ret[syms[i]] = std::move(str[i]);
|
|
return ret;
|
|
}
|
|
|
|
void lld::coff::writeLLDMapFile(const COFFLinkerContext &ctx) {
|
|
if (ctx.config.lldmapFile.empty())
|
|
return;
|
|
|
|
llvm::TimeTraceScope timeScope(".lldmap file");
|
|
std::error_code ec;
|
|
raw_fd_ostream os(ctx.config.lldmapFile, ec, sys::fs::OF_None);
|
|
if (ec)
|
|
fatal("cannot open " + ctx.config.lldmapFile + ": " + ec.message());
|
|
|
|
// Collect symbol info that we want to print out.
|
|
std::vector<DefinedRegular *> syms = getSymbols(ctx);
|
|
SymbolMapTy sectionSyms = getSectionSyms(syms);
|
|
DenseMap<DefinedRegular *, std::string> symStr = getSymbolStrings(ctx, syms);
|
|
|
|
// Print out the header line.
|
|
os << "Address Size Align Out In Symbol\n";
|
|
|
|
// Print out file contents.
|
|
for (OutputSection *sec : ctx.outputSections) {
|
|
writeHeader(os, sec->getRVA(), sec->getVirtualSize(), /*align=*/pageSize);
|
|
os << sec->name << '\n';
|
|
|
|
for (Chunk *c : sec->chunks) {
|
|
auto *sc = dyn_cast<SectionChunk>(c);
|
|
if (!sc)
|
|
continue;
|
|
|
|
writeHeader(os, sc->getRVA(), sc->getSize(), sc->getAlignment());
|
|
os << indent8 << sc->file->getName() << ":(" << sc->getSectionName()
|
|
<< ")\n";
|
|
for (DefinedRegular *sym : sectionSyms[sc])
|
|
os << symStr[sym] << '\n';
|
|
}
|
|
}
|
|
}
|