Initial revision

git-svn-id: https://nagiosplug.svn.sourceforge.net/svnroot/nagiosplug/nagiosplug/trunk@2 f882894a-f735-0410-b71e-b25c423dba1c

1 files changed, 610 insertions, 0 deletions

diff --git a/plugins/utils.c b/plugins/utils.c
new file mode 100644
index 00000000..49e4d3d7
--- /dev/null
+++ b/plugins/utils.c
@@ -0,0 +1,610 @@
+/*****************************************************************************
+ *
+ * utils.c
+ *
+ * Library of useful functions for plugins
+ *
+ * Copyright (c) 2000 Karl DeBisschop (karl@debisschop.net)
+ * License: GPL
+ *
+ * $Revision$
+ * $Date$
+ ****************************************************************************/
+
+#include "config.h"
+#include "common.h"
+#include "version.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
+#define STRLEN 64
+#define TXTBLK 128
+
+#define max(a,b) ((a)>(b))?(a):(b)
+
+char *
+my_basename (char *path)
+{
+        if (!strstr (path, "/"))
+                return path;
+        else
+                return 1 + strrchr (path, '/');
+}
+
+
+void
+support (void)
+{
+        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");
+}
+
+
+char *
+clean_revstring (const char *revstring)
+{
+        char plugin_revision[STRLEN];
+        if (sscanf (revstring,"$Revision: %[0-9.]",plugin_revision) == 1)
+                return strscpy (NULL, plugin_revision);
+        else
+          return strscpy (NULL, "N/A");
+}
+
+void
+print_revision (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");
+
+}
+
+
+void
+terminate (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);
+        }
+}
+
+int
+is_host (char *address)
+{
+        if (is_dotted_quad (address) || is_hostname (address))
+                return (TRUE);
+        return (FALSE);
+}
+
+int
+is_dotted_quad (char *address)
+{
+        int o1, o2, o3, o4;
+        char c[1];
+
+        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)
+                return FALSE;
+        else if (o1 < 0 || o2 < 0 || o3 < 0 || o4 < 0)
+                return FALSE;
+        else
+                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)
+                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);
+}
+
+int
+is_negative (char *number)
+{
+        if (is_numeric (number) && atof (number) < 0.0)
+                return (TRUE);
+        return (FALSE);
+}
+
+int
+is_nonnegative (char *number)
+{
+        if (is_numeric (number) && atof (number) >= 0.0)
+                return (TRUE);
+        return (FALSE);
+}
+
+int
+is_percentage (char *number)
+{
+        int x;
+        if (is_numeric (number) && (x = atof (number)) >= 0 && x <= 100)
+                return (TRUE);
+        return (FALSE);
+}
+
+int
+is_integer (char *number)
+{
+        long int n;
+
+        if (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);
+}
+
+int
+is_intpos (char *number)
+{
+        if (is_integer (number) && atoi (number) > 0)
+                return (TRUE);
+        return (FALSE);
+}
+
+int
+is_intneg (char *number)
+{
+        if (is_integer (number) && atoi (number) < 0)
+                return (TRUE);
+        return (FALSE);
+}
+
+int
+is_intnonneg (char *number)
+{
+        if (is_integer (number) && atoi (number) >= 0)
+                return (TRUE);
+        return (FALSE);
+}
+
+int
+is_intpercent (char *number)
+{
+        int i;
+        if (is_integer (number) && (i = atoi (number)) >= 0 && i <= 100)
+                return (TRUE);
+        return (FALSE);
+}
+
+int
+is_option (char *str)
+{
+        if (strspn (str, "-") == 1 || strspn (str, "-") == 2)
+                return TRUE;
+        return FALSE;
+}
+
+
+
+
+
+void
+strip (char *buffer)
+{
+        size_t x;
+        int i;
+
+        for (x = strlen (buffer); x >= 1; x--) {
+                i = x - 1;
+                if (buffer[i] == ' ' ||
+                                buffer[i] == '\r' || buffer[i] == '\n' || buffer[i] == '\t')
+                        buffer[i] = '\0';
+                else
+                        break;
+        }
+        return;
+}
+
+
+
+
+
+/******************************************************************************
+ *
+ * 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
+ * the destination string is lost.
+ *
+ * Example:
+ *
+ * char *str=NULL;
+ * str = strscpy("This is a line of text with no trailing newline");
+ *
+ *****************************************************************************/
+
+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)
+                l2 = strlen (src);
+        else
+                return dest;
+
+        if (dest)
+                len = strlen (dest);
+        else
+                len = 0;
+
+        dest = realloc (dest, len + l2 + 1);
+        if (dest == NULL)
+                terminate (STATE_UNKNOWN, "failed malloc in strscat\n");
+
+        strncpy (dest + len, src, l2);
+        dest[len + l2] = '\0';
+
+        return dest;
+}
+
+
+
+
+
+/******************************************************************************
+ *
+ * Returns a pointer to the next line of a multiline string buffer
+ *
+ * Given a pointer string, find the text following the next sequence
+ * of \r and \n characters. This has the effect of skipping blank
+ * lines as well
+ *
+ * Example:
+ *
+ * Given text as follows:
+ *
+ * ==============================
+ * This
+ * is
+ * a
+ * 
+ * multiline string buffer
+ * ==============================
+ *
+ * int i=0;
+ * char *str=NULL;
+ * char *ptr=NULL;
+ * str = strscpy(str,"This\nis\r\na\n\nmultiline string buffer\n");
+ * ptr = str;
+ * while (ptr) {
+ *   printf("%d %s",i++,firstword(ptr));
+ *   ptr = strnl(ptr);
+ * }
+ * 
+ * Produces the following:
+ *
+ * 1 This
+ * 2 is
+ * 3 a
+ * 4 multiline
+ *
+ * NOTE: The 'firstword()' function is conceptual only and does not
+ *       exist in this package.
+ *
+ * NOTE: Although the second 'ptr' variable is not strictly needed in
+ *       this example, it is good practice with these utilities. Once
+ *       the * pointer is advance in this manner, it may no longer be
+ *       handled with * realloc(). So at the end of the code fragment
+ *       above, * strscpy(str,"foo") work perfectly fine, but
+ *       strscpy(ptr,"foo") will * cause the the program to crash with
+ *       a segmentation fault.
+ *
+ *****************************************************************************/
+
+char *
+strnl (char *str)
+{
+        size_t len;
+        if (str == NULL)
+                return NULL;
+        str = strpbrk (str, "\r\n");
+        if (str == NULL)
+                return NULL;
+        len = strspn (str, "\r\n");
+        if (str[len] == '\0')
+                return NULL;
+        str += len;
+        if (strlen (str) == 0)
+                return NULL;
+        return 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
+ * the provided delimiter is copied.
+ *
+ * Example:
+ *
+ * str = strpcpy(str,"This is a line of text with no trailing newline","x");
+ * printf("%s\n",str);
+ *
+ * Produces:
+ *
+ *This is a line of te
+ *
+ *****************************************************************************/
+
+char *
+strpcpy (char *dest, const char *src, const char *str)
+{
+        size_t len;
+
+        if (src)
+                len = strcspn (src, str);
+        else
+                return NULL;
+
+        if (dest == NULL || strlen (dest) < len)
+                dest = realloc (dest, len + 1);
+        if (dest == NULL)
+                terminate (STATE_UNKNOWN, "failed realloc in strpcpy\n");
+
+        strncpy (dest, src, len);
+        dest[len] = '\0';
+
+        return dest;
+}
+
+
+
+
+
+/******************************************************************************
+ *
+ * Like strscat, except only the portion of the source string up to
+ * the provided delimiter is copied.
+ *
+ * str = strpcpy(str,"This is a line of text with no trailing newline","x");
+ * str = strpcat(str,"This is a line of text with no trailing newline","x");
+ * printf("%s\n",str);
+ * 
+ *This is a line of texThis is a line of tex
+ *
+ *****************************************************************************/
+
+char *
+strpcat (char *dest, const char *src, const char *str)
+{
+        size_t len, l2;
+
+        if (dest)
+                len = strlen (dest);
+        else
+                len = 0;
+
+        if (src) {
+                l2 = strcspn (src, str);
+        }
+        else {
+                return dest;
+        }
+
+        dest = realloc (dest, len + l2 + 1);
+        if (dest == NULL)
+                terminate (STATE_UNKNOWN, "failed malloc in strscat\n");
+
+        strncpy (dest + len, src, l2);
+        dest[len + l2] = '\0';
+
+        return dest;
+}