Code

Spent the day working on backwards compatability using getaddrinfo()
[nagiosplug.git] / plugins / check_http.c
index daddfc1993212ffa173c206fa92875c649df598a..4734d75c5c43b4fde66490986f89046399424786 100644 (file)
@@ -41,11 +41,11 @@ strings and regular expressions, check connection times, and report on\n\
 certificate expiration times.\n"
 
 #define OPTIONS "\
-\(-H <vhost> | -I <IP-address>) [-u <uri>] [-p <port>]\n\
+(-H <vhost> | -I <IP-address>) [-u <uri>] [-p <port>]\n\
             [-w <warn time>] [-c <critical time>] [-t <timeout>] [-L]\n\
             [-a auth] [-f <ok | warn | critcal | follow>] [-e <expect>]\n\
             [-s string] [-l] [-r <regex> | -R <case-insensitive regex>]\n\
-            [-P string]"
+            [-P string] [-m min_pg_size]"
 
 #define LONGOPTIONS "\
  -H, --hostname=ADDRESS\n\
@@ -75,6 +75,8 @@ certificate expiration times.\n"
    Wrap output in HTML link (obsoleted by urlize)\n\
  -f, --onredirect=<ok|warning|critical|follow>\n\
    How to handle redirected pages\n%s%s\
+-m, --min=INTEGER\n\
+   Minimum page size required (bytes)\n\
  -v, --verbose\n\
     Show details for command-line debugging (do not use with nagios server)\n\
  -h, --help\n\
@@ -151,7 +153,7 @@ the certificate is expired.\n"
 #ifdef HAVE_SSL
 int check_cert = FALSE;
 int days_till_exp;
-unsigned char *randbuff;
+char *randbuff = "";
 SSL_CTX *ctx;
 SSL *ssl;
 X509 *server_cert;
@@ -160,8 +162,10 @@ int check_certificate (X509 **);
 #endif
 
 #ifdef HAVE_REGEX_H
-#define REGS 2
-#define MAX_RE_SIZE 256
+enum {
+       REGS = 2,
+       MAX_RE_SIZE = 256
+};
 #include <regex.h>
 regex_t preg;
 regmatch_t pmatch[REGS];
@@ -178,15 +182,18 @@ struct timeval tv;
 
 #define server_port_check(use_ssl) (use_ssl ? HTTPS_PORT : HTTP_PORT)
 
-#define MAX_IPV4_HOSTLENGTH 64
 #define HDR_LOCATION "%*[Ll]%*[Oo]%*[Cc]%*[Aa]%*[Tt]%*[Ii]%*[Oo]%*[Nn]: "
 #define URI_HTTP "%[HTPShtps]://"
 #define URI_HOST "%[a-zA-Z0-9.-]"
 #define URI_PORT ":%[0-9]"
 #define URI_PATH "%[/a-zA-Z0-9._-=@,]"
 
-#define HTTP_PORT 80
-#define HTTPS_PORT 443
+enum {
+       MAX_IPV4_HOSTLENGTH = 64,
+       HTTP_PORT = 80,
+       HTTPS_PORT = 443
+};
+
 #define HTTP_EXPECT "HTTP/1."
 #define HTTP_URL "/"
 #define CRLF "\r\n"
@@ -198,8 +205,8 @@ char server_port_text[6] = "";
 char server_type[6] = "http";
 char *server_address = ""; 
 char *host_name = "";
-char *server_url = HTTP_URL;
-int server_url_length = 1;
+char *server_url = "";
+int server_url_length;
 int server_expect_yn = 0;
 char server_expect[MAX_INPUT_BUFFER] = HTTP_EXPECT;
 char string_expect[MAX_INPUT_BUFFER] = "";
@@ -213,6 +220,7 @@ int onredirect = STATE_OK;
 int use_ssl = FALSE;
 int verbose = FALSE;
 int sd;
+int min_page_len = 0;
 char *http_method = "GET";
 char *http_post_data = "";
 char buffer[MAX_INPUT_BUFFER];
