1 //===- lib/TextAPI/SymbolSet.cpp - TAPI Symbol Set ------------*- C++-*----===// 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/TextAPI/SymbolSet.h" 10 11 using namespace llvm; 12 using namespace llvm::MachO; 13 14 Symbol *SymbolSet::addGlobalImpl(EncodeKind Kind, StringRef Name, 15 SymbolFlags Flags) { 16 Name = copyString(Name); 17 auto Result = Symbols.try_emplace(SymbolsMapKey{Kind, Name}, nullptr); 18 if (Result.second) 19 Result.first->second = 20 new (Allocator) Symbol{Kind, Name, TargetList(), Flags}; 21 return Result.first->second; 22 } 23 24 Symbol *SymbolSet::addGlobal(EncodeKind Kind, StringRef Name, SymbolFlags Flags, 25 const Target &Targ) { 26 auto *Sym = addGlobalImpl(Kind, Name, Flags); 27 Sym->addTarget(Targ); 28 return Sym; 29 } 30 31 const Symbol *SymbolSet::findSymbol(EncodeKind Kind, StringRef Name, 32 ObjCIFSymbolKind ObjCIF) const { 33 if (auto result = Symbols.lookup({Kind, Name})) 34 return result; 35 if ((ObjCIF == ObjCIFSymbolKind::None) || (ObjCIF > ObjCIFSymbolKind::EHType)) 36 return nullptr; 37 assert(ObjCIF <= ObjCIFSymbolKind::EHType && 38 "expected single ObjCIFSymbolKind enum value"); 39 // Non-complete ObjC Interfaces are represented as global symbols. 40 if (ObjCIF == ObjCIFSymbolKind::Class) 41 return Symbols.lookup( 42 {EncodeKind::GlobalSymbol, (ObjC2ClassNamePrefix + Name).str()}); 43 if (ObjCIF == ObjCIFSymbolKind::MetaClass) 44 return Symbols.lookup( 45 {EncodeKind::GlobalSymbol, (ObjC2MetaClassNamePrefix + Name).str()}); 46 return Symbols.lookup( 47 {EncodeKind::GlobalSymbol, (ObjC2EHTypePrefix + Name).str()}); 48 } 49