Code

fixes from richard brodie (tracker id 1216576)
[nagiosplug.git] / plugins / check_mysql.c
index 50836f91109f4ec9f0b51be0e81eb04ac3ec89ef..20901e639d908afd7f211469d8e50fe69f95a56b 100644 (file)
 * This plugin is for testing a mysql server.
 ******************************************************************************/
 
-#define PROGNAME "check_mysql"
+const char *progname = "check_mysql";
+const char *revision = "$Revision$";
+const char *copyright = "1999-2004";
+const char *email = "nagiosplug-devel@lists.sourceforge.net";
+
+#define SLAVERESULTSIZE 40
 
 #include "common.h"
 #include "utils.h"
+#include "netutils.h"
 
 #include <mysql/mysql.h>
 #include <mysql/errmsg.h>
@@ -27,191 +33,185 @@ char *db_host = NULL;
 char *db_pass = NULL;
 char *db = NULL;
 unsigned int db_port = MYSQL_PORT;
+int check_slave = 0;
 
 int process_arguments (int, char **);
-int call_getopt (int, char **);
 int validate_arguments (void);
-int check_disk (int usp, int free_disk);
 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 slaveresult[SLAVERESULTSIZE];
 
-       if (process_arguments (argc, argv) != OK)
-               usage ("Invalid command arguments supplied\n");
+       setlocale (LC_ALL, "");
+       bindtextdomain (PACKAGE, LOCALEDIR);
+       textdomain (PACKAGE);
+
+       if (process_arguments (argc, argv) == ERROR)
+               usage4 (_("Could not parse arguments"));
 
        /* initialize mysql  */
        mysql_init (&mysql);
 
        /* 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_errno (&mysql) == CR_UNKNOWN_HOST) {
-                       printf ("%s\n", mysql_error (&mysql));
-                       return STATE_WARNING;
-
-               }
-               else if (mysql_errno (&mysql) == CR_VERSION_ERROR) {
-                       printf ("%s\n", mysql_error (&mysql));
-                       return STATE_WARNING;
-
-               }
-               else if (mysql_errno (&mysql) == CR_OUT_OF_MEMORY) {
-                       printf ("%s\n", mysql_error (&mysql));
-                       return STATE_WARNING;
-
-               }
-               else if (mysql_errno (&mysql) == CR_IPSOCK_ERROR) {
-                       printf ("%s\n", mysql_error (&mysql));
-                       return STATE_WARNING;
-
-               }
-               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;
-               }
-
+       if (!mysql_real_connect(&mysql,db_host,db_user,db_pass,db,db_port,NULL,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));
        }
 
        /* get the server stats */
-       sprintf (result, mysql_stat (&mysql));
+       result = strdup (mysql_stat (&mysql));
 
        /* 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 (mysql_errno (&mysql) == CR_SERVER_GONE_ERROR) {
-                       printf ("%s\n", mysql_error (&mysql));
-                       return STATE_CRITICAL;
+       if(check_slave) {
+               /* check the slave status */
+               if (mysql_query (&mysql, "show slave status") != 0) {
+                       mysql_close (&mysql);
+                       die (STATE_CRITICAL, _("slave query error: %s\n"), mysql_error (&mysql));
+               }
 
+               /* store the result */
+               if ( (res = mysql_store_result (&mysql)) == NULL) {
+                       mysql_close (&mysql);
+                       die (STATE_CRITICAL, _("slave store_result error: %s\n"), mysql_error (&mysql));
                }
-               else if (mysql_errno (&mysql) == CR_SERVER_LOST) {
-                       printf ("%s\n", mysql_error (&mysql));
-                       return STATE_CRITICAL;
 
+               /* fetch the first row */
+               if ( (row = mysql_fetch_row (res)) == NULL) {
+                       mysql_free_result (res);
+                       mysql_close (&mysql);
+                       die (STATE_CRITICAL, _("slave fetch row error: %s\n"), mysql_error (&mysql));
                }
-               else if (mysql_errno (&mysql) == CR_UNKNOWN_ERROR) {
-                       printf ("%s\n", mysql_error (&mysql));
-                       return STATE_UNKNOWN;
+
+               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);
+                       }
+
+               } else {
+                       /* mysql 4.x.x */
+                       int slave_io_field = -1 , slave_sql_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 (0 == strcmp(fields[i].name, "Slave_IO_Running"))
+                               {
+                                       slave_io_field = i;
+                                       continue;
+                               }
+                               if (0 == strcmp(fields[i].name, "Slave_SQL_Running"))
+                               {
+                                       slave_sql_field = i;
+                                       continue;
+                               }
+                       }
+                       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");
+                       }
+                        
+                       snprintf (slaveresult, SLAVERESULTSIZE, "Slave IO: %s Slave SQL: %s", row[slave_io_field], row[slave_sql_field]);
+                       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);
+                       }
                }
 
