1299bba404
The SymbolSet class bump-ptr-allocates Symbol objects, but Symbol has a non-trivial destructor (since Symbol's Targets member is a SmallVector): we need to explicitly destroy the Symbol objects to ensure that no memory is leaked. rdar://154778728
54 lines
2.0 KiB
C++
54 lines
2.0 KiB
C++
//===- lib/TextAPI/SymbolSet.cpp - TAPI Symbol Set ------------*- C++-*----===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/TextAPI/SymbolSet.h"
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::MachO;
|
|
|
|
SymbolSet::~SymbolSet() {
|
|
for (auto &[Key, Sym] : Symbols)
|
|
Sym->~Symbol();
|
|
}
|
|
|
|
Symbol *SymbolSet::addGlobalImpl(EncodeKind Kind, StringRef Name,
|
|
SymbolFlags Flags) {
|
|
Name = copyString(Name);
|
|
auto Result = Symbols.try_emplace(SymbolsMapKey{Kind, Name}, nullptr);
|
|
if (Result.second)
|
|
Result.first->second =
|
|
new (Allocator) Symbol{Kind, Name, TargetList(), Flags};
|
|
return Result.first->second;
|
|
}
|
|
|
|
Symbol *SymbolSet::addGlobal(EncodeKind Kind, StringRef Name, SymbolFlags Flags,
|
|
const Target &Targ) {
|
|
auto *Sym = addGlobalImpl(Kind, Name, Flags);
|
|
Sym->addTarget(Targ);
|
|
return Sym;
|
|
}
|
|
|
|
const Symbol *SymbolSet::findSymbol(EncodeKind Kind, StringRef Name,
|
|
ObjCIFSymbolKind ObjCIF) const {
|
|
if (auto result = Symbols.lookup({Kind, Name}))
|
|
return result;
|
|
if ((ObjCIF == ObjCIFSymbolKind::None) || (ObjCIF > ObjCIFSymbolKind::EHType))
|
|
return nullptr;
|
|
assert(ObjCIF <= ObjCIFSymbolKind::EHType &&
|
|
"expected single ObjCIFSymbolKind enum value");
|
|
// Non-complete ObjC Interfaces are represented as global symbols.
|
|
if (ObjCIF == ObjCIFSymbolKind::Class)
|
|
return Symbols.lookup(
|
|
{EncodeKind::GlobalSymbol, (ObjC2ClassNamePrefix + Name).str()});
|
|
if (ObjCIF == ObjCIFSymbolKind::MetaClass)
|
|
return Symbols.lookup(
|
|
{EncodeKind::GlobalSymbol, (ObjC2MetaClassNamePrefix + Name).str()});
|
|
return Symbols.lookup(
|
|
{EncodeKind::GlobalSymbol, (ObjC2EHTypePrefix + Name).str()});
|
|
}
|