2 * collectd - src/notify_email.c
3 * Copyright (C) 2008 Oleg King
5 * This program is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License as published by the
7 * Free Software Foundation; either version 2 of the License, or (at your
8 * option) any later version.
10 * This program is distributed in the hope that it will be useful, but
11 * WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * General Public License for more details.
15 * You should have received a copy of the GNU General Public License along
16 * with this program; if not, write to the Free Software Foundation, Inc.,
17 * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
20 * Oleg King <king2 at kaluga.ru>
27 #include <auth-client.h>
32 static const char *config_keys[] =
42 static int config_keys_num = STATIC_ARRAY_SIZE (config_keys);
45 static int emails_len = 0;
47 static smtp_session_t session;
48 static smtp_message_t message;
49 static auth_context_t authctx = NULL;
51 static int smtp_port = 25;
52 static char *smtp_host = NULL;
53 static char *smtp_user = NULL;
54 static char *smtp_password = NULL;
55 static char *smtp_from = NULL;
56 static char *smtp_subject = NULL;
58 #define DEFAULT_SMTP_HOST "localhost"
59 #define DEFAULT_SMTP_FROM "root@localhost"
60 #define DEFAULT_SMTP_SUBJECT "Collectd notify: %s@%s"
63 /* Callback to get username and password */
64 static int authinteract (auth_client_request_t request, char **result, int fields, void *arg)
67 for (i = 0; i < fields; i++)
69 if (request[i].flags & AUTH_USER)
70 result[i] = smtp_user;
71 else if (request[i].flags & AUTH_PASS)
72 result[i] = smtp_password;
79 /* Callback to print the recipient status */
80 static void print_recipient_status (smtp_recipient_t recipient, const char *mailbox, void *arg)
82 const smtp_status_t *status;
84 status = smtp_recipient_status (recipient);
85 if (status->text[strlen(status->text) - 2] == '\r')
86 status->text[strlen(status->text) - 2] = 0;
87 INFO ("notify_email: notify sent to %s: %d %s", mailbox, status->code, status->text);
90 /* Callback to monitor SMTP activity */
91 static void monitor_cb (const char *buf, int buflen, int writing, void *arg)
93 char log_str[MAXSTRING];
95 sstrncpy (log_str, buf, sizeof (log_str));
97 log_str[buflen - 2] = 0; /* replace \n with \0 */
99 if (writing == SMTP_CB_HEADERS) {
100 DEBUG ("SMTP --- H: %s", log_str);
103 DEBUG (writing ? "SMTP >>> C: %s" : "SMTP <<< S: %s", log_str);
106 static int notify_email_init()
108 char server[MAXSTRING];
111 if ( !(session = smtp_create_session()) ) {
112 ERROR ("notify_email plugin: cannot create SMTP session");
116 smtp_set_monitorcb (session, monitor_cb, NULL, 1);
117 smtp_set_hostname (session, hostname_g);
118 ssnprintf(server, sizeof (server), "%s:%i",
119 (smtp_host == NULL) ? DEFAULT_SMTP_HOST : smtp_host,
121 smtp_set_server (session, server);
123 if (smtp_user && smtp_password) {
124 authctx = auth_create_context ();
125 auth_set_mechanism_flags (authctx, AUTH_PLUGIN_PLAIN, 0);
126 auth_set_interact_cb (authctx, authinteract, NULL);
129 if ( !smtp_auth_set_context (session, authctx)) {
130 ERROR ("notify_email plugin: cannot set SMTP auth context");
138 static int notify_email_shutdown()
140 smtp_destroy_session (session);
141 auth_destroy_context (authctx);
147 static int notify_email_config (const char *key, const char *value)
149 if (strcasecmp (key, "EmailTo") == 0)
153 tmp = (char **) realloc ((void *) emails, (emails_len + 1) * sizeof (char *));
155 ERROR ("notify_email: realloc failed.");
160 emails[emails_len] = strdup (value);
161 if (emails[emails_len] == NULL) {
162 ERROR ("notify_email: strdup failed.");
167 else if (0 == strcasecmp (key, "SMTPHost")) {
169 smtp_host = strdup (value);
171 else if (0 == strcasecmp (key, "SMTPPort")) {
172 int port_tmp = atoi (value);
173 if (port_tmp < 1 || port_tmp > 65535)
175 WARNING ("notify_email plugin: Invalid SMTP port: %i", port_tmp);
178 smtp_port = port_tmp;
180 else if (0 == strcasecmp (key, "SMTPUser")) {
182 smtp_user = strdup (value);
184 else if (0 == strcasecmp (key, "SMTPPassword")) {
185 sfree (smtp_password);
186 smtp_password = strdup (value);
188 else if (0 == strcasecmp (key, "SMTPFrom")) {
190 smtp_from = strdup (value);
192 else if (0 == strcasecmp (key, "SMTPSubject")) {
193 sfree (smtp_subject);
194 smtp_subject = strdup (value);
200 } /* int notify_email_config (const char *, const char *) */
203 static int notify_email_notification (const notification_t *n)
205 smtp_recipient_t recipient;
207 struct tm timestamp_tm;
208 char timestamp_str[64];
211 char subject[MAXSTRING];
214 int buf_len = sizeof (buf);
217 ssnprintf (severity, sizeof (severity), "%s",
218 (n->severity == NOTIF_FAILURE) ? "FAILURE"
219 : ((n->severity == NOTIF_WARNING) ? "WARNING"
220 : ((n->severity == NOTIF_OKAY) ? "OKAY" : "UNKNOWN")));
222 ssnprintf (subject, sizeof (subject),
223 (smtp_subject == NULL) ? DEFAULT_SMTP_SUBJECT : smtp_subject,
226 localtime_r (&n->time, ×tamp_tm);
227 strftime (timestamp_str, sizeof (timestamp_str), "%Y-%m-%d %H:%M:%S", ×tamp_tm);
228 timestamp_str[sizeof (timestamp_str) - 1] = '\0';
230 /* Let's make RFC822 message text with \r\n EOLs */
231 ssnprintf (buf, buf_len,
232 "MIME-Version: 1.0\r\n"
233 "Content-Type: text/plain;\r\n"
234 "Content-Transfer-Encoding: 8bit\r\n"
246 if ( !(message = smtp_add_message (session))) {
247 ERROR ("notify_email plugin: cannot set SMTP message");
250 smtp_set_reverse_path (message, smtp_from);
251 smtp_set_header (message, "To", NULL, NULL);
252 smtp_set_message_str (message, buf);
254 for (i = 0; i < emails_len; i++)
255 recipient = smtp_add_recipient (message, emails[i]);
257 /* Initiate a connection to the SMTP server and transfer the message. */
258 if (!smtp_start_session (session)) {
260 ERROR ("SMTP server problem: %s", smtp_strerror (smtp_errno (), buf, sizeof buf));
263 const smtp_status_t *status;
264 /* Report on the success or otherwise of the mail transfer. */
265 status = smtp_message_transfer_status (message);
266 DEBUG ("SMTP server report: %d %s", status->code, (status->text != NULL) ? status->text : "\n");
267 smtp_enumerate_recipients (message, print_recipient_status, NULL);
271 } /* int notify_email_notification */
273 void module_register (void)
275 plugin_register_init ("notify_email", notify_email_init);
276 plugin_register_shutdown ("notify_email", notify_email_shutdown);
277 plugin_register_config ("notify_email", notify_email_config,
278 config_keys, config_keys_num);
279 plugin_register_notification ("notify_email", notify_email_notification);
280 } /* void module_register (void) */