Code

Stop coredump on Solaris if arg not specified (Jason Kau - 1465288)
[nagiosplug.git] / plugins / utils.c
index 6e52dab75e0759c501910d30ba3a16452acad595..bb4ffbc22ff1920a138ea80f158bd7be04304689 100644 (file)
  * $Date$
  ****************************************************************************/
 
-#include "config.h"
+#define LOCAL_TIMEOUT_ALARM_HANDLER
+
 #include "common.h"
-#include "version.h"
+#include "utils.h"
 #include <stdarg.h>
 #include <limits.h>
 
-extern int timeout_interval;
-
-char *my_basename (char *);
-void support (void);
-char *clean_revstring (const char *);
-void print_revision (char *, const char *);
-void terminate (int, const char *fmt, ...);
-RETSIGTYPE timeout_alarm_handler (int);
-
-int is_host (char *);
-int is_dotted_quad (char *);
-int is_hostname (char *);
-
-int is_integer (char *);
-int is_intpos (char *);
-int is_intneg (char *);
-int is_intnonneg (char *);
-int is_intpercent (char *);
-
-int is_numeric (char *);
-int is_positive (char *);
-int is_negative (char *);
-int is_nonnegative (char *);
-int is_percentage (char *);
-
-int is_option (char *str);
-
-void strip (char *);
-char *strscpy (char *dest, const char *src);
-char *strscat (char *dest, const char *src);
-char *strnl (char *str);
-char *ssprintf (char *str, const char *fmt, ...);
-char *strpcpy (char *dest, const char *src, const char *str);
-char *strpcat (char *dest, const char *src, const char *str);
-
-#define LABELLEN 63
+#include <arpa/inet.h>
+
+extern void print_usage (void);
+extern const char *progname;
+
 #define STRLEN 64
 #define TXTBLK 128
 
-#define max(a,b) ((a)>(b))?(a):(b)
-
 /* **************************************************************************
- * max_state(result, STATE_x)
- * compares STATE_x to result and returns result if STATE_x is less than a based on the following
+ * max_state(STATE_x, STATE_y)
+ * compares STATE_x to  STATE_y and returns result based on the following
  * STATE_UNKNOWN < STATE_OK < STATE_WARNING < STATE_CRITICAL
  *
  * Note that numerically the above does not hold
  ****************************************************************************/
 
 int
-max_state(int a, int b)
+max_state (int a, int b)
 {
-       if(a == STATE_CRITICAL){
-               return a;
-       }
-       else if (a == STATE_WARNING) {
-
-               if (b == STATE_CRITICAL){
-                       return b;
-               }else {
-                       return a;
-               }
-       } 
-       else if (a == STATE_OK) {
-               
-               if ( b== STATE_CRITICAL || b == STATE_WARNING) {
-                       return b;
-               }else{
-                       return a;
-               }
-       }
-       else {
-               /* a == UNKNOWN */
-               return b;
-       }
-               
+       if (a == STATE_CRITICAL || b == STATE_CRITICAL)
+               return STATE_CRITICAL;
+       else if (a == STATE_WARNING || b == STATE_WARNING)
+               return STATE_WARNING;
+       else if (a == STATE_OK || b == STATE_OK)
+               return STATE_OK;
+       else if (a == STATE_UNKNOWN || b == STATE_UNKNOWN)
+               return STATE_UNKNOWN;
+       else if (a == STATE_DEPENDENT || b == STATE_DEPENDENT)
+               return STATE_DEPENDENT;
+       else
+               return max (a, b);
+}
 
+void usage (const char *msg)
+{
+       printf ("%s\n", msg);
+       print_usage ();
+       exit (STATE_UNKNOWN);
 }
 
-char *
-my_basename (char *path)
+void usage_va (const char *fmt, ...)
 {
-       if (!strstr (path, "/"))
-               return path;
-       else
-               return 1 + strrchr (path, '/');
+       va_list ap;
+       printf("%s: ", progname);
+       va_start(ap, fmt);
+       vprintf(fmt, ap);
+       va_end(ap);
+       printf("\n");
+       exit (STATE_UNKNOWN);
 }
 
