IRObjectFile.h 2.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091
  1. //===- IRObjectFile.h - LLVM IR object file implementation ------*- C++ -*-===//
  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 declares the IRObjectFile template class.
  10. //
  11. //===----------------------------------------------------------------------===//
  12. #ifndef LLVM_OBJECT_IROBJECTFILE_H
  13. #define LLVM_OBJECT_IROBJECTFILE_H
  14. #include "llvm/ADT/PointerUnion.h"
  15. #include "llvm/Object/IRSymtab.h"
  16. #include "llvm/Object/ModuleSymbolTable.h"
  17. #include "llvm/Object/SymbolicFile.h"
  18. namespace llvm {
  19. class BitcodeModule;
  20. class Mangler;
  21. class Module;
  22. class GlobalValue;
  23. class Triple;
  24. namespace object {
  25. class ObjectFile;
  26. class IRObjectFile : public SymbolicFile {
  27. std::vector<std::unique_ptr<Module>> Mods;
  28. ModuleSymbolTable SymTab;
  29. IRObjectFile(MemoryBufferRef Object,
  30. std::vector<std::unique_ptr<Module>> Mods);
  31. public:
  32. ~IRObjectFile() override;
  33. void moveSymbolNext(DataRefImpl &Symb) const override;
  34. Error printSymbolName(raw_ostream &OS, DataRefImpl Symb) const override;
  35. Expected<uint32_t> getSymbolFlags(DataRefImpl Symb) const override;
  36. basic_symbol_iterator symbol_begin() const override;
  37. basic_symbol_iterator symbol_end() const override;
  38. StringRef getTargetTriple() const;
  39. static bool classof(const Binary *v) {
  40. return v->isIR();
  41. }
  42. using module_iterator =
  43. pointee_iterator<std::vector<std::unique_ptr<Module>>::const_iterator,
  44. const Module>;
  45. module_iterator module_begin() const { return module_iterator(Mods.begin()); }
  46. module_iterator module_end() const { return module_iterator(Mods.end()); }
  47. iterator_range<module_iterator> modules() const {
  48. return make_range(module_begin(), module_end());
  49. }
  50. /// Finds and returns bitcode embedded in the given object file, or an
  51. /// error code if not found.
  52. static Expected<MemoryBufferRef> findBitcodeInObject(const ObjectFile &Obj);
  53. /// Finds and returns bitcode in the given memory buffer (which may
  54. /// be either a bitcode file or a native object file with embedded bitcode),
  55. /// or an error code if not found.
  56. static Expected<MemoryBufferRef>
  57. findBitcodeInMemBuffer(MemoryBufferRef Object);
  58. static Expected<std::unique_ptr<IRObjectFile>> create(MemoryBufferRef Object,
  59. LLVMContext &Context);
  60. };
  61. /// The contents of a bitcode file and its irsymtab. Any underlying data
  62. /// for the irsymtab are owned by Symtab and Strtab.
  63. struct IRSymtabFile {
  64. std::vector<BitcodeModule> Mods;
  65. SmallVector<char, 0> Symtab, Strtab;
  66. irsymtab::Reader TheReader;
  67. };
  68. /// Reads a bitcode file, creating its irsymtab if necessary.
  69. Expected<IRSymtabFile> readIRSymtab(MemoryBufferRef MBRef);
  70. }
  71. }
  72. #endif