Code

Included patch for check_ldap from Patch #753621 by Ziya Suzen
[nagiosplug.git] / plugins / check_ldap.c
index 6491e5ba1f4703a25d25abb69d027e242f96f317..8eb895691e3cf89d7b3d3fd9152b53cb10810af6 100644 (file)
@@ -7,7 +7,7 @@
  * 
  * Last Modified: $Date$
  *
- * Command line: check_ldap -h <host> -b <base_dn> -p <port> -w <warn_time> -w <crit_time>
+ * Command line: check_ldap -H <host> -b <base_dn> -p <port> -w <warn_time> -w <crit_time>
  *
  * Description:
  *
@@ -21,7 +21,7 @@
  *****************************************************************************/
 
 const char *progname = "check_ldap";
-#define REVISION "$Revision$"
+const char *revision = "$Revision$";
 
 #include "config.h"
 #include "common.h"
@@ -31,7 +31,13 @@ const char *progname = "check_ldap";
 #include <lber.h>
 #include <ldap.h>
 
-#define UNKNOWN -1
+enum {
+       UNDEFINED = -1,
+#ifdef HAVE_LDAP_SET_OPTION
+       DEFAULT_PROTOCOL = 2,
+#endif
+       DEFAULT_PORT = 389
+};
 
 int process_arguments (int, char **);
 int validate_arguments (void);
@@ -40,9 +46,16 @@ void print_usage (void);
 
 char ld_defattr[] = "(objectclass=*)";
 char *ld_attr = ld_defattr;
-char *ld_host = NULL, *ld_base = NULL, *ld_passwd = NULL, *ld_binddn = NULL;
-unsigned int ld_port = 389;
-int warn_time = UNKNOWN, crit_time = UNKNOWN;
+char *ld_host = "";
+char *ld_base = "";
+char *ld_passwd = NULL;
+char *ld_binddn = NULL;
+unsigned int ld_port = DEFAULT_PORT;
+#ifdef HAVE_LDAP_SET_OPTION
+int ld_protocol = DEFAULT_PROTOCOL;
+#endif
+int warn_time = UNDEFINED;
+int crit_time = UNDEFINED;
 
 int
 main (int argc, char *argv[])
@@ -73,6 +86,14 @@ main (int argc, char *argv[])
                return STATE_CRITICAL;
        }
 