+void usage2(const char *msg, const char *arg)
+{
+       printf ("%s: %s - %s\n", progname, msg, arg?arg:"(null)" );
+       print_usage ();
+       exit (STATE_UNKNOWN);
+}
 
 void
-support (void)
+usage3 (const char *msg, int arg)
 {
-       printf
-               ("Send email to nagios-users@lists.sourceforge.net if you have questions\n"
-                "regarding use of this software. To submit patches or suggest improvements,\n"
-                "send email to nagiosplug-devel@lists.sourceforge.net\n");
+       printf ("%s: %s - %c\n", progname, msg, arg);
+       print_usage();
+       exit (STATE_UNKNOWN);
 }
 
+void
+usage4 (const char *msg)
+{
+       printf ("%s: %s\n", progname, msg);
+       print_usage();
+       exit (STATE_UNKNOWN);
+}
 
 char *
 clean_revstring (const char *revstring)
@@ -127,24 +103,35 @@ clean_revstring (const char *revstring)
 }
 
 void
-print_revision (char *command_name, const char *revision_string)
+print_revision (const char *command_name, const char *revision_string)
 {
        char plugin_revision[STRLEN];
 
        if (sscanf (revision_string, "$Revision: %[0-9.]", plugin_revision) != 1)
                strncpy (plugin_revision, "N/A", STRLEN);
-       printf ("%s (nagios-plugins %s) %s\n",
-                                       my_basename (command_name), VERSION, plugin_revision);
-       printf
-               ("The nagios plugins come with ABSOLUTELY NO WARRANTY. You may redistribute\n"
-                "copies of the plugins under the terms of the GNU General Public License.\n"
-                "For more information about these matters, see the file named COPYING.\n");
-
+       printf ("%s (%s %s) %s\n",
+                                       command_name, PACKAGE, VERSION, plugin_revision);
 }
 
+const char *
+state_text (int result)
+{
+       switch (result) {
+       case STATE_OK:
+               return "OK";
+       case STATE_WARNING:
+               return "WARNING";
+       case STATE_CRITICAL:
+               return "CRITICAL";
+       case STATE_DEPENDENT:
+               return "DEPENDENT";
+       default:
+               return "UNKNOWN";
+       }
+}
 
 void
