snort plugin: Improve error handling.
[collectd.git] / src / utils_cmd_getthreshold.c
1 /**
2  * collectd - src/utils_cms_getthreshold.c
3  * Copyright (C) 2008,2009  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_threshold.h"
27 #include "utils_parse_option.h" /* for `parse_string' */
28 #include "utils_cmd_getthreshold.h"
29
30 #define print_to_socket(fh, ...) \
31   if (fprintf (fh, __VA_ARGS__) < 0) { \
32     char errbuf[1024]; \
33     WARNING ("handle_getthreshold: failed to write to socket #%i: %s", \
34         fileno (fh), sstrerror (errno, errbuf, sizeof (errbuf))); \
35     return -1; \
36   }
37
38 int handle_getthreshold (FILE *fh, char *buffer)
39 {
40   char *command;
41   char *identifier;
42   char *identifier_copy;
43
44   char *host;
45   char *plugin;
46   char *plugin_instance;
47   char *type;
48   char *type_instance;
49
50   value_list_t vl;
51   threshold_t threshold;
52
53   int   status;
54   size_t i;
55
56   if ((fh == NULL) || (buffer == NULL))
57     return (-1);
58
59   DEBUG ("utils_cmd_getthreshold: handle_getthreshold (fh = %p, buffer = %s);",
60       (void *) fh, buffer);
61
62   command = NULL;
63   status = parse_string (&buffer, &command);
64   if (status != 0)
65   {
66     print_to_socket (fh, "-1 Cannot parse command.\n");
67     return (-1);
68   }
69   assert (command != NULL);
70
71   if (strcasecmp ("GETTHRESHOLD", command) != 0)
72   {
73     print_to_socket (fh, "-1 Unexpected command: `%s'.\n", command);
74     return (-1);
75   }
76
77   identifier = NULL;
78   status = parse_string (&buffer, &identifier);
79   if (status != 0)
80   {
81     print_to_socket (fh, "-1 Cannot parse identifier.\n");
82     return (-1);
83   }
84   assert (identifier != NULL);
85
86   if (*buffer != 0)
87   {
88     print_to_socket (fh, "-1 Garbage after end of command: %s\n", buffer);
89     return (-1);
90   }
91
92   /* parse_identifier() modifies its first argument,
93    * returning pointers into it */
94   identifier_copy = sstrdup (identifier);
95
96   status = parse_identifier (identifier_copy, &host,
97       &plugin, &plugin_instance,
98       &type, &type_instance);
99   if (status != 0)
100   {
101     DEBUG ("handle_getthreshold: Cannot parse identifier `%s'.", identifier);
102     print_to_socket (fh, "-1 Cannot parse identifier `%s'.\n", identifier);
103     sfree (identifier_copy);
104     return (-1);
105   }
106
107   memset (&vl, 0, sizeof (vl));
108   sstrncpy (vl.host, host, sizeof (vl.host));
109   sstrncpy (vl.plugin, plugin, sizeof (vl.plugin));
110   if (plugin_instance != NULL)
111     sstrncpy (vl.plugin_instance, plugin_instance, sizeof (vl.plugin_instance));
112   sstrncpy (vl.type, type, sizeof (vl.type));
113   if (type_instance != NULL)
114     sstrncpy (vl.type_instance, type_instance, sizeof (vl.type_instance));
115   sfree (identifier_copy);
116
117   memset (&threshold, 0, sizeof (threshold));
118   status = ut_search_threshold (&vl, &threshold);
119   if (status == ENOENT)
120   {
121     print_to_socket (fh, "-1 No threshold found for identifier %s\n",
122         identifier);
123     return (0);
124   }
125   else if (status != 0)
126   {
127     print_to_socket (fh, "-1 Error while looking up threshold: %i\n",
128         status);
129     return (-1);
130   }
131
132   /* Lets count the number of lines we'll return. */
133   i = 0;
134   if (threshold.host[0] != 0)            i++;
135   if (threshold.plugin[0] != 0)          i++;
136   if (threshold.plugin_instance[0] != 0) i++;
137   if (threshold.type[0] != 0)            i++;
138   if (threshold.type_instance[0] != 0)   i++;
139   if (threshold.data_source[0] != 0)     i++;
140   if (!isnan (threshold.warning_min))    i++;
141   if (!isnan (threshold.warning_max))    i++;
142   if (!isnan (threshold.failure_min))    i++;
143   if (!isnan (threshold.failure_max))    i++;
144   if (threshold.hysteresis > 0.0)        i++;
145   if (threshold.hits > 1)                i++;
146
147   /* Print the response */
148   print_to_socket (fh, "%zu Threshold found\n", i);
149
150   if (threshold.host[0] != 0)
151     print_to_socket (fh, "Host: %s\n", threshold.host)
152   if (threshold.plugin[0] != 0)
153     print_to_socket (fh, "Plugin: %s\n", threshold.plugin)
154   if (threshold.plugin_instance[0] != 0)
155     print_to_socket (fh, "Plugin Instance: %s\n", threshold.plugin_instance)
156   if (threshold.type[0] != 0)
157     print_to_socket (fh, "Type: %s\n", threshold.type)
158   if (threshold.type_instance[0] != 0)
159     print_to_socket (fh, "Type Instance: %s\n", threshold.type_instance)
160   if (threshold.data_source[0] != 0)
161     print_to_socket (fh, "Data Source: %s\n", threshold.data_source)
162   if (!isnan (threshold.warning_min))
163     print_to_socket (fh, "Warning Min: %g\n", threshold.warning_min)
164   if (!isnan (threshold.warning_max))
165     print_to_socket (fh, "Warning Max: %g\n", threshold.warning_max)
166   if (!isnan (threshold.failure_min))
167     print_to_socket (fh, "Failure Min: %g\n", threshold.failure_min)
168   if (!isnan (threshold.failure_max))
169     print_to_socket (fh, "Failure Max: %g\n", threshold.failure_max)
170   if (threshold.hysteresis > 0.0)
171     print_to_socket (fh, "Hysteresis: %g\n", threshold.hysteresis)
172   if (threshold.hits > 1)
173     print_to_socket (fh, "Hits: %i\n", threshold.hits)
174
175   return (0);
176 } /* int handle_getthreshold */
177
178 /* vim: set sw=2 sts=2 ts=8 et : */