forked from OSchip/llvm-project
83 lines
3.0 KiB
C++
83 lines
3.0 KiB
C++
//===- RecordingMemoryManager.h - LLI MCJIT recording memory manager ------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This memory manager allocates local storage and keeps a record of each
|
|
// allocation. Iterators are provided for all data and code allocations.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef RECORDINGMEMORYMANAGER_H
|
|
#define RECORDINGMEMORYMANAGER_H
|
|
|
|
#include "llvm/ADT/SmallVector.h"
|
|
#include "llvm/ExecutionEngine/JITMemoryManager.h"
|
|
#include "llvm/Support/ErrorHandling.h"
|
|
#include "llvm/Support/Memory.h"
|
|
#include <utility>
|
|
|
|
namespace llvm {
|
|
|
|
class RecordingMemoryManager : public JITMemoryManager {
|
|
public:
|
|
typedef std::pair<sys::MemoryBlock, unsigned> Allocation;
|
|
|
|
private:
|
|
SmallVector<Allocation, 16> AllocatedDataMem;
|
|
SmallVector<Allocation, 16> AllocatedCodeMem;
|
|
|
|
// FIXME: This is part of a work around to keep sections near one another
|
|
// when MCJIT performs relocations after code emission but before
|
|
// the generated code is moved to the remote target.
|
|
sys::MemoryBlock Near;
|
|
sys::MemoryBlock allocateSection(uintptr_t Size);
|
|
|
|
public:
|
|
RecordingMemoryManager() {}
|
|
virtual ~RecordingMemoryManager();
|
|
|
|
typedef SmallVectorImpl<Allocation>::const_iterator const_data_iterator;
|
|
typedef SmallVectorImpl<Allocation>::const_iterator const_code_iterator;
|
|
|
|
const_data_iterator data_begin() const { return AllocatedDataMem.begin(); }
|
|
const_data_iterator data_end() const { return AllocatedDataMem.end(); }
|
|
const_code_iterator code_begin() const { return AllocatedCodeMem.begin(); }
|
|
const_code_iterator code_end() const { return AllocatedCodeMem.end(); }
|
|
|
|
uint8_t *allocateCodeSection(uintptr_t Size, unsigned Alignment,
|
|
unsigned SectionID);
|
|
|
|
uint8_t *allocateDataSection(uintptr_t Size, unsigned Alignment,
|
|
unsigned SectionID, bool IsReadOnly);
|
|
|
|
void *getPointerToNamedFunction(const std::string &Name,
|
|
bool AbortOnFailure = true);
|
|
|
|
bool finalizeMemory(std::string *ErrMsg) { return false; }
|
|
|
|
// The following obsolete JITMemoryManager calls are stubbed out for
|
|
// this model.
|
|
void setMemoryWritable();
|
|
void setMemoryExecutable();
|
|
void setPoisonMemory(bool poison);
|
|
void AllocateGOT();
|
|
uint8_t *getGOTBase() const;
|
|
uint8_t *startFunctionBody(const Function *F, uintptr_t &ActualSize);
|
|
uint8_t *allocateStub(const GlobalValue* F, unsigned StubSize,
|
|
unsigned Alignment);
|
|
void endFunctionBody(const Function *F, uint8_t *FunctionStart,
|
|
uint8_t *FunctionEnd);
|
|
uint8_t *allocateSpace(intptr_t Size, unsigned Alignment);
|
|
uint8_t *allocateGlobal(uintptr_t Size, unsigned Alignment);
|
|
void deallocateFunctionBody(void *Body);
|
|
};
|
|
|
|
} // end namespace llvm
|
|
|
|
#endif
|