1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
// -*- c++ -*-
// Distributed under the BSD 2-Clause License.
// See accompanying file LICENSE for details.
#include "execute.h"
#include <cstring>
#if !defined(_WIN32)
#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <spawn.h>
#else
#include <processthreadsapi.h>
#include <synchapi.h>
#endif
#include <iostream>
/*
https://blog.famzah.net/2009/11/20/a-much-faster-popen-and-system-implementation-for-linux/
https://github.com/famzah/popen-noshell/commit/1f9eaf4eeef348d1efe0f3c7fe8ab670653cfbb1
https://blog.famzah.net/2018/12/19/posix_spawn-performance-benchmarks-and-usage-examples/
https://stackoverflow.com/questions/4259629/what-is-the-difference-between-fork-and-vfork/5207945#5207945
*/
#if !defined(_WIN32)
namespace
{
int parent_waitpid(pid_t pid)
{
int status;
if(waitpid(pid, &status, 0) != pid)
{
return 1;
}
return WEXITSTATUS(status);
}
} // namespace ::
#endif
int execute(const std::string& command,
const std::vector<std::string>& args,
bool verbose)
{
std::vector<const char*> argv;
argv.push_back(command.data());
for(const auto& arg : args)
{
argv.push_back(arg.data());
}
argv.push_back(nullptr);
std::string cmd;
if(verbose)
{
for(const auto& arg : argv)
{
if(arg == nullptr)
{
break;
}
if(!cmd.empty())
{
cmd += " ";
}
cmd += arg;
}
std::cout << cmd << "\n";
}
#if !defined(_WIN32)
#if 1
auto pid = vfork();
if(pid == 0)
{
// TODO: use execvpe to set LD_LIBRARY_PATH
execv(command.data(), (char**)argv.data());
std::cout << "Could not execute " << command << ": " <<
strerror(errno) << "\n";
_exit(1); // execv only returns if an error occurred
}
return parent_waitpid(pid);
#elif 0
pid_t pid;
const char * envp[] = {nullptr}; // TODO: use this to set LD_LIBRARY_PATH
if(posix_spawn(&pid, command.data(), nullptr, nullptr,
(char**)argv.data(), envp))
{
return 1;
}
return parent_waitpid(pid);
#else
return system(cmd.data());
#endif
#else
STARTUPINFO si;
PROCESS_INFORMATION pi;
ZeroMemory(&si,sizeof(si));
si.cb=sizeof(si);
ZeroMemory(&pi,sizeof(pi));
// TODO: Use SetDllDirectory(...) to set DLL search directory
if(!CreateProcess(nullptr, //"C:/WINDOWS/notepad.exe",
//"notepad.exe c:/readme.txt",0,0,0,0,0,0,&si,&pi))
(char*)cmd.data(),0,0,0,0,0,0,&si,&pi))
{
return 1; // Could not start process;
}
// Now 'pi.hProcess' contains the process HANDLE, which you can use to
// wait for it like this:
const DWORD infinite = 0xFFFFFFFF;
WaitForSingleObject(pi.hProcess, infinite);
DWORD exitCode{0};
GetExitCodeProcess(pi.hProcess, &exitCode);
return exitCode;
#endif
}
|
