From 1c70ae69ea43e6c431a59bd81921c173005b0541 Mon Sep 17 00:00:00 2001
From: Karl DeBisschop <kdebisschop@users.sourceforge.net>
Date: Sun, 1 Jun 2003 05:19:36 +0000
Subject: allow user to limit the number of bytes to be fetched

git-svn-id: https://nagiosplug.svn.sourceforge.net/svnroot/nagiosplug/nagiosplug/branches/release-1.3.0@528 f882894a-f735-0410-b71e-b25c423dba1c

diff --git a/plugins/check_tcp.c b/plugins/check_tcp.c
index 8a2dcc5..444166d 100644
--- a/plugins/check_tcp.c
+++ b/plugins/check_tcp.c
@@ -82,6 +82,7 @@ char *server_send = NULL;
 char *server_quit = NULL;
 char **server_expect = NULL;
 int server_expect_count = 0;
+int maxbytes = 0;
 char **warn_codes = NULL;
 int warn_codes_count = 0;
 char **crit_codes = NULL;
@@ -260,6 +261,8 @@ main (int argc, char **argv)
 			asprintf (&status, "%s%s", status, buffer);
 			if (buffer[i-2] == '\r' && buffer[i-1] == '\n')
 				break;
+			if (maxbytes>0 && strlen(status)>maxbytes)
+				break;
 		}
 
 		/* return a CRITICAL status if we couldn't read any data */
@@ -347,6 +350,7 @@ process_arguments (int argc, char **argv)
 		{"port", required_argument, 0, 'p'},
 		{"send", required_argument, 0, 's'},
 		{"expect", required_argument, 0, 'e'},
+		{"maxbytes", required_argument, 0, 'm'},
 		{"quit", required_argument, 0, 'q'},
 		{"delay", required_argument, 0, 'd'},
 		{"verbose", no_argument, 0, 'v'},
@@ -379,10 +383,10 @@ process_arguments (int argc, char **argv)
 	while (1) {
 #ifdef HAVE_GETOPT_H
 		c =
-			getopt_long (argc, argv, "+hVvH:s:e:q:c:w:t:p:C:W:d:S", long_options,
+			getopt_long (argc, argv, "+hVvH:s:e:q:m:c:w:t:p:C:W:d:S", long_options,
 									 &option_index);
 #else
-		c = getopt (argc, argv, "+hVvH:s:e:q:c:w:t:p:C:W:d:S");
+		c = getopt (argc, argv, "+hVvH:s:e:q:m:c:w:t:p:C:W:d:S");
 #endif
 
 		if (c == -1 || c == EOF || c == 1)
@@ -448,6 +452,10 @@ process_arguments (int argc, char **argv)
 				server_expect = realloc (server_expect, ++server_expect_count);
 			server_expect[server_expect_count - 1] = optarg;
 			break;
+		case 'm':
+			if (!is_intpos (optarg))
+				usage ("Maxbytes must be a positive integer\n");
+			maxbytes = atoi (optarg);
 		case 'q':
 			server_quit = optarg;
 			break;
@@ -481,8 +489,9 @@ void
 print_usage (void)
 {
 	printf
-		("Usage: %s -H host -p port [-w warn_time] [-c crit_time] [-s send]\n"
-		 "         [-e expect] [-W wait] [-t to_sec] [-v]\n", progname);
+		("Usage: %s -H host -p port [-w warn_time] [-c crit_time] [-s send_string]\n"
+		 "         [-e expect_string] [-q quit_string] [-m maxbytes] [-W wait]\n"
+		 "         [-t to_sec] [-v]\n", progname);
 }
 
 
@@ -508,8 +517,12 @@ print_help (void)
 		 " -s, --send=STRING\n"
 		 "    String to send to the server\n"
 		 " -e, --expect=STRING\n"
-		 "    String to expect in server response"
-		 " -W, --wait=INTEGER\n"
+		 "    String to expect in server response\n"
+		 " -q, --quit=STRING\n"
+		 "    String to send server to initiate a clean close of the connection\n"
+		 " -m, --maxbytes=INTEGER\n"
+		 "    Close connection once more than this number of bytes are received\n"
+		 " -d, --delay=INTEGER\n"
 		 "    Seconds to wait between sending string and polling for response\n"
 		 " -w, --warning=DOUBLE\n"
 		 "    Response time to result in warning status (seconds)\n"
@@ -525,6 +538,12 @@ print_help (void)
 		 "    Print version information\n", DEFAULT_SOCKET_TIMEOUT);
 }
 
+/*
+		 " -W, --warning-codes=STRING\n"
+		 "    \n"
+		 " -C, --critical-code=STRING\n"
+		 "    \n"
+*/
 
 #ifdef HAVE_SSL
 int
-- 
cgit v0.10-9-g596f