lib/Support/SystemUtils.cpp

   1 //===- SystemUtils.cpp - Utilities for low-level system tasks -------------===//
   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 file contains functions used to do a variety of low-level, often
  11 // system-specific, tasks.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "llvm/Support/SystemUtils.h"
  16 #include "llvm/System/Process.h"
  17 #include "llvm/System/Program.h"
  18 #include "llvm/Support/raw_ostream.h"
  19 using namespace llvm;
  20
  21 bool llvm::CheckBitcodeOutputToConsole(raw_ostream &stream_to_check,
  22                                        bool print_warning) {
  23   if (stream_to_check.is_displayed()) {
  24     if (print_warning) {
  25       errs() << "WARNING: You're attempting to print out a bitcode file.\n"
  26                 "This is inadvisable as it may cause display problems. If\n"
  27                 "you REALLY want to taste LLVM bitcode first-hand, you\n"
  28                 "can force output with the `-f' option.\n\n";
  29     }
  30     return true;
  31   }
  32   return false;
  33 }
  34
  35 /// PrependMainExecutablePath - Prepend the path to the program being executed
  36 /// to \p ExeName, given the value of argv[0] and the address of main()
  37 /// itself. This allows us to find another LLVM tool if it is built in the same
  38 /// directory. An empty string is returned on error; note that this function
  39 /// just mainpulates the path and doesn't check for executability.
  40 /// @brief Find a named executable.
  41 sys::Path llvm::PrependMainExecutablePath(const std::string &ExeName,
  42                                           const char *Argv0, void *MainAddr) {
  43   // Check the directory that the calling program is in.  We can do
  44   // this if ProgramPath contains at least one / character, indicating that it
  45   // is a relative path to the executable itself.
  46   sys::Path Result = sys::Path::GetMainExecutable(Argv0, MainAddr);
  47   Result.eraseComponent();
  48
  49   if (!Result.isEmpty()) {
  50     Result.appendComponent(ExeName);
  51     Result.appendSuffix(sys::Path::GetEXESuffix());
  52   }
  53
  54   return Result;
  55 }