@@ -230,6 +238,10 @@ main (int argc, char **argv)
 {
        int result = STATE_UNKNOWN;
 
+       /* Set default URL. Must be malloced for subsequent realloc if --onredirect=follow */
+       asprintf (&server_url, "%s", HTTP_URL);
+       server_url_length = strlen(server_url);
+
        if (process_arguments (argc, argv) == ERROR)
                usage ("check_http: could not parse arguments\n");
 
@@ -284,7 +296,6 @@ process_arguments (int argc, char **argv)
 {
        int c = 1;
 
-#ifdef HAVE_GETOPT_H
        int option_index = 0;
        static struct option long_options[] = {
                STD_LONG_OPTS,
@@ -302,9 +313,9 @@ process_arguments (int argc, char **argv)
                {"linespan", no_argument, 0, 'l'},
                {"onredirect", required_argument, 0, 'f'},
                {"certificate", required_argument, 0, 'C'},
+               {"min", required_argument, 0, 'm'},
                {0, 0, 0, 0}
        };
-#endif
 
        if (argc < 2)
                return ERROR;
@@ -322,14 +333,8 @@ process_arguments (int argc, char **argv)
                        strcpy (argv[c], "-n");
        }
 
-#define OPTCHARS "Vvht:c:w:H:P:I:a:e:p:s:R:r:u:f:C:nlLS"
-
        while (1) {
-#ifdef HAVE_GETOPT_H
-               c = getopt_long (argc, argv, OPTCHARS, long_options, &option_index);
-#else
-               c = getopt (argc, argv, OPTCHARS);
-#endif
+               c = getopt_long (argc, argv, "Vvht:c:w:H:P:I:a:e:p:s:R:r:u:f:C:nlLSm:", long_options, &option_index);
                if (c == -1 || c == EOF)
                        break;
 
@@ -460,6 +465,9 @@ process_arguments (int argc, char **argv)
                case 'v': /* verbose */
                        verbose = TRUE;
                        break;
+               case 'm': /* min_page_length */
+                       min_page_len = atoi (optarg);
+                       break;
                }
        }
 
@@ -541,6 +549,7 @@ check_http (void)
        char *x = NULL;
        char *orig_url = NULL;
        double elapsed_time;
+       int page_len = 0;
 #ifdef HAVE_SSL
        int sslerr;
 #endif
@@ -549,8 +558,9 @@ check_http (void)
 #ifdef HAVE_SSL
        if (use_ssl == TRUE) {
 
-               if (connect_SSL () != OK)
+               if (connect_SSL () != OK) {
                        terminate (STATE_CRITICAL, "Unable to open TCP socket");
+               }
 
                if ((server_cert = SSL_get_peer_certificate (ssl)) != NULL) {
                        X509_free (server_cert);
@@ -560,112 +570,52 @@ check_http (void)
                        return STATE_CRITICAL;
                }
 
-               asprintf (&buf, "%s %s HTTP/1.0\r\n", http_method, server_url);
-               if (SSL_write (ssl, buf, strlen (buf)) == -1) {
-                       ERR_print_errors_fp (stderr);
-                       return STATE_CRITICAL;
-               }
-
-               /* optionally send the host header info (not clear if it's usable) */
-               if (strcmp (host_name, "")) {
-                       asprintf (&buf, "Host: %s\r\n", host_name);
-                       if (SSL_write (ssl, buf, strlen (buf)) == -1) {
-                               ERR_print_errors_fp (stderr);
-                               return STATE_CRITICAL;
-                       }
-               }
-
-               /* send user agent */
-               asprintf (&buf, "User-Agent: check_http/%s (nagios-plugins %s)\r\n",
-                        clean_revstring (REVISION), PACKAGE_VERSION);
-               if (SSL_write (ssl, buf, strlen (buf)) == -1) {
-                       ERR_print_errors_fp (stderr);
-                       return STATE_CRITICAL;
-               }
-
-               /* optionally send the authentication info */
-               if (strcmp (user_auth, "")) {
-                       auth = base64 (user_auth, strlen (user_auth));
-                       asprintf (&buf, "Authorization: Basic %s\r\n", auth);
-                       if (SSL_write (ssl, buf, strlen (buf)) == -1) {
-                               ERR_print_errors_fp (stderr);
-                               return STATE_CRITICAL;
-                       }
-               }
-
-               /* either send http POST data */
-               if (strlen (http_post_data)) {
-                       asprintf (&buf, "Content-Type: application/x-www-form-urlencoded\r\n");
-                       if (SSL_write (ssl, buf, strlen (buf)) == -1) {
-                               ERR_print_errors_fp (stderr);
-                               return STATE_CRITICAL;
-                       }
-                       asprintf (&buf, "Content-Length: %i\r\n\r\n", strlen (http_post_data));
-                       if (SSL_write (ssl, buf, strlen (buf)) == -1) {
-                               ERR_print_errors_fp (stderr);
-                               return STATE_CRITICAL;
-                       }
-                       if (SSL_write (ssl, http_post_data, strlen (http_post_data)) == -1) {
-                               ERR_print_errors_fp (stderr);
-                               return STATE_CRITICAL;
-                       }
-                       asprintf (&buf, CRLF);
-                       if (SSL_write (ssl, buf, strlen (buf)) == -1) {
-                               ERR_print_errors_fp (stderr);
-                               return STATE_CRITICAL;
-                       }
-               }
-               else {
-                       /* or just a newline so the server knows we're done with the request */
-                       asprintf (&buf, "\r\n");
-                       if (SSL_write (ssl, buf, strlen (buf)) == -1) {
-                               ERR_print_errors_fp (stderr);
-                               return STATE_CRITICAL;
-                       }
-               }
-
        }
        else {
 #endif
                if (my_tcp_connect (server_address, server_port, &sd) != STATE_OK)
                        terminate (STATE_CRITICAL, "Unable to open TCP socket");
-               asprintf (&buf, "%s %s HTTP/1.0\r\n", http_method, server_url);
-               send (sd, buf, strlen (buf), 0);
+#ifdef HAVE_SSL
+       }
+#endif
 
