Fix: correct file name in diagnostics.

llvm-svn: 172405

1 files changed, 5 insertions, 4 deletions

diff --git a/clang-tools-extra/clang-format/ClangFormat.cpp b/clang-tools-extra/clang-format/ClangFormat.cpp
index da63130d4ca..507f55188f6 100644
--- a/clang-tools-extra/clang-format/ClangFormat.cpp
+++ b/clang-tools-extra/clang-format/ClangFormat.cpp
@@ -44,10 +44,11 @@ static cl::opt<std::string> FileName(cl::Positional, cl::desc("[<file>]"),
 namespace clang {
 namespace format {
 
-static FileID createInMemoryFile(const MemoryBuffer *Source,
+static FileID createInMemoryFile(StringRef FileName, const MemoryBuffer *Source,
                                  SourceManager &Sources, FileManager &Files) {
-  const FileEntry *Entry =
-      Files.getVirtualFile("<stdio>", Source->getBufferSize(), 0);
+  const FileEntry *Entry = Files.getVirtualFile(FileName == "-" ? "<stdin>" :
+                                                    FileName,
+                                                Source->getBufferSize(), 0);
   Sources.overrideFileContents(Entry, Source, true);
   return Sources.createFileID(Entry, SourceLocation(), SrcMgr::C_User);
 }
@@ -63,7 +64,7 @@ static void format() {
     llvm::errs() << ec.message() << "\n";
     return;
   }
-  FileID ID = createInMemoryFile(Code.get(), Sources, Files);
+  FileID ID = createInMemoryFile(FileName, Code.get(), Sources, Files);
   Lexer Lex(ID, Sources.getBuffer(ID), Sources, getFormattingLangOpts());
   SourceLocation Start =
       Sources.getLocForStartOfFile(ID).getLocWithOffset(Offset);