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

   1 /* dn2id.c - routines to deal with the dn2id index */
   2 /* $OpenLDAP$ */
   3 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
   4  *
   5  * Copyright 2000-2008 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
  22 #include "back-bdb.h"
  23 #include "idl.h"
  24 #include "lutil.h"
  25
  26 #define bdb_dn2id_lock                                  BDB_SYMBOL(dn2id_lock)
  27
  28 static int
  29 bdb_dn2id_lock( struct bdb_info *bdb, struct berval *dn,
  30         int rw, DB_TXN *txn, DB_LOCK *lock )
  31 {
  32         int       rc;
  33         DBT       lockobj;
  34         int       db_rw;
  35
  36         if (!txn)
  37                 return 0;
  38
  39         if (rw)
  40                 db_rw = DB_LOCK_WRITE;
  41         else
  42                 db_rw = DB_LOCK_READ;
  43
  44         lockobj.data = dn->bv_val;
  45         lockobj.size = dn->bv_len;
  46
  47         rc = LOCK_GET(bdb->bi_dbenv, TXN_ID(txn), DB_LOCK_NOWAIT,
  48                                         &lockobj, db_rw, lock);
  49         return rc;
  50 }
  51
  52 #ifndef BDB_HIER
  53 int
  54 bdb_dn2id_add(
  55         Operation *op,
  56         DB_TXN *txn,
  57         EntryInfo *eip,
  58         Entry           *e )
  59 {
  60         struct bdb_info *bdb = (struct bdb_info *) op->o_bd->be_private;
  61         DB *db = bdb->bi_dn2id->bdi_db;
  62         int             rc;
  63         DBT             key, data;
  64         ID              nid;
  65         char            *buf;
  66         struct berval   ptr, pdn;
  67
  68         Debug( LDAP_DEBUG_TRACE, "=> bdb_dn2id_add 0x%lx: \"%s\"\n",
  69                 e->e_id, e->e_ndn, 0 );
  70         assert( e->e_id != NOID );
  71
  72         DBTzero( &key );
  73         key.size = e->e_nname.bv_len + 2;
  74         key.ulen = key.size;
  75         key.flags = DB_DBT_USERMEM;
  76         buf = op->o_tmpalloc( key.size, op->o_tmpmemctx );
  77         key.data = buf;
  78         buf[0] = DN_BASE_PREFIX;
  79         ptr.bv_val = buf + 1;
  80         ptr.bv_len = e->e_nname.bv_len;
  81         AC_MEMCPY( ptr.bv_val, e->e_nname.bv_val, e->e_nname.bv_len );
  82         ptr.bv_val[ptr.bv_len] = '\0';
  83
  84         DBTzero( &data );
  85         data.data = &nid;
  86         data.size = sizeof( nid );
  87         BDB_ID2DISK( e->e_id, &nid );
  88
  89         /* store it -- don't override */
  90         rc = db->put( db, txn, &key, &data, DB_NOOVERWRITE );
  91         if( rc != 0 ) {
  92                 char buf[ SLAP_TEXT_BUFLEN ];
  93                 snprintf( buf, sizeof( buf ), "%s => bdb_dn2id_add dn=\"%s\" ID=0x%lx",
  94                         op->o_log_prefix, e->e_name.bv_val, e->e_id );
  95                 Debug( LDAP_DEBUG_ANY, "%s: put failed: %s %d\n",
  96                         buf, db_strerror(rc), rc );
  97                 goto done;
  98         }
  99
 100 #ifndef BDB_MULTIPLE_SUFFIXES
 101         if( !be_issuffix( op->o_bd, &ptr ))
 102 #endif
 103         {
 104                 buf[0] = DN_SUBTREE_PREFIX;
 105                 rc = db->put( db, txn, &key, &data, DB_NOOVERWRITE );
 106                 if( rc != 0 ) {
 107                         Debug( LDAP_DEBUG_ANY,
 108                         "=> bdb_dn2id_add 0x%lx: subtree (%s) put failed: %d\n",
 109                         e->e_id, ptr.bv_val, rc );
 110                         goto done;
 111                 }
 112
 113 #ifdef BDB_MULTIPLE_SUFFIXES
 114         if( !be_issuffix( op->o_bd, &ptr ))
 115 #endif
 116         {
 117                 dnParent( &ptr, &pdn );
 118
 119                 key.size = pdn.bv_len + 2;
 120                 key.ulen = key.size;
 121                 pdn.bv_val[-1] = DN_ONE_PREFIX;
 122                 key.data = pdn.bv_val-1;
 123                 ptr = pdn;
 124
 125                 rc = bdb_idl_insert_key( op->o_bd, db, txn, &key, e->e_id );
 126
 127                 if( rc != 0 ) {
 128                         Debug( LDAP_DEBUG_ANY,
 129                                 "=> bdb_dn2id_add 0x%lx: parent (%s) insert failed: %d\n",
 130                                         e->e_id, ptr.bv_val, rc );
 131                         goto done;
 132                 }
 133         }
 134
 135 #ifndef BDB_MULTIPLE_SUFFIXES
 136         while( !be_issuffix( op->o_bd, &ptr ))
 137 #else
 138         for (;;)
 139 #endif
 140         {
 141                 ptr.bv_val[-1] = DN_SUBTREE_PREFIX;
 142
 143                 rc = bdb_idl_insert_key( op->o_bd, db, txn, &key, e->e_id );
 144
 145                 if( rc != 0 ) {
 146                         Debug( LDAP_DEBUG_ANY,
 147                                 "=> bdb_dn2id_add 0x%lx: subtree (%s) insert failed: %d\n",
 148                                         e->e_id, ptr.bv_val, rc );
 149                         break;
 150                 }
 151 #ifdef BDB_MULTIPLE_SUFFIXES
 152                 if( be_issuffix( op->o_bd, &ptr )) break;
 153 #endif
 154                 dnParent( &ptr, &pdn );
 155
 156                 key.size = pdn.bv_len + 2;
 157                 key.ulen = key.size;
 158                 key.data = pdn.bv_val - 1;
 159                 ptr = pdn;
 160         }
 161         }
 162
 163 done:
 164         op->o_tmpfree( buf, op->o_tmpmemctx );
 165         Debug( LDAP_DEBUG_TRACE, "<= bdb_dn2id_add 0x%lx: %d\n", e->e_id, rc, 0 );
 166         return rc;
 167 }
 168
 169 int
 170 bdb_dn2id_delete(
 171         Operation *op,
 172         DB_TXN *txn,
 173         EntryInfo       *eip,
 174         Entry           *e )
 175 {
 176         struct bdb_info *bdb = (struct bdb_info *) op->o_bd->be_private;
 177         DB *db = bdb->bi_dn2id->bdi_db;
 178         char            *buf;
 179         DBT             key;
 180         DB_LOCK lock;
 181         struct berval   pdn, ptr;
 182         int             rc;
 183
 184         Debug( LDAP_DEBUG_TRACE, "=> bdb_dn2id_delete 0x%lx: \"%s\"\n",
 185                 e->e_id, e->e_ndn, 0 );
 186
 187         DBTzero( &key );
 188         key.size = e->e_nname.bv_len + 2;
 189         buf = op->o_tmpalloc( key.size, op->o_tmpmemctx );
 190         key.data = buf;
 191         key.flags = DB_DBT_USERMEM;
 192         buf[0] = DN_BASE_PREFIX;
 193         ptr.bv_val = buf+1;
 194         ptr.bv_len = e->e_nname.bv_len;
 195         AC_MEMCPY( ptr.bv_val, e->e_nname.bv_val, e->e_nname.bv_len );
 196         ptr.bv_val[ptr.bv_len] = '\0';
 197
 198         /* We hold this lock until the TXN completes */
 199         rc = bdb_dn2id_lock( bdb, &e->e_nname, 1, txn, &lock );
 200         if ( rc ) goto done;
 201
 202         /* delete it */
 203         rc = db->del( db, txn, &key, 0 );
 204         if( rc != 0 ) {
 205                 Debug( LDAP_DEBUG_ANY, "=> bdb_dn2id_delete 0x%lx: delete failed: %s %d\n",
 206                         e->e_id, db_strerror(rc), rc );
 207                 goto done;
 208         }
 209
 210 #ifndef BDB_MULTIPLE_SUFFIXES
 211         if( !be_issuffix( op->o_bd, &ptr ))
 212 #endif
 213         {
 214                 buf[0] = DN_SUBTREE_PREFIX;
 215                 rc = bdb_idl_delete_key( op->o_bd, db, txn, &key, e->e_id );
 216                 if( rc != 0 ) {
 217                         Debug( LDAP_DEBUG_ANY,
 218                         "=> bdb_dn2id_delete 0x%lx: subtree (%s) delete failed: %d\n",
 219                         e->e_id, ptr.bv_val, rc );
 220                         goto done;
 221                 }
 222
 223 #ifdef BDB_MULTIPLE_SUFFIXES
 224         if( !be_issuffix( op->o_bd, &ptr ))
 225 #endif
 226         {
 227                 dnParent( &ptr, &pdn );
 228
 229                 key.size = pdn.bv_len + 2;
 230                 key.ulen = key.size;
 231                 pdn.bv_val[-1] = DN_ONE_PREFIX;
 232                 key.data = pdn.bv_val - 1;
 233                 ptr = pdn;
 234
 235                 rc = bdb_idl_delete_key( op->o_bd, db, txn, &key, e->e_id );
 236
 237                 if( rc != 0 ) {
 238                         Debug( LDAP_DEBUG_ANY,
 239                                 "=> bdb_dn2id_delete 0x%lx: parent (%s) delete failed: %d\n",
 240                                 e->e_id, ptr.bv_val, rc );
 241                         goto done;
 242                 }
 243         }
 244
 245 #ifndef BDB_MULTIPLE_SUFFIXES
 246         while( !be_issuffix( op->o_bd, &ptr ))
 247 #else
 248         for (;;)
 249 #endif
 250         {
 251                 ptr.bv_val[-1] = DN_SUBTREE_PREFIX;
 252
 253                 rc = bdb_idl_delete_key( op->o_bd, db, txn, &key, e->e_id );
 254                 if( rc != 0 ) {
 255                         Debug( LDAP_DEBUG_ANY,
 256                                 "=> bdb_dn2id_delete 0x%lx: subtree (%s) delete failed: %d\n",
 257                                 e->e_id, ptr.bv_val, rc );
 258                         goto done;
 259                 }
 260 #ifdef BDB_MULTIPLE_SUFFIXES
 261                 if( be_issuffix( op->o_bd, &ptr )) break;
 262 #endif
 263                 dnParent( &ptr, &pdn );
 264
 265                 key.size = pdn.bv_len + 2;
 266                 key.ulen = key.size;
 267                 key.data = pdn.bv_val - 1;
 268                 ptr = pdn;
 269         }
 270         }
 271
 272 done:
 273         op->o_tmpfree( buf, op->o_tmpmemctx );
 274         Debug( LDAP_DEBUG_TRACE, "<= bdb_dn2id_delete 0x%lx: %d\n", e->e_id, rc, 0 );
 275         return rc;
 276 }
 277
 278 int
 279 bdb_dn2id(
 280         Operation *op,
 281         struct berval   *dn,
 282         EntryInfo *ei,
 283         DB_TXN *txn,
 284         DB_LOCK *lock )
 285 {
 286         struct bdb_info *bdb = (struct bdb_info *) op->o_bd->be_private;
 287         DB *db = bdb->bi_dn2id->bdi_db;
 288         DBC     *cursor;
 289         int             rc;
 290         DBT             key, data;
 291         ID              nid;
 292
 293         Debug( LDAP_DEBUG_TRACE, "=> bdb_dn2id(\"%s\")\n", dn->bv_val, 0, 0 );
 294
 295         DBTzero( &key );
 296         key.size = dn->bv_len + 2;
 297         key.data = op->o_tmpalloc( key.size, op->o_tmpmemctx );
 298         ((char *)key.data)[0] = DN_BASE_PREFIX;
 299         AC_MEMCPY( &((char *)key.data)[1], dn->bv_val, key.size - 1 );
 300
 301         /* store the ID */
 302         DBTzero( &data );
 303         data.data = &nid;
 304         data.ulen = sizeof(ID);
 305         data.flags = DB_DBT_USERMEM;
 306
 307         rc = db->cursor( db, txn, &cursor, bdb->bi_db_opflags );
 308         if ( rc ) goto func_leave;
 309
 310         rc = bdb_dn2id_lock( bdb, dn, 0, txn, lock );
 311         if ( rc ) goto nolock;
 312
 313         /* fetch it */
 314         rc = cursor->c_get( cursor, &key, &data, DB_SET );
 315
 316 nolock:
 317         cursor->c_close( cursor );
 318 func_leave:
 319
 320         if( rc != 0 ) {
 321                 Debug( LDAP_DEBUG_TRACE, "<= bdb_dn2id: get failed: %s (%d)\n",
 322                         db_strerror( rc ), rc, 0 );
 323         } else {
 324                 BDB_DISK2ID( &nid, &ei->bei_id );
 325                 Debug( LDAP_DEBUG_TRACE, "<= bdb_dn2id: got id=0x%lx\n",
 326                         ei->bei_id, 0, 0 );
 327         }
 328         op->o_tmpfree( key.data, op->o_tmpmemctx );
 329         return rc;
 330 }
 331
 332 int
 333 bdb_dn2id_children(
 334         Operation *op,
 335         DB_TXN *txn,
 336         Entry *e )
 337 {
 338         DBT             key, data;
 339         struct bdb_info *bdb = (struct bdb_info *) op->o_bd->be_private;
 340         DB *db = bdb->bi_dn2id->bdi_db;
 341         ID              id;
 342         int             rc;
 343
 344         Debug( LDAP_DEBUG_TRACE, "=> bdb_dn2id_children(\"%s\")\n",
 345                 e->e_nname.bv_val, 0, 0 );
 346         DBTzero( &key );
 347         key.size = e->e_nname.bv_len + 2;
 348         key.data = op->o_tmpalloc( key.size, op->o_tmpmemctx );
 349         ((char *)key.data)[0] = DN_ONE_PREFIX;
 350         AC_MEMCPY( &((char *)key.data)[1], e->e_nname.bv_val, key.size - 1 );
 351
 352         if ( bdb->bi_idl_cache_size ) {
 353                 rc = bdb_idl_cache_get( bdb, db, &key, NULL );
 354                 if ( rc != LDAP_NO_SUCH_OBJECT ) {
 355                         op->o_tmpfree( key.data, op->o_tmpmemctx );
 356                         return rc;
 357                 }
 358         }
 359         /* we actually could do a empty get... */
 360         DBTzero( &data );
 361         data.data = &id;
 362         data.ulen = sizeof(id);
 363         data.flags = DB_DBT_USERMEM;
 364         data.doff = 0;
 365         data.dlen = sizeof(id);
 366
 367         rc = db->get( db, txn, &key, &data, bdb->bi_db_opflags );
 368         op->o_tmpfree( key.data, op->o_tmpmemctx );
 369
 370         Debug( LDAP_DEBUG_TRACE, "<= bdb_dn2id_children(\"%s\"): %s (%d)\n",
 371                 e->e_nname.bv_val,
 372                 rc == 0 ? "" : ( rc == DB_NOTFOUND ? "no " :
 373                         db_strerror(rc) ), rc );
 374
 375         return rc;
 376 }
 377
 378 int
 379 bdb_dn2idl(
 380         Operation *op,
 381         DB_TXN *txn,
 382         struct berval *ndn,
 383         EntryInfo *ei,
 384         ID *ids,
 385         ID *stack )
 386 {
 387         int             rc;
 388         DBT             key;
 389         struct bdb_info *bdb = (struct bdb_info *) op->o_bd->be_private;
 390         DB *db = bdb->bi_dn2id->bdi_db;
 391         int prefix = ( op->ors_scope == LDAP_SCOPE_ONELEVEL )
 392                 ? DN_ONE_PREFIX : DN_SUBTREE_PREFIX;
 393
 394         Debug( LDAP_DEBUG_TRACE, "=> bdb_dn2idl(\"%s\")\n",
 395                 ndn->bv_val, 0, 0 );
 396
 397 #ifndef BDB_MULTIPLE_SUFFIXES
 398         if ( prefix == DN_SUBTREE_PREFIX
 399                 && ( ei->bei_id == 0 ||
 400                 ( ei->bei_parent->bei_id == 0 && op->o_bd->be_suffix[0].bv_len ))) {
 401                 BDB_IDL_ALL(bdb, ids);
 402                 return 0;
 403         }
 404 #endif
 405
 406         DBTzero( &key );
 407         key.size = ndn->bv_len + 2;
 408         key.ulen = key.size;
 409         key.flags = DB_DBT_USERMEM;
 410         key.data = op->o_tmpalloc( key.size, op->o_tmpmemctx );
 411         ((char *)key.data)[0] = prefix;
 412         AC_MEMCPY( &((char *)key.data)[1], ndn->bv_val, key.size - 1 );
 413
 414         BDB_IDL_ZERO( ids );
 415         rc = bdb_idl_fetch_key( op->o_bd, db, txn, &key, ids, NULL, 0 );
 416
 417         if( rc != 0 ) {
 418                 Debug( LDAP_DEBUG_TRACE,
 419                         "<= bdb_dn2idl: get failed: %s (%d)\n",
 420                         db_strerror( rc ), rc, 0 );
 421
 422         } else {
 423                 Debug( LDAP_DEBUG_TRACE,
 424                         "<= bdb_dn2idl: id=%ld first=%ld last=%ld\n",
 425                         (long) ids[0],
 426                         (long) BDB_IDL_FIRST( ids ), (long) BDB_IDL_LAST( ids ) );
 427         }
 428
 429         op->o_tmpfree( key.data, op->o_tmpmemctx );
 430         return rc;
 431 }
 432
 433 #else   /* BDB_HIER */
 434 /* Management routines for a hierarchically structured database.
 435  *
 436  * Instead of a ldbm-style dn2id database, we use a hierarchical one. Each
 437  * entry in this database is a struct diskNode, keyed by entryID and with
 438  * the data containing the RDN and entryID of the node's children. We use
 439  * a B-Tree with sorted duplicates to store all the children of a node under
 440  * the same key. Also, the first item under the key contains the entry's own
 441  * rdn and the ID of the node's parent, to allow bottom-up tree traversal as
 442  * well as top-down. To keep this info first in the list, the high bit of all
 443  * subsequent nrdnlen's is always set. This means we can only accomodate
 444  * RDNs up to length 32767, but that's fine since full DNs are already
 445  * restricted to 8192.
 446  *
 447  * The diskNode is a variable length structure. This definition is not
 448  * directly usable for in-memory manipulation.
 449  */
 450 typedef struct diskNode {
 451         unsigned char nrdnlen[2];
 452         char nrdn[1];
 453         char rdn[1];                        /* variable placement */
 454         unsigned char entryID[sizeof(ID)];  /* variable placement */
 455 } diskNode;
 456
 457 /* Sort function for the sorted duplicate data items of a dn2id key.
 458  * Sorts based on normalized RDN, in length order.
 459  */
 460 int
 461 hdb_dup_compare(
 462         DB *db,
 463         const DBT *usrkey,
 464         const DBT *curkey
 465 )
 466 {
 467         diskNode *un, *cn;
 468         int rc;
 469
 470         un = (diskNode *)usrkey->data;
 471         cn = (diskNode *)curkey->data;
 472
 473         /* data is not aligned, cannot compare directly */
 474         rc = un->nrdnlen[0] - cn->nrdnlen[0];
 475         if ( rc ) return rc;
 476         rc = un->nrdnlen[1] - cn->nrdnlen[1];
 477         if ( rc ) return rc;
 478
 479         return strcmp( un->nrdn, cn->nrdn );
 480 }
 481
 482 /* This function constructs a full DN for a given entry.
 483  */
 484 int hdb_fix_dn(
 485         Entry *e,
 486         int checkit )
 487 {
 488         EntryInfo *ei;
 489         int rlen = 0, nrlen = 0;
 490         char *ptr, *nptr;
 491         int max = 0;
 492
 493         if ( !e->e_id )
 494                 return 0;
 495
 496         /* count length of all DN components */
 497         for ( ei = BEI(e); ei && ei->bei_id; ei=ei->bei_parent ) {
 498                 rlen += ei->bei_rdn.bv_len + 1;
 499                 nrlen += ei->bei_nrdn.bv_len + 1;
 500                 if (ei->bei_modrdns > max) max = ei->bei_modrdns;
 501         }
 502
 503         /* See if the entry DN was invalidated by a subtree rename */
 504         if ( checkit ) {
 505                 if ( BEI(e)->bei_modrdns >= max ) {
 506                         return 0;
 507                 }
 508                 /* We found a mismatch, tell the caller to lock it */
 509                 if ( checkit == 1 ) {
 510                         return 1;
 511                 }
 512                 /* checkit == 2. do the fix. */
 513                 free( e->e_name.bv_val );
 514                 free( e->e_nname.bv_val );
 515         }
 516
 517         e->e_name.bv_len = rlen - 1;
 518         e->e_nname.bv_len = nrlen - 1;
 519         e->e_name.bv_val = ch_malloc(rlen);
 520         e->e_nname.bv_val = ch_malloc(nrlen);
 521         ptr = e->e_name.bv_val;
 522         nptr = e->e_nname.bv_val;
 523         for ( ei = BEI(e); ei && ei->bei_id; ei=ei->bei_parent ) {
 524                 ptr = lutil_strcopy(ptr, ei->bei_rdn.bv_val);
 525                 nptr = lutil_strcopy(nptr, ei->bei_nrdn.bv_val);
 526                 if ( ei->bei_parent ) {
 527                         *ptr++ = ',';
 528                         *nptr++ = ',';
 529                 }
 530         }
 531         BEI(e)->bei_modrdns = max;
 532         ptr[-1] = '\0';
 533         nptr[-1] = '\0';
 534
 535         return 0;
 536 }
 537
 538 /* We add two elements to the DN2ID database - a data item under the parent's
 539  * entryID containing the child's RDN and entryID, and an item under the
 540  * child's entryID containing the parent's entryID.
 541  */
 542 int
 543 hdb_dn2id_add(
 544         Operation       *op,
 545         DB_TXN *txn,
 546         EntryInfo       *eip,
 547         Entry           *e )
 548 {
 549         struct bdb_info *bdb = (struct bdb_info *) op->o_bd->be_private;
 550         DB *db = bdb->bi_dn2id->bdi_db;
 551         DBT             key, data;
 552         ID              nid;
 553         int             rc, rlen, nrlen;
 554         diskNode *d;
 555         char *ptr;
 556
 557         Debug( LDAP_DEBUG_TRACE, "=> hdb_dn2id_add 0x%lx: \"%s\"\n",
 558                 e->e_id, e->e_ndn, 0 );
 559
 560         nrlen = dn_rdnlen( op->o_bd, &e->e_nname );
 561         if (nrlen) {
 562                 rlen = dn_rdnlen( op->o_bd, &e->e_name );
 563         } else {
 564                 nrlen = e->e_nname.bv_len;
 565                 rlen = e->e_name.bv_len;
 566         }
 567
 568         d = op->o_tmpalloc(sizeof(diskNode) + rlen + nrlen, op->o_tmpmemctx);
 569         d->nrdnlen[1] = nrlen & 0xff;
 570         d->nrdnlen[0] = (nrlen >> 8) | 0x80;
 571         ptr = lutil_strncopy( d->nrdn, e->e_nname.bv_val, nrlen );
 572         *ptr++ = '\0';
 573         ptr = lutil_strncopy( ptr, e->e_name.bv_val, rlen );
 574         *ptr++ = '\0';
 575         BDB_ID2DISK( e->e_id, ptr );
 576
 577         DBTzero(&key);
 578         DBTzero(&data);
 579         key.size = sizeof(ID);
 580         key.flags = DB_DBT_USERMEM;
 581         BDB_ID2DISK( eip->bei_id, &nid );
 582
 583         key.data = &nid;
 584
 585         /* Need to make dummy root node once. Subsequent attempts
 586          * will fail harmlessly.
 587          */
 588         if ( eip->bei_id == 0 ) {
 589                 diskNode dummy = {{0, 0}, "", "", ""};
 590                 data.data = &dummy;
 591                 data.size = sizeof(diskNode);
 592                 data.flags = DB_DBT_USERMEM;
 593
 594                 db->put( db, txn, &key, &data, DB_NODUPDATA );
 595         }
 596
 597         data.data = d;
 598         data.size = sizeof(diskNode) + rlen + nrlen;
 599         data.flags = DB_DBT_USERMEM;
 600
 601         rc = db->put( db, txn, &key, &data, DB_NODUPDATA );
 602
 603         if (rc == 0) {
 604                 BDB_ID2DISK( e->e_id, &nid );
 605                 BDB_ID2DISK( eip->bei_id, ptr );
 606                 d->nrdnlen[0] ^= 0x80;
 607
 608                 rc = db->put( db, txn, &key, &data, DB_NODUPDATA );
 609         }
 610
 611         /* Update all parents' IDL cache entries */
 612         if ( rc == 0 && bdb->bi_idl_cache_size ) {
 613                 ID tmp[2];
 614                 char *ptr = ((char *)&tmp[1])-1;
 615                 key.data = ptr;
 616                 key.size = sizeof(ID)+1;
 617                 tmp[1] = eip->bei_id;
 618                 *ptr = DN_ONE_PREFIX;
 619                 bdb_idl_cache_add_id( bdb, db, &key, e->e_id );
 620                 if ( eip->bei_parent ) {
 621                         *ptr = DN_SUBTREE_PREFIX;
 622                         for (; eip && eip->bei_parent->bei_id; eip = eip->bei_parent) {
 623                                 tmp[1] = eip->bei_id;
 624                                 bdb_idl_cache_add_id( bdb, db, &key, e->e_id );
 625                         }
 626                 }
 627         }
 628
 629         op->o_tmpfree( d, op->o_tmpmemctx );
 630         Debug( LDAP_DEBUG_TRACE, "<= hdb_dn2id_add 0x%lx: %d\n", e->e_id, rc, 0 );
 631
 632         return rc;
 633 }
 634
 635 int
 636 hdb_dn2id_delete(
 637         Operation       *op,
 638         DB_TXN *txn,
 639         EntryInfo       *eip,
 640         Entry   *e )
 641 {
 642         struct bdb_info *bdb = (struct bdb_info *) op->o_bd->be_private;
 643         DB *db = bdb->bi_dn2id->bdi_db;
 644         DBT             key, data;
 645         DBC     *cursor;
 646         diskNode *d;
 647         int rc;
 648         ID      nid;
 649         unsigned char dlen[2];
 650         DB_LOCK lock;
 651
 652         Debug( LDAP_DEBUG_TRACE, "=> hdb_dn2id_delete 0x%lx: \"%s\"\n",
 653                 e->e_id, e->e_ndn, 0 );
 654
 655         DBTzero(&key);
 656         key.size = sizeof(ID);
 657         key.ulen = key.size;
 658         key.flags = DB_DBT_USERMEM;
 659         BDB_ID2DISK( eip->bei_id, &nid );
 660
 661         DBTzero(&data);
 662         data.size = sizeof(diskNode) + BEI(e)->bei_nrdn.bv_len - sizeof(ID) - 1;
 663         data.ulen = data.size;
 664         data.dlen = data.size;
 665         data.flags = DB_DBT_USERMEM | DB_DBT_PARTIAL;
 666
 667         key.data = &nid;
 668
 669         d = op->o_tmpalloc( data.size, op->o_tmpmemctx );
 670         d->nrdnlen[1] = BEI(e)->bei_nrdn.bv_len & 0xff;
 671         d->nrdnlen[0] = (BEI(e)->bei_nrdn.bv_len >> 8) | 0x80;
 672         dlen[0] = d->nrdnlen[0];
 673         dlen[1] = d->nrdnlen[1];
 674         strcpy( d->nrdn, BEI(e)->bei_nrdn.bv_val );
 675         data.data = d;
 676
 677         rc = db->cursor( db, txn, &cursor, bdb->bi_db_opflags );
 678         if ( rc ) goto func_leave;
 679
 680         /* We hold this lock until the TXN completes */
 681         rc = bdb_dn2id_lock( bdb, &e->e_nname, 1, txn, &lock );
 682         if ( rc ) goto nolock;
 683
 684         /* Delete our ID from the parent's list */
 685         rc = cursor->c_get( cursor, &key, &data, DB_GET_BOTH_RANGE );
 686         if ( rc == 0 ) {
 687                 if ( dlen[1] == d->nrdnlen[1] && dlen[0] == d->nrdnlen[0] &&
 688                         !strcmp( d->nrdn, BEI(e)->bei_nrdn.bv_val ))
 689                         rc = cursor->c_del( cursor, 0 );
 690                 else
 691                         rc = DB_NOTFOUND;
 692         }
 693
 694         /* Delete our ID from the tree. With sorted duplicates, this
 695          * will leave any child nodes still hanging around. This is OK
 696          * for modrdn, which will add our info back in later.
 697          */
 698         if ( rc == 0 ) {
 699                 BDB_ID2DISK( e->e_id, &nid );
 700                 rc = cursor->c_get( cursor, &key, &data, DB_SET );
 701                 if ( rc == 0 )
 702                         rc = cursor->c_del( cursor, 0 );
 703         }
 704
 705 nolock:
 706         cursor->c_close( cursor );
 707 func_leave:
 708         op->o_tmpfree( d, op->o_tmpmemctx );
 709
 710         /* Delete IDL cache entries */
 711         if ( rc == 0 && bdb->bi_idl_cache_size ) {
 712                 ID tmp[2];
 713                 char *ptr = ((char *)&tmp[1])-1;
 714                 key.data = ptr;
 715                 key.size = sizeof(ID)+1;
 716                 tmp[1] = eip->bei_id;
 717                 *ptr = DN_ONE_PREFIX;
 718                 bdb_idl_cache_del_id( bdb, db, &key, e->e_id );
 719                 if ( eip ->bei_parent ) {
 720                         *ptr = DN_SUBTREE_PREFIX;
 721                         for (; eip && eip->bei_parent->bei_id; eip = eip->bei_parent) {
 722                                 tmp[1] = eip->bei_id;
 723                                 bdb_idl_cache_del_id( bdb, db, &key, e->e_id );
 724                         }
 725                 }
 726         }
 727         Debug( LDAP_DEBUG_TRACE, "<= hdb_dn2id_delete 0x%lx: %d\n", e->e_id, rc, 0 );
 728         return rc;
 729 }
 730
 731
 732 int
 733 hdb_dn2id(
 734         Operation       *op,
 735         struct berval   *in,
 736         EntryInfo       *ei,
 737         DB_TXN *txn,
 738         DB_LOCK *lock )
 739 {
 740         struct bdb_info *bdb = (struct bdb_info *) op->o_bd->be_private;
 741         DB *db = bdb->bi_dn2id->bdi_db;
 742         DBT             key, data;
 743         DBC     *cursor;
 744         int             rc = 0, nrlen;
 745         diskNode *d;
 746         char    *ptr;
 747         unsigned char dlen[2];
 748         ID idp, parentID;
 749
 750         Debug( LDAP_DEBUG_TRACE, "=> hdb_dn2id(\"%s\")\n", in->bv_val, 0, 0 );
 751
 752         nrlen = dn_rdnlen( op->o_bd, in );
 753         if (!nrlen) nrlen = in->bv_len;
 754
 755         DBTzero(&key);
 756         key.size = sizeof(ID);
 757         key.data = &idp;
 758         key.ulen = sizeof(ID);
 759         key.flags = DB_DBT_USERMEM;
 760         parentID = ( ei->bei_parent != NULL ) ? ei->bei_parent->bei_id : 0;
 761         BDB_ID2DISK( parentID, &idp );
 762
 763         DBTzero(&data);
 764         data.size = sizeof(diskNode) + nrlen - sizeof(ID) - 1;
 765         data.ulen = data.size * 3;
 766         data.dlen = data.ulen;
 767         data.flags = DB_DBT_USERMEM | DB_DBT_PARTIAL;
 768
 769         rc = db->cursor( db, txn, &cursor, bdb->bi_db_opflags );
 770         if ( rc ) return rc;
 771
 772         d = op->o_tmpalloc( data.size * 3, op->o_tmpmemctx );
 773         d->nrdnlen[1] = nrlen & 0xff;
 774         d->nrdnlen[0] = (nrlen >> 8) | 0x80;
 775         dlen[0] = d->nrdnlen[0];
 776         dlen[1] = d->nrdnlen[1];
 777         ptr = lutil_strncopy( d->nrdn, in->bv_val, nrlen );
 778         *ptr = '\0';
 779         data.data = d;
 780
 781         rc = bdb_dn2id_lock( bdb, in, 0, txn, lock );
 782         if ( rc ) goto func_leave;
 783
 784         rc = cursor->c_get( cursor, &key, &data, DB_GET_BOTH_RANGE );
 785         if ( rc == 0 && (dlen[1] != d->nrdnlen[1] || dlen[0] != d->nrdnlen[0] ||
 786                 strncmp( d->nrdn, in->bv_val, nrlen ))) {
 787                 rc = DB_NOTFOUND;
 788         }
 789         if ( rc == 0 ) {
 790                 ptr = (char *) data.data + data.size - sizeof(ID);
 791                 BDB_DISK2ID( ptr, &ei->bei_id );
 792                 ei->bei_rdn.bv_len = data.size - sizeof(diskNode) - nrlen;
 793                 ptr = d->nrdn + nrlen + 1;
 794                 ber_str2bv( ptr, ei->bei_rdn.bv_len, 1, &ei->bei_rdn );
 795                 if ( ei->bei_parent != NULL && !ei->bei_parent->bei_dkids ) {
 796                         db_recno_t dkids;
 797                         /* How many children does the parent have? */
 798                         /* FIXME: do we need to lock the parent
 799                          * entryinfo? Seems safe...
 800                          */
 801                         cursor->c_count( cursor, &dkids, 0 );
 802                         ei->bei_parent->bei_dkids = dkids;
 803                 }
 804         }
 805
 806 func_leave:
 807         cursor->c_close( cursor );
 808         op->o_tmpfree( d, op->o_tmpmemctx );
 809         if( rc != 0 ) {
 810                 Debug( LDAP_DEBUG_TRACE, "<= hdb_dn2id: get failed: %s (%d)\n",
 811                         db_strerror( rc ), rc, 0 );
 812         } else {
 813                 Debug( LDAP_DEBUG_TRACE, "<= hdb_dn2id: got id=0x%lx\n",
 814                         ei->bei_id, 0, 0 );
 815         }
 816
 817         return rc;
 818 }
 819
 820 int
 821 hdb_dn2id_parent(
 822         Operation *op,
 823         DB_TXN *txn,
 824         EntryInfo *ei,
 825         ID *idp )
 826 {
 827         struct bdb_info *bdb = (struct bdb_info *) op->o_bd->be_private;
 828         DB *db = bdb->bi_dn2id->bdi_db;
 829         DBT             key, data;
 830         DBC     *cursor;
 831         int             rc = 0;
 832         diskNode *d;
 833         char    *ptr;
 834         ID      nid;
 835
 836         DBTzero(&key);
 837         key.size = sizeof(ID);
 838         key.data = &nid;
 839         key.ulen = sizeof(ID);
 840         key.flags = DB_DBT_USERMEM;
 841         BDB_ID2DISK( ei->bei_id, &nid );
 842
 843         DBTzero(&data);
 844         data.flags = DB_DBT_USERMEM;
 845
 846         rc = db->cursor( db, txn, &cursor, bdb->bi_db_opflags );
 847         if ( rc ) return rc;
 848
 849         data.ulen = sizeof(diskNode) + (SLAP_LDAPDN_MAXLEN * 2);
 850         d = op->o_tmpalloc( data.ulen, op->o_tmpmemctx );
 851         data.data = d;
 852
 853         rc = cursor->c_get( cursor, &key, &data, DB_SET );
 854         if ( rc == 0 ) {
 855                 if (d->nrdnlen[0] & 0x80) {
 856                         rc = LDAP_OTHER;
 857                 } else {
 858                         db_recno_t dkids;
 859                         ptr = (char *) data.data + data.size - sizeof(ID);
 860                         BDB_DISK2ID( ptr, idp );
 861                         ei->bei_nrdn.bv_len = (d->nrdnlen[0] << 8) | d->nrdnlen[1];
 862                         ber_str2bv( d->nrdn, ei->bei_nrdn.bv_len, 1, &ei->bei_nrdn );
 863                         ei->bei_rdn.bv_len = data.size - sizeof(diskNode) -
 864                                 ei->bei_nrdn.bv_len;
 865                         ptr = d->nrdn + ei->bei_nrdn.bv_len + 1;
 866                         ber_str2bv( ptr, ei->bei_rdn.bv_len, 1, &ei->bei_rdn );
 867                         /* How many children does this node have? */
 868                         cursor->c_count( cursor, &dkids, 0 );
 869                         ei->bei_dkids = dkids;
 870                 }
 871         }
 872         cursor->c_close( cursor );
 873         op->o_tmpfree( d, op->o_tmpmemctx );
 874         return rc;
 875 }
 876
 877 int
 878 hdb_dn2id_children(
 879         Operation *op,
 880         DB_TXN *txn,
 881         Entry *e )
 882 {
 883         struct bdb_info *bdb = (struct bdb_info *) op->o_bd->be_private;
 884         DB *db = bdb->bi_dn2id->bdi_db;
 885         DBT             key, data;
 886         DBC             *cursor;
 887         int             rc;
 888         ID              id;
 889         diskNode d;
 890
 891         DBTzero(&key);
 892         key.size = sizeof(ID);
 893         key.data = &e->e_id;
 894         key.flags = DB_DBT_USERMEM;
 895         BDB_ID2DISK( e->e_id, &id );
 896
 897         /* IDL cache is in host byte order */
 898         if ( bdb->bi_idl_cache_size ) {
 899                 rc = bdb_idl_cache_get( bdb, db, &key, NULL );
 900                 if ( rc != LDAP_NO_SUCH_OBJECT ) {
 901                         return rc;
 902                 }
 903         }
 904
 905         key.data = &id;
 906         DBTzero(&data);
 907         data.data = &d;
 908         data.ulen = sizeof(d);
 909         data.flags = DB_DBT_USERMEM | DB_DBT_PARTIAL;
 910         data.dlen = sizeof(d);
 911
 912         rc = db->cursor( db, txn, &cursor, bdb->bi_db_opflags );
 913         if ( rc ) return rc;
 914
 915         rc = cursor->c_get( cursor, &key, &data, DB_SET );
 916         if ( rc == 0 ) {
 917                 db_recno_t dkids;
 918                 rc = cursor->c_count( cursor, &dkids, 0 );
 919                 if ( rc == 0 ) {
 920                         BEI(e)->bei_dkids = dkids;
 921                         if ( dkids < 2 ) rc = DB_NOTFOUND;
 922                 }
 923         }
 924         cursor->c_close( cursor );
 925         return rc;
 926 }
 927
 928 /* bdb_dn2idl:
 929  * We can't just use bdb_idl_fetch_key because
 930  * 1 - our data items are longer than just an entry ID
 931  * 2 - our data items are sorted alphabetically by nrdn, not by ID.
 932  *
 933  * We descend the tree recursively, so we define this cookie
 934  * to hold our necessary state information. The bdb_dn2idl_internal
 935  * function uses this cookie when calling itself.
 936  */
 937
 938 struct dn2id_cookie {
 939         struct bdb_info *bdb;
 940         Operation *op;
 941         DB_TXN *txn;
 942         EntryInfo *ei;
 943         ID *ids;
 944         ID *tmp;
 945         ID *buf;
 946         DB *db;
 947         DBC *dbc;
 948         DBT key;
 949         DBT data;
 950         ID dbuf;
 951         ID id;
 952         ID nid;
 953         int rc;
 954         int depth;
 955         char need_sort;
 956         char prefix;
 957 };
 958
 959 static int
 960 apply_func(
 961         void *data,
 962         void *arg )
 963 {
 964         EntryInfo *ei = data;
 965         ID *idl = arg;
 966
 967         bdb_idl_append_one( idl, ei->bei_id );
 968         return 0;
 969 }
 970
 971 static int
 972 hdb_dn2idl_internal(
 973         struct dn2id_cookie *cx
 974 )
 975 {
 976         BDB_IDL_ZERO( cx->tmp );
 977
 978         if ( cx->bdb->bi_idl_cache_size ) {
 979                 char *ptr = ((char *)&cx->id)-1;
 980
 981                 cx->key.data = ptr;
 982                 cx->key.size = sizeof(ID)+1;
 983                 if ( cx->prefix == DN_SUBTREE_PREFIX ) {
 984                         ID *ids = cx->depth ? cx->tmp : cx->ids;
 985                         *ptr = cx->prefix;
 986                         cx->rc = bdb_idl_cache_get(cx->bdb, cx->db, &cx->key, ids);
 987                         if ( cx->rc == LDAP_SUCCESS ) {
 988                                 if ( cx->depth ) {
 989                                         bdb_idl_append( cx->ids, cx->tmp );
 990                                         cx->need_sort = 1;
 991                                 }
 992                                 return cx->rc;
 993                         }
 994                 }
 995                 *ptr = DN_ONE_PREFIX;
 996                 cx->rc = bdb_idl_cache_get(cx->bdb, cx->db, &cx->key, cx->tmp);
 997                 if ( cx->rc == LDAP_SUCCESS ) {
 998                         goto gotit;
 999                 }
1000                 if ( cx->rc == DB_NOTFOUND ) {
1001                         return cx->rc;
1002                 }
1003         }
1004
1005         bdb_cache_entryinfo_lock( cx->ei );
1006
1007         /* If number of kids in the cache differs from on-disk, load
1008          * up all the kids from the database
1009          */
1010         if ( cx->ei->bei_ckids+1 != cx->ei->bei_dkids ) {
1011                 EntryInfo ei;
1012                 db_recno_t dkids = cx->ei->bei_dkids;
1013                 ei.bei_parent = cx->ei;
1014
1015                 /* Only one thread should load the cache */
1016                 while ( cx->ei->bei_state & CACHE_ENTRY_ONELEVEL ) {
1017                         bdb_cache_entryinfo_unlock( cx->ei );
1018                         ldap_pvt_thread_yield();
1019                         bdb_cache_entryinfo_lock( cx->ei );
1020                         if ( cx->ei->bei_ckids+1 == cx->ei->bei_dkids ) {
1021                                 goto synced;
1022                         }
1023                 }
1024
1025                 cx->ei->bei_state |= CACHE_ENTRY_ONELEVEL;
1026
1027                 bdb_cache_entryinfo_unlock( cx->ei );
1028
1029                 cx->rc = cx->db->cursor( cx->db, NULL, &cx->dbc,
1030                         cx->bdb->bi_db_opflags );
1031                 if ( cx->rc )
1032                         goto done_one;
1033
1034                 cx->data.data = &cx->dbuf;
1035                 cx->data.ulen = sizeof(ID);
1036                 cx->data.dlen = sizeof(ID);
1037                 cx->data.flags = DB_DBT_USERMEM | DB_DBT_PARTIAL;
1038
1039                 /* The first item holds the parent ID. Ignore it. */
1040                 cx->key.data = &cx->nid;
1041                 cx->key.size = sizeof(ID);
1042                 cx->rc = cx->dbc->c_get( cx->dbc, &cx->key, &cx->data, DB_SET );
1043                 if ( cx->rc ) {
1044                         cx->dbc->c_close( cx->dbc );
1045                         goto done_one;
1046                 }
1047
1048                 /* If the on-disk count is zero we've never checked it.
1049                  * Count it now.
1050                  */
1051                 if ( !dkids ) {
1052                         cx->dbc->c_count( cx->dbc, &dkids, 0 );
1053                         cx->ei->bei_dkids = dkids;
1054                 }
1055
1056                 cx->data.data = cx->buf;
1057                 cx->data.ulen = BDB_IDL_UM_SIZE * sizeof(ID);
1058                 cx->data.flags = DB_DBT_USERMEM;
1059
1060                 if ( dkids > 1 ) {
1061                         /* Fetch the rest of the IDs in a loop... */
1062                         while ( (cx->rc = cx->dbc->c_get( cx->dbc, &cx->key, &cx->data,
1063                                 DB_MULTIPLE | DB_NEXT_DUP )) == 0 ) {
1064                                 u_int8_t *j;
1065                                 size_t len;
1066                                 void *ptr;
1067                                 DB_MULTIPLE_INIT( ptr, &cx->data );
1068                                 while (ptr) {
1069                                         DB_MULTIPLE_NEXT( ptr, &cx->data, j, len );
1070                                         if (j) {
1071                                                 EntryInfo *ei2;
1072                                                 diskNode *d = (diskNode *)j;
1073                                                 short nrlen;
1074
1075                                                 BDB_DISK2ID( j + len - sizeof(ID), &ei.bei_id );
1076                                                 nrlen = ((d->nrdnlen[0] ^ 0x80) << 8) | d->nrdnlen[1];
1077                                                 ei.bei_nrdn.bv_len = nrlen;
1078                                                 /* nrdn/rdn are set in-place.
1079                                                  * hdb_cache_load will copy them as needed
1080                                                  */
1081                                                 ei.bei_nrdn.bv_val = d->nrdn;
1082                                                 ei.bei_rdn.bv_len = len - sizeof(diskNode)
1083                                                         - ei.bei_nrdn.bv_len;
1084                                                 ei.bei_rdn.bv_val = d->nrdn + ei.bei_nrdn.bv_len + 1;
1085                                                 bdb_idl_append_one( cx->tmp, ei.bei_id );
1086                                                 hdb_cache_load( cx->bdb, &ei, &ei2 );
1087                                         }
1088                                 }
1089                         }
1090                 }
1091
1092                 cx->rc = cx->dbc->c_close( cx->dbc );
1093 done_one:
1094                 bdb_cache_entryinfo_lock( cx->ei );
1095                 cx->ei->bei_state ^= CACHE_ENTRY_ONELEVEL;
1096                 bdb_cache_entryinfo_unlock( cx->ei );
1097                 if ( cx->rc )
1098                         return cx->rc;
1099
1100         } else {
1101                 /* The in-memory cache is in sync with the on-disk data.
1102                  * do we have any kids?
1103                  */
1104 synced:
1105                 cx->rc = 0;
1106                 if ( cx->ei->bei_ckids > 0 ) {
1107                         /* Walk the kids tree; order is irrelevant since bdb_idl_sort
1108                          * will sort it later.
1109                          */
1110                         avl_apply( cx->ei->bei_kids, apply_func,
1111                                 cx->tmp, -1, AVL_POSTORDER );
1112                 }
1113                 bdb_cache_entryinfo_unlock( cx->ei );
1114         }
1115
1116         if ( !BDB_IDL_IS_RANGE( cx->tmp ) && cx->tmp[0] > 3 )
1117                 bdb_idl_sort( cx->tmp, cx->buf );
1118         if ( cx->bdb->bi_idl_cache_max_size && !BDB_IDL_IS_ZERO( cx->tmp )) {
1119                 char *ptr = ((char *)&cx->id)-1;
1120                 cx->key.data = ptr;
1121                 cx->key.size = sizeof(ID)+1;
1122                 *ptr = DN_ONE_PREFIX;
1123                 bdb_idl_cache_put( cx->bdb, cx->db, &cx->key, cx->tmp, cx->rc );
1124         }
1125
1126 gotit:
1127         if ( !BDB_IDL_IS_ZERO( cx->tmp )) {
1128                 if ( cx->prefix == DN_SUBTREE_PREFIX ) {
1129                         bdb_idl_append( cx->ids, cx->tmp );
1130                         cx->need_sort = 1;
1131                         if ( !(cx->ei->bei_state & CACHE_ENTRY_NO_GRANDKIDS)) {
1132                                 ID *save, idcurs;
1133                                 EntryInfo *ei = cx->ei;
1134                                 int nokids = 1;
1135                                 save = cx->op->o_tmpalloc( BDB_IDL_SIZEOF( cx->tmp ),
1136                                         cx->op->o_tmpmemctx );
1137                                 BDB_IDL_CPY( save, cx->tmp );
1138
1139                                 idcurs = 0;
1140                                 cx->depth++;
1141                                 for ( cx->id = bdb_idl_first( save, &idcurs );
1142                                         cx->id != NOID;
1143                                         cx->id = bdb_idl_next( save, &idcurs )) {
1144                                         cx->ei = bdb_cache_find_info( cx->bdb, cx->id );
1145                                         if ( !cx->ei ||
1146                                                 ( cx->ei->bei_state & CACHE_ENTRY_NO_KIDS ))
1147                                                 continue;
1148
1149                                         BDB_ID2DISK( cx->id, &cx->nid );
1150                                         hdb_dn2idl_internal( cx );
1151                                         if ( !BDB_IDL_IS_ZERO( cx->tmp ))
1152                                                 nokids = 0;
1153                                 }
1154                                 cx->depth--;
1155                                 cx->op->o_tmpfree( save, cx->op->o_tmpmemctx );
1156                                 if ( nokids ) {
1157                                         bdb_cache_entryinfo_lock( ei );
1158                                         ei->bei_state |= CACHE_ENTRY_NO_GRANDKIDS;
1159                                         bdb_cache_entryinfo_unlock( ei );
1160                                 }
1161                         }
1162                         /* Make sure caller knows it had kids! */
1163                         cx->tmp[0]=1;
1164
1165                         cx->rc = 0;
1166                 } else {
1167                         BDB_IDL_CPY( cx->ids, cx->tmp );
1168                 }
1169         }
1170         return cx->rc;
1171 }
1172
1173 int
1174 hdb_dn2idl(
1175         Operation       *op,
1176         DB_TXN *txn,
1177         struct berval *ndn,
1178         EntryInfo       *ei,
1179         ID *ids,
1180         ID *stack )
1181 {
1182         struct bdb_info *bdb = (struct bdb_info *)op->o_bd->be_private;
1183         struct dn2id_cookie cx;
1184
1185         Debug( LDAP_DEBUG_TRACE, "=> hdb_dn2idl(\"%s\")\n",
1186                 ndn->bv_val, 0, 0 );
1187
1188 #ifndef BDB_MULTIPLE_SUFFIXES
1189         if ( op->ors_scope != LDAP_SCOPE_ONELEVEL &&
1190                 ( ei->bei_id == 0 ||
1191                 ( ei->bei_parent->bei_id == 0 && op->o_bd->be_suffix[0].bv_len )))
1192         {
1193                 BDB_IDL_ALL( bdb, ids );
1194                 return 0;
1195         }
1196 #endif
1197
1198         cx.id = ei->bei_id;
1199         BDB_ID2DISK( cx.id, &cx.nid );
1200         cx.ei = ei;
1201         cx.bdb = bdb;
1202         cx.db = cx.bdb->bi_dn2id->bdi_db;
1203         cx.prefix = (op->ors_scope == LDAP_SCOPE_ONELEVEL) ?
1204                 DN_ONE_PREFIX : DN_SUBTREE_PREFIX;
1205         cx.ids = ids;
1206         cx.tmp = stack;
1207         cx.buf = stack + BDB_IDL_UM_SIZE;
1208         cx.op = op;
1209         cx.txn = txn;
1210         cx.need_sort = 0;
1211         cx.depth = 0;
1212
1213         if ( cx.prefix == DN_SUBTREE_PREFIX ) {
1214                 ids[0] = 1;
1215                 ids[1] = cx.id;
1216         } else {
1217                 BDB_IDL_ZERO( ids );
1218         }
1219         if ( cx.ei->bei_state & CACHE_ENTRY_NO_KIDS )
1220                 return LDAP_SUCCESS;
1221
1222         DBTzero(&cx.key);
1223         cx.key.ulen = sizeof(ID);
1224         cx.key.size = sizeof(ID);
1225         cx.key.flags = DB_DBT_USERMEM;
1226
1227         DBTzero(&cx.data);
1228
1229         hdb_dn2idl_internal(&cx);
1230         if ( cx.need_sort ) {
1231                 char *ptr = ((char *)&cx.id)-1;
1232                 if ( !BDB_IDL_IS_RANGE( cx.ids ) && cx.ids[0] > 3 )
1233                         bdb_idl_sort( cx.ids, cx.tmp );
1234                 cx.key.data = ptr;
1235                 cx.key.size = sizeof(ID)+1;
1236                 *ptr = cx.prefix;
1237                 cx.id = ei->bei_id;
1238                 if ( cx.bdb->bi_idl_cache_max_size )
1239                         bdb_idl_cache_put( cx.bdb, cx.db, &cx.key, cx.ids, cx.rc );
1240         }
1241
1242         if ( cx.rc == DB_NOTFOUND )
1243                 cx.rc = LDAP_SUCCESS;
1244
1245         return cx.rc;
1246 }
1247 #endif  /* BDB_HIER */