src/lib/emailer.cc

   1 /*
   2     Copyright (C) 2015 Carl Hetherington <cth@carlh.net>
   3
   4     This program is free software; you can redistribute it and/or modify
   5     it under the terms of the GNU General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or
   7     (at your option) any later version.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12     GNU General Public License for more details.
  13
  14     You should have received a copy of the GNU General Public License
  15     along with this program; if not, write to the Free Software
  16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 */
  19
  20 #include "compose.hpp"
  21 #include "job.h"
  22 #include "data.h"
  23 #include "config.h"
  24 #include "emailer.h"
  25 #include "exceptions.h"
  26 #include <curl/curl.h>
  27 #include <boost/algorithm/string.hpp>
  28 #include <boost/date_time/c_local_time_adjustor.hpp>
  29 #include <boost/foreach.hpp>
  30
  31 #include "i18n.h"
  32
  33 using std::string;
  34 using std::stringstream;
  35 using std::min;
  36 using std::list;
  37 using std::cout;
  38 using std::pair;
  39 using boost::shared_ptr;
  40
  41 Emailer::Emailer (string from, string to, string subject, string body)
  42         : _from (from)
  43         , _to (to)
  44         , _subject (subject)
  45         , _body (body)
  46         , _offset (0)
  47 {
  48         boost::algorithm::replace_all (_body, "\n", "\r\n");
  49 }
  50
  51 void
  52 Emailer::add_cc (string cc)
  53 {
  54         _cc.push_back (cc);
  55 }
  56
  57 void
  58 Emailer::add_bcc (string bcc)
  59 {
  60         _bcc.push_back (bcc);
  61 }
  62
  63 void
  64 Emailer::add_attachment (boost::filesystem::path file, string name, string mime_type)
  65 {
  66         Attachment a;
  67         a.file = file;
  68         a.name = name;
  69         a.mime_type = mime_type;
  70         _attachments.push_back (a);
  71 }
  72
  73 static size_t
  74 curl_data_shim (void* ptr, size_t size, size_t nmemb, void* userp)
  75 {
  76         return reinterpret_cast<Emailer*>(userp)->get_data (ptr, size, nmemb);
  77 }
  78
  79 size_t
  80 Emailer::get_data (void* ptr, size_t size, size_t nmemb)
  81 {
  82         size_t const t = min (_email.length() - _offset, size * nmemb);
  83         memcpy (ptr, _email.substr(_offset, size * nmemb).c_str(), size * nmemb);
  84         _offset += t;
  85         return t;
  86 }
  87
  88 void
  89 Emailer::send (shared_ptr<Job> job)
  90 {
  91         char date_buffer[32];
  92         time_t now = time (0);
  93         strftime (date_buffer, sizeof(date_buffer), "%a, %d %b %Y %H:%M:%S ", localtime (&now));
  94
  95         boost::posix_time::ptime const utc_now = boost::posix_time::second_clock::universal_time ();
  96         boost::posix_time::ptime const local_now = boost::date_time::c_local_adjustor<boost::posix_time::ptime>::utc_to_local (utc_now);
  97         boost::posix_time::time_duration offset = local_now - utc_now;
  98         sprintf (date_buffer + strlen(date_buffer), "%s%02d%02d", (offset.hours() >= 0 ? "+" : "-"), abs (offset.hours()), offset.minutes());
  99
 100         stringstream email;
 101
 102         email << "Date: " << date_buffer << "\r\n"
 103               << "To: " << _to << "\r\n"
 104               << "From: " << _from << "\r\n";
 105
 106         if (!_cc.empty ()) {
 107                 email << "Cc: " << address_list (_cc);
 108         }
 109
 110         if (!_bcc.empty ()) {
 111                 email << "Bcc: " << address_list (_bcc);
 112         }
 113
 114         string const chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890";
 115         string boundary;
 116         for (int i = 0; i < 32; ++i) {
 117                 boundary += chars[rand() % chars.length()];
 118         }
 119
 120         if (!_attachments.empty ()) {
 121                 email << "MIME-Version: 1.0\r\n"
 122                       << "Content-Type: multipart/alternative; boundary=" << boundary << "\r\n";
 123         }
 124
 125         email << "Subject: " << _subject << "\r\n"
 126               << "User-Agent: DCP-o-matic\r\n"
 127               << "\r\n";
 128
 129         if (!_attachments.empty ()) {
 130                 email << "--" << boundary << "\r\n"
 131                       << "Content-Type: text/plain; charset=utf-8\r\n\r\n";
 132         }
 133
 134         email << _body;
 135
 136         BOOST_FOREACH (Attachment i, _attachments) {
 137                 email << "\r\n\r\n--" << boundary << "\r\n"
 138                       << "Content-Type: " << i.mime_type << "; name=" << i.name << "\r\n"
 139                       << "Content-Transfer-Encoding: Base64\r\n"
 140                       << "Content-Disposition: attachment; filename=" << i.name << "\r\n\r\n";
 141
 142                 BIO* b64 = BIO_new (BIO_f_base64());
 143
 144                 BIO* bio = BIO_new (BIO_s_mem());
 145                 bio = BIO_push (b64, bio);
 146
 147                 Data data (i.file);
 148                 BIO_write (bio, data.data().get(), data.size());
 149                 (void) BIO_flush (bio);
 150
 151                 char* out;
 152                 long int bytes = BIO_get_mem_data (bio, &out);
 153                 email << string (out, bytes);
 154
 155                 BIO_free_all (b64);
 156         }
 157
 158         if (!_attachments.empty ()) {
 159                 email << "\r\n--" << boundary << "--\r\n";
 160         }
 161
 162         _email = email.str ();
 163
 164         curl_global_init (CURL_GLOBAL_DEFAULT);
 165
 166         CURL* curl = curl_easy_init ();
 167         if (!curl) {
 168                 throw NetworkError ("Could not initialise libcurl");
 169         }
 170
 171         CURLM* mcurl = curl_multi_init ();
 172         if (!mcurl) {
 173                 throw NetworkError ("Could not initialise libcurl");
 174         }
 175
 176         curl_easy_setopt (curl, CURLOPT_URL, String::compose (
 177                                   "smtp://%1:%2",
 178                                   Config::instance()->mail_server().c_str(),
 179                                   Config::instance()->mail_port()
 180                                   ).c_str());
 181
 182         if (!Config::instance()->mail_user().empty ()) {
 183                 curl_easy_setopt (curl, CURLOPT_USERNAME, Config::instance()->mail_user().c_str());
 184         }
 185         if (!Config::instance()->mail_password().empty ()) {
 186                 curl_easy_setopt (curl, CURLOPT_PASSWORD, Config::instance()->mail_password().c_str());
 187         }
 188
 189         curl_easy_setopt (curl, CURLOPT_MAIL_FROM, _from.c_str());
 190
 191         struct curl_slist* recipients = curl_slist_append (0, _to.c_str());
 192         BOOST_FOREACH (string i, _cc) {
 193                 recipients = curl_slist_append (recipients, i.c_str());
 194         }
 195         BOOST_FOREACH (string i, _bcc) {
 196                 recipients = curl_slist_append (recipients, i.c_str());
 197         }
 198
 199         curl_easy_setopt (curl, CURLOPT_MAIL_RCPT, recipients);
 200
 201         curl_easy_setopt (curl, CURLOPT_READFUNCTION, curl_data_shim);
 202         curl_easy_setopt (curl, CURLOPT_READDATA, this);
 203         curl_easy_setopt (curl, CURLOPT_UPLOAD, 1L);
 204
 205         curl_easy_setopt (curl, CURLOPT_USE_SSL, (long) CURLUSESSL_TRY);
 206         curl_easy_setopt (curl, CURLOPT_SSL_VERIFYPEER, 0L);
 207         curl_easy_setopt (curl, CURLOPT_SSL_VERIFYHOST, 0L);
 208         curl_easy_setopt (curl, CURLOPT_VERBOSE, 1L);
 209
 210         _notes_buffer.reset (new char[65536]);
 211         FILE* notes = fmemopen (_notes_buffer.get(), 65536, "w");
 212         curl_easy_setopt (curl, CURLOPT_STDERR, notes);
 213
 214         curl_multi_add_handle (mcurl, curl);
 215
 216         time_t start = time (0);
 217
 218         int still_running = 1;
 219         curl_multi_perform (mcurl, &still_running);
 220
 221         while (still_running) {
 222
 223                 fd_set fdread;
 224                 fd_set fdwrite;
 225                 fd_set fdexcep;
 226
 227                 FD_ZERO (&fdread);
 228                 FD_ZERO (&fdwrite);
 229                 FD_ZERO (&fdexcep);
 230
 231                 struct timeval timeout;
 232                 timeout.tv_sec = 1;
 233                 timeout.tv_usec = 0;
 234
 235                 long curl_timeout = -1;
 236                 curl_multi_timeout (mcurl, &curl_timeout);
 237                 if (curl_timeout >= 0) {
 238                         timeout.tv_sec = curl_timeout / 1000;
 239                         if (timeout.tv_sec > 1) {
 240                                 timeout.tv_sec = 1;
 241                         } else {
 242                                 timeout.tv_usec = (curl_timeout % 1000) * 1000;
 243                         }
 244                 }
 245
 246                 int maxfd = -1;
 247                 CURLMcode mc = curl_multi_fdset (mcurl, &fdread, &fdwrite, &fdexcep, &maxfd);
 248
 249                 if (mc != CURLM_OK) {
 250                         fclose (notes);
 251                         throw KDMError (String::compose ("Failed to send KDM email to %1", _to));
 252                 }
 253
 254                 int rc;
 255                 if (maxfd == -1) {
 256 #ifdef DCPOMATIC_WINDOWS
 257                         Sleep (100);
 258                         rc = 0;
 259 #else
 260                         struct timeval wait = { 0, 100 * 1000};
 261                         rc = select (0, 0, 0, 0, &wait);
 262 #endif
 263                 } else {
 264                         rc = select (maxfd + 1, &fdread, &fdwrite, &fdexcep, &timeout);
 265                 }
 266
 267                 if (rc < 0) {
 268                         throw KDMError ("Failed to send KDM email");
 269                 }
 270
 271                 mc = curl_multi_perform (mcurl, &still_running);
 272                 if (mc != CURLM_OK) {
 273                         fclose (notes);
 274                         throw KDMError (String::compose ("Failed to send KDM email (%1)", curl_multi_strerror (mc)));
 275                 }
 276
 277                 if (job) {
 278                         job->set_progress_unknown ();
 279                 }
 280
 281                 if ((time(0) - start) > 10) {
 282                         fclose (notes);
 283                         throw KDMError (_("Failed to send KDM email (timed out)"));
 284                 }
 285         }
 286
 287         int messages;
 288         do {
 289                 CURLMsg* m = curl_multi_info_read (mcurl, &messages);
 290                 if (m && m->data.result != CURLE_OK) {
 291                         fclose (notes);
 292                         throw KDMError (String::compose ("Failed to send KDM email (%1)", curl_easy_strerror (m->data.result)));
 293                 }
 294         } while (messages > 0);
 295
 296         /* XXX: we should do this stuff when an exception is thrown, but curl_multi_remove_handle
 297            seems to hang if we try that.
 298         */
 299
 300         curl_slist_free_all (recipients);
 301         curl_multi_remove_handle (mcurl, curl);
 302         curl_multi_cleanup (mcurl);
 303         curl_easy_cleanup (curl);
 304         curl_global_cleanup ();
 305
 306         fclose (notes);
 307 }
 308
 309 string
 310 Emailer::address_list (list<string> addresses)
 311 {
 312         string o;
 313         BOOST_FOREACH (string i, addresses) {
 314                 o += i + ", ";
 315         }
 316
 317         return o.substr (0, o.length() - 2);
 318 }
 319
 320 string
 321 Emailer::notes () const
 322 {
 323         return string (_notes_buffer.get());
 324 }