gclib2/modules/Core/djobs.cxx

   1 /*
   2  * (c) Oleg Puchinin 2006
   3  * graycardinalster@gmail.com
   4  *
   5  */
   6
   7 #include "gclib2.h"
   8 #include <sys/types.h>
   9 #include <sys/wait.h>
  10 #include <errno.h>
  11
  12 DJobs::DJobs ()
  13 {
  14
  15 }
  16
  17 DJobs::~DJobs ()
  18 {
  19
  20 }
  21
  22 /// Добавить процесс (на контроль).
  23 __djob_t * DJobs::add (int pid, char * user_data)
  24 {
  25         __djob_t * j;
  26         j = alloc_job ();
  27         j->pid = pid;
  28         j->user_data = user_data;
  29         add_tail (LPCHAR(j));
  30         return j;
  31 }
  32
  33 __djob_t * DJobs::attach (__djob_t * j)
  34 {
  35         if (! j)
  36                 return NULL;
  37
  38         return (__djob_t *) add_tail (LPCHAR(j));
  39 }
  40
  41 __djob_t * DJobs::zombie ()
  42 {
  43         pid_t pid;
  44         int status;
  45
  46         if (! first ())
  47                 return NULL;
  48
  49         pid = waitpid (-1, &status, WNOHANG);
  50         if (pid <= 0)
  51                 return NULL;
  52
  53         return __do_zombie (pid, status);
  54 }
  55
  56 __djob_t * DJobs::__do_zombie (pid_t pid, int exit_status)
  57 {
  58         __djob_t * j = by_pid (pid);
  59         if (j) {
  60                 rm ();
  61                 j->exit_status = exit_status;
  62         }
  63         return j;
  64 }
  65
  66 __djob_t * DJobs::wait_all ()
  67 {
  68         __djob_t * j;
  69         int status;
  70         pid_t pid;
  71
  72         while (first ()) {
  73                 pid = waitpid (-1, &status, 0);
  74                 if (pid <= 0)
  75                         break;
  76                 j = __do_zombie (pid, status);
  77                 return j;
  78         }
  79
  80         return NULL;
  81 }
  82
  83 __djob_t * DJobs::alloc_job ()
  84 {
  85         __djob_t * j;
  86         j = CNEW (__djob_t, 1);
  87         Dexec_init (j);
  88         return j;
  89 }
  90
  91 __djob_t * DJobs::by_pid (pid_t pid)
  92 {
  93         __djob_t * j;
  94         first ();
  95         while ((j = (__djob_t *) get ()) && j) {
  96                 if (j->pid == pid)
  97                         return j;
  98                 next ();
  99         }
 100         return NULL;
 101 }
 102
 103 __djob_t * DJobs::fork ()
 104 {
 105         pid_t pid;
 106         __djob_t *j;
 107
 108         pid = ::fork ();
 109         if (pid < 0)
 110                 return NULL;
 111
 112         j = alloc_job ();
 113         Dexec_init (j);
 114         if (pid > 0)
 115                 j->pid = pid;
 116         else {
 117                 j->pid = getpid ();
 118                 j->child = 1;
 119         }
 120
 121         attach (j);
 122         return j;
 123 }
 124
 125
 126