15 #include <util/pragma_push.def>
17 #pragma warning(disable:4668)
19 #pragma warning(disable:5039)
24 #include <util/pragma_pop.def>
36 #include <sys/types.h>
49 int run(
const std::string &what,
const std::vector<std::string> &argv)
51 return run(what, argv,
"",
"",
"");
55 #define STDIN_FILENO 0
56 #define STDOUT_FILENO 1
57 #define STDERR_FILENO 2
67 fdt result_fd = INVALID_HANDLE_VALUE;
70 SECURITY_ATTRIBUTES SecurityAttributes;
71 ZeroMemory(&SecurityAttributes,
sizeof SecurityAttributes);
72 SecurityAttributes.bInheritHandle =
true;
79 result_fd = GetStdHandle(STD_INPUT_HANDLE);
81 result_fd = CreateFileW(
87 FILE_ATTRIBUTE_READONLY,
94 result_fd = GetStdHandle(STD_OUTPUT_HANDLE);
96 result_fd = CreateFileW(
102 FILE_ATTRIBUTE_NORMAL,
109 result_fd = GetStdHandle(STD_ERROR_HANDLE);
111 result_fd = CreateFileW(
117 FILE_ATTRIBUTE_NORMAL,
125 if(result_fd == INVALID_HANDLE_VALUE)
126 perror((
"Failed to open " + name +
" file " +
file).c_str());
133 int flags = 0, mode = 0;
145 flags = O_CREAT | O_WRONLY;
146 mode = S_IRUSR | S_IWUSR;
147 name = fd == STDOUT_FILENO ?
"stdout" :
"stderr";
154 const fdt result_fd = open(
file.c_str(), flags, mode);
157 perror((
"Failed to open " + name +
" file " +
file).c_str());
166 std::wstring quote_windows_arg(
const std::wstring &src)
168 if(src.find_first_of(L
" \t\n\v\"") == src.npos)
171 std::wstring result = L
"\"";
173 for(
auto it = src.begin();; ++it)
175 std::size_t NumberBackslashes = 0;
177 while(it != src.end() && *it == L
'\\')
191 result.append(NumberBackslashes * 2, L
'\\');
201 result.append(NumberBackslashes * 2 + 1, L
'\\');
202 result.push_back(*it);
210 result.append(NumberBackslashes, L
'\\');
211 result.push_back(*it);
215 result.push_back(L
'"');
222 const std::string &what,
223 const std::vector<std::string> &argv,
224 const std::string &std_input,
225 const std::string &std_output,
226 const std::string &std_error)
230 std::wstring cmdline;
233 cmdline = quote_windows_arg(
widen(what));
235 for(std::size_t i = 1; i < argv.size(); i++)
238 cmdline += quote_windows_arg(
widen(argv[i]));
241 PROCESS_INFORMATION piProcInfo;
242 STARTUPINFOW siStartInfo;
244 ZeroMemory(&piProcInfo,
sizeof piProcInfo);
245 ZeroMemory(&siStartInfo,
sizeof siStartInfo);
247 siStartInfo.cb =
sizeof siStartInfo;
253 siStartInfo.dwFlags |= STARTF_USESTDHANDLES;
256 std::vector<wchar_t> mutable_cmdline(cmdline.begin(), cmdline.end());
257 mutable_cmdline.push_back(0);
258 wchar_t *cmdline_ptr = mutable_cmdline.data();
260 BOOL bSuccess = CreateProcessW(
274 if(!std_input.empty())
275 CloseHandle(siStartInfo.hStdInput);
276 if(!std_output.empty())
277 CloseHandle(siStartInfo.hStdOutput);
278 if(!std_error.empty())
279 CloseHandle(siStartInfo.hStdError);
284 WaitForSingleObject(piProcInfo.hProcess, INFINITE);
286 if(!std_input.empty())
287 CloseHandle(siStartInfo.hStdInput);
288 if(!std_output.empty())
289 CloseHandle(siStartInfo.hStdOutput);
290 if(!std_error.empty())
291 CloseHandle(siStartInfo.hStdError);
296 if(!GetExitCodeProcess(piProcInfo.hProcess, &exit_code))
298 CloseHandle(piProcInfo.hProcess);
299 CloseHandle(piProcInfo.hThread);
303 CloseHandle(piProcInfo.hProcess);
304 CloseHandle(piProcInfo.hThread);
313 if(stdin_fd == -1 || stdout_fd == -1 || stderr_fd == -1)
317 sigset_t new_mask, old_mask;
318 sigemptyset(&new_mask);
319 sigprocmask(SIG_SETMASK, &new_mask, &old_mask);
322 pid_t childpid = fork();
330 sigprocmask(SIG_SETMASK, &old_mask,
nullptr);
332 std::vector<char *> _argv(argv.size()+1);
333 for(std::size_t i=0; i<argv.size(); i++)
334 _argv[i]=strdup(argv[i].c_str());
336 _argv[argv.size()]=
nullptr;
338 if(stdin_fd!=STDIN_FILENO)
339 dup2(stdin_fd, STDIN_FILENO);
340 if(stdout_fd!=STDOUT_FILENO)
341 dup2(stdout_fd, STDOUT_FILENO);
342 if(stderr_fd != STDERR_FILENO)
343 dup2(stderr_fd, STDERR_FILENO);
346 execvp(what.c_str(), _argv.data());
349 perror(std::string(
"execvp "+what+
" failed").c_str());
358 sigprocmask(SIG_SETMASK, &old_mask,
nullptr);
363 while(waitpid(childpid, &status, 0)==-1)
371 perror(
"Waiting for child process failed");
372 if(stdin_fd!=STDIN_FILENO)
374 if(stdout_fd!=STDOUT_FILENO)
376 if(stderr_fd != STDERR_FILENO)
384 if(stdin_fd!=STDIN_FILENO)
386 if(stdout_fd!=STDOUT_FILENO)
388 if(stderr_fd != STDERR_FILENO)
391 return WEXITSTATUS(status);
397 sigprocmask(SIG_SETMASK, &old_mask,
nullptr);
399 if(stdin_fd!=STDIN_FILENO)
401 if(stdout_fd!=STDOUT_FILENO)
403 if(stderr_fd != STDERR_FILENO)
417 if(src.find(
' ')==std::string::npos &&
418 src.find(
'"')==std::string::npos &&
419 src.find(
'&')==std::string::npos &&
420 src.find(
'|')==std::string::npos &&
421 src.find(
'(')==std::string::npos &&
422 src.find(
')')==std::string::npos &&
423 src.find(
'<')==std::string::npos &&
424 src.find(
'>')==std::string::npos &&
425 src.find(
'^')==std::string::npos)
435 for(
const char ch : src)
450 if(src.find(
' ')==std::string::npos &&
451 src.find(
'"')==std::string::npos &&
452 src.find(
'*')==std::string::npos &&
453 src.find(
'$')==std::string::npos &&
454 src.find(
'\\')==std::string::npos &&
455 src.find(
'?')==std::string::npos &&
456 src.find(
'&')==std::string::npos &&
457 src.find(
'|')==std::string::npos &&
458 src.find(
'>')==std::string::npos &&
459 src.find(
'<')==std::string::npos &&
460 src.find(
'^')==std::string::npos &&
461 src.find(
'\'')==std::string::npos)
472 for(
const char ch : src)
486 const std::string &what,
487 const std::vector<std::string> &argv,
488 const std::string &std_input,
489 std::ostream &std_output,
490 const std::string &std_error)
495 int result =
run(what, argv, std_input, tmpi(), std_error);
497 std::ifstream instream(tmpi());
500 std_output << instream.rdbuf();
509 for(
const auto &arg : argv)
520 if(!std_input.empty())
523 if(!std_error.empty())
526 FILE *stream=popen(command.c_str(),
"r");
531 while((ch=fgetc(stream))!=EOF)
532 std_output << (
unsigned char)ch;
534 return pclose(stream);