src/openldap.c

   1 /**
   2  * collectd - src/openldap.c
   3  * Copyright (C) 2011       Kimo Rosenbaum
   4  * Copyright (C) 2014-2015  Marc Fournier
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a
   7  * copy of this software and associated documentation files (the "Software"),
   8  * to deal in the Software without restriction, including without limitation
   9  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
  10  * and/or sell copies of the Software, and to permit persons to whom the
  11  * Software is furnished to do so, subject to the following conditions:
  12  *
  13  * The above copyright notice and this permission notice shall be included in
  14  * all copies or substantial portions of the Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  21  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  22  * DEALINGS IN THE SOFTWARE.
  23  *
  24  * Authors:
  25  *   Kimo Rosenbaum <kimor79 at yahoo.com>
  26  *   Marc Fournier <marc.fournier at camptocamp.com>
  27  **/
  28
  29 #include "collectd.h"
  30
  31 #include "plugin.h"
  32 #include "utils/common/common.h"
  33
  34 #if defined(__APPLE__)
  35 #pragma clang diagnostic push
  36 #pragma clang diagnostic warning "-Wdeprecated-declarations"
  37 #endif
  38
  39 #include <lber.h>
  40 #include <ldap.h>
  41
  42 struct cldap_s /* {{{ */
  43 {
  44   char *name;
  45
  46   char *binddn;
  47   char *password;
  48   char *cacert;
  49   char *host;
  50   bool starttls;
  51   int timeout;
  52   char *url;
  53   bool verifyhost;
  54   int version;
  55
  56   LDAP *ld;
  57 };
  58 typedef struct cldap_s cldap_t; /* }}} */
  59
  60 static void cldap_free(void *arg) /* {{{ */
  61 {
  62   cldap_t *st = arg;
  63
  64   if (st == NULL)
  65     return;
  66
  67   sfree(st->binddn);
  68   sfree(st->password);
  69   sfree(st->cacert);
  70   sfree(st->host);
  71   sfree(st->name);
  72   sfree(st->url);
  73   if (st->ld)
  74     ldap_unbind_ext_s(st->ld, NULL, NULL);
  75
  76   sfree(st);
  77 } /* }}} void cldap_free */
  78
  79 /* initialize ldap for each host */
  80 static int cldap_init_host(cldap_t *st) /* {{{ */
  81 {
  82   int rc;
  83
  84   if (st->ld) {
  85     DEBUG("openldap plugin: Already connected to %s", st->url);
  86     return 0;
  87   }
  88
  89   rc = ldap_initialize(&st->ld, st->url);
  90   if (rc != LDAP_SUCCESS) {
  91     ERROR("openldap plugin: ldap_initialize failed: %s", ldap_err2string(rc));
  92     if (st->ld != NULL)
  93       ldap_unbind_ext_s(st->ld, NULL, NULL);
  94     st->ld = NULL;
  95     return (-1);
  96   }
  97
  98   ldap_set_option(st->ld, LDAP_OPT_PROTOCOL_VERSION, &st->version);
  99
 100   ldap_set_option(st->ld, LDAP_OPT_TIMEOUT,
 101                   &(const struct timeval){st->timeout, 0});
 102
 103   ldap_set_option(st->ld, LDAP_OPT_RESTART, LDAP_OPT_ON);
 104
 105   if (st->cacert != NULL)
 106     ldap_set_option(st->ld, LDAP_OPT_X_TLS_CACERTFILE, st->cacert);
 107
 108   if (st->verifyhost == false) {
 109     int never = LDAP_OPT_X_TLS_NEVER;
 110     ldap_set_option(st->ld, LDAP_OPT_X_TLS_REQUIRE_CERT, &never);
 111   }
 112
 113   if (st->starttls) {
 114     rc = ldap_start_tls_s(st->ld, NULL, NULL);
 115     if (rc != LDAP_SUCCESS) {
 116       ERROR("openldap plugin: Failed to start tls on %s: %s", st->url,
 117             ldap_err2string(rc));
 118       ldap_unbind_ext_s(st->ld, NULL, NULL);
 119       st->ld = NULL;
 120       return (-1);
 121     }
 122   }
 123
 124   struct berval cred;
 125   if (st->password != NULL) {
 126     cred.bv_val = st->password;
 127     cred.bv_len = strlen(st->password);
 128   } else {
 129     cred.bv_val = "";
 130     cred.bv_len = 0;
 131   }
 132
 133   rc = ldap_sasl_bind_s(st->ld, st->binddn, LDAP_SASL_SIMPLE, &cred, NULL, NULL,
 134                         NULL);
 135   if (rc != LDAP_SUCCESS) {
 136     ERROR("openldap plugin: Failed to bind to %s: %s", st->url,
 137           ldap_err2string(rc));
 138     ldap_unbind_ext_s(st->ld, NULL, NULL);
 139     st->ld = NULL;
 140     return (-1);
 141   } else {
 142     DEBUG("openldap plugin: Successfully connected to %s", st->url);
 143     return 0;
 144   }
 145 } /* }}} static cldap_init_host */
 146
 147 static void cldap_submit_value(const char *type,
 148                                const char *type_instance, /* {{{ */
 149                                value_t value, cldap_t *st) {
 150   value_list_t vl = VALUE_LIST_INIT;
 151
 152   vl.values = &value;
 153   vl.values_len = 1;
 154
 155   if ((st->host != NULL) && (strcmp("localhost", st->host) != 0))
 156     sstrncpy(vl.host, st->host, sizeof(vl.host));
 157
 158   sstrncpy(vl.plugin, "openldap", sizeof(vl.plugin));
 159   if (st->name != NULL)
 160     sstrncpy(vl.plugin_instance, st->name, sizeof(vl.plugin_instance));
 161
 162   sstrncpy(vl.type, type, sizeof(vl.type));
 163   if (type_instance != NULL)
 164     sstrncpy(vl.type_instance, type_instance, sizeof(vl.type_instance));
 165
 166   plugin_dispatch_values(&vl);
 167 } /* }}} void cldap_submit_value */
 168
 169 static void cldap_submit_derive(const char *type,
 170                                 const char *type_instance, /* {{{ */
 171                                 derive_t d, cldap_t *st) {
 172   cldap_submit_value(type, type_instance, (value_t){.derive = d}, st);
 173 } /* }}} void cldap_submit_derive */
 174
 175 static void cldap_submit_gauge(const char *type,
 176                                const char *type_instance, /* {{{ */
 177                                gauge_t g, cldap_t *st) {
 178   cldap_submit_value(type, type_instance, (value_t){.gauge = g}, st);
 179 } /* }}} void cldap_submit_gauge */
 180
 181 static int cldap_read_host(user_data_t *ud) /* {{{ */
 182 {
 183   cldap_t *st;
 184   LDAPMessage *result;
 185   char *dn;
 186   int rc;
 187   int status;
 188
 189   char *attrs[9] = {
 190       "monitorCounter", "monitorOpCompleted", "monitorOpInitiated",
 191       "monitoredInfo",  "olmBDBEntryCache",   "olmBDBDNCache",
 192       "olmBDBIDLCache", "namingContexts",     NULL};
 193
 194   if ((ud == NULL) || (ud->data == NULL)) {
 195     ERROR("openldap plugin: cldap_read_host: Invalid user data.");
 196     return -1;
 197   }
 198
 199   st = (cldap_t *)ud->data;
 200
 201   status = cldap_init_host(st);
 202   if (status != 0)
 203     return -1;
 204
 205   rc = ldap_search_ext_s(st->ld, "cn=Monitor", LDAP_SCOPE_SUBTREE,
 206                          "(|(!(cn=* *))(cn=Database*))", attrs, 0, NULL, NULL,
 207                          NULL, 0, &result);
 208
 209   if (rc != LDAP_SUCCESS) {
 210     ERROR("openldap plugin: Failed to execute search: %s", ldap_err2string(rc));
 211     ldap_msgfree(result);
 212     ldap_unbind_ext_s(st->ld, NULL, NULL);
 213     st->ld = NULL;
 214     return (-1);
 215   }
 216
 217   for (LDAPMessage *e = ldap_first_entry(st->ld, result); e != NULL;
 218        e = ldap_next_entry(st->ld, e)) {
 219     if ((dn = ldap_get_dn(st->ld, e)) != NULL) {
 220       unsigned long long counter = 0;
 221       unsigned long long opc = 0;
 222       unsigned long long opi = 0;
 223       unsigned long long info = 0;
 224
 225       struct berval counter_data;
 226       struct berval opc_data;
 227       struct berval opi_data;
 228       struct berval info_data;
 229       struct berval olmbdb_data;
 230       struct berval nc_data;
 231
 232       struct berval **counter_list;
 233       struct berval **opc_list;
 234       struct berval **opi_list;
 235       struct berval **info_list;
 236       struct berval **olmbdb_list;
 237       struct berval **nc_list;
 238
 239       if ((counter_list = ldap_get_values_len(st->ld, e, "monitorCounter")) !=
 240           NULL) {
 241         counter_data = *counter_list[0];
 242         counter = atoll(counter_data.bv_val);
 243       }
 244
 245       if ((opc_list = ldap_get_values_len(st->ld, e, "monitorOpCompleted")) !=
 246           NULL) {
 247         opc_data = *opc_list[0];
 248         opc = atoll(opc_data.bv_val);
 249       }
 250
 251       if ((opi_list = ldap_get_values_len(st->ld, e, "monitorOpInitiated")) !=
 252           NULL) {
 253         opi_data = *opi_list[0];
 254         opi = atoll(opi_data.bv_val);
 255       }
 256
 257       if ((info_list = ldap_get_values_len(st->ld, e, "monitoredInfo")) !=
 258           NULL) {
 259         info_data = *info_list[0];
 260         info = atoll(info_data.bv_val);
 261       }
 262
 263       if (strcmp(dn, "cn=Total,cn=Connections,cn=Monitor") == 0) {
 264         cldap_submit_derive("total_connections", NULL, counter, st);
 265       } else if (strcmp(dn, "cn=Current,cn=Connections,cn=Monitor") == 0) {
 266         cldap_submit_gauge("current_connections", NULL, counter, st);
 267       } else if (strcmp(dn, "cn=Operations,cn=Monitor") == 0) {
 268         cldap_submit_derive("operations", "completed", opc, st);
 269         cldap_submit_derive("operations", "initiated", opi, st);
 270       } else if (strcmp(dn, "cn=Bind,cn=Operations,cn=Monitor") == 0) {
 271         cldap_submit_derive("operations", "bind-completed", opc, st);
 272         cldap_submit_derive("operations", "bind-initiated", opi, st);
 273       } else if (strcmp(dn, "cn=UnBind,cn=Operations,cn=Monitor") == 0) {
 274         cldap_submit_derive("operations", "unbind-completed", opc, st);
 275         cldap_submit_derive("operations", "unbind-initiated", opi, st);
 276       } else if (strcmp(dn, "cn=Search,cn=Operations,cn=Monitor") == 0) {
 277         cldap_submit_derive("operations", "search-completed", opc, st);
 278         cldap_submit_derive("operations", "search-initiated", opi, st);
 279       } else if (strcmp(dn, "cn=Compare,cn=Operations,cn=Monitor") == 0) {
 280         cldap_submit_derive("operations", "compare-completed", opc, st);
 281         cldap_submit_derive("operations", "compare-initiated", opi, st);
 282       } else if (strcmp(dn, "cn=Modify,cn=Operations,cn=Monitor") == 0) {
 283         cldap_submit_derive("operations", "modify-completed", opc, st);
 284         cldap_submit_derive("operations", "modify-initiated", opi, st);
 285       } else if (strcmp(dn, "cn=Modrdn,cn=Operations,cn=Monitor") == 0) {
 286         cldap_submit_derive("operations", "modrdn-completed", opc, st);
 287         cldap_submit_derive("operations", "modrdn-initiated", opi, st);
 288       } else if (strcmp(dn, "cn=Add,cn=Operations,cn=Monitor") == 0) {
 289         cldap_submit_derive("operations", "add-completed", opc, st);
 290         cldap_submit_derive("operations", "add-initiated", opi, st);
 291       } else if (strcmp(dn, "cn=Delete,cn=Operations,cn=Monitor") == 0) {
 292         cldap_submit_derive("operations", "delete-completed", opc, st);
 293         cldap_submit_derive("operations", "delete-initiated", opi, st);
 294       } else if (strcmp(dn, "cn=Abandon,cn=Operations,cn=Monitor") == 0) {
 295         cldap_submit_derive("operations", "abandon-completed", opc, st);
 296         cldap_submit_derive("operations", "abandon-initiated", opi, st);
 297       } else if (strcmp(dn, "cn=Extended,cn=Operations,cn=Monitor") == 0) {
 298         cldap_submit_derive("operations", "extended-completed", opc, st);
 299         cldap_submit_derive("operations", "extended-initiated", opi, st);
 300       } else if ((strncmp(dn, "cn=Database", 11) == 0) &&
 301                  ((nc_list = ldap_get_values_len(st->ld, e,
 302                                                  "namingContexts")) != NULL)) {
 303         nc_data = *nc_list[0];
 304         char typeinst[DATA_MAX_NAME_LEN];
 305
 306         if ((olmbdb_list =
 307                  ldap_get_values_len(st->ld, e, "olmBDBEntryCache")) != NULL) {
 308           olmbdb_data = *olmbdb_list[0];
 309           snprintf(typeinst, sizeof(typeinst), "bdbentrycache-%s",
 310                    nc_data.bv_val);
 311           cldap_submit_gauge("cache_size", typeinst, atoll(olmbdb_data.bv_val),
 312                              st);
 313           ldap_value_free_len(olmbdb_list);
 314         }
 315
 316         if ((olmbdb_list = ldap_get_values_len(st->ld, e, "olmBDBDNCache")) !=
 317             NULL) {
 318           olmbdb_data = *olmbdb_list[0];
 319           snprintf(typeinst, sizeof(typeinst), "bdbdncache-%s", nc_data.bv_val);
 320           cldap_submit_gauge("cache_size", typeinst, atoll(olmbdb_data.bv_val),
 321                              st);
 322           ldap_value_free_len(olmbdb_list);
 323         }
 324
 325         if ((olmbdb_list = ldap_get_values_len(st->ld, e, "olmBDBIDLCache")) !=
 326             NULL) {
 327           olmbdb_data = *olmbdb_list[0];
 328           snprintf(typeinst, sizeof(typeinst), "bdbidlcache-%s",
 329                    nc_data.bv_val);
 330           cldap_submit_gauge("cache_size", typeinst, atoll(olmbdb_data.bv_val),
 331                              st);
 332           ldap_value_free_len(olmbdb_list);
 333         }
 334
 335         ldap_value_free_len(nc_list);
 336       } else if (strcmp(dn, "cn=Bytes,cn=Statistics,cn=Monitor") == 0) {
 337         cldap_submit_derive("derive", "statistics-bytes", counter, st);
 338       } else if (strcmp(dn, "cn=PDU,cn=Statistics,cn=Monitor") == 0) {
 339         cldap_submit_derive("derive", "statistics-pdu", counter, st);
 340       } else if (strcmp(dn, "cn=Entries,cn=Statistics,cn=Monitor") == 0) {
 341         cldap_submit_derive("derive", "statistics-entries", counter, st);
 342       } else if (strcmp(dn, "cn=Referrals,cn=Statistics,cn=Monitor") == 0) {
 343         cldap_submit_derive("derive", "statistics-referrals", counter, st);
 344       } else if (strcmp(dn, "cn=Open,cn=Threads,cn=Monitor") == 0) {
 345         cldap_submit_gauge("threads", "threads-open", info, st);
 346       } else if (strcmp(dn, "cn=Starting,cn=Threads,cn=Monitor") == 0) {
 347         cldap_submit_gauge("threads", "threads-starting", info, st);
 348       } else if (strcmp(dn, "cn=Active,cn=Threads,cn=Monitor") == 0) {
 349         cldap_submit_gauge("threads", "threads-active", info, st);
 350       } else if (strcmp(dn, "cn=Pending,cn=Threads,cn=Monitor") == 0) {
 351         cldap_submit_gauge("threads", "threads-pending", info, st);
 352       } else if (strcmp(dn, "cn=Backload,cn=Threads,cn=Monitor") == 0) {
 353         cldap_submit_gauge("threads", "threads-backload", info, st);
 354       } else if (strcmp(dn, "cn=Read,cn=Waiters,cn=Monitor") == 0) {
 355         cldap_submit_derive("derive", "waiters-read", counter, st);
 356       } else if (strcmp(dn, "cn=Write,cn=Waiters,cn=Monitor") == 0) {
 357         cldap_submit_derive("derive", "waiters-write", counter, st);
 358       }
 359
 360       ldap_value_free_len(counter_list);
 361       ldap_value_free_len(opc_list);
 362       ldap_value_free_len(opi_list);
 363       ldap_value_free_len(info_list);
 364     }
 365
 366     ldap_memfree(dn);
 367   }
 368
 369   ldap_msgfree(result);
 370   return 0;
 371 } /* }}} int cldap_read_host */
 372
 373 /* Configuration handling functions {{{
 374  *
 375  * <Plugin ldap>
 376  *   <Instance "plugin_instance1">
 377  *     URL "ldap://localhost"
 378  *     ...
 379  *   </Instance>
 380  * </Plugin>
 381  */
 382
 383 static int cldap_config_add(oconfig_item_t *ci) /* {{{ */
 384 {
 385   cldap_t *st;
 386   int status;
 387
 388   st = calloc(1, sizeof(*st));
 389   if (st == NULL) {
 390     ERROR("openldap plugin: calloc failed.");
 391     return -1;
 392   }
 393
 394   status = cf_util_get_string(ci, &st->name);
 395   if (status != 0) {
 396     sfree(st);
 397     return status;
 398   }
 399
 400   st->starttls = false;
 401   st->timeout = (long)CDTIME_T_TO_TIME_T(plugin_get_interval());
 402   st->verifyhost = true;
 403   st->version = LDAP_VERSION3;
 404
 405   for (int i = 0; i < ci->children_num; i++) {
 406     oconfig_item_t *child = ci->children + i;
 407
 408     if (strcasecmp("BindDN", child->key) == 0)
 409       status = cf_util_get_string(child, &st->binddn);
 410     else if (strcasecmp("Password", child->key) == 0)
 411       status = cf_util_get_string(child, &st->password);
 412     else if (strcasecmp("CACert", child->key) == 0)
 413       status = cf_util_get_string(child, &st->cacert);
 414     else if (strcasecmp("StartTLS", child->key) == 0)
 415       status = cf_util_get_boolean(child, &st->starttls);
 416     else if (strcasecmp("Timeout", child->key) == 0)
 417       status = cf_util_get_int(child, &st->timeout);
 418     else if (strcasecmp("URL", child->key) == 0)
 419       status = cf_util_get_string(child, &st->url);
 420     else if (strcasecmp("VerifyHost", child->key) == 0)
 421       status = cf_util_get_boolean(child, &st->verifyhost);
 422     else if (strcasecmp("Version", child->key) == 0)
 423       status = cf_util_get_int(child, &st->version);
 424     else {
 425       WARNING("openldap plugin: Option `%s' not allowed here.", child->key);
 426       status = -1;
 427     }
 428
 429     if (status != 0)
 430       break;
 431   }
 432
 433   /* Check if struct is complete.. */
 434   if ((status == 0) && (st->url == NULL)) {
 435     ERROR("openldap plugin: Instance `%s': "
 436           "No URL has been configured.",
 437           st->name);
 438     status = -1;
 439   }
 440
 441   /* Check if URL is valid */
 442   if ((status == 0) && (st->url != NULL)) {
 443     LDAPURLDesc *ludpp;
 444
 445     if (ldap_url_parse(st->url, &ludpp) != 0) {
 446       ERROR("openldap plugin: Instance `%s': "
 447             "Invalid URL: `%s'",
 448             st->name, st->url);
 449       status = -1;
 450     }
 451
 452     if ((status == 0) && (ludpp->lud_host != NULL))
 453       st->host = strdup(ludpp->lud_host);
 454
 455     ldap_free_urldesc(ludpp);
 456   }
 457
 458   if (status != 0) {
 459     cldap_free(st);
 460     return -1;
 461   }
 462
 463   char callback_name[3 * DATA_MAX_NAME_LEN] = {0};
 464
 465   snprintf(callback_name, sizeof(callback_name), "openldap/%s/%s",
 466            (st->host != NULL) ? st->host : hostname_g,
 467            (st->name != NULL) ? st->name : "default");
 468
 469   return plugin_register_complex_read(/* group = */ NULL,
 470                                       /* name      = */ callback_name,
 471                                       /* callback  = */ cldap_read_host,
 472                                       /* interval  = */ 0,
 473                                       &(user_data_t){
 474                                           .data = st, .free_func = cldap_free,
 475                                       });
 476 } /* }}} int cldap_config_add */
 477
 478 static int cldap_config(oconfig_item_t *ci) /* {{{ */
 479 {
 480   int status = 0;
 481
 482   for (int i = 0; i < ci->children_num; i++) {
 483     oconfig_item_t *child = ci->children + i;
 484
 485     if (strcasecmp("Instance", child->key) == 0)
 486       cldap_config_add(child);
 487     else
 488       WARNING("openldap plugin: The configuration option "
 489               "\"%s\" is not allowed here. Did you "
 490               "forget to add an <Instance /> block "
 491               "around the configuration?",
 492               child->key);
 493   } /* for (ci->children) */
 494
 495   return status;
 496 } /* }}} int cldap_config */
 497
 498 /* }}} End of configuration handling functions */
 499
 500 static int cldap_init(void) /* {{{ */
 501 {
 502   /* Initialize LDAP library while still single-threaded as recommended in
 503    * ldap_initialize(3) */
 504   int debug_level;
 505   ldap_get_option(NULL, LDAP_OPT_DEBUG_LEVEL, &debug_level);
 506   return 0;
 507 } /* }}} int cldap_init */
 508
 509 void module_register(void) /* {{{ */
 510 {
 511   plugin_register_complex_config("openldap", cldap_config);
 512   plugin_register_init("openldap", cldap_init);
 513 } /* }}} void module_register */
 514
 515 #if defined(__APPLE__)
 516 #pragma clang diagnostic pop
 517 #endif