Fixed issues with initializing hostname.
[collectd.git] / src / utils_cmd_putnotif.c
1 /**
2  * collectd - src/utils_cmd_putnotif.c
3  * Copyright (C) 2008       Florian octo Forster
4  *
5  * Permission is hereby granted, free of charge, to any person obtaining a
6  * copy of this software and associated documentation files (the "Software"),
7  * to deal in the Software without restriction, including without limitation
8  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
9  * and/or sell copies of the Software, and to permit persons to whom the
10  * Software is furnished to do so, subject to the following conditions:
11  *
12  * The above copyright notice and this permission notice shall be included in
13  * all copies or substantial portions of the Software.
14  *
15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
21  * DEALINGS IN THE SOFTWARE.
22  *
23  * Authors:
24  *   Florian octo Forster <octo at collectd.org>
25  **/
26
27 #include "collectd.h"
28
29 #include "common.h"
30 #include "plugin.h"
31
32 #include "utils_cmd_putnotif.h"
33 #include "utils_parse_option.h"
34
35 #define print_to_socket(fh, ...)                                               \
36   do {                                                                         \
37     if (fprintf(fh, __VA_ARGS__) < 0) {                                        \
38       char errbuf[1024];                                                       \
39       WARNING("handle_putnotif: failed to write to socket #%i: %s",            \
40               fileno(fh), sstrerror(errno, errbuf, sizeof(errbuf)));           \
41       return -1;                                                               \
42     }                                                                          \
43     fflush(fh);                                                                \
44   } while (0)
45
46 static int set_option_severity(notification_t *n, const char *value) {
47   if (strcasecmp(value, "Failure") == 0)
48     n->severity = NOTIF_FAILURE;
49   else if (strcasecmp(value, "Warning") == 0)
50     n->severity = NOTIF_WARNING;
51   else if (strcasecmp(value, "Okay") == 0)
52     n->severity = NOTIF_OKAY;
53   else
54     return -1;
55
56   return 0;
57 } /* int set_option_severity */
58
59 static int set_option_time(notification_t *n, const char *value) {
60   char *endptr = NULL;
61   double tmp;
62
63   errno = 0;
64   tmp = strtod(value, &endptr);
65   if ((errno != 0)         /* Overflow */
66       || (endptr == value) /* Invalid string */
67       || (endptr == NULL)  /* This should not happen */
68       || (*endptr != 0))   /* Trailing chars */
69     return -1;
70
71   n->time = DOUBLE_TO_CDTIME_T(tmp);
72
73   return 0;
74 } /* int set_option_time */
75
76 static int set_option(notification_t *n, const char *option,
77                       const char *value) {
78   if ((n == NULL) || (option == NULL) || (value == NULL))
79     return -1;
80
81   DEBUG("utils_cmd_putnotif: set_option (option = %s, value = %s);", option,
82         value);
83
84   /* Add a meta option in the form: <type>:<key> */
85   if (option[0] != '\0' && option[1] == ':') {
86     /* Refuse empty key */
87     if (option[2] == '\0')
88       return 1;
89
90     if (option[0] == 's')
91       return plugin_notification_meta_add_string(n, option + 2, value);
92     else
93       return 1;
94   }
95
96   if (strcasecmp("severity", option) == 0)
97     return set_option_severity(n, value);
98   else if (strcasecmp("time", option) == 0)
99     return set_option_time(n, value);
100   else if (strcasecmp("message", option) == 0)
101     sstrncpy(n->message, value, sizeof(n->message));
102   else if (strcasecmp("host", option) == 0)
103     sstrncpy(n->host, value, sizeof(n->host));
104   else if (strcasecmp("plugin", option) == 0)
105     sstrncpy(n->plugin, value, sizeof(n->plugin));
106   else if (strcasecmp("plugin_instance", option) == 0)
107     sstrncpy(n->plugin_instance, value, sizeof(n->plugin_instance));
108   else if (strcasecmp("type", option) == 0)
109     sstrncpy(n->type, value, sizeof(n->type));
110   else if (strcasecmp("type_instance", option) == 0)
111     sstrncpy(n->type_instance, value, sizeof(n->type_instance));
112   else
113     return 1;
114
115   return 0;
116 } /* int set_option */
117
118 int handle_putnotif(FILE *fh, char *buffer) {
119   char *command;
120   notification_t n = {0};
121   int status;
122
123   if ((fh == NULL) || (buffer == NULL))
124     return -1;
125
126   DEBUG("utils_cmd_putnotif: handle_putnotif (fh = %p, buffer = %s);",
127         (void *)fh, buffer);
128
129   command = NULL;
130   status = parse_string(&buffer, &command);
131   if (status != 0) {
132     print_to_socket(fh, "-1 Cannot parse command.\n");
133     return -1;
134   }
135   assert(command != NULL);
136
137   if (strcasecmp("PUTNOTIF", command) != 0) {
138     print_to_socket(fh, "-1 Unexpected command: `%s'.\n", command);
139     return -1;
140   }
141
142   status = 0;
143   while (*buffer != 0) {
144     char *key;
145     char *value;
146
147     status = parse_option(&buffer, &key, &value);
148     if (status != 0) {
149       print_to_socket(fh, "-1 Malformed option.\n");
150       break;
151     }
152
153     status = set_option(&n, key, value);
154     if (status != 0) {
155       print_to_socket(fh, "-1 Error parsing option `%s'\n", key);
156       break;
157     }
158   } /* for (i) */
159
160   /* Check for required fields and complain if anything is missing. */
161   if ((status == 0) && (n.severity == 0)) {
162     print_to_socket(fh, "-1 Option `severity' missing.\n");
163     status = -1;
164   }
165   if ((status == 0) && (n.time == 0)) {
166     print_to_socket(fh, "-1 Option `time' missing.\n");
167     status = -1;
168   }
169   if ((status == 0) && (strlen(n.message) == 0)) {
170     print_to_socket(fh, "-1 No message or message of length 0 given.\n");
171     status = -1;
172   }
173
174   /* If status is still zero the notification is fine and we can finally
175    * dispatch it. */
176   if (status == 0) {
177     plugin_dispatch_notification(&n);
178     print_to_socket(fh, "0 Success\n");
179   }
180
181   return 0;
182 } /* int handle_putnotif */