X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Flib%2Femailer.cc;h=dc216e90cc087c30d5f553703786f34566e07bc1;hb=e9cca90a162e3e4b1db4a5e5188831beaa69d44e;hp=916d1718248a08cf5921204469a723979d45d555;hpb=969064d0e45b5885374452ff1ac2e15377bd76eb;p=dcpomatic.git diff --git a/src/lib/emailer.cc b/src/lib/emailer.cc index 916d17182..dc216e90c 100644 --- a/src/lib/emailer.cc +++ b/src/lib/emailer.cc @@ -1,25 +1,24 @@ /* - Copyright (C) 2015 Carl Hetherington + Copyright (C) 2015-2019 Carl Hetherington - This program is free software; you can redistribute it and/or modify + This file is part of DCP-o-matic. + + DCP-o-matic is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. - This program is distributed in the hope that it will be useful, + DCP-o-matic is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License - along with this program; if not, write to the Free Software - Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. + along with DCP-o-matic. If not, see . */ #include "compose.hpp" -#include "job.h" -#include "data.h" #include "config.h" #include "emailer.h" #include "exceptions.h" @@ -31,21 +30,29 @@ #include "i18n.h" using std::string; -using std::stringstream; using std::min; using std::list; using std::cout; using std::pair; using boost::shared_ptr; +using dcp::Data; -Emailer::Emailer (string from, string to, string subject, string body) +Emailer::Emailer (string from, list to, string subject, string body) : _from (from) , _to (to) , _subject (subject) - , _body (body) + , _body (fix (body)) , _offset (0) { - boost::algorithm::replace_all (_body, "\n", "\r\n"); + +} + +string +Emailer::fix (string s) const +{ + boost::algorithm::replace_all (s, "\n", "\r\n"); + boost::algorithm::replace_all (s, "\0", " "); + return s; } void @@ -86,35 +93,33 @@ size_t Emailer::get_data (void* ptr, size_t size, size_t nmemb) { size_t const t = min (_email.length() - _offset, size * nmemb); - memcpy (ptr, _email.substr(_offset, size * nmemb).c_str(), size * nmemb); + memcpy (ptr, _email.substr (_offset, t).c_str(), t); _offset += t; return t; } void -Emailer::send (shared_ptr job) +Emailer::send (string server, int port, EmailProtocol protocol, string user, string password) { - char date_buffer[32]; + char date_buffer[128]; time_t now = time (0); strftime (date_buffer, sizeof(date_buffer), "%a, %d %b %Y %H:%M:%S ", localtime (&now)); boost::posix_time::ptime const utc_now = boost::posix_time::second_clock::universal_time (); boost::posix_time::ptime const local_now = boost::date_time::c_local_adjustor::utc_to_local (utc_now); boost::posix_time::time_duration offset = local_now - utc_now; - sprintf (date_buffer + strlen(date_buffer), "%s%02d%02d", (offset.hours() >= 0 ? "+" : "-"), abs (offset.hours()), offset.minutes()); + sprintf (date_buffer + strlen(date_buffer), "%s%02d%02d", (offset.hours() >= 0 ? "+" : "-"), int(abs(offset.hours())), int(offset.minutes())); - stringstream email; - - email << "Date: " << date_buffer << "\r\n" - << "To: " << _to << "\r\n" - << "From: " << _from << "\r\n"; + _email = "Date: " + string(date_buffer) + "\r\n" + "To: " + address_list (_to) + "\r\n" + "From: " + _from + "\r\n"; if (!_cc.empty ()) { - email << "Cc: " << address_list (_cc); + _email += "Cc: " + address_list (_cc) + "\r\n"; } if (!_bcc.empty ()) { - email << "Bcc: " << address_list (_bcc); + _email += "Bcc: " + address_list (_bcc) + "\r\n"; } string const chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890"; @@ -124,26 +129,26 @@ Emailer::send (shared_ptr job) } if (!_attachments.empty ()) { - email << "MIME-Version: 1.0\r\n" - << "Content-Type: multipart/mixed; boundary=" << boundary << "\r\n"; + _email += "MIME-Version: 1.0\r\n" + "Content-Type: multipart/mixed; boundary=" + boundary + "\r\n"; } - email << "Subject: " << _subject << "\r\n" - << "User-Agent: DCP-o-matic\r\n" - << "\r\n"; + _email += "Subject: " + _subject + "\r\n" + "User-Agent: DCP-o-matic\r\n" + "\r\n"; if (!_attachments.empty ()) { - email << "--" << boundary << "\r\n" - << "Content-Type: text/plain; charset=utf-8\r\n\r\n"; + _email += "--" + boundary + "\r\n" + + "Content-Type: text/plain; charset=utf-8\r\n\r\n"; } - email << _body; + _email += _body; BOOST_FOREACH (Attachment i, _attachments) { - email << "\r\n\r\n--" << boundary << "\r\n" - << "Content-Type: " << i.mime_type << "; name=" << i.name << "\r\n" - << "Content-Transfer-Encoding: Base64\r\n" - << "Content-Disposition: attachment; filename=" << i.name << "\r\n\r\n"; + _email += "\r\n\r\n--" + boundary + "\r\n" + "Content-Type: " + i.mime_type + "; name=" + i.name + "\r\n" + "Content-Transfer-Encoding: Base64\r\n" + "Content-Disposition: attachment; filename=" + i.name + "\r\n\r\n"; BIO* b64 = BIO_new (BIO_f_base64()); @@ -156,17 +161,15 @@ Emailer::send (shared_ptr job) char* out; long int bytes = BIO_get_mem_data (bio, &out); - email << string (out, bytes); + _email += fix (string (out, bytes)); BIO_free_all (b64); } if (!_attachments.empty ()) { - email << "\r\n--" << boundary << "--\r\n"; + _email += "\r\n--" + boundary + "--\r\n"; } - _email = email.str (); - curl_global_init (CURL_GLOBAL_DEFAULT); CURL* curl = curl_easy_init (); @@ -174,27 +177,26 @@ Emailer::send (shared_ptr job) throw NetworkError ("Could not initialise libcurl"); } - CURLM* mcurl = curl_multi_init (); - if (!mcurl) { - throw NetworkError ("Could not initialise libcurl"); + if ((protocol == EMAIL_PROTOCOL_AUTO && port == 465) || protocol == EMAIL_PROTOCOL_SSL) { + /* "SSL" or "Implicit TLS"; I think curl wants us to use smtps here */ + curl_easy_setopt (curl, CURLOPT_URL, String::compose("smtps://%1:%2", server, port).c_str()); + } else { + curl_easy_setopt (curl, CURLOPT_URL, String::compose("smtp://%1:%2", server, port).c_str()); } - curl_easy_setopt (curl, CURLOPT_URL, String::compose ( - "smtp://%1:%2", - Config::instance()->mail_server().c_str(), - Config::instance()->mail_port() - ).c_str()); - - if (!Config::instance()->mail_user().empty ()) { - curl_easy_setopt (curl, CURLOPT_USERNAME, Config::instance()->mail_user().c_str()); + if (!user.empty ()) { + curl_easy_setopt (curl, CURLOPT_USERNAME, user.c_str ()); } - if (!Config::instance()->mail_password().empty ()) { - curl_easy_setopt (curl, CURLOPT_PASSWORD, Config::instance()->mail_password().c_str()); + if (!password.empty ()) { + curl_easy_setopt (curl, CURLOPT_PASSWORD, password.c_str()); } curl_easy_setopt (curl, CURLOPT_MAIL_FROM, _from.c_str()); - struct curl_slist* recipients = curl_slist_append (0, _to.c_str()); + struct curl_slist* recipients = 0; + BOOST_FOREACH (string i, _to) { + recipients = curl_slist_append (recipients, i.c_str()); + } BOOST_FOREACH (string i, _cc) { recipients = curl_slist_append (recipients, i.c_str()); } @@ -208,98 +210,21 @@ Emailer::send (shared_ptr job) curl_easy_setopt (curl, CURLOPT_READDATA, this); curl_easy_setopt (curl, CURLOPT_UPLOAD, 1L); - curl_easy_setopt (curl, CURLOPT_USE_SSL, (long) CURLUSESSL_TRY); + if (protocol == EMAIL_PROTOCOL_AUTO || protocol == EMAIL_PROTOCOL_STARTTLS) { + curl_easy_setopt (curl, CURLOPT_USE_SSL, (long) CURLUSESSL_TRY); + } curl_easy_setopt (curl, CURLOPT_SSL_VERIFYPEER, 0L); curl_easy_setopt (curl, CURLOPT_SSL_VERIFYHOST, 0L); curl_easy_setopt (curl, CURLOPT_VERBOSE, 1L); curl_easy_setopt (curl, CURLOPT_DEBUGFUNCTION, curl_debug_shim); curl_easy_setopt (curl, CURLOPT_DEBUGDATA, this); - curl_multi_add_handle (mcurl, curl); - - time_t start = time (0); - - int still_running = 1; - curl_multi_perform (mcurl, &still_running); - - while (still_running) { - - fd_set fdread; - fd_set fdwrite; - fd_set fdexcep; - - FD_ZERO (&fdread); - FD_ZERO (&fdwrite); - FD_ZERO (&fdexcep); - - struct timeval timeout; - timeout.tv_sec = 1; - timeout.tv_usec = 0; - - long curl_timeout = -1; - curl_multi_timeout (mcurl, &curl_timeout); - if (curl_timeout >= 0) { - timeout.tv_sec = curl_timeout / 1000; - if (timeout.tv_sec > 1) { - timeout.tv_sec = 1; - } else { - timeout.tv_usec = (curl_timeout % 1000) * 1000; - } - } - - int maxfd = -1; - CURLMcode mc = curl_multi_fdset (mcurl, &fdread, &fdwrite, &fdexcep, &maxfd); - - if (mc != CURLM_OK) { - throw KDMError (String::compose ("Failed to send KDM email to %1", _to)); - } - - int rc; - if (maxfd == -1) { -#ifdef DCPOMATIC_WINDOWS - Sleep (100); - rc = 0; -#else - struct timeval wait = { 0, 100 * 1000}; - rc = select (0, 0, 0, 0, &wait); -#endif - } else { - rc = select (maxfd + 1, &fdread, &fdwrite, &fdexcep, &timeout); - } - - if (rc < 0) { - throw KDMError ("Failed to send KDM email"); - } - - mc = curl_multi_perform (mcurl, &still_running); - if (mc != CURLM_OK) { - throw KDMError (String::compose ("Failed to send KDM email (%1)", curl_multi_strerror (mc))); - } - - if (job) { - job->set_progress_unknown (); - } - - if ((time(0) - start) > 10) { - throw KDMError (_("Failed to send KDM email (timed out)")); - } + CURLcode const r = curl_easy_perform (curl); + if (r != CURLE_OK) { + throw KDMError (_("Failed to send email"), curl_easy_strerror (r)); } - int messages; - do { - CURLMsg* m = curl_multi_info_read (mcurl, &messages); - if (m && m->data.result != CURLE_OK) { - throw KDMError (String::compose ("Failed to send KDM email (%1)", curl_easy_strerror (m->data.result))); - } - } while (messages > 0); - - /* XXX: we should do this stuff when an exception is thrown, but curl_multi_remove_handle - seems to hang if we try that. - */ - curl_slist_free_all (recipients); - curl_multi_remove_handle (mcurl, curl); - curl_multi_cleanup (mcurl); curl_easy_cleanup (curl); curl_global_cleanup (); }