LTOModule.h revision ef194ed74033eba099f4f391ffdfc176f9aa6f26 
1//===-LTOModule.h - LLVM Link Time Optimizer ------------------------------===//
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 LTOModule class.
11//
12//===----------------------------------------------------------------------===//
13
14#ifndef LTO_MODULE_H
15#define LTO_MODULE_H
16
17#include "llvm/Module.h"
18#include "llvm/ADT/OwningPtr.h"
19#include "llvm/Target/TargetMachine.h"
20#include "llvm/ADT/StringMap.h"
21
22#include "llvm-c/lto.h"
23
24#include <vector>
25#include <string>
26
27
28// forward references to llvm classes
29namespace llvm {
30    class Mangler;
31    class MemoryBuffer;
32    class GlobalValue;
33    class Value;
34    class Function;
35}
36
37
38//
39// C++ class which implements the opaque lto_module_t
40//
41class LTOModule {
42public:
43
44    static bool              isBitcodeFile(const void* mem, size_t length);
45    static bool              isBitcodeFile(const char* path);
46
47    static bool              isBitcodeFileForTarget(const void* mem,
48                                    size_t length, const char* triplePrefix);
49
50    static bool              isBitcodeFileForTarget(const char* path,
51                                                    const char* triplePrefix);
52
53    static LTOModule*        makeLTOModule(const char* path, std::string& errMsg);
54    static LTOModule*        makeLTOModule(const void* mem, size_t length,
55                                                            std::string& errMsg);
56
57    const char*              getTargetTriple();
58    uint32_t                 getSymbolCount();
59    lto_symbol_attributes    getSymbolAttributes(uint32_t index);
60    const char*              getSymbolName(uint32_t index);
61
62    llvm::Module *           getLLVVMModule() { return _module.get(); }
63
64private:
65                            LTOModule(llvm::Module* m, llvm::TargetMachine* t);
66
67    void                    lazyParseSymbols();
68    void                    addDefinedSymbol(llvm::GlobalValue* def,
69                                                    llvm::Mangler& mangler,
70                                                    bool isFunction);
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    static bool             isTargetMatch(llvm::MemoryBuffer* memBuffer,
80                                                    const char* triplePrefix);
81
82    static LTOModule*       makeLTOModule(llvm::MemoryBuffer* buffer,
83                                                        std::string& errMsg);
84
85    typedef llvm::StringMap<uint8_t> StringSet;
86
87    struct NameAndAttributes {
88        const char*            name;
89        lto_symbol_attributes  attributes;
90    };
91
92    llvm::OwningPtr<llvm::Module>           _module;
93    llvm::OwningPtr<llvm::TargetMachine>    _target;
94    bool                                    _symbolsParsed;
95    std::vector<NameAndAttributes>          _symbols;
96    // _defines and _undefines only needed to disambiguate tentative definitions
97    StringSet                               _defines;
98    StringSet                               _undefines;
99};
100
101#endif // LTO_MODULE_H
102
103