+               /* 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;
 
-       if (argc < 1)
-               return ERROR;
-
-       c = 0;
-       while ((c += (call_getopt (argc - c, &argv[c]))) < argc) {
-
-               if (is_option (argv[c]))
-                       continue;
-
-               if (db_host == NULL)
-                       if (is_host (argv[c])) {
-                               db_host = argv[c];
-                       }
-                       else {
-                               usage ("Invalid host name");
-                       }
-               else if (db_user == NULL)
-                       db_user = argv[c];
-               else if (db_pass == NULL)
-                       db_pass = argv[c];
-               else if (db == NULL)
-                       db = argv[c];
-               else if (is_intnonneg (argv[c]))
-                       db_port = atoi (argv[c]);
-       }
-
-       if (db_host == NULL)
-               db_host = strscpy (db_host, "127.0.0.1");
-
-       return validate_arguments ();
-}
-
-
-
-
-
-
-int
-call_getopt (int argc, char **argv)
-{
-       int c, i = 0;
-
-#ifdef HAVE_GETOPT_H
-       int option_index = 0;
-       static struct option long_options[] = {
+       int option = 0;
+       static struct option longopts[] = {
                {"hostname", required_argument, 0, 'H'},
                {"database", required_argument, 0, 'd'},
                {"username", required_argument, 0, 'u'},
                {"password", required_argument, 0, 'p'},
                {"port", required_argument, 0, 'P'},
+               {"check-slave", no_argument, 0, 'S'},
                {"verbose", no_argument, 0, 'v'},
                {"version", no_argument, 0, 'V'},
                {"help", no_argument, 0, 'h'},
                {0, 0, 0, 0}
        };
-#endif
 
-       while (1) {
-#ifdef HAVE_GETOPT_H
-               c =
-                       getopt_long (argc, argv, "+hVP:p:u:d:H:", long_options, &option_index);
-#else
-               c = getopt (argc, argv, "+?hVP:p:u:d:H:");
-#endif
+       if (argc < 1)
+               return ERROR;
 
-               i++;
+       while (1) {
+               c = getopt_long (argc, argv, "hVSP:p:u:d:H:", longopts, &option);
 
-               if (c == -1 || c == EOF || c == 1)
+               if (c == -1 || c == EOF)
                        break;
 
-               switch (c) {
-               case 'P':
-               case 'p':
-               case 'u':
-               case 'd':
-               case 'H':
-                       i++;
-               }
-
                switch (c) {
                case 'H':                                                                       /* hostname */
                        if (is_host (optarg)) {
                                db_host = optarg;
                        }
                        else {
-                               usage ("Invalid host name\n");
+                               usage2 (_("Invalid hostname/address"), optarg);
                        }
                        break;
                case 'd':                                                                       /* hostname */
@@ -226,72 +226,108 @@ call_getopt (int argc, char **argv)
                case 'P':                                                                       /* critical time threshold */
                        db_port = atoi (optarg);
                        break;
+               case 'S':
+                       check_slave = 1;                                                        /* check-slave */
+                       break;
                case 'V':                                                                       /* version */
-                       print_revision (my_basename (argv[0]), "$Revision$");
+                       print_revision (progname, revision);
                        exit (STATE_OK);
                case 'h':                                                                       /* help */
                        print_help ();
                        exit (STATE_OK);
                case '?':                                                                       /* help */
-                       usage ("Invalid argument\n");
+                       usage2 (_("Unknown argument"), optarg);
                }
        }
-       return i;
-}
 
+       c = optind;
 
+       while ( argc > c ) {
 
+               if (strlen(db_host) == 0)
+                       if (is_host (argv[c])) {
+                               db_host = argv[c++];
+                       }
+                       else {
+                               usage2 (_("Invalid hostname/address"), optarg);
+                       }
+               else if (strlen(db_user) == 0)
+                       db_user = argv[c++];
+               else if (strlen(db_pass) == 0)
+                       db_pass = argv[c++];
+               else if (strlen(db) == 0)
+                       db = argv[c++];
+               else if (is_intnonneg (argv[c]))
+                       db_port = atoi (argv[c++]);
+               else
+                       break;
+       }
+
+       return validate_arguments ();
+}
 
 
 int
 validate_arguments (void)
 {
-       return OK;
-}
+       if (db_user == NULL)
+               db_user = strdup("");
 
+       if (db_host == NULL)
+               db_host = strdup("");
 
+       if (db_pass == NULL)
+               db_pass == strdup("");
 
+       if (db == NULL)
+               db = strdup("");
+
+       return OK;
+}
 
 
 void
 print_help (void)
 {
-       print_revision (PROGNAME, "$Revision$");
-       printf
-               ("Copyright (c) 2000 Didi Rieder/Karl DeBisschop\n\n"
-                "This plugin is for testing a mysql server.\n");
+       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"));
+
        print_usage ();
-       printf
-               ("\nThere are no required arguments. By default, the local database with\n"
-                "a server listening on MySQL standard port %d will be checked\n\n"
-                "Options:\n"
-                " -d, --database=STRING\n"
-                "   Check database with indicated name\n"
-                " -H, --hostname=STRING or IPADDRESS\n"
-                "   Check server on the indicated host\n"
-                " -P, --port=INTEGER\n"
-                "   Make connection on the indicated port\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"
-                " -h, --help\n"
-                "    Print detailed help screen\n"
-                " -V, --version\n" "    Print version information\n\n", MYSQL_PORT);
-       support ();
-}
 
+       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\
+ -S, --check-slave\n\
+   Check if the slave thread is running properly.\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));
+}
 
 
 void
 print_usage (void)
 {
-       printf
-               ("Usage: %s [-d database] [-H host] [-P port] [-u user] [-p password]\n"
-                "       %s --help\n"
-                "       %s --version\n", PROGNAME, PROGNAME, PROGNAME);
+       printf ("\
+Usage: %s [-d database] [-H host] [-P port] [-u user] [-p password] [-S]\n",
+               progname);
 }