--- /dev/null
+/*
+ Copyright (C) 2015 Carl Hetherington <cth@carlh.net>
+
+ This program 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,
+ 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.
+
+*/
+
+#include "compose.hpp"
+#include "job.h"
+#include "data.h"
+#include "config.h"
+#include "emailer.h"
+#include "exceptions.h"
+#include <curl/curl.h>
+#include <boost/algorithm/string.hpp>
+#include <boost/date_time/c_local_time_adjustor.hpp>
+#include <boost/foreach.hpp>
+
+#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;
+
+Emailer::Emailer (string from, string to, string subject, string body)
+ : _from (from)
+ , _to (to)
+ , _subject (subject)
+ , _body (body)
+ , _offset (0)
+{
+ boost::algorithm::replace_all (_body, "\n", "\r\n");
+}
+
+void
+Emailer::add_cc (string cc)
+{
+ _cc.push_back (cc);
+}
+
+void
+Emailer::add_bcc (string bcc)
+{
+ _bcc.push_back (bcc);
+}
+
+void
+Emailer::add_attachment (boost::filesystem::path attachment, string mime_type)
+{
+ _attachments.push_back (make_pair (attachment, mime_type));
+}
+
+static size_t
+curl_data_shim (void* ptr, size_t size, size_t nmemb, void* userp)
+{
+ return reinterpret_cast<Emailer*>(userp)->get_data (ptr, size, nmemb);
+}
+
+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);
+ _offset += t;
+ return t;
+}
+
+void
+Emailer::send (shared_ptr<Job> job)
+{
+ char date_buffer[32];
+ 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<boost::posix_time::ptime>::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());
+
+ stringstream email;
+
+ email << "Date: " << date_buffer << "\r\n"
+ << "To: " << _to << "\r\n"
+ << "From: " << _from << "\r\n";
+
+ if (!_cc.empty ()) {
+ email << "Cc: " << address_list (_cc);
+ }
+
+ if (!_bcc.empty ()) {
+ email << "Bcc: " << address_list (_bcc);
+ }
+
+ string const chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890";
+ string boundary;
+ for (int i = 0; i < 32; ++i) {
+ boundary += chars[rand() % chars.length()];
+ }
+
+ if (!_attachments.empty ()) {
+ email << "MIME-Version: 1.0\r\n"
+ << "Content-Type: multipart/alternative; boundary=" << boundary << "\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 << _body;
+
+ for (list<pair<boost::filesystem::path, string> >::const_iterator i = _attachments.begin(); i != _attachments.end(); ++i) {
+ email << "\r\n\r\n--" << boundary << "\r\n"
+ << "Content-Type: " << i->second << "; name=" << i->first.leaf() << "\r\n"
+ << "Content-Transfer-Encoding: Base64\r\n"
+ << "Content-Disposition: attachment; filename=" << i->first.leaf() << "\r\n\r\n";
+
+ BIO* b64 = BIO_new (BIO_f_base64());
+
+ BIO* bio = BIO_new (BIO_s_mem());
+ bio = BIO_push (b64, bio);
+
+ Data data (i->first);
+ BIO_write (bio, data.data().get(), data.size());
+ (void) BIO_flush (bio);
+
+ char* out;
+ long int bytes = BIO_get_mem_data (bio, &out);
+ email << string (out, bytes);
+
+ BIO_free_all (b64);
+ }
+
+ if (!_attachments.empty ()) {
+ email << "\r\n--" << boundary << "--\r\n";
+ }
+
+ _email = email.str ();
+
+ curl_global_init (CURL_GLOBAL_DEFAULT);
+
+ CURL* curl = curl_easy_init ();
+ if (!curl) {
+ throw NetworkError ("Could not initialise libcurl");
+ }
+
+ CURLM* mcurl = curl_multi_init ();
+ if (!mcurl) {
+ throw NetworkError ("Could not initialise libcurl");
+ }
+
+ 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 (!Config::instance()->mail_password().empty ()) {
+ curl_easy_setopt (curl, CURLOPT_PASSWORD, Config::instance()->mail_password().c_str());
+ }
+
+ curl_easy_setopt (curl, CURLOPT_MAIL_FROM, _from.c_str());
+
+ struct curl_slist* recipients = curl_slist_append (0, _to.c_str());
+ BOOST_FOREACH (string i, _cc) {
+ recipients = curl_slist_append (recipients, i.c_str());
+ }
+ BOOST_FOREACH (string i, _bcc) {
+ recipients = curl_slist_append (recipients, i.c_str());
+ }
+
+ curl_easy_setopt (curl, CURLOPT_MAIL_RCPT, recipients);
+
+ curl_easy_setopt (curl, CURLOPT_READFUNCTION, curl_data_shim);
+ curl_easy_setopt (curl, CURLOPT_READDATA, this);
+ curl_easy_setopt (curl, CURLOPT_UPLOAD, 1L);
+
+ 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);
+
+ _notes_buffer.reset (new char[65536]);
+ FILE* notes = fmemopen (_notes_buffer.get(), 65536, "w");
+ curl_easy_setopt (curl, CURLOPT_STDERR, notes);
+
+ 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) {
+ fclose (notes);
+ 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) {
+ fclose (notes);
+ 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) {
+ fclose (notes);
+ throw KDMError (_("Failed to send KDM email (timed out)"));
+ }
+ }
+
+ int messages;
+ do {
+ CURLMsg* m = curl_multi_info_read (mcurl, &messages);
+ if (m && m->data.result != CURLE_OK) {
+ fclose (notes);
+ 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 ();
+
+ fclose (notes);
+}
+
+string
+Emailer::address_list (list<string> addresses)
+{
+ string o;
+ BOOST_FOREACH (string i, addresses) {
+ o += i + ", ";
+ }
+
+ return o.substr (0, o.length() - 2);
+}
+
+string
+Emailer::notes () const
+{
+ return string (_notes_buffer.get());
+}
projects / dcpomatic.git / commitdiff