X-Git-Url: https://git.tokkee.org/?a=blobdiff_plain;f=plugins%2Fcheck_real.c;h=272aab61d85d0531e99c7b188ee44dbd2c4aade3;hb=3179e551c29af098567d5109dca56d834506fdce;hp=ba746b81ced0f14eaa12ceb928f0403cf12e65ca;hpb=44a321cb8a42d6c0ea2d96a1086a17f2134c89cc;p=nagiosplug.git diff --git a/plugins/check_real.c b/plugins/check_real.c index ba746b8..272aab6 100644 --- a/plugins/check_real.c +++ b/plugins/check_real.c @@ -48,24 +48,23 @@ #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; @@ -117,7 +116,7 @@ main (int argc, char **argv) terminate (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 @@ -163,7 +162,7 @@ main (int argc, char **argv) } /* 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 */ @@ -190,7 +189,7 @@ main (int argc, char **argv) } 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 @@ -275,53 +274,6 @@ process_arguments (int argc, char **argv) { 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'}, @@ -337,43 +289,28 @@ call_getopt (int argc, char **argv) {"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; @@ -385,7 +322,7 @@ call_getopt (int argc, char **argv) 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 */ @@ -426,7 +363,7 @@ call_getopt (int argc, char **argv) } break; case 'V': /* version */ - print_revision (PROGNAME, "$Revision$"); + print_revision (progname, "$Revision$"); exit (STATE_OK); case 'h': /* help */ print_help (); @@ -435,7 +372,24 @@ call_getopt (int argc, char **argv) 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 (); } @@ -455,7 +409,7 @@ validate_arguments (void) 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"); @@ -499,163 +453,5 @@ print_usage (void) ("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 [-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; - } - -*/