check_dns.c 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697
  1. /*****************************************************************************
  2. *
  3. * Nagios check_dns plugin
  4. *
  5. * License: GPL
  6. * Copyright (c) 2000-2018 Nagios Plugins Development Team
  7. *
  8. * Description:
  9. *
  10. * This file contains the check_dns plugin
  11. *
  12. * LIMITATION: nslookup on Solaris 7 can return output over 2 lines, which
  13. * will not be picked up by this plugin
  14. *
  15. *
  16. * This program is free software: you can redistribute it and/or modify
  17. * it under the terms of the GNU General Public License as published by
  18. * the Free Software Foundation, either version 3 of the License, or
  19. * (at your option) any later version.
  20. *
  21. * This program is distributed in the hope that it will be useful,
  22. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  23. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  24. * GNU General Public License for more details.
  25. *
  26. * You should have received a copy of the GNU General Public License
  27. * along with this program. If not, see <http://www.gnu.org/licenses/>.
  28. *
  29. *
  30. *****************************************************************************/
  31. #define IF_RECORD(label, querytype, verb_str, comp_str) if (strstr (chld_out.line[i], label) && (strncmp(query_type, querytype, query_size) == 0 || strncmp(query_type, "-querytype=ANY", query_size) == 0)) { \
  32. if (verbose) { \
  33. printf(verb_str); \
  34. } \
  35. temp_buffer = rindex (chld_out.line[i], comp_str); \
  36. addresses[n_addresses++] = check_new_address(temp_buffer); \
  37. memset(query_found, '\0', sizeof(query_found)); \
  38. strncpy(query_found, querytype, sizeof(query_found));
  39. const char *progname = "check_dns";
  40. const char *copyright = "2000-2018";
  41. const char *email = "devel@nagios-plugins.org";
  42. #include "common.h"
  43. #include "utils.h"
  44. #include "utils_base.h"
  45. #include "netutils.h"
  46. #include "runcmd.h"
  47. int process_arguments (int, char **);
  48. int validate_arguments (void);
  49. int error_scan (char *);
  50. void print_help (void);
  51. void print_usage (void);
  52. /* Allow up to 4096 input length, this is helpful
  53. when the TXT records returned have multiple 255 legth values returned */
  54. #define ADDRESS_LENGTH 4096
  55. char query_address[ADDRESS_LENGTH] = "";
  56. char dns_server[ADDRESS_LENGTH] = "";
  57. char tmp_dns_server[ADDRESS_LENGTH] = "";
  58. char ptr_server[ADDRESS_LENGTH] = "";
  59. char query_type[16] = "";
  60. int query_set = FALSE;
  61. int verbose = FALSE;
  62. char **expected_address = NULL;
  63. int expected_address_cnt = 0;
  64. int expect_authority = FALSE;
  65. int accept_cname = FALSE;
  66. thresholds *time_thresholds = NULL;
  67. static int
  68. qstrcmp(const void *p1, const void *p2)
  69. {
  70. /* The actual arguments to this function are "pointers to
  71. pointers to char", but strcmp() arguments are "pointers
  72. to char", hence the following cast plus dereference */
  73. return strcmp(* (char * const *) p1, * (char * const *) p2);
  74. }
  75. char *
  76. check_new_address(char *temp_buffer)
  77. {
  78. temp_buffer++;
  79. /* Strip leading spaces */
  80. for (; *temp_buffer != '\0' && *temp_buffer == ' '; temp_buffer++) {
  81. /* NOOP */;
  82. }
  83. strip(temp_buffer);
  84. if (temp_buffer==NULL || strlen(temp_buffer)==0) {
  85. die (STATE_CRITICAL, "%s%s%s\n", _("DNS CRITICAL - '"), NSLOOKUP_COMMAND, _("' returned empty host name string"));
  86. }
  87. return temp_buffer;
  88. }
  89. int
  90. main (int argc, char **argv)
  91. {
  92. char *command_line = NULL;
  93. char input_buffer[MAX_INPUT_BUFFER];
  94. char *address = NULL; /* comma separated str with addrs/ptrs (sorted) */
  95. char **addresses = NULL;
  96. int n_addresses = 0;
  97. char *msg = NULL;
  98. char query_found[24] = "";
  99. int query_size = 24;
  100. char *temp_buffer = NULL;
  101. int non_authoritative = FALSE;
  102. int result = STATE_UNKNOWN;
  103. double elapsed_time;
  104. long microsec;
  105. struct timeval tv;
  106. int parse_address = FALSE; /* This flag scans for Address: but only after Name: */
  107. output chld_out, chld_err;
  108. size_t i;
  109. setlocale (LC_ALL, "");
  110. bindtextdomain (PACKAGE, LOCALEDIR);
  111. textdomain (PACKAGE);
  112. /* Set signal handling and alarm */
  113. if (signal (SIGALRM, runcmd_timeout_alarm_handler) == SIG_ERR) {
  114. usage_va(_("Cannot catch SIGALRM"));
  115. }
  116. /* Parse extra opts if any */
  117. argv=np_extra_opts (&argc, argv, progname);
  118. if (process_arguments (argc, argv) == ERROR) {
  119. usage_va(_("Could not parse arguments"));
  120. }
  121. /* get the command to run */
  122. xasprintf (&command_line, "%s %s %s %s", NSLOOKUP_COMMAND, query_type, query_address, dns_server);
  123. alarm (timeout_interval);
  124. gettimeofday (&tv, NULL);
  125. if (verbose) {
  126. printf ("%s\n", command_line);
  127. }
  128. /* run the command */
  129. if((np_runcmd(command_line, &chld_out, &chld_err, 0)) != 0) {
  130. msg = strdup(_("nslookup returned an error status"));
  131. result = STATE_WARNING;
  132. }
  133. /* scan stdout */
  134. for(i = 0; i < chld_out.lines; i++) {
  135. if (addresses == NULL) {
  136. addresses = malloc(sizeof(*addresses)*10);
  137. }
  138. else if (!(n_addresses % 10)) {
  139. addresses = realloc(addresses,sizeof(*addresses) * (n_addresses + 10));
  140. }
  141. if (verbose) {
  142. puts(chld_out.line[i]);
  143. }
  144. /* bug ID: 2946553 - Older versions of bind will use all available dns
  145. + servers, we have to match the one specified */
  146. if (strlen(dns_server) > 0 && strstr(chld_out.line[i], "Server:")) {
  147. temp_buffer = strchr(chld_out.line[i], ':');
  148. temp_buffer++;
  149. if (temp_buffer > (char *)1) {
  150. /* Strip leading tabs */
  151. for (; *temp_buffer != '\0' && *temp_buffer == '\t'; temp_buffer++) {
  152. /* NOOP */;
  153. }
  154. strip(temp_buffer);
  155. if (temp_buffer==NULL || strlen(temp_buffer)==0) {
  156. die (STATE_CRITICAL, "%s%s%s\n", _("DNS CRITICAL - '"), NSLOOKUP_COMMAND, _("' returned empty server string"));
  157. }
  158. if (strcmp(temp_buffer, dns_server) != 0) {
  159. die (STATE_CRITICAL, "%s %s\n", _("DNS CRITICAL - No response from DNS server:"), dns_server);
  160. }
  161. }
  162. }
  163. /* Provide the server name\ip to error_scan when not using -s */
  164. else if (strlen(tmp_dns_server) == 0) {
  165. temp_buffer = strchr(chld_out.line[i], ':');
  166. temp_buffer++;
  167. if (temp_buffer > (char *)1) {
  168. /* Strip leading tabs */
  169. for (; *temp_buffer != '\0' && *temp_buffer == '\t'; temp_buffer++) {
  170. /* NOOP */;
  171. }
  172. strip(temp_buffer);
  173. strncpy(tmp_dns_server, temp_buffer, ADDRESS_LENGTH);
  174. }
  175. }
  176. if (strstr (chld_out.line[i], "Authoritative answers can be found from:")) {
  177. break;
  178. }
  179. /* the server is responding, we just got the host name...*/
  180. if (strstr (chld_out.line[i], "Name:")) {
  181. parse_address = TRUE;
  182. }
  183. /* begin handling types of records */
  184. IF_RECORD("AAAA address", "-querytype=AAAA", "Found AAAA record\n", ' ') }
  185. else IF_RECORD("exchanger =", "-querytype=MX", "Found MX record\n", '=') }
  186. else IF_RECORD("service =", "-querytype=SRV", "Found SRV record\n", ' ') }
  187. else IF_RECORD("nameserver =", "-querytype=NS", "Found NS record\n", ' ') }
  188. else IF_RECORD("dname =", "-querytype=DNAME", "Found DNAME record\n", ' ') }
  189. else IF_RECORD("protocol =", "-querytype=WKS", "Found WKS record\n", ' ') }
  190. else if (strstr (chld_out.line[i], "text =") && (strncmp(query_type, "-querytype=TXT", query_size) == 0 || strncmp(query_type, "-querytype=ANY", query_size) == 0)) {
  191. if (verbose) {
  192. printf("Found TXT record\n");
  193. }
  194. temp_buffer = index(chld_out.line[i], '"');
  195. --temp_buffer;
  196. addresses[n_addresses++] = check_new_address(temp_buffer);
  197. memset(query_found, '\0', sizeof(query_found));
  198. strncpy(query_found, "-querytype=TXT", sizeof(query_found));
  199. }
  200. /* only matching for origin records, if requested other fields could be included at a later date */
  201. else IF_RECORD("origin =", "-querytype=SOA", "Found SOA record\n", ' ') }
  202. /* cnames cannot use macro as we must check for accepting them separately */
  203. else if (accept_cname && strstr (chld_out.line[i], "canonical name =") && (strncmp(query_type, "-querytype=CNAME", query_size) == 0 || strncmp(query_type, "-querytype=ANY", query_size) == 0)) {
  204. if (verbose) {
  205. printf("Found CNAME record\n");
  206. }
  207. temp_buffer = index (chld_out.line[i], '=');
  208. addresses[n_addresses++] = check_new_address(temp_buffer);
  209. strncpy(query_found, "-querytype=CNAME", sizeof(query_found));
  210. }
  211. /* does not need strncmp as we want A at all times unless another record match */
  212. else if (parse_address == TRUE && (strstr (chld_out.line[i], "Address:") || strstr (chld_out.line[i], "Addresses:"))) {
  213. temp_buffer = index (chld_out.line[i], ':');
  214. char *temp_address = check_new_address(temp_buffer);
  215. addresses[n_addresses++] = temp_address;
  216. /* Bind 9.11.x onwards reports AAAA records the same as A records */
  217. /* It is assumed that nslookup returns a valid IPv4 or IPv6 address */
  218. /* Hence we'll look for a colon : in the address to detect an IPv6 address */
  219. if (strstr(temp_address, ":")) {
  220. if (verbose) {
  221. printf("Found AAAA record\n");
  222. }
  223. strncpy(query_found, "-querytype=AAAA", sizeof(query_found));
  224. }
  225. else {
  226. if (verbose) {
  227. printf("Found A record\n");
  228. }
  229. strncpy(query_found, "-querytype=A", sizeof(query_found));
  230. }
  231. }
  232. /* must be after other records with "name" as an identifier, as ptr does not spefify */
  233. else IF_RECORD("name =", "-querytype=PTR", "Found PTR record\n", ' ') }
  234. /* needed for non-query ptr\reverse lookup checks */
  235. else if (strstr(chld_out.line[i], ".in-addr.arpa") && !query_set) {
  236. if ((temp_buffer = strstr(chld_out.line[i], "name = "))) {
  237. addresses[n_addresses++] = strdup(temp_buffer);
  238. }
  239. else {
  240. xasprintf(&msg, "%s %s %s %s", _("Warning plugin error"));
  241. result = STATE_WARNING;
  242. }
  243. }
  244. if (strstr (chld_out.line[i], _("Non-authoritative answer:"))) {
  245. non_authoritative = TRUE;
  246. }
  247. int tmp = error_scan(chld_out.line[i]);
  248. result = (result == STATE_UNKNOWN)
  249. ? tmp
  250. : (result < tmp)
  251. ? tmp : result;
  252. if (result != STATE_OK) {
  253. msg = strchr (chld_out.line[i], ':');
  254. if(msg) {
  255. msg++;
  256. }
  257. else {
  258. msg = chld_out.line[i];
  259. }
  260. break;
  261. }
  262. } /*This is the end of the scan stdout loop */
  263. /* scan stderr */
  264. for(i = 0; i < chld_err.lines; i++) {
  265. if (verbose) {
  266. puts(chld_err.line[i]);
  267. }
  268. if (error_scan (chld_err.line[i]) != STATE_OK) {
  269. result = max_state (result, error_scan (chld_err.line[i]));
  270. msg = strchr(chld_err.line[i], ':');
  271. if(msg) {
  272. msg++;
  273. }
  274. else {
  275. msg = chld_err.line[i];
  276. }
  277. }
  278. }
  279. if (addresses) {
  280. int i,slen;
  281. char *adrp;
  282. qsort(addresses, n_addresses, sizeof(*addresses), qstrcmp);
  283. for(i=0, slen=1; i < n_addresses; i++) {
  284. slen += strlen(addresses[i])+1;
  285. }
  286. adrp = address = malloc(slen);
  287. for(i=0; i < n_addresses; i++) {
  288. if (i) *adrp++ = ',';
  289. strcpy(adrp, addresses[i]);
  290. adrp += strlen(addresses[i]);
  291. }
  292. *adrp = 0;
  293. }
  294. else {
  295. die (STATE_CRITICAL, "%s%s%s\n", _("DNS CRITICAL - '"), NSLOOKUP_COMMAND, _("' msg parsing exited with no address"));
  296. }
  297. /* compare to expected address */
  298. if (result == STATE_OK && expected_address_cnt > 0) {
  299. result = STATE_CRITICAL;
  300. temp_buffer = "";
  301. for (i=0; i<expected_address_cnt; i++) {
  302. /* check if we get a match and prepare an error string */
  303. if (strcasecmp(address, expected_address[i]) == 0) result = STATE_OK;
  304. xasprintf(&temp_buffer, "%s%s; ", temp_buffer, expected_address[i]);
  305. }
  306. if (result == STATE_CRITICAL) {
  307. /* Strip off last semicolon... */
  308. temp_buffer[strlen(temp_buffer)-2] = '\0';
  309. xasprintf(&msg, "%s%s%s%s%s", _("expected '"), temp_buffer, _("' but got '"), address, "'");
  310. }
  311. }
  312. /* check if authoritative */
  313. if (result == STATE_OK && expect_authority && non_authoritative) {
  314. result = STATE_CRITICAL;
  315. if (strncmp(dns_server, "", 1)) {
  316. xasprintf(&msg, "%s %s %s %s", _("server"), dns_server, _("is not authoritative for"), query_address);
  317. }
  318. else {
  319. xasprintf(&msg, "%s %s", _("there is no authoritative server for"), query_address);
  320. }
  321. }
  322. /* compare query type to query found, if query type is ANY we can skip as any record is accepted*/
  323. if (result == STATE_OK && strncmp(query_type, "", 1) && (strncmp(query_type, "-querytype=ANY", 15) != 0)) {
  324. if (strncmp(query_type, query_found, 16) != 0) {
  325. if (verbose) {
  326. printf( "%s %s %s %s %s\n", _("Failed query for"), query_type, _("only found"), query_found, _(", or nothing"));
  327. }
  328. result = STATE_CRITICAL;
  329. xasprintf(&msg, "%s %s %s %s", _("query type of"), query_type, _("was not found for"), query_address);
  330. }
  331. }
  332. microsec = deltime (tv);
  333. elapsed_time = (double)microsec / 1.0e6;
  334. if (result == STATE_OK) {
  335. result = get_status(elapsed_time, time_thresholds);
  336. if (result == STATE_OK) {
  337. printf ("%s %s: ", _("DNS"), _("OK"));
  338. }
  339. else if (result == STATE_WARNING) {
  340. printf ("%s %s: ", _("DNS"), _("WARNING"));
  341. }
  342. else if (result == STATE_CRITICAL) {
  343. printf ("%s %s: ", _("DNS"), _("CRITICAL"));
  344. }
  345. printf (ngettext("%.3f second response time", "%.3f seconds response time", elapsed_time), elapsed_time);
  346. printf (". %s %s %s", query_address, _("returns"), address);
  347. if ((time_thresholds->warning != NULL) && (time_thresholds->critical != NULL)) {
  348. printf ("|%s\n", fperfdata ("time", elapsed_time, "s",
  349. TRUE, time_thresholds->warning->end,
  350. TRUE, time_thresholds->critical->end,
  351. TRUE, 0, FALSE, 0));
  352. }
  353. else if ((time_thresholds->warning == NULL) && (time_thresholds->critical != NULL)) {
  354. printf ("|%s\n", fperfdata ("time", elapsed_time, "s",
  355. FALSE, 0,
  356. TRUE, time_thresholds->critical->end,
  357. TRUE, 0, FALSE, 0));
  358. }
  359. else if ((time_thresholds->warning != NULL) && (time_thresholds->critical == NULL)) {
  360. printf ("|%s\n", fperfdata ("time", elapsed_time, "s",
  361. TRUE, time_thresholds->warning->end,
  362. FALSE, 0,
  363. TRUE, 0, FALSE, 0));
  364. }
  365. else {
  366. printf ("|%s\n", fperfdata ("time", elapsed_time, "s", FALSE, 0, FALSE, 0, TRUE, 0, FALSE, 0));
  367. }
  368. }
  369. else if (result == STATE_WARNING) {
  370. printf ("%s %s\n", _("DNS WARNING -"), !strcmp (msg, "") ? _("Probably a non-existent host/domain") : msg);
  371. }
  372. else if (result == STATE_CRITICAL) {
  373. printf ("%s %s\n", _("DNS CRITICAL -"), !strcmp (msg, "") ? _("Probably a non-existent host/domain") : msg);
  374. }
  375. else {
  376. printf ("%s %s\n", _("DNS UNKNOWN -"), !strcmp (msg, "") ? _("Probably a non-existent host/domain") : msg);
  377. }
  378. return result;
  379. }
  380. int
  381. error_scan (char *input_buffer)
  382. {
  383. /* the DNS lookup timed out */
  384. if (strstr (input_buffer, _("Note: nslookup is deprecated and may be removed from future releases.")) ||
  385. strstr (input_buffer, _("Consider using the `dig' or `host' programs instead. Run nslookup with")) ||
  386. strstr (input_buffer, _("the `-sil[ent]' option to prevent this message from appearing.")))
  387. {
  388. return STATE_OK;
  389. }
  390. /* DNS server is not running... */
  391. else if (strstr (input_buffer, "No response from server")) {
  392. die (STATE_CRITICAL, "%s %s\n", _("No response from DNS"), (strlen(dns_server)==0)?tmp_dns_server:dns_server);
  393. }
  394. /* Host name is valid, but server doesn't have records... */
  395. else if (strstr (input_buffer, "No records") || strstr (input_buffer, "No answer")) {
  396. die (STATE_CRITICAL, "%s %s %s\n", _("DNS"), (strlen(dns_server)==0)?tmp_dns_server:dns_server, _("has no records"));
  397. }
  398. /* Connection was refused */
  399. else if (strstr (input_buffer, "Connection refused") ||
  400. strstr (input_buffer, "Couldn't find server") ||
  401. strstr (input_buffer, "Refused") ||
  402. (strstr (input_buffer, "** server can't find") &&
  403. strstr (input_buffer, ": REFUSED")))
  404. {
  405. die (STATE_CRITICAL, "%s %s %s\n", _("Connection to DNS"), (strlen(dns_server)==0)?tmp_dns_server:dns_server, _("was refused"));
  406. }
  407. /* Query refused (usually by an ACL in the namserver) */
  408. else if (strstr (input_buffer, "Query refused")) {
  409. die (STATE_CRITICAL, "%s %s\n", _("Query was refused by DNS server at"), (strlen(dns_server)==0)?tmp_dns_server:dns_server);
  410. }
  411. /* No information (e.g. nameserver IP has two PTR records) */
  412. else if (strstr (input_buffer, "No information")) {
  413. die (STATE_CRITICAL, "%s %s\n", _("No information returned by DNS server at"), (strlen(dns_server)==0)?tmp_dns_server:dns_server);
  414. }
  415. /* Host or domain name does not exist */
  416. else if (strstr (input_buffer, "Non-existent") ||
  417. strstr (input_buffer, "** server can't find") ||
  418. strstr (input_buffer,"NXDOMAIN"))
  419. {
  420. die (STATE_CRITICAL, "%s %s %s\n", _("Domain"), query_address, _("was not found by the server"));
  421. }
  422. /* Network is unreachable */
  423. else if (strstr (input_buffer, "Network is unreachable")) {
  424. die (STATE_CRITICAL, "%s\n", _("Network is unreachable"));
  425. }
  426. /* Internal server failure */
  427. else if (strstr (input_buffer, "Server failure")) {
  428. die (STATE_CRITICAL, "%s %s\n", _("DNS failure for"), (strlen(dns_server)==0)?tmp_dns_server:dns_server);
  429. }
  430. /* Request error or the DNS lookup timed out */
  431. else if (strstr (input_buffer, "Format error") ||
  432. strstr (input_buffer, "Timed out"))
  433. {
  434. return STATE_WARNING;
  435. }
  436. return STATE_OK;
  437. }
  438. /* process command-line arguments */
  439. int
  440. process_arguments (int argc, char **argv)
  441. {
  442. int c;
  443. char *warning = NULL;
  444. char *critical = NULL;
  445. int opt_index = 0;
  446. static struct option long_opts[] = {
  447. {"help", no_argument, 0, 'h'},
  448. {"version", no_argument, 0, 'V'},
  449. {"verbose", no_argument, 0, 'v'},
  450. {"timeout", required_argument, 0, 't'},
  451. {"hostname", required_argument, 0, 'H'},
  452. {"server", required_argument, 0, 's'},
  453. {"reverse-server", required_argument, 0, 'r'},
  454. {"querytype", required_argument, 0, 'q'},
  455. {"expected-address", required_argument, 0, 'a'},
  456. {"expect-authority", no_argument, 0, 'A'},
  457. {"accept-cname", no_argument, 0, 'n'},
  458. {"warning", required_argument, 0, 'w'},
  459. {"critical", required_argument, 0, 'c'},
  460. {0, 0, 0, 0}
  461. };
  462. if (argc < 2) {
  463. return ERROR;
  464. }
  465. for (c = 1; c < argc; c++) {
  466. if (strcmp ("-to", argv[c]) == 0) {
  467. strcpy (argv[c], "-t");
  468. }
  469. }
  470. while (1) {
  471. c = getopt_long (argc, argv, "hVvAnt:H:s:r:a:q:w:c:", long_opts, &opt_index);
  472. if (c == -1 || c == EOF) {
  473. break;
  474. }
  475. switch (c) {
  476. /* help */
  477. case 'h':
  478. print_help ();
  479. exit (STATE_OK);
  480. /* version */
  481. case 'V':
  482. print_revision (progname, NP_VERSION);
  483. exit (STATE_OK);
  484. /* verbose */
  485. case 'v':
  486. verbose = TRUE;
  487. break;
  488. /* timeout period */
  489. case 't':
  490. timeout_interval = parse_timeout_string (optarg);
  491. break;
  492. /* hostname */
  493. case 'H':
  494. if (strlen (optarg) >= ADDRESS_LENGTH) {
  495. die (STATE_UNKNOWN, "%s\n", _("Input buffer overflow"));
  496. }
  497. strcpy (query_address, optarg);
  498. break;
  499. /* server name */
  500. case 's':
  501. /* TODO: this host_or_die check is probably unnecessary.
  502. * Better to confirm nslookup response matches */
  503. host_or_die(optarg);
  504. if (strlen (optarg) >= ADDRESS_LENGTH) {
  505. die (STATE_UNKNOWN, "%s\n", _("Input buffer overflow"));
  506. }
  507. strcpy (dns_server, optarg);
  508. break;
  509. /* reverse server name */
  510. case 'r':
  511. /* TODO: Is this host_or_die necessary? */
  512. host_or_die(optarg);
  513. if (strlen (optarg) >= ADDRESS_LENGTH) {
  514. die (STATE_UNKNOWN, "%s\n", _("Input buffer overflow"));
  515. }
  516. strcpy (ptr_server, optarg);
  517. break;
  518. /* expected address */
  519. case 'a':
  520. if (strlen (optarg) >= ADDRESS_LENGTH) {
  521. die (STATE_UNKNOWN, "%s\n", _("Input buffer overflow"));
  522. }
  523. expected_address = (char **)realloc(expected_address, (expected_address_cnt+1) * sizeof(char*));
  524. expected_address[expected_address_cnt] = strdup(optarg);
  525. expected_address_cnt++;
  526. break;
  527. /* querytype -- A or AAAA or ANY or SRV or TXT, etc. */
  528. case 'q':
  529. if (strlen (optarg) < 1 || strlen (optarg) > 5) {
  530. die (STATE_UNKNOWN, "%s\n", _("Missing valid querytype parameter. Try using 'A' or 'AAAA' or 'SRV' or 'ANY'"));
  531. }
  532. strntoupper(optarg, strlen(optarg));
  533. strcpy(query_type, "-querytype=");
  534. strcat(query_type, optarg);
  535. query_set = TRUE;
  536. /* logic is set such that we must accept cnames if they are querying for them */
  537. if (strcmp(query_type, "-querytype=CNAME") != 0) {
  538. break;
  539. }
  540. /* accept cname responses as a result */
  541. case 'n':
  542. accept_cname = TRUE;
  543. break;
  544. /* expect authority */
  545. case 'A':
  546. expect_authority = TRUE;
  547. break;
  548. case 'w':
  549. warning = optarg;
  550. break;
  551. case 'c':
  552. critical = optarg;
  553. break;
  554. /* args not parsable */
  555. default:
  556. usage5();
  557. }
  558. }
  559. set_thresholds(&time_thresholds, warning, critical);
  560. return validate_arguments ();
  561. }
  562. int
  563. validate_arguments ()
  564. {
  565. if (query_address[0] == 0) {
  566. return ERROR;
  567. }
  568. /* Bind 9.11.x onwards performs a query for both A and AAAA records */
  569. /* The previous default behavior of nslookup was just A records. */
  570. /* To ensure that exisitng users of this plugin do not get incorrect results */
  571. /* set the querytype to A if it has not already been specified. */
  572. /* If an end user wants both A and AAAA then they need to use ANY. */
  573. if (strcmp(query_type, "") == 0) {
  574. /*query_type = "-querytype=A";*/
  575. strcpy(query_type, "-querytype=");
  576. strcat(query_type, "A");
  577. query_set = TRUE;
  578. }
  579. return OK;
  580. }
  581. void
  582. print_help (void)
  583. {
  584. print_revision (progname, NP_VERSION);
  585. printf ("%s\n", "Copyright (c) 1999 Ethan Galstad <nagios@nagios.org>");
  586. printf (COPYRIGHT, copyright, email);
  587. printf ("%s\n", _("This plugin uses the nslookup program to obtain the IP address for the given host/domain query."));
  588. printf ("%s\n", _("An optional DNS server to use may be specified."));
  589. printf ("%s\n", _("If no DNS server is specified, the default server(s) specified in /etc/resolv.conf will be used."));
  590. printf ("\n\n");
  591. print_usage ();
  592. printf (UT_HELP_VRSN);
  593. printf (UT_EXTRA_OPTS);
  594. printf ("%s\n", " -H, --hostname=HOST");
  595. printf (" %s\n", _("The name or address you want to query"));
  596. printf ("%s\n", " -s, --server=HOST");
  597. printf (" %s\n", _("Optional DNS server you want to use for the lookup"));
  598. printf ("%s\n", " -q, --querytype=TYPE");
  599. printf (" %s\n", _("Optional DNS record query type where TYPE =(A, AAAA, SRV, TXT, MX, ANY)"));
  600. printf (" %s\n", _("The default query type is 'A' (IPv4 host entry)"));
  601. printf (" %s\n", _("BIND 9.11.x onwards supports both 'A' and 'AAAA', if you want both use 'ANY'"));
  602. printf ("%s\n", " -a, --expected-address=IP-ADDRESS|HOST");
  603. printf (" %s\n", _("Optional IP-ADDRESS you expect the DNS server to return. HOST must end with"));
  604. printf (" %s\n", _("a dot (.). This option can be repeated multiple times (Returns OK if any"));
  605. printf (" %s\n", _("value match). If multiple addresses are returned at once, you have to match"));
  606. printf (" %s\n", _("the whole string of addresses separated with commas (sorted alphabetically)."));
  607. printf (" %s\n", _("If you would like to test for the presence of a cname, combine with -n param."));
  608. printf ("%s\n", " -A, --expect-authority");
  609. printf (" %s\n", _("Optionally expect the DNS server to be authoritative for the lookup"));
  610. printf ("%s\n", " -n, --accept-cname");
  611. printf (" %s\n", _("Optionally accept cname responses as a valid result to a query"));
  612. printf (" %s\n", _("The default is to ignore cname responses as part of the result"));
  613. printf ("%s\n", " -w, --warning=seconds");
  614. printf (" %s\n", _("Return warning if elapsed time exceeds value. Default off"));
  615. printf ("%s\n", " -c, --critical=seconds");
  616. printf (" %s\n", _("Return critical if elapsed time exceeds value. Default off"));
  617. printf (UT_CONN_TIMEOUT, DEFAULT_SOCKET_TIMEOUT);
  618. printf (UT_SUPPORT);
  619. }
  620. void
  621. print_usage (void)
  622. {
  623. printf ("%s\n", _("Usage:"));
  624. printf ("%s %s\n", progname, "-H host [-s server] [-q type ] [-a expected-address] [-A] [-n] [-t timeout] [-w warn] [-c crit]");
  625. }