Code

Stop coredump on Solaris if arg not specified (Jason Kau - 1465288)
[nagiosplug.git] / plugins / utils.c
index 5f770a1f5a508c339238f45209bc829a36452131..bb4ffbc22ff1920a138ea80f158bd7be04304689 100644 (file)
@@ -13,7 +13,6 @@
 
 #define LOCAL_TIMEOUT_ALARM_HANDLER
 
-#include "config.h"
 #include "common.h"
 #include "utils.h"
 #include <stdarg.h>
@@ -52,39 +51,47 @@ max_state (int a, int b)
                return max (a, b);
 }
 
-void usage (char *msg)
+void usage (const char *msg)
 {
-       printf (msg);
+       printf ("%s\n", msg);
        print_usage ();
        exit (STATE_UNKNOWN);
 }
 
-void usage2(char *msg, char *arg)
+void usage_va (const char *fmt, ...)
 {
-       printf ("%s: %s - %s\n",progname,msg,arg);
+       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
-usage3 (char *msg, char arg)
+usage3 (const char *msg, int arg)
 {
        printf ("%s: %s - %c\n", progname, msg, arg);
        print_usage();
        exit (STATE_UNKNOWN);
 }
 
-
 void
-support (void)
+usage4 (const char *msg)
 {
-       printf (_("\n\
-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\n", progname, msg);
+       print_usage();
+       exit (STATE_UNKNOWN);
 }
 
-
 char *
 clean_revstring (const char *revstring)
 {
@@ -104,14 +111,9 @@ print_revision (const char *command_name, const char *revision_string)
                strncpy (plugin_revision, "N/A", STRLEN);
        printf ("%s (%s %s) %s\n",
                                        command_name, PACKAGE, 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"));
-
 }
 
-char *
+const char *
 state_text (int result)
 {
        switch (result) {
@@ -130,17 +132,6 @@ state_text (int result)
 
 void
 die (int result, const char *fmt, ...)
-{
-       va_list ap;
-       printf ("%s %s: ", sizeof (char) + index(progname, '_'), state_text(result));
-       va_start (ap, fmt);
-       vprintf (fmt, ap);
-       va_end (ap);
-       exit (result);
-}
-
-void
-terminate (int result, const char *fmt, ...)
 {
        va_list ap;
        va_start (ap, fmt);
@@ -153,7 +144,7 @@ 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);
        }
@@ -274,7 +265,155 @@ is_option (char *str)
                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
@@ -298,6 +437,16 @@ delta_time (struct timeval tv)
 
 
 
+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;
+}
+
+
+
 
 void
 strip (char *buffer)
@@ -317,9 +466,6 @@ strip (char *buffer)
 }
 
 
-
-
-
 /******************************************************************************
  *
  * Copies one string to another. Any previously existing data in
@@ -345,42 +491,6 @@ strscpy (char *dest, const char *src)
 
 
 
-
-
-/******************************************************************************
- *
- * 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)
-{
-
-       if (dest == NULL)
-               return src;
-       if (src != NULL)
-               asprintf (&dest, "%s%s", dest, src);
-
-       return dest;
-}
-
-
-
-
-
 /******************************************************************************
  *
  * Returns a pointer to the next line of a multiline string buffer
@@ -450,9 +560,6 @@ strnl (char *str)
 }
 
 
-
-
-
 /******************************************************************************
  *
  * Like strscpy, except only the portion of the source string up to
@@ -482,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';
@@ -492,8 +599,6 @@ strpcpy (char *dest, const char *src, const char *str)
 
 
 
-
-
 /******************************************************************************
  *
  * Like strscat, except only the portion of the source string up to
@@ -526,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;
+}