Code

Fix translations when extra-opts aren't enabled
[nagiosplug.git] / plugins / check_mysql.c
index 16a1f1335cf8f68f23575e21beebe5bbf679b357..ce190621d60f507cbcccf59eb39490ba6b721639 100644 (file)
-/******************************************************************************
-*
-* CHECK_MYSQL.C
-*
-* Program: Mysql plugin for Nagios
+/*****************************************************************************
+* 
+* Nagios check_mysql plugin
+* 
 * License: GPL
 * Copyright (c) 1999 Didi Rieder (adrieder@sbox.tu-graz.ac.at)
-*  portions (c) 2000 Karl DeBisschop (kdebisschop@users.sourceforge.net)
+* Copyright (c) 2000 Karl DeBisschop (kdebisschop@users.sourceforge.net)
+* Copyright (c) 1999-2009 Nagios Plugins Development Team
 * 
-* $Id$
-*
 * Description:
-*
-* This plugin is for testing a mysql server.
-******************************************************************************/
+* 
+* This file contains the check_mysql plugin
+* 
+* This program tests connections to a mysql server
+* 
+* 
+* 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/>.
+* 
+* 
+*****************************************************************************/
 
 const char *progname = "check_mysql";
-const char *revision = "$Revision$";
-const char *copyright = "1999-2002";
+const char *copyright = "1999-2007";
 const char *email = "nagiosplug-devel@lists.sourceforge.net";
 
+#define SLAVERESULTSIZE 70
+
 #include "common.h"
 #include "utils.h"
+#include "utils_base.h"
 #include "netutils.h"
-#include <mysql/mysql.h>
-#include <mysql/errmsg.h>
-
-void
-print_usage (void)
-{
-       printf (_("\
-Usage: %s [-d database] [-H host] [-P port] [-u user] [-p password]\n"),
-               progname);
-       printf (_(UT_HLP_VRS), progname, progname);
-}
-
-void
-print_help (void)
-{
-       char *myport;
-       asprintf (&myport, "%d", MYSQL_PORT);
-
-       print_revision (progname, revision);
-
-       printf (_(COPYRIGHT), copyright, email);
 
-       printf (_("This program tests connections to a mysql server\n"));
+#include <mysql.h>
+#include <errmsg.h>
 
-       print_usage ();
-
-       printf (_(UT_HELP_VRSN));
-
-       printf (_(UT_HOST_PORT), 'P', myport);
-
-       printf (_("\
- -d, --database=STRING\n\
-   Check database with indicated name\n\
- -u, --username=STRING\n\
-   Connect using the indicated username\n\
- -p, --password=STRING\n\
-   Use the indicated password to authenticate the connection\n\
-   ==> IMPORTANT: THIS FORM OF AUTHENTICATION IS NOT SECURE!!! <==\n\
-   Your clear-text password will be visible as a process table entry\n"));
-
-       printf (_("\n\
-There are no required arguments. By default, the local database with\n\
-a server listening on MySQL standard port %d will be checked\n"), MYSQL_PORT);
-
-       printf (_(UT_SUPPORT));
-}
-\f
-char *db_user = "";
-char *db_host = "";
-char *db_pass = "";
-char *db = "";
+char *db_user = NULL;
+char *db_host = NULL;
+char *db_socket = NULL;
+char *db_pass = NULL;
+char *db = NULL;
 unsigned int db_port = MYSQL_PORT;
+int check_slave = 0, warn_sec = 0, crit_sec = 0;
+int verbose = 0;
+
+thresholds *my_threshold = NULL;
 
 int process_arguments (int, char **);
 int validate_arguments (void);
