diff options
Diffstat (limited to 'plugins')
-rw-r--r-- | plugins/check_dns.c | 370 | ||||
-rw-r--r-- | plugins/check_dns.d/config.h | 34 |
2 files changed, 246 insertions, 158 deletions
diff --git a/plugins/check_dns.c b/plugins/check_dns.c index e1e7c00e..a4c8e9b1 100644 --- a/plugins/check_dns.c +++ b/plugins/check_dns.c | |||
@@ -39,26 +39,22 @@ const char *email = "devel@monitoring-plugins.org"; | |||
39 | #include "netutils.h" | 39 | #include "netutils.h" |
40 | #include "runcmd.h" | 40 | #include "runcmd.h" |
41 | 41 | ||
42 | static int process_arguments(int /*argc*/, char ** /*argv*/); | 42 | #include "states.h" |
43 | static int validate_arguments(void); | 43 | #include "check_dns.d/config.h" |
44 | static int error_scan(char * /*input_buffer*/, bool *); | 44 | |
45 | typedef struct { | ||
46 | int errorcode; | ||
47 | check_dns_config config; | ||
48 | } check_dns_config_wrapper; | ||
49 | static check_dns_config_wrapper process_arguments(int /*argc*/, char ** /*argv*/); | ||
50 | static check_dns_config_wrapper validate_arguments(check_dns_config_wrapper /*config_wrapper*/); | ||
51 | static mp_state_enum error_scan(char * /*input_buffer*/, bool * /*is_nxdomain*/, const char /*dns_server*/[ADDRESS_LENGTH]); | ||
45 | static bool ip_match_cidr(const char * /*addr*/, const char * /*cidr_ro*/); | 52 | static bool ip_match_cidr(const char * /*addr*/, const char * /*cidr_ro*/); |
46 | static unsigned long ip2long(const char * /*src*/); | 53 | static unsigned long ip2long(const char * /*src*/); |
47 | static void print_help(void); | 54 | static void print_help(void); |
48 | void print_usage(void); | 55 | void print_usage(void); |
49 | 56 | ||
50 | #define ADDRESS_LENGTH 256 | ||
51 | static char query_address[ADDRESS_LENGTH] = ""; | ||
52 | static char dns_server[ADDRESS_LENGTH] = ""; | ||
53 | static char ptr_server[ADDRESS_LENGTH] = ""; | ||
54 | static bool verbose = false; | 57 | static bool verbose = false; |
55 | static char **expected_address = NULL; | ||
56 | static int expected_address_cnt = 0; | ||
57 | static bool expect_nxdomain = false; | ||
58 | |||
59 | static bool expect_authority = false; | ||
60 | static bool all_match = false; | ||
61 | static thresholds *time_thresholds = NULL; | ||
62 | 58 | ||
63 | static int qstrcmp(const void *p1, const void *p2) { | 59 | static int qstrcmp(const void *p1, const void *p2) { |
64 | /* The actual arguments to this function are "pointers to | 60 | /* The actual arguments to this function are "pointers to |
@@ -68,23 +64,6 @@ static int qstrcmp(const void *p1, const void *p2) { | |||
68 | } | 64 | } |
69 | 65 | ||
70 | int main(int argc, char **argv) { | 66 | int main(int argc, char **argv) { |
71 | char *command_line = NULL; | ||
72 | char input_buffer[MAX_INPUT_BUFFER]; | ||
73 | char *address = NULL; /* comma separated str with addrs/ptrs (sorted) */ | ||
74 | char **addresses = NULL; | ||
75 | int n_addresses = 0; | ||
76 | char *msg = NULL; | ||
77 | char *temp_buffer = NULL; | ||
78 | bool non_authoritative = false; | ||
79 | int result = STATE_UNKNOWN; | ||
80 | double elapsed_time; | ||
81 | long microsec; | ||
82 | struct timeval tv; | ||
83 | bool parse_address = false; /* This flag scans for Address: but only after Name: */ | ||
84 | output chld_out; | ||
85 | output chld_err; | ||
86 | bool is_nxdomain = false; | ||
87 | |||
88 | setlocale(LC_ALL, ""); | 67 | setlocale(LC_ALL, ""); |
89 | bindtextdomain(PACKAGE, LOCALEDIR); | 68 | bindtextdomain(PACKAGE, LOCALEDIR); |
90 | textdomain(PACKAGE); | 69 | textdomain(PACKAGE); |
@@ -97,39 +76,62 @@ int main(int argc, char **argv) { | |||
97 | /* Parse extra opts if any */ | 76 | /* Parse extra opts if any */ |
98 | argv = np_extra_opts(&argc, argv, progname); | 77 | argv = np_extra_opts(&argc, argv, progname); |
99 | 78 | ||
100 | if (process_arguments(argc, argv) == ERROR) { | 79 | check_dns_config_wrapper tmp = process_arguments(argc, argv); |
80 | |||
81 | if (tmp.errorcode == ERROR) { | ||
101 | usage_va(_("Could not parse arguments")); | 82 | usage_va(_("Could not parse arguments")); |
102 | } | 83 | } |
103 | 84 | ||
85 | const check_dns_config config = tmp.config; | ||
86 | |||
87 | char *command_line = NULL; | ||
104 | /* get the command to run */ | 88 | /* get the command to run */ |
105 | xasprintf(&command_line, "%s %s %s", NSLOOKUP_COMMAND, query_address, dns_server); | 89 | xasprintf(&command_line, "%s %s %s", NSLOOKUP_COMMAND, config.query_address, config.dns_server); |
106 | 90 | ||
91 | struct timeval tv; | ||
107 | alarm(timeout_interval); | 92 | alarm(timeout_interval); |
108 | gettimeofday(&tv, NULL); | 93 | gettimeofday(&tv, NULL); |
109 | 94 | ||
110 | if (verbose) | 95 | if (verbose) { |
111 | printf("%s\n", command_line); | 96 | printf("%s\n", command_line); |
97 | } | ||
112 | 98 | ||
99 | output chld_out; | ||
100 | output chld_err; | ||
101 | char *msg = NULL; | ||
102 | mp_state_enum result = STATE_UNKNOWN; | ||
113 | /* run the command */ | 103 | /* run the command */ |
114 | if ((np_runcmd(command_line, &chld_out, &chld_err, 0)) != 0) { | 104 | if ((np_runcmd(command_line, &chld_out, &chld_err, 0)) != 0) { |
115 | msg = (char *)_("nslookup returned an error status"); | 105 | msg = (char *)_("nslookup returned an error status"); |
116 | result = STATE_WARNING; | 106 | result = STATE_WARNING; |
117 | } | 107 | } |
118 | 108 | ||
119 | /* scan stdout */ | 109 | /* ===== |
110 | * scan stdout, main results get retrieved here | ||
111 | * ===== | ||
112 | */ | ||
113 | char *address = NULL; /* comma separated str with addrs/ptrs (sorted) */ | ||
114 | char **addresses = NULL; // All addresses parsed from stdout | ||
115 | size_t n_addresses = 0; // counter for retrieved addresses | ||
116 | bool non_authoritative = false; | ||
117 | bool is_nxdomain = false; | ||
118 | bool parse_address = false; /* This flag scans for Address: but only after Name: */ | ||
120 | for (size_t i = 0; i < chld_out.lines; i++) { | 119 | for (size_t i = 0; i < chld_out.lines; i++) { |
121 | if (addresses == NULL) | 120 | if (addresses == NULL) { |
122 | addresses = malloc(sizeof(*addresses) * 10); | 121 | addresses = malloc(sizeof(*addresses) * 10); |
123 | else if (!(n_addresses % 10)) | 122 | } else if (!(n_addresses % 10)) { |
124 | addresses = realloc(addresses, sizeof(*addresses) * (n_addresses + 10)); | 123 | addresses = realloc(addresses, sizeof(*addresses) * (n_addresses + 10)); |
124 | } | ||
125 | 125 | ||
126 | if (verbose) | 126 | if (verbose) { |
127 | puts(chld_out.line[i]); | 127 | puts(chld_out.line[i]); |
128 | } | ||
128 | 129 | ||
129 | if (strcasestr(chld_out.line[i], ".in-addr.arpa") || strcasestr(chld_out.line[i], ".ip6.arpa")) { | 130 | if (strcasestr(chld_out.line[i], ".in-addr.arpa") || strcasestr(chld_out.line[i], ".ip6.arpa")) { |
130 | if ((temp_buffer = strstr(chld_out.line[i], "name = "))) | 131 | char *temp_buffer = NULL; |
132 | if ((temp_buffer = strstr(chld_out.line[i], "name = "))) { | ||
131 | addresses[n_addresses++] = strdup(temp_buffer + 7); | 133 | addresses[n_addresses++] = strdup(temp_buffer + 7); |
132 | else { | 134 | } else { |
133 | msg = (char *)_("Warning plugin error"); | 135 | msg = (char *)_("Warning plugin error"); |
134 | result = STATE_WARNING; | 136 | result = STATE_WARNING; |
135 | } | 137 | } |
@@ -137,37 +139,47 @@ int main(int argc, char **argv) { | |||
137 | 139 | ||
138 | /* bug ID: 2946553 - Older versions of bind will use all available dns | 140 | /* bug ID: 2946553 - Older versions of bind will use all available dns |
139 | servers, we have to match the one specified */ | 141 | servers, we have to match the one specified */ |
140 | if (strstr(chld_out.line[i], "Server:") && strlen(dns_server) > 0) { | 142 | if (strstr(chld_out.line[i], "Server:") && strlen(config.dns_server) > 0) { |
141 | temp_buffer = strchr(chld_out.line[i], ':'); | 143 | char *temp_buffer = strchr(chld_out.line[i], ':'); |
144 | if (temp_buffer == NULL) { | ||
145 | die(STATE_UNKNOWN, _("'%s' returned a weirdly formatted Server line\n"), NSLOOKUP_COMMAND); | ||
146 | } | ||
147 | |||
142 | temp_buffer++; | 148 | temp_buffer++; |
143 | 149 | ||
144 | /* Strip leading tabs */ | 150 | /* Strip leading tabs */ |
145 | for (; *temp_buffer != '\0' && *temp_buffer == '\t'; temp_buffer++) | 151 | for (; *temp_buffer != '\0' && *temp_buffer == '\t'; temp_buffer++) { |
146 | /* NOOP */; | 152 | /* NOOP */; |
153 | } | ||
147 | 154 | ||
148 | strip(temp_buffer); | 155 | strip(temp_buffer); |
149 | if (temp_buffer == NULL || strlen(temp_buffer) == 0) { | 156 | if (strlen(temp_buffer) == 0) { |
150 | die(STATE_CRITICAL, _("DNS CRITICAL - '%s' returned empty server string\n"), NSLOOKUP_COMMAND); | 157 | die(STATE_CRITICAL, _("DNS CRITICAL - '%s' returned empty server string\n"), NSLOOKUP_COMMAND); |
151 | } | 158 | } |
152 | 159 | ||
153 | if (strcmp(temp_buffer, dns_server) != 0) { | 160 | if (strcmp(temp_buffer, config.dns_server) != 0) { |
154 | die(STATE_CRITICAL, _("DNS CRITICAL - No response from DNS %s\n"), dns_server); | 161 | die(STATE_CRITICAL, _("DNS CRITICAL - No response from DNS %s\n"), config.dns_server); |
155 | } | 162 | } |
156 | } | 163 | } |
157 | 164 | ||
158 | /* the server is responding, we just got the host name... */ | 165 | /* the server is responding, we just got the host name... */ |
159 | if (strstr(chld_out.line[i], "Name:")) | 166 | if (strstr(chld_out.line[i], "Name:")) { |
160 | parse_address = true; | 167 | parse_address = true; |
161 | else if (parse_address && (strstr(chld_out.line[i], "Address:") || strstr(chld_out.line[i], "Addresses:"))) { | 168 | } else if (parse_address && (strstr(chld_out.line[i], "Address:") || strstr(chld_out.line[i], "Addresses:"))) { |
162 | temp_buffer = index(chld_out.line[i], ':'); | 169 | char *temp_buffer = strchr(chld_out.line[i], ':'); |
170 | if (temp_buffer == NULL) { | ||
171 | die(STATE_UNKNOWN, _("'%s' returned a weirdly formatted Address line\n"), NSLOOKUP_COMMAND); | ||
172 | } | ||
173 | |||
163 | temp_buffer++; | 174 | temp_buffer++; |
164 | 175 | ||
165 | /* Strip leading spaces */ | 176 | /* Strip leading spaces */ |
166 | while (*temp_buffer == ' ') | 177 | while (*temp_buffer == ' ') { |
167 | temp_buffer++; | 178 | temp_buffer++; |
179 | } | ||
168 | 180 | ||
169 | strip(temp_buffer); | 181 | strip(temp_buffer); |
170 | if (temp_buffer == NULL || strlen(temp_buffer) == 0) { | 182 | if (strlen(temp_buffer) == 0) { |
171 | die(STATE_CRITICAL, _("DNS CRITICAL - '%s' returned empty host name string\n"), NSLOOKUP_COMMAND); | 183 | die(STATE_CRITICAL, _("DNS CRITICAL - '%s' returned empty host name string\n"), NSLOOKUP_COMMAND); |
172 | } | 184 | } |
173 | 185 | ||
@@ -176,65 +188,71 @@ int main(int argc, char **argv) { | |||
176 | non_authoritative = true; | 188 | non_authoritative = true; |
177 | } | 189 | } |
178 | 190 | ||
179 | result = error_scan(chld_out.line[i], &is_nxdomain); | 191 | result = error_scan(chld_out.line[i], &is_nxdomain, config.dns_server); |
180 | if (result != STATE_OK) { | 192 | if (result != STATE_OK) { |
181 | msg = strchr(chld_out.line[i], ':'); | 193 | msg = strchr(chld_out.line[i], ':'); |
182 | if (msg) | 194 | if (msg) { |
183 | msg++; | 195 | msg++; |
196 | } | ||
184 | break; | 197 | break; |
185 | } | 198 | } |
186 | } | 199 | } |
187 | 200 | ||
201 | char input_buffer[MAX_INPUT_BUFFER]; | ||
188 | /* scan stderr */ | 202 | /* scan stderr */ |
189 | for (size_t i = 0; i < chld_err.lines; i++) { | 203 | for (size_t i = 0; i < chld_err.lines; i++) { |
190 | if (verbose) | 204 | if (verbose) { |
191 | puts(chld_err.line[i]); | 205 | puts(chld_err.line[i]); |
206 | } | ||
192 | 207 | ||
193 | if (error_scan(chld_err.line[i], &is_nxdomain) != STATE_OK) { | 208 | if (error_scan(chld_err.line[i], &is_nxdomain, config.dns_server) != STATE_OK) { |
194 | result = max_state(result, error_scan(chld_err.line[i], &is_nxdomain)); | 209 | result = max_state(result, error_scan(chld_err.line[i], &is_nxdomain, config.dns_server)); |
195 | msg = strchr(input_buffer, ':'); | 210 | msg = strchr(input_buffer, ':'); |
196 | if (msg) | 211 | if (msg) { |
197 | msg++; | 212 | msg++; |
198 | else | 213 | } else { |
199 | msg = input_buffer; | 214 | msg = input_buffer; |
215 | } | ||
200 | } | 216 | } |
201 | } | 217 | } |
202 | 218 | ||
203 | if (is_nxdomain && !expect_nxdomain) { | 219 | if (is_nxdomain && !config.expect_nxdomain) { |
204 | die(STATE_CRITICAL, _("Domain '%s' was not found by the server\n"), query_address); | 220 | die(STATE_CRITICAL, _("Domain '%s' was not found by the server\n"), config.query_address); |
205 | } | 221 | } |
206 | 222 | ||
207 | if (addresses) { | 223 | if (addresses) { |
208 | int i; | 224 | size_t slen = 1; |
209 | int slen; | 225 | char *adrp = NULL; |
210 | char *adrp; | ||
211 | qsort(addresses, n_addresses, sizeof(*addresses), qstrcmp); | 226 | qsort(addresses, n_addresses, sizeof(*addresses), qstrcmp); |
212 | for (i = 0, slen = 1; i < n_addresses; i++) { | 227 | for (size_t i = 0; i < n_addresses; i++) { |
213 | slen += strlen(addresses[i]) + 1; | 228 | slen += strlen(addresses[i]) + 1; |
214 | } | 229 | } |
230 | |||
231 | // Temporary pointer adrp gets moved, address stays on the beginning | ||
215 | adrp = address = malloc(slen); | 232 | adrp = address = malloc(slen); |
216 | for (i = 0; i < n_addresses; i++) { | 233 | for (size_t i = 0; i < n_addresses; i++) { |
217 | if (i) | 234 | if (i) { |
218 | *adrp++ = ','; | 235 | *adrp++ = ','; |
236 | } | ||
219 | strcpy(adrp, addresses[i]); | 237 | strcpy(adrp, addresses[i]); |
220 | adrp += strlen(addresses[i]); | 238 | adrp += strlen(addresses[i]); |
221 | } | 239 | } |
222 | *adrp = 0; | 240 | *adrp = 0; |
223 | } else | 241 | } else { |
224 | die(STATE_CRITICAL, _("DNS CRITICAL - '%s' msg parsing exited with no address\n"), NSLOOKUP_COMMAND); | 242 | die(STATE_CRITICAL, _("DNS CRITICAL - '%s' msg parsing exited with no address\n"), NSLOOKUP_COMMAND); |
243 | } | ||
225 | 244 | ||
226 | /* compare to expected address */ | 245 | /* compare to expected address */ |
227 | if (result == STATE_OK && expected_address_cnt > 0) { | 246 | if (result == STATE_OK && config.expected_address_cnt > 0) { |
228 | result = STATE_CRITICAL; | 247 | result = STATE_CRITICAL; |
229 | temp_buffer = ""; | 248 | char *temp_buffer = ""; |
230 | unsigned long expect_match = (1 << expected_address_cnt) - 1; | 249 | unsigned long expect_match = (1 << config.expected_address_cnt) - 1; |
231 | unsigned long addr_match = (1 << n_addresses) - 1; | 250 | unsigned long addr_match = (1 << n_addresses) - 1; |
232 | 251 | ||
233 | for (int i = 0; i < expected_address_cnt; i++) { | 252 | for (size_t i = 0; i < config.expected_address_cnt; i++) { |
234 | int j; | ||
235 | /* check if we get a match on 'raw' ip or cidr */ | 253 | /* check if we get a match on 'raw' ip or cidr */ |
236 | for (j = 0; j < n_addresses; j++) { | 254 | for (size_t j = 0; j < n_addresses; j++) { |
237 | if (strcmp(addresses[j], expected_address[i]) == 0 || ip_match_cidr(addresses[j], expected_address[i])) { | 255 | if (strcmp(addresses[j], config.expected_address[i]) == 0 || ip_match_cidr(addresses[j], config.expected_address[i])) { |
238 | result = STATE_OK; | 256 | result = STATE_OK; |
239 | addr_match &= ~(1 << j); | 257 | addr_match &= ~(1 << j); |
240 | expect_match &= ~(1 << i); | 258 | expect_match &= ~(1 << i); |
@@ -242,11 +260,12 @@ int main(int argc, char **argv) { | |||
242 | } | 260 | } |
243 | 261 | ||
244 | /* prepare an error string */ | 262 | /* prepare an error string */ |
245 | xasprintf(&temp_buffer, "%s%s; ", temp_buffer, expected_address[i]); | 263 | xasprintf(&temp_buffer, "%s%s; ", temp_buffer, config.expected_address[i]); |
246 | } | 264 | } |
247 | /* check if expected_address must cover all in addresses and none may be missing */ | 265 | /* check if expected_address must cover all in addresses and none may be missing */ |
248 | if (all_match && (expect_match != 0 || addr_match != 0)) | 266 | if (config.all_match && (expect_match != 0 || addr_match != 0)) { |
249 | result = STATE_CRITICAL; | 267 | result = STATE_CRITICAL; |
268 | } | ||
250 | if (result == STATE_CRITICAL) { | 269 | if (result == STATE_CRITICAL) { |
251 | /* Strip off last semicolon... */ | 270 | /* Strip off last semicolon... */ |
252 | temp_buffer[strlen(temp_buffer) - 2] = '\0'; | 271 | temp_buffer[strlen(temp_buffer) - 2] = '\0'; |
@@ -254,28 +273,29 @@ int main(int argc, char **argv) { | |||
254 | } | 273 | } |
255 | } | 274 | } |
256 | 275 | ||
257 | if (expect_nxdomain) { | 276 | if (config.expect_nxdomain) { |
258 | if (!is_nxdomain) { | 277 | if (!is_nxdomain) { |
259 | result = STATE_CRITICAL; | 278 | result = STATE_CRITICAL; |
260 | xasprintf(&msg, _("Domain '%s' was found by the server: '%s'\n"), query_address, address); | 279 | xasprintf(&msg, _("Domain '%s' was found by the server: '%s'\n"), config.query_address, address); |
261 | } else { | 280 | } else { |
262 | if (address != NULL) | 281 | if (address != NULL) { |
263 | free(address); | 282 | free(address); |
283 | } | ||
264 | address = "NXDOMAIN"; | 284 | address = "NXDOMAIN"; |
265 | } | 285 | } |
266 | } | 286 | } |
267 | 287 | ||
268 | /* check if authoritative */ | 288 | /* check if authoritative */ |
269 | if (result == STATE_OK && expect_authority && non_authoritative) { | 289 | if (result == STATE_OK && config.expect_authority && non_authoritative) { |
270 | result = STATE_CRITICAL; | 290 | result = STATE_CRITICAL; |
271 | xasprintf(&msg, _("server %s is not authoritative for %s"), dns_server, query_address); | 291 | xasprintf(&msg, _("server %s is not authoritative for %s"), config.dns_server, config.query_address); |
272 | } | 292 | } |
273 | 293 | ||
274 | microsec = deltime(tv); | 294 | long microsec = deltime(tv); |
275 | elapsed_time = (double)microsec / 1.0e6; | 295 | double elapsed_time = (double)microsec / 1.0e6; |
276 | 296 | ||
277 | if (result == STATE_OK) { | 297 | if (result == STATE_OK) { |
278 | result = get_status(elapsed_time, time_thresholds); | 298 | result = get_status(elapsed_time, config.time_thresholds); |
279 | if (result == STATE_OK) { | 299 | if (result == STATE_OK) { |
280 | printf("DNS %s: ", _("OK")); | 300 | printf("DNS %s: ", _("OK")); |
281 | } else if (result == STATE_WARNING) { | 301 | } else if (result == STATE_WARNING) { |
@@ -284,24 +304,26 @@ int main(int argc, char **argv) { | |||
284 | printf("DNS %s: ", _("CRITICAL")); | 304 | printf("DNS %s: ", _("CRITICAL")); |
285 | } | 305 | } |
286 | printf(ngettext("%.3f second response time", "%.3f seconds response time", elapsed_time), elapsed_time); | 306 | printf(ngettext("%.3f second response time", "%.3f seconds response time", elapsed_time), elapsed_time); |
287 | printf(_(". %s returns %s"), query_address, address); | 307 | printf(_(". %s returns %s"), config.query_address, address); |
288 | if ((time_thresholds->warning != NULL) && (time_thresholds->critical != NULL)) { | 308 | if ((config.time_thresholds->warning != NULL) && (config.time_thresholds->critical != NULL)) { |
289 | printf("|%s\n", fperfdata("time", elapsed_time, "s", true, time_thresholds->warning->end, true, time_thresholds->critical->end, | 309 | printf("|%s\n", fperfdata("time", elapsed_time, "s", true, config.time_thresholds->warning->end, true, |
290 | true, 0, false, 0)); | 310 | config.time_thresholds->critical->end, true, 0, false, 0)); |
291 | } else if ((time_thresholds->warning == NULL) && (time_thresholds->critical != NULL)) { | 311 | } else if ((config.time_thresholds->warning == NULL) && (config.time_thresholds->critical != NULL)) { |
292 | printf("|%s\n", fperfdata("time", elapsed_time, "s", false, 0, true, time_thresholds->critical->end, true, 0, false, 0)); | 312 | printf("|%s\n", fperfdata("time", elapsed_time, "s", false, 0, true, config.time_thresholds->critical->end, true, 0, false, 0)); |
293 | } else if ((time_thresholds->warning != NULL) && (time_thresholds->critical == NULL)) { | 313 | } else if ((config.time_thresholds->warning != NULL) && (config.time_thresholds->critical == NULL)) { |
294 | printf("|%s\n", fperfdata("time", elapsed_time, "s", true, time_thresholds->warning->end, false, 0, true, 0, false, 0)); | 314 | printf("|%s\n", fperfdata("time", elapsed_time, "s", true, config.time_thresholds->warning->end, false, 0, true, 0, false, 0)); |
295 | } else | 315 | } else { |
296 | printf("|%s\n", fperfdata("time", elapsed_time, "s", false, 0, false, 0, true, 0, false, 0)); | 316 | printf("|%s\n", fperfdata("time", elapsed_time, "s", false, 0, false, 0, true, 0, false, 0)); |
297 | } else if (result == STATE_WARNING) | 317 | } |
318 | } else if (result == STATE_WARNING) { | ||
298 | printf(_("DNS WARNING - %s\n"), !strcmp(msg, "") ? _(" Probably a non-existent host/domain") : msg); | 319 | printf(_("DNS WARNING - %s\n"), !strcmp(msg, "") ? _(" Probably a non-existent host/domain") : msg); |
299 | else if (result == STATE_CRITICAL) | 320 | } else if (result == STATE_CRITICAL) { |
300 | printf(_("DNS CRITICAL - %s\n"), !strcmp(msg, "") ? _(" Probably a non-existent host/domain") : msg); | 321 | printf(_("DNS CRITICAL - %s\n"), !strcmp(msg, "") ? _(" Probably a non-existent host/domain") : msg); |
301 | else | 322 | } else { |
302 | printf(_("DNS UNKNOWN - %s\n"), !strcmp(msg, "") ? _(" Probably a non-existent host/domain") : msg); | 323 | printf(_("DNS UNKNOWN - %s\n"), !strcmp(msg, "") ? _(" Probably a non-existent host/domain") : msg); |
324 | } | ||
303 | 325 | ||
304 | return result; | 326 | exit(result); |
305 | } | 327 | } |
306 | 328 | ||
307 | bool ip_match_cidr(const char *addr, const char *cidr_ro) { | 329 | bool ip_match_cidr(const char *addr, const char *cidr_ro) { |
@@ -329,64 +351,69 @@ unsigned long ip2long(const char *src) { | |||
329 | : 0; | 351 | : 0; |
330 | } | 352 | } |
331 | 353 | ||
332 | int error_scan(char *input_buffer, bool *is_nxdomain) { | 354 | mp_state_enum error_scan(char *input_buffer, bool *is_nxdomain, const char dns_server[ADDRESS_LENGTH]) { |
333 | 355 | ||
334 | const int nxdomain = strstr(input_buffer, "Non-existent") || strstr(input_buffer, "** server can't find") || | 356 | const int nxdomain = strstr(input_buffer, "Non-existent") || strstr(input_buffer, "** server can't find") || |
335 | strstr(input_buffer, "** Can't find") || strstr(input_buffer, "NXDOMAIN"); | 357 | strstr(input_buffer, "** Can't find") || strstr(input_buffer, "NXDOMAIN"); |
336 | if (nxdomain) | 358 | if (nxdomain) { |
337 | *is_nxdomain = true; | 359 | *is_nxdomain = true; |
360 | } | ||
338 | 361 | ||
339 | /* the DNS lookup timed out */ | 362 | /* the DNS lookup timed out */ |
340 | if (strstr(input_buffer, _("Note: nslookup is deprecated and may be removed from future releases.")) || | 363 | if (strstr(input_buffer, _("Note: nslookup is deprecated and may be removed from future releases.")) || |
341 | strstr(input_buffer, _("Consider using the `dig' or `host' programs instead. Run nslookup with")) || | 364 | strstr(input_buffer, _("Consider using the `dig' or `host' programs instead. Run nslookup with")) || |
342 | strstr(input_buffer, _("the `-sil[ent]' option to prevent this message from appearing."))) | 365 | strstr(input_buffer, _("the `-sil[ent]' option to prevent this message from appearing."))) { |
343 | return STATE_OK; | 366 | return STATE_OK; |
367 | } | ||
344 | 368 | ||
345 | /* DNS server is not running... */ | 369 | /* DNS server is not running... */ |
346 | else if (strstr(input_buffer, "No response from server")) | 370 | else if (strstr(input_buffer, "No response from server")) { |
347 | die(STATE_CRITICAL, _("No response from DNS %s\n"), dns_server); | 371 | die(STATE_CRITICAL, _("No response from DNS %s\n"), dns_server); |
348 | else if (strstr(input_buffer, "no servers could be reached")) | 372 | } else if (strstr(input_buffer, "no servers could be reached")) { |
349 | die(STATE_CRITICAL, _("No response from DNS %s\n"), dns_server); | 373 | die(STATE_CRITICAL, _("No response from DNS %s\n"), dns_server); |
374 | } | ||
350 | 375 | ||
351 | /* Host name is valid, but server doesn't have records... */ | 376 | /* Host name is valid, but server doesn't have records... */ |
352 | else if (strstr(input_buffer, "No records")) | 377 | else if (strstr(input_buffer, "No records")) { |
353 | die(STATE_CRITICAL, _("DNS %s has no records\n"), dns_server); | 378 | die(STATE_CRITICAL, _("DNS %s has no records\n"), dns_server); |
379 | } | ||
354 | 380 | ||
355 | /* Connection was refused */ | 381 | /* Connection was refused */ |
356 | else if (strstr(input_buffer, "Connection refused") || strstr(input_buffer, "Couldn't find server") || | 382 | else if (strstr(input_buffer, "Connection refused") || strstr(input_buffer, "Couldn't find server") || |
357 | strstr(input_buffer, "Refused") || (strstr(input_buffer, "** server can't find") && strstr(input_buffer, ": REFUSED"))) | 383 | strstr(input_buffer, "Refused") || (strstr(input_buffer, "** server can't find") && strstr(input_buffer, ": REFUSED"))) { |
358 | die(STATE_CRITICAL, _("Connection to DNS %s was refused\n"), dns_server); | 384 | die(STATE_CRITICAL, _("Connection to DNS %s was refused\n"), dns_server); |
385 | } | ||
359 | 386 | ||
360 | /* Query refused (usually by an ACL in the namserver) */ | 387 | /* Query refused (usually by an ACL in the namserver) */ |
361 | else if (strstr(input_buffer, "Query refused")) | 388 | else if (strstr(input_buffer, "Query refused")) { |
362 | die(STATE_CRITICAL, _("Query was refused by DNS server at %s\n"), dns_server); | 389 | die(STATE_CRITICAL, _("Query was refused by DNS server at %s\n"), dns_server); |
390 | } | ||
363 | 391 | ||
364 | /* No information (e.g. nameserver IP has two PTR records) */ | 392 | /* No information (e.g. nameserver IP has two PTR records) */ |
365 | else if (strstr(input_buffer, "No information")) | 393 | else if (strstr(input_buffer, "No information")) { |
366 | die(STATE_CRITICAL, _("No information returned by DNS server at %s\n"), dns_server); | 394 | die(STATE_CRITICAL, _("No information returned by DNS server at %s\n"), dns_server); |
395 | } | ||
367 | 396 | ||
368 | /* Network is unreachable */ | 397 | /* Network is unreachable */ |
369 | else if (strstr(input_buffer, "Network is unreachable")) | 398 | else if (strstr(input_buffer, "Network is unreachable")) { |
370 | die(STATE_CRITICAL, _("Network is unreachable\n")); | 399 | die(STATE_CRITICAL, _("Network is unreachable\n")); |
400 | } | ||
371 | 401 | ||
372 | /* Internal server failure */ | 402 | /* Internal server failure */ |
373 | else if (strstr(input_buffer, "Server failure")) | 403 | else if (strstr(input_buffer, "Server failure")) { |
374 | die(STATE_CRITICAL, _("DNS failure for %s\n"), dns_server); | 404 | die(STATE_CRITICAL, _("DNS failure for %s\n"), dns_server); |
405 | } | ||
375 | 406 | ||
376 | /* Request error or the DNS lookup timed out */ | 407 | /* Request error or the DNS lookup timed out */ |
377 | else if (strstr(input_buffer, "Format error") || strstr(input_buffer, "Timed out")) | 408 | else if (strstr(input_buffer, "Format error") || strstr(input_buffer, "Timed out")) { |
378 | return STATE_WARNING; | 409 | return STATE_WARNING; |
410 | } | ||
379 | 411 | ||
380 | return STATE_OK; | 412 | return STATE_OK; |
381 | } | 413 | } |
382 | 414 | ||
383 | /* process command-line arguments */ | 415 | /* process command-line arguments */ |
384 | int process_arguments(int argc, char **argv) { | 416 | check_dns_config_wrapper process_arguments(int argc, char **argv) { |
385 | int c; | ||
386 | char *warning = NULL; | ||
387 | char *critical = NULL; | ||
388 | |||
389 | int opt_index = 0; | ||
390 | static struct option long_opts[] = {{"help", no_argument, 0, 'h'}, | 417 | static struct option long_opts[] = {{"help", no_argument, 0, 'h'}, |
391 | {"version", no_argument, 0, 'V'}, | 418 | {"version", no_argument, 0, 'V'}, |
392 | {"verbose", no_argument, 0, 'v'}, | 419 | {"verbose", no_argument, 0, 'v'}, |
@@ -402,20 +429,34 @@ int process_arguments(int argc, char **argv) { | |||
402 | {"critical", required_argument, 0, 'c'}, | 429 | {"critical", required_argument, 0, 'c'}, |
403 | {0, 0, 0, 0}}; | 430 | {0, 0, 0, 0}}; |
404 | 431 | ||
405 | if (argc < 2) | 432 | check_dns_config_wrapper result = { |
406 | return ERROR; | 433 | .config = check_dns_config_init(), |
434 | .errorcode = OK, | ||
435 | }; | ||
407 | 436 | ||
408 | for (c = 1; c < argc; c++) | 437 | if (argc < 2) { |
409 | if (strcmp("-to", argv[c]) == 0) | 438 | result.errorcode = ERROR; |
410 | strcpy(argv[c], "-t"); | 439 | return result; |
440 | } | ||
411 | 441 | ||
412 | while (1) { | 442 | for (int index = 1; index < argc; index++) { |
413 | c = getopt_long(argc, argv, "hVvALnt:H:s:r:a:w:c:", long_opts, &opt_index); | 443 | if (strcmp("-to", argv[index]) == 0) { |
444 | strcpy(argv[index], "-t"); | ||
445 | } | ||
446 | } | ||
447 | |||
448 | char *warning = NULL; | ||
449 | char *critical = NULL; | ||
450 | int opt_index = 0; | ||
451 | int index = 0; | ||
452 | while (true) { | ||
453 | index = getopt_long(argc, argv, "hVvALnt:H:s:r:a:w:c:", long_opts, &opt_index); | ||
414 | 454 | ||
415 | if (c == -1 || c == EOF) | 455 | if (index == -1 || index == EOF) { |
416 | break; | 456 | break; |
457 | } | ||
417 | 458 | ||
418 | switch (c) { | 459 | switch (index) { |
419 | case 'h': /* help */ | 460 | case 'h': /* help */ |
420 | print_help(); | 461 | print_help(); |
421 | exit(STATE_UNKNOWN); | 462 | exit(STATE_UNKNOWN); |
@@ -429,54 +470,63 @@ int process_arguments(int argc, char **argv) { | |||
429 | timeout_interval = atoi(optarg); | 470 | timeout_interval = atoi(optarg); |
430 | break; | 471 | break; |
431 | case 'H': /* hostname */ | 472 | case 'H': /* hostname */ |
432 | if (strlen(optarg) >= ADDRESS_LENGTH) | 473 | if (strlen(optarg) >= ADDRESS_LENGTH) { |
433 | die(STATE_UNKNOWN, _("Input buffer overflow\n")); | 474 | die(STATE_UNKNOWN, _("Input buffer overflow\n")); |
434 | strcpy(query_address, optarg); | 475 | } |
476 | strcpy(result.config.query_address, optarg); | ||
435 | break; | 477 | break; |
436 | case 's': /* server name */ | 478 | case 's': /* server name */ |
437 | /* TODO: this host_or_die check is probably unnecessary. | 479 | /* TODO: this host_or_die check is probably unnecessary. |
438 | * Better to confirm nslookup response matches */ | 480 | * Better to confirm nslookup response matches */ |
439 | host_or_die(optarg); | 481 | host_or_die(optarg); |
440 | if (strlen(optarg) >= ADDRESS_LENGTH) | 482 | if (strlen(optarg) >= ADDRESS_LENGTH) { |
441 | die(STATE_UNKNOWN, _("Input buffer overflow\n")); | 483 | die(STATE_UNKNOWN, _("Input buffer overflow\n")); |
442 | strcpy(dns_server, optarg); | 484 | } |
485 | strcpy(result.config.dns_server, optarg); | ||
443 | break; | 486 | break; |
444 | case 'r': /* reverse server name */ | 487 | case 'r': /* reverse server name */ |
445 | /* TODO: Is this host_or_die necessary? */ | 488 | /* TODO: Is this host_or_die necessary? */ |
489 | // TODO This does not do anything!!! 2025-03-08 rincewind | ||
446 | host_or_die(optarg); | 490 | host_or_die(optarg); |
447 | if (strlen(optarg) >= ADDRESS_LENGTH) | 491 | if (strlen(optarg) >= ADDRESS_LENGTH) { |
448 | die(STATE_UNKNOWN, _("Input buffer overflow\n")); | 492 | die(STATE_UNKNOWN, _("Input buffer overflow\n")); |
493 | } | ||
494 | static char ptr_server[ADDRESS_LENGTH] = ""; | ||
449 | strcpy(ptr_server, optarg); | 495 | strcpy(ptr_server, optarg); |
450 | break; | 496 | break; |
451 | case 'a': /* expected address */ | 497 | case 'a': /* expected address */ |
452 | if (strlen(optarg) >= ADDRESS_LENGTH) | 498 | if (strlen(optarg) >= ADDRESS_LENGTH) { |
453 | die(STATE_UNKNOWN, _("Input buffer overflow\n")); | 499 | die(STATE_UNKNOWN, _("Input buffer overflow\n")); |
500 | } | ||
454 | if (strchr(optarg, ',') != NULL) { | 501 | if (strchr(optarg, ',') != NULL) { |
455 | char *comma = strchr(optarg, ','); | 502 | char *comma = strchr(optarg, ','); |
456 | while (comma != NULL) { | 503 | while (comma != NULL) { |
457 | expected_address = (char **)realloc(expected_address, (expected_address_cnt + 1) * sizeof(char **)); | 504 | result.config.expected_address = |
458 | expected_address[expected_address_cnt] = strndup(optarg, comma - optarg); | 505 | (char **)realloc(result.config.expected_address, (result.config.expected_address_cnt + 1) * sizeof(char **)); |
459 | expected_address_cnt++; | 506 | result.config.expected_address[result.config.expected_address_cnt] = strndup(optarg, comma - optarg); |
507 | result.config.expected_address_cnt++; | ||
460 | optarg = comma + 1; | 508 | optarg = comma + 1; |
461 | comma = strchr(optarg, ','); | 509 | comma = strchr(optarg, ','); |
462 | } | 510 | } |
463 | expected_address = (char **)realloc(expected_address, (expected_address_cnt + 1) * sizeof(char **)); | 511 | result.config.expected_address = |
464 | expected_address[expected_address_cnt] = strdup(optarg); | 512 | (char **)realloc(result.config.expected_address, (result.config.expected_address_cnt + 1) * sizeof(char **)); |
465 | expected_address_cnt++; | 513 | result.config.expected_address[result.config.expected_address_cnt] = strdup(optarg); |
514 | result.config.expected_address_cnt++; | ||
466 | } else { | 515 | } else { |
467 | expected_address = (char **)realloc(expected_address, (expected_address_cnt + 1) * sizeof(char **)); | 516 | result.config.expected_address = |
468 | expected_address[expected_address_cnt] = strdup(optarg); | 517 | (char **)realloc(result.config.expected_address, (result.config.expected_address_cnt + 1) * sizeof(char **)); |
469 | expected_address_cnt++; | 518 | result.config.expected_address[result.config.expected_address_cnt] = strdup(optarg); |
519 | result.config.expected_address_cnt++; | ||
470 | } | 520 | } |
471 | break; | 521 | break; |
472 | case 'n': /* expect NXDOMAIN */ | 522 | case 'n': /* expect NXDOMAIN */ |
473 | expect_nxdomain = true; | 523 | result.config.expect_nxdomain = true; |
474 | break; | 524 | break; |
475 | case 'A': /* expect authority */ | 525 | case 'A': /* expect authority */ |
476 | expect_authority = true; | 526 | result.config.expect_authority = true; |
477 | break; | 527 | break; |
478 | case 'L': /* all must match */ | 528 | case 'L': /* all must match */ |
479 | all_match = true; | 529 | result.config.all_match = true; |
480 | break; | 530 | break; |
481 | case 'w': | 531 | case 'w': |
482 | warning = optarg; | 532 | warning = optarg; |
@@ -489,38 +539,42 @@ int process_arguments(int argc, char **argv) { | |||
489 | } | 539 | } |
490 | } | 540 | } |
491 | 541 | ||
492 | c = optind; | 542 | index = optind; |
493 | if (strlen(query_address) == 0 && c < argc) { | 543 | if (strlen(result.config.query_address) == 0 && index < argc) { |
494 | if (strlen(argv[c]) >= ADDRESS_LENGTH) | 544 | if (strlen(argv[index]) >= ADDRESS_LENGTH) { |
495 | die(STATE_UNKNOWN, _("Input buffer overflow\n")); | 545 | die(STATE_UNKNOWN, _("Input buffer overflow\n")); |
496 | strcpy(query_address, argv[c++]); | 546 | } |
547 | strcpy(result.config.query_address, argv[index++]); | ||
497 | } | 548 | } |
498 | 549 | ||
499 | if (strlen(dns_server) == 0 && c < argc) { | 550 | if (strlen(result.config.dns_server) == 0 && index < argc) { |
500 | /* TODO: See -s option */ | 551 | /* TODO: See -s option */ |
501 | host_or_die(argv[c]); | 552 | host_or_die(argv[index]); |
502 | if (strlen(argv[c]) >= ADDRESS_LENGTH) | 553 | if (strlen(argv[index]) >= ADDRESS_LENGTH) { |
503 | die(STATE_UNKNOWN, _("Input buffer overflow\n")); | 554 | die(STATE_UNKNOWN, _("Input buffer overflow\n")); |
504 | strcpy(dns_server, argv[c++]); | 555 | } |
556 | strcpy(result.config.dns_server, argv[index++]); | ||
505 | } | 557 | } |
506 | 558 | ||
507 | set_thresholds(&time_thresholds, warning, critical); | 559 | set_thresholds(&result.config.time_thresholds, warning, critical); |
508 | 560 | ||
509 | return validate_arguments(); | 561 | return validate_arguments(result); |
510 | } | 562 | } |
511 | 563 | ||
512 | int validate_arguments(void) { | 564 | check_dns_config_wrapper validate_arguments(check_dns_config_wrapper config_wrapper) { |
513 | if (query_address[0] == 0) { | 565 | if (config_wrapper.config.query_address[0] == 0) { |
514 | printf("missing --host argument\n"); | 566 | printf("missing --host argument\n"); |
515 | return ERROR; | 567 | config_wrapper.errorcode = ERROR; |
568 | return config_wrapper; | ||
516 | } | 569 | } |
517 | 570 | ||
518 | if (expected_address_cnt > 0 && expect_nxdomain) { | 571 | if (config_wrapper.config.expected_address_cnt > 0 && config_wrapper.config.expect_nxdomain) { |
519 | printf("--expected-address and --expect-nxdomain cannot be combined\n"); | 572 | printf("--expected-address and --expect-nxdomain cannot be combined\n"); |
520 | return ERROR; | 573 | config_wrapper.errorcode = ERROR; |
574 | return config_wrapper; | ||
521 | } | 575 | } |
522 | 576 | ||
523 | return OK; | 577 | return config_wrapper; |
524 | } | 578 | } |
525 | 579 | ||
526 | void print_help(void) { | 580 | void print_help(void) { |
diff --git a/plugins/check_dns.d/config.h b/plugins/check_dns.d/config.h new file mode 100644 index 00000000..9ec4eb82 --- /dev/null +++ b/plugins/check_dns.d/config.h | |||
@@ -0,0 +1,34 @@ | |||
1 | #pragma once | ||
2 | |||
3 | #include "../../config.h" | ||
4 | #include "thresholds.h" | ||
5 | #include <stddef.h> | ||
6 | |||
7 | #define ADDRESS_LENGTH 256 | ||
8 | |||
9 | typedef struct { | ||
10 | bool all_match; | ||
11 | char dns_server[ADDRESS_LENGTH]; | ||
12 | char query_address[ADDRESS_LENGTH]; | ||
13 | bool expect_nxdomain; | ||
14 | bool expect_authority; | ||
15 | char **expected_address; | ||
16 | size_t expected_address_cnt; | ||
17 | |||
18 | thresholds *time_thresholds; | ||
19 | } check_dns_config; | ||
20 | |||
21 | check_dns_config check_dns_config_init() { | ||
22 | check_dns_config tmp = { | ||
23 | .all_match = false, | ||
24 | .dns_server = "", | ||
25 | .query_address = "", | ||
26 | .expect_nxdomain = false, | ||
27 | .expect_authority = false, | ||
28 | .expected_address = NULL, | ||
29 | .expected_address_cnt = 0, | ||
30 | |||
31 | .time_thresholds = NULL, | ||
32 | }; | ||
33 | return tmp; | ||
34 | } | ||