mirror of
https://github.com/rust-lang/rust
synced 2024-09-15 22:50:55 +00:00
Add LLVM ExecutionEngine API
This commit is contained in:
parent
c38e73fef5
commit
2c028452b5
|
@ -22,7 +22,8 @@ LLVM_EXTRA_INCDIRS_$(1)= -iquote $(S)src/llvm/include \
|
||||||
-iquote $$(CFG_LLVM_BUILD_DIR_$(1))/include
|
-iquote $$(CFG_LLVM_BUILD_DIR_$(1))/include
|
||||||
endif
|
endif
|
||||||
|
|
||||||
RUSTLLVM_OBJS_CS_$(1) := $$(addprefix rustllvm/, RustWrapper.cpp PassWrapper.cpp)
|
RUSTLLVM_OBJS_CS_$(1) := $$(addprefix rustllvm/, \
|
||||||
|
ExecutionEngineWrapper.cpp RustWrapper.cpp PassWrapper.cpp)
|
||||||
|
|
||||||
RUSTLLVM_DEF_$(1) := $(1)/rustllvm/rustllvm$(CFG_DEF_SUFFIX_$(1))
|
RUSTLLVM_DEF_$(1) := $(1)/rustllvm/rustllvm$(CFG_DEF_SUFFIX_$(1))
|
||||||
|
|
||||||
|
|
|
@ -466,6 +466,9 @@ pub enum Builder_opaque {}
|
||||||
pub enum ExecutionEngine_opaque {}
|
pub enum ExecutionEngine_opaque {}
|
||||||
pub type ExecutionEngineRef = *mut ExecutionEngine_opaque;
|
pub type ExecutionEngineRef = *mut ExecutionEngine_opaque;
|
||||||
#[allow(missing_copy_implementations)]
|
#[allow(missing_copy_implementations)]
|
||||||
|
pub enum RustJITMemoryManager_opaque {}
|
||||||
|
pub type RustJITMemoryManagerRef = *mut RustJITMemoryManager_opaque;
|
||||||
|
#[allow(missing_copy_implementations)]
|
||||||
pub enum MemoryBuffer_opaque {}
|
pub enum MemoryBuffer_opaque {}
|
||||||
pub type MemoryBufferRef = *mut MemoryBuffer_opaque;
|
pub type MemoryBufferRef = *mut MemoryBuffer_opaque;
|
||||||
#[allow(missing_copy_implementations)]
|
#[allow(missing_copy_implementations)]
|
||||||
|
@ -1064,7 +1067,18 @@ pub fn LLVMInsertIntoBuilderWithName(Builder: BuilderRef,
|
||||||
Instr: ValueRef,
|
Instr: ValueRef,
|
||||||
Name: *const c_char);
|
Name: *const c_char);
|
||||||
pub fn LLVMDisposeBuilder(Builder: BuilderRef);
|
pub fn LLVMDisposeBuilder(Builder: BuilderRef);
|
||||||
|
|
||||||
|
/* Execution engine */
|
||||||
|
pub fn LLVMRustCreateJITMemoryManager(morestack: *const ())
|
||||||
|
-> RustJITMemoryManagerRef;
|
||||||
|
pub fn LLVMBuildExecutionEngine(Mod: ModuleRef,
|
||||||
|
MM: RustJITMemoryManagerRef) -> ExecutionEngineRef;
|
||||||
pub fn LLVMDisposeExecutionEngine(EE: ExecutionEngineRef);
|
pub fn LLVMDisposeExecutionEngine(EE: ExecutionEngineRef);
|
||||||
|
pub fn LLVMExecutionEngineFinalizeObject(EE: ExecutionEngineRef);
|
||||||
|
pub fn LLVMRustLoadDynamicLibrary(path: *const c_char) -> Bool;
|
||||||
|
pub fn LLVMExecutionEngineAddModule(EE: ExecutionEngineRef, M: ModuleRef);
|
||||||
|
pub fn LLVMExecutionEngineRemoveModule(EE: ExecutionEngineRef, M: ModuleRef)
|
||||||
|
-> Bool;
|
||||||
|
|
||||||
/* Metadata */
|
/* Metadata */
|
||||||
pub fn LLVMSetCurrentDebugLocation(Builder: BuilderRef, L: ValueRef);
|
pub fn LLVMSetCurrentDebugLocation(Builder: BuilderRef, L: ValueRef);
|
||||||
|
|
112
src/rustllvm/ExecutionEngineWrapper.cpp
Normal file
112
src/rustllvm/ExecutionEngineWrapper.cpp
Normal file
|
@ -0,0 +1,112 @@
|
||||||
|
// 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();
|
||||||
|
}
|
|
@ -72,6 +72,7 @@ typedef struct OpaqueRustString *RustStringRef;
|
||||||
typedef struct LLVMOpaqueTwine *LLVMTwineRef;
|
typedef struct LLVMOpaqueTwine *LLVMTwineRef;
|
||||||
typedef struct LLVMOpaqueDebugLoc *LLVMDebugLocRef;
|
typedef struct LLVMOpaqueDebugLoc *LLVMDebugLocRef;
|
||||||
typedef struct LLVMOpaqueSMDiagnostic *LLVMSMDiagnosticRef;
|
typedef struct LLVMOpaqueSMDiagnostic *LLVMSMDiagnosticRef;
|
||||||
|
typedef struct LLVMOpaqueRustJITMemoryManager *LLVMRustJITMemoryManagerRef;
|
||||||
|
|
||||||
extern "C" void
|
extern "C" void
|
||||||
rust_llvm_string_write_impl(RustStringRef str, const char *ptr, size_t size);
|
rust_llvm_string_write_impl(RustStringRef str, const char *ptr, size_t size);
|
||||||
|
|
Loading…
Reference in a new issue