diff options
Diffstat (limited to 'plugins/check_ldap.c')
-rw-r--r-- | plugins/check_ldap.c | 517 |
1 files changed, 517 insertions, 0 deletions
diff --git a/plugins/check_ldap.c b/plugins/check_ldap.c new file mode 100644 index 0000000..15113b1 --- /dev/null +++ b/plugins/check_ldap.c @@ -0,0 +1,517 @@ +/***************************************************************************** +* +* Monitoring check_ldap plugin +* +* License: GPL +* Copyright (c) 2000-2008 Monitoring Plugins Development Team +* +* Description: +* +* This file contains the check_ldap plugin +* +* +* 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/>. +* +* +*****************************************************************************/ + +/* progname may be check_ldaps */ +char *progname = "check_ldap"; +const char *copyright = "2000-2008"; +const char *email = "devel@monitoring-plugins.org"; + +#include "common.h" +#include "netutils.h" +#include "utils.h" + +#include <lber.h> +#define LDAP_DEPRECATED 1 +#include <ldap.h> + +enum { + UNDEFINED = 0, +#ifdef HAVE_LDAP_SET_OPTION + DEFAULT_PROTOCOL = 2, +#endif + DEFAULT_PORT = 389 +}; + +int process_arguments (int, char **); +int validate_arguments (void); +void print_help (void); +void print_usage (void); + +char ld_defattr[] = "(objectclass=*)"; +char *ld_attr = ld_defattr; +char *ld_host = NULL; +char *ld_base = NULL; +char *ld_passwd = NULL; +char *ld_binddn = NULL; +int ld_port = -1; +#ifdef HAVE_LDAP_SET_OPTION +int ld_protocol = DEFAULT_PROTOCOL; +#endif +#ifndef LDAP_OPT_SUCCESS +# define LDAP_OPT_SUCCESS LDAP_SUCCESS +#endif +double warn_time = UNDEFINED; +double crit_time = UNDEFINED; +thresholds *entries_thresholds = NULL; +struct timeval tv; +char* warn_entries = NULL; +char* crit_entries = NULL; +int starttls = FALSE; +int ssl_on_connect = FALSE; +int verbose = 0; + +/* for ldap tls */ + +char *SERVICE = "LDAP"; + +int +main (int argc, char *argv[]) +{ + + LDAP *ld; + LDAPMessage *result; + + /* should be int result = STATE_UNKNOWN; */ + + int status = STATE_UNKNOWN; + long microsec; + double elapsed_time; + + /* for ldap tls */ + + int tls; + int version=3; + + int status_entries = STATE_OK; + int num_entries = 0; + + setlocale (LC_ALL, ""); + bindtextdomain (PACKAGE, LOCALEDIR); + textdomain (PACKAGE); + + if (strstr(argv[0],"check_ldaps")) { + xasprintf (&progname, "check_ldaps"); + } + + /* Parse extra opts if any */ + argv=np_extra_opts (&argc, argv, progname); + + if (process_arguments (argc, argv) == ERROR) + usage4 (_("Could not parse arguments")); + + if (strstr(argv[0],"check_ldaps") && ! starttls && ! ssl_on_connect) + starttls = TRUE; + + /* initialize alarm signal handling */ + signal (SIGALRM, socket_timeout_alarm_handler); + + /* set socket timeout */ + alarm (socket_timeout); + + /* get the start time */ + gettimeofday (&tv, NULL); + + /* initialize ldap */ +#ifdef HAVE_LDAP_INIT + if (!(ld = ldap_init (ld_host, ld_port))) { + printf ("Could not connect to the server at port %i\n", ld_port); + return STATE_CRITICAL; + } +#else + if (!(ld = ldap_open (ld_host, ld_port))) { + if (verbose) + ldap_perror(ld, "ldap_open"); + printf (_("Could not connect to the server at port %i\n"), ld_port); + return STATE_CRITICAL; + } +#endif /* HAVE_LDAP_INIT */ + +#ifdef HAVE_LDAP_SET_OPTION + /* set ldap options */ + if (ldap_set_option (ld, LDAP_OPT_PROTOCOL_VERSION, &ld_protocol) != + LDAP_OPT_SUCCESS ) { + printf(_("Could not set protocol version %d\n"), ld_protocol); + return STATE_CRITICAL; + } +#endif + + if (ld_port == LDAPS_PORT || ssl_on_connect) { + xasprintf (&SERVICE, "LDAPS"); +#if defined(HAVE_LDAP_SET_OPTION) && defined(LDAP_OPT_X_TLS) + /* ldaps: set option tls */ + tls = LDAP_OPT_X_TLS_HARD; + + if (ldap_set_option (ld, LDAP_OPT_X_TLS, &tls) != LDAP_SUCCESS) + { + if (verbose) + ldap_perror(ld, "ldaps_option"); + printf (_("Could not init TLS at port %i!\n"), ld_port); + return STATE_CRITICAL; + } +#else + printf (_("TLS not supported by the libraries!\n")); + return STATE_CRITICAL; +#endif /* LDAP_OPT_X_TLS */ + } else if (starttls) { + xasprintf (&SERVICE, "LDAP-TLS"); +#if defined(HAVE_LDAP_SET_OPTION) && defined(HAVE_LDAP_START_TLS_S) + /* ldap with startTLS: set option version */ + if (ldap_get_option(ld,LDAP_OPT_PROTOCOL_VERSION, &version) == LDAP_OPT_SUCCESS ) + { + if (version < LDAP_VERSION3) + { + version = LDAP_VERSION3; + ldap_set_option(ld, LDAP_OPT_PROTOCOL_VERSION, &version); + } + } + /* call start_tls */ + if (ldap_start_tls_s(ld, NULL, NULL) != LDAP_SUCCESS) + { + if (verbose) + ldap_perror(ld, "ldap_start_tls"); + printf (_("Could not init startTLS at port %i!\n"), ld_port); + return STATE_CRITICAL; + } +#else + printf (_("startTLS not supported by the library, needs LDAPv3!\n")); + return STATE_CRITICAL; +#endif /* HAVE_LDAP_START_TLS_S */ + } + + /* bind to the ldap server */ + if (ldap_bind_s (ld, ld_binddn, ld_passwd, LDAP_AUTH_SIMPLE) != + LDAP_SUCCESS) { + if (verbose) + ldap_perror(ld, "ldap_bind"); + printf (_("Could not bind to the LDAP server\n")); + return STATE_CRITICAL; + } + + /* do a search of all objectclasses in the base dn */ + if (ldap_search_s (ld, ld_base, (crit_entries!=NULL || warn_entries!=NULL) ? LDAP_SCOPE_SUBTREE : LDAP_SCOPE_BASE, ld_attr, NULL, 0, &result) + != LDAP_SUCCESS) { + if (verbose) + ldap_perror(ld, "ldap_search"); + printf (_("Could not search/find objectclasses in %s\n"), ld_base); + return STATE_CRITICAL; + } else if (crit_entries!=NULL || warn_entries!=NULL) { + num_entries = ldap_count_entries(ld, result); + } + + /* unbind from the ldap server */ + ldap_unbind (ld); + + /* reset the alarm handler */ + alarm (0); + + /* calculate the elapsed time and compare to thresholds */ + + microsec = deltime (tv); + elapsed_time = (double)microsec / 1.0e6; + + if (crit_time!=UNDEFINED && elapsed_time>crit_time) + status = STATE_CRITICAL; + else if (warn_time!=UNDEFINED && elapsed_time>warn_time) + status = STATE_WARNING; + else + status = STATE_OK; + + if(entries_thresholds != NULL) { + if (verbose) { + printf ("entries found: %d\n", num_entries); + print_thresholds("entry thresholds", entries_thresholds); + } + status_entries = get_status(num_entries, entries_thresholds); + if (status_entries == STATE_CRITICAL) { + status = STATE_CRITICAL; + } else if (status != STATE_CRITICAL) { + status = status_entries; + } + } + + /* print out the result */ + if (crit_entries!=NULL || warn_entries!=NULL) { + printf (_("LDAP %s - found %d entries in %.3f seconds|%s %s\n"), + state_text (status), + num_entries, + elapsed_time, + fperfdata ("time", elapsed_time, "s", + (int)warn_time, warn_time, + (int)crit_time, crit_time, + TRUE, 0, FALSE, 0), + sperfdata ("entries", (double)num_entries, "", + warn_entries, + crit_entries, + TRUE, 0.0, FALSE, 0.0)); + } else { + printf (_("LDAP %s - %.3f seconds response time|%s\n"), + state_text (status), + elapsed_time, + fperfdata ("time", elapsed_time, "s", + (int)warn_time, warn_time, + (int)crit_time, crit_time, + TRUE, 0, FALSE, 0)); + } + + return status; +} + +/* process command-line arguments */ +int +process_arguments (int argc, char **argv) +{ + int c; + + int option = 0; + /* initialize the long option struct */ + static struct option longopts[] = { + {"help", no_argument, 0, 'h'}, + {"version", no_argument, 0, 'V'}, + {"timeout", required_argument, 0, 't'}, + {"hostname", required_argument, 0, 'H'}, + {"base", required_argument, 0, 'b'}, + {"attr", required_argument, 0, 'a'}, + {"bind", required_argument, 0, 'D'}, + {"pass", required_argument, 0, 'P'}, +#ifdef HAVE_LDAP_SET_OPTION + {"ver2", no_argument, 0, '2'}, + {"ver3", no_argument, 0, '3'}, +#endif + {"starttls", no_argument, 0, 'T'}, + {"ssl", no_argument, 0, 'S'}, + {"use-ipv4", no_argument, 0, '4'}, + {"use-ipv6", no_argument, 0, '6'}, + {"port", required_argument, 0, 'p'}, + {"warn", required_argument, 0, 'w'}, + {"crit", required_argument, 0, 'c'}, + {"warn-entries", required_argument, 0, 'W'}, + {"crit-entries", required_argument, 0, 'C'}, + {"verbose", no_argument, 0, 'v'}, + {0, 0, 0, 0} + }; + + if (argc < 2) + return ERROR; + + for (c = 1; c < argc; c++) { + if (strcmp ("-to", argv[c]) == 0) + strcpy (argv[c], "-t"); + } + + while (1) { + c = getopt_long (argc, argv, "hvV234TS6t:c:w:H:b:p:a:D:P:C:W:", longopts, &option); + + if (c == -1 || c == EOF) + break; + + switch (c) { + case 'h': /* help */ + print_help (); + exit (STATE_UNKNOWN); + case 'V': /* version */ + print_revision (progname, NP_VERSION); + exit (STATE_UNKNOWN); + case 't': /* timeout period */ + if (!is_intnonneg (optarg)) + usage2 (_("Timeout interval must be a positive integer"), optarg); + else + socket_timeout = atoi (optarg); + break; + case 'H': + ld_host = optarg; + break; + case 'b': + ld_base = optarg; + break; + case 'p': + ld_port = atoi (optarg); + break; + case 'a': + ld_attr = optarg; + break; + case 'D': + ld_binddn = optarg; + break; + case 'P': + ld_passwd = optarg; + break; + case 'w': + warn_time = strtod (optarg, NULL); + break; + case 'c': + crit_time = strtod (optarg, NULL); + break; + case 'W': + warn_entries = optarg; + break; + case 'C': + crit_entries = optarg; + break; +#ifdef HAVE_LDAP_SET_OPTION + case '2': + ld_protocol = 2; + break; + case '3': + ld_protocol = 3; + break; +#endif + case '4': + address_family = AF_INET; + break; + case 'v': + verbose++; + break; + case 'T': + if (! ssl_on_connect) + starttls = TRUE; + else + usage_va(_("%s cannot be combined with %s"), "-T/--starttls", "-S/--ssl"); + break; + case 'S': + if (! starttls) { + ssl_on_connect = TRUE; + if (ld_port == -1) + ld_port = LDAPS_PORT; + } else + usage_va(_("%s cannot be combined with %s"), "-S/--ssl", "-T/--starttls"); + break; + case '6': +#ifdef USE_IPV6 + address_family = AF_INET6; +#else + usage (_("IPv6 support not available\n")); +#endif + break; + default: + usage5 (); + } + } + + c = optind; + if (ld_host == NULL && is_host(argv[c])) + ld_host = strdup (argv[c++]); + + if (ld_base == NULL && argv[c]) + ld_base = strdup (argv[c++]); + + if (ld_port == -1) + ld_port = DEFAULT_PORT; + + return validate_arguments (); +} + + +int +validate_arguments () +{ + if (ld_host==NULL || strlen(ld_host)==0) + usage4 (_("Please specify the host name\n")); + + if (ld_base==NULL) + usage4 (_("Please specify the LDAP base\n")); + + if (crit_entries!=NULL || warn_entries!=NULL) { + set_thresholds(&entries_thresholds, + warn_entries, crit_entries); + } + if (ld_passwd==NULL) + ld_passwd = getenv("LDAP_PASSWORD"); + + return OK; +} + + +void +print_help (void) +{ + char *myport; + xasprintf (&myport, "%d", DEFAULT_PORT); + + print_revision (progname, NP_VERSION); + + printf ("Copyright (c) 1999 Didi Rieder (adrieder@sbox.tu-graz.ac.at)\n"); + printf (COPYRIGHT, copyright, email); + + printf ("\n\n"); + + print_usage (); + + printf (UT_HELP_VRSN); + printf (UT_EXTRA_OPTS); + + printf (UT_HOST_PORT, 'p', myport); + + printf (UT_IPv46); + + printf (" %s\n", "-a [--attr]"); + printf (" %s\n", _("ldap attribute to search (default: \"(objectclass=*)\"")); + printf (" %s\n", "-b [--base]"); + printf (" %s\n", _("ldap base (eg. ou=my unit, o=my org, c=at")); + printf (" %s\n", "-D [--bind]"); + printf (" %s\n", _("ldap bind DN (if required)")); + printf (" %s\n", "-P [--pass]"); + printf (" %s\n", _("ldap password (if required, or set the password through environment variable 'LDAP_PASSWORD')")); + printf (" %s\n", "-T [--starttls]"); + printf (" %s\n", _("use starttls mechanism introduced in protocol version 3")); + printf (" %s\n", "-S [--ssl]"); + printf (" %s %i\n", _("use ldaps (ldap v2 ssl method). this also sets the default port to"), LDAPS_PORT); + +#ifdef HAVE_LDAP_SET_OPTION + printf (" %s\n", "-2 [--ver2]"); + printf (" %s\n", _("use ldap protocol version 2")); + printf (" %s\n", "-3 [--ver3]"); + printf (" %s\n", _("use ldap protocol version 3")); + printf (" (%s %d)\n", _("default protocol version:"), DEFAULT_PROTOCOL); +#endif + + printf (UT_WARN_CRIT); + + printf (" %s\n", "-W [--warn-entries]"); + printf (" %s\n", _("Number of found entries to result in warning status")); + printf (" %s\n", "-C [--crit-entries]"); + printf (" %s\n", _("Number of found entries to result in critical status")); + + printf (UT_CONN_TIMEOUT, DEFAULT_SOCKET_TIMEOUT); + + printf (UT_VERBOSE); + + printf ("\n"); + printf ("%s\n", _("Notes:")); + printf (" %s\n", _("If this plugin is called via 'check_ldaps', method 'STARTTLS' will be")); + printf (_(" implied (using default port %i) unless --port=636 is specified. In that case\n"), DEFAULT_PORT); + printf (" %s\n", _("'SSL on connect' will be used no matter how the plugin was called.")); + printf (" %s\n", _("This detection is deprecated, please use 'check_ldap' with the '--starttls' or '--ssl' flags")); + printf (" %s\n", _("to define the behaviour explicitly instead.")); + printf (" %s\n", _("The parameters --warn-entries and --crit-entries are optional.")); + + printf (UT_SUPPORT); +} + +void +print_usage (void) +{ + printf ("%s\n", _("Usage:")); + printf (" %s -H <host> -b <base_dn> [-p <port>] [-a <attr>] [-D <binddn>]",progname); + printf ("\n [-P <password>] [-w <warn_time>] [-c <crit_time>] [-t timeout]%s\n", +#ifdef HAVE_LDAP_SET_OPTION + "\n [-2|-3] [-4|-6]" +#else + "" +#endif + ); +} |