瀏覽代碼

check_snmp: add timeout handler befor running snmpget

Sven Nierlein 13 年之前
父節點
當前提交
4933146e00
共有 1 個文件被更改,包括 10 次插入 和 0 次删除
  1. 10 0
      plugins/check_snmp.c

+ 10 - 0
plugins/check_snmp.c

@@ -33,6 +33,7 @@ const char *copyright = "1999-2007";
 const char *email = "nagiosplug-devel@lists.sourceforge.net";
 
 #include "common.h"
+#include "runcmd.h"
 #include "utils.h"
 #include "utils_cmd.h"
 
@@ -330,9 +331,18 @@ main (int argc, char **argv)
 	if (verbose)
 		printf ("%s\n", cl_hidden_auth);
 
+	/* Set signal handling and alarm */
+	if (signal (SIGALRM, runcmd_timeout_alarm_handler) == SIG_ERR) {
+		usage4 (_("Cannot catch SIGALRM"));
+	}
+	alarm(timeout_interval * retries + 5);
+
 	/* Run the command */
 	return_code = cmd_run_array (command_line, &chld_out, &chld_err, 0);
 
+	/* disable alarm again */
+	alarm(0);
+
 	/* Due to net-snmp sometimes showing stderr messages with poorly formed MIBs,
 	   only return state unknown if return code is non zero or there is no stdout.
 	   Do this way so that if there is stderr, will get added to output, which helps problem diagnosis