diff options
author | Dan Gohman <gohman@apple.com> | 2009-07-15 17:29:42 +0000 |
---|---|---|
committer | Dan Gohman <gohman@apple.com> | 2009-07-15 17:29:42 +0000 |
commit | a1bdcedc3879510a874d24c450e07feb170d9cd6 (patch) | |
tree | 87232c6e8e70de1345065d6e50540c4577f68dc3 /tools/opt | |
parent | 2286f8dc4cec0625f7d7a14e2570926cf8599646 (diff) | |
download | external_llvm-a1bdcedc3879510a874d24c450e07feb170d9cd6.zip external_llvm-a1bdcedc3879510a874d24c450e07feb170d9cd6.tar.gz external_llvm-a1bdcedc3879510a874d24c450e07feb170d9cd6.tar.bz2 |
Add a Force option to raw_fd_ostream to specify whether opening
an existing file is considered an error. Convert several tools
to use raw_fd_ostream instead of std::ostream, and to use this
new option instead of doing a manual check.
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@75801 91177308-0d34-0410-b5e6-96231b3b80d8
Diffstat (limited to 'tools/opt')
-rw-r--r-- | tools/opt/opt.cpp | 29 |
1 files changed, 11 insertions, 18 deletions
diff --git a/tools/opt/opt.cpp b/tools/opt/opt.cpp index 6891619..c92fad8 100644 --- a/tools/opt/opt.cpp +++ b/tools/opt/opt.cpp @@ -35,8 +35,6 @@ #include "llvm/Support/raw_ostream.h" #include "llvm/LinkAllPasses.h" #include "llvm/LinkAllVMCore.h" -#include <iostream> -#include <fstream> #include <memory> #include <algorithm> using namespace llvm; @@ -342,21 +340,16 @@ int main(int argc, char **argv) { // Figure out what stream we are supposed to write to... // FIXME: cout is not binary! - std::ostream *Out = &std::cout; // Default to printing to stdout... + raw_ostream *Out = &outs(); // Default to printing to stdout... if (OutputFilename != "-") { - if (!Force && std::ifstream(OutputFilename.c_str())) { - // If force is not specified, make sure not to overwrite a file! - cerr << argv[0] << ": error opening '" << OutputFilename - << "': file exists!\n" - << "Use -f command line argument to force output\n"; - return 1; - } - std::ios::openmode io_mode = std::ios::out | std::ios::trunc | - std::ios::binary; - Out = new std::ofstream(OutputFilename.c_str(), io_mode); - - if (!Out->good()) { - cerr << argv[0] << ": error opening " << OutputFilename << "!\n"; + std::string ErrorInfo; + Out = new raw_fd_ostream(OutputFilename.c_str(), /*Binary=*/true, + Force, ErrorInfo); + if (!ErrorInfo.empty()) { + errs() << ErrorInfo << '\n'; + if (!Force) + errs() << "Use -f command line argument to force output\n"; + delete Out; return 1; } @@ -479,13 +472,13 @@ int main(int argc, char **argv) { // Write bitcode out to disk or cout as the last step... if (!NoOutput && !AnalyzeOnly) - Passes.add(CreateBitcodeWriterPass(*Out)); + Passes.add(createBitcodeWriterPass(*Out)); // Now that we have all of the passes ready, run them. Passes.run(*M.get()); // Delete the ofstream. - if (Out != &std::cout) + if (Out != &outs()) delete Out; return 0; |