-               /* optionally send the host header info */
-               if (strcmp (host_name, "")) {
-                       asprintf (&buf, "Host: %s\r\n", host_name);
-                       send (sd, buf, strlen (buf), 0);
-               }
+       asprintf (&buf, "%s %s HTTP/1.0\r\n", http_method, server_url);
 
-               /* send user agent */
-               asprintf (&buf,
-                        "User-Agent: check_http/%s (nagios-plugins %s)\r\n",
-                        clean_revstring (REVISION), PACKAGE_VERSION);
-               send (sd, buf, strlen (buf), 0);
+       /* optionally send the host header info (not clear if it's usable) */
+       if (strcmp (host_name, ""))
+               asprintf (&buf, "%sHost: %s\r\n", buf, host_name);
 
-               /* optionally send the authentication info */
-               if (strcmp (user_auth, "")) {
-                       auth = base64 (user_auth, strlen (user_auth));
-                       asprintf (&buf, "Authorization: Basic %s\r\n", auth);
-                       send (sd, buf, strlen (buf), 0);
-               }
+       /* send user agent */
+       asprintf (&buf, "%sUser-Agent: check_http/%s (nagios-plugins %s)\r\n",
+                 buf, clean_revstring (REVISION), PACKAGE_VERSION);
 
-               /* either send http POST data */
-               /* written by Chris Henesy <lurker@shadowtech.org> */
-               if (strlen (http_post_data)) {
-                       asprintf (&buf, "Content-Type: application/x-www-form-urlencoded\r\n");
-                       send (sd, buf, strlen (buf), 0);
-                       asprintf (&buf, "Content-Length: %i\r\n\r\n", strlen (http_post_data));
-                       send (sd, buf, strlen (buf), 0);
-                       send (sd, http_post_data, strlen (http_post_data), 0);
-                       send (sd, CRLF, strlen (CRLF), 0);
-               }
-               else {
-                       /* or just a newline so the server knows we're done with the request */
-                       asprintf (&buf, "\r\n");
-                       send (sd, buf, strlen (buf), 0);
+       /* optionally send the authentication info */
+       if (strcmp (user_auth, "")) {
+               auth = base64 (user_auth, strlen (user_auth));
+               asprintf (&buf, "%sAuthorization: Basic %s\r\n", buf, auth);
+       }
+
+       /* either send http POST data */
+       if (strlen (http_post_data)) {
+               asprintf (&buf, "%sContent-Type: application/x-www-form-urlencoded\r\n", buf);
+               asprintf (&buf, "%sContent-Length: %i\r\n\r\n", buf, strlen (http_post_data));
+               asprintf (&buf, "%s%s%s", buf, http_post_data, CRLF);
+       }
+       else {
+               /* or just a newline so the server knows we're done with the request */
+               asprintf (&buf, "%s%s", buf, CRLF);
+       }
+
+#ifdef HAVE_SSL
+       if (use_ssl == TRUE) {
+               if (SSL_write (ssl, buf, strlen (buf)) == -1) {
+                       ERR_print_errors_fp (stderr);
+                       return STATE_CRITICAL;
                }
+       }
+       else {
+#endif
+               send (sd, buf, strlen (buf), 0);
 #ifdef HAVE_SSL
        }
 #endif
@@ -831,21 +781,21 @@ check_http (void)
                                        pos += (size_t) strcspn (pos, "\r\n");
                                        pos += (size_t) strspn (pos, "\r\n");
                                } /* end while (pos) */
