xref: /freebsd/contrib/llvm-project/llvm/lib/TextAPI/RecordVisitor.cpp (revision 0fca6ea1d4eea4c934cfff25ac9ee8ad6fe95583)
1 //===- RecordVisitor.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 TAPI Record Visitor.
10 ///
11 //===----------------------------------------------------------------------===//
12 
13 #include "llvm/TextAPI/RecordVisitor.h"
14 
15 using namespace llvm;
16 using namespace llvm::MachO;
17 
~RecordVisitor()18 RecordVisitor::~RecordVisitor() {}
visitObjCInterface(const ObjCInterfaceRecord &)19 void RecordVisitor::visitObjCInterface(const ObjCInterfaceRecord &) {}
visitObjCCategory(const ObjCCategoryRecord &)20 void RecordVisitor::visitObjCCategory(const ObjCCategoryRecord &) {}
21 
shouldSkipRecord(const Record & R,const bool RecordUndefs)22 static bool shouldSkipRecord(const Record &R, const bool RecordUndefs) {
23   if (R.isExported())
24     return false;
25 
26   // Skip non exported symbols unless for flat namespace libraries.
27   return !(RecordUndefs && R.isUndefined());
28 }
29 
visitGlobal(const GlobalRecord & GR)30 void SymbolConverter::visitGlobal(const GlobalRecord &GR) {
31   auto [SymName, SymKind, InterfaceType] = parseSymbol(GR.getName());
32   if (shouldSkipRecord(GR, RecordUndefs))
33     return;
34   Symbols->addGlobal(SymKind, SymName, GR.getFlags(), Targ);
35 
36   if (InterfaceType == ObjCIFSymbolKind::None) {
37     Symbols->addGlobal(SymKind, SymName, GR.getFlags(), Targ);
38     return;
39   }
40 
41   // It is impossible to hold a complete ObjCInterface with a single
42   // GlobalRecord, so continue to treat this symbol a generic global.
43   Symbols->addGlobal(EncodeKind::GlobalSymbol, GR.getName(), GR.getFlags(),
44                      Targ);
45 }
46 
addIVars(const ArrayRef<ObjCIVarRecord * > IVars,StringRef ContainerName)47 void SymbolConverter::addIVars(const ArrayRef<ObjCIVarRecord *> IVars,
48                                StringRef ContainerName) {
49   for (auto *IV : IVars) {
50     if (shouldSkipRecord(*IV, RecordUndefs))
51       continue;
52     std::string Name =
53         ObjCIVarRecord::createScopedName(ContainerName, IV->getName());
54     Symbols->addGlobal(EncodeKind::ObjectiveCInstanceVariable, Name,
55                        IV->getFlags(), Targ);
56   }
57 }
58 
visitObjCInterface(const ObjCInterfaceRecord & ObjCR)59 void SymbolConverter::visitObjCInterface(const ObjCInterfaceRecord &ObjCR) {
60   if (!shouldSkipRecord(ObjCR, RecordUndefs)) {
61     if (ObjCR.isCompleteInterface()) {
62       Symbols->addGlobal(EncodeKind::ObjectiveCClass, ObjCR.getName(),
63                          ObjCR.getFlags(), Targ);
64       if (ObjCR.hasExceptionAttribute())
65         Symbols->addGlobal(EncodeKind::ObjectiveCClassEHType, ObjCR.getName(),
66                            ObjCR.getFlags(), Targ);
67     } else {
68       // Because there is not a complete interface, visit individual symbols
69       // instead.
70       if (ObjCR.isExportedSymbol(ObjCIFSymbolKind::EHType))
71         Symbols->addGlobal(EncodeKind::GlobalSymbol,
72                            (ObjC2EHTypePrefix + ObjCR.getName()).str(),
73                            ObjCR.getFlags(), Targ);
74       if (ObjCR.isExportedSymbol(ObjCIFSymbolKind::Class))
75         Symbols->addGlobal(EncodeKind::GlobalSymbol,
76                            (ObjC2ClassNamePrefix + ObjCR.getName()).str(),
77                            ObjCR.getFlags(), Targ);
78       if (ObjCR.isExportedSymbol(ObjCIFSymbolKind::MetaClass))
79         Symbols->addGlobal(EncodeKind::GlobalSymbol,
80                            (ObjC2MetaClassNamePrefix + ObjCR.getName()).str(),
81                            ObjCR.getFlags(), Targ);
82     }
83   }
84 
85   addIVars(ObjCR.getObjCIVars(), ObjCR.getName());
86   for (const auto *Cat : ObjCR.getObjCCategories())
87     addIVars(Cat->getObjCIVars(), ObjCR.getName());
88 }
89 
visitObjCCategory(const ObjCCategoryRecord & Cat)90 void SymbolConverter::visitObjCCategory(const ObjCCategoryRecord &Cat) {
91   addIVars(Cat.getObjCIVars(), Cat.getSuperClassName());
92 }
93