1 #include "cetlib/PluginTypeDeducer.h"
2 #include "fhiclcpp/ParameterSet.h"
3 #include "fhiclcpp/types/ConfigurationTable.h"
4 #include "fhiclcpp/types/Sequence.h"
5 #include "fhiclcpp/types/TableFragment.h"
7 #include "messagefacility/MessageService/ELdestination.h"
8 #include "messagefacility/Utilities/ELseverityLevel.h"
9 #if MESSAGEFACILITY_HEX_VERSION < 0x20201 // v2_02_01 is s67
10 #include "messagefacility/MessageService/MessageDrop.h"
12 #include "messagefacility/MessageLogger/MessageLogger.h"
14 #include "messagefacility/Utilities/exception.h"
17 #include <arpa/inet.h>
20 #include <netinet/in.h>
23 #include <boost/thread.hpp>
28 #include <QtCore/QString>
29 #include "cetlib/compiler_macros.h"
33 using mf::ELseverityLevel;
35 using mf::service::ELdestination;
49 using strings_t = fhicl::Sequence<std::string>::default_type;
53 fhicl::Atom<std::string>
host =
54 fhicl::Atom<std::string>{fhicl::Name{
"host"}, fhicl::Comment{
"SMTP Server hostname"},
"smtp.fnal.gov"};
56 fhicl::Atom<int>
port = fhicl::Atom<int>{fhicl::Name{
"port"}, fhicl::Comment{
"SMTP Server port"}, 25};
58 fhicl::Sequence<std::string>
to = fhicl::Sequence<std::string>{
59 fhicl::Name{
"to_addresses"}, fhicl::Comment{
"The list of email addresses that SMTP mfPlugin should sent to"},
62 fhicl::Atom<std::string>
from =
63 fhicl::Atom<std::string>{fhicl::Name{
"from_address"}, fhicl::Comment{
"Source email address"}};
65 fhicl::Atom<std::string>
subject = fhicl::Atom<std::string>{
66 fhicl::Name{
"subject"}, fhicl::Comment{
"Subject of the email message"},
"MessageFacility SMTP Message Digest"};
69 fhicl::Name{
"message_header"}, fhicl::Comment{
"String to preface messages with in email body"},
""};
72 fhicl::Atom<bool>{fhicl::Name{
"use_smtps"}, fhicl::Comment{
"Use SMTPS protocol"},
false};
74 fhicl::Atom<std::string>
user =
75 fhicl::Atom<std::string>{fhicl::Name{
"smtp_username"}, fhicl::Comment{
"Username for SMTP server"},
""};
77 fhicl::Atom<std::string>
pw =
78 fhicl::Atom<std::string>{fhicl::Name{
"smtp_password"}, fhicl::Comment{
"Password for SMTP server"},
""};
81 fhicl::Atom<bool>{fhicl::Name{
"verify_host_ssl_certificate"},
82 fhicl::Comment{
"Whether to run full SSL verify on SMTP server in SMTPS mode"},
true};
84 fhicl::Atom<size_t>
sendInterval = fhicl::Atom<size_t>{fhicl::Name{
"email_send_interval_seconds"},
85 fhicl::Comment{
"Only send email every N seconds"}, 15};
100 while (sending_thread_active_) usleep(1000);
108 virtual void routePayload(
const std::ostringstream& o,
const ErrorObj& msg)
override;
111 void send_message_();
112 std::string generateMessageId_()
const;
113 std::string dateTimeNow_();
114 std::string to_html(std::string msgString,
const ErrorObj& msg);
116 std::string smtp_host_;
118 std::vector<std::string> to_;
120 std::string subject_;
121 std::string message_prefix_;
125 std::string hostname_;
126 std::string hostaddr_;
130 std::string username_;
131 std::string password_;
132 bool ssl_verify_host_cert_;
134 std::atomic<bool> sending_thread_active_;
135 std::atomic<bool> abort_sleep_;
136 size_t send_interval_s_;
137 mutable std::mutex message_mutex_;
138 std::ostringstream message_contents_;
149 : ELdestination(pset().elDestConfig()), smtp_host_(pset().host()), port_(pset().port()), to_(pset().to()), from_(pset().from()), subject_(pset().subject()), message_prefix_(pset().messageHeader()), pid_(static_cast<long>(getpid())), use_ssl_(pset().useSmtps()), username_(pset().user()), password_(pset().pw()), ssl_verify_host_cert_(pset().verifyCert()), sending_thread_active_(false), abort_sleep_(false), send_interval_s_(pset().sendInterval())
152 char hostname_c[1024];
153 hostname_ = (gethostname(hostname_c, 1023) == 0) ? hostname_c :
"Unkonwn Host";
156 hostent* host =
nullptr;
157 host = gethostbyname(hostname_c);
162 char* ip = inet_ntoa(*(
struct in_addr*)host->h_addr);
168 struct ifaddrs* ifAddrStruct =
nullptr;
169 struct ifaddrs* ifa =
nullptr;
170 void* tmpAddrPtr =
nullptr;
172 if (getifaddrs(&ifAddrStruct))
175 hostaddr_ =
"127.0.0.1";
180 for (ifa = ifAddrStruct; ifa !=
nullptr; ifa = ifa->ifa_next)
182 if (ifa->ifa_addr->sa_family == AF_INET)
185 tmpAddrPtr = &((
struct sockaddr_in*)ifa->ifa_addr)->sin_addr;
186 char addressBuffer[INET_ADDRSTRLEN];
187 inet_ntop(AF_INET, tmpAddrPtr, addressBuffer, INET_ADDRSTRLEN);
188 hostaddr_ = addressBuffer;
191 else if (ifa->ifa_addr->sa_family == AF_INET6)
194 tmpAddrPtr = &((
struct sockaddr_in6*)ifa->ifa_addr)->sin6_addr;
195 char addressBuffer[INET6_ADDRSTRLEN];
196 inet_ntop(AF_INET6, tmpAddrPtr, addressBuffer, INET6_ADDRSTRLEN);
197 hostaddr_ = addressBuffer;
201 if (!hostaddr_.empty() && hostaddr_.compare(
"127.0.0.1") && hostaddr_.compare(
"::1"))
break;
204 if (hostaddr_.empty())
205 hostaddr_ =
"127.0.0.1";
211 std::ostringstream pid_ostr;
212 pid_ostr <<
"/proc/" << pid_ <<
"/exe";
213 exe = realpath(pid_ostr.str().c_str(), NULL);
215 size_t end = exe.find(
'\0');
216 size_t start = exe.find_last_of(
'/', end);
218 app_ = exe.substr(start + 1, end - start - 1);
221 std::string ELSMTP::to_html(std::string msgString,
const ErrorObj& msg)
223 auto sevid = msg.xid().severity().getLevel();
225 QString text_ = QString(
"<font color=");
229 case mf::ELseverityLevel::ELsev_success:
230 case mf::ELseverityLevel::ELsev_zeroSeverity:
231 case mf::ELseverityLevel::ELsev_unspecified:
232 text_ += QString(
"#505050>");
235 case mf::ELseverityLevel::ELsev_info:
236 text_ += QString(
"#008000>");
239 case mf::ELseverityLevel::ELsev_warning:
240 text_ += QString(
"#E08000>");
243 case mf::ELseverityLevel::ELsev_error:
244 case mf::ELseverityLevel::ELsev_severe:
245 case mf::ELseverityLevel::ELsev_highestSeverity:
246 text_ += QString(
"#FF0000>");
254 text_ += QString(
"<pre>") + QString(msgString.c_str()).toHtmlEscaped()
257 text_ += QString(
"</font>");
258 return text_.toStdString();
266 std::lock_guard<std::mutex> lk(message_mutex_);
267 message_contents_ << to_html(oss.str(), msg);
269 if (!sending_thread_active_)
271 sending_thread_active_ =
true;
272 boost::thread t([=] { send_message_(); });
277 void ELSMTP::send_message_()
280 while (!abort_sleep_ && slept < send_interval_s_ * 1000000)
288 std::lock_guard<std::mutex> lk(message_mutex_);
289 payload = message_contents_.str();
290 message_contents_.str(
"");
292 std::string destination = (use_ssl_ ?
"smtps://" :
"smtp://") + smtp_host_ +
":" + std::to_string(port_);
294 std::vector<const char*> to;
295 to.reserve(to_.size());
296 std::string toString;
297 for (
size_t i = 0; i < to_.size(); ++i)
299 to.push_back(to_[i].c_str());
301 if (i < to_.size() - 1)
307 std::ostringstream headers_builder;
309 headers_builder <<
"Date: " << dateTimeNow_() <<
"\r\n";
310 headers_builder <<
"To: " << toString <<
"\r\n";
311 headers_builder <<
"From: " << from_ <<
"\r\n";
312 headers_builder <<
"Message-ID: <" + generateMessageId_() +
"@" + from_.substr(from_.find(
'@') + 1) +
">\r\n";
313 headers_builder <<
"Subject: " << subject_ <<
" @ " << dateTimeNow_() <<
" from PID " << getpid() <<
"\r\n";
314 headers_builder <<
"Content-Type: text/html; charset=\"UTF-8\"\r\n";
315 headers_builder <<
"\r\n";
317 std::string headers = headers_builder.str();
318 std::ostringstream message_builder;
319 message_builder << headers <<
"<html><body><p>" << message_prefix_ <<
"</p>" << payload <<
"</body></html>";
320 std::string payloadWithHeaders = message_builder.str();
324 send_message_ssl(destination.c_str(), &to[0], to_.size(), from_.c_str(), payloadWithHeaders.c_str(),
325 payloadWithHeaders.size(), username_.c_str(), password_.c_str(), !ssl_verify_host_cert_);
329 send_message(destination.c_str(), &to[0], to_.size(), from_.c_str(), payloadWithHeaders.c_str(),
330 payloadWithHeaders.size());
332 sending_thread_active_ =
false;
336 std::string ELSMTP::generateMessageId_()
const
338 const size_t MESSAGE_ID_LEN = 37;
345 ret.resize(MESSAGE_ID_LEN);
346 size_t datelen = std::strftime(&ret[0], MESSAGE_ID_LEN,
"%Y%m%d%H%M%S", &t);
347 static const std::string alphaNum{
349 "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
350 "abcdefghijklmnopqrstuvwxyz"};
352 std::uniform_int_distribution<> dis(0, alphaNum.length() - 1);
353 std::generate_n(ret.begin() + datelen, MESSAGE_ID_LEN - datelen, [&]() {
return alphaNum[dis(gen)]; });
357 std::string ELSMTP::dateTimeNow_()
359 const int RFC5322_TIME_LEN = 32;
362 ret.resize(RFC5322_TIME_LEN);
369 strftime(&ret[0], RFC5322_TIME_LEN,
"%a, %d %b %Y %H:%M:%S %z", t);
381 #ifndef EXTERN_C_FUNC_DECLARE_START
382 #define EXTERN_C_FUNC_DECLARE_START extern "C" {
385 EXTERN_C_FUNC_DECLARE_START
386 auto makePlugin(
const std::string&,
const fhicl::ParameterSet& pset)
388 return std::make_unique<mfplugins::ELSMTP>(pset);
392 DEFINE_BASIC_PLUGINTYPE_FUNC(mf::service::ELdestination)
fhicl::Atom< std::string > from
"from_address" (REQUIRED): Source email address
fhicl::Atom< std::string > user
"smtp_username" (Default: ""): Username for SMTP server
fhicl::Sequence< std::string > to
"to_addresses" (Default: {}): The list of email addresses that SMTP mfPlugin should sent to ...
fhicl::Atom< size_t > sendInterval
"email_send_interval_seconds" (Default: 15): Only send email every N seconds
fhicl::WrappedTable< Config > Parameters
Used for ParameterSet validation.
fhicl::Atom< std::string > host
"host" (Default: "smtp.fnal.gov"): SMTP Server hostname
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.
fhicl::Sequence< std::string >::default_type strings_t
Array of strings.
SMTP Message Facility destination plugin (Using libcurl)
fhicl::Atom< std::string > messageHeader
"message_header" (Default: ""): String to preface messages with in email body
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.
fhicl::Atom< std::string > subject
"subject" (Default: "MessageFacility SMTP Message Digest"): Subject of the email message ...
virtual void routePayload(const std::ostringstream &o, const ErrorObj &msg) override
Serialize a MessageFacility message to the output.
fhicl::TableFragment< ELdestination::Config > elDestConfig
ELDestination common config parameters.
fhicl::Atom< bool > verifyCert
"verify_host_ssl_certificate" (Default: true): Whether to run full SSL verify on SMTP server in SMTPS...
ELSMTP(Parameters const &pset)
ELSMTP Constructor
fhicl::Atom< std::string > pw
"smtp_password" (Default: ""): Password for SMTP server
fhicl::Atom< bool > useSmtps
"use_smtps" (Default: false): Use SMTPS protocol
Configuration parameters for ELSMTP.
fhicl::Atom< int > port
"port" (Default: 25): SMTP Server port