multimeter plugin: Check the return value of `write'.
[collectd.git] / src / utils_cmd_putnotif.c
1 /**
2  * collectd - src/utils_cms_putnotif.c
3  * Copyright (C) 2008  Florian octo Forster
4  *
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; only version 2 of the License is applicable.
8  *
9  * This program is distributed in the hope that it will be useful, but
10  * WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License along
15  * with this program; if not, write to the Free Software Foundation, Inc.,
16  * 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
17  *
18  * Author:
19  *   Florian octo Forster <octo at verplant.org>
20  **/
21
22 #include "collectd.h"
23 #include "common.h"
24 #include "plugin.h"
25
26 #define print_to_socket(fh, ...) \
27   if (fprintf (fh, __VA_ARGS__) < 0) { \
28     char errbuf[1024]; \
29     WARNING ("handle_putnotif: failed to write to socket #%i: %s", \
30         fileno (fh), sstrerror (errno, errbuf, sizeof (errbuf))); \
31     return -1; \
32   }
33
34 static int parse_option_severity (notification_t *n, char *value)
35 {
36   if (strcasecmp (value, "Failure") == 0)
37     n->severity = NOTIF_FAILURE;
38   else if (strcasecmp (value, "Warning") == 0)
39     n->severity = NOTIF_WARNING;
40   else if (strcasecmp (value, "Okay") == 0)
41     n->severity = NOTIF_OKAY;
42   else
43     return (-1);
44
45   return (0);
46 } /* int parse_option_severity */
47
48 static int parse_option_time (notification_t *n, char *value)
49 {
50   time_t tmp;
51   
52   tmp = (time_t) atoi (value);
53   if (tmp <= 0)
54     return (-1);
55
56   n->time = tmp;
57
58   return (0);
59 } /* int parse_option_time */
60
61 static int parse_option (notification_t *n, char *buffer)
62 {
63   char *option = buffer;
64   char *value;
65
66   if ((n == NULL) || (option == NULL))
67     return (-1);
68
69   value = strchr (option, '=');
70   if (value == NULL)
71     return (-1);
72   *value = '\0'; value++;
73
74   if (strcasecmp ("severity", option) == 0)
75     return (parse_option_severity (n, value));
76   else if (strcasecmp ("time", option) == 0)
77     return (parse_option_time (n, value));
78   else if (strcasecmp ("host", option) == 0)
79     sstrncpy (n->host, value, sizeof (n->host));
80   else if (strcasecmp ("plugin", option) == 0)
81     sstrncpy (n->plugin, value, sizeof (n->plugin));
82   else if (strcasecmp ("plugin_instance", option) == 0)
83     sstrncpy (n->plugin_instance, value, sizeof (n->plugin_instance));
84   else if (strcasecmp ("type", option) == 0)
85     sstrncpy (n->type, value, sizeof (n->type));
86   else if (strcasecmp ("type_instance", option) == 0)
87     sstrncpy (n->type_instance, value, sizeof (n->type_instance));
88   else
89     return (1);
90
91   return (0);
92 } /* int parse_option */
93
94 static int parse_message (notification_t *n, char **fields, int fields_num)
95 {
96   int status;
97
98   /* Strip off the leading `message=' */
99   fields[0] += strlen ("message=");
100
101   status = strjoin (n->message, sizeof (n->message), fields, fields_num, " ");
102   if (status < 0)
103     return (-1);
104
105   return (0);
106 } /* int parse_message */
107
108 int handle_putnotif (FILE *fh, char **fields, int fields_num)
109 {
110   notification_t n;
111   int status;
112   int i;
113
114   /* Required fields: `PUTNOTIF', severity, time, message */
115   if (fields_num < 4)
116   {
117     DEBUG ("cmd putnotif: Wrong number of fields: %i", fields_num);
118     print_to_socket (fh, "-1 Wrong number of fields: Got %i, "
119         "expected at least 4.\n",
120         fields_num);
121     return (-1);
122   }
123
124   memset (&n, '\0', sizeof (n));
125
126   status = 0;
127   for (i = 1; i < fields_num; i++)
128   {
129     if (strncasecmp (fields[i], "message=", strlen ("message=")) == 0)
130     {
131       status = parse_message (&n, fields + i, fields_num - i);
132       if (status != 0)
133       {
134         print_to_socket (fh, "-1 Error parsing the message. Have you hit the "
135             "limit of %u bytes?\n", (unsigned int) sizeof (n.message));
136       }
137       break;
138     }
139     else
140     {
141       status = parse_option (&n, fields[i]);
142       if (status != 0)
143       {
144         print_to_socket (fh, "-1 Error parsing option `%s'\n", fields[i]);
145         break;
146       }
147     }
148   } /* for (i) */
149
150   /* Check for required fields and complain if anything is missing. */
151   if ((status == 0) && (n.severity == 0))
152   {
153     print_to_socket (fh, "-1 Option `severity' missing.\n");
154     status = -1;
155   }
156   if ((status == 0) && (n.time == 0))
157   {
158     print_to_socket (fh, "-1 Option `time' missing.\n");
159     status = -1;
160   }
161   if ((status == 0) && (strlen (n.message) == 0))
162   {
163     print_to_socket (fh, "-1 No message or message of length 0 given.\n");
164     status = -1;
165   }
166
167   /* If status is still zero the notification is fine and we can finally
168    * dispatch it. */
169   if (status == 0)
170   {
171     plugin_dispatch_notification (&n);
172     print_to_socket (fh, "0 Success\n");
173   }
174
175   return (0);
176 } /* int handle_putnotif */
177
178 /* vim: set shiftwidth=2 softtabstop=2 tabstop=8 : */