X-Git-Url: https://git.tokkee.org/?a=blobdiff_plain;f=plugins%2Fcheck_pgsql.c;h=e66eeaff91e9d3dc439422b169a91b93f2211851;hb=2ab4d8fc278413b01ff9c4e4496f892398f80a4c;hp=2dce0b35d2ef533f254fec2f1f93ccc48b618a26;hpb=b07cec408f61acd3c34ad1c03b212bba9b66c454;p=nagiosplug.git diff --git a/plugins/check_pgsql.c b/plugins/check_pgsql.c index 2dce0b3..e66eeaf 100644 --- a/plugins/check_pgsql.c +++ b/plugins/check_pgsql.c @@ -1,28 +1,34 @@ /****************************************************************************** - * - * Program: PostgreSQL plugin for Nagios - * License: GPL - * - * License Information: - * - * 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 2 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, write to the Free Software - * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. - * - * $Id$ - * + + 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 2 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, write to the Free Software + Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. + + $Id$ + *****************************************************************************/ +const char *progname = "check_pgsql"; +const char *revision = "$Revision$"; +const char *copyright = "1999-2004"; +const char *email = "nagiosplug-devel@lists.sourceforge.net"; + +#include "common.h" +#include "utils.h" + +#include "netutils.h" +#include + #define DEFAULT_DB "template1" #define DEFAULT_HOST "127.0.0.1" @@ -32,10 +38,7 @@ enum { DEFAULT_CRIT = 8 }; -#include "config.h" -#include "common.h" -#include "utils.h" -#include + int process_arguments (int, char **); int validate_arguments (void); @@ -52,17 +55,12 @@ char *pgtty = NULL; char dbName[NAMEDATALEN] = DEFAULT_DB; char *pguser = NULL; char *pgpasswd = NULL; -int twarn = DEFAULT_WARN; -int tcrit = DEFAULT_CRIT; +double twarn = (double)DEFAULT_WARN; +double tcrit = (double)DEFAULT_CRIT; PGconn *conn; /*PGresult *res;*/ -const char *progname = "check_pgsql"; -const char *revision = "$Revision$"; -const char *copyright = "1999-2003"; -const char *email = "nagiosplug-devel@lists.sourceforge.net"; - /****************************************************************************** @@ -117,81 +115,14 @@ Please note that all tags must be lowercase to use the DocBook XML DTD. Functions -@@ ******************************************************************************/ - - - - -void -print_help (void) -{ - char *myport; - - asprintf (&myport, "%d", DEFAULT_PORT); - - print_revision (progname, revision); - - printf (_(COPYRIGHT), copyright, email); - - printf (_("Test whether a PostgreSQL DBMS is accepting connections.\n\n")); - - print_usage (); - - printf (_(UT_HELP_VRSN)); - - printf (_(UT_HOST_PORT), 'P', myport); - - printf (_(UT_IPv46)); - - printf (S_("\ - -d, --database=STRING\n\ - Database to check (default: %s)\n\ - -l, --logname = STRING\n\ - Login name of user\n\ - -p, --password = STRING\n\ - Password (BIG SECURITY ISSUE)\n"), DEFAULT_DB); - - printf (_(UT_WARN_CRIT)); - - printf (_(UT_TIMEOUT), DEFAULT_SOCKET_TIMEOUT); - - printf (_(UT_VERBOSE)); - - printf (S_("\nAll parameters are optional.\n\ -\n\ -This plugin tests a PostgreSQL DBMS to determine whether it is active and\n\ -accepting queries. In its current operation, it simply connects to the\n\ -specified database, and then disconnects. If no database is specified, it\n\ -connects to the template1 database, which is present in every functioning \n\ -PostgreSQL DBMS.\n")); - printf (S_("\n\ -The plugin will connect to a local postmaster if no host is specified. To\n\ -connect to a remote host, be sure that the remote postmaster accepts TCP/IP\n\ -connections (start the postmaster with the -i option).\n")); - printf (S_("\n\ -Typically, the nagios user (unless the --logname option is used) should be\n\ -able to connect to the database without a password. The plugin can also send\n\ -a password, but no effort is made to obsure or encrypt the password.\n")); - support (); -} -void -print_usage (void) -{ - printf (S_("\ -Usage:\n %s [-H ] [-P ] [-c ] [-w ]\n\ - [-t ]"), progname); - printf (S_("[-d ] [-l ] [-p ]\n")); - printf (S_("\ - %s (-h | --help) for detailed help\n\ - %s (-V | --version) for version information\n"), - progname, progname); -} int main (int argc, char **argv) { int elapsed_time; + int status = STATE_UNKNOWN; /* begin, by setting the parameters for a backend connection if the * parameters are null, then the system will try to use reasonable @@ -201,13 +132,16 @@ main (int argc, char **argv) pgoptions = NULL; /* special options to start up the backend server */ pgtty = NULL; /* debugging tty for the backend server */ + setlocale (LC_ALL, ""); + bindtextdomain (PACKAGE, LOCALEDIR); + textdomain (PACKAGE); + if (process_arguments (argc, argv) == ERROR) - usage ("Could not parse arguments"); + usage4 (_("Could not parse arguments")); /* Set signal handling and alarm */ if (signal (SIGALRM, timeout_alarm_handler) == SIG_ERR) { - printf (_("Cannot catch SIGALRM")); - return STATE_UNKNOWN; + usage4 (_("Cannot catch SIGALRM")); } alarm (timeout_interval); @@ -220,29 +154,28 @@ main (int argc, char **argv) /* check to see that the backend connection was successfully made */ if (PQstatus (conn) == CONNECTION_BAD) { - printf (_("PGSQL: CRITICAL - no connection to '%s' (%s).\n"), dbName, - PQerrorMessage (conn)); + printf (_("CRITICAL - no connection to '%s' (%s).\n"), + dbName, PQerrorMessage (conn)); PQfinish (conn); return STATE_CRITICAL; } else if (elapsed_time > tcrit) { - PQfinish (conn); - printf (_("PGSQL: CRITICAL - database %s (%d sec.)\n"), dbName, - elapsed_time); - return STATE_CRITICAL; + status = STATE_CRITICAL; } else if (elapsed_time > twarn) { - PQfinish (conn); - printf (_("PGSQL: WARNING - database %s (%d sec.)\n"), dbName, elapsed_time); - return STATE_WARNING; + status = STATE_WARNING; } else { - PQfinish (conn); - printf (_("PGSQL: ok - database %s (%d sec.)\n"), dbName, elapsed_time); - return STATE_OK; + status = STATE_OK; } + PQfinish (conn); + printf (_(" %s - database %s (%d sec.)|%s\n"), + state_text(status), dbName, elapsed_time, + fperfdata("time", elapsed_time, "s", + (int)twarn, twarn, (int)tcrit, tcrit, TRUE, 0, FALSE,0)); + return status; } - + /* process command-line arguments */ @@ -251,8 +184,8 @@ process_arguments (int argc, char **argv) { int c; - int option_index = 0; - static struct option long_options[] = { + int option = 0; + static struct option longopts[] = { {"help", no_argument, 0, 'h'}, {"version", no_argument, 0, 'V'}, {"timeout", required_argument, 0, 't'}, @@ -269,14 +202,16 @@ process_arguments (int argc, char **argv) while (1) { c = getopt_long (argc, argv, "hVt:c:w:H:P:d:l:p:a:", - long_options, &option_index); + longopts, &option); if (c == EOF) break; switch (c) { case '?': /* usage */ - usage3 (_("Unknown argument"), optopt); + printf (_("%s: Unknown argument: %s\n\n"), progname, optarg); + print_usage (); + exit (STATE_UNKNOWN); case 'h': /* help */ print_help (); exit (STATE_OK); @@ -285,39 +220,45 @@ process_arguments (int argc, char **argv) exit (STATE_OK); case 't': /* timeout period */ if (!is_integer (optarg)) - usage2 (_("Timeout Interval must be an integer"), optarg); - timeout_interval = atoi (optarg); + usage2 (_("Timeout interval must be a positive integer"), optarg); + else + timeout_interval = atoi (optarg); break; case 'c': /* critical time threshold */ - if (!is_integer (optarg)) - usage2 (_("Invalid critical threshold"), optarg); - tcrit = atoi (optarg); + if (!is_nonnegative (optarg)) + usage2 (_("Critical threshold must be a positive integer"), optarg); + else + tcrit = strtod (optarg, NULL); break; case 'w': /* warning time threshold */ - if (!is_integer (optarg)) - usage2 (_("Invalid critical threshold"), optarg); - twarn = atoi (optarg); + if (!is_nonnegative (optarg)) + usage2 (_("Critical threshold must be a positive integer"), optarg); + else + twarn = strtod (optarg, NULL); break; case 'H': /* host */ if (!is_host (optarg)) - usage2 (_("You gave an invalid host name"), optarg); - pghost = optarg; + usage2 (_("Invalid hostname/address"), optarg); + else + pghost = optarg; break; case 'P': /* port */ if (!is_integer (optarg)) - usage2 (_("Port must be an integer"), optarg); - pgport = optarg; + usage2 (_("Port must be a positive integer"), optarg); + else + pgport = optarg; break; case 'd': /* database name */ - if (!is_pg_dbname (optarg)) + if (!is_pg_dbname (optarg)) /* checks length and valid chars */ usage2 (_("Database name is not valid"), optarg); - strncpy (dbName, optarg, NAMEDATALEN - 1); - dbName[NAMEDATALEN - 1] = 0; + else /* we know length, and know optarg is terminated, so us strcpy */ + strcpy (dbName, optarg); break; case 'l': /* login name */ if (!is_pg_logname (optarg)) - usage2 (_("user name is not valid"), optarg); - pguser = optarg; + usage2 (_("User name is not valid"), optarg); + else + pguser = optarg; break; case 'p': /* authentication password */ case 'a': @@ -350,12 +291,13 @@ first character cannot be a number, however. -@@ ******************************************************************************/ + + int validate_arguments () { return OK; } - /****************************************************************************** @@ -378,6 +320,8 @@ first character cannot be a number, however. -@@ ******************************************************************************/ + + int is_pg_dbname (char *dbname) { @@ -418,6 +362,8 @@ should be added. -@@ ******************************************************************************/ + + int is_pg_logname (char *username) { @@ -433,3 +379,69 @@ is_pg_logname (char *username) -@@ ******************************************************************************/ + + + +void +print_help (void) +{ + char *myport; + + asprintf (&myport, "%d", DEFAULT_PORT); + + print_revision (progname, revision); + + printf (COPYRIGHT, copyright, email); + + printf (_("Test whether a PostgreSQL Database is accepting connections.\n\n")); + + print_usage (); + + printf (_(UT_HELP_VRSN)); + + printf (_(UT_HOST_PORT), 'P', myport); + + printf (_(UT_IPv46)); + + printf (S_("\ + -d, --database=STRING\n\ + Database to check (default: %s)\n\ + -l, --logname = STRING\n\ + Login name of user\n\ + -p, --password = STRING\n\ + Password (BIG SECURITY ISSUE)\n"), DEFAULT_DB); + + printf (_(UT_WARN_CRIT)); + + printf (_(UT_TIMEOUT), DEFAULT_SOCKET_TIMEOUT); + + printf (_(UT_VERBOSE)); + + printf (S_("\nAll parameters are optional.\n\ +\n\ +This plugin tests a PostgreSQL DBMS to determine whether it is active and\n\ +accepting queries. In its current operation, it simply connects to the\n\ +specified database, and then disconnects. If no database is specified, it\n\ +connects to the template1 database, which is present in every functioning \n\ +PostgreSQL DBMS.\n")); + printf (S_("\n\ +The plugin will connect to a local postmaster if no host is specified. To\n\ +connect to a remote host, be sure that the remote postmaster accepts TCP/IP\n\ +connections (start the postmaster with the -i option).\n")); + printf (S_("\n\ +Typically, the nagios user (unless the --logname option is used) should be\n\ +able to connect to the database without a password. The plugin can also send\n\ +a password, but no effort is made to obsure or encrypt the password.\n")); + + printf (_(UT_SUPPORT)); +} + + + +void +print_usage (void) +{ + printf ("\ +Usage: %s [-H ] [-P ] [-c ] [-w ]\n\ + [-t ] [-d ] [-l ] [-p ]\n", progname); +}