diff --git a/plugins/utils.c b/plugins/utils.c
index 8bec1cf12c2d811edfbbcbee42e67334956f535c..0d250677c218ced9d6936826d455220d890d1722 100644 (file)
--- a/plugins/utils.c
+++ b/plugins/utils.c
#include <stdarg.h>
#include <limits.h>
+#include <arpa/inet.h>
+
extern int timeout_interval;
+extern const char *progname;
-char *my_basename (char *);
void support (void);
char *clean_revstring (const char *);
-void print_revision (char *, const char *);
+void print_revision (const 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_addr (char *);
+int resolve_host_or_addr (char *, int);
+int is_inet_addr (char *);
+int is_inet6_addr (char *);
int is_hostname (char *);
int is_integer (char *);
int is_option (char *str);
+double delta_time (struct timeval tv);
+
void strip (char *);
char *strscpy (char *dest, const char *src);
-char *strscat (char *dest, const char *src);
+char *strscat (char *dest, 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 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
****************************************************************************/
+#define max(a,b) (((a)>(b))?(a):(b))
+
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 (char *msg)
+{
+ printf (msg);
+ print_usage ();
+ exit (STATE_UNKNOWN);
+}
+void usage2(char *msg, char *arg)
+{
+ printf ("%s: %s - %s\n",progname,msg,arg);
+ print_usage ();
+ exit (STATE_UNKNOWN);
}
-char *
-my_basename (char *path)
+void
+usage3 (char *msg, char arg)
{
- if (!strstr (path, "/"))
- return path;
- else
- return 1 + strrchr (path, '/');
+ printf ("%s: %s - %c\n", progname, msg, arg);
+ print_usage();
+ exit (STATE_UNKNOWN);
}
}
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);
+ progname, 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"
int
is_host (char *address)
{
- if (is_dotted_quad (address) || is_hostname (address))
+ if (is_addr (address) || is_hostname (address))
return (TRUE);
+
return (FALSE);
}
int
-is_dotted_quad (char *address)
+is_addr (char *address)
{
- int o1, o2, o3, o4;
- char c[1];
+ if (is_inet_addr (address) || is_inet6_addr (address))
+ return (TRUE);
- 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);
+}
+
+int
+resolve_host_or_addr (char *address, int family)
+{
+ struct addrinfo hints;
+ struct addrinfo *res;
+ int retval;
+
+ memset (&hints, 0, sizeof (hints));
+ hints.ai_family = family;
+ retval = getaddrinfo (address, NULL, &hints, &res);
+
+ if (retval != 0)
return FALSE;
- else
+ else {
+ freeaddrinfo (res);
return TRUE;
+ }
+}
+
+int
+is_inet_addr (char *address)
+{
+ return resolve_host_or_addr (address, AF_INET);
+}
+
+int
+is_inet6_addr (char *address)
+{
+ return resolve_host_or_addr (address, AF_INET6);
}
/* from RFC-1035
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;
+ return resolve_host_or_addr (s1, AF_UNSPEC);
}
int
{
char tmp[1];
float x;
- if (sscanf (number, "%f%c", &x, tmp) == 1)
- return (TRUE);
- return (FALSE);
+
+ if (!number)
+ return FALSE;
+ else if (sscanf (number, "%f%c", &x, tmp) == 1)
+ return TRUE;
+ else
+ 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
{
int x;
if (is_numeric (number) && (x = atof (number)) >= 0 && x <= 100)
- return (TRUE);
- return (FALSE);
+ return TRUE;
+ else
+ return FALSE;
}
int
{
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
{
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;
}
+#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)
+{
+ struct timeval now;
+
+ gettimeofday (&now, NULL);
+ return ((double)(now.tv_sec - tv.tv_sec) + (double)(now.tv_usec - tv.tv_usec) / (double)1000000);
+}
+
+
void
/******************************************************************************
*
- * 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:
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");
+ return NULL;
- strncpy (dest, src, len);
+ asprintf (&dest, "%s", src);
return dest;
}
*****************************************************************************/
char *
-strscat (char *dest, const char *src)
+strscat (char *dest, 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 src;
+ if (src != NULL)
+ asprintf (&dest, "%s%s", dest, src);
return dest;
}
-/******************************************************************************
- *
- * 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