diff --git a/llvm/test/tools/llvm-dwp/X86/missing_dwo_id.test b/llvm/test/tools/llvm-dwp/X86/missing_dwo_id.test --- a/llvm/test/tools/llvm-dwp/X86/missing_dwo_id.test +++ b/llvm/test/tools/llvm-dwp/X86/missing_dwo_id.test @@ -1,3 +1,7 @@ +RUN: rm -f %t RUN: not llvm-dwp %p/../Inputs/missing_dwo_id.dwo -o %t 2>&1 | FileCheck %s +Make sure we did not leave behind a temporary file: +RUN: not ls %t + CHECK: error: {{.*}}missing_dwo_id.dwo': compile unit missing dwo_id diff --git a/llvm/tools/llvm-dwp/llvm-dwp.cpp b/llvm/tools/llvm-dwp/llvm-dwp.cpp --- a/llvm/tools/llvm-dwp/llvm-dwp.cpp +++ b/llvm/tools/llvm-dwp/llvm-dwp.cpp @@ -39,6 +39,7 @@ #include "llvm/Support/Path.h" #include "llvm/Support/TargetRegistry.h" #include "llvm/Support/TargetSelect.h" +#include "llvm/Support/ToolOutputFile.h" #include "llvm/Support/raw_ostream.h" using namespace llvm; @@ -700,23 +701,23 @@ // Create the output file. std::error_code EC; - raw_fd_ostream OutFile(OutputFilename, EC, sys::fs::F_None); + ToolOutputFile OutFile(OutputFilename, EC, sys::fs::F_None); Optional BOS; raw_pwrite_stream *OS; if (EC) return error(Twine(OutputFilename) + ": " + EC.message(), Context); - if (OutFile.supportsSeeking()) { - OS = &OutFile; + if (OutFile.os().supportsSeeking()) { + OS = &OutFile.os(); } else { - BOS.emplace(OutFile); + BOS.emplace(OutFile.os()); OS = BOS.getPointer(); } MCTargetOptions MCOptions = InitMCTargetOptionsFromFlags(); std::unique_ptr MS(TheTarget->createMCObjectStreamer( TheTriple, MC, std::unique_ptr(MAB), - MAB->createObjectWriter(*OS), std::unique_ptr(MCE), - *MSTI, MCOptions.MCRelaxAll, MCOptions.MCIncrementalLinkerCompatible, + MAB->createObjectWriter(*OS), std::unique_ptr(MCE), *MSTI, + MCOptions.MCRelaxAll, MCOptions.MCIncrementalLinkerCompatible, /*DWARFMustBeAtTheEnd*/ false)); if (!MS) return error("no object streamer for target " + TripleName, Context); @@ -739,4 +740,6 @@ } MS->Finish(); + OutFile.keep(); + return 0; }