diff options
Diffstat (limited to 'llvm/lib/Support/SourceMgr.cpp')
| -rw-r--r-- | llvm/lib/Support/SourceMgr.cpp | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/llvm/lib/Support/SourceMgr.cpp b/llvm/lib/Support/SourceMgr.cpp index f2bbaab..299615a 100644 --- a/llvm/lib/Support/SourceMgr.cpp +++ b/llvm/lib/Support/SourceMgr.cpp @@ -69,11 +69,11 @@ unsigned SourceMgr::AddIncludeFile(const std::string &Filename, ErrorOr<std::unique_ptr<MemoryBuffer>> SourceMgr::OpenIncludeFile(const std::string &Filename, std::string &IncludedFile) { - if (!FS) - reportFatalInternalError("Opening include file from SourceMgr without VFS"); + auto GetFile = [this](StringRef Path) { + return FS ? FS->getBufferForFile(Path) : MemoryBuffer::getFile(Path); + }; - ErrorOr<std::unique_ptr<MemoryBuffer>> NewBufOrErr = - FS->getBufferForFile(Filename); + ErrorOr<std::unique_ptr<MemoryBuffer>> NewBufOrErr = GetFile(Filename); SmallString<64> Buffer(Filename); // If the file didn't exist directly, see if it's in an include path. @@ -81,7 +81,7 @@ SourceMgr::OpenIncludeFile(const std::string &Filename, ++i) { Buffer = IncludeDirectories[i]; sys::path::append(Buffer, Filename); - NewBufOrErr = FS->getBufferForFile(Buffer); + NewBufOrErr = GetFile(Buffer); } if (NewBufOrErr) |