1 //===-- BitReader.cpp -----------------------------------------------------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 10 #include "llvm-c/BitReader.h" 11 #include "llvm-c/Core.h" 12 #include "llvm/Bitcode/BitcodeReader.h" 13 #include "llvm/IR/DiagnosticPrinter.h" 14 #include "llvm/IR/LLVMContext.h" 15 #include "llvm/IR/Module.h" 16 #include "llvm/Support/MemoryBuffer.h" 17 #include "llvm/Support/raw_ostream.h" 18 #include <cstring> 19 #include <string> 20 21 using namespace llvm; 22 23 /* Builds a module from the bitcode in the specified memory buffer, returning a 24 reference to the module via the OutModule parameter. Returns 0 on success. 25 Optionally returns a human-readable error message via OutMessage. */ 26 LLVMBool LLVMParseBitcode(LLVMMemoryBufferRef MemBuf, LLVMModuleRef *OutModule, 27 char **OutMessage) { 28 return LLVMParseBitcodeInContext(LLVMGetGlobalContext(), MemBuf, OutModule, 29 OutMessage); 30 } 31 32 LLVMBool LLVMParseBitcode2(LLVMMemoryBufferRef MemBuf, 33 LLVMModuleRef *OutModule) { 34 return LLVMParseBitcodeInContext2(LLVMGetGlobalContext(), MemBuf, OutModule); 35 } 36 37 LLVMBool LLVMParseBitcodeInContext(LLVMContextRef ContextRef, 38 LLVMMemoryBufferRef MemBuf, 39 LLVMModuleRef *OutModule, 40 char **OutMessage) { 41 MemoryBufferRef Buf = unwrap(MemBuf)->getMemBufferRef(); 42 LLVMContext &Ctx = *unwrap(ContextRef); 43 44 Expected<std::unique_ptr<Module>> ModuleOrErr = parseBitcodeFile(Buf, Ctx); 45 if (Error Err = ModuleOrErr.takeError()) { 46 std::string Message; 47 handleAllErrors(std::move(Err), [&](ErrorInfoBase &EIB) { 48 Message = EIB.message(); 49 }); 50 if (OutMessage) 51 *OutMessage = strdup(Message.c_str()); 52 *OutModule = wrap((Module *)nullptr); 53 return 1; 54 } 55 56 *OutModule = wrap(ModuleOrErr.get().release()); 57 return 0; 58 } 59 60 LLVMBool LLVMParseBitcodeInContext2(LLVMContextRef ContextRef, 61 LLVMMemoryBufferRef MemBuf, 62 LLVMModuleRef *OutModule) { 63 MemoryBufferRef Buf = unwrap(MemBuf)->getMemBufferRef(); 64 LLVMContext &Ctx = *unwrap(ContextRef); 65 66 ErrorOr<std::unique_ptr<Module>> ModuleOrErr = 67 expectedToErrorOrAndEmitErrors(Ctx, parseBitcodeFile(Buf, Ctx)); 68 if (ModuleOrErr.getError()) { 69 *OutModule = wrap((Module *)nullptr); 70 return 1; 71 } 72 73 *OutModule = wrap(ModuleOrErr.get().release()); 74 return 0; 75 } 76 77 /* Reads a module from the specified path, returning via the OutModule parameter 78 a module provider which performs lazy deserialization. Returns 0 on success. 79 Optionally returns a human-readable error message via OutMessage. */ 80 LLVMBool LLVMGetBitcodeModuleInContext(LLVMContextRef ContextRef, 81 LLVMMemoryBufferRef MemBuf, 82 LLVMModuleRef *OutM, char **OutMessage) { 83 LLVMContext &Ctx = *unwrap(ContextRef); 84 std::unique_ptr<MemoryBuffer> Owner(unwrap(MemBuf)); 85 Expected<std::unique_ptr<Module>> ModuleOrErr = 86 getOwningLazyBitcodeModule(std::move(Owner), Ctx); 87 Owner.release(); 88 89 if (Error Err = ModuleOrErr.takeError()) { 90 std::string Message; 91 handleAllErrors(std::move(Err), [&](ErrorInfoBase &EIB) { 92 Message = EIB.message(); 93 }); 94 if (OutMessage) 95 *OutMessage = strdup(Message.c_str()); 96 *OutM = wrap((Module *)nullptr); 97 return 1; 98 } 99 100 *OutM = wrap(ModuleOrErr.get().release()); 101 102 return 0; 103 } 104 105 LLVMBool LLVMGetBitcodeModuleInContext2(LLVMContextRef ContextRef, 106 LLVMMemoryBufferRef MemBuf, 107 LLVMModuleRef *OutM) { 108 LLVMContext &Ctx = *unwrap(ContextRef); 109 std::unique_ptr<MemoryBuffer> Owner(unwrap(MemBuf)); 110 111 ErrorOr<std::unique_ptr<Module>> ModuleOrErr = expectedToErrorOrAndEmitErrors( 112 Ctx, getOwningLazyBitcodeModule(std::move(Owner), Ctx)); 113 Owner.release(); 114 115 if (ModuleOrErr.getError()) { 116 *OutM = wrap((Module *)nullptr); 117 return 1; 118 } 119 120 *OutM = wrap(ModuleOrErr.get().release()); 121 return 0; 122 } 123 124 LLVMBool LLVMGetBitcodeModule(LLVMMemoryBufferRef MemBuf, LLVMModuleRef *OutM, 125 char **OutMessage) { 126 return LLVMGetBitcodeModuleInContext(LLVMGetGlobalContext(), MemBuf, OutM, 127 OutMessage); 128 } 129 130 LLVMBool LLVMGetBitcodeModule2(LLVMMemoryBufferRef MemBuf, 131 LLVMModuleRef *OutM) { 132 return LLVMGetBitcodeModuleInContext2(LLVMGetGlobalContext(), MemBuf, OutM); 133 } 134