1 #include "cetlib/PluginTypeDeducer.h"
2 #include "fhiclcpp/ParameterSet.h"
3 #if MESSAGEFACILITY_HEX_VERSION >= 0x20103
4 #include "fhiclcpp/types/ConfigurationTable.h"
5 #include "fhiclcpp/types/Sequence.h"
6 #include "fhiclcpp/types/TableFragment.h"
10 #include "messagefacility/MessageService/ELdestination.h"
11 #include "messagefacility/Utilities/ELseverityLevel.h"
12 #if MESSAGEFACILITY_HEX_VERSION < 0x20201 // v2_02_01 is s67
13 # include "messagefacility/MessageService/MessageDrop.h"
15 # include "messagefacility/MessageLogger/MessageLogger.h"
17 #include "messagefacility/Utilities/exception.h"
25 #include <arpa/inet.h>
28 #include <netinet/in.h>
29 #include <boost/thread.hpp>
31 #include <QtCore/QString>
33 #include "cetlib/compiler_macros.h"
37 using mf::service::ELdestination;
38 using mf::ELseverityLevel;
46 #if MESSAGEFACILITY_HEX_VERSION >= 0x20103
49 using strings_t = fhicl::Sequence<std::string>::default_type;
51 fhicl::TableFragment<ELdestination::Config> elDestConfig;
52 fhicl::Atom<std::string> host{ fhicl::Name{
"host" },fhicl::Comment{
"SMTP Server hostname" },
"smtp.fnal.gov" };
53 fhicl::Atom<int> port{ fhicl::Name{
"port" },fhicl::Comment{
"SMTP Server port" },25 };
54 fhicl::Sequence<std::string> to{ fhicl::Name{
"to_addresses" }, fhicl::Comment{
"The list of email addresses that SMTP mfPlugin should sent to" } , strings_t {} };
55 fhicl::Atom<std::string> from{ fhicl::Name{
"from_address" },fhicl::Comment{
"Source email address" } };
56 fhicl::Atom<std::string> subject{ fhicl::Name{
"subject" },fhicl::Comment{
"Subject of the email message" },
"MessageFacility SMTP Message Digest" };
57 fhicl::Atom<std::string> messageHeader{ fhicl::Name{
"message_header" },fhicl::Comment{
"String to preface messages with in email body" },
"" };
58 fhicl::Atom<bool> useSmtps{ fhicl::Name{
"use_smtps" },fhicl::Comment{
"Use SMTPS protocol" },
false };
59 fhicl::Atom<std::string> user{ fhicl::Name{
"smtp_username" },fhicl::Comment{
"Username for SMTP server" },
"" };
60 fhicl::Atom<std::string> pw{ fhicl::Name{
"smtp_password" },fhicl::Comment{
"Password for SMTP server" },
"" };
61 fhicl::Atom<bool> verifyCert{ fhicl::Name{
"verify_host_ssl_certificate" },fhicl::Comment{
"Whether to run full SSL verify on SMTP server in SMTPS mode" },
true };
62 fhicl::Atom<size_t> sendInterval{ fhicl::Name{
"email_send_interval_seconds" },fhicl::Comment{
"Only send email every N seconds" }, 15 };
64 using Parameters = fhicl::WrappedTable<Config>;
67 #if MESSAGEFACILITY_HEX_VERSION < 0x20103 // v2_01_03 is s58, pre v2_01_03 is s50
68 ELSMTP(
const fhicl::ParameterSet& pset);
70 ELSMTP(Parameters
const& pset);
76 while (sending_thread_active_) usleep(1000);
79 virtual void routePayload(
const std::ostringstream&,
const ErrorObj& msg)
override;
83 std::string generateMessageId_()
const;
84 std::string dateTimeNow_();
85 std::string to_html(std::string msgString,
const ErrorObj& msg);
87 std::string smtp_host_;
89 std::vector<std::string> to_;
92 std::string message_prefix_;
96 std::string hostname_;
97 std::string hostaddr_;
101 std::string username_;
102 std::string password_;
103 bool ssl_verify_host_cert_;
105 std::atomic<bool> sending_thread_active_;
106 std::atomic<bool> abort_sleep_;
107 size_t send_interval_s_;
108 mutable std::mutex message_mutex_;
109 std::ostringstream message_contents_;
120 #if MESSAGEFACILITY_HEX_VERSION < 0x20103 // v2_01_03 is s58, pre v2_01_03 is s50
121 ELSMTP::ELSMTP(
const fhicl::ParameterSet& pset)
122 : ELdestination(pset)
123 , smtp_host_(pset.get<std::string>(
"host",
"smtp.fnal.gov"))
124 , port_(pset.get<int>(
"port", 25))
125 , to_(pset.get<std::vector<std::string>>(
"to_addresses"))
126 , from_(pset.get<std::string>(
"from_address"))
127 , subject_(pset.get<std::string>(
"subject",
"MessageFacility SMTP Message Digest"))
128 , message_prefix_(pset.get<std::string>(
"message_header",
""))
129 , pid_(static_cast<long>(getpid()))
130 , use_ssl_(pset.get<bool>(
"use_smtps", false))
131 , username_(pset.get<std::string>(
"smtp_username",
""))
132 , password_(pset.get<std::string>(
"smtp_password",
""))
133 , ssl_verify_host_cert_(pset.get<bool>(
"verify_host_ssl_certificate", true))
134 , sending_thread_active_(false)
135 , abort_sleep_(false)
136 , send_interval_s_(pset.get<size_t>(
"email_send_interval_seconds", 15))
139 : ELdestination(pset().elDestConfig())
140 , smtp_host_(pset().host())
141 , port_(pset().port())
143 , from_(pset().from())
144 , subject_(pset().subject())
145 , message_prefix_(pset().messageHeader())
146 , pid_(static_cast<long>(getpid()))
147 , use_ssl_(pset().useSmtps())
148 , username_(pset().user())
149 , password_(pset().pw())
150 , ssl_verify_host_cert_(pset().verifyCert())
151 , sending_thread_active_(false)
152 , abort_sleep_(false)
153 , send_interval_s_(pset().sendInterval())
157 char hostname_c[1024];
158 hostname_ = (gethostname(hostname_c, 1023) == 0) ? hostname_c :
"Unkonwn Host";
161 hostent* host =
nullptr;
162 host = gethostbyname(hostname_c);
167 char* ip = inet_ntoa(*(
struct in_addr *)host->h_addr);
173 struct ifaddrs* ifAddrStruct =
nullptr;
174 struct ifaddrs* ifa =
nullptr;
175 void* tmpAddrPtr =
nullptr;
177 if (getifaddrs(&ifAddrStruct))
180 hostaddr_ =
"127.0.0.1";
185 for (ifa = ifAddrStruct; ifa !=
nullptr; ifa = ifa->ifa_next)
187 if (ifa->ifa_addr->sa_family == AF_INET)
190 tmpAddrPtr = &((
struct sockaddr_in *)ifa->ifa_addr)->sin_addr;
191 char addressBuffer[INET_ADDRSTRLEN];
192 inet_ntop(AF_INET, tmpAddrPtr, addressBuffer, INET_ADDRSTRLEN);
193 hostaddr_ = addressBuffer;
196 else if (ifa->ifa_addr->sa_family == AF_INET6)
199 tmpAddrPtr = &((
struct sockaddr_in6 *)ifa->ifa_addr)->sin6_addr;
200 char addressBuffer[INET6_ADDRSTRLEN];
201 inet_ntop(AF_INET6, tmpAddrPtr, addressBuffer, INET6_ADDRSTRLEN);
202 hostaddr_ = addressBuffer;
206 if (!hostaddr_.empty()
207 && hostaddr_.compare(
"127.0.0.1")
208 && hostaddr_.compare(
"::1"))
212 if (hostaddr_.empty())
213 hostaddr_ =
"127.0.0.1";
219 std::ostringstream pid_ostr;
220 pid_ostr <<
"/proc/" << pid_ <<
"/exe";
221 exe = realpath(pid_ostr.str().c_str(), NULL);
223 size_t end = exe.find(
'\0');
224 size_t start = exe.find_last_of(
'/', end);
226 app_ = exe.substr(start + 1, end - start - 1);
230 std::string ELSMTP::to_html(std::string msgString,
const ErrorObj& msg)
232 auto sevid = msg.xid().severity().getLevel();
234 QString text_ = QString(
"<font color=");
238 case mf::ELseverityLevel::ELsev_success:
239 case mf::ELseverityLevel::ELsev_zeroSeverity:
240 case mf::ELseverityLevel::ELsev_unspecified:
241 text_ += QString(
"#505050>");
244 case mf::ELseverityLevel::ELsev_info:
245 text_ += QString(
"#008000>");
248 case mf::ELseverityLevel::ELsev_warning:
249 text_ += QString(
"#E08000>");
252 case mf::ELseverityLevel::ELsev_error:
253 case mf::ELseverityLevel::ELsev_severe:
254 case mf::ELseverityLevel::ELsev_highestSeverity:
255 text_ += QString(
"#FF0000>");
262 text_ += QString(
"<pre>")
263 + QString(msgString.c_str()).toHtmlEscaped()
266 text_ += QString(
"</font>");
267 return text_.toStdString();
273 void ELSMTP::routePayload(
const std::ostringstream& oss,
const ErrorObj& msg)
275 std::unique_lock<std::mutex>(message_mutex_);
276 message_contents_ << to_html(oss.str(), msg);
278 if (!sending_thread_active_)
280 sending_thread_active_ =
true;
281 boost::thread t([=] { send_message_(); });
286 void ELSMTP::send_message_()
289 while (!abort_sleep_ && slept < send_interval_s_ * 1000000)
297 std::unique_lock<std::mutex> lk(message_mutex_);
298 payload = message_contents_.str();
299 message_contents_.str(
"");
301 std::string destination = (use_ssl_ ?
"smtps://" :
"smtp://") + smtp_host_ +
":" + std::to_string(port_);
304 std::vector<const char*> to;
305 to.reserve(to_.size());
306 std::string toString;
307 for (
size_t i = 0; i < to_.size(); ++i)
309 to.push_back(to_[i].c_str());
311 if (i < to_.size() - 1)
317 std::ostringstream headers_builder;
319 headers_builder <<
"Date: " << dateTimeNow_() <<
"\r\n";
320 headers_builder <<
"To: " << toString <<
"\r\n";
321 headers_builder <<
"From: " << from_ <<
"\r\n";
322 headers_builder <<
"Message-ID: <" + generateMessageId_() +
"@" + from_.substr(from_.find(
'@') + 1) +
">\r\n";
323 headers_builder <<
"Subject: " << subject_ <<
" @ " << dateTimeNow_() <<
" from PID " << getpid() <<
"\r\n";
324 headers_builder <<
"Content-Type: text/html; charset=\"UTF-8\"\r\n";
325 headers_builder <<
"\r\n";
327 std::string headers = headers_builder.str();
328 std::ostringstream message_builder;
329 message_builder << headers <<
"<html><body><p>" << message_prefix_ <<
"</p>" << payload <<
"</body></html>";
330 std::string payloadWithHeaders = message_builder.str();
335 send_message_ssl(destination.c_str(), &to[0], to_.size(), from_.c_str(), payloadWithHeaders.c_str(), payloadWithHeaders.size(), username_.c_str(), password_.c_str(), !ssl_verify_host_cert_);
339 send_message(destination.c_str(), &to[0], to_.size(), from_.c_str(), payloadWithHeaders.c_str(), payloadWithHeaders.size());
341 sending_thread_active_ =
false;
345 std::string ELSMTP::generateMessageId_()
const
347 const size_t MESSAGE_ID_LEN = 37;
354 ret.resize(MESSAGE_ID_LEN);
355 size_t datelen = std::strftime(&ret[0], MESSAGE_ID_LEN,
"%Y%m%d%H%M%S", &t);
356 static const std::string alphaNum{
358 "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
359 "abcdefghijklmnopqrstuvwxyz" };
361 std::uniform_int_distribution<> dis(0, alphaNum.length() - 1);
362 std::generate_n(ret.begin() + datelen, MESSAGE_ID_LEN - datelen, [&]() {
return alphaNum[dis(gen)]; });
366 std::string ELSMTP::dateTimeNow_()
368 const int RFC5322_TIME_LEN = 32;
371 ret.resize(RFC5322_TIME_LEN);
378 strftime(&ret[0], RFC5322_TIME_LEN,
"%a, %d %b %Y %H:%M:%S %z", t);
390 #ifndef EXTERN_C_FUNC_DECLARE_START
391 #define EXTERN_C_FUNC_DECLARE_START extern "C" {
394 EXTERN_C_FUNC_DECLARE_START
395 auto makePlugin(
const std::string&,
396 const fhicl::ParameterSet& pset)
398 return std::make_unique<mfplugins::ELSMTP>(pset);
402 DEFINE_BASIC_PLUGINTYPE_FUNC(mf::service::ELdestination)
void send_message_ssl(const char *dest, const char *to[], size_t to_size, const char *from, const char *payload, size_t payload_size, const char *username, const char *pw, int disableVerify)
Sends a message to the given SMTP server, using SSL encryption.
SMTP Message Facility destination plugin (Using libcurl)
void send_message(const char *dest, const char *to[], size_t to_size, const char *from, const char *payload, size_t payload_size)
Sends a message to the given SMTP server.