diff --git a/plugins/check_real.c b/plugins/check_real.c
index ba746b81ced0f14eaa12ceb928f0403cf12e65ca..875f9c40c84738d1913ecc01b1e3132c48851085 100644 (file)
--- a/plugins/check_real.c
+++ b/plugins/check_real.c
#include "netutils.h"
#include "utils.h"
-#define PROGNAME "check_real"
+const char *progname = "check_real";
#define PORT 554
#define EXPECT "RTSP/1."
#define URL ""
int process_arguments (int, char **);
-int call_getopt (int, char **);
int validate_arguments (void);
int check_disk (int usp, int free_disk);
void print_help (void);
void print_usage (void);
int server_port = PORT;
-char *server_address = NULL;
-char *host_name = NULL;
+char *server_address = "";
+char *host_name = "";
char *server_url = NULL;
-char *server_expect = NULL;
+char *server_expect = EXPECT;
int warning_time = 0;
int check_warning_time = FALSE;
int critical_time = 0;
/* try to connect to the host at the given port number */
if (my_tcp_connect (server_address, server_port, &sd) != STATE_OK)
- terminate (STATE_CRITICAL, "Unable to connect to %s on port %d\n",
+ die (STATE_CRITICAL, "Unable to connect to %s on port %d\n",
server_address, server_port);
/* Part I - Server Check */
/* send the OPTIONS request */
- sprintf (buffer, "OPTIONS rtsp://%s:%d RTSP/1.0\n", host_name, server_port);
+ sprintf (buffer, "OPTIONS rtsp://%s:%d RTSP/1.0\r\n", host_name, server_port);
result = send (sd, buffer, strlen (buffer), 0);
/* send the header sync */
- sprintf (buffer, "CSeq: 1\n");
+ sprintf (buffer, "CSeq: 1\r\n");
result = send (sd, buffer, strlen (buffer), 0);
/* send a newline so the server knows we're done with the request */
- sprintf (buffer, "\n");
+ sprintf (buffer, "\r\n");
result = send (sd, buffer, strlen (buffer), 0);
/* watch for the REAL connection string */
/* return a CRITICAL status if we couldn't read any data */
if (result == -1)
- terminate (STATE_CRITICAL, "No data received from %s\n", host_name);
+ die (STATE_CRITICAL, "No data received from %s\n", host_name);
/* make sure we find the response we are looking for */
- if (!strstr (buffer, EXPECT)) {
+ if (!strstr (buffer, server_expect)) {
if (server_port == PORT)
printf ("Invalid REAL response received from host\n");
else
}
/* Part II - Check stream exists and is ok */
- if ((result == STATE_OK) && (server_url != NULL)) {
+ if ((result == STATE_OK )&& (server_url != NULL) ) {
/* Part I - Server Check */
}
else {
/* make sure we find the response we are looking for */
- if (!strstr (buffer, EXPECT)) {
+ if (!strstr (buffer, server_expect)) {
if (server_port == PORT)
printf ("Invalid REAL response received from host\n");
else
{
int c;
- if (argc < 2)
- return ERROR;
-
- for (c = 1; c < argc; c++) {
- if (strcmp ("-to", argv[c]) == 0)
- strcpy (argv[c], "-t");
- else if (strcmp ("-wt", argv[c]) == 0)
- strcpy (argv[c], "-w");
- else if (strcmp ("-ct", argv[c]) == 0)
- strcpy (argv[c], "-c");
- }
-
-
-
- c = 0;
- while ((c += (call_getopt (argc - c, &argv[c]))) < argc) {
-
- if (is_option (argv[c]))
- continue;
-
- if (server_address == NULL) {
- if (is_host (argv[c])) {
- server_address = argv[c];
- }
- else {
- usage ("Invalid host name");
- }
- }
- }
-
- if (server_expect == NULL)
- server_expect = strscpy (NULL, EXPECT);
-
- return validate_arguments ();
-}
-
-
-
-
-
-
-int
-call_getopt (int argc, char **argv)
-{
- int c, i = 0;
-
-#ifdef HAVE_GETOPT_H
int option_index = 0;
static struct option long_options[] = {
{"hostname", required_argument, 0, 'H'},
{"help", no_argument, 0, 'h'},
{0, 0, 0, 0}
};
-#endif
+
+ if (argc < 2)
+ return ERROR;
+
+ for (c = 1; c < argc; c++) {
+ if (strcmp ("-to", argv[c]) == 0)
+ strcpy (argv[c], "-t");
+ else if (strcmp ("-wt", argv[c]) == 0)
+ strcpy (argv[c], "-w");
+ else if (strcmp ("-ct", argv[c]) == 0)
+ strcpy (argv[c], "-c");
+ }
while (1) {
-#ifdef HAVE_GETOPT_H
- c =
- getopt_long (argc, argv, "+hVI:H:e:u:p:w:c:t:", long_options,
+ c = getopt_long (argc, argv, "+hVI:H:e:u:p:w:c:t:", long_options,
&option_index);
-#else
- c = getopt (argc, argv, "+?hVI:H:e:u:p:w:c:t");
-#endif
- i++;
-
- if (c == -1 || c == EOF || c == 1)
+ if (c == -1 || c == EOF)
break;
- switch (c) {
- case 'I':
- case 'H':
- case 'e':
- case 'u':
- case 'p':
- case 'w':
- case 'c':
- case 't':
- i++;
- }
-
switch (c) {
case 'I': /* hostname */
- if (is_host (optarg)) {
- server_address = optarg;
- }
- else {
- usage ("Invalid host name\n");
- }
- break;
case 'H': /* hostname */
if (is_host (optarg)) {
server_address = optarg;
case 'e': /* string to expect in response header */
server_expect = optarg;
break;
- case 'u': /* string to expect in response header */
+ case 'u': /* server URL */
server_url = optarg;
break;
case 'p': /* port */
}
break;
case 'V': /* version */
- print_revision (PROGNAME, "$Revision$");
+ print_revision (progname, "$Revision$");
exit (STATE_OK);
case 'h': /* help */
print_help ();
usage ("Invalid argument\n");
}
}
- return i;
+
+ c = optind;
+ if (strlen(server_address)==0 && argc>c) {
+ if (is_host (argv[c])) {
+ server_address = argv[c++];
+ }
+ else {
+ usage ("Invalid host name");
+ }
+ }
+
+ if (strlen(server_address) == 0)
+ usage ("You must provide a server to check\n");
+
+ if (strlen(host_name) == 0)
+ asprintf (&host_name, "%s", server_address);
+
+ return validate_arguments ();
}
void
print_help (void)
{
- print_revision (PROGNAME, "$Revision$");
+ print_revision (progname, "$Revision$");
printf
("Copyright (c) 2000 Pedro Leite (leite@cic.ua.pt)/Karl DeBisschop\n\n"
"This plugin tests the REAL service on the specified host.\n\n");
("Usage: %s -H host [-e expect] [-p port] [-w warn] [-c crit]\n"
" [-t timeout] [-v]\n"
" %s --help\n"
- " %s --version\n", PROGNAME, PROGNAME, PROGNAME);
+ " %s --version\n", progname, progname, progname);
}
-
-
-
-
-/*
-// process command-line arguments
-int
-process_arguments (int argc, char **argv)
-{
- int x;
-
- // no options were supplied
- if (argc < 2)
- return ERROR;
-
- // first option is always the server name/address
- strncpy (server_address, argv[1], sizeof (server_address) - 1);
- server_address[sizeof (server_address) - 1] = 0;
-
- // set the host name to the server address (until its overridden)
- strcpy (host_name, server_address);
-
- // process all remaining arguments
- for (x = 3; x <= argc; x++)
- {
-
- // we got the string to expect from the server
- if (!strcmp (argv[x - 1], "-e"))
- {
- if (x < argc)
- {
- strncpy (server_expect, argv[x], sizeof (server_expect) - 1);
- server_expect[sizeof (server_expect) - 1] = 0;
- x++;
- }
- else
- return ERROR;
- }
-
- // we got the URL to check
- else if (!strcmp (argv[x - 1], "-u"))
- {
- if (x < argc)
- {
- strncpy (server_url, argv[x], sizeof (server_url) - 1);
- server_url[sizeof (server_url) - 1] = 0;
- x++;
- }
- else
- return ERROR;
- }
-
- // we go the host name to use in the host header
- else if (!strcmp (argv[x - 1], "-hn"))
- {
- if (x < argc)
- {
- strncpy (host_name, argv[x], sizeof (host_name) - 1);
- host_name[sizeof (host_name) - 1] = 0;
- x++;
- }
- else
- return ERROR;
- }
-
- // we got the port number to use
- else if (!strcmp (argv[x - 1], "-p"))
- {
- if (x < argc)
- {
- server_port = atoi (argv[x]);
- x++;
- }
- else
- return ERROR;
- }
-
- // we got the socket timeout
- else if (!strcmp (argv[x - 1], "-to"))
- {
- if (x < argc)
- {
- socket_timeout = atoi (argv[x]);
- if (socket_timeout <= 0)
- return ERROR;
- x++;
- }
- else
- return ERROR;
- }
-
- // we got the warning threshold time
- else if (!strcmp (argv[x - 1], "-wt"))
- {
- if (x < argc)
- {
- warning_time = atoi (argv[x]);
- check_warning_time = TRUE;
- x++;
- }
- else
- return ERROR;
- }
-
- // we got the critical threshold time
- else if (!strcmp (argv[x - 1], "-ct"))
- {
- if (x < argc)
- {
- critical_time = atoi (argv[x]);
- check_critical_time = TRUE;
- x++;
- }
- else
- return ERROR;
- }
-
- // else we got something else...
- else
- return ERROR;
- }
-
- return OK;
-}
-
- result = process_arguments (argc, argv);
-
- if (result != OK)
- {
-
- printf ("Incorrect number of arguments supplied\n");
- printf ("\n");
- print_revision(argv[0],"$Revision$");
- printf ("Copyright (c) 1999 Pedro Leite (leite@cic.ua.pt)\n");
- printf ("Last Modified: 30-10-1999\n");
- printf ("License: GPL\n");
- printf ("\n");
- printf ("Usage: %s <host_address> [-e expect] [-u url] [-p port] [-hn host_name] [-wt warn_time]\n",argv[0]);
- printf(" [-ct crit_time] [-to to_sec] [-a auth]\n");
- printf ("\n");
- printf ("Options:\n");
- printf (" [expect] = String to expect in first line of server response - default is \"%s\"\n", EXPECT);
- printf (" [url] = Optional URL to GET - default is root document\n");
- printf (" [port] = Optional port number to use - default is %d\n", PORT);
- printf (" [host_name] = Optional host name argument to GET command - used for servers using host headers\n");
- printf (" [warn_time] = Response time in seconds necessary to result in a warning status\n");
- printf (" [crit_time] = Response time in seconds necessary to result in a critical status\n");
- printf (" [to_sec] = Number of seconds before connection attempt times out - default is %d seconds\n", DEFAULT_SOCKET_TIMEOUT);
- printf (" [auth] = Optional username:password for sites requiring basic authentication\n");
- printf ("\n");
- printf ("This plugin attempts to contact the REAL service on the specified host.\n");
- printf ("If possible, supply an IP address for the host address, as this will bypass the DNS lookup.\n");
- printf ("\n");
-
- return STATE_UNKNOWN;
- }
-
-*/