source/externaltask.cpp

   1 #include <cstdlib>
   2 #include <unistd.h>
   3 #include <sys/wait.h>
   4 #include <msp/fs/dir.h>
   5 #include <msp/io/print.h>
   6 #include "externaltask.h"
   7
   8 using namespace std;
   9 using namespace Msp;
  10
  11 ExternalTask::ExternalTask(const vector<string> &argv, const FS::Path &work_dir):
  12         pid(-1),
  13         exit_code(-1)
  14 {
  15         if((pid = fork()))
  16         {
  17                 if(pid==-1)
  18                         exit_code = 1;
  19                 else
  20                         exit_code = 0;
  21         }
  22         else
  23         {
  24                 vector<const char *> cargv(argv.size()+1);
  25                 for(unsigned i=0; i<argv.size(); ++i)
  26                         cargv[i] = argv[i].c_str();
  27                 cargv.back() = 0;
  28
  29                 if(!work_dir.empty())
  30                         FS::chdir(work_dir);
  31                 execvp(cargv.front(), const_cast<char *const *>(&cargv.front()));
  32                 IO::print("Couldn't execute %s\n", argv.front());
  33                 exit(1);
  34         }
  35 }
  36
  37 Task::Status ExternalTask::check()
  38 {
  39         if(pid>0)
  40         {
  41                 int status;
  42                 if(waitpid(pid, &status, WNOHANG)==pid)
  43                 {
  44                         if(WIFEXITED(status))
  45                                 exit_code = WEXITSTATUS(status);
  46                         else if(WIFSIGNALED(status))
  47                                 exit_code = 256+WTERMSIG(status);
  48                         else
  49                                 exit_code = 1025;
  50                         pid = 0;
  51
  52                         signal_finished.emit(!exit_code);
  53                 }
  54                 else
  55                         return RUNNING;
  56         }
  57
  58         return exit_code ? ERROR : SUCCESS;
  59 }