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 
  36 class 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
  83 uint8_t *allocateCodeSection(uintptr_t Size, unsigned Alignment, unsigned SectionID);
  84 uint8_t *allocateDataSection(uintptr_t Size, unsigned Alignment, unsigned SectionID);
  85 unsigned char* startExceptionTable(const llvm::Function* F,
  86                                    uintptr_t& ActualSize);
  87 void deallocateExceptionTable(void *ptr);
  88 void endExceptionTable(const llvm::Function* F,
  89                        unsigned char* TableStart,
  90                        unsigned char* TableEnd,
  91                        unsigned char* FrameRegister);
  92 #else
  93 uint8_t *allocateCodeSection(uintptr_t Size, unsigned Alignment, unsigned SectionID, llvm::StringRef SectionName);
  94 uint8_t *allocateDataSection(uintptr_t Size, unsigned Alignment, unsigned SectionID, llvm::StringRef SectionName, bool IsReadOnly);
  95 bool finalizeMemory(std::string *ErrMsg = 0);
  96 #endif
  97 
  98 };
  99 
 100 #endif // SHARE_VM_SHARK_SHARKMEMORYMANAGER_HPP