1//===--- ToolOutputFile.cpp - Implement the ToolOutputFile class --------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// This implements the ToolOutputFile class.
10//
11//===----------------------------------------------------------------------===//
12
13#include "llvm/Support/ToolOutputFile.h"
14#include "llvm/Support/FileSystem.h"
15#include "llvm/Support/Signals.h"
16using namespace llvm;
17
18static bool isStdout(StringRef Filename) { return Filename == "-"; }
19
20ToolOutputFile::CleanupInstaller::CleanupInstaller(StringRef Filename)
21    : Filename(std::string(Filename)), Keep(false) {
22  // Arrange for the file to be deleted if the process is killed.
23  if (!isStdout(Filename))
24    sys::RemoveFileOnSignal(Filename);
25}
26
27ToolOutputFile::CleanupInstaller::~CleanupInstaller() {
28  if (isStdout(Filename))
29    return;
30
31  // Delete the file if the client hasn't told us not to.
32  if (!Keep)
33    sys::fs::remove(Filename);
34
35  // Ok, the file is successfully written and closed, or deleted. There's no
36  // further need to clean it up on signals.
37  sys::DontRemoveFileOnSignal(Filename);
38}
39
40ToolOutputFile::ToolOutputFile(StringRef Filename, std::error_code &EC,
41                               sys::fs::OpenFlags Flags)
42    : Installer(Filename) {
43  if (isStdout(Filename)) {
44    OS = &outs();
45    EC = std::error_code();
46    return;
47  }
48  OSHolder.emplace(Filename, EC, Flags);
49  OS = &*OSHolder;
50  // If open fails, no cleanup is needed.
51  if (EC)
52    Installer.Keep = true;
53}
54
55ToolOutputFile::ToolOutputFile(StringRef Filename, int FD)
56    : Installer(Filename) {
57  OSHolder.emplace(FD, true);
58  OS = &*OSHolder;
59}
60