diff options
author | Dan Gohman <gohman@apple.com> | 2010-08-20 01:12:13 +0000 |
---|---|---|
committer | Dan Gohman <gohman@apple.com> | 2010-08-20 01:12:13 +0000 |
commit | 2df9504fec1ddf198321c7fe8c968154b4edbff3 (patch) | |
tree | 7b0f0b89f6bb7737748413e96c7f1aa088cd6cb3 | |
parent | d5826a33a5a7c298a8934541d11cda042028be3b (diff) |
Use tool_output_file in llvm-extract and llvm-link too.
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@111604 91177308-0d34-0410-b5e6-96231b3b80d8
-rw-r--r-- | tools/llvm-extract/llvm-extract.cpp | 12 | ||||
-rw-r--r-- | tools/llvm-link/llvm-link.cpp | 19 |
2 files changed, 13 insertions, 18 deletions
diff --git a/tools/llvm-extract/llvm-extract.cpp b/tools/llvm-extract/llvm-extract.cpp index 440d1bdcb50..190ab2e6b18 100644 --- a/tools/llvm-extract/llvm-extract.cpp +++ b/tools/llvm-extract/llvm-extract.cpp @@ -116,14 +116,9 @@ int main(int argc, char **argv) { Passes.add(createDeadTypeEliminationPass()); // Remove dead types... Passes.add(createStripDeadPrototypesPass()); // Remove dead func decls - // Make sure that the Output file gets unlinked from the disk if we get a - // SIGINT - if (OutputFilename != "-") - sys::RemoveFileOnSignal(sys::Path(OutputFilename)); - std::string ErrorInfo; - raw_fd_ostream Out(OutputFilename.c_str(), ErrorInfo, - raw_fd_ostream::F_Binary); + tool_output_file Out(OutputFilename.c_str(), ErrorInfo, + raw_fd_ostream::F_Binary); if (!ErrorInfo.empty()) { errs() << ErrorInfo << '\n'; return 1; @@ -136,5 +131,8 @@ int main(int argc, char **argv) { Passes.run(*M.get()); + // Declare success. + Out.keep(); + return 0; } diff --git a/tools/llvm-link/llvm-link.cpp b/tools/llvm-link/llvm-link.cpp index f7dad3da5f6..f487d2452df 100644 --- a/tools/llvm-link/llvm-link.cpp +++ b/tools/llvm-link/llvm-link.cpp @@ -116,19 +116,13 @@ int main(int argc, char **argv) { if (DumpAsm) errs() << "Here's the assembly:\n" << *Composite; std::string ErrorInfo; - std::auto_ptr<raw_ostream> - Out(new raw_fd_ostream(OutputFilename.c_str(), ErrorInfo, - raw_fd_ostream::F_Binary)); + tool_output_file Out(OutputFilename.c_str(), ErrorInfo, + raw_fd_ostream::F_Binary); if (!ErrorInfo.empty()) { errs() << ErrorInfo << '\n'; return 1; } - // Make sure that the Out file gets unlinked from the disk if we get a - // SIGINT - if (OutputFilename != "-") - sys::RemoveFileOnSignal(sys::Path(OutputFilename)); - if (verifyModule(*Composite)) { errs() << argv[0] << ": linked module is broken!\n"; return 1; @@ -136,9 +130,12 @@ int main(int argc, char **argv) { if (Verbose) errs() << "Writing bitcode...\n"; if (OutputAssembly) { - *Out << *Composite; - } else if (Force || !CheckBitcodeOutputToConsole(*Out, true)) - WriteBitcodeToFile(Composite.get(), *Out); + Out << *Composite; + } else if (Force || !CheckBitcodeOutputToConsole(Out, true)) + WriteBitcodeToFile(Composite.get(), Out); + + // Declare success. + Out.keep(); return 0; } |