84 lines · cpp
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-exception6//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(MangledNameStream, Name, DL);26 }27 return ES.intern(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>(G);46 47 auto Flags = JITSymbolFlags::fromGlobalValue(*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 an55 // __emutls.t symbol too.56 if (GV->hasInitializer()) {57 const auto *InitVal = GV->getInitializer();58 59 // Skip zero-initializers.60 if (isa<ConstantAggregateZero>(InitVal))61 continue;62 const auto *InitIntValue = dyn_cast<ConstantInt>(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(*G);77 if (SymbolToDefinition)78 (*SymbolToDefinition)[MangledName] = G;79 }80}81 82} // End namespace orc.83} // End namespace llvm.84