]> git.lyx.org Git - lyx.git/blobdiff - src/support/forkedcall.C
hopefully fix tex2lyx linking.
[lyx.git] / src / support / forkedcall.C
index 2e809bdcaa849a235bb9f70dcbd3061bc71c093b..e1d5ef50cfd04655cc54f83479cc4438c06fda5c 100644 (file)
@@ -1,5 +1,5 @@
 /**
- *  \file forkedcall.C
+ * \file forkedcall.C
  * This file is part of LyX, the document processor.
  * Licence details can be found in the file COPYING.
  *
@@ -8,7 +8,7 @@
  * Interface cleaned up by
  * \author Angus Leeming
  *
- * Full author contact details are available in file CREDITS
+ * Full author contact details are available in file CREDITS.
  *
  * An instance of Class Forkedcall represents a single child process.
  *
 
 #include <config.h>
 
-#ifdef __GNUG__
-#pragma implementation
-#endif
+#include "support/forkedcall.h"
+#include "support/forkedcontr.h"
+#include "support/lstrings.h"
+#include "support/lyxlib.h"
+#include "support/filetools.h"
+#include "support/os.h"
 
-#include "forkedcall.h"
-#include "forkedcontr.h"
-#include "lstrings.h"
-#include "lyxlib.h"
-#include "filetools.h"
-#include "os.h"
 #include "debug.h"
+
 #include "frontends/Timeout.h"
 
 #include <boost/bind.hpp>
 
+#include <vector>
 #include <cerrno>
-#include <sys/types.h>
-#include <sys/wait.h>
-#include <csignal>
-#include <cstdlib>
-#include <unistd.h>
+
+#ifdef _WIN32
+# define SIGHUP 1
+# define SIGKILL 9
+# include <process.h>
+# include <windows.h>
+
+#else
+# include <csignal>
+# include <cstdlib>
+# ifdef HAVE_UNISTD_H
+#  include <unistd.h>
+# endif
+# include <sys/wait.h>
+#endif
 
 using std::endl;
+using std::string;
+using std::vector;
 
 #ifndef CXX_GLOBAL_CSTD
 using std::strerror;
 #endif
 
+namespace lyx {
+namespace support {
+
 
 namespace {
 
@@ -63,7 +77,7 @@ public:
                if (secs > 0) {
                        new Murder(secs, pid);
                } else if (pid != 0) {
-                       lyx::kill(pid, SIGKILL);
+                       lyx::support::kill(pid, SIGKILL);
                }
        }
 
@@ -71,7 +85,7 @@ public:
        void kill()
        {
                if (pid_ != 0) {
-                       lyx::kill(pid_, SIGKILL);
+                       lyx::support::kill(pid_, SIGKILL);
                }
                lyxerr << "Killed " << pid_ << std::endl;
                delete this;
@@ -114,8 +128,8 @@ void ForkedProcess::emitSignal()
 }
 
 
-// Wait for child process to finish.
-int ForkedProcess::runBlocking() 
+// Spawn the child process
+int ForkedProcess::run(Starttype type)
 {
        retval_  = 0;
        pid_ = generateChild();
@@ -124,44 +138,58 @@ int ForkedProcess::runBlocking()
                return retval_;
        }
 
-       retval_ = waitForChild();
+       switch (type) {
+       case Wait:
+               retval_ = waitForChild();
+               break;
+       case DontWait: {
+               // Integrate into the Controller
+               ForkedcallsController & contr = ForkedcallsController::get();
+               contr.addCall(*this);
+               break;
+       }
+       }
+
        return retval_;
 }
 
 
-// Do not wait for child process to finish.
-int ForkedProcess::runNonBlocking()
+bool ForkedProcess::running() const
 {
-       retval_ = 0;
-       pid_ = generateChild();
-       if (pid_ <= 0) { // child or fork failed.
-               retval_ = 1;
-               return retval_;
-       }
-
-       // Non-blocking execution.
-       // Integrate into the Controller
-       ForkedcallsController & contr = ForkedcallsController::get();
-       contr.addCall(*this);
+       if (!pid())
+               return false;
+
+#if !defined (_WIN32)
+       // Un-UNIX like, but we don't have much use for
+       // knowing if a zombie exists, so just reap it first.
+       int waitstatus;
+       waitpid(pid(), &waitstatus, WNOHANG);
+#endif
 
-       return retval_;
+       // Racy of course, but it will do.
+       if (lyx::support::kill(pid(), 0) && errno == ESRCH)
+               return false;
+       return true;
 }
 
+
 void ForkedProcess::kill(int tol)
 {
-       lyxerr << "ForkedProcess::kill(" << tol << ")" << std::endl;
+       lyxerr << "ForkedProcess::kill(" << tol << ')' << endl;
        if (pid() == 0) {
                lyxerr << "Can't kill non-existent process!" << endl;
                return;
        }
 
-       int const tolerance = std::max(0, tol);
+       // The weird (std::max)(a,b) signature prevents expansion
+       // of an evil MSVC macro.
+       int const tolerance = (std::max)(0, tol);
        if (tolerance == 0) {
                // Kill it dead NOW!
                Murder::killItDead(0, pid());
 
        } else {
-               int ret = lyx::kill(pid(), SIGHUP);
+               int ret = lyx::support::kill(pid(), SIGHUP);
 
                // The process is already dead if wait_for_death is false
                bool const wait_for_death = (ret == 0 && errno != ESRCH);
@@ -178,6 +206,29 @@ int ForkedProcess::waitForChild()
 {
        // We'll pretend that the child returns 1 on all error conditions.
        retval_ = 1;
+
+#if defined (_WIN32)
+       HANDLE const hProcess = HANDLE(pid_);
+
+       DWORD const wait_status = ::WaitForSingleObject(hProcess, INFINITE);
+
+       switch (wait_status) {
+       case WAIT_OBJECT_0: {
+               DWORD exit_code = 0;
+               if (!GetExitCodeProcess(hProcess, &exit_code)) {
+                       lyxerr << "GetExitCodeProcess failed waiting for child\n"
+                              << getChildErrorMessage() << std::endl;
+               } else
+                       retval_ = exit_code;
+               break;
+       }
+       case WAIT_FAILED:
+               lyxerr << "WaitForSingleObject failed waiting for child\n"
+                      << getChildErrorMessage() << std::endl;
+               break;
+       }
+
+#else
        int status;
        bool wait = true;
        while (wait) {
@@ -206,6 +257,7 @@ int ForkedProcess::waitForChild()
                        wait = false;
                }
        }
+#endif
        return retval_;
 }
 
@@ -219,7 +271,7 @@ int Forkedcall::startscript(Starttype wait, string const & what)
 
        command_ = what;
        signal_.reset();
-       return runBlocking();
+       return run(Wait);
 }
 
 
@@ -228,47 +280,100 @@ int Forkedcall::startscript(string const & what, SignalTypePtr signal)
        command_ = what;
        signal_  = signal;
 
-       return runNonBlocking();
+       return run(DontWait);
 }
 
 
 // generate child in background
 int Forkedcall::generateChild()
 {
-       // Split command_ up into a char * array
-       int const MAX_ARGV = 255;
-       char *argv[MAX_ARGV];
-
-       string line = command_;
-       int index = 0;
-       for (; index < MAX_ARGV-1; ++index) {
-               string word;
-               line = split(line, word, ' ');
-               if (word.empty())
-                       break;
-
-               char * tmp = new char[word.length() + 1];
-               word.copy(tmp, word.length());
-               tmp[word.length()] = '\0';
-
-               argv[index] = tmp;
+       string line = trim(command_);
+       if (line.empty())
+               return 1;
+
+       // Split the input command up into an array of words stored
+       // in a contiguous block of memory. The array contains pointers
+       // to each word.
+       // Don't forget the terminating `\0' character.
+       char const * const c_str = line.c_str();
+       vector<char> vec(c_str, c_str + line.size() + 1);
+
+       // Splitting the command up into an array of words means replacing
+       // the whitespace between words with '\0'. Life is complicated
+       // however, because words protected by quotes can contain whitespace.
+       //
+       // The strategy we adopt is:
+       // 1. If we're not inside quotes, then replace white space with '\0'.
+       // 2. If we are inside quotes, then don't replace the white space
+       //    but do remove the quotes themselves. We do this naively by
+       //    replacing the quote with '\0' which is fine if quotes
+       //    delimit the entire word.
+       char inside_quote = 0;
+       vector<char>::iterator it = vec.begin();
+       vector<char>::iterator const end = vec.end();
+       for (; it != end; ++it) {
+               char const c = *it;
+               if (!inside_quote) {
+                       if (c == ' ')
+                               *it = '\0';
+                       else if (c == '\'' || c == '"') {
+#if defined (_WIN32)
+                               // How perverse!
+                               // spawnvp *requires* the quotes or it will
+                               // split the arg at the internal whitespace!
+                               // Make shure the quote is a DOS-style one.
+                               *it = '"';
+#else
+                               *it = '\0';
+#endif
+                               inside_quote = c;
+                       }
+               } else if (c == inside_quote) {
+#if defined (_WIN32)
+                       *it = '"';
+#else
+                       *it = '\0';
+#endif
+                       inside_quote = 0;
+               }
+       }
+
+       // Build an array of pointers to each word.
+       it = vec.begin();
+       vector<char *> argv;
+       char prev = '\0';
+       for (; it != end; ++it) {
+               if (*it != '\0' && prev == '\0')
+                       argv.push_back(&*it);
+               prev = *it;
+       }
+       argv.push_back(0);
+
+       // Debug output.
+       if (lyxerr.debugging(Debug::FILES)) {
+               vector<char *>::iterator ait = argv.begin();
+               vector<char *>::iterator const aend = argv.end();
+               lyxerr << "<command>\n\t" << line
+                      << "\n\tInterpretted as:\n\n";
+               for (; ait != aend; ++ait)
+                       if (*ait)
+                               lyxerr << '\t'<< *ait << '\n';
+               lyxerr << "</command>" << std::endl;
        }
-       argv[index] = 0;
 
-#ifndef __EMX__
+#ifdef _WIN32
+       pid_t const cpid = spawnvp(_P_NOWAIT, argv[0], &*argv.begin());
+#else // POSIX
        pid_t const cpid = ::fork();
        if (cpid == 0) {
                // Child
-               execvp(argv[0], argv);
+               execvp(argv[0], &*argv.begin());
 
                // If something goes wrong, we end up here
                lyxerr << "execvp of \"" << command_ << "\" failed: "
                       << strerror(errno) << endl;
                _exit(1);
        }
-#else
-       pid_t const cpid = spawnvp(P_SESSION|P_DEFAULT|P_MINIMIZE|P_BACKGROUND,
-                                  syscmd, argv);
 #endif
 
        if (cpid < 0) {
@@ -276,12 +381,8 @@ int Forkedcall::generateChild()
                lyxerr << "Could not fork: " << strerror(errno) << endl;
        }
 
-       // Clean-up.
-       for (int i = 0; i < MAX_ARGV; ++i) {
-               if (argv[i] == 0)
-                       break;
-               delete [] argv[i];
-       }
-
        return cpid;
 }
+
+} // namespace support
+} // namespace lyx