blob: 8bf81a2f838ba88daf1b11d6e2116e37fca490de (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
//===- ReaderWrappers.cpp - Parse bitcode from file or buffer -------------===//
//
// The LLVM Compiler Infrastructure
//
// This file was developed by Chris Lattner and is distributed under
// the University of Illinois Open Source License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
// This file implements loading and parsing a bitcode file and parsing a
// module from a memory buffer.
//
//===----------------------------------------------------------------------===//
#include "llvm/Bitcode/ReaderWriter.h"
#include "BitcodeReader.h"
#include "llvm/System/MappedFile.h"
using namespace llvm;
//===----------------------------------------------------------------------===//
// BitcodeFileReader - Read from an mmap'able file descriptor.
namespace {
/// BitcodeFileReader - parses bitcode from a file.
///
class BitcodeFileReader : public BitcodeReader {
private:
std::string Filename;
sys::MappedFile File;
BitcodeFileReader(const BitcodeFileReader&); // DO NOT IMPLEMENT
void operator=(const BitcodeFileReader&); // DO NOT IMPLEMENT
public:
BitcodeFileReader(const std::string &FN) : Filename(FN) {}
bool Read(std::string *ErrMsg);
void FreeState() {
BitcodeReader::FreeState();
File.close();
}
};
}
bool BitcodeFileReader::Read(std::string *ErrMsg) {
if (File.open(sys::Path(Filename), sys::MappedFile::READ_ACCESS, ErrMsg))
return true;
if (!File.map(ErrMsg)) {
File.close();
return true;
}
unsigned char *Buffer = reinterpret_cast<unsigned char*>(File.base());
if (!ParseBitcode(Buffer, File.size(), Filename))
return false;
assert(getErrorString() && "Didn't set an error string?");
if (ErrMsg) *ErrMsg = getErrorString();
return true;
}
//===----------------------------------------------------------------------===//
// External interface
//===----------------------------------------------------------------------===//
/// getBitcodeModuleProvider - lazy function-at-a-time loading from a file.
///
ModuleProvider *llvm::getBitcodeModuleProvider(const std::string &Filename,
std::string *ErrMsg) {
if (Filename != std::string("-")) {
BitcodeFileReader *R = new BitcodeFileReader(Filename);
if (R->Read(ErrMsg)) {
delete R;
return 0;
}
return R;
}
assert(0 && "FIXME: stdin reading unimp!");
#if 0
// Read from stdin
BytecodeStdinReader *R = new BytecodeStdinReader();
if (R->Read(ErrMsg)) {
delete R;
return 0;
}
return R;
#endif
}
/// ParseBitcodeFile - Read the specified bitcode file, returning the module.
/// If an error occurs, return null and fill in *ErrMsg if non-null.
Module *llvm::ParseBitcodeFile(const std::string &Filename,std::string *ErrMsg){
ModuleProvider *MP = getBitcodeModuleProvider(Filename, ErrMsg);
if (!MP) return 0;
Module *M = MP->releaseModule(ErrMsg);
delete MP;
return M;
}
|
