Code

Fix for regex input of '|', being output causing problems with Nagios' parsing of
[nagiosplug.git] / plugins / utils.c
index 5dbdb4de0c7f4beaa4233facade3beccfcf6e0a0..45373909560552b462ccbb11ecdcc51c1a7a2ca8 100644 (file)
@@ -1,20 +1,32 @@
 /*****************************************************************************
- *
- * utils.c
- *
- * Library of useful functions for plugins
- *
- * Copyright (c) 2000 Karl DeBisschop (karl@debisschop.net)
- * License: GPL
- *
- * $Revision$
- * $Date$
- ****************************************************************************/
+* 
+* Library of useful functions for plugins
+* 
+* License: GPL
+* Copyright (c) 2000 Karl DeBisschop (karl@debisschop.net)
+* Copyright (c) 2002-2007 Nagios Plugin Development Team
+* 
+* This program is free software: you can redistribute it and/or modify
+* it under the terms of the GNU General Public License as published by
+* the Free Software Foundation, either version 3 of the License, or
+* (at your option) any later version.
+* 
+* This program is distributed in the hope that it will be useful,
+* but WITHOUT ANY WARRANTY; without even the implied warranty of
+* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+* GNU General Public License for more details.
+* 
+* You should have received a copy of the GNU General Public License
+* along with this program.  If not, see <http://www.gnu.org/licenses/>.
+* 
+* 
+*****************************************************************************/
 
 #define LOCAL_TIMEOUT_ALARM_HANDLER
 
 #include "common.h"
 #include "utils.h"
+#include "utils_base.h"
 #include <stdarg.h>
 #include <limits.h>
 
@@ -51,16 +63,54 @@ max_state (int a, int b)
                return max (a, b);
 }
 
+/* **************************************************************************
+ * max_state_alt(STATE_x, STATE_y)
+ * compares STATE_x to  STATE_y and returns result based on the following
+ * STATE_OK < STATE_DEPENDENT < STATE_UNKNOWN < STATE_WARNING < STATE_CRITICAL
+ *
+ * The main difference between max_state_alt and max_state it that it doesn't
+ * allow setting a default to UNKNOWN. It will instead prioritixe any valid
+ * non-OK state.
+ ****************************************************************************/
+
+int
+max_state_alt (int a, int 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_UNKNOWN || b == STATE_UNKNOWN)
+               return STATE_UNKNOWN;
+       else if (a == STATE_DEPENDENT || b == STATE_DEPENDENT)
+               return STATE_DEPENDENT;
+       else if (a == STATE_OK || b == STATE_OK)
+               return STATE_OK;
+       else
+               return max (a, b);
+}
+
 void usage (const char *msg)
 {
-       printf ("%s", msg);
+       printf ("%s\n", msg);
        print_usage ();
        exit (STATE_UNKNOWN);
 }
 
+void usage_va (const char *fmt, ...)
+{
+       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);
+       printf ("%s: %s - %s\n", progname, msg, arg?arg:"(null)" );
        print_usage ();
        exit (STATE_UNKNOWN);
 }
@@ -73,41 +123,28 @@ usage3 (const char *msg, int arg)
        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)
+void
+usage5 (void)
 {
-       char plugin_revision[STRLEN];
-       if (sscanf (revstring,"$Revision: %[0-9.]",plugin_revision) == 1)
-               return strscpy (NULL, plugin_revision);
-       else
-         return strscpy (NULL, "N/A");
+       print_usage();
+       exit (STATE_UNKNOWN);
 }
 
 void
-print_revision (const char *command_name, const char *revision_string)
+print_revision (const char *command_name, const char *revision)
 {
        char plugin_revision[STRLEN];
 
-       if (sscanf (revision_string, "$Revision: %[0-9.]", plugin_revision) != 1)
-               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"));
-
+       printf ("%s v%s (%s %s)\n",
+                command_name, revision, PACKAGE, VERSION);
 }
 
 const char *
@@ -127,23 +164,13 @@ state_text (int result)
        }
 }
 
-void
-die (int result, const char *fmt, ...)
-{
-       va_list ap;
-       va_start (ap, fmt);
-       vprintf (fmt, ap);
-       va_end (ap);
-       exit (result);
-}
-
 void
 timeout_alarm_handler (int signo)
 {
        if (signo == SIGALRM) {
-               printf ("CRITICAL - Plugin timed out after %d seconds\n",
-                                               timeout_interval);
-               exit (STATE_CRITICAL);
+               printf (_("%s - Plugin timed out after %d seconds\n"),
+                                               state_text(timeout_state), timeout_interval);
+               exit (timeout_state);
        }
 }
 
@@ -262,8 +289,6 @@ is_option (char *str)
                return FALSE;
 }
 
-
-
 #ifdef NEED_GETTIMEOFDAY
 int
 gettimeofday (struct timeval *tv, struct timezone *tz)
@@ -315,9 +340,6 @@ strip (char *buffer)
 }
 
 
-
-
-
 /******************************************************************************
  *
  * Copies one string to another. Any previously existing data in
@@ -343,8 +365,6 @@ strscpy (char *dest, const char *src)
 
 
 
-
-
 /******************************************************************************
  *
  * Returns a pointer to the next line of a multiline string buffer
@@ -414,9 +434,6 @@ strnl (char *str)
 }
 
 
-
-
-
 /******************************************************************************
  *
  * Like strscpy, except only the portion of the source string up to
@@ -446,7 +463,7 @@ strpcpy (char *dest, const char *src, const char *str)
        if (dest == NULL || strlen (dest) < len)
                dest = realloc (dest, len + 1);
        if (dest == NULL)
-               die (STATE_UNKNOWN, "failed realloc in strpcpy\n");
+               die (STATE_UNKNOWN, _("failed realloc in strpcpy\n"));
 
        strncpy (dest, src, len);
        dest[len] = '\0';
@@ -456,8 +473,6 @@ strpcpy (char *dest, const char *src, const char *str)
 
 
 
-
-
 /******************************************************************************
  *
  * Like strscat, except only the portion of the source string up to
@@ -490,7 +505,7 @@ strpcat (char *dest, const char *src, const char *str)
 
        dest = realloc (dest, len + l2 + 1);
        if (dest == NULL)
-               die (STATE_UNKNOWN, "failed malloc in strscat\n");
+               die (STATE_UNKNOWN, _("failed malloc in strscat\n"));
 
        strncpy (dest + len, src, l2);
        dest[len + l2] = '\0';
@@ -498,9 +513,6 @@ strpcat (char *dest, const char *src, const char *str)
        return dest;
 }
 
-
-
-
 /******************************************************************************
  *
  * Print perfdata in a standard format
@@ -521,8 +533,8 @@ char *perfdata (const char *label,
 {
        char *data = NULL;
 
-       if (index (label, '"'))
-               asprintf (&data, "\"%s\"=%ld%s;", label, val, uom);
+       if (strpbrk (label, "'= "))
+               asprintf (&data, "'%s'=%ld%s;", label, val, uom);
        else
                asprintf (&data, "%s=%ld%s;", label, val, uom);
 
@@ -544,3 +556,47 @@ char *perfdata (const char *label,
 
        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;
+}