llvm/lib/ExecutionEngine/Orc/TargetProcess/TargetExecutionUtils.cpp

   1 //===--- TargetExecutionUtils.cpp - Execution utils for target processes --===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "llvm/ExecutionEngine/Orc/TargetProcess/TargetExecutionUtils.h"
  10
  11 #include <vector>
  12
  13 namespace llvm {
  14 namespace orc {
  15
  16 int runAsMain(int (*Main)(int, char *[]), ArrayRef<std::string> Args,
  17               Optional<StringRef> ProgramName) {
  18   std::vector<std::unique_ptr<char[]>> ArgVStorage;
  19   std::vector<char *> ArgV;
  20
  21   ArgVStorage.reserve(Args.size() + (ProgramName ? 1 : 0));
  22   ArgV.reserve(Args.size() + 1 + (ProgramName ? 1 : 0));
  23
  24   if (ProgramName) {
  25     ArgVStorage.push_back(std::make_unique<char[]>(ProgramName->size() + 1));
  26     llvm::copy(*ProgramName, &ArgVStorage.back()[0]);
  27     ArgVStorage.back()[ProgramName->size()] = '\0';
  28     ArgV.push_back(ArgVStorage.back().get());
  29   }
  30
  31   for (const auto &Arg : Args) {
  32     ArgVStorage.push_back(std::make_unique<char[]>(Arg.size() + 1));
  33     llvm::copy(Arg, &ArgVStorage.back()[0]);
  34     ArgVStorage.back()[Arg.size()] = '\0';
  35     ArgV.push_back(ArgVStorage.back().get());
  36   }
  37   ArgV.push_back(nullptr);
  38
  39   return Main(Args.size() + !!ProgramName, ArgV.data());
  40 }
  41
  42 } // End namespace orc.
  43 } // End namespace llvm.