X-Git-Url: https://git.tokkee.org/?a=blobdiff_plain;f=plugins%2Futils.c;h=0d250677c218ced9d6936826d455220d890d1722;hb=831d03cd768cb4d1f96d3cabd6262a0a55f904e7;hp=0d1729e0d7cba768414d5b4564be4504f6ceddc2;hpb=d02fedd5aab5d8d6047774a19eb21b27022e693d;p=nagiosplug.git diff --git a/plugins/utils.c b/plugins/utils.c index 0d1729e..0d25067 100644 --- a/plugins/utils.c +++ b/plugins/utils.c @@ -17,17 +17,22 @@ #include #include +#include + 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 *); @@ -44,11 +49,12 @@ int is_percentage (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); @@ -56,53 +62,53 @@ 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); } -char * -my_basename (char *path) +void usage2(char *msg, char *arg) { - if (!strstr (path, "/")) - return path; - else - return 1 + strrchr (path, '/'); + printf ("%s: %s - %s\n",progname,msg,arg); + print_usage (); + exit (STATE_UNKNOWN); +} + +void +usage3 (char *msg, char arg) +{ + printf ("%s: %s - %c\n", progname, msg, arg); + print_usage(); + exit (STATE_UNKNOWN); } @@ -127,14 +133,14 @@ 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); + 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" @@ -166,25 +172,50 @@ timeout_alarm_handler (int signo) 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_addr (char *address) +{ + if (is_inet_addr (address) || is_inet6_addr (address)) return (TRUE); + return (FALSE); } int -is_dotted_quad (char *address) +resolve_host_or_addr (char *address, int family) { - int o1, o2, o3, o4; - char c[1]; + struct addrinfo hints; + struct addrinfo *res; + int retval; - 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) + 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 @@ -197,22 +228,7 @@ is_dotted_quad (char *address) 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 @@ -220,33 +236,40 @@ is_numeric (char *number) { 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 @@ -254,8 +277,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 +287,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,27 +330,39 @@ 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; } +#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; - struct timezone tz; - double et; gettimeofday (&now, NULL); return ((double)(now.tv_sec - tv.tv_sec) + (double)(now.tv_usec - tv.tv_usec) / (double)1000000); @@ -353,13 +394,7 @@ 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,20 +407,10 @@ 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"); + return NULL; - strncpy (dest, src, len); + asprintf (&dest, "%s", src); return dest; } @@ -413,9 +438,11 @@ strscpy (char *dest, const char *src) *****************************************************************************/ char * -strscat (char *dest, const char *src) +strscat (char *dest, char *src) { + if (dest == NULL) + return src; if (src != NULL) asprintf (&dest, "%s%s", dest, src); @@ -498,72 +525,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