diff options
Diffstat (limited to 'src/rustllvm/RustWrapper.cpp')
| -rw-r--r-- | src/rustllvm/RustWrapper.cpp | 278 |
1 files changed, 1 insertions, 277 deletions
diff --git a/src/rustllvm/RustWrapper.cpp b/src/rustllvm/RustWrapper.cpp index 0a148d334ab..484cded3147 100644 --- a/src/rustllvm/RustWrapper.cpp +++ b/src/rustllvm/RustWrapper.cpp @@ -34,282 +34,6 @@ extern "C" const char *LLVMRustGetLastError(void) { return LLVMRustError; } -// Custom memory manager for MCJITting. It needs special features -// that the generic JIT memory manager doesn't entail. Based on -// code from LLI, change where needed for Rust. -class RustMCJITMemoryManager : public JITMemoryManager { -public: - SmallVector<sys::MemoryBlock, 16> AllocatedDataMem; - SmallVector<sys::MemoryBlock, 16> AllocatedCodeMem; - SmallVector<sys::MemoryBlock, 16> FreeCodeMem; - void* __morestack; - DenseSet<DynamicLibrary*> crates; - - RustMCJITMemoryManager(void* sym) : __morestack(sym) { } - ~RustMCJITMemoryManager(); - - bool loadCrate(const char*, std::string*); - - virtual uint8_t *allocateCodeSection(uintptr_t Size, unsigned Alignment, - unsigned SectionID); - - virtual uint8_t *allocateDataSection(uintptr_t Size, unsigned Alignment, - unsigned SectionID, bool isReadOnly); - bool finalizeMemory(std::string *ErrMsg) { return false; } - - virtual bool applyPermissions(std::string *Str); - - virtual void *getPointerToNamedFunction(const std::string &Name, - bool AbortOnFailure = true); - - // Invalidate instruction cache for code sections. Some platforms with - // separate data cache and instruction cache require explicit cache flush, - // otherwise JIT code manipulations (like resolved relocations) will get to - // the data cache but not to the instruction cache. - virtual void invalidateInstructionCache(); - - // The MCJITMemoryManager doesn't use the following functions, so we don't - // need implement them. - virtual void setMemoryWritable() { - llvm_unreachable("Unimplemented call"); - } - virtual void setMemoryExecutable() { - llvm_unreachable("Unimplemented call"); - } - virtual void setPoisonMemory(bool poison) { - llvm_unreachable("Unimplemented call"); - } - virtual void AllocateGOT() { - llvm_unreachable("Unimplemented call"); - } - virtual uint8_t *getGOTBase() const { - llvm_unreachable("Unimplemented call"); - return 0; - } - virtual uint8_t *startFunctionBody(const Function *F, - uintptr_t &ActualSize){ - llvm_unreachable("Unimplemented call"); - return 0; - } - virtual uint8_t *allocateStub(const GlobalValue* F, unsigned StubSize, - unsigned Alignment) { - llvm_unreachable("Unimplemented call"); - return 0; - } - virtual void endFunctionBody(const Function *F, uint8_t *FunctionStart, - uint8_t *FunctionEnd) { - llvm_unreachable("Unimplemented call"); - } - virtual uint8_t *allocateSpace(intptr_t Size, unsigned Alignment) { - llvm_unreachable("Unimplemented call"); - return 0; - } - virtual uint8_t *allocateGlobal(uintptr_t Size, unsigned Alignment) { - llvm_unreachable("Unimplemented call"); - return 0; - } - virtual void deallocateFunctionBody(void *Body) { - llvm_unreachable("Unimplemented call"); - } - virtual uint8_t* startExceptionTable(const Function* F, - uintptr_t &ActualSize) { - llvm_unreachable("Unimplemented call"); - return 0; - } - virtual void endExceptionTable(const Function *F, uint8_t *TableStart, - uint8_t *TableEnd, uint8_t* FrameRegister) { - llvm_unreachable("Unimplemented call"); - } - virtual void deallocateExceptionTable(void *ET) { - llvm_unreachable("Unimplemented call"); - } -}; - -bool RustMCJITMemoryManager::loadCrate(const char* file, std::string* err) { - DynamicLibrary crate = DynamicLibrary::getPermanentLibrary(file, - err); - - if(crate.isValid()) { - crates.insert(&crate); - - return true; - } - - return false; -} - -uint8_t *RustMCJITMemoryManager::allocateDataSection(uintptr_t Size, - unsigned Alignment, - unsigned SectionID, - bool isReadOnly) { - if (!Alignment) - Alignment = 16; - uint8_t *Addr = (uint8_t*)calloc((Size + Alignment - 1)/Alignment, Alignment); - AllocatedDataMem.push_back(sys::MemoryBlock(Addr, Size)); - return Addr; -} - -bool RustMCJITMemoryManager::applyPermissions(std::string *Str) { - // Empty. - return true; -} - -uint8_t *RustMCJITMemoryManager::allocateCodeSection(uintptr_t Size, - unsigned Alignment, - unsigned SectionID) { - if (!Alignment) - Alignment = 16; - unsigned NeedAllocate = Alignment * ((Size + Alignment - 1)/Alignment + 1); - uintptr_t Addr = 0; - // Look in the list of free code memory regions and use a block there if one - // is available. - for (int i = 0, e = FreeCodeMem.size(); i != e; ++i) { - sys::MemoryBlock &MB = FreeCodeMem[i]; - if (MB.size() >= NeedAllocate) { - Addr = (uintptr_t)MB.base(); - uintptr_t EndOfBlock = Addr + MB.size(); - // Align the address. - Addr = (Addr + Alignment - 1) & ~(uintptr_t)(Alignment - 1); - // Store cutted free memory block. - FreeCodeMem[i] = sys::MemoryBlock((void*)(Addr + Size), - EndOfBlock - Addr - Size); - return (uint8_t*)Addr; - } - } - - // No pre-allocated free block was large enough. Allocate a new memory region. - sys::MemoryBlock MB = sys::Memory::AllocateRWX(NeedAllocate, 0, 0); - - AllocatedCodeMem.push_back(MB); - Addr = (uintptr_t)MB.base(); - uintptr_t EndOfBlock = Addr + MB.size(); - // Align the address. - Addr = (Addr + Alignment - 1) & ~(uintptr_t)(Alignment - 1); - // The AllocateRWX may allocate much more memory than we need. In this case, - // we store the unused memory as a free memory block. - unsigned FreeSize = EndOfBlock-Addr-Size; - if (FreeSize > 16) - FreeCodeMem.push_back(sys::MemoryBlock((void*)(Addr + Size), FreeSize)); - - // Return aligned address - return (uint8_t*)Addr; -} - -void RustMCJITMemoryManager::invalidateInstructionCache() { - for (int i = 0, e = AllocatedCodeMem.size(); i != e; ++i) - sys::Memory::InvalidateInstructionCache(AllocatedCodeMem[i].base(), - AllocatedCodeMem[i].size()); -} - -void *RustMCJITMemoryManager::getPointerToNamedFunction(const std::string &Name, - bool AbortOnFailure) { -#ifdef __linux__ - // Force the following functions to be linked in to anything that uses the - // JIT. This is a hack designed to work around the all-too-clever Glibc - // strategy of making these functions work differently when inlined vs. when - // not inlined, and hiding their real definitions in a separate archive file - // that the dynamic linker can't see. For more info, search for - // 'libc_nonshared.a' on Google, or read http://llvm.org/PR274. - if (Name == "stat") return (void*)(intptr_t)&stat; - if (Name == "fstat") return (void*)(intptr_t)&fstat; - if (Name == "lstat") return (void*)(intptr_t)&lstat; - if (Name == "stat64") return (void*)(intptr_t)&stat64; - if (Name == "fstat64") return (void*)(intptr_t)&fstat64; - if (Name == "lstat64") return (void*)(intptr_t)&lstat64; - if (Name == "atexit") return (void*)(intptr_t)&atexit; - if (Name == "mknod") return (void*)(intptr_t)&mknod; -#endif - - if (Name == "__morestack" || Name == "___morestack") return &__morestack; - - const char *NameStr = Name.c_str(); - - // Look through loaded crates and main for symbols. - - void *Ptr = sys::DynamicLibrary::SearchForAddressOfSymbol(NameStr); - if (Ptr) return Ptr; - - // If it wasn't found and if it starts with an underscore ('_') character, - // try again without the underscore. - if (NameStr[0] == '_') { - Ptr = sys::DynamicLibrary::SearchForAddressOfSymbol(NameStr+1); - if (Ptr) return Ptr; - } - - if (AbortOnFailure) - report_fatal_error("Program used external function '" + Name + - "' which could not be resolved!"); - return 0; -} - -RustMCJITMemoryManager::~RustMCJITMemoryManager() { - for (unsigned i = 0, e = AllocatedCodeMem.size(); i != e; ++i) - sys::Memory::ReleaseRWX(AllocatedCodeMem[i]); - for (unsigned i = 0, e = AllocatedDataMem.size(); i != e; ++i) - free(AllocatedDataMem[i].base()); -} - -extern "C" void* -LLVMRustPrepareJIT(void* __morestack) { - // An execution engine will take ownership of this later - // and clean it up for us. - - return (void*) new RustMCJITMemoryManager(__morestack); -} - -extern "C" bool -LLVMRustLoadCrate(void* mem, const char* crate) { - RustMCJITMemoryManager* manager = (RustMCJITMemoryManager*) mem; - std::string Err; - - assert(manager); - - if(!manager->loadCrate(crate, &Err)) { - LLVMRustError = Err.c_str(); - return false; - } - - return true; -} - -extern "C" LLVMExecutionEngineRef -LLVMRustBuildJIT(void* mem, - LLVMModuleRef M, - bool EnableSegmentedStacks) { - - InitializeNativeTarget(); - InitializeNativeTargetAsmPrinter(); - InitializeNativeTargetAsmParser(); - - std::string Err; - TargetOptions Options; - Options.JITEmitDebugInfo = true; - Options.NoFramePointerElim = true; - Options.EnableSegmentedStacks = EnableSegmentedStacks; - RustMCJITMemoryManager* MM = (RustMCJITMemoryManager*) mem; - assert(MM); - - ExecutionEngine* EE = EngineBuilder(unwrap(M)) - .setErrorStr(&Err) - .setTargetOptions(Options) - .setJITMemoryManager(MM) - .setUseMCJIT(true) - .setAllocateGVsWithCode(false) - .create(); - - if(!EE || Err != "") { - LLVMRustError = Err.c_str(); - // The EngineBuilder only takes ownership of these two structures if the - // create() call is successful, but here it wasn't successful. - LLVMDisposeModule(M); - delete MM; - return NULL; - } - - MM->invalidateInstructionCache(); - return wrap(EE); -} - extern "C" void LLVMRustSetNormalizedTarget(LLVMModuleRef M, const char *triple) { unwrap(M)->setTargetTriple(Triple::normalize(triple)); @@ -566,7 +290,7 @@ extern "C" LLVMValueRef LLVMDIBuilderCreateStructType( unwrapDI<DIType>(DerivedFrom), unwrapDI<DIArray>(Elements), RunTimeLang, - unwrapDI<MDNode*>(VTableHolder), + unwrapDI<DIType>(VTableHolder), UniqueId)); } |