+void print_help (void);
+void print_usage (void);
 
 int
 main (int argc, char **argv)
 {
 
        MYSQL mysql;
-       char result[1024];
+       MYSQL_RES *res;
+       MYSQL_ROW row;
+
+       /* should be status */
+
+       char *result = NULL;
+       char *error = NULL;
+       char slaveresult[SLAVERESULTSIZE];
 
-       if (process_arguments (argc, argv) != OK)
-               usage (_("Invalid command arguments supplied\n"));
+       setlocale (LC_ALL, "");
+       bindtextdomain (PACKAGE, LOCALEDIR);
+       textdomain (PACKAGE);
+
+       /* Parse extra opts if any */
+       argv=np_extra_opts (&argc, argv, progname);
+
+       if (process_arguments (argc, argv) == ERROR)
+               usage4 (_("Could not parse arguments"));
 
        /* initialize mysql  */
        mysql_init (&mysql);
 
+       mysql_options(&mysql,MYSQL_READ_DEFAULT_GROUP,"client");
+
        /* establish a connection to the server and error checking */
-       if (!mysql_real_connect
-                       (&mysql, db_host, db_user, db_pass, db, db_port, NULL, 0)) {
+       if (!mysql_real_connect(&mysql,db_host,db_user,db_pass,db,db_port,db_socket,0)) {
+               if (mysql_errno (&mysql) == CR_UNKNOWN_HOST)
+                       die (STATE_WARNING, "%s\n", mysql_error (&mysql));
+               else if (mysql_errno (&mysql) == CR_VERSION_ERROR)
+                       die (STATE_WARNING, "%s\n", mysql_error (&mysql));
+               else if (mysql_errno (&mysql) == CR_OUT_OF_MEMORY)
+                       die (STATE_WARNING, "%s\n", mysql_error (&mysql));
+               else if (mysql_errno (&mysql) == CR_IPSOCK_ERROR)
+                       die (STATE_WARNING, "%s\n", mysql_error (&mysql));
+               else if (mysql_errno (&mysql) == CR_SOCKET_CREATE_ERROR)
+                       die (STATE_WARNING, "%s\n", mysql_error (&mysql));
+               else
+                       die (STATE_CRITICAL, "%s\n", mysql_error (&mysql));
+       }
 
-               if (mysql_errno (&mysql) == CR_UNKNOWN_HOST) {
-                       printf ("%s\n", mysql_error (&mysql));
-                       return STATE_WARNING;
+       /* get the server stats */
+       result = strdup (mysql_stat (&mysql));
 
-               }
-               else if (mysql_errno (&mysql) == CR_VERSION_ERROR) {
-                       printf ("%s\n", mysql_error (&mysql));
-                       return STATE_WARNING;
+       /* error checking once more */
+       if (mysql_error (&mysql)) {
+               if (mysql_errno (&mysql) == CR_SERVER_GONE_ERROR)
+                       die (STATE_CRITICAL, "%s\n", mysql_error (&mysql));
+               else if (mysql_errno (&mysql) == CR_SERVER_LOST)
+                       die (STATE_CRITICAL, "%s\n", mysql_error (&mysql));
+               else if (mysql_errno (&mysql) == CR_UNKNOWN_ERROR)
+                       die (STATE_CRITICAL, "%s\n", mysql_error (&mysql));
+       }
 
+       if(check_slave) {
+               /* check the slave status */
+               if (mysql_query (&mysql, "show slave status") != 0) {
+                       error = strdup(mysql_error(&mysql));
+                       mysql_close (&mysql);
+                       die (STATE_CRITICAL, _("slave query error: %s\n"), error);
                }
-               else if (mysql_errno (&mysql) == CR_OUT_OF_MEMORY) {
-                       printf ("%s\n", mysql_error (&mysql));
-                       return STATE_WARNING;
 
+               /* store the result */
+               if ( (res = mysql_store_result (&mysql)) == NULL) {
+                       error = strdup(mysql_error(&mysql));
+                       mysql_close (&mysql);
+                       die (STATE_CRITICAL, _("slave store_result error: %s\n"), error);
                }
-               else if (mysql_errno (&mysql) == CR_IPSOCK_ERROR) {
-                       printf ("%s\n", mysql_error (&mysql));
-                       return STATE_WARNING;
 
+               /* Check there is some data */
+               if (mysql_num_rows(res) == 0) {
+                       mysql_close(&mysql);
+                       die (STATE_WARNING, "%s\n", _("No slaves defined"));
                }
-               else if (mysql_errno (&mysql) == CR_SOCKET_CREATE_ERROR) {
-                       printf ("%s\n", mysql_error (&mysql));
-                       return STATE_WARNING;
 
-               }
-               else {
-                       printf ("%s\n", mysql_error (&mysql));
-                       return STATE_CRITICAL;
+               /* fetch the first row */
+               if ( (row = mysql_fetch_row (res)) == NULL) {
+                       error = strdup(mysql_error(&mysql));
+                       mysql_free_result (res);
+                       mysql_close (&mysql);
+                       die (STATE_CRITICAL, _("slave fetch row error: %s\n"), error);
                }
 
-       }
+               if (mysql_field_count (&mysql) == 12) {
+                       /* mysql 3.23.x */
+                       snprintf (slaveresult, SLAVERESULTSIZE, _("Slave running: %s"), row[6]);
+                       if (strcmp (row[6], "Yes") != 0) {
+                               mysql_free_result (res);
+                               mysql_close (&mysql);
+                               die (STATE_CRITICAL, "%s\n", slaveresult);
+                       }
 
-       /* get the server stats */
-       sprintf (result, mysql_stat (&mysql));
+               } else {
+                       /* mysql 4.x.x */
+                       int slave_io_field = -1 , slave_sql_field = -1, seconds_behind_field = -1, i, num_fields;
+                       MYSQL_FIELD* fields;
+
+                       num_fields = mysql_num_fields(res);
+                       fields = mysql_fetch_fields(res);
+                       for(i = 0; i < num_fields; i++) {
+                               if (strcmp(fields[i].name, "Slave_IO_Running") == 0) {
+                                       slave_io_field = i;
+                                       continue;
+                               }
+                               if (strcmp(fields[i].name, "Slave_SQL_Running") == 0) {
+                                       slave_sql_field = i;
+                                       continue;
+                               }
+                               if (strcmp(fields[i].name, "Seconds_Behind_Master") == 0) {
+                                       seconds_behind_field = i;
+                                       continue;
+                               }
+                       }
 
-       /* error checking once more */
-       if (mysql_error (&mysql)) {
+                       if ((slave_io_field < 0) || (slave_sql_field < 0) || (num_fields == 0)) {
+                               mysql_free_result (res);
+                               mysql_close (&mysql);
+                               die (STATE_CRITICAL, "Slave status unavailable\n");
+                       }
 
-               if (mysql_errno (&mysql) == CR_SERVER_GONE_ERROR) {
-                       printf ("%s\n", mysql_error (&mysql));
-                       return STATE_CRITICAL;
+                       snprintf (slaveresult, SLAVERESULTSIZE, "Slave IO: %s Slave SQL: %s Seconds Behind Master: %s", row[slave_io_field], row[slave_sql_field], seconds_behind_field!=-1?row[seconds_behind_field]:"Unknown");
+                       if (strcmp (row[slave_io_field], "Yes") != 0 || strcmp (row[slave_sql_field], "Yes") != 0) {
+                               mysql_free_result (res);
+                               mysql_close (&mysql);
+                               die (STATE_CRITICAL, "%s\n", slaveresult);
+                       }
 
-               }
-               else if (mysql_errno (&mysql) == CR_SERVER_LOST) {
-                       printf ("%s\n", mysql_error (&mysql));
-                       return STATE_CRITICAL;
+                       if (verbose >=3) {
+                               if (seconds_behind_field == -1) {
+                                       printf("seconds_behind_field not found\n");
+                               } else {
+                                       printf ("seconds_behind_field(index %d)=%s\n", seconds_behind_field, row[seconds_behind_field]);
+                               }
+                       }
 
-               }
-               else if (mysql_errno (&mysql) == CR_UNKNOWN_ERROR) {
-                       printf ("%s\n", mysql_error (&mysql));
-                       return STATE_UNKNOWN;
+                       if ((seconds_behind_field != -1) && (strcmp (row[seconds_behind_field], "NULL") != 0)) {
+                               double value = atof(row[seconds_behind_field]);
+                               int status;
+
+                               status = get_status(value, my_threshold);
+
+                               if (status == STATE_WARNING) {
+                                       printf("SLOW_SLAVE %s: %s\n", _("WARNING"), slaveresult);
+                                       exit(STATE_WARNING);
+                               } else if (status == STATE_CRITICAL) {
+                                       printf("SLOW_SLAVE %s: %s\n", _("CRITICAL"), slaveresult);
+                                       exit(STATE_CRITICAL);
+                               }
+                       }
                }
 
+               /* free the result */
+               mysql_free_result (res);
        }
 
        /* close the connection */
        mysql_close (&mysql);
 
        /* print out the result of stats */
-       printf ("%s\n", result);
+       if (check_slave) {
+               printf ("%s %s\n", result, slaveresult);
+       } else {
+               printf ("%s\n", result);
+       }
 
        return STATE_OK;
 }
 
 
-
-
-
 /* process command-line arguments */
 int
 process_arguments (int argc, char **argv)
 {
        int c;
+       char *warning = NULL;
+       char *critical = NULL;
 
-       int option_index = 0;
-       static struct option long_options[] = {
+       int option = 0;
+       static struct option longopts[] = {
                {"hostname", required_argument, 0, 'H'},
+               {"socket", required_argument, 0, 's'},
                {"database", required_argument, 0, 'd'},
                {"username", required_argument, 0, 'u'},
                {"password", required_argument, 0, 'p'},
                {"port", required_argument, 0, 'P'},
+               {"critical", required_argument, 0, 'c'},
+               {"warning", required_argument, 0, 'w'},
+               {"check-slave", no_argument, 0, 'S'},
                {"verbose", no_argument, 0, 'v'},
                {"version", no_argument, 0, 'V'},
                {"help", no_argument, 0, 'h'},
@@ -186,7 +262,7 @@ process_arguments (int argc, char **argv)
                return ERROR;
 
        while (1) {
-               c = getopt_long (argc, argv, "hVP:p:u:d:H:", long_options, &option_index);
+               c = getopt_long (argc, argv, "hvVSP:p:u:d:H:s:c:w:", longopts, &option);
 
                if (c == -1 || c == EOF)
                        break;
@@ -197,48 +273,71 @@ process_arguments (int argc, char **argv)
                                db_host = optarg;
                        }
                        else {
-                               usage (_("Invalid host name\n"));
+                               usage2 (_("Invalid hostname/address"), optarg);
                        }
                        break;
-               case 'd':                                                                       /* hostname */
+               case 's':                                                                       /* socket */
+                       db_socket = optarg;
+                       break;
+               case 'd':                                                                       /* database */
                        db = optarg;
                        break;
                case 'u':                                                                       /* username */
                        db_user = optarg;
                        break;
                case 'p':                                                                       /* authentication information: password */
-                       db_pass = optarg;
+                       db_pass = strdup(optarg);
+
+                       /* Delete the password from process list */
+                       while (*optarg != '\0') {
+                               *optarg = 'X';
+                               optarg++;
+                       }
                        break;
                case 'P':                                                                       /* critical time threshold */
                        db_port = atoi (optarg);
                        break;
+               case 'S':
+                       check_slave = 1;                                                        /* check-slave */
+                       break;
+               case 'w':
+                       warning = optarg;
+                       break;
+               case 'c':
+                       critical = optarg;
+                       break;
                case 'V':                                                                       /* version */
-                       print_revision (progname, revision);
+                       print_revision (progname, NP_VERSION);
                        exit (STATE_OK);
                case 'h':                                                                       /* help */
                        print_help ();
                        exit (STATE_OK);
+               case 'v':
+                       verbose++;
+                       break;
                case '?':                                                                       /* help */
-                       usage (_("Invalid argument\n"));
+                       usage5 ();
                }
        }
 
        c = optind;
 
+       set_thresholds(&my_threshold, warning, critical);
+
        while ( argc > c ) {
 
-               if (strlen(db_host) == 0)
+               if (db_host == NULL)
                        if (is_host (argv[c])) {
                                db_host = argv[c++];
                        }
                        else {
-                               usage ("Invalid host name");
+                               usage2 (_("Invalid hostname/address"), argv[c]);
                        }
-               else if (strlen(db_user) == 0)
+               else if (db_user == NULL)
                        db_user = argv[c++];
-               else if (strlen(db_pass) == 0)
+               else if (db_pass == NULL)
                        db_pass = argv[c++];
-               else if (strlen(db) == 0)
+               else if (db == NULL)
                        db = argv[c++];
                else if (is_intnonneg (argv[c]))
                        db_port = atoi (argv[c++]);
@@ -250,11 +349,84 @@ process_arguments (int argc, char **argv)
 }
 
 
-
-
-
 int
 validate_arguments (void)
 {
+       if (db_user == NULL)
+               db_user = strdup("");
+
+       if (db_host == NULL)
+               db_host = strdup("");
+
+       if (db == NULL)
+               db = strdup("");
+
        return OK;
 }
+
+
+void
+print_help (void)
+{
+       char *myport;
+       asprintf (&myport, "%d", MYSQL_PORT);
+
+       print_revision (progname, NP_VERSION);
+
+       printf (_(COPYRIGHT), copyright, email);
+
+       printf ("%s\n", _("This program tests connections to a MySQL server"));
+
+  printf ("\n\n");
+
+       print_usage ();
+
+  printf (UT_HELP_VRSN);
+       printf (UT_EXTRA_OPTS);
+
+  printf (UT_HOST_PORT, 'P', myport);
+  printf (" %s\n", "-s, --socket=STRING");
+  printf ("    %s\n", _("Use the specified socket (has no effect if -H is used)"));
+
+  printf (" %s\n", "-d, --database=STRING");
+  printf ("    %s\n", _("Check database with indicated name"));
+  printf (" %s\n", "-u, --username=STRING");
+  printf ("    %s\n", _("Connect using the indicated username"));
+  printf (" %s\n", "-p, --password=STRING");
+  printf ("    %s\n", _("Use the indicated password to authenticate the connection"));
+  printf ("    ==> %s <==\n", _("IMPORTANT: THIS FORM OF AUTHENTICATION IS NOT SECURE!!!"));
+  printf ("    %s\n", _("Your clear-text password could be visible as a process table entry"));
+  printf (" %s\n", "-S, --check-slave");
+  printf ("    %s\n", _("Check if the slave thread is running properly."));
+  printf (" %s\n", "-w, --warning");
+  printf ("    %s\n", _("Exit with WARNING status if slave server is more than INTEGER seconds"));
+  printf ("    %s\n", _("behind master"));
+  printf (" %s\n", "-c, --critical");
+  printf ("    %s\n", _("Exit with CRITICAL status if slave server is more then INTEGER seconds"));
+  printf ("    %s\n", _("behind master"));
+
+  printf ("\n");
+  printf (" %s\n", _("There are no required arguments. By default, the local database is checked"));
+  printf (" %s\n", _("using the default unix socket. You can force TCP on localhost by using an"));
+  printf (" %s\n", _("IP address or FQDN ('localhost' will use the socket as well)."));
+
+#ifdef NP_EXTRA_OPTS
+       printf ("\n");
+       printf ("%s\n", _("Notes:"));
+       printf (UT_EXTRA_OPTS_NOTES);
+       printf ("\n");
+       printf (" %s\n", _("You must specify -p with an empty string to force an empty password,"));
+       printf (" %s\n", _("overriding any my.cnf settings."));
+#endif
+
+       printf (UT_SUPPORT);
+}
+
+
+void
+print_usage (void)
+{
+       printf (_("Usage:"));
+  printf (" %s [-d database] [-H host] [-P port] [-s socket]\n",progname);
+  printf ("       [-u user] [-p password] [-S]\n");
+}