1/* 2 * Copyright (c) 1999, 2010, Oracle and/or its affiliates. All rights reserved. 3 * Copyright 2009 Red Hat, Inc. 4 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 5 * 6 * This code is free software; you can redistribute it and/or modify it 7 * under the terms of the GNU General Public License version 2 only, as 8 * published by the Free Software Foundation. 9 * 10 * This code is distributed in the hope that it will be useful, but WITHOUT 11 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 12 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 13 * version 2 for more details (a copy is included in the LICENSE file that 14 * accompanied this code). 15 * 16 * You should have received a copy of the GNU General Public License version 17 * 2 along with this work; if not, write to the Free Software Foundation, 18 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 19 * 20 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 21 * or visit www.oracle.com if you need additional information or have any 22 * questions. 23 * 24 */ 25 26#ifndef SHARE_VM_SHARK_SHARKMEMORYMANAGER_HPP 27#define SHARE_VM_SHARK_SHARKMEMORYMANAGER_HPP 28 29#include "shark/llvmHeaders.hpp" 30#include "shark/sharkEntry.hpp" 31 32// SharkMemoryManager wraps the LLVM JIT Memory Manager. We could use 33// this to run our own memory allocation policies, but for now all we 34// use it for is figuring out where the resulting native code ended up. 35 36class SharkMemoryManager : public llvm::JITMemoryManager { 37 public: 38 SharkMemoryManager() 39 : _mm(llvm::JITMemoryManager::CreateDefaultMemManager()) {} 40 41 private: 42 llvm::JITMemoryManager* _mm; 43 44 private: 45 llvm::JITMemoryManager* mm() const { 46 return _mm; 47 } 48 49 private: 50 std::map<const llvm::Function*, SharkEntry*> _entry_map; 51 52 public: 53 void set_entry_for_function(const llvm::Function* function, 54 SharkEntry* entry) { 55 _entry_map[function] = entry; 56 } 57 SharkEntry* get_entry_for_function(const llvm::Function* function) { 58 return _entry_map[function]; 59 } 60 61 public: 62 void AllocateGOT(); 63 unsigned char* getGOTBase() const; 64 unsigned char* allocateStub(const llvm::GlobalValue* F, 65 unsigned StubSize, 66 unsigned Alignment); 67 unsigned char* startFunctionBody(const llvm::Function* F, 68 uintptr_t& ActualSize); 69 void endFunctionBody(const llvm::Function* F, 70 unsigned char* FunctionStart, 71 unsigned char* FunctionEnd); 72 73 void *getPointerToNamedFunction(const std::string &Name, bool AbortOnFailure = true); 74 void setPoisonMemory(bool); 75 uint8_t* allocateGlobal(uintptr_t, unsigned int); 76 void setMemoryWritable(); 77 void setMemoryExecutable(); 78 void deallocateFunctionBody(void *ptr); 79 unsigned char *allocateSpace(intptr_t Size, 80 unsigned int Alignment); 81 82#if SHARK_LLVM_VERSION <= 32 83uint8_t *allocateCodeSection(uintptr_t Size, unsigned Alignment, unsigned SectionID); 84uint8_t *allocateDataSection(uintptr_t Size, unsigned Alignment, unsigned SectionID); 85unsigned char* startExceptionTable(const llvm::Function* F, 86 uintptr_t& ActualSize); 87void deallocateExceptionTable(void *ptr); 88void endExceptionTable(const llvm::Function* F, 89 unsigned char* TableStart, 90 unsigned char* TableEnd, 91 unsigned char* FrameRegister); 92#else 93uint8_t *allocateCodeSection(uintptr_t Size, unsigned Alignment, unsigned SectionID, llvm::StringRef SectionName); 94uint8_t *allocateDataSection(uintptr_t Size, unsigned Alignment, unsigned SectionID, llvm::StringRef SectionName, bool IsReadOnly); 95bool finalizeMemory(std::string *ErrMsg = 0); 96#endif 97 98}; 99 100#endif // SHARE_VM_SHARK_SHARKMEMORYMANAGER_HPP 101