-terminate (int result, const char *fmt, ...)
+die (int result, const char *fmt, ...)
 {
        va_list ap;
        va_start (ap, fmt);
@@ -157,96 +144,51 @@ void
 timeout_alarm_handler (int signo)
 {
        if (signo == SIGALRM) {
-               printf ("CRITICAL - Plugin timed out after %d seconds\n",
+               printf (_("CRITICAL - Plugin timed out after %d seconds\n"),
                                                timeout_interval);
                exit (STATE_CRITICAL);
        }
 }
 
 int
-is_host (char *address)
-{
-       if (is_dotted_quad (address) || is_hostname (address))
-               return (TRUE);
-       return (FALSE);
-}
-
-int
-is_dotted_quad (char *address)
+is_numeric (char *number)
 {
-       int o1, o2, o3, o4;
-       char c[1];
+       char tmp[1];
+       float x;
 
-       if (sscanf (address, "%d.%d.%d.%d%c", &o1, &o2, &o3, &o4, c) != 4)
-               return FALSE;
-       else if (o1 > 255 || o2 > 255 || o3 > 255 || o4 > 255)
+       if (!number)
                return FALSE;
-       else if (o1 < 0 || o2 < 0 || o3 < 0 || o4 < 0)
-               return FALSE;
-       else
+       else if (sscanf (number, "%f%c", &x, tmp) == 1)
                return TRUE;
-}
-
-/* from RFC-1035
- * 
- * The labels must follow the rules for ARPANET host names.  They must
- * start with a letter, end with a letter or digit, and have as interior
- * characters only letters, digits, and hyphen.  There are also some
- * restrictions on the length.  Labels must be 63 characters or less. */
-
-int
-is_hostname (char *s1)
-{
-       if (strlen (s1) > 63)
-               return FALSE;
-       if (strcspn
-                       (s1,
-                        "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUWVXYZ0123456789-.") !=
-                       0) return FALSE;
-       if (strspn (s1, "0123456789-.") == 1)
+       else
                return FALSE;
-       while ((s1 = index (s1, '.'))) {
-               s1++;
-               if (strspn (s1, "0123456789-.") == 1) {
-                       printf ("%s\n", s1);
-                       return FALSE;
-               }
-       }
-       return TRUE;
-}
-
-int
-is_numeric (char *number)
-{
-       char tmp[1];
-       float x;
-       if (sscanf (number, "%f%c", &x, tmp) == 1)
-               return (TRUE);
-       return (FALSE);
 }
 
 int
 is_positive (char *number)
 {
        if (is_numeric (number) && atof (number) > 0.0)
-               return (TRUE);
-       return (FALSE);
+               return TRUE;
+       else
+               return FALSE;
 }
 
 int
 is_negative (char *number)
 {
        if (is_numeric (number) && atof (number) < 0.0)
-               return (TRUE);
-       return (FALSE);
+               return TRUE;
+       else
+               return FALSE;
 }
 
 int
 is_nonnegative (char *number)
 {
        if (is_numeric (number) && atof (number) >= 0.0)
-               return (TRUE);
-       return (FALSE);
+               return TRUE;
+       else
+               return FALSE;
 }
 
 int
@@ -254,8 +196,9 @@ is_percentage (char *number)
 {
        int x;
        if (is_numeric (number) && (x = atof (number)) >= 0 && x <= 100)
-               return (TRUE);
-       return (FALSE);
+               return TRUE;
+       else
+               return FALSE;
 }
 
 int
@@ -263,37 +206,42 @@ is_integer (char *number)
 {
        long int n;
 
-       if (strspn (number, "-0123456789 ") != strlen (number))
-               return (FALSE);
+       if (!number || (strspn (number, "-0123456789 ") != strlen (number)))
+               return FALSE;
 
        n = strtol (number, NULL, 10);
+
        if (errno != ERANGE && n >= INT_MIN && n <= INT_MAX)
-               return (TRUE);
-       return (FALSE);
+               return TRUE;
+       else
+               return FALSE;
 }
 
 int
 is_intpos (char *number)
 {
        if (is_integer (number) && atoi (number) > 0)
-               return (TRUE);
-       return (FALSE);
+               return TRUE;
+       else
+               return FALSE;
 }
 
 int
 is_intneg (char *number)
 {
        if (is_integer (number) && atoi (number) < 0)
-               return (TRUE);
-       return (FALSE);
+               return TRUE;
+       else
+               return FALSE;
 }
 
 int
 is_intnonneg (char *number)
 {
        if (is_integer (number) && atoi (number) >= 0)
-               return (TRUE);
-       return (FALSE);
+               return TRUE;
+       else
+               return FALSE;
 }
 
 int
@@ -301,30 +249,200 @@ is_intpercent (char *number)
 {
        int i;
        if (is_integer (number) && (i = atoi (number)) >= 0 && i <= 100)
-               return (TRUE);
-       return (FALSE);
+               return TRUE;
+       else
+               return FALSE;
 }
 
 int
 is_option (char *str)
 {
-       if (strspn (str, "-") == 1 || strspn (str, "-") == 2)
+       if (!str)
+               return FALSE;
+       else if (strspn (str, "-") == 1 || strspn (str, "-") == 2)
                return TRUE;
-       return FALSE;
+       else
+               return FALSE;
 }
 
+void set_range_start (range *this, double value) {
+       this->start = value;
+       this->start_infinity = FALSE;
+}
+
+void set_range_end (range *this, double value) {
+       this->end = value;
+       this->end_infinity = FALSE;
+}
+
+range
+*parse_range_string (char *str) {
+       range *temp_range;
+       double start;
+       double end;
+       char *end_str;
+
+       temp_range = (range *) malloc(sizeof(range));
+
+       /* Set defaults */
+       temp_range->start = 0;
+       temp_range->start_infinity = FALSE;
+       temp_range->end = 0;
+       temp_range->end_infinity = TRUE;
+       temp_range->alert_on = OUTSIDE;
+
+       if (str[0] == '@') {
+               temp_range->alert_on = INSIDE;
+               str++;
+       }
+
+       end_str = index(str, ':');
+       if (end_str != NULL) {
+               if (str[0] == '~') {
+                       temp_range->start_infinity = TRUE;
+               } else {
+                       start = strtod(str, NULL);      /* Will stop at the ':' */
+                       set_range_start(temp_range, start);
+               }
+               end_str++;              /* Move past the ':' */
+       } else {
+               end_str = str;
+       }
+       end = strtod(end_str, NULL);
+       if (strcmp(end_str, "") != 0) {
+               set_range_end(temp_range, end);
+       }
+
+       if (temp_range->start_infinity == TRUE || 
+               temp_range->end_infinity == TRUE ||
+               temp_range->start <= temp_range->end) {
+               return temp_range;
+       }
+       free(temp_range);
+       return NULL;
+}
+
+/* returns 0 if okay, otherwise 1 */
+int
+_set_thresholds(thresholds **my_thresholds, char *warn_string, char *critical_string)
+{
+       thresholds *temp_thresholds = NULL;
+
+       temp_thresholds = malloc(sizeof(temp_thresholds));
+
+       temp_thresholds->warning = NULL;
+       temp_thresholds->critical = NULL;
+
+       if (warn_string != NULL) {
+               if ((temp_thresholds->warning = parse_range_string(warn_string)) == NULL) {
+                       return 1;
+               }
+       }
+       if (critical_string != NULL) {
+               if ((temp_thresholds->critical = parse_range_string(critical_string)) == NULL) {
+                       return 1;
+               }
+       }
+
+       if (*my_thresholds != 0) {
+               /* printf("Freeing here: %d\n", *my_thresholds); */
+               free(*my_thresholds);
+       }
+       *my_thresholds = temp_thresholds;
+
+       return 0;
+}
+
+void
+set_thresholds(thresholds **my_thresholds, char *warn_string, char *critical_string)
+{
+       if (_set_thresholds(my_thresholds, warn_string, critical_string) == 0) {
+               return;
+       } else {
+               usage("Range format incorrect");
+       }
+}
+
+/* Returns TRUE if alert should be raised based on the range */
+int
+check_range(double value, range *my_range)
+{
+       int false = FALSE;
+       int true = TRUE;
+       
+       if (my_range->alert_on == INSIDE) {
+               false = TRUE;
+               true = FALSE;
+       }
+
+       if (my_range->end_infinity == FALSE && my_range->start_infinity == FALSE) {
+               if ((my_range->start <= value) && (value <= my_range->end)) {
+                       return false;
+               } else {
+                       return true;
+               }
+       } else if (my_range->start_infinity == FALSE && my_range->end_infinity == TRUE) {
+               if (my_range->start <= value) {
+                       return false;
+               } else {
+                       return true;
+               }
+       } else if (my_range->start_infinity == TRUE && my_range->end_infinity == FALSE) {
+               if (value <= my_range->end) {
+                       return false;
+               } else {
+                       return true;
+               }
+       } else {
+               return false;
+       }
+}
+
+/* Returns status */
+int
+get_status(double value, thresholds *my_thresholds)
+{
+       if (my_thresholds->critical != NULL) {
+               if (check_range(value, my_thresholds->critical) == TRUE) {
+                       return STATE_CRITICAL;
+               }
+       }
+       if (my_thresholds->warning != NULL) {
+               if (check_range(value, my_thresholds->warning) == TRUE) {
+                       return STATE_WARNING;
+               }
+       }
+       return STATE_OK;
+}
+
+#ifdef NEED_GETTIMEOFDAY
+int
+gettimeofday (struct timeval *tv, struct timezone *tz)
+{
+       tv->tv_usec = 0;
+       tv->tv_sec = (long) time ((time_t) 0);
+}
+#endif
 
 
 
 double
-delta_time (struct timeval *tv)
+delta_time (struct timeval tv)
 {
-       struct timeval *pt;
-       struct timezone *tz;
+       struct timeval now;
+
+       gettimeofday (&now, NULL);
+       return ((double)(now.tv_sec - tv.tv_sec) + (double)(now.tv_usec - tv.tv_usec) / (double)1000000);
+}
+
 
-       gettimeofday (pt, tz);
 
-       return (pt->tv_sec - tv->tv_sec + (pt->tv_usec - tv->tv_usec) / 1000000);
+long
+deltime (struct timeval tv)
+{
+       struct timeval now;
+       gettimeofday (&now, NULL);
+       return (now.tv_sec - tv.tv_sec)*1000000 + now.tv_usec - tv.tv_usec;
 }
 
 
@@ -348,18 +466,9 @@ strip (char *buffer)
 }
 
 
-
-
-
 /******************************************************************************
  *
- * Copies one string to another
- *
- * Given a pointer destination string, which may or may not already
- * hold some text, and a source string with additional text (possibly
- * NULL or empty), returns a pointer to a a copy of the source
- * string. Uses realloc to free memory held by the dest argument if
- * new storage space is required, and any previously existing data in
+ * Copies one string to another. Any previously existing data in
  * the destination string is lost.
  *
  * Example:
@@ -372,77 +481,16 @@ strip (char *buffer)
 char *
 strscpy (char *dest, const char *src)
 {
-       size_t len;
-
        if (src == NULL)
-               return dest;
-
-       len = strlen (src) + 1;
-       if (dest == NULL)
-               dest = malloc (len);
-       else if (strlen (dest) < len)
-               dest = realloc (dest, len);
-       if (dest == NULL)
-               terminate (STATE_UNKNOWN, "failed realloc in strscpy\n");
-
-       strncpy (dest, src, len);
-
-       return dest;
-}
-
-
-
-
-
-/******************************************************************************
- *
- * Concatenates one string to the end of another
- *
- * Given a pointer destination string, which may or may not already
- * hold some text, and a source string with additional text (possibly
- * NULL or empty), returns a pointer to a string that is the first
- * string with the second concatenated to it. Uses realloc to free 
- * memory held by the dest argument if new storage space is required.
- *
- * Example:
- *
- * char *str=NULL;
- * str = strscpy("This is a line of text with no trailing newline");
- * str = strscat(str,"\n");
- *
- *****************************************************************************/
-
-char *
-strscat (char *dest, const char *src)
-{
-       size_t len, l2;
-
-       if (src == NULL)
-               return dest;
-       else
-               l2 = strlen (src);
-
-       if (dest == NULL) {
-               len = 0;
-               dest = malloc (l2 + 1);
-       } else {
-               len = strlen (dest);
-               dest = realloc (dest, len + l2 + 1);
-       }
-
-       if (dest == NULL)
-               terminate (STATE_UNKNOWN, "failed malloc in strscat\n");
+               return NULL;
 
-       strncpy (dest + len, src, l2);
-       dest[len + l2] = '\0';
+       asprintf (&dest, "%s", src);
 
        return dest;
 }
 
 
 
-
-
 /******************************************************************************
  *
  * Returns a pointer to the next line of a multiline string buffer
@@ -512,75 +560,6 @@ strnl (char *str)
 }
 
 
-
-
-
-/******************************************************************************
- *
- * Does a formatted print to a string variable
- *
- * Given a pointer destination string, which may or may not already
- * hold some text, and a source string with additional text (possibly
- * NULL or empty), returns a pointer to a string that cntains the
- * results of the specified formatted print
- *
- * Example:
- *
- * char *str=NULL;
- * str = ssprintf(str,"%d %s",1,"string");
- *
- *****************************************************************************/
-
-char *
-ssprintf (char *ptr, const char *fmt, ...)
-{
-       va_list ap;
-       int nchars;
-       size_t size;
-       char *str = NULL;
-
-       if (str == NULL) {
-               str = malloc (TXTBLK);
-               if (str == NULL)
-                       terminate (STATE_UNKNOWN, "malloc failed in ssprintf");
-               size = TXTBLK;
-       }
-       else
-               size = max (strlen (str), TXTBLK);
-
-       va_start (ap, fmt);
-
-       while (1) {
-
-               nchars = vsnprintf (str, size, fmt, ap);
-
-               if (nchars > -1)
-                       if (nchars < (int) size) {
-                               va_end (ap);
-                               str[nchars] = '\0';
-                               if (ptr)
-                                       free (ptr);
-                               return str;
-                       }
-                       else {
-                               size = (size_t) (nchars + 1);
-                       }
-
-               else
-                       size *= 2;
-
-               str = realloc (str, size);
-
-               if (str == NULL)
-                       terminate (STATE_UNKNOWN, "realloc failed in ssprintf");
-       }
-
-}
-
-
-
-
-
 /******************************************************************************
  *
  * Like strscpy, except only the portion of the source string up to
@@ -610,7 +589,7 @@ strpcpy (char *dest, const char *src, const char *str)
        if (dest == NULL || strlen (dest) < len)
                dest = realloc (dest, len + 1);
        if (dest == NULL)
-               terminate (STATE_UNKNOWN, "failed realloc in strpcpy\n");
+               die (STATE_UNKNOWN, _("failed realloc in strpcpy\n"));
 
        strncpy (dest, src, len);
        dest[len] = '\0';
@@ -620,8 +599,6 @@ strpcpy (char *dest, const char *src, const char *str)
 
 
 
-
-
 /******************************************************************************
  *
  * Like strscat, except only the portion of the source string up to
@@ -654,10 +631,129 @@ strpcat (char *dest, const char *src, const char *str)
 
        dest = realloc (dest, len + l2 + 1);
        if (dest == NULL)
-               terminate (STATE_UNKNOWN, "failed malloc in strscat\n");
+               die (STATE_UNKNOWN, _("failed malloc in strscat\n"));
 
        strncpy (dest + len, src, l2);
        dest[len + l2] = '\0';
 
        return dest;
 }
+
+
+/******************************************************************************
+ *
+ * Print perfdata in a standard format
+ *
+ ******************************************************************************/
+
+char *perfdata (const char *label,
+ long int val,
+ const char *uom,
+ int warnp,
+ long int warn,
+ int critp,
+ long int crit,
+ int minp,
+ long int minv,
+ int maxp,
+ long int maxv)
+{
+       char *data = NULL;
+
+       if (strpbrk (label, "'= "))
+               asprintf (&data, "'%s'=%ld%s;", label, val, uom);
+       else
+               asprintf (&data, "%s=%ld%s;", label, val, uom);
+
+       if (warnp)
+               asprintf (&data, "%s%ld;", data, warn);
+       else
+               asprintf (&data, "%s;", data);
+
+       if (critp)
+               asprintf (&data, "%s%ld;", data, crit);
+       else
+               asprintf (&data, "%s;", data);
+
+       if (minp)
+               asprintf (&data, "%s%ld", data, minv);
+
+       if (maxp)
+               asprintf (&data, "%s;%ld", data, maxv);
+
+       return data;
+}
+
+
+char *fperfdata (const char *label,
+ double val,
+ const char *uom,
+ int warnp,
+ double warn,
+ int critp,
+ double crit,
+ int minp,
+ double minv,
+ int maxp,
+ double maxv)
+{
+       char *data = NULL;
+
+       if (strpbrk (label, "'= "))
+               asprintf (&data, "'%s'=", label);
+       else
+               asprintf (&data, "%s=", label);
+
+       asprintf (&data, "%s%f", data, val);
+       asprintf (&data, "%s%s;", data, uom);
+
+       if (warnp)
+               asprintf (&data, "%s%f", data, warn);
+
+       asprintf (&data, "%s;", data);
+
+       if (critp)
+               asprintf (&data, "%s%f", data, crit);
+
+       asprintf (&data, "%s;", data);
+
+       if (minp)
+               asprintf (&data, "%s%f", data, minv);
+
+       if (maxp) {
+               asprintf (&data, "%s;", data);
+               asprintf (&data, "%s%f", data, maxv);
+       }
+
+       return data;
+}
+
+char *np_escaped_string (const char *string) {
+       char *data;
+       int i, j=0;
+       data = strdup(string);
+       for (i=0; data[i]; i++) {
+               if (data[i] == '\\') {
+                       switch(data[++i]) {
+                               case 'n':
+                                       data[j++] = '\n';
+                                       break;
+                               case 'r':
+                                       data[j++] = '\r';
+                                       break;
+                               case 't':
+                                       data[j++] = '\t';
+                                       break;
+                               case '\\':
+                                       data[j++] = '\\';
+                                       break;
+                               default:
+                                       data[j++] = data[i];
+                       }
+               } else {
+                       data[j++] = data[i];
+               }
+       }
+       data[j] = '\0';
+       return data;
+}