LLVM  6.0.0svn
IRReader.cpp
Go to the documentation of this file.
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"
15 #include "llvm/IR/LLVMContext.h"
16 #include "llvm/IR/Module.h"
18 #include "llvm/Support/SourceMgr.h"
19 #include "llvm/Support/Timer.h"
21 #include <system_error>
22 
23 using namespace llvm;
24 
25 namespace llvm {
26  extern bool TimePassesIsEnabled;
27 }
28 
29 static const char *const TimeIRParsingGroupName = "irparse";
30 static const char *const TimeIRParsingGroupDescription = "LLVM IR Parsing";
31 static const char *const TimeIRParsingName = "parse";
32 static const char *const TimeIRParsingDescription = "Parse IR";
33 
34 static std::unique_ptr<Module>
35 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())) {
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 
54 std::unique_ptr<Module> llvm::getLazyIRFileModule(StringRef Filename,
55  SMDiagnostic &Err,
57  bool ShouldLazyLoadMetadata) {
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 
70 std::unique_ptr<Module> llvm::parseIR(MemoryBufferRef Buffer, SMDiagnostic &Err,
72  bool UpgradeDebugInfo) {
73  NamedRegionTimer T(TimeIRParsingName, TimeIRParsingDescription,
74  TimeIRParsingGroupName, TimeIRParsingGroupDescription,
76  if (isBitcode((const unsigned char *)Buffer.getBufferStart(),
77  (const unsigned char *)Buffer.getBufferEnd())) {
79  parseBitcodeFile(Buffer, Context);
80  if (Error E = ModuleOrErr.takeError()) {
81  handleAllErrors(std::move(E), [&](ErrorInfoBase &EIB) {
83  EIB.message());
84  });
85  return nullptr;
86  }
87  return std::move(ModuleOrErr.get());
88  }
89 
90  return parseAssembly(Buffer, Err, Context, nullptr, UpgradeDebugInfo);
91 }
92 
93 std::unique_ptr<Module> llvm::parseIRFile(StringRef Filename, SMDiagnostic &Err,
95  bool UpgradeDebugInfo) {
98  if (std::error_code EC = FileOrErr.getError()) {
99  Err = SMDiagnostic(Filename, SourceMgr::DK_Error,
100  "Could not open input file: " + EC.message());
101  return nullptr;
102  }
103 
104  return parseIR(FileOrErr.get()->getMemBufferRef(), Err, Context,
106 }
107 
108 //===----------------------------------------------------------------------===//
109 // C API.
110 //===----------------------------------------------------------------------===//
111 
113  LLVMMemoryBufferRef MemBuf, LLVMModuleRef *OutM,
114  char **OutMessage) {
115  SMDiagnostic Diag;
116 
117  std::unique_ptr<MemoryBuffer> MB(unwrap(MemBuf));
118  *OutM =
119  wrap(parseIR(MB->getMemBufferRef(), Diag, *unwrap(ContextRef)).release());
120 
121  if(!*OutM) {
122  if (OutMessage) {
123  std::string buf;
124  raw_string_ostream os(buf);
125 
126  Diag.print(nullptr, os, false);
127  os.flush();
128 
129  *OutMessage = strdup(buf.c_str());
130  }
131  return 1;
132  }
133 
134  return 0;
135 }
std::unique_ptr< Module > parseAssembly(MemoryBufferRef F, SMDiagnostic &Err, LLVMContext &Context, SlotMapping *Slots=nullptr, bool UpgradeDebugInfo=true)
parseAssemblyFile and parseAssemblyString are wrappers around this function.
Definition: Parser.cpp:35
static const char *const TimeIRParsingGroupDescription
Definition: IRReader.cpp:30
Represents either an error or a value T.
Definition: ErrorOr.h:69
std::unique_ptr< Module > parseIR(MemoryBufferRef Buffer, SMDiagnostic &Err, LLVMContext &Context, bool UpgradeDebugInfo=true)
If the given MemoryBuffer holds a bitcode image, return a Module for it.
Definition: IRReader.cpp:70
LLVMContext & Context
Compute iterated dominance frontiers using a linear time algorithm.
Definition: AllocatorList.h:24
struct LLVMOpaqueModule * LLVMModuleRef
The top-level container for all other LLVM Intermediate Representation (IR) objects.
Definition: Types.h:62
void print(const char *ProgName, raw_ostream &S, bool ShowColors=true, bool ShowKindLabel=true) const
Definition: SourceMgr.cpp:345
const char * getBufferEnd() const
Definition: MemoryBuffer.h:175
struct LLVMOpaqueMemoryBuffer * LLVMMemoryBufferRef
LLVM uses a polymorphic type hierarchy which C cannot represent, therefore parameters must be passed ...
Definition: Types.h:49
virtual std::string message() const
Return the error message as a string.
Definition: Error.h:55
Error takeError()
Take ownership of the stored error.
Definition: Error.h:537
Base class for error info classes.
Definition: Error.h:47
Attribute unwrap(LLVMAttributeRef Attr)
Definition: Attributes.h:195
This class is basically a combination of TimeRegion and Timer.
Definition: Timer.h:160
Tagged union holding either a T or a Error.
Definition: CachePruning.h:23
#define T
struct LLVMOpaqueContext * LLVMContextRef
The top-level container for all LLVM global data.
Definition: Types.h:54
static const char *const TimeIRParsingName
Definition: IRReader.cpp:31
This is an important class for using LLVM in a threaded context.
Definition: LLVMContext.h:69
static GCRegistry::Add< CoreCLRGC > E("coreclr", "CoreCLR-compatible GC")
std::error_code getError() const
Definition: ErrorOr.h:172
bool UpgradeDebugInfo(Module &M)
Check the debug info version number, if it is out-dated, drop the debug info.
int LLVMBool
Definition: Types.h:29
std::unique_ptr< Module > getLazyIRFileModule(StringRef Filename, SMDiagnostic &Err, LLVMContext &Context, bool ShouldLazyLoadMetadata=false)
If the given file holds a bitcode image, return a Module for it which does lazy deserialization of fu...
Definition: IRReader.cpp:54
void handleAllErrors(Error E, HandlerTs &&... Handlers)
Behaves the same as handleErrors, except that it requires that all errors be handled by the given han...
Definition: Error.h:889
Module.h This file contains the declarations for the Module class.
bool isBitcode(const unsigned char *BufPtr, const unsigned char *BufEnd)
isBitcode - Return true if the given bytes are the magic bytes for LLVM IR bitcode, either with or without a wrapper.
LLVMBool LLVMParseIRInContext(LLVMContextRef ContextRef, LLVMMemoryBufferRef MemBuf, LLVMModuleRef *OutM, char **OutMessage)
Read LLVM IR from a memory buffer and convert it into an in-memory Module object. ...
Definition: IRReader.cpp:112
reference get()
Returns a reference to the stored T value.
Definition: Error.h:517
StringRef getBufferIdentifier() const
Definition: MemoryBuffer.h:172
Expected< std::unique_ptr< Module > > parseBitcodeFile(MemoryBufferRef Buffer, LLVMContext &Context)
Read the specified bitcode file, returning the module.
static ErrorOr< std::unique_ptr< MemoryBuffer > > getFileOrSTDIN(const Twine &Filename, int64_t FileSize=-1, bool RequiresNullTerminator=true)
Open the specified file as a MemoryBuffer, or open stdin if the Filename is "-".
LLVMAttributeRef wrap(Attribute Attr)
Definition: Attributes.h:190
static const char *const TimeIRParsingGroupName
Definition: IRReader.cpp:29
Expected< std::unique_ptr< Module > > getOwningLazyBitcodeModule(std::unique_ptr< MemoryBuffer > &&Buffer, LLVMContext &Context, bool ShouldLazyLoadMetadata=false, bool IsImporting=false)
Like getLazyBitcodeModule, except that the module takes ownership of the memory buffer if successful...
const char * getBufferStart() const
Definition: MemoryBuffer.h:174
A raw_ostream that writes to an std::string.
Definition: raw_ostream.h:462
Lightweight error class with error context and mandatory checking.
Definition: Error.h:156
StringRef - Represent a constant reference to a string, i.e.
Definition: StringRef.h:49
std::unique_ptr< Module > parseIRFile(StringRef Filename, SMDiagnostic &Err, LLVMContext &Context, bool UpgradeDebugInfo=true)
If the given file holds a bitcode image, return a Module for it.
Definition: IRReader.cpp:93
bool TimePassesIsEnabled
If the user specifies the -time-passes argument on an LLVM tool command line then the value of this b...
Definition: IRReader.cpp:26
static const char *const TimeIRParsingDescription
Definition: IRReader.cpp:32
static std::unique_ptr< Module > getLazyIRModule(std::unique_ptr< MemoryBuffer > Buffer, SMDiagnostic &Err, LLVMContext &Context, bool ShouldLazyLoadMetadata)
Definition: IRReader.cpp:35
reference get()
Definition: ErrorOr.h:169
Instances of this class encapsulate one diagnostic report, allowing printing to a raw_ostream as a ca...
Definition: SourceMgr.h:233