1 //===-LTOModule.h - LLVM Link Time Optimizer ------------------------------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This file declares the LTOModule class.
12 //===----------------------------------------------------------------------===//
17 #include "llvm/Module.h"
18 #include "llvm/ADT/OwningPtr.h"
19 #include "llvm/Target/TargetMachine.h"
20 #include "llvm/ADT/StringMap.h"
22 #include "llvm-c/lto.h"
28 // forward references to llvm classes
39 // C++ class which implements the opaque lto_module_t
43 static bool isBitcodeFile(const void* mem, size_t length);
44 static bool isBitcodeFile(const char* path);
46 static bool isBitcodeFileForTarget(const void* mem,
47 size_t length, const char* triplePrefix);
49 static bool isBitcodeFileForTarget(const char* path,
50 const char* triplePrefix);
52 static LTOModule* makeLTOModule(const char* path,
54 static LTOModule* makeLTOModule(int fd, const char *path,
57 static LTOModule* makeLTOModule(int fd, const char *path,
62 static LTOModule* makeLTOModule(const void* mem, size_t length,
65 const char* getTargetTriple();
66 void setTargetTriple(const char*);
67 uint32_t getSymbolCount();
68 lto_symbol_attributes getSymbolAttributes(uint32_t index);
69 const char* getSymbolName(uint32_t index);
71 llvm::Module * getLLVVMModule() { return _module.get(); }
72 const std::vector<const char*> &getAsmUndefinedRefs() {
73 return _asm_undefines;
77 LTOModule(llvm::Module* m, llvm::TargetMachine* t);
79 bool ParseSymbols(std::string &errMsg);
80 void addDefinedSymbol(llvm::GlobalValue* def,
81 llvm::Mangler& mangler,
83 void addPotentialUndefinedSymbol(llvm::GlobalValue* decl,
84 llvm::Mangler &mangler);
85 void addDefinedFunctionSymbol(llvm::Function* f,
86 llvm::Mangler &mangler);
87 void addDefinedDataSymbol(llvm::GlobalValue* v,
88 llvm::Mangler &mangler);
89 bool addAsmGlobalSymbols(llvm::MCContext &Context,
91 void addAsmGlobalSymbol(const char *,
92 lto_symbol_attributes scope);
93 void addAsmGlobalSymbolUndef(const char *);
94 void addObjCClass(llvm::GlobalVariable* clgv);
95 void addObjCCategory(llvm::GlobalVariable* clgv);
96 void addObjCClassRef(llvm::GlobalVariable* clgv);
97 bool objcClassNameFromExpression(llvm::Constant* c,
100 static bool isTargetMatch(llvm::MemoryBuffer* memBuffer,
101 const char* triplePrefix);
103 static LTOModule* makeLTOModule(llvm::MemoryBuffer* buffer,
104 std::string& errMsg);
105 static llvm::MemoryBuffer* makeBuffer(const void* mem, size_t length);
107 typedef llvm::StringMap<uint8_t> StringSet;
109 struct NameAndAttributes {
111 lto_symbol_attributes attributes;
114 llvm::OwningPtr<llvm::Module> _module;
115 llvm::OwningPtr<llvm::TargetMachine> _target;
116 std::vector<NameAndAttributes> _symbols;
117 // _defines and _undefines only needed to disambiguate tentative definitions
119 llvm::StringMap<NameAndAttributes> _undefines;
120 std::vector<const char*> _asm_undefines;
123 #endif // LTO_MODULE_H