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)
169 if(src.find_first_of(L
" \t\n\v\"") == src.npos && !src.empty())
172 std::wstring result = L
"\"";
174 for(
auto it = src.begin();; ++it)
176 std::size_t NumberBackslashes = 0;
178 while(it != src.end() && *it == L
'\\')
192 result.append(NumberBackslashes * 2, L
'\\');
202 result.append(NumberBackslashes * 2 + 1, L
'\\');
203 result.push_back(*it);
211 result.append(NumberBackslashes, L
'\\');
212 result.push_back(*it);
216 result.push_back(L
'"');
223 const std::string &what,
224 const std::vector<std::string> &argv,
225 const std::string &std_input,
226 const std::string &std_output,
227 const std::string &std_error)
231 std::wstring cmdline;
234 cmdline = quote_windows_arg(
widen(what));
236 for(std::size_t i = 1; i < argv.size(); i++)
239 cmdline += quote_windows_arg(
widen(argv[i]));
242 PROCESS_INFORMATION piProcInfo;
243 STARTUPINFOW siStartInfo;
245 ZeroMemory(&piProcInfo,
sizeof piProcInfo);
246 ZeroMemory(&siStartInfo,
sizeof siStartInfo);
248 siStartInfo.cb =
sizeof siStartInfo;
254 siStartInfo.dwFlags |= STARTF_USESTDHANDLES;
257 std::vector<wchar_t> mutable_cmdline(cmdline.begin(), cmdline.end());
258 mutable_cmdline.push_back(0);
259 wchar_t *cmdline_ptr = mutable_cmdline.data();
261 BOOL bSuccess = CreateProcessW(
275 if(!std_input.empty())
276 CloseHandle(siStartInfo.hStdInput);
277 if(!std_output.empty())
278 CloseHandle(siStartInfo.hStdOutput);
279 if(!std_error.empty())
280 CloseHandle(siStartInfo.hStdError);
285 WaitForSingleObject(piProcInfo.hProcess, INFINITE);
287 if(!std_input.empty())
288 CloseHandle(siStartInfo.hStdInput);
289 if(!std_output.empty())
290 CloseHandle(siStartInfo.hStdOutput);
291 if(!std_error.empty())
292 CloseHandle(siStartInfo.hStdError);
297 if(!GetExitCodeProcess(piProcInfo.hProcess, &exit_code))
299 CloseHandle(piProcInfo.hProcess);
300 CloseHandle(piProcInfo.hThread);
304 CloseHandle(piProcInfo.hProcess);
305 CloseHandle(piProcInfo.hThread);
314 if(stdin_fd == -1 || stdout_fd == -1 || stderr_fd == -1)
318 sigset_t new_mask, old_mask;
319 sigemptyset(&new_mask);
320 sigprocmask(SIG_SETMASK, &new_mask, &old_mask);
323 pid_t childpid = fork();
331 sigprocmask(SIG_SETMASK, &old_mask,
nullptr);
333 std::vector<char *> _argv(argv.size()+1);
334 for(std::size_t i=0; i<argv.size(); i++)
335 _argv[i]=strdup(argv[i].c_str());
337 _argv[argv.size()]=
nullptr;
339 if(stdin_fd!=STDIN_FILENO)
340 dup2(stdin_fd, STDIN_FILENO);
341 if(stdout_fd!=STDOUT_FILENO)
342 dup2(stdout_fd, STDOUT_FILENO);
343 if(stderr_fd != STDERR_FILENO)
344 dup2(stderr_fd, STDERR_FILENO);
347 execvp(what.c_str(), _argv.data());
350 perror(std::string(
"execvp "+what+
" failed").c_str());
359 sigprocmask(SIG_SETMASK, &old_mask,
nullptr);
364 while(waitpid(childpid, &status, 0)==-1)
372 perror(
"Waiting for child process failed");
373 if(stdin_fd!=STDIN_FILENO)
375 if(stdout_fd!=STDOUT_FILENO)
377 if(stderr_fd != STDERR_FILENO)
385 if(stdin_fd!=STDIN_FILENO)
387 if(stdout_fd!=STDOUT_FILENO)
389 if(stderr_fd != STDERR_FILENO)
392 return WEXITSTATUS(status);
398 sigprocmask(SIG_SETMASK, &old_mask,
nullptr);
400 if(stdin_fd!=STDIN_FILENO)
402 if(stdout_fd!=STDOUT_FILENO)
404 if(stderr_fd != STDERR_FILENO)
418 if(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 &&
426 src.find(
'^')==std::string::npos)
436 for(
const char ch : src)
451 if(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 &&
462 src.find(
'\'')==std::string::npos)
473 for(
const char ch : src)
487 const std::string &what,
488 const std::vector<std::string> &argv,
489 const std::string &std_input,
490 std::ostream &std_output,
491 const std::string &std_error)
496 int result =
run(what, argv, std_input, tmpi(), std_error);
498 std::ifstream instream(tmpi());
501 std_output << instream.rdbuf();
510 for(
const auto &arg : argv)
521 if(!std_input.empty())
524 if(!std_error.empty())
527 FILE *stream=popen(command.c_str(),
"r");
532 while((ch=fgetc(stream))!=EOF)
533 std_output << (
unsigned char)ch;
535 return pclose(stream);