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