From d890de10af6a4d01fe0b35f09ccd674d771e4732 Mon Sep 17 00:00:00 2001 From: Dan Gohman Date: Wed, 15 Jul 2009 17:04:50 +0000 Subject: [PATCH] Add a raw_ostream version of CheckBitcodeOutputToConsole. git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@75796 91177308-0d34-0410-b5e6-96231b3b80d8 --- include/llvm/Support/SystemUtils.h | 10 ++++++++++ lib/Support/SystemUtils.cpp | 15 +++++++++++++++ 2 files changed, 25 insertions(+) diff --git a/include/llvm/Support/SystemUtils.h b/include/llvm/Support/SystemUtils.h index 9a33fa31f22..95124a047da 100644 --- a/include/llvm/Support/SystemUtils.h +++ b/include/llvm/Support/SystemUtils.h @@ -29,6 +29,16 @@ bool CheckBitcodeOutputToConsole( bool print_warning = true ///< Control whether warnings are printed ); +/// Determine if the raw_ostream provided is connected to the outs() and +/// displayed or not (to a console window). If so, generate a warning message +/// advising against display of bitcode and return true. Otherwise just return +/// false +/// @brief Check for output written to a console +bool CheckBitcodeOutputToConsole( + raw_ostream* stream_to_check, ///< The stream to be checked + bool print_warning = true ///< Control whether warnings are printed +); + /// FindExecutable - Find a named executable, giving the argv[0] of program /// being executed. This allows us to find another LLVM tool if it is built into /// the same directory, but that directory is neither the current directory, nor diff --git a/lib/Support/SystemUtils.cpp b/lib/Support/SystemUtils.cpp index c86e200fd95..49a87e405ac 100644 --- a/lib/Support/SystemUtils.cpp +++ b/lib/Support/SystemUtils.cpp @@ -19,6 +19,21 @@ #include using namespace llvm; +bool llvm::CheckBitcodeOutputToConsole(raw_ostream* stream_to_check, + bool print_warning) { + if (stream_to_check == &outs() && + sys::Process::StandardOutIsDisplayed()) { + if (print_warning) { + cerr << "WARNING: You're attempting to print out a bitcode file.\n" + << "This is inadvisable as it may cause display problems. If\n" + << "you REALLY want to taste LLVM bitcode first-hand, you\n" + << "can force output with the `-f' option.\n\n"; + } + return true; + } + return false; +} + bool llvm::CheckBitcodeOutputToConsole(std::ostream* stream_to_check, bool print_warning) { if (stream_to_check == cout.stream() && -- 2.34.1