evolution/updateitem.cc

   1 #include "updateitem.h"
   2 #include "genericcommand.h"
   3 #include "exchangeitadaptor.h"
   4 #include "wvfile.h"
   5 #include "wvbase64.h"
   6 #include "wvstream.h"
   7 #include "wvtnefconv.h"
   8 #include "wvloopback.h"
   9 #include "wvencoderstream.h"
  10
  11 UpdateItem::UpdateItem(WvStream &_stream,
  12                        ExchangeItAdaptor *_adaptor,
  13                        WvStringParm _uid) :
  14     log("UpdateItem", WvLog::Debug1),
  15     stream(_stream),
  16     adaptor(_adaptor),
  17     uid(_uid)
  18 {
  19     assert(adaptor);
  20 }
  21
  22 bool UpdateItem::update(bool careful)
  23 {
  24     // FIXME (22/12/03): this code is inefficient, shouldn't be
  25     // allocating all the time. probably not worth the effort
  26     // until we fix the scalability problem on the server though
  27     // It also tickles a bug in WvTnef, namely that it expects good
  28     // data right away; with larger items, the Base64Decoder can't
  29     // "keep up" (pcolijn)
  30
  31     WvString line;
  32     WvStream *out_stream = NULL;
  33
  34     WvEncoderStream b64_stream(new WvLoopback());
  35     b64_stream.writechain.append(new WvBase64Decoder(), true);
  36     b64_stream.auto_flush(false);
  37
  38     log("Reading tnef");
  39
  40     while (!!(line = stream.blocking_getline(TIMEOUT)))
  41     {
  42         log(WvLog::Debug5, "Read:\n'%s'\nfrom stream\n", line);
  43         b64_stream.write(line);
  44     }
  45
  46     b64_stream.finish_write();
  47     bool success = true;
  48
  49     // In some situations we may not receive a TNEF, and that's ok
  50     if (b64_stream.isreadable())
  51     {
  52         WvString tnef_path("%s/%s", adaptor->get_storage_path(), uid);
  53         WvFile file(tnef_path, (O_WRONLY|O_CREAT), 0600);
  54         if (file.isok())
  55             out_stream = &file;
  56         else if (!careful)
  57         {
  58             log("Could not open %s for writing: %s\n",
  59                 tnef_path, file.errstr());
  60             file.close();
  61             return false;
  62         }
  63
  64         WvTnef tnef(&b64_stream, out_stream);
  65
  66         if (tnef.isok())
  67             adaptor->update_item(tnef, uid);
  68         else
  69             success = false;
  70
  71         file.close();
  72     }
  73
  74     return success;
  75 }
  76