X-Git-Url: https://git.tokkee.org/?a=blobdiff_plain;f=plugins%2Fcheck_mysql.c;h=3194ecede85b73f7ef572d06467848d066772e16;hb=7c69ccaee2baba7d1e8347b8b9e1f45a18551a6c;hp=18324c81784ff100ae02fc40f7a7022b943b5715;hpb=f2f3481179c0b11430cbdb2459e21e51ff265661;p=nagiosplug.git

diff --git a/plugins/check_mysql.c b/plugins/check_mysql.c
index 18324c8..3194ece 100644
--- a/plugins/check_mysql.c
+++ b/plugins/check_mysql.c
@@ -15,142 +15,194 @@
 ******************************************************************************/
 
 const char *progname = "check_mysql";
-#define REVISION "$Revision$"
-#define COPYRIGHT "1999-2002"
+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>
 
-char *db_user = "";
-char *db_host = "";
-char *db_pass = "";
-char *db = "";
+char *db_user = NULL;
+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 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 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;
-		}
+	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_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;
 
-#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
 
 	if (argc < 1)
 		return ERROR;
 
 	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
+		c = getopt_long (argc, argv, "hVSP:p:u:d:H:", longopts, &option);
 
 		if (c == -1 || c == EOF)
 			break;
@@ -161,7 +213,7 @@ process_arguments (int argc, char **argv)
 				db_host = optarg;
 			}
 			else {
-				usage ("Invalid host name\n");
+				usage2 (_("Invalid hostname/address"), optarg);
 			}
 			break;
 		case 'd':									/* hostname */
@@ -176,14 +228,17 @@ process_arguments (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 (progname, 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);
 		}
 	}
 
@@ -196,7 +251,7 @@ process_arguments (int argc, char **argv)
 				db_host = argv[c++];
 			}
 			else {
-				usage ("Invalid host name");
+				usage2 (_("Invalid hostname/address"), optarg);
 			}
 		else if (strlen(db_user) == 0)
 			db_user = argv[c++];
@@ -214,58 +269,67 @@ process_arguments (int argc, char **argv)
 }
 
 
-
-
-
 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);
 }