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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
|
#include "cru/common/platform/unix/PosixSpawnSubProcess.h"
#include "cru/common/Exception.h"
#include "cru/common/Format.h"
#include "cru/common/Guard.h"
#include "cru/common/String.h"
#include "cru/common/SubProcess.h"
#include "cru/common/log/Logger.h"
#include <signal.h>
#include <spawn.h>
#include <sys/wait.h>
#include <unistd.h>
#include <memory>
#include <unordered_map>
namespace cru::platform::unix {
PosixSpawnSubProcess::PosixSpawnSubProcess(SubProcessStartInfo start_info)
: PlatformSubProcessBase(std::move(start_info)),
pid_(0),
exit_code_(0),
stdin_pipe_(UnixPipe::Usage::Send),
stdout_pipe_(UnixPipe::Usage::Receive),
stderr_pipe_(UnixPipe::Usage::Receive) {
stdin_stream_ = std::make_unique<UnixFileStream>(
stdin_pipe_.GetSelfFileDescriptor(), false, false, true, false);
stdout_stream_ = std::make_unique<UnixFileStream>(
stdout_pipe_.GetSelfFileDescriptor(), false, true, false, false);
stderr_stream_ = std::make_unique<UnixFileStream>(
stderr_pipe_.GetSelfFileDescriptor(), false, true, false, false);
stdout_buffer_stream_ =
std::make_unique<io::AutoReadStream>(stdout_stream_.get(), false);
stderr_buffer_stream_ =
std::make_unique<io::AutoReadStream>(stdout_stream_.get(), false);
}
PosixSpawnSubProcess::~PosixSpawnSubProcess() {}
io::Stream* PosixSpawnSubProcess::GetStdinStream() {
return stdin_stream_.get();
}
io::Stream* PosixSpawnSubProcess::GetStdoutStream() {
return stdout_buffer_stream_.get();
}
io::Stream* PosixSpawnSubProcess::GetStderrStream() {
return stderr_buffer_stream_.get();
}
namespace {
char** CreateCstrArray(const std::vector<String>& argv) {
std::vector<Buffer> utf8_argv;
for (const auto& arg : argv) {
utf8_argv.push_back(arg.ToUtf8Buffer());
}
char** result = new char*[utf8_argv.size() + 1];
for (int i = 0; i < utf8_argv.size(); i++) {
result[i] = reinterpret_cast<char*>(utf8_argv[i].Detach());
}
result[utf8_argv.size()] = nullptr;
return result;
}
char** CreateCstrArray(const std::unordered_map<String, String>& envp) {
std::vector<String> str_array;
for (auto& [key, value] : envp) {
str_array.push_back(cru::Format(u"{}={}", key, value));
}
return CreateCstrArray(str_array);
}
void DestroyCstrArray(char** argv) {
int index = 0;
char* arg = argv[index];
while (arg) {
delete[] arg;
index++;
arg = argv[index];
}
delete[] argv;
}
} // namespace
void PosixSpawnSubProcess::PlatformCreateProcess() {
int error;
auto check_error = [&error](String message) {
if (error == 0) return;
std::unique_ptr<ErrnoException> inner(new ErrnoException({}, error));
throw SubProcessFailedToStartException(std::move(message),
std::move(inner));
};
posix_spawn_file_actions_t file_actions;
error = posix_spawn_file_actions_init(&file_actions);
check_error(u"Failed to call posix_spawn_file_actions_init.");
Guard file_actions_guard(
[&file_actions] { posix_spawn_file_actions_destroy(&file_actions); });
// dup2 stdin/stdout/stderr
error = posix_spawn_file_actions_adddup2(
&file_actions, stdin_pipe_.GetOtherFileDescriptor(), STDIN_FILENO);
check_error(u"Failed to call posix_spawn_file_actions_adddup2 on stdin.");
error = posix_spawn_file_actions_adddup2(
&file_actions, stdout_pipe_.GetOtherFileDescriptor(), STDOUT_FILENO);
check_error(u"Failed to call posix_spawn_file_actions_adddup2 on stdout.");
error = posix_spawn_file_actions_adddup2(
&file_actions, stderr_pipe_.GetOtherFileDescriptor(), STDERR_FILENO);
check_error(u"Failed to call posix_spawn_file_actions_adddup2 on stderr.");
posix_spawnattr_t attr;
error = posix_spawnattr_init(&attr);
check_error(u"Failed to call posix_spawnattr_init.");
Guard attr_guard([&attr] { posix_spawnattr_destroy(&attr); });
#ifdef CRU_PLATFORM_OSX
error = posix_spawnattr_setflags(&attr, POSIX_SPAWN_CLOEXEC_DEFAULT);
check_error(u"Failed to set flag POSIX_SPAWN_CLOEXEC_DEFAULT (osx).");
#endif
auto exe = start_info_.program.ToUtf8();
auto argv = CreateCstrArray(start_info_.arguments);
Guard argv_guard([argv] { DestroyCstrArray(argv); });
auto envp = CreateCstrArray(start_info_.environments);
Guard envp_guard([envp] { DestroyCstrArray(envp); });
error = posix_spawnp(&pid_, exe.c_str(), &file_actions, &attr, argv, envp);
check_error(u"Failed to call posix_spawnp.");
}
SubProcessExitResult PosixSpawnSubProcess::PlatformWaitForProcess() {
int wstatus;
while (waitpid(pid_, &wstatus, 0) == -1) {
if (errno == EINTR) {
CRU_LOG_INFO(u"Waitpid is interrupted by a signal. Call it again.");
continue;
}
std::unique_ptr<ErrnoException> inner(new ErrnoException({}, errno));
throw SubProcessInternalException(
u"Failed to call waitpid on a subprocess.", std::move(inner));
}
if (WIFEXITED(wstatus)) {
return SubProcessExitResult::Normal(WEXITSTATUS(wstatus));
} else if (WIFEXITED(wstatus)) {
return SubProcessExitResult::Signal(WTERMSIG(wstatus), WCOREDUMP(wstatus));
} else {
return SubProcessExitResult::Unknown();
}
}
void PosixSpawnSubProcess::PlatformKillProcess() {
int error = kill(pid_, SIGKILL);
if (error != 0) {
std::unique_ptr<ErrnoException> inner(new ErrnoException({}, errno));
throw SubProcessInternalException(u"Failed to call kill on a subprocess.",
std::move(inner));
}
}
} // namespace cru::platform::unix
|
