140 lines · cpp
1//===---- IRReader.cpp - Reader for LLVM IR files -------------------------===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8 9#include "llvm/IRReader/IRReader.h"10#include "llvm-c/IRReader.h"11#include "llvm/AsmParser/AsmParserContext.h"12#include "llvm/AsmParser/Parser.h"13#include "llvm/Bitcode/BitcodeReader.h"14#include "llvm/IR/LLVMContext.h"15#include "llvm/IR/Module.h"16#include "llvm/Support/MemoryBuffer.h"17#include "llvm/Support/SourceMgr.h"18#include "llvm/Support/Timer.h"19#include "llvm/Support/raw_ostream.h"20#include <optional>21#include <system_error>22 23using namespace llvm;24 25namespace llvm {26 extern bool TimePassesIsEnabled;27}28 29const char TimeIRParsingGroupName[] = "irparse";30const char TimeIRParsingGroupDescription[] = "LLVM IR Parsing";31const char TimeIRParsingName[] = "parse";32const char TimeIRParsingDescription[] = "Parse IR";33 34std::unique_ptr<Module>35llvm::getLazyIRModule(std::unique_ptr<MemoryBuffer> Buffer, SMDiagnostic &Err,36 LLVMContext &Context, bool ShouldLazyLoadMetadata) {37 if (isBitcode((const unsigned char *)Buffer->getBufferStart(),38 (const unsigned char *)Buffer->getBufferEnd())) {39 Expected<std::unique_ptr<Module>> ModuleOrErr = getOwningLazyBitcodeModule(40 std::move(Buffer), Context, ShouldLazyLoadMetadata);41 if (Error E = ModuleOrErr.takeError()) {42 handleAllErrors(std::move(E), [&](ErrorInfoBase &EIB) {43 Err = SMDiagnostic(Buffer->getBufferIdentifier(), SourceMgr::DK_Error,44 EIB.message());45 });46 return nullptr;47 }48 return std::move(ModuleOrErr.get());49 }50 51 return parseAssembly(Buffer->getMemBufferRef(), Err, Context);52}53 54std::unique_ptr<Module> llvm::getLazyIRFileModule(StringRef Filename,55 SMDiagnostic &Err,56 LLVMContext &Context,57 bool ShouldLazyLoadMetadata) {58 ErrorOr<std::unique_ptr<MemoryBuffer>> FileOrErr =59 MemoryBuffer::getFileOrSTDIN(Filename);60 if (std::error_code EC = FileOrErr.getError()) {61 Err = SMDiagnostic(Filename, SourceMgr::DK_Error,62 "Could not open input file: " + EC.message());63 return nullptr;64 }65 66 return getLazyIRModule(std::move(FileOrErr.get()), Err, Context,67 ShouldLazyLoadMetadata);68}69 70std::unique_ptr<Module> llvm::parseIR(MemoryBufferRef Buffer, SMDiagnostic &Err,71 LLVMContext &Context,72 ParserCallbacks Callbacks,73 llvm::AsmParserContext *ParserContext) {74 NamedRegionTimer T(TimeIRParsingName, TimeIRParsingDescription,75 TimeIRParsingGroupName, TimeIRParsingGroupDescription,76 TimePassesIsEnabled);77 if (isBitcode((const unsigned char *)Buffer.getBufferStart(),78 (const unsigned char *)Buffer.getBufferEnd())) {79 Expected<std::unique_ptr<Module>> ModuleOrErr =80 parseBitcodeFile(Buffer, Context, Callbacks);81 if (Error E = ModuleOrErr.takeError()) {82 handleAllErrors(std::move(E), [&](ErrorInfoBase &EIB) {83 Err = SMDiagnostic(Buffer.getBufferIdentifier(), SourceMgr::DK_Error,84 EIB.message());85 });86 return nullptr;87 }88 return std::move(ModuleOrErr.get());89 }90 91 return parseAssembly(Buffer, Err, Context, nullptr,92 Callbacks.DataLayout.value_or(93 [](StringRef, StringRef) { return std::nullopt; }),94 ParserContext);95}96 97std::unique_ptr<Module> llvm::parseIRFile(StringRef Filename, SMDiagnostic &Err,98 LLVMContext &Context,99 ParserCallbacks Callbacks,100 AsmParserContext *ParserContext) {101 ErrorOr<std::unique_ptr<MemoryBuffer>> FileOrErr =102 MemoryBuffer::getFileOrSTDIN(Filename, /*IsText=*/true);103 if (std::error_code EC = FileOrErr.getError()) {104 Err = SMDiagnostic(Filename, SourceMgr::DK_Error,105 "Could not open input file: " + EC.message());106 return nullptr;107 }108 109 return parseIR(FileOrErr.get()->getMemBufferRef(), Err, Context, Callbacks,110 ParserContext);111}112 113//===----------------------------------------------------------------------===//114// C API.115//===----------------------------------------------------------------------===//116 117LLVMBool LLVMParseIRInContext(LLVMContextRef ContextRef,118 LLVMMemoryBufferRef MemBuf, LLVMModuleRef *OutM,119 char **OutMessage) {120 SMDiagnostic Diag;121 122 std::unique_ptr<MemoryBuffer> MB(unwrap(MemBuf));123 *OutM =124 wrap(parseIR(MB->getMemBufferRef(), Diag, *unwrap(ContextRef)).release());125 126 if(!*OutM) {127 if (OutMessage) {128 std::string buf;129 raw_string_ostream os(buf);130 131 Diag.print(nullptr, os, false);132 133 *OutMessage = strdup(buf.c_str());134 }135 return 1;136 }137 138 return 0;139}140