diff options
Diffstat (limited to 'Build/source/texk/dvisvgm/dvisvgm-1.4/src/Process.cpp')
| -rw-r--r-- | Build/source/texk/dvisvgm/dvisvgm-1.4/src/Process.cpp | 188 |
1 files changed, 188 insertions, 0 deletions
diff --git a/Build/source/texk/dvisvgm/dvisvgm-1.4/src/Process.cpp b/Build/source/texk/dvisvgm/dvisvgm-1.4/src/Process.cpp new file mode 100644 index 00000000000..c9f38b004d0 --- /dev/null +++ b/Build/source/texk/dvisvgm/dvisvgm-1.4/src/Process.cpp @@ -0,0 +1,188 @@ +/************************************************************************* +** Process.cpp ** +** ** +** This file is part of dvisvgm -- the DVI to SVG converter ** +** Copyright (C) 2005-2013 Martin Gieseking <martin.gieseking@uos.de> ** +** ** +** This program is free software; you can redistribute it and/or ** +** modify it under the terms of the GNU General Public License as ** +** published by the Free Software Foundation; either version 3 of ** +** the License, or (at your option) any later version. ** +** ** +** This program is distributed in the hope that it will be useful, but ** +** WITHOUT ANY WARRANTY; without even the implied warranty of ** +** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the ** +** GNU General Public License for more details. ** +** ** +** You should have received a copy of the GNU General Public License ** +** along with this program; if not, see <http://www.gnu.org/licenses/>. ** +*************************************************************************/ + +#ifdef __WIN32__ + #include <windows.h> +#else + #include <fcntl.h> + #include <sys/wait.h> + #include <unistd.h> + #include <signal.h> +#endif + +#include <cstdlib> +#include "FileSystem.h" +#include "Process.h" +#include "SignalHandler.h" +#include "macros.h" + +using namespace std; + +Process::Process (const string &cmd, const string ¶mstr) + : _cmd(cmd), _paramstr(paramstr) +{ +} + + +#ifdef __WIN32__ +static void pipe_read (HANDLE handle, string &out) { + char buf[4096]; + out.clear(); + for (;;) { + DWORD num_chars; + bool success = ReadFile(handle, buf, 1024, &num_chars, NULL); + if (!success || num_chars == 0) + break; + out.append(buf, num_chars); + } + // remove trailing space + if (!out.empty()) { + int pos = out.size()-1; + while (pos >= 0 && isspace(out[pos])) + pos--; + out.erase(pos+1); + } +} + + +static inline void close_handle (HANDLE handle) { + if (handle != NULL) + CloseHandle(handle); +} + +#else + +/** Extracts whitespace-sparated parameters from a string. + * @param[in] paramstr the parameter string + * @param[out] params vector holding the extracted parameters */ +static void split_paramstr (string paramstr, vector<const char*> ¶ms) { + size_t left=0, right=0; // index of first and last character of current parameter + char quote=0; // current quote character, 0=none + const size_t len = paramstr.length(); + while (left <= right && right < len) { + while (left < len && isspace(paramstr[left])) + ++left; + if (left < len && (paramstr[left] == '"' || paramstr[left] == '\'')) + quote = paramstr[left++]; + right = left; + while (right < len && (quote || !isspace(paramstr[right]))) { + if (quote && paramstr[right] == quote) { + quote=0; + break; + } + else + ++right; + } + if (right < len) + paramstr[right]=0; + if (left < len) + params.push_back(¶mstr[left]); + left = ++right; + } +} + +#endif + + +/** Runs the process and waits until it's finished. + * @param[out] out takes the output written to stdout by the executed process + * @return true if process terminated properly + * @throw SignalException if CTRL-C was pressed during execution */ +bool Process::run (string *out) { +#ifdef __WIN32__ + SECURITY_ATTRIBUTES sa; + ZeroMemory(&sa, sizeof(SECURITY_ATTRIBUTES)); + sa.nLength = sizeof(SECURITY_ATTRIBUTES); + sa.bInheritHandle = true; + + STARTUPINFO si; + ZeroMemory(&si, sizeof(STARTUPINFO)); + si.cb = sizeof(STARTUPINFO); + si.dwFlags = STARTF_USESTDHANDLES; + HANDLE devnull = NULL; + HANDLE proc_read_handle = NULL; + HANDLE proc_write_handle = NULL; + if (out) { + CreatePipe(&proc_read_handle, &proc_write_handle, &sa, 0); + SetHandleInformation(proc_read_handle, HANDLE_FLAG_INHERIT, 0); + si.hStdOutput = proc_write_handle; + } + else { + devnull = CreateFile("nul", GENERIC_READ|GENERIC_WRITE, FILE_SHARE_READ|FILE_SHARE_WRITE, &sa, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL); + si.hStdOutput = devnull; + } + si.hStdInput = GetStdHandle(STD_INPUT_HANDLE); + si.hStdError = GetStdHandle(STD_ERROR_HANDLE); + + PROCESS_INFORMATION pi; + ZeroMemory(&pi, sizeof(pi)); + DWORD exitcode = DWORD(-1); + string cmdline = _cmd+" "+_paramstr; + bool success = CreateProcess(NULL, (LPSTR)cmdline.c_str(), NULL, NULL, true, 0, NULL, NULL, &si, &pi); + if (success) { + WaitForSingleObject(pi.hProcess, INFINITE); + GetExitCodeProcess(pi.hProcess, &exitcode); + CloseHandle(pi.hProcess); + CloseHandle(pi.hThread); + } + close_handle(proc_write_handle); // must be closed before reading from pipe to prevent blocking + if (success && out) + pipe_read(proc_read_handle, *out); + close_handle(proc_read_handle); + close_handle(devnull); + return exitcode == 0; +#else + pid_t pid = fork(); + if (pid == 0) { // child process + if (!out) { + int devnull = open(FileSystem::DEVNULL, O_WRONLY); + if (devnull >= 0) { + dup2(devnull, STDOUT_FILENO); + dup2(devnull, STDERR_FILENO); + close(devnull); + } + } + vector<const char*> params; + params.push_back(_cmd.c_str()); + split_paramstr(_paramstr, params); + params.push_back(0); // trailing NULL marks end + execvp(_cmd.c_str(), const_cast<char* const*>(¶ms[0])); + exit(1); + } + if (pid > 0) { // main process + int status; + for (;;) { + waitpid(pid, &status, WNOHANG); + if (WIFEXITED(status)) // child process exited normally + return WEXITSTATUS(status) == 0; + + try { + SignalHandler::instance().check(); + } + catch (SignalException &e) { // caught ctrl-c + kill(pid, SIGKILL); + throw; + } + } + } + return false; +#endif +} + |