diff --git a/llvm/include/llvm/Support/ToolOutputFile.h b/llvm/include/llvm/Support/ToolOutputFile.h --- a/llvm/include/llvm/Support/ToolOutputFile.h +++ b/llvm/include/llvm/Support/ToolOutputFile.h @@ -38,8 +38,11 @@ ~CleanupInstaller(); } Installer; - /// The contained stream. This is intentionally declared after Installer. - raw_fd_ostream OS; + /// Storage for the stream, if we're owning our own stream. + std::unique_ptr OSHolder; + + /// The actual stream to use. + raw_fd_ostream *OS; public: /// This constructor's arguments are passed to raw_fd_ostream's @@ -50,7 +53,7 @@ ToolOutputFile(StringRef Filename, int FD); /// Return the contained raw_fd_ostream. - raw_fd_ostream &os() { return OS; } + raw_fd_ostream &os() { return *OS; } /// Indicate that the tool's job wrt this output file has been successful and /// the file should not be deleted. diff --git a/llvm/include/llvm/Support/raw_ostream.h b/llvm/include/llvm/Support/raw_ostream.h --- a/llvm/include/llvm/Support/raw_ostream.h +++ b/llvm/include/llvm/Support/raw_ostream.h @@ -496,13 +496,13 @@ void clear_error() { EC = std::error_code(); } }; -/// This returns a reference to a raw_ostream for standard output. Use it like: -/// outs() << "foo" << "bar"; -raw_ostream &outs(); +/// This returns a reference to a raw_fd_ostream for standard output. Use it +/// like: outs() << "foo" << "bar"; +raw_fd_ostream &outs(); -/// This returns a reference to a raw_ostream for standard error. Use it like: -/// errs() << "foo" << "bar"; -raw_ostream &errs(); +/// This returns a reference to a raw_fd_ostream for standard error. Use it +/// like: errs() << "foo" << "bar"; +raw_fd_ostream &errs(); /// This returns a reference to a raw_ostream which simply discards output. raw_ostream &nulls(); diff --git a/llvm/lib/Support/ToolOutputFile.cpp b/llvm/lib/Support/ToolOutputFile.cpp --- a/llvm/lib/Support/ToolOutputFile.cpp +++ b/llvm/lib/Support/ToolOutputFile.cpp @@ -15,31 +15,43 @@ #include "llvm/Support/Signals.h" using namespace llvm; +static bool isStdout(StringRef Filename) { return Filename == "-"; } + ToolOutputFile::CleanupInstaller::CleanupInstaller(StringRef Filename) : Filename(std::string(Filename)), Keep(false) { // Arrange for the file to be deleted if the process is killed. - if (Filename != "-") + if (isStdout(Filename)) sys::RemoveFileOnSignal(Filename); } ToolOutputFile::CleanupInstaller::~CleanupInstaller() { + if (isStdout(Filename)) + return; + // Delete the file if the client hasn't told us not to. - if (!Keep && Filename != "-") + if (!Keep) sys::fs::remove(Filename); // Ok, the file is successfully written and closed, or deleted. There's no // further need to clean it up on signals. - if (Filename != "-") - sys::DontRemoveFileOnSignal(Filename); + sys::DontRemoveFileOnSignal(Filename); } ToolOutputFile::ToolOutputFile(StringRef Filename, std::error_code &EC, sys::fs::OpenFlags Flags) - : Installer(Filename), OS(Filename, EC, Flags) { + : Installer(Filename) { + if (isStdout(Filename)) { + OS = &outs(); + EC = std::error_code(); + return; + } + OSHolder = std::make_unique(Filename, EC, Flags); + OS = OSHolder.get(); // If open fails, no cleanup is needed. if (EC) Installer.Keep = true; } ToolOutputFile::ToolOutputFile(StringRef Filename, int FD) - : Installer(Filename), OS(FD, true) {} + : Installer(Filename), OSHolder(new raw_fd_ostream(FD, true)), + OS(OSHolder.get()) {} diff --git a/llvm/lib/Support/raw_ostream.cpp b/llvm/lib/Support/raw_ostream.cpp --- a/llvm/lib/Support/raw_ostream.cpp +++ b/llvm/lib/Support/raw_ostream.cpp @@ -864,9 +864,7 @@ // outs(), errs(), nulls() //===----------------------------------------------------------------------===// -/// outs() - This returns a reference to a raw_ostream for standard output. -/// Use it like: outs() << "foo" << "bar"; -raw_ostream &llvm::outs() { +raw_fd_ostream &llvm::outs() { // Set buffer settings to model stdout behavior. std::error_code EC; static raw_fd_ostream S("-", EC, sys::fs::OF_None); @@ -874,9 +872,7 @@ return S; } -/// errs() - This returns a reference to a raw_ostream for standard error. -/// Use it like: errs() << "foo" << "bar"; -raw_ostream &llvm::errs() { +raw_fd_ostream &llvm::errs() { // Set standard error to be unbuffered by default. static raw_fd_ostream S(STDERR_FILENO, false, true); return S;