108 lines
3.7 KiB
C++
108 lines
3.7 KiB
C++
//===-- Program.cpp - Implement OS Program Concept --------------*- C++ -*-===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file implements the operating system Program concept.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Support/Program.h"
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include "llvm/Config/llvm-config.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include <system_error>
|
|
using namespace llvm;
|
|
using namespace sys;
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
//=== WARNING: Implementation here must contain only TRULY operating system
|
|
//=== independent code.
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
static bool Execute(ProcessInfo &PI, StringRef Program,
|
|
ArrayRef<StringRef> Args, Optional<ArrayRef<StringRef>> Env,
|
|
ArrayRef<Optional<StringRef>> Redirects,
|
|
unsigned MemoryLimit, std::string *ErrMsg,
|
|
BitVector *AffinityMask);
|
|
|
|
int sys::ExecuteAndWait(StringRef Program, ArrayRef<StringRef> Args,
|
|
Optional<ArrayRef<StringRef>> Env,
|
|
ArrayRef<Optional<StringRef>> Redirects,
|
|
unsigned SecondsToWait, unsigned MemoryLimit,
|
|
std::string *ErrMsg, bool *ExecutionFailed,
|
|
Optional<ProcessStatistics> *ProcStat,
|
|
BitVector *AffinityMask) {
|
|
assert(Redirects.empty() || Redirects.size() == 3);
|
|
ProcessInfo PI;
|
|
if (Execute(PI, Program, Args, Env, Redirects, MemoryLimit, ErrMsg,
|
|
AffinityMask)) {
|
|
if (ExecutionFailed)
|
|
*ExecutionFailed = false;
|
|
ProcessInfo Result =
|
|
Wait(PI, SecondsToWait, /*WaitUntilTerminates=*/SecondsToWait == 0,
|
|
ErrMsg, ProcStat);
|
|
return Result.ReturnCode;
|
|
}
|
|
|
|
if (ExecutionFailed)
|
|
*ExecutionFailed = true;
|
|
|
|
return -1;
|
|
}
|
|
|
|
ProcessInfo sys::ExecuteNoWait(StringRef Program, ArrayRef<StringRef> Args,
|
|
Optional<ArrayRef<StringRef>> Env,
|
|
ArrayRef<Optional<StringRef>> Redirects,
|
|
unsigned MemoryLimit, std::string *ErrMsg,
|
|
bool *ExecutionFailed, BitVector *AffinityMask) {
|
|
assert(Redirects.empty() || Redirects.size() == 3);
|
|
ProcessInfo PI;
|
|
if (ExecutionFailed)
|
|
*ExecutionFailed = false;
|
|
if (!Execute(PI, Program, Args, Env, Redirects, MemoryLimit, ErrMsg,
|
|
AffinityMask))
|
|
if (ExecutionFailed)
|
|
*ExecutionFailed = true;
|
|
|
|
return PI;
|
|
}
|
|
|
|
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);
|
|
}
|
|
|
|
void sys::printArg(raw_ostream &OS, StringRef Arg, bool Quote) {
|
|
const bool Escape = Arg.find_first_of(" \"\\$") != StringRef::npos;
|
|
|
|
if (!Quote && !Escape) {
|
|
OS << Arg;
|
|
return;
|
|
}
|
|
|
|
// Quote and escape. This isn't really complete, but good enough.
|
|
OS << '"';
|
|
for (const auto c : Arg) {
|
|
if (c == '"' || c == '\\' || c == '$')
|
|
OS << '\\';
|
|
OS << c;
|
|
}
|
|
OS << '"';
|
|
}
|
|
|
|
// Include the platform-specific parts of this class.
|
|
#ifdef LLVM_ON_UNIX
|
|
#include "Unix/Program.inc"
|
|
#endif
|
|
#ifdef _WIN32
|
|
#include "Windows/Program.inc"
|
|
#endif
|