From 96cd7af935dd2813e66335a76c4715ced5a36cec Mon Sep 17 00:00:00 2001 From: Eli Friedman Date: Mon, 18 May 2009 08:44:04 +0000 Subject: [PATCH] Tweak MemoryBuffer::getSTDIN so that it returns after the first EOF. It doesn't matter for piped input, but it's annoying when typing at the console. git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@71998 91177308-0d34-0410-b5e6-96231b3b80d8 --- lib/Support/MemoryBuffer.cpp | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/lib/Support/MemoryBuffer.cpp b/lib/Support/MemoryBuffer.cpp index a4963d92fb6..e35c626c408 100644 --- a/lib/Support/MemoryBuffer.cpp +++ b/lib/Support/MemoryBuffer.cpp @@ -258,13 +258,16 @@ public: MemoryBuffer *MemoryBuffer::getSTDIN() { char Buffer[4096*4]; - + std::vector FileData; - + // Read in all of the data from stdin, we cannot mmap stdin. sys::Program::ChangeStdinToBinary(); - while (size_t ReadBytes = fread(Buffer, sizeof(char), 4096*4, stdin)) + size_t ReadBytes; + do { + ReadBytes = fread(Buffer, sizeof(char), sizeof(Buffer), stdin); FileData.insert(FileData.end(), Buffer, Buffer+ReadBytes); + } while (ReadBytes == sizeof(Buffer)); FileData.push_back(0); // &FileData[Size] is invalid. So is &*FileData.end(). size_t Size = FileData.size(); -- 2.34.1