IRReader.cpp revision 36b56886974eae4f9c5ebc96befd3e7bfe5de338
1//===---- IRReader.cpp - Reader for LLVM IR files -------------------------===// 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/IRReader/IRReader.h" 11#include "llvm-c/Core.h" 12#include "llvm-c/IRReader.h" 13#include "llvm/AsmParser/Parser.h" 14#include "llvm/Bitcode/ReaderWriter.h" 15#include "llvm/IR/LLVMContext.h" 16#include "llvm/IR/Module.h" 17#include "llvm/Support/MemoryBuffer.h" 18#include "llvm/Support/SourceMgr.h" 19#include "llvm/Support/Timer.h" 20#include "llvm/Support/raw_ostream.h" 21#include "llvm/Support/system_error.h" 22 23using namespace llvm; 24 25namespace llvm { 26 extern bool TimePassesIsEnabled; 27} 28 29static const char *const TimeIRParsingGroupName = "LLVM IR Parsing"; 30static const char *const TimeIRParsingName = "Parse IR"; 31 32 33Module *llvm::getLazyIRModule(MemoryBuffer *Buffer, SMDiagnostic &Err, 34 LLVMContext &Context) { 35 if (isBitcode((const unsigned char *)Buffer->getBufferStart(), 36 (const unsigned char *)Buffer->getBufferEnd())) { 37 std::string ErrMsg; 38 ErrorOr<Module *> ModuleOrErr = getLazyBitcodeModule(Buffer, Context); 39 if (error_code EC = ModuleOrErr.getError()) { 40 Err = SMDiagnostic(Buffer->getBufferIdentifier(), SourceMgr::DK_Error, 41 EC.message()); 42 // ParseBitcodeFile does not take ownership of the Buffer in the 43 // case of an error. 44 delete Buffer; 45 return NULL; 46 } 47 return ModuleOrErr.get(); 48 } 49 50 return ParseAssembly(Buffer, 0, Err, Context); 51} 52 53Module *llvm::getLazyIRFileModule(const std::string &Filename, SMDiagnostic &Err, 54 LLVMContext &Context) { 55 std::unique_ptr<MemoryBuffer> File; 56 if (error_code ec = MemoryBuffer::getFileOrSTDIN(Filename, File)) { 57 Err = SMDiagnostic(Filename, SourceMgr::DK_Error, 58 "Could not open input file: " + ec.message()); 59 return 0; 60 } 61 62 return getLazyIRModule(File.release(), Err, Context); 63} 64 65Module *llvm::ParseIR(MemoryBuffer *Buffer, SMDiagnostic &Err, 66 LLVMContext &Context) { 67 NamedRegionTimer T(TimeIRParsingName, TimeIRParsingGroupName, 68 TimePassesIsEnabled); 69 if (isBitcode((const unsigned char *)Buffer->getBufferStart(), 70 (const unsigned char *)Buffer->getBufferEnd())) { 71 ErrorOr<Module *> ModuleOrErr = parseBitcodeFile(Buffer, Context); 72 Module *M = 0; 73 if (error_code EC = ModuleOrErr.getError()) 74 Err = SMDiagnostic(Buffer->getBufferIdentifier(), SourceMgr::DK_Error, 75 EC.message()); 76 else 77 M = ModuleOrErr.get(); 78 // parseBitcodeFile does not take ownership of the Buffer. 79 delete Buffer; 80 return M; 81 } 82 83 return ParseAssembly(Buffer, 0, Err, Context); 84} 85 86Module *llvm::ParseIRFile(const std::string &Filename, SMDiagnostic &Err, 87 LLVMContext &Context) { 88 std::unique_ptr<MemoryBuffer> File; 89 if (error_code ec = MemoryBuffer::getFileOrSTDIN(Filename, File)) { 90 Err = SMDiagnostic(Filename, SourceMgr::DK_Error, 91 "Could not open input file: " + ec.message()); 92 return 0; 93 } 94 95 return ParseIR(File.release(), Err, Context); 96} 97 98//===----------------------------------------------------------------------===// 99// C API. 100//===----------------------------------------------------------------------===// 101 102LLVMBool LLVMParseIRInContext(LLVMContextRef ContextRef, 103 LLVMMemoryBufferRef MemBuf, LLVMModuleRef *OutM, 104 char **OutMessage) { 105 SMDiagnostic Diag; 106 107 *OutM = wrap(ParseIR(unwrap(MemBuf), Diag, *unwrap(ContextRef))); 108 109 if(!*OutM) { 110 if (OutMessage) { 111 std::string buf; 112 raw_string_ostream os(buf); 113 114 Diag.print(NULL, os, false); 115 os.flush(); 116 117 *OutMessage = strdup(buf.c_str()); 118 } 119 return 1; 120 } 121 122 return 0; 123} 124