2004-08-29 21:20:41 +02:00
|
|
|
//===-- Program.cpp - Implement OS Program Concept --------------*- C++ -*-===//
|
2005-04-22 00:55:34 +02:00
|
|
|
//
|
2019-01-19 09:50:56 +01:00
|
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
2005-04-22 00:55:34 +02:00
|
|
|
//
|
2004-08-29 21:20:41 +02:00
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
//
|
2014-06-20 03:36:00 +02:00
|
|
|
// This file implements the operating system Program concept.
|
2004-08-29 21:20:41 +02:00
|
|
|
//
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
|
2010-11-29 19:16:10 +01:00
|
|
|
#include "llvm/Support/Program.h"
|
2015-03-23 19:07:13 +01:00
|
|
|
#include "llvm/ADT/StringRef.h"
|
2018-04-30 16:59:11 +02:00
|
|
|
#include "llvm/Config/llvm-config.h"
|
2014-06-12 19:38:55 +02:00
|
|
|
#include <system_error>
|
2010-04-18 05:33:55 +02:00
|
|
|
using namespace llvm;
|
2004-08-29 21:20:41 +02:00
|
|
|
using namespace sys;
|
|
|
|
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
//=== WARNING: Implementation here must contain only TRULY operating system
|
2005-04-22 00:55:34 +02:00
|
|
|
//=== independent code.
|
2004-08-29 21:20:41 +02:00
|
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
|
2018-06-12 19:43:52 +02:00
|
|
|
static bool Execute(ProcessInfo &PI, StringRef Program,
|
|
|
|
ArrayRef<StringRef> Args, Optional<ArrayRef<StringRef>> Env,
|
|
|
|
ArrayRef<Optional<StringRef>> Redirects,
|
2017-09-06 18:28:33 +02:00
|
|
|
unsigned MemoryLimit, std::string *ErrMsg);
|
2013-06-13 22:06:28 +02:00
|
|
|
|
2018-06-12 19:43:52 +02:00
|
|
|
int sys::ExecuteAndWait(StringRef Program, ArrayRef<StringRef> Args,
|
|
|
|
Optional<ArrayRef<StringRef>> Env,
|
2017-09-13 19:03:37 +02:00
|
|
|
ArrayRef<Optional<StringRef>> Redirects,
|
|
|
|
unsigned SecondsToWait, unsigned MemoryLimit,
|
|
|
|
std::string *ErrMsg, bool *ExecutionFailed) {
|
|
|
|
assert(Redirects.empty() || Redirects.size() == 3);
|
2013-10-01 16:28:18 +02:00
|
|
|
ProcessInfo PI;
|
2018-06-12 19:43:52 +02:00
|
|
|
if (Execute(PI, Program, Args, Env, Redirects, MemoryLimit, ErrMsg)) {
|
2013-10-01 16:28:18 +02:00
|
|
|
if (ExecutionFailed)
|
|
|
|
*ExecutionFailed = false;
|
2014-05-31 03:36:02 +02:00
|
|
|
ProcessInfo Result = Wait(
|
2017-09-06 18:28:33 +02:00
|
|
|
PI, SecondsToWait, /*WaitUntilTerminates=*/SecondsToWait == 0, ErrMsg);
|
2013-10-01 16:28:18 +02:00
|
|
|
return Result.ReturnCode;
|
2013-03-27 00:35:00 +01:00
|
|
|
}
|
2013-10-01 16:28:18 +02:00
|
|
|
|
|
|
|
if (ExecutionFailed)
|
|
|
|
*ExecutionFailed = true;
|
|
|
|
|
2013-03-27 00:35:00 +01:00
|
|
|
return -1;
|
2009-07-18 23:43:12 +02:00
|
|
|
}
|
|
|
|
|
2018-06-12 19:43:52 +02:00
|
|
|
ProcessInfo sys::ExecuteNoWait(StringRef Program, ArrayRef<StringRef> Args,
|
|
|
|
Optional<ArrayRef<StringRef>> Env,
|
2017-09-13 19:03:37 +02:00
|
|
|
ArrayRef<Optional<StringRef>> Redirects,
|
2017-09-06 18:28:33 +02:00
|
|
|
unsigned MemoryLimit, std::string *ErrMsg,
|
2013-10-01 16:28:18 +02:00
|
|
|
bool *ExecutionFailed) {
|
2017-09-13 19:03:37 +02:00
|
|
|
assert(Redirects.empty() || Redirects.size() == 3);
|
2013-10-01 16:28:18 +02:00
|
|
|
ProcessInfo PI;
|
|
|
|
if (ExecutionFailed)
|
|
|
|
*ExecutionFailed = false;
|
2018-06-12 19:43:52 +02:00
|
|
|
if (!Execute(PI, Program, Args, Env, Redirects, MemoryLimit, ErrMsg))
|
2013-10-01 16:28:18 +02:00
|
|
|
if (ExecutionFailed)
|
|
|
|
*ExecutionFailed = true;
|
|
|
|
|
|
|
|
return PI;
|
2009-07-18 23:43:12 +02:00
|
|
|
}
|
|
|
|
|
2018-06-10 22:57:14 +02:00
|
|
|
bool sys::commandLineFitsWithinSystemLimits(StringRef Program,
|
|
|
|
ArrayRef<const char *> Args) {
|
|
|
|
SmallVector<StringRef, 8> StringRefArgs;
|
|
|
|
StringRefArgs.reserve(Args.size());
|
|
|
|
for (const char *A : Args)
|
|
|
|
StringRefArgs.emplace_back(A);
|
|
|
|
return commandLineFitsWithinSystemLimits(Program, StringRefArgs);
|
|
|
|
}
|
|
|
|
|
2004-08-29 21:20:41 +02:00
|
|
|
// Include the platform-specific parts of this class.
|
2004-12-27 07:16:25 +01:00
|
|
|
#ifdef LLVM_ON_UNIX
|
2005-01-10 00:29:00 +01:00
|
|
|
#include "Unix/Program.inc"
|
2004-12-27 07:16:25 +01:00
|
|
|
#endif
|
2018-04-29 02:45:03 +02:00
|
|
|
#ifdef _WIN32
|
2010-11-29 19:16:10 +01:00
|
|
|
#include "Windows/Program.inc"
|
2004-12-27 07:16:25 +01:00
|
|
|
#endif
|