llvm/lib/Object/TapiFile.cpp

   1 //===- TapiFile.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 // This file defines the Text-based Dynamcic Library Stub format.
  10 //
  11 //===----------------------------------------------------------------------===//
  12
  13 #include "llvm/Object/TapiFile.h"
  14 #include "llvm/ADT/StringRef.h"
  15 #include "llvm/BinaryFormat/MachO.h"
  16 #include "llvm/Support/MemoryBufferRef.h"
  17 #include "llvm/TextAPI/ArchitectureSet.h"
  18 #include "llvm/TextAPI/InterfaceFile.h"
  19 #include "llvm/TextAPI/Platform.h"
  20 #include "llvm/TextAPI/Symbol.h"
  21
  22 using namespace llvm;
  23 using namespace MachO;
  24 using namespace object;
  25
  26 static uint32_t getFlags(const Symbol *Sym) {
  27   uint32_t Flags = BasicSymbolRef::SF_Global;
  28   if (Sym->isUndefined())
  29     Flags |= BasicSymbolRef::SF_Undefined;
  30   else
  31     Flags |= BasicSymbolRef::SF_Exported;
  32
  33   if (Sym->isWeakDefined() || Sym->isWeakReferenced())
  34     Flags |= BasicSymbolRef::SF_Weak;
  35
  36   return Flags;
  37 }
  38
  39 static SymbolRef::Type getType(const Symbol *Sym) {
  40   SymbolRef::Type Type = SymbolRef::ST_Unknown;
  41   if (Sym->isData())
  42     Type = SymbolRef::ST_Data;
  43   else if (Sym->isText())
  44     Type = SymbolRef::ST_Function;
  45
  46   return Type;
  47 }
  48
  49 TapiFile::TapiFile(MemoryBufferRef Source, const InterfaceFile &Interface,
  50                    Architecture Arch)
  51     : SymbolicFile(ID_TapiFile, Source), Arch(Arch),
  52       FileKind(Interface.getFileType()) {
  53   for (const auto *Symbol : Interface.symbols()) {
  54     if (!Symbol->getArchitectures().has(Arch))
  55       continue;
  56
  57     switch (Symbol->getKind()) {
  58     case EncodeKind::GlobalSymbol:
  59       Symbols.emplace_back(StringRef(), Symbol->getName(), getFlags(Symbol),
  60                            ::getType(Symbol));
  61       break;
  62     case EncodeKind::ObjectiveCClass:
  63       if (Interface.getPlatforms().count(PLATFORM_MACOS) && Arch == AK_i386) {
  64         Symbols.emplace_back(ObjC1ClassNamePrefix, Symbol->getName(),
  65                              getFlags(Symbol), ::getType(Symbol));
  66       } else {
  67         Symbols.emplace_back(ObjC2ClassNamePrefix, Symbol->getName(),
  68                              getFlags(Symbol), ::getType(Symbol));
  69         Symbols.emplace_back(ObjC2MetaClassNamePrefix, Symbol->getName(),
  70                              getFlags(Symbol), ::getType(Symbol));
  71       }
  72       break;
  73     case EncodeKind::ObjectiveCClassEHType:
  74       Symbols.emplace_back(ObjC2EHTypePrefix, Symbol->getName(),
  75                            getFlags(Symbol), ::getType(Symbol));
  76       break;
  77     case EncodeKind::ObjectiveCInstanceVariable:
  78       Symbols.emplace_back(ObjC2IVarPrefix, Symbol->getName(), getFlags(Symbol),
  79                            ::getType(Symbol));
  80       break;
  81     }
  82   }
  83 }
  84
  85 TapiFile::~TapiFile() = default;
  86
  87 void TapiFile::moveSymbolNext(DataRefImpl &DRI) const { DRI.d.a++; }
  88
  89 Error TapiFile::printSymbolName(raw_ostream &OS, DataRefImpl DRI) const {
  90   assert(DRI.d.a < Symbols.size() && "Attempt to access symbol out of bounds");
  91   const Symbol &Sym = Symbols[DRI.d.a];
  92   OS << Sym.Prefix << Sym.Name;
  93   return Error::success();
  94 }
  95
  96 Expected<SymbolRef::Type> TapiFile::getSymbolType(DataRefImpl DRI) const {
  97   assert(DRI.d.a < Symbols.size() && "Attempt to access symbol out of bounds");
  98   return Symbols[DRI.d.a].Type;
  99 }
 100
 101 Expected<uint32_t> TapiFile::getSymbolFlags(DataRefImpl DRI) const {
 102   assert(DRI.d.a < Symbols.size() && "Attempt to access symbol out of bounds");
 103   return Symbols[DRI.d.a].Flags;
 104 }
 105
 106 basic_symbol_iterator TapiFile::symbol_begin() const {
 107   DataRefImpl DRI;
 108   DRI.d.a = 0;
 109   return BasicSymbolRef{DRI, this};
 110 }
 111
 112 basic_symbol_iterator TapiFile::symbol_end() const {
 113   DataRefImpl DRI;
 114   DRI.d.a = Symbols.size();
 115   return BasicSymbolRef{DRI, this};
 116 }