contrib/llvm/tools/lli/RecordingMemoryManager.h

   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:
  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
  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(); }
  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
  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