Line data    Source code

       1             : //===--- ToolOutputFile.cpp - Implement the ToolOutputFile class --------===//
       2             : //
       3             : //                     The LLVM Compiler Infrastructure
       4             : //
       5             : // This file is distributed under the University of Illinois Open Source
       6             : // License. See LICENSE.TXT for details.
       7             : //
       8             : //===----------------------------------------------------------------------===//
       9             : //
      10             : // This implements the ToolOutputFile class.
      11             : //
      12             : //===----------------------------------------------------------------------===//
      13             : 
      14             : #include "llvm/Support/ToolOutputFile.h"
      15             : #include "llvm/Support/FileSystem.h"
      16             : #include "llvm/Support/Signals.h"
      17             : using namespace llvm;
      18             : 
      19       42935 : ToolOutputFile::CleanupInstaller::CleanupInstaller(StringRef Filename)
      20       42935 :     : Filename(Filename), Keep(false) {
      21             :   // Arrange for the file to be deleted if the process is killed.
      22             :   if (Filename != "-")
      23        4639 :     sys::RemoveFileOnSignal(Filename);
      24       42935 : }
      25             : 
      26       85238 : ToolOutputFile::CleanupInstaller::~CleanupInstaller() {
      27             :   // Delete the file if the client hasn't told us not to.
      28       42619 :   if (!Keep && Filename != "-")
      29         318 :     sys::fs::remove(Filename);
      30             : 
      31             :   // Ok, the file is successfully written and closed, or deleted. There's no
      32             :   // further need to clean it up on signals.
      33       42619 :   if (Filename != "-")
      34        4580 :     sys::DontRemoveFileOnSignal(Filename);
      35       42619 : }
      36             : 
      37       42918 : ToolOutputFile::ToolOutputFile(StringRef Filename, std::error_code &EC,
      38       42918 :                                sys::fs::OpenFlags Flags)
      39       42918 :     : Installer(Filename), OS(Filename, EC, Flags) {
      40             :   // If open fails, no cleanup is needed.
      41       42918 :   if (EC)
      42           0 :     Installer.Keep = true;
      43       42918 : }
      44             : 
      45          17 : ToolOutputFile::ToolOutputFile(StringRef Filename, int FD)
      46          17 :     : Installer(Filename), OS(FD, true) {}