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
44 static bool isBitcodeFile(const void* mem, size_t length);
45 static bool isBitcodeFile(const char* path);
47 static bool isBitcodeFileForTarget(const void* mem,
48 size_t length, const char* triplePrefix);
50 static bool isBitcodeFileForTarget(const char* path,
51 const char* triplePrefix);
53 static LTOModule* makeLTOModule(const char* path, std::string& errMsg);
54 static LTOModule* makeLTOModule(const void* mem, size_t length,
57 const char* getTargetTriple();
58 uint32_t getSymbolCount();
59 lto_symbol_attributes getSymbolAttributes(uint32_t index);
60 const char* getSymbolName(uint32_t index);
62 llvm::Module * getLLVVMModule() { return _module.get(); }
65 LTOModule(llvm::Module* m, llvm::TargetMachine* t);
67 void lazyParseSymbols();
68 void addDefinedSymbol(llvm::GlobalValue* def,
69 llvm::Mangler& mangler,
71 void addPotentialUndefinedSymbol(llvm::GlobalValue* decl,
72 llvm::Mangler &mangler);
73 void findExternalRefs(llvm::Value* value,
74 llvm::Mangler& mangler);
75 void addDefinedFunctionSymbol(llvm::Function* f,
76 llvm::Mangler &mangler);
77 void addDefinedDataSymbol(llvm::GlobalValue* v,
78 llvm::Mangler &mangler);
79 void addAsmGlobalSymbol(const char *);
80 static bool isTargetMatch(llvm::MemoryBuffer* memBuffer,
81 const char* triplePrefix);
83 static LTOModule* makeLTOModule(llvm::MemoryBuffer* buffer,
85 static llvm::MemoryBuffer* makeBuffer(const void* mem, size_t length);
87 typedef llvm::StringMap<uint8_t> StringSet;
89 struct NameAndAttributes {
91 lto_symbol_attributes attributes;
94 llvm::OwningPtr<llvm::Module> _module;
95 llvm::OwningPtr<llvm::TargetMachine> _target;
97 std::vector<NameAndAttributes> _symbols;
98 // _defines and _undefines only needed to disambiguate tentative definitions
100 StringSet _undefines;
103 extern std::string getFeatureString(const char *TargetTriple);
105 #endif // LTO_MODULE_H