git.sur5r.net Git - openldap/blob - servers/slapd/back-bdb/monitor.c

   1 /* monitor.c - monitor bdb backend */
   2 /* $OpenLDAP$ */
   3 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
   4  *
   5  * Copyright 2000-2007 The OpenLDAP Foundation.
   6  * All rights reserved.
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted only as authorized by the OpenLDAP
  10  * Public License.
  11  *
  12  * A copy of this license is available in the file LICENSE in the
  13  * top-level directory of the distribution or, alternatively, at
  14  * <http://www.OpenLDAP.org/license.html>.
  15  */
  16
  17 #include "portable.h"
  18
  19 #include <stdio.h>
  20 #include <ac/string.h>
  21 #include <ac/unistd.h>
  22 #include <ac/stdlib.h>
  23 #include <ac/errno.h>
  24 #include <sys/stat.h>
  25 #include "lutil.h"
  26 #include "back-bdb.h"
  27
  28 #include "../back-monitor/back-monitor.h"
  29
  30 #include "config.h"
  31
  32 static ObjectClass              *oc_olmBDBDatabase;
  33
  34 static AttributeDescription     *ad_olmBDBEntryCache,
  35         *ad_olmBDBDNCache, *ad_olmBDBIDLCache,
  36         *ad_olmDbDirectory;
  37
  38 #ifdef BDB_MONITOR_IDX
  39 static int
  40 bdb_monitor_idx_entry_add(
  41         struct bdb_info *bdb,
  42         Entry           *e );
  43
  44 static AttributeDescription     *ad_olmBDBNotIndexed;
  45 #endif /* BDB_MONITOR_IDX */
  46
  47 /*
  48  * NOTE: there's some confusion in monitor OID arc;
  49  * by now, let's consider:
  50  *
  51  * Subsystems monitor attributes        1.3.6.1.4.1.4203.666.1.55.0
  52  * Databases monitor attributes         1.3.6.1.4.1.4203.666.1.55.0.1
  53  * BDB database monitor attributes      1.3.6.1.4.1.4203.666.1.55.0.1.1
  54  *
  55  * Subsystems monitor objectclasses     1.3.6.1.4.1.4203.666.3.16.0
  56  * Databases monitor objectclasses      1.3.6.1.4.1.4203.666.3.16.0.1
  57  * BDB database monitor objectclasses   1.3.6.1.4.1.4203.666.3.16.0.1.1
  58  */
  59
  60 static struct {
  61         char                    *name;
  62         char                    *oid;
  63 }               s_oid[] = {
  64         { "olmBDBAttributes",                   "olmDatabaseAttributes:1" },
  65         { "olmBDBObjectClasses",                "olmDatabaseObjectClasses:1" },
  66
  67         { NULL }
  68 };
  69
  70 static struct {
  71         char                    *desc;
  72         AttributeDescription    **ad;
  73 }               s_at[] = {
  74         { "( olmBDBAttributes:1 "
  75                 "NAME ( 'olmBDBEntryCache' ) "
  76                 "DESC 'Number of items in Entry Cache' "
  77                 "SUP monitorCounter "
  78                 "NO-USER-MODIFICATION "
  79                 "USAGE dSAOperation )",
  80                 &ad_olmBDBEntryCache },
  81
  82         { "( olmBDBAttributes:2 "
  83                 "NAME ( 'olmBDBDNCache' ) "
  84                 "DESC 'Number of items in DN Cache' "
  85                 "SUP monitorCounter "
  86                 "NO-USER-MODIFICATION "
  87                 "USAGE dSAOperation )",
  88                 &ad_olmBDBDNCache },
  89
  90         { "( olmBDBAttributes:3 "
  91                 "NAME ( 'olmBDBIDLCache' ) "
  92                 "DESC 'Number of items in IDL Cache' "
  93                 "SUP monitorCounter "
  94                 "NO-USER-MODIFICATION "
  95                 "USAGE dSAOperation )",
  96                 &ad_olmBDBIDLCache },
  97
  98         { "( olmBDBAttributes:4 "
  99                 "NAME ( 'olmDbDirectory' ) "
 100                 "DESC 'Path name of the directory "
 101                         "where the database environment resides' "
 102                 "SUP monitoredInfo "
 103                 "NO-USER-MODIFICATION "
 104                 "USAGE dSAOperation )",
 105                 &ad_olmDbDirectory },
 106
 107 #ifdef BDB_MONITOR_IDX
 108         { "( olmBDBAttributes:5 "
 109                 "NAME ( 'olmBDBNotIndexed' ) "
 110                 "DESC 'Missing indexes resulting from candidate selection' "
 111                 "SUP monitoredInfo "
 112                 "NO-USER-MODIFICATION "
 113                 "USAGE dSAOperation )",
 114                 &ad_olmBDBNotIndexed },
 115 #endif /* BDB_MONITOR_IDX */
 116
 117         { NULL }
 118 };
 119
 120 static struct {
 121         char            *desc;
 122         ObjectClass     **oc;
 123 }               s_oc[] = {
 124         /* augments an existing object, so it must be AUXILIARY
 125          * FIXME: derive from some ABSTRACT "monitoredEntity"? */
 126         { "( olmBDBObjectClasses:1 "
 127                 "NAME ( 'olmBDBDatabase' ) "
 128                 "SUP top AUXILIARY "
 129                 "MAY ( "
 130                         "olmBDBEntryCache "
 131                         "$ olmBDBDNCache "
 132                         "$ olmBDBIDLCache "
 133                         "$ olmDbDirectory "
 134 #ifdef BDB_MONITOR_IDX
 135                         "$ olmBDBNotIndexed "
 136 #endif /* BDB_MONITOR_IDX */
 137                         ") )",
 138                 &oc_olmBDBDatabase },
 139
 140         { NULL }
 141 };
 142
 143 static int
 144 bdb_monitor_update(
 145         Operation       *op,
 146         SlapReply       *rs,
 147         Entry           *e,
 148         void            *priv )
 149 {
 150         struct bdb_info         *bdb = (struct bdb_info *) priv;
 151         Attribute               *a;
 152
 153         char                    buf[ BUFSIZ ];
 154         struct berval           bv;
 155
 156         assert( ad_olmBDBEntryCache != NULL );
 157
 158         a = attr_find( e->e_attrs, ad_olmBDBEntryCache );
 159         assert( a != NULL );
 160         bv.bv_val = buf;
 161         bv.bv_len = snprintf( buf, sizeof( buf ), "%d", bdb->bi_cache.c_cursize );
 162         ber_bvreplace( &a->a_vals[ 0 ], &bv );
 163
 164         a = attr_find( e->e_attrs, ad_olmBDBDNCache );
 165         assert( a != NULL );
 166         bv.bv_len = snprintf( buf, sizeof( buf ), "%d", bdb->bi_cache.c_eiused );
 167         ber_bvreplace( &a->a_vals[ 0 ], &bv );
 168
 169         a = attr_find( e->e_attrs, ad_olmBDBIDLCache );
 170         assert( a != NULL );
 171         bv.bv_len = snprintf( buf, sizeof( buf ), "%d", bdb->bi_idl_cache_size );
 172         ber_bvreplace( &a->a_vals[ 0 ], &bv );
 173
 174 #ifdef BDB_MONITOR_IDX
 175         bdb_monitor_idx_entry_add( bdb, e );
 176 #endif /* BDB_MONITOR_IDX */
 177
 178         return SLAP_CB_CONTINUE;
 179 }
 180
 181 #if 0   /* uncomment if required */
 182 static int
 183 bdb_monitor_modify(
 184         Operation       *op,
 185         SlapReply       *rs,
 186         Entry           *e,
 187         void            *priv )
 188 {
 189         return SLAP_CB_CONTINUE;
 190 }
 191 #endif
 192
 193 static int
 194 bdb_monitor_free(
 195         Entry           *e,
 196         void            **priv )
 197 {
 198         struct berval   values[ 2 ];
 199         Modification    mod = { 0 };
 200
 201         const char      *text;
 202         char            textbuf[ SLAP_TEXT_BUFLEN ];
 203
 204         int             i, rc;
 205
 206         /* NOTE: if slap_shutdown != 0, priv might have already been freed */
 207         *priv = NULL;
 208
 209         /* Remove objectClass */
 210         mod.sm_op = LDAP_MOD_DELETE;
 211         mod.sm_desc = slap_schema.si_ad_objectClass;
 212         mod.sm_values = values;
 213         values[ 0 ] = oc_olmBDBDatabase->soc_cname;
 214         BER_BVZERO( &values[ 1 ] );
 215
 216         rc = modify_delete_values( e, &mod, 1, &text,
 217                 textbuf, sizeof( textbuf ) );
 218         /* don't care too much about return code... */
 219
 220         /* remove attrs */
 221         for ( i = 0; s_at[ i ].desc != NULL; i++ ) {
 222                 mod.sm_desc = *s_at[ i ].ad;
 223                 mod.sm_values = NULL;
 224                 rc = modify_delete_values( e, &mod, 1, &text,
 225                         textbuf, sizeof( textbuf ) );
 226                 /* don't care too much about return code... */
 227         }
 228
 229         return SLAP_CB_CONTINUE;
 230 }
 231
 232 #define bdb_monitor_initialize  BDB_SYMBOL(monitor_initialize)
 233
 234 /*
 235  * call from within bdb_initialize()
 236  */
 237 static int
 238 bdb_monitor_initialize( void )
 239 {
 240         int             i, code;
 241         ConfigArgs c;
 242         char    *argv[ 3 ];
 243
 244         static int      bdb_monitor_initialized = 0;
 245
 246         if ( backend_info( "monitor" ) == NULL ) {
 247                 return -1;
 248         }
 249
 250         if ( bdb_monitor_initialized++ ) {
 251                 return 0;
 252         }
 253
 254         /* register schema here */
 255
 256         argv[ 0 ] = "back-bdb/back-hdb monitor";
 257         c.argv = argv;
 258         c.argc = 3;
 259         c.fname = argv[0];
 260
 261         for ( i = 0; s_oid[ i ].name; i++ ) {
 262                 c.lineno = i;
 263                 argv[ 1 ] = s_oid[ i ].name;
 264                 argv[ 2 ] = s_oid[ i ].oid;
 265
 266                 if ( parse_oidm( &c, 0, NULL ) != 0 ) {
 267                         Debug( LDAP_DEBUG_ANY,
 268                                 "bdb_monitor_initialize: unable to add "
 269                                 "objectIdentifier \"%s=%s\"\n",
 270                                 s_oid[ i ].name, s_oid[ i ].oid, 0 );
 271                         return 1;
 272                 }
 273         }
 274
 275         for ( i = 0; s_at[ i ].desc != NULL; i++ ) {
 276                 code = register_at( s_at[ i ].desc, s_at[ i ].ad, 1 );
 277                 if ( code != LDAP_SUCCESS ) {
 278                         Debug( LDAP_DEBUG_ANY,
 279                                 "bdb_monitor_initialize: register_at failed\n",
 280                                 0, 0, 0 );
 281                 }
 282                 (*s_at[ i ].ad)->ad_type->sat_flags |= SLAP_AT_HIDE;
 283         }
 284
 285         for ( i = 0; s_oc[ i ].desc != NULL; i++ ) {
 286                 code = register_oc( s_oc[ i ].desc, s_oc[ i ].oc, 1 );
 287                 if ( code != LDAP_SUCCESS ) {
 288                         Debug( LDAP_DEBUG_ANY,
 289                                 "bdb_monitor_initialize: register_oc failed\n",
 290                                 0, 0, 0 );
 291                 }
 292                 (*s_oc[ i ].oc)->soc_flags |= SLAP_OC_HIDE;
 293         }
 294
 295         return 0;
 296 }
 297
 298 /*
 299  * call from within bdb_db_init()
 300  */
 301 int
 302 bdb_monitor_db_init( BackendDB *be )
 303 {
 304         struct bdb_info         *bdb = (struct bdb_info *) be->be_private;
 305
 306         if ( SLAP_GLUE_SUBORDINATE( be ) ) {
 307                 return 0;
 308         }
 309
 310         if ( bdb_monitor_initialize() == LDAP_SUCCESS ) {
 311                 /* monitoring in back-bdb is on by default */
 312                 SLAP_DBFLAGS( be ) |= SLAP_DBFLAG_MONITORING;
 313         }
 314
 315 #ifdef BDB_MONITOR_IDX
 316         bdb->bi_idx = NULL;
 317         ldap_pvt_thread_mutex_init( &bdb->bi_idx_mutex );
 318 #endif /* BDB_MONITOR_IDX */
 319
 320         bdb->bi_monitor.bdm_scope = -1;
 321
 322         return 0;
 323 }
 324
 325 /*
 326  * call from within bdb_db_open()
 327  */
 328 int
 329 bdb_monitor_db_open( BackendDB *be )
 330 {
 331         struct bdb_info         *bdb = (struct bdb_info *) be->be_private;
 332         Attribute               *a, *next;
 333         monitor_callback_t      *cb = NULL;
 334         struct berval           suffix, *filter, *base;
 335         char                    *ptr;
 336         int                     rc = 0;
 337         BackendInfo             *mi;
 338         monitor_extra_t         *mbe;
 339
 340         if ( !SLAP_DBMONITORING( be ) ) {
 341                 return 0;
 342         }
 343
 344         if ( SLAP_GLUE_SUBORDINATE( be ) ) {
 345                 return 0;
 346         }
 347
 348         mi = backend_info( "monitor" );
 349         if ( !mi || !mi->bi_extra ) {
 350                 SLAP_DBFLAGS( be ) ^= SLAP_DBFLAG_MONITORING;
 351                 return 0;
 352         }
 353         mbe = mi->bi_extra;
 354
 355         /* don't bother if monitor is not configured */
 356         if ( !mbe->is_configured() ) {
 357                 static int warning = 0;
 358
 359                 if ( warning++ == 0 ) {
 360                         Debug( LDAP_DEBUG_ANY, "bdb_monitor_open: "
 361                                 "monitoring disabled; "
 362                                 "configure monitor database to enable\n",
 363                                 0, 0, 0 );
 364                 }
 365
 366                 return 0;
 367         }
 368
 369         if ( bdb->bi_monitor.bdm_scope == -1 ) {
 370                 bdb->bi_monitor.bdm_scope = LDAP_SCOPE_ONELEVEL;
 371         }
 372         base = &bdb->bi_monitor.bdm_nbase;
 373         BER_BVSTR( base, "cn=databases,cn=monitor" );
 374         filter = &bdb->bi_monitor.bdm_filter;
 375         BER_BVZERO( filter );
 376
 377         suffix.bv_len = ldap_bv2escaped_filter_value_len( &be->be_nsuffix[ 0 ] );
 378         if ( suffix.bv_len == be->be_nsuffix[ 0 ].bv_len ) {
 379                 suffix = be->be_nsuffix[ 0 ];
 380
 381         } else {
 382                 ldap_bv2escaped_filter_value( &be->be_nsuffix[ 0 ], &suffix );
 383         }
 384
 385         if ( BER_BVISEMPTY( &suffix ) ) {
 386                 /* frontend also has empty suffix, sigh! */
 387                 filter->bv_len = STRLENOF( "(&(namingContexts:distinguishedNameMatch:=" )
 388                         + suffix.bv_len + STRLENOF( ")(!(cn=frontend)))" );
 389                 ptr = filter->bv_val = ch_malloc( filter->bv_len + 1 );
 390                 ptr = lutil_strcopy( ptr, "(&(namingContexts:distinguishedNameMatch:=" );
 391                 ptr = lutil_strncopy( ptr, suffix.bv_val, suffix.bv_len );
 392                 ptr = lutil_strcopy( ptr, ")(!(cn=frontend)))" );
 393
 394         } else {
 395                 /* just look for the naming context */
 396                 filter->bv_len = STRLENOF( "(namingContexts:distinguishedNameMatch:=" )
 397                         + suffix.bv_len + STRLENOF( ")" );
 398                 ptr = filter->bv_val = ch_malloc( filter->bv_len + 1 );
 399                 ptr = lutil_strcopy( ptr, "(namingContexts:distinguishedNameMatch:=" );
 400                 ptr = lutil_strncopy( ptr, suffix.bv_val, suffix.bv_len );
 401                 ptr = lutil_strcopy( ptr, ")" );
 402         }
 403         ptr[ 0 ] = '\0';
 404         assert( filter->bv_len == ptr - filter->bv_val );
 405
 406         if ( suffix.bv_val != be->be_nsuffix[ 0 ].bv_val ) {
 407                 ch_free( suffix.bv_val );
 408         }
 409
 410         /* alloc as many as required (plus 1 for objectClass) */
 411         a = attrs_alloc( 1 + 4 );
 412         if ( a == NULL ) {
 413                 rc = 1;
 414                 goto cleanup;
 415         }
 416
 417         a->a_desc = slap_schema.si_ad_objectClass;
 418         value_add_one( &a->a_vals, &oc_olmBDBDatabase->soc_cname );
 419         a->a_nvals = a->a_vals;
 420         next = a->a_next;
 421
 422         {
 423                 struct berval   bv = BER_BVC( "0" );
 424
 425                 next->a_desc = ad_olmBDBEntryCache;
 426                 value_add_one( &next->a_vals, &bv );
 427                 next->a_nvals = next->a_vals;
 428                 next = next->a_next;
 429
 430                 next->a_desc = ad_olmBDBDNCache;
 431                 value_add_one( &next->a_vals, &bv );
 432                 next->a_nvals = next->a_vals;
 433                 next = next->a_next;
 434
 435                 next->a_desc = ad_olmBDBIDLCache;
 436                 value_add_one( &next->a_vals, &bv );
 437                 next->a_nvals = next->a_vals;
 438                 next = next->a_next;
 439         }
 440
 441         {
 442                 struct berval   bv, nbv;
 443                 ber_len_t       pathlen = 0, len = 0;
 444                 char            path[ PATH_MAX ] = { '\0' };
 445                 char            *fname = bdb->bi_dbenv_home,
 446                                 *ptr;
 447
 448                 len = strlen( fname );
 449                 if ( fname[ 0 ] != '/' ) {
 450                         /* get full path name */
 451                         getcwd( path, sizeof( path ) );
 452                         pathlen = strlen( path );
 453
 454                         if ( fname[ 0 ] == '.' && fname[ 1 ] == '/' ) {
 455                                 fname += 2;
 456                                 len -= 2;
 457                         }
 458                 }
 459
 460                 bv.bv_len = pathlen + STRLENOF( "/" ) + len;
 461                 ptr = bv.bv_val = ch_malloc( bv.bv_len + STRLENOF( "/" ) + 1 );
 462                 if ( pathlen ) {
 463                         ptr = lutil_strncopy( ptr, path, pathlen );
 464                         ptr[ 0 ] = '/';
 465                         ptr++;
 466                 }
 467                 ptr = lutil_strncopy( ptr, fname, len );
 468                 if ( ptr[ -1 ] != '/' ) {
 469                         ptr[ 0 ] = '/';
 470                         ptr++;
 471                 }
 472                 ptr[ 0 ] = '\0';
 473
 474                 attr_normalize_one( ad_olmDbDirectory, &bv, &nbv, NULL );
 475
 476                 next->a_desc = ad_olmDbDirectory;
 477                 next->a_vals = ch_calloc( sizeof( struct berval ), 2 );
 478                 next->a_vals[ 0 ] = bv;
 479
 480                 if ( BER_BVISNULL( &nbv ) ) {
 481                         next->a_nvals = next->a_vals;
 482
 483                 } else {
 484                         next->a_nvals = ch_calloc( sizeof( struct berval ), 2 );
 485                         next->a_nvals[ 0 ] = nbv;
 486                 }
 487
 488                 next = next->a_next;
 489         }
 490
 491         cb = ch_calloc( sizeof( monitor_callback_t ), 1 );
 492         cb->mc_update = bdb_monitor_update;
 493 #if 0   /* uncomment if required */
 494         cb->mc_modify = bdb_monitor_modify;
 495 #endif
 496         cb->mc_free = bdb_monitor_free;
 497         cb->mc_private = (void *)bdb;
 498
 499         /* make sure the database is registered; then add monitor attributes */
 500         rc = mbe->register_database( be );
 501         if ( rc == 0 ) {
 502                 rc = mbe->register_entry_attrs( NULL, a, cb,
 503                         base, bdb->bi_monitor.bdm_scope, filter );
 504         }
 505
 506 cleanup:;
 507         if ( rc != 0 ) {
 508                 if ( cb != NULL ) {
 509                         ch_free( cb );
 510                         cb = NULL;
 511                 }
 512
 513                 if ( a != NULL ) {
 514                         attrs_free( a );
 515                         a = NULL;
 516                 }
 517
 518                 if ( !BER_BVISNULL( filter ) ) {
 519                         ch_free( filter->bv_val );
 520                         BER_BVZERO( filter );
 521                 }
 522         }
 523
 524         /* store for cleanup */
 525         bdb->bi_monitor.bdm_cb = (void *)cb;
 526
 527         /* we don't need to keep track of the attributes, because
 528          * bdb_monitor_free() takes care of everything */
 529         if ( a != NULL ) {
 530                 attrs_free( a );
 531         }
 532
 533         return rc;
 534 }
 535
 536 /*
 537  * call from within bdb_db_close()
 538  */
 539 int
 540 bdb_monitor_db_close( BackendDB *be )
 541 {
 542         struct bdb_info         *bdb = (struct bdb_info *) be->be_private;
 543
 544         if ( SLAP_GLUE_SUBORDINATE( be ) ) {
 545                 return 0;
 546         }
 547
 548         if ( !BER_BVISNULL( &bdb->bi_monitor.bdm_filter ) ) {
 549                 BackendInfo             *mi = backend_info( "monitor" );
 550                 monitor_extra_t         *mbe;
 551
 552                 if ( mi && &mi->bi_extra ) {
 553                         mbe = mi->bi_extra;
 554                         mbe->unregister_entry_callback( NULL,
 555                                 (monitor_callback_t *)bdb->bi_monitor.bdm_cb,
 556                                 &bdb->bi_monitor.bdm_nbase,
 557                                 bdb->bi_monitor.bdm_scope,
 558                                 &bdb->bi_monitor.bdm_filter );
 559                 }
 560
 561                 if ( !BER_BVISNULL( &bdb->bi_monitor.bdm_filter ) ) {
 562                         ch_free( bdb->bi_monitor.bdm_filter.bv_val );
 563                 }
 564
 565                 memset( &bdb->bi_monitor, 0, sizeof( bdb->bi_monitor ) );
 566         }
 567
 568         return 0;
 569 }
 570
 571 /*
 572  * call from within bdb_db_destroy()
 573  */
 574 int
 575 bdb_monitor_db_destroy( BackendDB *be )
 576 {
 577         if ( SLAP_GLUE_SUBORDINATE( be ) ) {
 578                 return 0;
 579         }
 580
 581 #ifdef BDB_MONITOR_IDX
 582         {
 583                 struct bdb_info         *bdb = (struct bdb_info *) be->be_private;
 584
 585                 /* TODO: free tree */
 586                 ldap_pvt_thread_mutex_destroy( &bdb->bi_idx_mutex );
 587                 avl_free( bdb->bi_idx, ch_free );
 588         }
 589 #endif /* BDB_MONITOR_IDX */
 590
 591         return 0;
 592 }
 593
 594 #ifdef BDB_MONITOR_IDX
 595
 596 #define BDB_MONITOR_IDX_TYPES   (4)
 597
 598 typedef struct monitor_idx_t monitor_idx_t;
 599
 600 struct monitor_idx_t {
 601         AttributeDescription    *idx_ad;
 602         unsigned long           idx_count[BDB_MONITOR_IDX_TYPES];
 603 };
 604
 605 static int
 606 bdb_monitor_bitmask2key( slap_mask_t bitmask )
 607 {
 608         int     key;
 609
 610         for ( key = 0; key < 8*sizeof(slap_mask_t) && !( bitmask & 0x1U ); key++ ) {
 611                 bitmask >>= 1;
 612         }
 613
 614         return key;
 615 }
 616
 617 static struct berval idxbv[] = {
 618         BER_BVC( "present=" ),
 619         BER_BVC( "equality=" ),
 620         BER_BVC( "approx=" ),
 621         BER_BVC( "substr=" ),
 622         BER_BVNULL
 623 };
 624
 625 static ber_len_t
 626 bdb_monitor_idx2len( monitor_idx_t *idx )
 627 {
 628         int             i;
 629         ber_len_t       len = 0;
 630
 631         for ( i = 0; i < BDB_MONITOR_IDX_TYPES; i++ ) {
 632                 if ( idx->idx_count[ i ] != 0 ) {
 633                         len += idxbv[i].bv_len;
 634                 }
 635         }
 636
 637         return len;
 638 }
 639
 640 static int
 641 monitor_idx_cmp( const void *p1, const void *p2 )
 642 {
 643         const monitor_idx_t     *idx1 = (const monitor_idx_t *)p1;
 644         const monitor_idx_t     *idx2 = (const monitor_idx_t *)p2;
 645
 646         return SLAP_PTRCMP( idx1->idx_ad, idx2->idx_ad );
 647 }
 648
 649 static int
 650 monitor_idx_dup( void *p1, void *p2 )
 651 {
 652         monitor_idx_t   *idx1 = (monitor_idx_t *)p1;
 653         monitor_idx_t   *idx2 = (monitor_idx_t *)p2;
 654
 655         return SLAP_PTRCMP( idx1->idx_ad, idx2->idx_ad ) == 0 ? -1 : 0;
 656 }
 657
 658 int
 659 bdb_monitor_idx_add(
 660         struct bdb_info         *bdb,
 661         AttributeDescription    *desc,
 662         slap_mask_t             type )
 663 {
 664         monitor_idx_t           idx_dummy = { 0 },
 665                                 *idx;
 666         int                     rc = 0, key;
 667
 668         idx_dummy.idx_ad = desc;
 669         key = bdb_monitor_bitmask2key( type ) - 1;
 670         if ( key >= BDB_MONITOR_IDX_TYPES ) {
 671                 /* invalid index type */
 672                 return -1;
 673         }
 674
 675         ldap_pvt_thread_mutex_lock( &bdb->bi_idx_mutex );
 676
 677         idx = (monitor_idx_t *)avl_find( bdb->bi_idx,
 678                 (caddr_t)&idx_dummy, monitor_idx_cmp );
 679         if ( idx == NULL ) {
 680                 idx = (monitor_idx_t *)ch_calloc( sizeof( monitor_idx_t ), 1 );
 681                 idx->idx_ad = desc;
 682                 idx->idx_count[ key ] = 1;
 683
 684                 switch ( avl_insert( &bdb->bi_idx, (caddr_t)idx,
 685                         monitor_idx_cmp, monitor_idx_dup ) )
 686                 {
 687                 case 0:
 688                         break;
 689
 690                 default:
 691                         ch_free( idx );
 692                         rc = -1;
 693                 }
 694
 695         } else {
 696                 idx->idx_count[ key ]++;
 697         }
 698
 699         ldap_pvt_thread_mutex_unlock( &bdb->bi_idx_mutex );
 700
 701         return rc;
 702 }
 703
 704 static int
 705 bdb_monitor_idx_apply( void *v_idx, void *v_valp )
 706 {
 707         monitor_idx_t   *idx = (monitor_idx_t *)v_idx;
 708         BerVarray       *valp = (BerVarray *)v_valp;
 709
 710         struct berval   bv;
 711         char            *ptr;
 712         char            count_buf[ BDB_MONITOR_IDX_TYPES ][ SLAP_TEXT_BUFLEN ];
 713         ber_len_t       count_len[ BDB_MONITOR_IDX_TYPES ],
 714                         idx_len;
 715         int             i, num = 0;
 716
 717         idx_len = bdb_monitor_idx2len( idx );
 718
 719         bv.bv_len = 0;
 720         for ( i = 0; i < BDB_MONITOR_IDX_TYPES; i++ ) {
 721                 if ( idx->idx_count[ i ] == 0 ) {
 722                         continue;
 723                 }
 724
 725                 count_len[ i ] = snprintf( count_buf[ i ],
 726                         sizeof( count_buf[ i ] ), "%lu", idx->idx_count[ i ] );
 727                 bv.bv_len += count_len[ i ];
 728                 num++;
 729         }
 730
 731         bv.bv_len += idx->idx_ad->ad_cname.bv_len
 732                 + num
 733                 + idx_len;
 734         ptr = bv.bv_val = ch_malloc( bv.bv_len + 1 );
 735         ptr = lutil_strcopy( ptr, idx->idx_ad->ad_cname.bv_val );
 736         for ( i = 0; i < BDB_MONITOR_IDX_TYPES; i++ ) {
 737                 if ( idx->idx_count[ i ] == 0 ) {
 738                         continue;
 739                 }
 740
 741                 ptr[ 0 ] = '#';
 742                 ++ptr;
 743                 ptr = lutil_strcopy( ptr, idxbv[ i ].bv_val );
 744                 ptr = lutil_strcopy( ptr, count_buf[ i ] );
 745         }
 746
 747         ber_bvarray_add( valp, &bv );
 748
 749         return 0;
 750 }
 751
 752 static int
 753 bdb_monitor_idx_entry_add(
 754         struct bdb_info *bdb,
 755         Entry           *e )
 756 {
 757         BerVarray       vals = NULL;
 758         Attribute       *a;
 759
 760         a = attr_find( e->e_attrs, ad_olmBDBNotIndexed );
 761
 762         ldap_pvt_thread_mutex_lock( &bdb->bi_idx_mutex );
 763
 764         avl_apply( bdb->bi_idx, bdb_monitor_idx_apply,
 765                 &vals, -1, AVL_INORDER );
 766
 767         ldap_pvt_thread_mutex_unlock( &bdb->bi_idx_mutex );
 768
 769         if ( vals != NULL ) {
 770                 if ( a != NULL ) {
 771                         assert( a->a_nvals == a->a_vals );
 772
 773                         ber_bvarray_free( a->a_vals );
 774
 775                 } else {
 776                         Attribute       **ap;
 777
 778                         for ( ap = &e->e_attrs; *ap != NULL; ap = &(*ap)->a_next )
 779                                 ;
 780                         *ap = attr_alloc( ad_olmBDBNotIndexed );
 781                         a = *ap;
 782                 }
 783                 a->a_vals = vals;
 784                 a->a_nvals = a->a_vals;
 785         }
 786
 787         return 0;
 788 }
 789
 790 #endif /* BDB_MONITOR_IDX */