1//===- IRObjectFile.h - LLVM IR object file implementation ------*- C++ -*-===// 2// 3// The LLVM Compiler Infrastructure 4// 5// This file is distributed under the University of Illinois Open Source 6// License. See LICENSE.TXT for details. 7// 8//===----------------------------------------------------------------------===// 9// 10// This file declares the IRObjectFile template class. 11// 12//===----------------------------------------------------------------------===// 13 14#ifndef LLVM_OBJECT_IROBJECTFILE_H 15#define LLVM_OBJECT_IROBJECTFILE_H 16 17#include "llvm/Object/SymbolicFile.h" 18 19namespace llvm { 20class Mangler; 21class Module; 22class GlobalValue; 23 24namespace object { 25class ObjectFile; 26 27class IRObjectFile : public SymbolicFile { 28 std::unique_ptr<Module> M; 29 std::unique_ptr<Mangler> Mang; 30 std::vector<std::pair<std::string, uint32_t>> AsmSymbols; 31 32public: 33 IRObjectFile(MemoryBufferRef Object, std::unique_ptr<Module> M); 34 ~IRObjectFile() override; 35 void moveSymbolNext(DataRefImpl &Symb) const override; 36 std::error_code printSymbolName(raw_ostream &OS, 37 DataRefImpl Symb) const override; 38 uint32_t getSymbolFlags(DataRefImpl Symb) const override; 39 GlobalValue *getSymbolGV(DataRefImpl Symb); 40 const GlobalValue *getSymbolGV(DataRefImpl Symb) const { 41 return const_cast<IRObjectFile *>(this)->getSymbolGV(Symb); 42 } 43 basic_symbol_iterator symbol_begin_impl() const override; 44 basic_symbol_iterator symbol_end_impl() const override; 45 46 const Module &getModule() const { 47 return const_cast<IRObjectFile*>(this)->getModule(); 48 } 49 Module &getModule() { 50 return *M; 51 } 52 std::unique_ptr<Module> takeModule(); 53 54 static inline bool classof(const Binary *v) { 55 return v->isIR(); 56 } 57 58 /// \brief Finds and returns bitcode embedded in the given object file, or an 59 /// error code if not found. 60 static ErrorOr<MemoryBufferRef> findBitcodeInObject(const ObjectFile &Obj); 61 62 /// \brief Finds and returns bitcode in the given memory buffer (which may 63 /// be either a bitcode file or a native object file with embedded bitcode), 64 /// or an error code if not found. 65 static ErrorOr<MemoryBufferRef> 66 findBitcodeInMemBuffer(MemoryBufferRef Object); 67 68 static ErrorOr<std::unique_ptr<IRObjectFile>> create(MemoryBufferRef Object, 69 LLVMContext &Context); 70}; 71} 72} 73 74#endif 75