1 //===- RecordingMemoryManager.h - LLI MCJIT recording memory manager ------===//
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 memory manager allocates local storage and keeps a record of each
11 // allocation. Iterators are provided for all data and code allocations.
13 //===----------------------------------------------------------------------===//
15 #ifndef RECORDINGMEMORYMANAGER_H
16 #define RECORDINGMEMORYMANAGER_H
18 #include "llvm/ADT/SmallVector.h"
19 #include "llvm/ExecutionEngine/JITMemoryManager.h"
20 #include "llvm/Support/ErrorHandling.h"
21 #include "llvm/Support/Memory.h"
26 class RecordingMemoryManager : public JITMemoryManager {
28 typedef std::pair<sys::MemoryBlock, unsigned> Allocation;
31 SmallVector<Allocation, 16> AllocatedDataMem;
32 SmallVector<Allocation, 16> AllocatedCodeMem;
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);
41 RecordingMemoryManager() {}
42 virtual ~RecordingMemoryManager();
44 typedef SmallVectorImpl<Allocation>::const_iterator const_data_iterator;
45 typedef SmallVectorImpl<Allocation>::const_iterator const_code_iterator;
47 const_data_iterator data_begin() const { return AllocatedDataMem.begin(); }
48 const_data_iterator data_end() const { return AllocatedDataMem.end(); }
49 const_code_iterator code_begin() const { return AllocatedCodeMem.begin(); }
50 const_code_iterator code_end() const { return AllocatedCodeMem.end(); }
52 uint8_t *allocateCodeSection(uintptr_t Size, unsigned Alignment,
53 unsigned SectionID, StringRef SectionName);
55 uint8_t *allocateDataSection(uintptr_t Size, unsigned Alignment,
56 unsigned SectionID, StringRef SectionName,
59 void *getPointerToNamedFunction(const std::string &Name,
60 bool AbortOnFailure = true);
62 bool finalizeMemory(std::string *ErrMsg) { return false; }
64 // The following obsolete JITMemoryManager calls are stubbed out for
66 void setMemoryWritable();
67 void setMemoryExecutable();
68 void setPoisonMemory(bool poison);
70 uint8_t *getGOTBase() const;
71 uint8_t *startFunctionBody(const Function *F, uintptr_t &ActualSize);
72 uint8_t *allocateStub(const GlobalValue* F, unsigned StubSize,
74 void endFunctionBody(const Function *F, uint8_t *FunctionStart,
75 uint8_t *FunctionEnd);
76 uint8_t *allocateSpace(intptr_t Size, unsigned Alignment);
77 uint8_t *allocateGlobal(uintptr_t Size, unsigned Alignment);
78 void deallocateFunctionBody(void *Body);
81 } // end namespace llvm