113 lines
3.0 KiB
C++
113 lines
3.0 KiB
C++
|
// Copyright 2014 The Rust Project Developers. See the COPYRIGHT
|
||
|
|
// file at the top-level directory of this distribution and at
|
||
|
|
// http://rust-lang.org/COPYRIGHT.
|
||
|
|
//
|
||
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
||
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
||
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
||
|
|
// option. This file may not be copied, modified, or distributed
|
||
|
|
// except according to those terms.
|
||
|
|
|
||
|
|
#include "rustllvm.h"
|
||
|
|
|
||
|
|
#include "llvm/ExecutionEngine/SectionMemoryManager.h"
|
||
|
|
|
||
|
|
using namespace llvm;
|
||
|
|
using namespace llvm::sys;
|
||
|
|
using namespace llvm::object;
|
||
|
|
|
||
|
|
class RustJITMemoryManager : public SectionMemoryManager
|
||
|
|
{
|
||
|
|
typedef SectionMemoryManager Base;
|
||
|
|
|
||
|
|
const void *morestack;
|
||
|
|
|
||
|
|
public:
|
||
|
|
|
||
|
|
RustJITMemoryManager(const void *morestack_ptr)
|
||
|
|
: morestack(morestack_ptr)
|
||
|
|
{}
|
||
|
|
|
||
|
|
uint64_t getSymbolAddress(const std::string &Name) override
|
||
|
|
{
|
||
|
|
if (Name == "__morestack" || Name == "___morestack")
|
||
|
|
return reinterpret_cast<uint64_t>(morestack);
|
||
|
|
|
||
|
|
return Base::getSymbolAddress(Name);
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
DEFINE_SIMPLE_CONVERSION_FUNCTIONS(RustJITMemoryManager, LLVMRustJITMemoryManagerRef)
|
||
|
|
|
||
|
|
extern "C" LLVMRustJITMemoryManagerRef LLVMRustCreateJITMemoryManager(void *morestack)
|
||
|
|
{
|
||
|
|
return wrap(new RustJITMemoryManager(morestack));
|
||
|
|
}
|
||
|
|
|
||
|
|
extern "C" LLVMBool LLVMRustLoadDynamicLibrary(const char *path)
|
||
|
|
{
|
||
|
|
std::string err;
|
||
|
|
DynamicLibrary lib = DynamicLibrary::getPermanentLibrary(path, &err);
|
||
|
|
|
||
|
|
if (!lib.isValid())
|
||
|
|
LLVMRustSetLastError(err.c_str());
|
||
|
|
|
||
|
|
return lib.isValid();
|
||
|
|
}
|
||
|
|
|
||
|
|
// Calls LLVMAddModule;
|
||
|
|
// exists for consistency with LLVMExecutionEngineRemoveModule
|
||
|
|
extern "C" void LLVMExecutionEngineAddModule(
|
||
|
|
LLVMExecutionEngineRef eeref, LLVMModuleRef mref)
|
||
|
|
{
|
||
|
|
LLVMAddModule(eeref, mref);
|
||
|
|
}
|
||
|
|
|
||
|
|
// LLVMRemoveModule exists in LLVM's C bindings,
|
||
|
|
// but it requires pointless parameters
|
||
|
|
extern "C" LLVMBool LLVMExecutionEngineRemoveModule(
|
||
|
|
LLVMExecutionEngineRef eeref, LLVMModuleRef mref)
|
||
|
|
{
|
||
|
|
ExecutionEngine *ee = unwrap(eeref);
|
||
|
|
Module *m = unwrap(mref);
|
||
|
|
|
||
|
|
return ee->removeModule(m);
|
||
|
|
}
|
||
|
|
|
||
|
|
extern "C" LLVMExecutionEngineRef LLVMBuildExecutionEngine(
|
||
|
|
LLVMModuleRef mod, LLVMRustJITMemoryManagerRef mref)
|
||
|
|
{
|
||
|
|
// These are necessary for code generation to work properly.
|
||
|
|
InitializeNativeTarget();
|
||
|
|
InitializeNativeTargetAsmPrinter();
|
||
|
|
InitializeNativeTargetAsmParser();
|
||
|
|
|
||
|
|
std::unique_ptr<Module> m(unwrap(mod));
|
||
|
|
RustJITMemoryManager *mm = unwrap(mref);
|
||
|
|
|
||
|
|
std::string error_str;
|
||
|
|
TargetOptions options;
|
||
|
|
|
||
|
|
options.JITEmitDebugInfo = true;
|
||
|
|
options.NoFramePointerElim = true;
|
||
|
|
|
||
|
|
ExecutionEngine *ee = EngineBuilder(std::move(m))
|
||
|
|
.setEngineKind(EngineKind::JIT)
|
||
|
|
.setErrorStr(&error_str)
|
||
|
|
.setMCJITMemoryManager(mm)
|
||
|
|
.setTargetOptions(options)
|
||
|
|
.create();
|
||
|
|
|
||
|
|
if (!ee)
|
||
|
|
LLVMRustSetLastError(error_str.c_str());
|
||
|
|
|
||
|
|
return wrap(ee);
|
||
|
|
}
|
||
|
|
|
||
|
|
extern "C" void LLVMExecutionEngineFinalizeObject(LLVMExecutionEngineRef eeref)
|
||
|
|
{
|
||
|
|
ExecutionEngine *ee = unwrap(eeref);
|
||
|
|
|
||
|
|
ee->finalizeObject();
|
||
|
|
}
|