diff options
Diffstat (limited to 'llvm/lib/Support/ToolOutputFile.cpp')
-rw-r--r-- | llvm/lib/Support/ToolOutputFile.cpp | 45 |
1 files changed, 45 insertions, 0 deletions
diff --git a/llvm/lib/Support/ToolOutputFile.cpp b/llvm/lib/Support/ToolOutputFile.cpp new file mode 100644 index 0000000000000..ed3a247f01155 --- /dev/null +++ b/llvm/lib/Support/ToolOutputFile.cpp @@ -0,0 +1,45 @@ +//===--- ToolOutputFile.cpp - Implement the ToolOutputFile class --------===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// +// +// This implements the ToolOutputFile class. +// +//===----------------------------------------------------------------------===// + +#include "llvm/Support/ToolOutputFile.h" +#include "llvm/Support/FileSystem.h" +#include "llvm/Support/Signals.h" +using namespace llvm; + +ToolOutputFile::CleanupInstaller::CleanupInstaller(StringRef Filename) + : Filename(Filename), Keep(false) { + // Arrange for the file to be deleted if the process is killed. + if (Filename != "-") + sys::RemoveFileOnSignal(Filename); +} + +ToolOutputFile::CleanupInstaller::~CleanupInstaller() { + // Delete the file if the client hasn't told us not to. + if (!Keep && Filename != "-") + 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); +} + +ToolOutputFile::ToolOutputFile(StringRef Filename, std::error_code &EC, + sys::fs::OpenFlags Flags) + : Installer(Filename), OS(Filename, EC, Flags) { + // If open fails, no cleanup is needed. + if (EC) + Installer.Keep = true; +} + +ToolOutputFile::ToolOutputFile(StringRef Filename, int FD) + : Installer(Filename), OS(FD, true) {} |