Code

Stop coredump on Solaris if arg not specified (Jason Kau - 1465288)
[nagiosplug.git] / plugins / utils.c
index 1d7c0db668ad1e53251fced58be2d4226e886103..bb4ffbc22ff1920a138ea80f158bd7be04304689 100644 (file)
@@ -53,14 +53,25 @@ max_state (int a, int 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);
 }
@@ -81,16 +92,6 @@ usage4 (const char *msg)
        exit (STATE_UNKNOWN);
 }
 
-void
-support (void)
-{
-       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"));
-}
-
-
 char *
 clean_revstring (const char *revstring)
 {
@@ -110,11 +111,6 @@ 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"));
-
 }
 
 const char *
@@ -148,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);
        }
@@ -269,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
@@ -445,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)
-               die (STATE_UNKNOWN, "failed realloc in strpcpy\n");
+               die (STATE_UNKNOWN, _("failed realloc in strpcpy\n"));
 
        strncpy (dest, src, len);
        dest[len] = '\0';
@@ -487,7 +631,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';
@@ -583,3 +727,33 @@ char *fperfdata (const char *label,
 
        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;
+}