1//===- Symbol.cpp ---------------------------------------------------------===//
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// Implements the Symbol.
10//
11//===----------------------------------------------------------------------===//
12
13#include "llvm/TextAPI/Symbol.h"
14#include <string>
15
16namespace llvm {
17namespace MachO {
18
19#if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
20LLVM_DUMP_METHOD void Symbol::dump(raw_ostream &OS) const {
21  std::string Result;
22  if (isUndefined())
23    Result += "(undef) ";
24  if (isWeakDefined())
25    Result += "(weak-def) ";
26  if (isWeakReferenced())
27    Result += "(weak-ref) ";
28  if (isThreadLocalValue())
29    Result += "(tlv) ";
30  switch (Kind) {
31  case SymbolKind::GlobalSymbol:
32    Result += Name.str();
33    break;
34  case SymbolKind::ObjectiveCClass:
35    Result += "(ObjC Class) " + Name.str();
36    break;
37  case SymbolKind::ObjectiveCClassEHType:
38    Result += "(ObjC Class EH) " + Name.str();
39    break;
40  case SymbolKind::ObjectiveCInstanceVariable:
41    Result += "(ObjC IVar) " + Name.str();
42    break;
43  }
44  OS << Result;
45}
46#endif
47
48Symbol::const_filtered_target_range
49Symbol::targets(ArchitectureSet Architectures) const {
50  std::function<bool(const Target &)> FN =
51      [Architectures](const Target &Target) {
52        return Architectures.has(Target.Arch);
53      };
54  return make_filter_range(Targets, FN);
55}
56
57bool Symbol::operator==(const Symbol &O) const {
58  // Older Tapi files do not express all these symbol flags. In those
59  // cases, ignore those differences.
60  auto RemoveFlag = [](const Symbol &Sym, SymbolFlags &Flag) {
61    if (Sym.isData())
62      Flag &= ~SymbolFlags::Data;
63    if (Sym.isText())
64      Flag &= ~SymbolFlags::Text;
65  };
66  SymbolFlags LHSFlags = Flags;
67  SymbolFlags RHSFlags = O.Flags;
68  // Ignore Text and Data for now.
69  RemoveFlag(*this, LHSFlags);
70  RemoveFlag(O, RHSFlags);
71  return std::tie(Name, Kind, Targets, LHSFlags) ==
72         std::tie(O.Name, O.Kind, O.Targets, RHSFlags);
73}
74
75SimpleSymbol parseSymbol(StringRef SymName, const SymbolFlags Flags) {
76  if (SymName.starts_with(ObjC1ClassNamePrefix))
77    return {SymName.drop_front(ObjC1ClassNamePrefix.size()),
78            SymbolKind::ObjectiveCClass};
79  if (SymName.starts_with(ObjC2ClassNamePrefix))
80    return {SymName.drop_front(ObjC2ClassNamePrefix.size()),
81            SymbolKind::ObjectiveCClass};
82  if (SymName.starts_with(ObjC2MetaClassNamePrefix))
83    return {SymName.drop_front(ObjC2MetaClassNamePrefix.size()),
84            SymbolKind::ObjectiveCClass};
85  if (SymName.starts_with(ObjC2EHTypePrefix)) {
86    // When classes without ehtype are used in try/catch blocks
87    // a weak-defined symbol is exported. In those cases, treat these as a
88    // global instead.
89    if ((Flags & SymbolFlags::WeakDefined) == SymbolFlags::WeakDefined)
90      return {SymName, SymbolKind::GlobalSymbol};
91    return {SymName.drop_front(ObjC2EHTypePrefix.size()),
92            SymbolKind::ObjectiveCClassEHType};
93  }
94
95  if (SymName.starts_with(ObjC2IVarPrefix))
96    return {SymName.drop_front(ObjC2IVarPrefix.size()),
97            SymbolKind::ObjectiveCInstanceVariable};
98  return {SymName, SymbolKind::GlobalSymbol};
99}
100
101} // end namespace MachO.
102} // end namespace llvm.
103