include/llvm/ExecutionEngine/RuntimeDyld.h

   1 //===-- RuntimeDyld.h - Run-time dynamic linker for MC-JIT ------*- C++ -*-===//
   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 // Interface for the runtime dynamic linker facilities of the MC-JIT.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #ifndef LLVM_RUNTIME_DYLD_H
  15 #define LLVM_RUNTIME_DYLD_H
  16
  17 #include "llvm/ADT/StringRef.h"
  18 #include "llvm/Support/Memory.h"
  19
  20 namespace llvm {
  21
  22 class RuntimeDyldImpl;
  23 class MemoryBuffer;
  24
  25 // RuntimeDyld clients often want to handle the memory management of
  26 // what gets placed where. For JIT clients, this is an abstraction layer
  27 // over the JITMemoryManager, which references objects by their source
  28 // representations in LLVM IR.
  29 // FIXME: As the RuntimeDyld fills out, additional routines will be needed
  30 //        for the varying types of objects to be allocated.
  31 class RTDyldMemoryManager {
  32   RTDyldMemoryManager(const RTDyldMemoryManager&);  // DO NOT IMPLEMENT
  33   void operator=(const RTDyldMemoryManager&);       // DO NOT IMPLEMENT
  34 public:
  35   RTDyldMemoryManager() {}
  36   virtual ~RTDyldMemoryManager();
  37
  38   /// allocateCodeSection - Allocate a memory block of (at least) the given
  39   /// size suitable for executable code.
  40   virtual uint8_t *allocateCodeSection(uintptr_t Size, unsigned Alignment,
  41                                        unsigned SectionID) = 0;
  42
  43   /// allocateDataSection - Allocate a memory block of (at least) the given
  44   /// size suitable for data.
  45   virtual uint8_t *allocateDataSection(uintptr_t Size, unsigned Alignment,
  46                                        unsigned SectionID) = 0;
  47
  48   // Allocate ActualSize bytes, or more, for the named function. Return
  49   // a pointer to the allocated memory and update Size to reflect how much
  50   // memory was acutally allocated.
  51   virtual uint8_t *startFunctionBody(const char *Name, uintptr_t &Size) = 0;
  52
  53   // Mark the end of the function, including how much of the allocated
  54   // memory was actually used.
  55   virtual void endFunctionBody(const char *Name, uint8_t *FunctionStart,
  56                                uint8_t *FunctionEnd) = 0;
  57 };
  58
  59 class RuntimeDyld {
  60   RuntimeDyld(const RuntimeDyld &);     // DO NOT IMPLEMENT
  61   void operator=(const RuntimeDyld &);  // DO NOT IMPLEMENT
  62
  63   // RuntimeDyldImpl is the actual class. RuntimeDyld is just the public
  64   // interface.
  65   RuntimeDyldImpl *Dyld;
  66   RTDyldMemoryManager *MM;
  67 protected:
  68   // Change the address associated with a section when resolving relocations.
  69   // Any relocations already associated with the symbol will be re-resolved.
  70   void reassignSectionAddress(unsigned SectionID, uint64_t Addr);
  71 public:
  72   RuntimeDyld(RTDyldMemoryManager*);
  73   ~RuntimeDyld();
  74
  75   bool loadObject(MemoryBuffer *InputBuffer);
  76   // Get the address of our local copy of the symbol. This may or may not
  77   // be the address used for relocation (clients can copy the data around
  78   // and resolve relocatons based on where they put it).
  79   void *getSymbolAddress(StringRef Name);
  80   // Resolve the relocations for all symbols we currently know about.
  81   void resolveRelocations();
  82
  83   /// mapSectionAddress - map a section to its target address space value.
  84   /// Map the address of a JIT section as returned from the memory manager
  85   /// to the address in the target process as the running code will see it.
  86   /// This is the address which will be used for relocation resolution.
  87   void mapSectionAddress(void *LocalAddress, uint64_t TargetAddress);
  88
  89   StringRef getErrorString();
  90 };
  91
  92 } // end namespace llvm
  93
  94 #endif