1706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach//===- RemoteTarget.cpp - LLVM Remote process JIT execution --------------===// 2706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach// 3706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach// The LLVM Compiler Infrastructure 4706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach// 5706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach// This file is distributed under the University of Illinois Open Source 6706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach// License. See LICENSE.TXT for details. 7706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach// 8706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach//===----------------------------------------------------------------------===// 9706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach// 10706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach// Implementation of the RemoteTarget class which executes JITed code in a 11706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach// separate address range from where it was built. 12706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach// 13706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach//===----------------------------------------------------------------------===// 14706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach 15706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach#include "RemoteTarget.h" 16dbf545719a22bf03403c1d0137ae0f5726f36de3Craig Topper#include "llvm/ADT/StringRef.h" 17dbf545719a22bf03403c1d0137ae0f5726f36de3Craig Topper#include "llvm/Support/DataTypes.h" 18dbf545719a22bf03403c1d0137ae0f5726f36de3Craig Topper#include "llvm/Support/Memory.h" 19706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach#include <stdlib.h> 20706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach#include <string> 210ab5c6c16b1b09d76c3ba2d70443b10bcc26169cAndrew Kaylor 22706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbachusing namespace llvm; 23706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach 240ab5c6c16b1b09d76c3ba2d70443b10bcc26169cAndrew Kaylor//////////////////////////////////////////////////////////////////////////////// 250ab5c6c16b1b09d76c3ba2d70443b10bcc26169cAndrew Kaylor// Simulated remote execution 260ab5c6c16b1b09d76c3ba2d70443b10bcc26169cAndrew Kaylor// 270ab5c6c16b1b09d76c3ba2d70443b10bcc26169cAndrew Kaylor// This implementation will simply move generated code and data to a new memory 280ab5c6c16b1b09d76c3ba2d70443b10bcc26169cAndrew Kaylor// location in the current executable and let it run from there. 290ab5c6c16b1b09d76c3ba2d70443b10bcc26169cAndrew Kaylor//////////////////////////////////////////////////////////////////////////////// 300ab5c6c16b1b09d76c3ba2d70443b10bcc26169cAndrew Kaylor 31706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbachbool RemoteTarget::allocateSpace(size_t Size, unsigned Alignment, 32706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach uint64_t &Address) { 33dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines sys::MemoryBlock *Prev = Allocations.size() ? &Allocations.back() : nullptr; 34706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach sys::MemoryBlock Mem = sys::Memory::AllocateRWX(Size, Prev, &ErrorMsg); 35dce4a407a24b04eebc6a376f8e62b41aaa7b071fStephen Hines if (Mem.base() == nullptr) 3636b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines return false; 37706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach if ((uintptr_t)Mem.base() % Alignment) { 38706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach ErrorMsg = "unable to allocate sufficiently aligned memory"; 3936b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines return false; 40706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach } 41706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach Address = reinterpret_cast<uint64_t>(Mem.base()); 4236b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines Allocations.push_back(Mem); 4336b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines return true; 44706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach} 45706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach 46706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbachbool RemoteTarget::loadData(uint64_t Address, const void *Data, size_t Size) { 47706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach memcpy ((void*)Address, Data, Size); 4836b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines return true; 49706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach} 50706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach 51706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbachbool RemoteTarget::loadCode(uint64_t Address, const void *Data, size_t Size) { 52706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach memcpy ((void*)Address, Data, Size); 532932284f045005bc3ed850f9c42524fadf3ae3d8Andrew Kaylor sys::MemoryBlock Mem((void*)Address, Size); 542932284f045005bc3ed850f9c42524fadf3ae3d8Andrew Kaylor sys::Memory::setExecutable(Mem, &ErrorMsg); 5536b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines return true; 56706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach} 57706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach 58706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbachbool RemoteTarget::executeCode(uint64_t Address, int &RetVal) { 59706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach int (*fn)(void) = (int(*)(void))Address; 60706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach RetVal = fn(); 6136b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines return true; 62706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach} 63706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach 6436b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hinesbool RemoteTarget::create() { 6536b56886974eae4f9c5ebc96befd3e7bfe5de338Stephen Hines return true; 66706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach} 67706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach 68706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbachvoid RemoteTarget::stop() { 69706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach for (unsigned i = 0, e = Allocations.size(); i != e; ++i) 70706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach sys::Memory::ReleaseRWX(Allocations[i]); 71706f03a35db7029b2dbd2925552eb0d0472dcbb4Jim Grosbach} 72