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