Line data    Source code

       1             : //===-- Program.cpp - Implement OS Program Concept --------------*- C++ -*-===//
       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 implements the operating system Program concept.
      11             : //
      12             : //===----------------------------------------------------------------------===//
      13             : 
      14             : #include "llvm/Support/Program.h"
      15             : #include "llvm/ADT/StringRef.h"
      16             : #include "llvm/Config/llvm-config.h"
      17             : #include <system_error>
      18             : using namespace llvm;
      19             : using namespace sys;
      20             : 
      21             : //===----------------------------------------------------------------------===//
      22             : //=== WARNING: Implementation here must contain only TRULY operating system
      23             : //===          independent code.
      24             : //===----------------------------------------------------------------------===//
      25             : 
      26             : static bool Execute(ProcessInfo &PI, StringRef Program,
      27             :                     ArrayRef<StringRef> Args, Optional<ArrayRef<StringRef>> Env,
      28             :                     ArrayRef<Optional<StringRef>> Redirects,
      29             :                     unsigned MemoryLimit, std::string *ErrMsg);
      30             : 
      31       19303 : int sys::ExecuteAndWait(StringRef Program, ArrayRef<StringRef> Args,
      32             :                         Optional<ArrayRef<StringRef>> Env,
      33             :                         ArrayRef<Optional<StringRef>> Redirects,
      34             :                         unsigned SecondsToWait, unsigned MemoryLimit,
      35             :                         std::string *ErrMsg, bool *ExecutionFailed) {
      36             :   assert(Redirects.empty() || Redirects.size() == 3);
      37       19303 :   ProcessInfo PI;
      38       19303 :   if (Execute(PI, Program, Args, Env, Redirects, MemoryLimit, ErrMsg)) {
      39       19301 :     if (ExecutionFailed)
      40       13534 :       *ExecutionFailed = false;
      41             :     ProcessInfo Result = Wait(
      42       19301 :         PI, SecondsToWait, /*WaitUntilTerminates=*/SecondsToWait == 0, ErrMsg);
      43       19301 :     return Result.ReturnCode;
      44             :   }
      45             : 
      46           2 :   if (ExecutionFailed)
      47           2 :     *ExecutionFailed = true;
      48             : 
      49             :   return -1;
      50             : }
      51             : 
      52           3 : ProcessInfo sys::ExecuteNoWait(StringRef Program, ArrayRef<StringRef> Args,
      53             :                                Optional<ArrayRef<StringRef>> Env,
      54             :                                ArrayRef<Optional<StringRef>> Redirects,
      55             :                                unsigned MemoryLimit, std::string *ErrMsg,
      56             :                                bool *ExecutionFailed) {
      57             :   assert(Redirects.empty() || Redirects.size() == 3);
      58           3 :   ProcessInfo PI;
      59           3 :   if (ExecutionFailed)
      60           3 :     *ExecutionFailed = false;
      61           3 :   if (!Execute(PI, Program, Args, Env, Redirects, MemoryLimit, ErrMsg))
      62           1 :     if (ExecutionFailed)
      63           1 :       *ExecutionFailed = true;
      64             : 
      65           3 :   return PI;
      66             : }
      67             : 
      68       13562 : bool sys::commandLineFitsWithinSystemLimits(StringRef Program,
      69             :                                             ArrayRef<const char *> Args) {
      70             :   SmallVector<StringRef, 8> StringRefArgs;
      71             :   StringRefArgs.reserve(Args.size());
      72     1546574 :   for (const char *A : Args)
      73     1533012 :     StringRefArgs.emplace_back(A);
      74       13562 :   return commandLineFitsWithinSystemLimits(Program, StringRefArgs);
      75             : }
      76             : 
      77             : // Include the platform-specific parts of this class.
      78             : #ifdef LLVM_ON_UNIX
      79             : #include "Unix/Program.inc"
      80             : #endif
      81             : #ifdef _WIN32
      82             : #include "Windows/Program.inc"
      83             : #endif