Logo Search packages:      
Sourcecode: nagios-plugins version File versions  Download package

check_ldap.c

/******************************************************************************

 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: check_ldap.c,v 1.30 2004/12/25 23:17:44 opensides Exp $
 
******************************************************************************/

/* progname may be check_ldaps */
char *progname = "check_ldap";
const char *revision = "$Revision: 1.30 $";
const char *copyright = "2000-2004";
const char *email = "nagiosplug-devel@lists.sourceforge.net";

#include "common.h"
#include "netutils.h"
#include "utils.h"

#include <lber.h>
#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 = DEFAULT_PORT;
#ifdef HAVE_LDAP_SET_OPTION
int ld_protocol = DEFAULT_PROTOCOL;
#endif
double warn_time = UNDEFINED;
double crit_time = UNDEFINED;
struct timeval tv;

/* 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;

      setlocale (LC_ALL, "");
      bindtextdomain (PACKAGE, LOCALEDIR);
      textdomain (PACKAGE);

      if (strstr(argv[0],"check_ldaps")) {
            asprintf (&progname, "check_ldaps");
      }
      
      if (process_arguments (argc, argv) == ERROR)
            usage4 (_("Could not parse arguments"));

      /* 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))) {
            /*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 (strstr(argv[0],"check_ldaps")) {
      /* with TLS */
            if ( ld_port == LDAPS_PORT ) {
                  asprintf (&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)
                  {
                        /*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"), ld_port);
                  return STATE_CRITICAL;
#endif /* LDAP_OPT_X_TLS */
            } else {
                  asprintf (&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)
                  {
                        /*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) {
            /*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, LDAP_SCOPE_BASE, ld_attr, NULL, 0, &result)
                  != LDAP_SUCCESS) {
            /*ldap_perror(ld, "ldap_search"); */
            printf (_("Could not search/find objectclasses in %s\n"), ld_base);
            return STATE_CRITICAL;
      }

      /* unbind from the ldap server */
      ldap_unbind (ld);

      /* reset the alarm handler */
      alarm (0);

      /* calcutate 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;

      /* print out the result */
      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'},
            {"host", 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
            {"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'},
            {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, "hV2346t:c:w:H:b:p:a:D:P:", longopts, &option);

            if (c == -1 || c == EOF)
                  break;

            switch (c) {
            case 'h':                                                   /* help */
                  print_help ();
                  exit (STATE_OK);
            case 'V':                                                   /* version */
                  print_revision (progname, revision);
                  exit (STATE_OK);
            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;
#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 '6':
#ifdef USE_IPV6
                  address_family = AF_INET6;
#else
                  usage (_("IPv6 support not available\n"));
#endif
                  break;
            default:
                  usage2 (_("Unknown argument"), optarg);
            }
      }

      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++]);

      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 || strlen(ld_base)==0)
            usage4 (_("Please specify the LDAP base\n"));

      return OK;
}


void
print_help (void)
{
      char *myport;
      asprintf (&myport, "%d", DEFAULT_PORT);

      print_revision (progname, revision);

      printf ("Copyright (c) 1999 Didi Rieder (adrieder@sbox.tu-graz.ac.at)\n");
      printf (COPYRIGHT, copyright, email);

      print_usage ();

      printf (_(UT_HELP_VRSN));

      printf (_(UT_HOST_PORT), 'p', myport);

      printf (_(UT_IPv46));

      printf (_("\
 -a [--attr]\n\
    ldap attribute to search (default: \"(objectclass=*)\"\n\
 -b [--base]\n\
    ldap base (eg. ou=my unit, o=my org, c=at)\n\
 -D [--bind]\n\
    ldap bind DN (if required)\n\
 -P [--pass]\n\
    ldap password (if required)\n"));

#ifdef HAVE_LDAP_SET_OPTION
      printf (_("\
 -2 [--ver2]\n\
     use ldap protocol version 2\n\
 -3 [--ver3]\n\
    use ldap protocol version 3\n\
    (default protocol version: %d)\n"),
              DEFAULT_PROTOCOL);
#endif

      printf (_(UT_WARN_CRIT));

      printf (_(UT_TIMEOUT), DEFAULT_SOCKET_TIMEOUT);

      printf (_(UT_VERBOSE));

      printf (_(UT_SUPPORT));
}


void
print_usage (void)
{
      printf ("\
Usage: %s -H <host> -b <base_dn> [-p <port>] [-a <attr>] [-D <binddn>]\n\
                  [-P <password>] [-w <warn_time>] [-c <crit_time>]\n\
                                                      [-t timeout]%s\n",
//(Note: all times are in seconds.)\n",
              progname,
#ifdef HAVE_LDAP_SET_OPTION
                  " [-2|-3] [-4|-6]"
#else
                  ""
#endif
                  );
}

Generated by  Doxygen 1.6.0   Back to index