+++ /dev/null
-#include <cstdlib>
-#include <unistd.h>
-#include <sys/wait.h>
-#include <msp/fs/dir.h>
-#include <msp/io/console.h>
-#include <msp/io/file.h>
-#include <msp/io/print.h>
-#include <msp/time/units.h>
-#include "externaltask.h"
-
-using namespace std;
-using namespace Msp;
-
-ExternalTask::ExternalTask(const vector<string> &a, const FS::Path &wd):
- argv(a),
- work_dir(wd),
- pid(-1),
- exit_code(-1),
- stdout_dest(PASSTHROUGH),
- stderr_dest(PASSTHROUGH),
- capture_pipe(0)
-{ }
-
-ExternalTask::~ExternalTask()
-{
- delete capture_pipe;
-}
-
-string ExternalTask::get_command() const
-{
- string cmd;
- for(vector<string>::const_iterator i=argv.begin(); i!=argv.end(); ++i)
- {
- if(i!=argv.begin())
- cmd += ' ';
-
- for(string::const_iterator j=i->begin(); j!=i->end(); ++j)
- {
- if(*j=='"' || *j=='\'' || *j==' ' || *j=='\\' || *j=='&')
- cmd += '\\';
- cmd += *j;
- }
- }
-
- return cmd;
-}
-
-void ExternalTask::start()
-{
- if(stdout_dest==CAPTURE || stderr_dest==CAPTURE)
- capture_pipe = new IO::Pipe;
-
- prepare();
-
- if((pid = fork()))
- {
- if(pid==-1)
- exit_code = 1026;
- else
- exit_code = 0;
- }
- else
- {
- vector<const char *> cargv(argv.size()+1);
- for(unsigned i=0; i<argv.size(); ++i)
- cargv[i] = argv[i].c_str();
- cargv.back() = 0;
-
- if(stdout_dest==IGNORE || stderr_dest==IGNORE)
- {
- IO::File devnull("/dev/null", IO::M_WRITE);
- if(stdout_dest==IGNORE)
- IO::cout.redirect(devnull);
- if(stderr_dest==IGNORE)
- IO::cerr.redirect(devnull);
- }
-
- if(capture_pipe)
- {
- if(stdout_dest==CAPTURE)
- IO::cout.redirect(*capture_pipe);
- if(stderr_dest==CAPTURE)
- IO::cerr.redirect(*capture_pipe);
- delete capture_pipe;
- }
-
- if(!work_dir.empty())
- FS::chdir(work_dir);
- execvp(cargv.front(), const_cast<char *const *>(&cargv.front()));
- IO::print("Couldn't execute %s\n", argv.front());
- exit(1);
- }
-}
-
-Task::Status ExternalTask::check()
-{
- return do_wait(false);
-}
-
-Task::Status ExternalTask::wait()
-{
- return do_wait(true);
-}
-
-Task::Status ExternalTask::do_wait(bool block)
-{
- if(pid>0)
- {
- int status;
- if(waitpid(pid, &status, (block ? 0 : WNOHANG))==pid)
- {
- if(WIFEXITED(status))
- exit_code = WEXITSTATUS(status);
- else if(WIFSIGNALED(status))
- exit_code = 256+WTERMSIG(status);
- else
- exit_code = 1025;
- pid = 0;
- }
-
- // Do this after waiting to avoid a race condition
- if(capture_pipe)
- {
- while(IO::poll(*capture_pipe, IO::P_INPUT, Time::zero))
- {
- char buf[1024];
- unsigned len = capture_pipe->read(buf, sizeof(buf));
- output.append(buf, len);
- if(len<sizeof(buf))
- break;
- }
- }
-
- if(pid>0)
- return RUNNING;
- else
- signal_finished.emit(!exit_code);
- }
-
- return exit_code ? ERROR : SUCCESS;
-}
-
-void ExternalTask::set_stdout(Destination d)
-{
- stdout_dest = d;
-}
-
-void ExternalTask::set_stderr(Destination d)
-{
- stderr_dest = d;
-}
-
-string ExternalTask::run_and_capture_output(const Arguments &argv, const FS::Path &wd)
-{
- ExternalTask task(argv, wd);
- task.set_stdout(CAPTURE);
- task.set_stderr(IGNORE);
- task.start();
- if(task.wait()!=SUCCESS)
- throw runtime_error(format("%s failed", argv.front()));
- return task.get_output();
-}