• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 //===- RecordingMemoryManager.h - LLI MCJIT recording memory manager ------===//
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 memory manager allocates local storage and keeps a record of each
11 // allocation. Iterators are provided for all data and code allocations.
12 //
13 //===----------------------------------------------------------------------===//
14 
15 #ifndef RECORDINGMEMORYMANAGER_H
16 #define RECORDINGMEMORYMANAGER_H
17 
18 #include "llvm/ADT/SmallVector.h"
19 #include "llvm/ExecutionEngine/JITMemoryManager.h"
20 #include "llvm/Support/ErrorHandling.h"
21 #include "llvm/Support/Memory.h"
22 #include <utility>
23 
24 namespace llvm {
25 
26 class RecordingMemoryManager : public JITMemoryManager {
27 public:
28   typedef std::pair<sys::MemoryBlock, unsigned> Allocation;
29 
30 private:
31   SmallVector<Allocation, 16> AllocatedDataMem;
32   SmallVector<Allocation, 16> AllocatedCodeMem;
33 
34   // FIXME: This is part of a work around to keep sections near one another
35   // when MCJIT performs relocations after code emission but before
36   // the generated code is moved to the remote target.
37   sys::MemoryBlock Near;
38   sys::MemoryBlock allocateSection(uintptr_t Size);
39 
40 public:
RecordingMemoryManager()41   RecordingMemoryManager() {}
42   virtual ~RecordingMemoryManager();
43 
44   typedef SmallVectorImpl<Allocation>::const_iterator const_data_iterator;
45   typedef SmallVectorImpl<Allocation>::const_iterator const_code_iterator;
46 
data_begin()47   const_data_iterator data_begin() const { return AllocatedDataMem.begin(); }
data_end()48   const_data_iterator   data_end() const { return AllocatedDataMem.end(); }
code_begin()49   const_code_iterator code_begin() const { return AllocatedCodeMem.begin(); }
code_end()50   const_code_iterator   code_end() const { return AllocatedCodeMem.end(); }
51 
52   uint8_t *allocateCodeSection(uintptr_t Size, unsigned Alignment,
53                                        unsigned SectionID);
54 
55   uint8_t *allocateDataSection(uintptr_t Size, unsigned Alignment,
56                                        unsigned SectionID, bool IsReadOnly);
57 
58   void *getPointerToNamedFunction(const std::string &Name,
59                                   bool AbortOnFailure = true);
60 
applyPermissions(std::string * ErrMsg)61   bool applyPermissions(std::string *ErrMsg) { return false; }
62 
63   // The following obsolete JITMemoryManager calls are stubbed out for
64   // this model.
65   void setMemoryWritable();
66   void setMemoryExecutable();
67   void setPoisonMemory(bool poison);
68   void AllocateGOT();
69   uint8_t *getGOTBase() const;
70   uint8_t *startFunctionBody(const Function *F, uintptr_t &ActualSize);
71   uint8_t *allocateStub(const GlobalValue* F, unsigned StubSize,
72                         unsigned Alignment);
73   void endFunctionBody(const Function *F, uint8_t *FunctionStart,
74                        uint8_t *FunctionEnd);
75   uint8_t *allocateSpace(intptr_t Size, unsigned Alignment);
76   uint8_t *allocateGlobal(uintptr_t Size, unsigned Alignment);
77   void deallocateFunctionBody(void *Body);
78   uint8_t* startExceptionTable(const Function* F, uintptr_t &ActualSize);
79   void endExceptionTable(const Function *F, uint8_t *TableStart,
80                          uint8_t *TableEnd, uint8_t* FrameRegister);
81   void deallocateExceptionTable(void *ET);
82 
83 };
84 
85 } // end namespace llvm
86 
87 #endif
88