-                               printf ("HTTP UNKNOWN: Could not find redirect location - %s%s",
+                               printf ("UNKNOWN - Could not find redirect location - %s%s",
                                        status_line, (display_html ? "</A>" : ""));
                                exit (STATE_UNKNOWN);
                        } /* end if (onredirect == STATE_DEPENDENT) */
                        
                        else if (onredirect == STATE_UNKNOWN)
-                               printf ("HTTP UNKNOWN");
+                               printf ("UNKNOWN");
                        else if (onredirect == STATE_OK)
-                               printf ("HTTP ok");
+                               printf ("OK");
                        else if (onredirect == STATE_WARNING)
-                               printf ("HTTP WARNING");
+                               printf ("WARNING");
                        else if (onredirect == STATE_CRITICAL)
-                               printf ("HTTP CRITICAL");
+                               printf ("CRITICAL");
                        elapsed_time = delta_time (tv);
-                       asprintf (&msg, ": %s - %7.3f second response time %s%s|time=%7.3f\n",
+                       asprintf (&msg, " - %s - %.3f second response time %s%s|time=%.3f\n",
                                 status_line, elapsed_time, timestamp,
                           (display_html ? "</A>" : ""), elapsed_time);
                        terminate (onredirect, msg);
@@ -857,7 +807,7 @@ check_http (void)
                
        /* check elapsed time */
        elapsed_time = delta_time (tv);
-       asprintf (&msg, "HTTP problem: %s - %7.3f second response time %s%s|time=%7.3f\n",
+       asprintf (&msg, "HTTP problem: %s - %.3f second response time %s%s|time=%.3f\n",
                       status_line, elapsed_time, timestamp,
                       (display_html ? "</A>" : ""), elapsed_time);
        if (check_critical_time == TRUE && elapsed_time > critical_time)
@@ -870,13 +820,13 @@ check_http (void)
 
        if (strlen (string_expect)) {
                if (strstr (page, string_expect)) {
-                       printf ("HTTP ok: %s - %7.3f second response time %s%s|time=%7.3f\n",
+                       printf ("HTTP OK %s - %.3f second response time %s%s|time=%.3f\n",
                                status_line, elapsed_time,
                                timestamp, (display_html ? "</A>" : ""), elapsed_time);
                        exit (STATE_OK);
                }
                else {
-                       printf ("HTTP CRITICAL: string not found%s|time=%7.3f\n",
+                       printf ("CRITICAL - string not found%s|time=%.3f\n",
                                (display_html ? "</A>" : ""), elapsed_time);
                        exit (STATE_CRITICAL);
                }
@@ -885,28 +835,35 @@ check_http (void)
        if (strlen (regexp)) {
                errcode = regexec (&preg, page, REGS, pmatch, 0);
                if (errcode == 0) {
-                       printf ("HTTP ok: %s - %7.3f second response time %s%s|time=%7.3f\n",
+                       printf ("HTTP OK %s - %.3f second response time %s%s|time=%.3f\n",
                                status_line, elapsed_time,
                                timestamp, (display_html ? "</A>" : ""), elapsed_time);
                        exit (STATE_OK);
                }
                else {
                        if (errcode == REG_NOMATCH) {
-                               printf ("HTTP CRITICAL: pattern not found%s|time=%7.3f\n",
+                               printf ("CRITICAL - pattern not found%s|time=%.3f\n",
                                        (display_html ? "</A>" : ""), elapsed_time);
                                exit (STATE_CRITICAL);
                        }
                        else {
                                regerror (errcode, &preg, errbuf, MAX_INPUT_BUFFER);
-                               printf ("Execute Error: %s\n", errbuf);
+                               printf ("CRITICAL - Execute Error: %s\n", errbuf);
                                exit (STATE_CRITICAL);
                        }
                }
        }
 #endif
 
+       /* make sure the page is of an appropriate size */
+       page_len = strlen (page);
+       if ((min_page_len > 0) && (page_len < min_page_len)) {
+               printf ("HTTP WARNING: page size too small%s|size=%i\n",
+                       (display_html ? "</A>" : ""), page_len );
+               exit (STATE_WARNING);
+       }
        /* We only get here if all tests have been passed */
-       asprintf (&msg, "HTTP ok: %s - %7.3f second response time %s%s|time=%7.3f\n",
+       asprintf (&msg, "HTTP OK %s - %.3f second response time %s%s|time=%.3f\n",
                        status_line, (float)elapsed_time,
                        timestamp, (display_html ? "</A>" : ""), elapsed_time);
        terminate (STATE_OK, msg);
@@ -920,14 +877,17 @@ int connect_SSL (void)
 {
        SSL_METHOD *meth;
 
-       asprintf (randbuff, "%s", "qwertyuiopasdfghjkl");
+       asprintf (&randbuff, "%s", "qwertyuiopasdfghjklqwertyuiopasdfghjkl");
        RAND_seed (randbuff, strlen (randbuff));
+       if (verbose)
+               printf("SSL seeding: %s\n", (RAND_status()==1 ? "OK" : "Failed") );
+
        /* Initialize SSL context */
        SSLeay_add_ssl_algorithms ();
        meth = SSLv23_client_method ();
        SSL_load_error_strings ();
        if ((ctx = SSL_CTX_new (meth)) == NULL) {
-               printf ("ERROR: Cannot create SSL context.\n");
+               printf ("CRITICAL -  Cannot create SSL context.\n");
                return STATE_CRITICAL;
        }
 
@@ -951,7 +911,7 @@ int connect_SSL (void)
                        ERR_print_errors_fp (stderr);
                }
                else {
-                       printf ("ERROR: Cannot initiate SSL handshake.\n");
+                       printf ("CRITICAL - Cannot initiate SSL handshake.\n");
                }
                SSL_free (ssl);
        }
@@ -979,7 +939,7 @@ check_certificate (X509 ** certificate)
        /* Generate tm structure to process timestamp */
        if (tm->type == V_ASN1_UTCTIME) {
                if (tm->length < 10) {
-                       printf ("ERROR: Wrong time format in certificate.\n");
+                       printf ("CRITICAL - Wrong time format in certificate.\n");
                        return STATE_CRITICAL;
                }
                else {
@@ -991,7 +951,7 @@ check_certificate (X509 ** certificate)
        }
        else {
                if (tm->length < 12) {
-                       printf ("ERROR: Wrong time format in certificate.\n");
+                       printf ("CRITICAL - Wrong time format in certificate.\n");
                        return STATE_CRITICAL;
                }
                else {
@@ -1020,20 +980,20 @@ check_certificate (X509 ** certificate)
                 stamp.tm_mday, stamp.tm_year + 1900, stamp.tm_hour, stamp.tm_min);
 
        if (days_left > 0 && days_left <= days_till_exp) {
-               printf ("Certificate expires in %d day(s) (%s).\n", days_left, timestamp);
+               printf ("WARNING - Certificate expires in %d day(s) (%s).\n", days_left, timestamp);
                return STATE_WARNING;
        }
        if (days_left < 0) {
-               printf ("Certificate expired on %s.\n", timestamp);
+               printf ("CRITICAL - Certificate expired on %s.\n", timestamp);
                return STATE_CRITICAL;
        }
 
        if (days_left == 0) {
-               printf ("Certificate expires today (%s).\n", timestamp);
+               printf ("WARNING - Certificate expires today (%s).\n", timestamp);
                return STATE_WARNING;
        }
 
-       printf ("Certificate will expire on %s.\n", timestamp);
+       printf ("OK - Certificate will expire on %s.\n", timestamp);
 
        return STATE_OK;
 }
@@ -1100,12 +1060,7 @@ void
 print_usage (void)
 {
        printf ("Usage:\n" " %s %s\n"
-#ifdef HAVE_GETOPT_H
                " %s (-h | --help) for detailed help\n"
                " %s (-V | --version) for version information\n",
-#else
-               " %s -h for detailed help\n"
-               " %s -V for version information\n",
-#endif
        progname, OPTIONS, progname, progname);
 }