1//===----------- Mangling.cpp -- Name Mangling Utilities for ORC ----------===//
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#include "llvm/ExecutionEngine/Orc/Mangling.h"
10#include "llvm/IR/Constants.h"
11#include "llvm/IR/Mangler.h"
12
13#define DEBUG_TYPE "orc"
14
15namespace llvm {
16namespace orc {
17
18MangleAndInterner::MangleAndInterner(ExecutionSession &ES, const DataLayout &DL)
19 : ES(ES), DL(DL) {}
20
21SymbolStringPtr MangleAndInterner::operator()(StringRef Name) {
22 std::string MangledName;
23 {
24 raw_string_ostream MangledNameStream(MangledName);
25 Mangler::getNameWithPrefix(OS&: MangledNameStream, GVName: Name, DL);
26 }
27 return ES.intern(SymName: MangledName);
28}
29
30void IRSymbolMapper::add(ExecutionSession &ES, const ManglingOptions &MO,
31 ArrayRef<GlobalValue *> GVs,
32 SymbolFlagsMap &SymbolFlags,
33 SymbolNameToDefinitionMap *SymbolToDefinition) {
34 if (GVs.empty())
35 return;
36
37 MangleAndInterner Mangle(ES, GVs[0]->getDataLayout());
38 for (auto *G : GVs) {
39 assert(G && "GVs cannot contain null elements");
40 if (!G->hasName() || G->isDeclaration() || G->hasLocalLinkage() ||
41 G->hasAvailableExternallyLinkage() || G->hasAppendingLinkage())
42 continue;
43
44 if (G->isThreadLocal() && MO.EmulatedTLS) {
45 auto *GV = cast<GlobalVariable>(Val: G);
46
47 auto Flags = JITSymbolFlags::fromGlobalValue(GV: *GV);
48
49 auto EmuTLSV = Mangle(("__emutls_v." + GV->getName()).str());
50 SymbolFlags[EmuTLSV] = Flags;
51 if (SymbolToDefinition)
52 (*SymbolToDefinition)[EmuTLSV] = GV;
53
54 // If this GV has a non-zero initializer we'll need to emit an
55 // __emutls.t symbol too.
56 if (GV->hasInitializer()) {
57 const auto *InitVal = GV->getInitializer();
58
59 // Skip zero-initializers.
60 if (isa<ConstantAggregateZero>(Val: InitVal))
61 continue;
62 const auto *InitIntValue = dyn_cast<ConstantInt>(Val: InitVal);
63 if (InitIntValue && InitIntValue->isZero())
64 continue;
65
66 auto EmuTLST = Mangle(("__emutls_t." + GV->getName()).str());
67 SymbolFlags[EmuTLST] = Flags;
68 if (SymbolToDefinition)
69 (*SymbolToDefinition)[EmuTLST] = GV;
70 }
71 continue;
72 }
73
74 // Otherwise we just need a normal linker mangling.
75 auto MangledName = Mangle(G->getName());
76 SymbolFlags[MangledName] = JITSymbolFlags::fromGlobalValue(GV: *G);
77 if (SymbolToDefinition)
78 (*SymbolToDefinition)[MangledName] = G;
79 }
80}
81
82} // End namespace orc.
83} // End namespace llvm.
84