6502/llvm-6502
1
0
Fork 0
mirror of https://github.com/c64scene-ar/llvm-6502.git synced 2025-06-13 04:38:24 +00:00
Code Issues Projects Releases Wiki Activity

Add basic error checking to MemoryBuffer::getSTDIN.

Browse Source 
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@104855 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Dan Gohman
2010-05-27 17:31:51 +00:00
parent 95131fcb67
commit 30377e7809
4 changed files with 26 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lib/Support/MemoryBuffer.cpp
View File

@ -137,7 +137,7 @@ MemoryBuffer *MemoryBuffer::getFileOrSTDIN(StringRef Filename,
int64_t FileSize,
struct stat *FileInfo) {
if (Filename == "-")
return getSTDIN();
return getSTDIN(ErrStr);
return getFile(Filename, ErrStr, FileSize, FileInfo);
}
@ -263,7 +263,7 @@ public:
};
}
MemoryBuffer *MemoryBuffer::getSTDIN() {
MemoryBuffer *MemoryBuffer::getSTDIN(std::string *ErrStr) {
char Buffer[4096*4];
std::vector<char> FileData;
@ -279,6 +279,11 @@ MemoryBuffer *MemoryBuffer::getSTDIN() {
FileData.insert(FileData.end(), Buffer, Buffer+ReadBytes);
} while (ReadBytes == sizeof(Buffer));
if (!feof(stdin)) {
if (ErrStr) *ErrStr = "error reading from stdin";
return 0;
}
FileData.push_back(0); // &FileData[Size] is invalid. So is &*FileData.end().
size_t Size = FileData.size();
MemoryBuffer *B = new STDINBufferFile();