//===-- Emitter.cpp - Write machine code to executable memory -------------===//
+//
+// The LLVM Compiler Infrastructure
+//
+// This file was developed by the LLVM research group and is distributed under
+// the University of Illinois Open Source License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
//
// This file defines a MachineCodeEmitter object that is used by Jello to write
// machine code to memory and remember where relocatable values lie.
//===----------------------------------------------------------------------===//
#define DEBUG_TYPE "jit"
-#ifndef _POSIX_MAPPED_FILES
-#define _POSIX_MAPPED_FILES
-#endif
-#include "VM.h"
+#include "JIT.h"
+#include "llvm/Constant.h"
+#include "llvm/Module.h"
#include "llvm/CodeGen/MachineCodeEmitter.h"
#include "llvm/CodeGen/MachineFunction.h"
#include "llvm/CodeGen/MachineConstantPool.h"
#include "llvm/Target/TargetData.h"
-#include "llvm/Module.h"
-#include "Support/Debug.h"
-#include "Support/Statistic.h"
-#include "Config/unistd.h"
-#include "Config/sys/mman.h"
-#include <stdio.h>
+#include "llvm/Support/Debug.h"
+#include "llvm/ADT/Statistic.h"
+#include "llvm/Support/SystemUtils.h"
+using namespace llvm;
namespace {
- Statistic<> NumBytes("jello", "Number of bytes of machine code compiled");
- VM *TheVM = 0;
+ Statistic<> NumBytes("jit", "Number of bytes of machine code compiled");
+ JIT *TheJIT = 0;
/// JITMemoryManager - Manage memory for the JIT code generation in a logical,
/// sane way. This splits a large block of MAP_NORESERVE'd memory into two
};
}
-// getMemory - Return a pointer to the specified number of bytes, which is
-// mapped as executable readable and writable.
-static void *getMemory(unsigned NumBytes) {
- if (NumBytes == 0) return 0;
- static const long pageSize = sysconf(_SC_PAGESIZE);
- unsigned NumPages = (NumBytes+pageSize-1)/pageSize;
-
-#if defined(i386) || defined(__i386__) || defined(__x86__)
- /* Linux and *BSD tend to have these flags named differently. */
-#if defined(MAP_ANON) && !defined(MAP_ANONYMOUS)
-# define MAP_ANONYMOUS MAP_ANON
-#endif /* defined(MAP_ANON) && !defined(MAP_ANONYMOUS) */
-#define fd 0
-#elif defined(sparc) || defined(__sparc__) || defined(__sparcv9)
-#define fd -1
-#else
- std::cerr << "This architecture is not supported by the JIT!\n";
- abort();
-#endif
-
- unsigned mmapFlags = MAP_PRIVATE|MAP_ANONYMOUS;
-#ifdef MAP_NORESERVE
- mmapFlags |= MAP_NORESERVE;
-#endif
-
- void *pa = mmap(0, pageSize*NumPages, PROT_READ|PROT_WRITE|PROT_EXEC,
- mmapFlags, fd, 0);
- if (pa == MAP_FAILED) {
- perror("mmap");
- abort();
- }
- return pa;
-}
-
JITMemoryManager::JITMemoryManager() {
// Allocate a 16M block of memory...
- MemBase = (unsigned char*)getMemory(16 << 20);
+ MemBase = (unsigned char*)AllocateRWXMemory(16 << 20);
FunctionBase = MemBase + 512*1024; // Use 512k for stubs
// Allocate stubs backwards from the function base, allocate functions forward
namespace {
- /// Emitter - The JIT implementation of the MachineCodeEmiter, which is used
+ /// Emitter - The JIT implementation of the MachineCodeEmitter, which is used
/// to output functions to memory for execution.
class Emitter : public MachineCodeEmitter {
JITMemoryManager MemMgr;
// constant pool.
std::vector<void*> ConstantPoolAddresses;
public:
- Emitter(VM &vm) { TheVM = &vm; }
+ Emitter(JIT &jit) { TheJIT = &jit; }
virtual void startFunction(MachineFunction &F);
virtual void finishFunction(MachineFunction &F);
virtual void* finishFunctionStub(const Function &F);
virtual void emitByte(unsigned char B);
virtual void emitWord(unsigned W);
+ virtual void emitWordAt(unsigned W, unsigned *Ptr);
virtual uint64_t getGlobalValueAddress(GlobalValue *V);
virtual uint64_t getGlobalValueAddress(const std::string &Name);
};
}
-MachineCodeEmitter *VM::createEmitter(VM &V) {
- return new Emitter(V);
+MachineCodeEmitter *JIT::createEmitter(JIT &jit) {
+ return new Emitter(jit);
}
void Emitter::startFunction(MachineFunction &F) {
CurByte = CurBlock = MemMgr.startFunctionBody();
- TheVM->addGlobalMapping(F.getFunction(), CurBlock);
+ TheJIT->addGlobalMapping(F.getFunction(), CurBlock);
}
void Emitter::finishFunction(MachineFunction &F) {
void Emitter::emitConstantPool(MachineConstantPool *MCP) {
const std::vector<Constant*> &Constants = MCP->getConstants();
+ if (Constants.empty()) return;
+
+ std::vector<unsigned> ConstantOffset;
+ ConstantOffset.reserve(Constants.size());
+
+ // Calculate how much space we will need for all the constants, and the offset
+ // each one will live in.
+ unsigned TotalSize = 0;
+ for (unsigned i = 0, e = Constants.size(); i != e; ++i) {
+ const Type *Ty = Constants[i]->getType();
+ unsigned Size = TheJIT->getTargetData().getTypeSize(Ty);
+ unsigned Alignment = TheJIT->getTargetData().getTypeAlignment(Ty);
+ // Make sure to take into account the alignment requirements of the type.
+ TotalSize = (TotalSize + Alignment-1) & ~(Alignment-1);
+
+ // Remember the offset this element lives at.
+ ConstantOffset.push_back(TotalSize);
+ TotalSize += Size; // Reserve space for the constant.
+ }
+
+ // Now that we know how much memory to allocate, do so.
+ char *Pool = new char[TotalSize];
+
+ // Actually output all of the constants, and remember their addresses.
for (unsigned i = 0, e = Constants.size(); i != e; ++i) {
- // For now we just allocate some memory on the heap, this can be
- // dramatically improved.
- const Type *Ty = ((Value*)Constants[i])->getType();
- void *Addr = malloc(TheVM->getTargetData().getTypeSize(Ty));
- TheVM->InitializeMemory(Constants[i], Addr);
+ void *Addr = Pool + ConstantOffset[i];
+ TheJIT->InitializeMemory(Constants[i], Addr);
ConstantPoolAddresses.push_back(Addr);
}
}
CurByte += sizeof(unsigned);
}
+void Emitter::emitWordAt(unsigned W, unsigned *Ptr) {
+ *Ptr = W;
+}
uint64_t Emitter::getGlobalValueAddress(GlobalValue *V) {
// Try looking up the function to see if it is already compiled, if not return
// 0.
- return (intptr_t)TheVM->getPointerToGlobalIfAvailable(V);
+ if (isa<Function>(V))
+ return (intptr_t)TheJIT->getPointerToGlobalIfAvailable(V);
+ else {
+ return (intptr_t)TheJIT->getOrEmitGlobalVariable(cast<GlobalVariable>(V));
+ }
}
uint64_t Emitter::getGlobalValueAddress(const std::string &Name) {
- return (intptr_t)TheVM->getPointerToNamedFunction(Name);
+ return (intptr_t)TheJIT->getPointerToNamedFunction(Name);
}
// getConstantPoolEntryAddress - Return the address of the 'ConstantNum' entry
}
uint64_t Emitter::forceCompilationOf(Function *F) {
- return (intptr_t)TheVM->getPointerToFunction(F);
+ return (intptr_t)TheJIT->getPointerToFunction(F);
}
// getPointerToNamedFunction - This function is used as a global wrapper to
-// VM::getPointerToNamedFunction for the purpose of resolving symbols when
+// JIT::getPointerToNamedFunction for the purpose of resolving symbols when
// bugpoint is debugging the JIT. In that scenario, we are loading an .so and
// need to resolve function(s) that are being mis-codegenerated, so we need to
// resolve their addresses at runtime, and this is the way to do it.
extern "C" {
void *getPointerToNamedFunction(const char *Name) {
- Module &M = TheVM->getModule();
+ Module &M = TheJIT->getModule();
if (Function *F = M.getNamedFunction(Name))
- return TheVM->getPointerToFunction(F);
- return TheVM->getPointerToNamedFunction(Name);
+ return TheJIT->getPointerToFunction(F);
+ return TheJIT->getPointerToNamedFunction(Name);
}
}