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