retchmail/wvsendmail.cc

   1 #include "wvsendmail.h"
   2
   3 int WvSendmailProc::num_sendmails = 0;
   4
   5 WvSendmailProc::WvSendmailProc(const char **argv, int _count,
   6                                const WvSendmailCallback &_cb)
   7     : WvPipe(argv[0], argv, true, false, false), count(_count), cb(_cb)
   8 {
   9     is_done = exited = false;
  10     num_sendmails++;
  11 }
  12
  13
  14 WvSendmailProc::~WvSendmailProc()
  15 {
  16     close();
  17
  18     int ret = finish();
  19
  20     // be certain the callback gets called
  21     if (!exited && cb)
  22         cb(count, (ret == 0));
  23
  24     num_sendmails--;
  25 }
  26
  27
  28 bool WvSendmailProc::pre_select(SelectInfo &si)
  29 {
  30     bool must = false;
  31
  32     if (is_done && !exited && si.msec_timeout > 20)
  33         si.msec_timeout = 20;
  34
  35     if (child_exited() && !exited && is_done)
  36     {
  37         exited = true;
  38
  39         // call the callback
  40         if (cb)
  41             cb(count, !exit_status());
  42
  43         si.msec_timeout = 0;
  44         must = true;
  45     }
  46
  47     // another hack because isok() returns true when it shouldn't really
  48     if ((exited || is_done) && si.wants.writable)
  49         must = true;
  50
  51     return must || WvPipe::pre_select(si);
  52 }
  53
  54
  55 bool WvSendmailProc::isok() const
  56 {
  57     // note: this means people will try to write to us even if the pipe
  58     // says it's invalid!
  59     return WvPipe::isok() || !exited;
  60 }
  61
  62
  63 size_t WvSendmailProc::uwrite(const void *buf, size_t count)
  64 {
  65     if (child_exited())
  66         return count; // fake it, because isok() is also faking it
  67     return WvPipe::uwrite(buf, count);
  68 }
  69
  70
  71 void WvSendmailProc::execute()
  72 {
  73     WvPipe::execute();
  74
  75     if (is_done && select(0, false, true, false))
  76         close();
  77 }
  78
  79
  80 void WvSendmailProc::done()
  81 {
  82     is_done = true;
  83     force_select(false, true);
  84 }