+#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
        /* bind to the ldap server */
        if (ldap_bind_s (ld, ld_binddn, ld_passwd, LDAP_AUTH_SIMPLE) !=
                        LDAP_SUCCESS) {
@@ -98,21 +119,21 @@ main (int argc, char *argv[])
        /* get the finish time */
        time (&time1);
 
-       /* calcutate the elapsed time */
+       /* calcutate the elapsed time and compare to thresholds */
        t_diff = time1 - time0;
 
-       /* check if warn_time or crit_time was exceeded */
-       if ((t_diff >= warn_time) && (t_diff < crit_time)) {
-               printf ("LDAP warning - %i seconds response time\n", t_diff);
-               return STATE_WARNING;
-       }
-       if (t_diff >= crit_time) {
-               printf ("LDAP critical - %i seconds response time\n", t_diff);
+       if (crit_time!=UNDEFINED && t_diff>=crit_time) {
+               printf ("LDAP CRITICAL - %i seconds response time\n", t_diff);
                return STATE_CRITICAL;
        }
 
+       if (warn_time!=UNDEFINED && t_diff>=warn_time) {
+               printf ("LDAP WARNING - %i seconds response time\n", t_diff);
+               return STATE_WARNING;
+       }
+
        /* print out the result */
-       printf ("LDAP ok - %i seconds response time\n", t_diff);
+       printf ("LDAP OK - %i seconds response time\n", t_diff);
 
        return STATE_OK;
 }
@@ -123,7 +144,6 @@ process_arguments (int argc, char **argv)
 {
        int c;
 
-#ifdef HAVE_GETOPT_H
        int option_index = 0;
        /* initialize the long option struct */
        static struct option longopts[] = {
@@ -135,12 +155,15 @@ process_arguments (int argc, char **argv)
                {"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
                {"port", required_argument, 0, 'p'},
                {"warn", required_argument, 0, 'w'},
                {"crit", required_argument, 0, 'c'},
                {0, 0, 0, 0}
        };
-#endif
 
        if (argc < 2)
                return ERROR;
@@ -151,11 +174,7 @@ process_arguments (int argc, char **argv)
        }
 
        while (1) {
-#ifdef HAVE_GETOPT_H
-               c =     getopt_long (argc, argv, "hVt:c:w:H:b:p:a:D:P:", longopts, &option_index);
-#else
-               c = getopt (argc, argv, "+?hVt:c:w:H:b:p:a:D:P:");
-#endif
+               c = getopt_long (argc, argv, "hV23t:c:w:H:b:p:a:D:P:", longopts, &option_index);
 
                if (c == -1 || c == EOF)
                        break;
@@ -165,11 +184,11 @@ process_arguments (int argc, char **argv)
                        print_help ();
                        exit (STATE_OK);
                case 'V':                                                                       /* version */
-                       print_revision (progname, REVISION);
+                       print_revision (progname, revision);
                        exit (STATE_OK);
                case 't':                                                                       /* timeout period */
                        if (!is_intnonneg (optarg))
-                               usage2 ("timeout interval must be an integer", optarg);
+                               usage2 ("timeout interval must be a positive integer", optarg);
                        socket_timeout = atoi (optarg);
                        break;
                case 'H':
@@ -196,8 +215,16 @@ process_arguments (int argc, char **argv)
                case 'c':
                        crit_time = atoi (optarg);
                        break;
+#ifdef HAVE_LDAP_SET_OPTION
+               case '2':
+                       ld_protocol = 2;
+                       break;
+               case '3':
+                       ld_protocol = 3;
+                       break;
+#endif
                default:
-                       usage ("check_ldap: could not parse arguments\n");
+                       usage ("check_ldap: could not parse unknown arguments\n");
                        break;
                }
        }
@@ -212,14 +239,15 @@ process_arguments (int argc, char **argv)
 int
 validate_arguments ()
 {
-       if (ld_host[0] == 0 ||
-                       ld_base[0] == 0 ||
-                       ld_port == UNKNOWN || warn_time == UNKNOWN || crit_time == UNKNOWN) {
-               return ERROR;
-       }
-       else {
+       if (strlen(ld_host) == 0)
+               usage ("please specify the host name\n");
+
+       if (strlen(ld_base) == 0)
+               usage ("please specify the LDAP base\n");
+
+       else
                return OK;
-       }
+
 }
 \f
 
@@ -228,7 +256,7 @@ validate_arguments ()
 void
 print_help ()
 {
-       print_revision (progname, REVISION);
+       print_revision (progname, revision);
        printf
                ("Copyright (c) 1999 Didi Rieder (adrieder@sbox.tu-graz.ac.at)\n"
                 "License: GPL\n" "\n");
@@ -241,10 +269,19 @@ print_help ()
                 "\t-b [--base] ... ldap base (eg. ou=my unit, o=my org, c=at)\n"
                 "\t-D [--bind] ... ldap bind DN (if required)\n"
                 "\t-P [--pass] ... ldap password (if required)\n"
-                "\t-p [--port] ... ldap port (normaly 389)\n"
+                "\t-p [--port] ... ldap port (default: %d)\n"
+#ifdef HAVE_LDAP_SET_OPTION
+                "\t-2 [--ver2] ... use ldap porotocol version 2\n"
+                "\t-3 [--ver3] ... use ldap porotocol version 3\n"
+                "\t\t(default protocol version: %d)\n"
+#endif
                 "\t-w [--warn] ... time in secs. - if the exceeds <warn> the STATE_WARNING will be returned\n"
                 "\t-c [--crit] ... time in secs. - if the exceeds <crit> the STATE_CRITICAL will be returned\n"
-                "\n");
+                "\n", DEFAULT_PORT
+#ifdef HAVE_LDAP_SET_OPTION
+                    , DEFAULT_PROTOCOL
+#endif
+               );
 }
 
 
@@ -252,7 +289,10 @@ void
 print_usage ()
 {
        printf
-               ("Usage: %s -H <host> -b <base_dn> -p <port> [-a <attr>] [-D <binddn>]\n"
+               ("Usage: %s -H <host> -b <base_dn> [-p <port>] [-a <attr>] [-D <binddn>]\n"
                 "         [-P <password>] [-w <warn_time>] [-c <crit_time>] [-t timeout]\n"
+#ifdef HAVE_LDAP_SET_OPTION
+                "         [-2|-3]\n"
+#endif
                 "(Note: all times are in seconds.)\n", progname);
 }