177595fc35642f990bfc5ad05b8e68d4056695ecaNick Kledzik//===-LTOModule.h - LLVM Link Time Optimizer ------------------------------===// 277595fc35642f990bfc5ad05b8e68d4056695ecaNick Kledzik// 377595fc35642f990bfc5ad05b8e68d4056695ecaNick Kledzik// The LLVM Compiler Infrastructure 477595fc35642f990bfc5ad05b8e68d4056695ecaNick Kledzik// 577595fc35642f990bfc5ad05b8e68d4056695ecaNick Kledzik// This file is distributed under the University of Illinois Open Source 677595fc35642f990bfc5ad05b8e68d4056695ecaNick Kledzik// License. See LICENSE.TXT for details. 73c6b29b7e96f4fd0c9f5aae521d52cbf2b4493c2Bill Wendling// 877595fc35642f990bfc5ad05b8e68d4056695ecaNick Kledzik//===----------------------------------------------------------------------===// 977595fc35642f990bfc5ad05b8e68d4056695ecaNick Kledzik// 103c6b29b7e96f4fd0c9f5aae521d52cbf2b4493c2Bill Wendling// This file declares the LTOModule class. 1177595fc35642f990bfc5ad05b8e68d4056695ecaNick Kledzik// 1277595fc35642f990bfc5ad05b8e68d4056695ecaNick Kledzik//===----------------------------------------------------------------------===// 1377595fc35642f990bfc5ad05b8e68d4056695ecaNick Kledzik 1477595fc35642f990bfc5ad05b8e68d4056695ecaNick Kledzik#ifndef LTO_MODULE_H 1577595fc35642f990bfc5ad05b8e68d4056695ecaNick Kledzik#define LTO_MODULE_H 1677595fc35642f990bfc5ad05b8e68d4056695ecaNick Kledzik 17f010c464a11444733ec67e31aace8bcebeaf2588Chandler Carruth#include "llvm-c/lto.h" 18f010c464a11444733ec67e31aace8bcebeaf2588Chandler Carruth#include "llvm/ADT/OwningPtr.h" 19f010c464a11444733ec67e31aace8bcebeaf2588Chandler Carruth#include "llvm/ADT/StringMap.h" 200b8c9a80f20772c3793201ab5b251d3520b9cea3Chandler Carruth#include "llvm/IR/Module.h" 21b9bff969e96a24da18d6c5d0d623a7d790d41fa2Bill Wendling#include "llvm/MC/MCContext.h" 22a7280fd61933f1268f69045cfbea46edb739726dBill Wendling#include "llvm/Target/Mangler.h" 2377595fc35642f990bfc5ad05b8e68d4056695ecaNick Kledzik#include "llvm/Target/TargetMachine.h" 24ef194ed74033eba099f4f391ffdfc176f9aa6f26Nick Kledzik#include <string> 25f010c464a11444733ec67e31aace8bcebeaf2588Chandler Carruth#include <vector> 26ef194ed74033eba099f4f391ffdfc176f9aa6f26Nick Kledzik 273bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling// Forward references to llvm classes. 28ef194ed74033eba099f4f391ffdfc176f9aa6f26Nick Kledziknamespace llvm { 2962cf01e4c5de703e7712c72d12f577153b373cbaBill Wendling class Function; 3030fe94ea43bd122fb4fd26b84bcf62f8096d4293Bill Wendling class GlobalValue; 3162cf01e4c5de703e7712c72d12f577153b373cbaBill Wendling class MemoryBuffer; 329ac0aaa6afe4162b4db32833f9f176ac05ce6f8fBill Wendling class TargetOptions; 3330fe94ea43bd122fb4fd26b84bcf62f8096d4293Bill Wendling class Value; 34ef194ed74033eba099f4f391ffdfc176f9aa6f26Nick Kledzik} 3577595fc35642f990bfc5ad05b8e68d4056695ecaNick Kledzik 36ab53bc76fd0f4863c9bdccf666fe8eed3d6bf702Bill Wendling//===----------------------------------------------------------------------===// 37ab53bc76fd0f4863c9bdccf666fe8eed3d6bf702Bill Wendling/// LTOModule - C++ class which implements the opaque lto_module_t type. 38ab53bc76fd0f4863c9bdccf666fe8eed3d6bf702Bill Wendling/// 391f291059b26350bd96d7723d28a35734fe0a1484Douglas Gregorstruct LTOModule { 4062cf01e4c5de703e7712c72d12f577153b373cbaBill Wendlingprivate: 413bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling typedef llvm::StringMap<uint8_t> StringSet; 423c6b29b7e96f4fd0c9f5aae521d52cbf2b4493c2Bill Wendling 433c6b29b7e96f4fd0c9f5aae521d52cbf2b4493c2Bill Wendling struct NameAndAttributes { 4424b878031d4f62baf0e9a0573aeebc9dfd54b546Bill Wendling const char *name; 4524b878031d4f62baf0e9a0573aeebc9dfd54b546Bill Wendling uint32_t attributes; 4624b878031d4f62baf0e9a0573aeebc9dfd54b546Bill Wendling bool isFunction; 47613abf3fa6c69f709fbc122caaa74435222c4bf3Rafael Espindola const llvm::GlobalValue *symbol; 483bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling }; 493bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling 503bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling llvm::OwningPtr<llvm::Module> _module; 513bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling llvm::OwningPtr<llvm::TargetMachine> _target; 523bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling std::vector<NameAndAttributes> _symbols; 533bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling 543bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling // _defines and _undefines only needed to disambiguate tentative definitions 553c6b29b7e96f4fd0c9f5aae521d52cbf2b4493c2Bill Wendling StringSet _defines; 563bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling llvm::StringMap<NameAndAttributes> _undefines; 573bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling std::vector<const char*> _asm_undefines; 583bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling llvm::MCContext _context; 593bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling 603bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling // Use mangler to add GlobalPrefix to names to match linker names. 613bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling llvm::Mangler _mangler; 623bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling 633bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling LTOModule(llvm::Module *m, llvm::TargetMachine *t); 6462cf01e4c5de703e7712c72d12f577153b373cbaBill Wendlingpublic: 653bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// isBitcodeFile - Returns 'true' if the file or memory contents is LLVM 663bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// bitcode. 673bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling static bool isBitcodeFile(const void *mem, size_t length); 683bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling static bool isBitcodeFile(const char *path); 693bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling 703bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// isBitcodeFileForTarget - Returns 'true' if the file or memory contents 713bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// is LLVM bitcode for the specified triple. 723c6b29b7e96f4fd0c9f5aae521d52cbf2b4493c2Bill Wendling static bool isBitcodeFileForTarget(const void *mem, 733bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling size_t length, 743bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling const char *triplePrefix); 753c6b29b7e96f4fd0c9f5aae521d52cbf2b4493c2Bill Wendling static bool isBitcodeFileForTarget(const char *path, 763bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling const char *triplePrefix); 773bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling 783bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// makeLTOModule - Create an LTOModule. N.B. These methods take ownership 793bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// of the buffer. 803bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling static LTOModule *makeLTOModule(const char* path, 813bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling std::string &errMsg); 823bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling static LTOModule *makeLTOModule(int fd, const char *path, 833bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling size_t size, std::string &errMsg); 843bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling static LTOModule *makeLTOModule(int fd, const char *path, 853bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling size_t map_size, 863bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling off_t offset, 873bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling std::string& errMsg); 883bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling static LTOModule *makeLTOModule(const void *mem, size_t length, 893bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling std::string &errMsg); 903bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling 913bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// getTargetTriple - Return the Module's target triple. 923bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling const char *getTargetTriple() { 933bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling return _module->getTargetTriple().c_str(); 943bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling } 953bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling 963bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// setTargetTriple - Set the Module's target triple. 973bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling void setTargetTriple(const char *triple) { 983bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling _module->setTargetTriple(triple); 993bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling } 1003bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling 1013bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// getSymbolCount - Get the number of symbols 1023bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling uint32_t getSymbolCount() { 1033bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling return _symbols.size(); 1043bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling } 1053bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling 1063bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// getSymbolAttributes - Get the attributes for a symbol at the specified 1073bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// index. 1083bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling lto_symbol_attributes getSymbolAttributes(uint32_t index) { 1093bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling if (index < _symbols.size()) 1103c6b29b7e96f4fd0c9f5aae521d52cbf2b4493c2Bill Wendling return lto_symbol_attributes(_symbols[index].attributes); 1113c6b29b7e96f4fd0c9f5aae521d52cbf2b4493c2Bill Wendling return lto_symbol_attributes(0); 1123bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling } 1133bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling 1143bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// getSymbolName - Get the name of the symbol at the specified index. 1153bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling const char *getSymbolName(uint32_t index) { 1163bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling if (index < _symbols.size()) 1173bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling return _symbols[index].name; 1183c6b29b7e96f4fd0c9f5aae521d52cbf2b4493c2Bill Wendling return NULL; 1193bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling } 1203bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling 1213bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// getLLVVMModule - Return the Module. 1223bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling llvm::Module *getLLVVMModule() { return _module.get(); } 1233bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling 1243bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// getAsmUndefinedRefs - 1253bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling const std::vector<const char*> &getAsmUndefinedRefs() { 1263bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling return _asm_undefines; 1273bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling } 12877595fc35642f990bfc5ad05b8e68d4056695ecaNick Kledzik 1299ac0aaa6afe4162b4db32833f9f176ac05ce6f8fBill Wendling /// getTargetOptions - Fill the TargetOptions object with the options 1309ac0aaa6afe4162b4db32833f9f176ac05ce6f8fBill Wendling /// specified on the command line. 1319ac0aaa6afe4162b4db32833f9f176ac05ce6f8fBill Wendling static void getTargetOptions(llvm::TargetOptions &Options); 1329ac0aaa6afe4162b4db32833f9f176ac05ce6f8fBill Wendling 13377595fc35642f990bfc5ad05b8e68d4056695ecaNick Kledzikprivate: 1343bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// parseSymbols - Parse the symbols from the module and model-level ASM and 1353bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// add them to either the defined or undefined lists. 1363bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling bool parseSymbols(std::string &errMsg); 13762cf01e4c5de703e7712c72d12f577153b373cbaBill Wendling 1383bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// addPotentialUndefinedSymbol - Add a symbol which isn't defined just yet 1393bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// to a list to be resolved later. 140613abf3fa6c69f709fbc122caaa74435222c4bf3Rafael Espindola void addPotentialUndefinedSymbol(const llvm::GlobalValue *dcl, bool isFunc); 14162cf01e4c5de703e7712c72d12f577153b373cbaBill Wendling 1423bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// addDefinedSymbol - Add a defined symbol to the list. 143613abf3fa6c69f709fbc122caaa74435222c4bf3Rafael Espindola void addDefinedSymbol(const llvm::GlobalValue *def, bool isFunction); 14462cf01e4c5de703e7712c72d12f577153b373cbaBill Wendling 1453bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// addDefinedFunctionSymbol - Add a function symbol as defined to the list. 146613abf3fa6c69f709fbc122caaa74435222c4bf3Rafael Espindola void addDefinedFunctionSymbol(const llvm::Function *f); 14762cf01e4c5de703e7712c72d12f577153b373cbaBill Wendling 1483bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// addDefinedDataSymbol - Add a data symbol as defined to the list. 149613abf3fa6c69f709fbc122caaa74435222c4bf3Rafael Espindola void addDefinedDataSymbol(const llvm::GlobalValue *v); 15062cf01e4c5de703e7712c72d12f577153b373cbaBill Wendling 1513bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// addAsmGlobalSymbols - Add global symbols from module-level ASM to the 1523bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// defined or undefined lists. 1533bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling bool addAsmGlobalSymbols(std::string &errMsg); 15462cf01e4c5de703e7712c72d12f577153b373cbaBill Wendling 1553bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// addAsmGlobalSymbol - Add a global symbol from module-level ASM to the 1563bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// defined list. 1573bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling void addAsmGlobalSymbol(const char *, lto_symbol_attributes scope); 15862cf01e4c5de703e7712c72d12f577153b373cbaBill Wendling 1593bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// addAsmGlobalSymbolUndef - Add a global symbol from module-level ASM to 1603bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// the undefined list. 1613bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling void addAsmGlobalSymbolUndef(const char *); 1623eb445feb22647e867a339f4c59b0a716b03a21aNick Kledzik 1633bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// addObjCClass - Parse i386/ppc ObjC class data structure. 164613abf3fa6c69f709fbc122caaa74435222c4bf3Rafael Espindola void addObjCClass(const llvm::GlobalVariable *clgv); 16577595fc35642f990bfc5ad05b8e68d4056695ecaNick Kledzik 1663bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// addObjCCategory - Parse i386/ppc ObjC category data structure. 167613abf3fa6c69f709fbc122caaa74435222c4bf3Rafael Espindola void addObjCCategory(const llvm::GlobalVariable *clgv); 16830fe94ea43bd122fb4fd26b84bcf62f8096d4293Bill Wendling 1693bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// addObjCClassRef - Parse i386/ppc ObjC class list data structure. 170613abf3fa6c69f709fbc122caaa74435222c4bf3Rafael Espindola void addObjCClassRef(const llvm::GlobalVariable *clgv); 17130fe94ea43bd122fb4fd26b84bcf62f8096d4293Bill Wendling 1723bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// objcClassNameFromExpression - Get string that the data pointer points 1733bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// to. 174613abf3fa6c69f709fbc122caaa74435222c4bf3Rafael Espindola bool objcClassNameFromExpression(const llvm::Constant* c, std::string &name); 17562cf01e4c5de703e7712c72d12f577153b373cbaBill Wendling 1763bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// isTargetMatch - Returns 'true' if the memory buffer is for the specified 1773bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// target triple. 1783bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling static bool isTargetMatch(llvm::MemoryBuffer *memBuffer, 1793bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling const char *triplePrefix); 18062cf01e4c5de703e7712c72d12f577153b373cbaBill Wendling 1813bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// makeLTOModule - Create an LTOModule (private version). N.B. This 1823bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// method takes ownership of the buffer. 1833bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling static LTOModule *makeLTOModule(llvm::MemoryBuffer *buffer, 1843bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling std::string &errMsg); 18562cf01e4c5de703e7712c72d12f577153b373cbaBill Wendling 1863bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling /// makeBuffer - Create a MemoryBuffer from a memory range. 1873bb17380b1c87fa23ef9fe208ce6f56d352df611Bill Wendling static llvm::MemoryBuffer *makeBuffer(const void *mem, size_t length); 18877595fc35642f990bfc5ad05b8e68d4056695ecaNick Kledzik}; 18977595fc35642f990bfc5ad05b8e68d4056695ecaNick Kledzik 19077595fc35642f990bfc5ad05b8e68d4056695ecaNick Kledzik#endif // LTO_MODULE_H 191