1 /**
2 * collectd - src/logfile.c
3 * Copyright (C) 2007 Sebastian Harl
4 * Copyright (C) 2007,2008 Florian Forster
5 *
6 * This program is free software; you can redistribute it and/or modify it
7 * under the terms of the GNU General Public License as published by the
8 * Free Software Foundation; only version 2 of the License is applicable.
9 *
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.
14 *
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
18 *
19 * Authors:
20 * Sebastian Harl <sh at tokkee.org>
21 * Florian Forster <octo at verplant.org>
22 **/
24 #include "collectd.h"
25 #include "common.h"
26 #include "plugin.h"
28 #include <pthread.h>
30 #define DEFAULT_LOGFILE LOCALSTATEDIR"/log/collectd.log"
32 #if COLLECT_DEBUG
33 static int log_level = LOG_DEBUG;
34 #else
35 static int log_level = LOG_INFO;
36 #endif /* COLLECT_DEBUG */
38 static pthread_mutex_t file_lock = PTHREAD_MUTEX_INITIALIZER;
40 static char *log_file = NULL;
41 static int print_timestamp = 1;
43 static const char *config_keys[] =
44 {
45 "LogLevel",
46 "File",
47 "Timestamp"
48 };
49 static int config_keys_num = STATIC_ARRAY_SIZE (config_keys);
51 static int logfile_config (const char *key, const char *value)
52 {
53 if (0 == strcasecmp (key, "LogLevel")) {
54 if ((0 == strcasecmp (value, "emerg"))
55 || (0 == strcasecmp (value, "alert"))
56 || (0 == strcasecmp (value, "crit"))
57 || (0 == strcasecmp (value, "err")))
58 log_level = LOG_ERR;
59 else if (0 == strcasecmp (value, "warning"))
60 log_level = LOG_WARNING;
61 else if (0 == strcasecmp (value, "notice"))
62 log_level = LOG_NOTICE;
63 else if (0 == strcasecmp (value, "info"))
64 log_level = LOG_INFO;
65 #if COLLECT_DEBUG
66 else if (0 == strcasecmp (value, "debug"))
67 log_level = LOG_DEBUG;
68 #endif /* COLLECT_DEBUG */
69 else
70 return 1;
71 }
72 else if (0 == strcasecmp (key, "File")) {
73 sfree (log_file);
74 log_file = strdup (value);
75 }
76 else if (0 == strcasecmp (key, "Timestamp")) {
77 if (IS_FALSE (value))
78 print_timestamp = 0;
79 else
80 print_timestamp = 1;
81 }
82 else {
83 return -1;
84 }
85 return 0;
86 } /* int logfile_config (const char *, const char *) */
88 static void logfile_print (const char *msg, time_t timestamp_time)
89 {
90 FILE *fh;
91 int do_close = 0;
92 struct tm timestamp_tm;
93 char timestamp_str[64];
95 if (print_timestamp)
96 {
97 localtime_r (×tamp_time, ×tamp_tm);
99 strftime (timestamp_str, sizeof (timestamp_str), "%Y-%m-%d %H:%M:%S",
100 ×tamp_tm);
101 timestamp_str[sizeof (timestamp_str) - 1] = '\0';
102 }
104 pthread_mutex_lock (&file_lock);
106 if (log_file == NULL)
107 {
108 fh = fopen (DEFAULT_LOGFILE, "a");
109 do_close = 1;
110 }
111 else if (strcasecmp (log_file, "stderr") == 0)
112 fh = stderr;
113 else if (strcasecmp (log_file, "stdout") == 0)
114 fh = stdout;
115 else
116 {
117 fh = fopen (log_file, "a");
118 do_close = 1;
119 }
121 if (fh == NULL)
122 {
123 char errbuf[1024];
124 fprintf (stderr, "logfile plugin: fopen (%s) failed: %s\n",
125 (log_file == NULL) ? DEFAULT_LOGFILE : log_file,
126 sstrerror (errno, errbuf, sizeof (errbuf)));
127 }
128 else
129 {
130 if (print_timestamp)
131 fprintf (fh, "[%s] %s\n", timestamp_str, msg);
132 else
133 fprintf (fh, "%s\n", msg);
135 if (do_close != 0)
136 fclose (fh);
137 }
139 pthread_mutex_unlock (&file_lock);
141 return;
142 } /* void logfile_print */
144 static void logfile_log (int severity, const char *msg,
145 user_data_t __attribute__((unused)) *user_data)
146 {
147 if (severity > log_level)
148 return;
150 logfile_print (msg, time (NULL));
151 } /* void logfile_log (int, const char *) */
153 static int logfile_notification (const notification_t *n,
154 user_data_t __attribute__((unused)) *user_data)
155 {
156 char buf[1024] = "";
157 char *buf_ptr = buf;
158 int buf_len = sizeof (buf);
159 int status;
161 status = ssnprintf (buf_ptr, buf_len, "Notification: severity = %s",
162 (n->severity == NOTIF_FAILURE) ? "FAILURE"
163 : ((n->severity == NOTIF_WARNING) ? "WARNING"
164 : ((n->severity == NOTIF_OKAY) ? "OKAY" : "UNKNOWN")));
165 if (status > 0)
166 {
167 buf_ptr += status;
168 buf_len -= status;
169 }
171 #define APPEND(bufptr, buflen, key, value) \
172 if ((buflen > 0) && (strlen (value) > 0)) { \
173 int status = ssnprintf (bufptr, buflen, ", %s = %s", key, value); \
174 if (status > 0) { \
175 bufptr += status; \
176 buflen -= status; \
177 } \
178 }
179 APPEND (buf_ptr, buf_len, "host", n->host);
180 APPEND (buf_ptr, buf_len, "plugin", n->plugin);
181 APPEND (buf_ptr, buf_len, "plugin_instance", n->plugin_instance);
182 APPEND (buf_ptr, buf_len, "type", n->type);
183 APPEND (buf_ptr, buf_len, "type_instance", n->type_instance);
184 APPEND (buf_ptr, buf_len, "message", n->message);
186 buf[sizeof (buf) - 1] = '\0';
188 logfile_print (buf,
189 (n->time > 0) ? n->time : time (NULL));
191 return (0);
192 } /* int logfile_notification */
194 void module_register (void)
195 {
196 plugin_register_config ("logfile", logfile_config,
197 config_keys, config_keys_num);
198 plugin_register_log ("logfile", logfile_log, /* user_data = */ NULL);
199 plugin_register_notification ("logfile", logfile_notification,
200 /* user_data = */ NULL);
201 } /* void module_register (void) */
203 /* vim: set sw=4 ts=4 tw=78 noexpandtab : */