Merge remote branch 'trenkel/collectd-4.9' into collectd-4.9
[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 #include "utils_parse_option.h"
27
28 #define print_to_socket(fh, ...) \
29   if (fprintf (fh, __VA_ARGS__) < 0) { \
30     char errbuf[1024]; \
31     WARNING ("handle_putnotif: failed to write to socket #%i: %s", \
32         fileno (fh), sstrerror (errno, errbuf, sizeof (errbuf))); \
33     return -1; \
34   }
35
36 static int set_option_severity (notification_t *n, const char *value)
37 {
38   if (strcasecmp (value, "Failure") == 0)
39     n->severity = NOTIF_FAILURE;
40   else if (strcasecmp (value, "Warning") == 0)
41     n->severity = NOTIF_WARNING;
42   else if (strcasecmp (value, "Okay") == 0)
43     n->severity = NOTIF_OKAY;
44   else
45     return (-1);
46
47   return (0);
48 } /* int set_option_severity */
49
50 static int set_option_time (notification_t *n, const char *value)
51 {
52   time_t tmp;
53   
54   tmp = (time_t) atoi (value);
55   if (tmp <= 0)
56     return (-1);
57
58   n->time = tmp;
59
60   return (0);
61 } /* int set_option_time */
62
63 static int set_option (notification_t *n, const char *option, const char *value)
64 {
65   if ((n == NULL) || (option == NULL) || (value == NULL))
66     return (-1);
67
68   DEBUG ("utils_cmd_putnotif: set_option (option = %s, value = %s);",
69       option, value);
70
71   if (strcasecmp ("severity", option) == 0)
72     return (set_option_severity (n, value));
73   else if (strcasecmp ("time", option) == 0)
74     return (set_option_time (n, value));
75   else if (strcasecmp ("message", option) == 0)
76     sstrncpy (n->message, value, sizeof (n->message));
77   else if (strcasecmp ("host", option) == 0)
78     sstrncpy (n->host, value, sizeof (n->host));
79   else if (strcasecmp ("plugin", option) == 0)
80     sstrncpy (n->plugin, value, sizeof (n->plugin));
81   else if (strcasecmp ("plugin_instance", option) == 0)
82     sstrncpy (n->plugin_instance, value, sizeof (n->plugin_instance));
83   else if (strcasecmp ("type", option) == 0)
84     sstrncpy (n->type, value, sizeof (n->type));
85   else if (strcasecmp ("type_instance", option) == 0)
86     sstrncpy (n->type_instance, value, sizeof (n->type_instance));
87   else
88     return (1);
89
90   return (0);
91 } /* int set_option */
92
93 int handle_putnotif (FILE *fh, char *buffer)
94 {
95   char *command;
96   notification_t n;
97   int status;
98
99   if ((fh == NULL) || (buffer == NULL))
100     return (-1);
101
102   DEBUG ("utils_cmd_putnotif: handle_putnotif (fh = %p, buffer = %s);",
103       (void *) fh, buffer);
104
105   command = NULL;
106   status = parse_string (&buffer, &command);
107   if (status != 0)
108   {
109     print_to_socket (fh, "-1 Cannot parse command.\n");
110     return (-1);
111   }
112   assert (command != NULL);
113
114   if (strcasecmp ("PUTNOTIF", command) != 0)
115   {
116     print_to_socket (fh, "-1 Unexpected command: `%s'.\n", command);
117     return (-1);
118   }
119
120   memset (&n, '\0', sizeof (n));
121
122   status = 0;
123   while (*buffer != 0)
124   {
125     char *key;
126     char *value;
127
128     status = parse_option (&buffer, &key, &value);
129     if (status != 0)
130     {
131       print_to_socket (fh, "-1 Malformed option.\n");
132       break;
133     }
134
135     status = set_option (&n, key, value);
136     if (status != 0)
137     {
138       print_to_socket (fh, "-1 Error parsing option `%s'\n", key);
139       break;
140     }
141   } /* for (i) */
142
143   /* Check for required fields and complain if anything is missing. */
144   if ((status == 0) && (n.severity == 0))
145   {
146     print_to_socket (fh, "-1 Option `severity' missing.\n");
147     status = -1;
148   }
149   if ((status == 0) && (n.time == 0))
150   {
151     print_to_socket (fh, "-1 Option `time' missing.\n");
152     status = -1;
153   }
154   if ((status == 0) && (strlen (n.message) == 0))
155   {
156     print_to_socket (fh, "-1 No message or message of length 0 given.\n");
157     status = -1;
158   }
159
160   /* If status is still zero the notification is fine and we can finally
161    * dispatch it. */
162   if (status == 0)
163   {
164     plugin_dispatch_notification (&n);
165     print_to_socket (fh, "0 Success\n");
166   }
167
168   return (0);
169 } /* int handle_putnotif */
170
171 /* vim: set shiftwidth=2 softtabstop=2 tabstop=8 : */