]> git.sur5r.net Git - openldap/blobdiff - servers/slapd/back-bdb/dn2id.c
Missed one db.bv_val in Debug statement
[openldap] / servers / slapd / back-bdb / dn2id.c
index b0088495731f4fdb373d4304b1b542659cf12b81..1be0a4bbfd240f187f0026c1b5b1c7ccabb06853 100644 (file)
 /* dn2id.c - routines to deal with the dn2id index */
 /* $OpenLDAP$ */
 /*
- * Copyright 1998-2000 The OpenLDAP Foundation, All Rights Reserved.
+ * Copyright 1998-2002 The OpenLDAP Foundation, All Rights Reserved.
  * COPYING RESTRICTIONS APPLY, see COPYRIGHT file
  */
 
 #include "portable.h"
 
 #include <stdio.h>
-
 #include <ac/string.h>
-#include <ac/socket.h>
 
 #include "back-bdb.h"
+#include "idl.h"
 
+#ifndef BDB_HIER
 int
-bdb_index_dn_add(
-    Backend    *be,
+bdb_dn2id_add(
+       BackendDB       *be,
        DB_TXN *txn,
-    const char *dn,
-    ID         id
-)
+       struct berval   *pbv,
+       Entry           *e )
 {
-       int             rc;
-       DBT             key, data;
        struct bdb_info *bdb = (struct bdb_info *) be->be_private;
        DB *db = bdb->bi_dn2id->bdi_db;
+       int             rc;
+       DBT             key, data;
+       char            *buf, *ptr, *pdn;
 
-       Debug( LDAP_DEBUG_TRACE, "=> bdb_index_dn_add( \"%s\", %ld )\n", dn, id, 0 );
-       assert( id != NOID );
+       Debug( LDAP_DEBUG_TRACE, "=> bdb_dn2id_add( \"%s\", 0x%08lx )\n",
+               e->e_ndn, (long) e->e_id, 0 );
+       assert( e->e_id != NOID );
 
        DBTzero( &key );
-       key.size = strlen( dn ) + 2;
-       key.data = ch_malloc( key.size );
-       ((char *)key.data)[0] = DN_BASE_PREFIX;
-       AC_MEMCPY( &((char *)key.data)[1], dn, key.size - 1 );
+       key.size = e->e_nname.bv_len + 2;
+       buf = ch_malloc( key.size );
+       key.data = buf;
+       buf[0] = DN_BASE_PREFIX;
+       ptr = buf + 1;
+       AC_MEMCPY( ptr, e->e_ndn, key.size - 1 );
 
        DBTzero( &data );
-       data.data = (char *) &id;
-       data.size = sizeof( id );
+       data.data = (char *) &e->e_id;
+       data.size = sizeof( e->e_id );
 
        /* store it -- don't override */
        rc = db->put( db, txn, &key, &data, DB_NOOVERWRITE );
        if( rc != 0 ) {
+               Debug( LDAP_DEBUG_ANY, "=> bdb_dn2id_add: put failed: %s %d\n",
+                       db_strerror(rc), rc, 0 );
                goto done;
        }
 
-       {
-               char *pdn = dn_parent( NULL, dn );
-               ((char *)(key.data))[0] = DN_ONE_PREFIX;
+       if( !be_issuffix( be, ptr )) {
+               buf[0] = DN_SUBTREE_PREFIX;
+               rc = bdb_idl_insert_key( be, db, txn, &key, e->e_id );
+               if( rc != 0 ) {
+                       Debug( LDAP_DEBUG_ANY,
+                       "=> bdb_dn2id_add: subtree (%s) insert failed: %d\n",
+                       ptr, rc, 0 );
+                       goto done;
+               }
+               
+               rc = dnParent( ptr, (const char **)&pdn );
+               if ( rc != LDAP_SUCCESS ) {
+                       Debug( LDAP_DEBUG_ANY,
+                               "=> bdb_dn2id_add: dnParent(\"%s\") failed\n",
+                               ptr, 0, 0 );
+                       goto done;
+               }
+       
+               key.size -= pdn - ptr;
+               pdn[-1] = DN_ONE_PREFIX;
+               key.data = pdn - 1;
+               ptr = pdn;
+
+               rc = bdb_idl_insert_key( be, db, txn, &key, e->e_id );
+
+               if( rc != 0 ) {
+                       Debug( LDAP_DEBUG_ANY,
+                               "=> bdb_dn2id_add: parent (%s) insert failed: %d\n",
+                                       ptr, rc, 0 );
+                       goto done;
+               }
+       }
 
-               if( pdn != NULL ) {
-                       key.size = strlen( pdn ) + 2;
-                       AC_MEMCPY( &((char*)key.data)[1],
-                               pdn, key.size - 1 );
+       while( !be_issuffix( be, ptr )) {
+               ptr[-1] = DN_SUBTREE_PREFIX;
 
-                       rc = bdb_idl_insert_key( be, db, txn, &key, id );
-                       free( pdn );
+               rc = bdb_idl_insert_key( be, db, txn, &key, e->e_id );
 
-                       if( rc != 0 ) {
-                               goto done;
-                       }
+               if( rc != 0 ) {
+                       Debug( LDAP_DEBUG_ANY,
+                               "=> bdb_dn2id_add: subtree (%s) insert failed: %d\n",
+                                       ptr, rc, 0 );
+                       break;
                }
+               rc = dnParent( ptr, &pdn );
+               if ( rc != LDAP_SUCCESS ) {
+                       Debug( LDAP_DEBUG_ANY,
+                               "=> bdb_dn2id_add: dnParent(\"%s\") failed\n",
+                               ptr, 0, 0 );
+                       goto done;
+               }
+
+               key.size -= pdn - ptr;
+               key.data = pdn - 1;
+               ptr = pdn;
        }
 
-       if ( rc != -1 ) {
-               char **subtree = dn_subtree( NULL, dn );
+done:
+       ch_free( buf );
+       Debug( LDAP_DEBUG_TRACE, "<= bdb_dn2id_add: %d\n", rc, 0, 0 );
+       return rc;
+}
 
-               if( subtree != NULL ) {
-                       int i;
-                       ((char *)key.data)[0] = DN_SUBTREE_PREFIX;
-                       for( i=0; subtree[i] != NULL; i++ ) {
-                               key.size = strlen( subtree[i] ) + 2;
-                               AC_MEMCPY( &((char *)key.data)[1],
-                                       subtree[i], key.size - 1 );
+int
+bdb_dn2id_delete(
+       BackendDB       *be,
+       DB_TXN *txn,
+       char    *pdn,
+       Entry           *e )
+{
+       struct bdb_info *bdb = (struct bdb_info *) be->be_private;
+       DB *db = bdb->bi_dn2id->bdi_db;
+       int             rc;
+       DBT             key;
+       char            *buf, *ptr;
 
-                               rc = bdb_idl_insert_key( be, db, txn, &key, id );
+       Debug( LDAP_DEBUG_TRACE, "=> bdb_dn2id_delete( \"%s\", 0x%08lx )\n",
+               e->e_ndn, e->e_id, 0 );
 
-                               if( rc != 0 ) {
-                                       goto done;
-                               }
-                       }
+       DBTzero( &key );
+       key.size = e->e_nname.bv_len + 2;
+       buf = ch_malloc( key.size );
+       key.data = buf;
+       key.flags = DB_DBT_USERMEM;
+       buf[0] = DN_BASE_PREFIX;
+       ptr = buf+1;
+       AC_MEMCPY( ptr, e->e_ndn, key.size - 1 );
+
+       /* delete it */
+       rc = db->del( db, txn, &key, 0 );
+       if( rc != 0 ) {
+               Debug( LDAP_DEBUG_ANY, "=> bdb_dn2id_delete: delete failed: %s %d\n",
+                       db_strerror(rc), rc, 0 );
+               goto done;
+       }
+
+       if( !be_issuffix( be, ptr )) {
+               buf[0] = DN_SUBTREE_PREFIX;
+               rc = bdb_idl_delete_key( be, db, txn, &key, e->e_id );
+               if( rc != 0 ) {
+                       Debug( LDAP_DEBUG_ANY,
+                       "=> bdb_dn2id_delete: subtree (%s) delete failed: %d\n",
+                       ptr, rc, 0 );
+                       goto done;
+               }
+
+               rc = dnParent( ptr, &pdn );
+               if ( rc != LDAP_SUCCESS ) {
+                       Debug( LDAP_DEBUG_ANY,
+                               "=> bdb_dn2id_delete: dnParent(\"%s\") failed\n",
+                               ptr, 0, 0 );
+                       goto done;
+               }
+
+               key.size -= pdn - ptr;
+               pdn[-1] = DN_ONE_PREFIX;
+               key.data = pdn - 1;
+               ptr = pdn;
 
-                       charray_free( subtree );
+               rc = bdb_idl_delete_key( be, db, txn, &key, e->e_id );
+
+               if( rc != 0 ) {
+                       Debug( LDAP_DEBUG_ANY,
+                               "=> bdb_dn2id_delete: parent (%s) delete failed: %d\n",
+                               ptr, rc, 0 );
+                       goto done;
                }
        }
 
+       while( !be_issuffix( be, ptr )) {
+               ptr[-1] = DN_SUBTREE_PREFIX;
+
+               rc = bdb_idl_delete_key( be, db, txn, &key, e->e_id );
+               if( rc != 0 ) {
+                       Debug( LDAP_DEBUG_ANY,
+                               "=> bdb_dn2id_delete: subtree (%s) delete failed: %d\n",
+                               ptr, rc, 0 );
+                       goto done;
+               }
+               rc = dnParent( ptr, &pdn );
+               if ( rc != LDAP_SUCCESS ) {
+                       Debug( LDAP_DEBUG_ANY,
+                               "=> bdb_dn2id_delete: dnParent(\"%s\") failed\n",
+                               ptr, 0, 0 );
+                       goto done;
+               }
+
+               key.size -= pdn - ptr;
+               key.data = pdn - 1;
+               ptr = pdn;
+       }
+
 done:
-       ch_free( key.data );
-       Debug( LDAP_DEBUG_TRACE, "<= bdb_index_dn_add %d\n", rc, 0, 0 );
-       return( rc );
+       ch_free( buf );
+       Debug( LDAP_DEBUG_TRACE, "<= bdb_dn2id_delete %d\n", rc, 0, 0 );
+       return rc;
 }
 
-#if 0
-ID
-dn2id(
-    Backend    *be,
-    const char *dn
-)
+int
+bdb_dn2id(
+       BackendDB       *be,
+       DB_TXN *txn,
+       struct berval   *dn,
+       ID *id )
 {
-       struct ldbminfo *li = (struct ldbminfo *) be->be_private;
-       DBCache *db;
-       ID              id;
-       Datum           key, data;
+       int             rc;
+       DBT             key, data;
+       struct bdb_info *bdb = (struct bdb_info *) be->be_private;
+       DB *db = bdb->bi_dn2id->bdi_db;
 
-       Debug( LDAP_DEBUG_TRACE, "=> dn2id( \"%s\" )\n", dn, 0, 0 );
+       Debug( LDAP_DEBUG_TRACE, "=> bdb_dn2id( \"%s\" )\n", dn->bv_val, 0, 0 );
 
-       /* first check the cache */
-       if ( (id = cache_find_entry_dn2id( be, &li->li_cache, dn )) != NOID ) {
-               Debug( LDAP_DEBUG_TRACE, "<= dn2id %ld (in cache)\n", id,
-                       0, 0 );
-               return( id );
+       assert (id);
+       *id = bdb_cache_find_entry_ndn2id(be, &bdb->bi_cache, dn);
+       if (*id != NOID) {
+               return 0;
        }
 
-       if ( (db = ldbm_cache_open( be, "dn2id", LDBM_SUFFIX, LDBM_WRCREAT ))
-               == NULL ) {
-               Debug( LDAP_DEBUG_ANY, "<= dn2id could not open dn2id%s\n",
-                       LDBM_SUFFIX, 0, 0 );
-               return( NOID );
+       DBTzero( &key );
+       key.size = dn->bv_len + 2;
+       key.data = ch_malloc( key.size );
+       ((char *)key.data)[0] = DN_BASE_PREFIX;
+       AC_MEMCPY( &((char *)key.data)[1], dn->bv_val, key.size - 1 );
+
+       /* store the ID */
+       DBTzero( &data );
+       data.data = id;
+       data.ulen = sizeof(ID);
+       data.flags = DB_DBT_USERMEM;
+
+       /* fetch it */
+       rc = db->get( db, txn, &key, &data, bdb->bi_db_opflags );
+
+       if( rc != 0 ) {
+               Debug( LDAP_DEBUG_TRACE, "<= bdb_dn2id: get failed: %s (%d)\n",
+                       db_strerror( rc ), rc, 0 );
+       } else {
+               Debug( LDAP_DEBUG_TRACE, "<= bdb_dn2id: got id=0x%08lx\n",
+                       *id, 0, 0 );
        }
 
-       ldbm_datum_init( key );
+       ch_free( key.data );
+       return rc;
+}
+
+int
+bdb_dn2id_matched(
+       BackendDB       *be,
+       DB_TXN *txn,
+       struct berval   *in,
+       ID *id,
+       ID *id2 )
+{
+       int             rc;
+       DBT             key, data;
+       struct bdb_info *bdb = (struct bdb_info *) be->be_private;
+       DB *db = bdb->bi_dn2id->bdi_db;
+       char            *buf;
+       struct  berval dn;
+       ID              cached_id;
+
+       Debug( LDAP_DEBUG_TRACE, "=> bdb_dn2id_matched( \"%s\" )\n", in->bv_val, 0, 0 );
+
+       DBTzero( &key );
+       key.size = in->bv_len + 2;
+       buf = ch_malloc( key.size );
+       key.data = buf;
+       dn.bv_val = buf+1;
+       dn.bv_len = key.size - 2;
+       AC_MEMCPY( dn.bv_val, in->bv_val, key.size - 1 );
+
+       /* store the ID */
+       DBTzero( &data );
+       data.data = id;
+       data.ulen = sizeof(ID);
+       data.flags = DB_DBT_USERMEM;
+
+       while(1) {
+               dn.bv_val[-1] = DN_BASE_PREFIX;
+
+               *id = NOID;
+
+               /* lookup cache */
+               cached_id = bdb_cache_find_entry_ndn2id(be, &bdb->bi_cache, &dn);
+               if (cached_id != NOID) {
+                       rc = 0;
+                       *id = cached_id;
+                       if ( dn.bv_val != buf+1 ) {
+                               *id2 = *id;
+                       }
+                       break;
+               } else {
+                       /* fetch it */
+                       rc = db->get(db, txn, &key, &data, bdb->bi_db_opflags );
+               }
+
+               if( rc == DB_NOTFOUND ) {
+                       char    *pdn = NULL;
 
-       key.dsize = strlen( dn ) + 2;
-       key.dptr = ch_malloc( key.dsize );
-       sprintf( key.dptr, "%c%s", DN_BASE_PREFIX, dn );
+                       if ( ! be_issuffix( be, dn.bv_val ) ) {
+                               rc = dnParent( dn.bv_val, &pdn );
+                               if ( rc != LDAP_SUCCESS ) {
+                                       Debug( LDAP_DEBUG_TRACE,
+                                               "<= bdb_dn2id_matched: dnParent(\"%s\") failed\n",
+                                               dn, 0, 0 );
+                                       break;
+                               }
+                       }
 
-       data = ldbm_cache_fetch( db, key );
+                       if( pdn == NULL || *pdn == '\0' ) {
+                               Debug( LDAP_DEBUG_TRACE,
+                                       "<= bdb_dn2id_matched: no match\n",
+                                       0, 0, 0 );
+                               break;
+                       }
 
-       ldbm_cache_close( be, db );
+                       key.size -= pdn - dn.bv_val;
+                       dn.bv_val = pdn;
+                       dn.bv_len = key.size - 2;
+                       key.data = pdn - 1;
+
+               } else if ( rc == 0 ) {
+                       if( data.size != sizeof( ID ) ) {
+                               Debug( LDAP_DEBUG_ANY,
+                                       "<= bdb_dn2id_matched: get size mismatch: "
+                                       "expected %ld, got %ld\n",
+                                       (long) sizeof(ID), (long) data.size, 0 );
+                       }
+
+                       if( dn.bv_val != buf+1 ) {
+                               *id2 = *id;
+                       }
 
-       free( key.dptr );
+                       Debug( LDAP_DEBUG_TRACE,
+                               "<= bdb_dn2id_matched: id=0x%08lx: %s %s\n",
+                               (long) *id, *id2 == 0 ? "entry" : "matched", dn.bv_val );
+                       break;
 
-       if ( data.dptr == NULL ) {
-               Debug( LDAP_DEBUG_TRACE, "<= dn2id NOID\n", 0, 0, 0 );
-               return( NOID );
+               } else {
+                       Debug( LDAP_DEBUG_ANY,
+                               "<= bdb_dn2id_matched: get failed: %s (%d)\n",
+                               db_strerror(rc), rc, 0 );
+                       break;
+               }
        }
 
-       AC_MEMCPY( (char *) &id, data.dptr, sizeof(ID) );
+       ch_free( buf );
+       return rc;
+}
+
+int
+bdb_dn2id_children(
+       BackendDB       *be,
+       DB_TXN *txn,
+       struct berval *dn )
+{
+       int             rc;
+       DBT             key, data;
+       struct bdb_info *bdb = (struct bdb_info *) be->be_private;
+       DB *db = bdb->bi_dn2id->bdi_db;
+       ID              id;
 
-       assert( id != NOID );
+       Debug( LDAP_DEBUG_TRACE, "=> bdb_dn2id_children( %s )\n",
+               dn->bv_val, 0, 0 );
 
-       ldbm_datum_free( db->dbc_db, data );
+       DBTzero( &key );
+       key.size = dn->bv_len + 2;
+       key.data = ch_malloc( key.size );
+       ((char *)key.data)[0] = DN_ONE_PREFIX;
+       AC_MEMCPY( &((char *)key.data)[1], dn->bv_val, key.size - 1 );
 
-       Debug( LDAP_DEBUG_TRACE, "<= dn2id %ld\n", id, 0, 0 );
-       return( id );
+       /* we actually could do a empty get... */
+       DBTzero( &data );
+       data.data = &id;
+       data.ulen = sizeof(id);
+       data.flags = DB_DBT_USERMEM;
+       data.doff = 0;
+       data.dlen = sizeof(id);
+
+       rc = db->get( db, txn, &key, &data, bdb->bi_db_opflags );
+       free( key.data );
+
+       Debug( LDAP_DEBUG_TRACE, "<= bdb_dn2id_children( %s ): %schildren (%d)\n",
+               dn->bv_val,
+               rc == 0 ? "" : ( rc == DB_NOTFOUND ? "no " :
+                       db_strerror(rc) ), rc );
+
+       return rc;
 }
 
-ID_BLOCK *
-dn2idl(
-    Backend    *be,
-    const char *dn,
-       int             prefix
-)
+int
+bdb_dn2idl(
+       BackendDB       *be,
+       struct berval   *dn,
+       int prefix,
+       ID *ids )
 {
-       DBCache *db;
-       Datum           key;
-       ID_BLOCK        *idl;
+       int             rc;
+       DBT             key;
+       struct bdb_info *bdb = (struct bdb_info *) be->be_private;
+       DB *db = bdb->bi_dn2id->bdi_db;
 
-       Debug( LDAP_DEBUG_TRACE, "=> dn2idl( \"%c%s\" )\n", prefix, dn, 0 );
+       Debug( LDAP_DEBUG_TRACE, "=> bdb_dn2idl( \"%s\" )\n", dn->bv_val, 0, 0 );
 
-       if ( (db = ldbm_cache_open( be, "dn2id", LDBM_SUFFIX, LDBM_WRCREAT ))
-               == NULL ) {
-               Debug( LDAP_DEBUG_ANY, "<= dn2idl could not open dn2id%s\n",
-                       LDBM_SUFFIX, 0, 0 );
-               return NULL;
+       if (prefix == DN_SUBTREE_PREFIX && be_issuffix(be, dn->bv_val))
+       {
+               BDB_IDL_ALL(bdb, ids);
+               return 0;
        }
 
-       ldbm_datum_init( key );
+       DBTzero( &key );
+       key.size = dn->bv_len + 2;
+       key.data = ch_malloc( key.size );
+       ((char *)key.data)[0] = prefix;
+       AC_MEMCPY( &((char *)key.data)[1], dn->bv_val, key.size - 1 );
 
-       key.dsize = strlen( dn ) + 2;
-       key.dptr = ch_malloc( key.dsize );
-       sprintf( key.dptr, "%c%s", prefix, dn );
+       rc = bdb_idl_fetch_key( be, db, NULL, &key, ids );
 
-       idl = idl_fetch( be, db, key );
+       if( rc != 0 ) {
+               Debug( LDAP_DEBUG_TRACE,
+                       "<= bdb_dn2idl: get failed: %s (%d)\n",
+                       db_strerror( rc ), rc, 0 );
+
+       } else {
+               Debug( LDAP_DEBUG_TRACE,
+                       "<= bdb_dn2idl: id=%ld first=%ld last=%ld\n",
+                       (long) ids[0],
+                       (long) BDB_IDL_FIRST( ids ), (long) BDB_IDL_LAST( ids ) );
+       }
 
-       ldbm_cache_close( be, db );
+       ch_free( key.data );
+       return rc;
+}
+#else  /* BDB_HIER */
+
+/* Experimental management routines for a hierarchically structured backend.
+ *
+ * Unsupported! Use at your own risk!
+ *
+ * Instead of a dn2id database, we use an id2parent database. Each entry in
+ * this database is a struct diskNode, containing the ID of the node's parent
+ * and the RDN of the node.
+ */
+typedef struct diskNode {
+       ID parent;
+       struct berval rdn;
+       struct berval nrdn;
+} diskNode;
+
+/* In bdb_db_open() we call bdb_build_tree() which reads the entire id2parent
+ * database into memory (into an AVL tree). Next we iterate through each node
+ * of this tree, connecting each child to its parent. The nodes in this AVL
+ * tree are a struct idNode. The immediate (Onelevel) children of a node are
+ * referenced in the i_kids AVL tree. With this arrangement, there is no need
+ * to maintain the DN_ONE_PREFIX or DN_SUBTREE_PREFIX database keys. Note that
+ * the DN of an entry is constructed by walking up the list of i_parent
+ * pointers, so no full DN is stored on disk anywhere. This makes modrdn
+ * extremely efficient, even when operating on a populated subtree.
+ *
+ * The idNode tree is searched directly from the root when performing id to
+ * entry lookups. The tree is traversed using the i_kids subtrees when
+ * performing dn to id lookups.
+ */
+typedef struct idNode {
+       ID i_id;
+       struct idNode *i_parent;
+       diskNode *i_rdn;
+       Avlnode *i_kids;
+       ldap_pvt_thread_rdwr_t i_kids_rdwr;
+} idNode;
+
+
+/* The main AVL tree is sorted in ID order. The i_kids AVL trees are
+ * sorted in lexical order. These are the various helper routines used
+ * for the searches and sorts.
+ */
+static int
+node_find_cmp(
+       ID id,
+       idNode *n
+)
+{
+       return id - n->i_id;
+}
 
-       free( key.dptr );
+static int
+node_frdn_cmp(
+       char *nrdn,
+       idNode *n
+)
+{
+       return strcmp(nrdn, n->i_rdn->nrdn.bv_val);
+}
 
-       return( idl );
+static int
+node_add_cmp(
+       idNode *a,
+       idNode *b
+)
+{
+       return a->i_id - b->i_id;
 }
 
+static int
+node_rdn_cmp(
+       idNode *a,
+       idNode *b
+)
+{
+#if 0
+       return strcmp(a->i_rdn->nrdn.bv_val, b->i_rdn->nrdn.bv_val);
+#endif
+       /* should be slightly better without ordering drawbacks */
+       return ber_bvcmp(&a->i_rdn->nrdn, &b->i_rdn->nrdn);
+}
 
-int
-dn2id_delete(
-    Backend    *be,
-    const char *dn,
-       ID id
+idNode * bdb_find_id_node(
+       ID id,
+       Avlnode *tree
 )
 {
-       DBCache *db;
-       Datum           key;
-       int             rc;
+       return avl_find(tree, (const void *)id, (AVL_CMP)node_find_cmp);
+}
 
-       Debug( LDAP_DEBUG_TRACE, "=> dn2id_delete( \"%s\", %ld )\n", dn, id, 0 );
+idNode * bdb_find_rdn_node(
+       char *nrdn,
+       Avlnode *tree
+)
+{
+       return avl_find(tree, (const void *)nrdn, (AVL_CMP)node_frdn_cmp);
+}
 
-       assert( id != NOID );
+/* This function links a node into its parent's i_kids tree. */
+int bdb_insert_kid(
+       idNode *a,
+       Avlnode *tree
+)
+{
+       int rc;
+
+       if (a->i_rdn->parent == 0)
+               return 0;
+       a->i_parent = bdb_find_id_node(a->i_rdn->parent, tree);
+       if (!a->i_parent)
+               return -1;
+       ldap_pvt_thread_rdwr_wlock(&a->i_parent->i_kids_rdwr);
+       rc = avl_insert( &a->i_parent->i_kids, (caddr_t) a,
+               (AVL_CMP)node_rdn_cmp, (AVL_DUP) avl_dup_error );
+       ldap_pvt_thread_rdwr_wunlock(&a->i_parent->i_kids_rdwr);
+       return rc;
+}
 
-       if ( (db = ldbm_cache_open( be, "dn2id", LDBM_SUFFIX, LDBM_WRCREAT ))
-           == NULL ) {
-               Debug( LDAP_DEBUG_ANY,
-                   "<= dn2id_delete could not open dn2id%s\n", LDBM_SUFFIX,
-                   0, 0 );
-               return( -1 );
+/* This function adds a node into the main AVL tree */
+idNode *bdb_add_node(
+       ID id,
+       char *d,
+       struct bdb_info *bdb
+)
+{
+       idNode *node;
+
+       node = (idNode *)ch_malloc(sizeof(idNode));
+       node->i_id = id;
+       node->i_parent = NULL;
+       node->i_kids = NULL;
+       node->i_rdn = (diskNode *)d;
+       node->i_rdn->rdn.bv_val += (long)d;
+       node->i_rdn->nrdn.bv_val += (long)d;
+       ldap_pvt_thread_rdwr_init(&node->i_kids_rdwr);
+       avl_insert( &bdb->bi_tree, (caddr_t) node,
+                       (AVL_CMP)node_add_cmp, (AVL_DUP) avl_dup_error );
+       if (id == 1)
+               bdb->bi_troot = node;
+       return node;
+}
+
+/* This function initializes the trees at startup time. */
+int bdb_build_tree(
+       Backend *be
+)
+{
+       struct bdb_info *bdb = (struct bdb_info *) be->be_private;
+       int i, rc;
+       DBC *cursor;
+       DBT key, data;
+       ID id;
+       idNode *node;
+       char **rdns;
+
+       bdb->bi_tree = NULL;
+
+       rc = bdb->bi_id2parent->bdi_db->cursor(
+               bdb->bi_id2parent->bdi_db, NULL, &cursor,
+               bdb->bi_db_opflags );
+       if( rc != 0 ) {
+               return NOID;
        }
 
+       /* When be_suffix is turned into struct berval or LDAPDN
+        * life will get a lot easier... Since no DNs live on disk, we
+        * need to operate on the be_suffix to fully qualify our DNs.
+        * We need to know how many components are in the suffix DN,
+        * so we can tell where the suffix ends and our nodes begin.
+        *
+        * Note that this code always uses be_suffix[0], so defining
+        * multiple suffixes for a single backend won't work!
+        */
+       rdns = ldap_explode_dn(be->be_nsuffix[0]->bv_val, 0);
+       for (i=0; rdns[i]; i++);
+       bdb->bi_nrdns = i;
+       charray_free(rdns);
 
-       {
-               char *pdn = dn_parent( NULL, dn );
+       DBTzero( &key );
+       DBTzero( &data );
+       key.data = (char *)&id;
+       key.ulen = sizeof( id );
+       key.flags = DB_DBT_USERMEM;
+       data.flags = DB_DBT_MALLOC;
 
-               if( pdn != NULL ) {
-                       ldbm_datum_init( key );
-                       key.dsize = strlen( pdn ) + 2;
-                       key.dptr = ch_malloc( key.dsize );
-                       sprintf( key.dptr, "%c%s", DN_ONE_PREFIX, pdn );
+       while (cursor->c_get( cursor, &key, &data, DB_NEXT ) == 0) {
+               bdb_add_node( id, data.data, bdb );
+       }
+       cursor->c_close( cursor );
 
-                       (void) idl_delete_key( be, db, key, id );
+       rc = avl_apply(bdb->bi_tree, (AVL_APPLY)bdb_insert_kid, bdb->bi_tree,
+               -1, AVL_INORDER );
 
-                       free( key.dptr );
-                       free( pdn );
-               }
+       return rc;
+}
+
+/* This function constructs a full DN for a given id. We really should
+ * be passing idNodes directly, to save some effort...
+ */
+int bdb_fix_dn(
+       BackendDB *be,
+       ID id,
+       Entry *e
+)
+{
+       struct bdb_info *bdb = (struct bdb_info *) be->be_private;
+       idNode *n, *o;
+       int rlen, nrlen;
+       char *ptr, *nptr;
+       
+       ldap_pvt_thread_rdwr_rlock(&bdb->bi_tree_rdwr);
+       o = bdb_find_id_node(id, bdb->bi_tree);
+       rlen = be->be_suffix[0]->bv_len + 1;
+       nrlen = be->be_nsuffix[0]->bv_len + 1;
+       for (n = o; n && n->i_parent; n=n->i_parent) {
+               rlen += n->i_rdn->rdn.bv_len + 1;
+               nrlen += n->i_rdn->nrdn.bv_len + 1;
+       }
+       e->e_name.bv_len = rlen - 1;
+       e->e_nname.bv_len = nrlen - 1;
+       e->e_name.bv_val = ch_malloc(rlen + nrlen);
+       e->e_nname.bv_val = e->e_name.bv_val + rlen;
+       ptr = e->e_name.bv_val;
+       nptr = e->e_nname.bv_val;
+       for (n = o; n && n->i_parent; n=n->i_parent) {
+               ptr = slap_strcopy(ptr, n->i_rdn->rdn.bv_val);
+               *ptr++ = ',';
+               nptr = slap_strcopy(nptr, n->i_rdn->nrdn.bv_val);
+               *nptr++ = ',';
        }
+       ldap_pvt_thread_rdwr_runlock(&bdb->bi_tree_rdwr);
 
-       {
-               char **subtree = dn_subtree( NULL, dn );
+       strcpy(ptr, be->be_suffix[0]->bv_val);
+       strcpy(nptr, be->be_nsuffix[0]->bv_val);
 
-               if( subtree != NULL ) {
-                       int i;
-                       for( i=0; subtree[i] != NULL; i++ ) {
-                               ldbm_datum_init( key );
-                               key.dsize = strlen( subtree[i] ) + 2;
-                               key.dptr = ch_malloc( key.dsize );
-                               sprintf( key.dptr, "%c%s",
-                                       DN_SUBTREE_PREFIX, subtree[i] );
+       return 0;
+}
 
-                               (void) idl_delete_key( be, db, key, id );
+int
+bdb_dn2id_add(
+       BackendDB       *be,
+       DB_TXN *txn,
+       struct berval   *pdn,
+       Entry           *e )
+{
+       struct bdb_info *bdb = (struct bdb_info *) be->be_private;
+       int             rc, rlen, nrlen;
+       DBT             key, data;
+       DB *db = bdb->bi_id2parent->bdi_db;
+       diskNode *d;
+       idNode *n;
+
+       nrlen = dn_rdnlen( be, &e->e_nname );
+       if (nrlen) {
+               rlen = dn_rdnlen( be, &e->e_name );
+       } else {
+               rlen = 0;
+       }
 
-                               free( key.dptr );
-                       }
+       d = ch_malloc(sizeof(diskNode) + rlen + nrlen + 2);
+       d->rdn.bv_len = rlen;
+       d->nrdn.bv_len = nrlen;
+       d->rdn.bv_val = (char *)(d+1);
+       d->nrdn.bv_val = d->rdn.bv_val + rlen + 1;
+       strncpy(d->rdn.bv_val, e->e_dn, rlen);
+       d->rdn.bv_val[rlen] = '\0';
+       strncpy(d->nrdn.bv_val, e->e_ndn, nrlen);
+       d->nrdn.bv_val[nrlen] = '\0';
+       d->rdn.bv_val -= (long)d;
+       d->nrdn.bv_val -= (long)d;
+
+       if (pdn->bv_len) {
+               bdb_dn2id(be, txn, pdn, &d->parent);
+       } else {
+               d->parent = 0;
+       }
+
+       DBTzero(&key);
+       DBTzero(&data);
+       key.data = &e->e_id;
+       key.size = sizeof(ID);
+       key.flags = DB_DBT_USERMEM;
 
-                       charray_free( subtree );
+       data.data = d;
+       data.size = sizeof(diskNode) + rlen + nrlen + 2;
+       data.flags = DB_DBT_USERMEM;
+
+       rc = db->put( db, txn, &key, &data, DB_NOOVERWRITE );
+
+       if (rc == 0) {
+               ldap_pvt_thread_rdwr_wlock(&bdb->bi_tree_rdwr);
+               n = bdb_add_node( e->e_id, data.data, bdb);
+               ldap_pvt_thread_rdwr_wunlock(&bdb->bi_tree_rdwr);
+
+               if (d->parent) {
+                       ldap_pvt_thread_rdwr_rlock(&bdb->bi_tree_rdwr);
+                       bdb_insert_kid(n, bdb->bi_tree);
+                       ldap_pvt_thread_rdwr_runlock(&bdb->bi_tree_rdwr);
                }
+       } else {
+               free(d);
        }
+       return rc;
+}
 
-       ldbm_datum_init( key );
+int
+bdb_dn2id_delete(
+       BackendDB       *be,
+       DB_TXN *txn,
+       char    *pdn,
+       Entry   *e )
+{
+       struct bdb_info *bdb = (struct bdb_info *) be->be_private;
+       int rc;
+       DBT             key;
+       DB *db = bdb->bi_id2parent->bdi_db;
+       idNode *n;
+
+       DBTzero(&key);
+       key.size = sizeof(e->e_id);
+       key.data = &e->e_id;
+
+       rc = db->del( db, txn, &key, 0);
+
+       ldap_pvt_thread_rdwr_wlock(&bdb->bi_tree_rdwr);
+       n = avl_delete(&bdb->bi_tree, (void *)e->e_id, (AVL_CMP)node_find_cmp);
+       if (n) {
+               if (n->i_parent) {
+                       ldap_pvt_thread_rdwr_wlock(&n->i_parent->i_kids_rdwr);
+                       avl_delete(&n->i_parent->i_kids, n->i_rdn->nrdn.bv_val,
+                               (AVL_CMP)node_frdn_cmp);
+                       ldap_pvt_thread_rdwr_wunlock(&n->i_parent->i_kids_rdwr);
+               }
+               free(n->i_rdn);
+               ldap_pvt_thread_rdwr_destroy(&n->i_kids_rdwr);
+               free(n);
+       }
+       if (e->e_id == 1)
+               bdb->bi_troot = NULL;
+       ldap_pvt_thread_rdwr_wunlock(&bdb->bi_tree_rdwr);
 
-       key.dsize = strlen( dn ) + 2;
-       key.dptr = ch_malloc( key.dsize );
-       sprintf( key.dptr, "%c%s", DN_BASE_PREFIX, dn );
+       return rc;
+}
 
-       rc = ldbm_cache_delete( db, key );
+int
+bdb_dn2id_matched(
+       BackendDB       *be,
+       DB_TXN *txn,
+       struct berval   *in,
+       ID *id,
+       ID *id2 )
+{
+       struct bdb_info *bdb = (struct bdb_info *) be->be_private;
+       int             i;
+       char            **rdns;
+       idNode *n, *p;
 
-       free( key.dptr );
+       if (!bdb->bi_troot)
+               return DB_NOTFOUND;
 
-       ldbm_cache_close( be, db );
+       p = bdb->bi_troot;
+       if (be_issuffix(be, in->bv_val)) {
+               *id = p->i_id;
+               return 0;
+       }
 
-       Debug( LDAP_DEBUG_TRACE, "<= dn2id_delete %d\n", rc, 0, 0 );
-       return( rc );
+       rdns = ldap_explode_dn(in->bv_val, 0);
+       for (i=0; rdns[i]; i++);
+       i -= bdb->bi_nrdns;
+       if (i < 0) {
+               charray_free(rdns);
+               return -1;
+       }
+       n = p;
+       ldap_pvt_thread_rdwr_rlock(&bdb->bi_tree_rdwr);
+       for (--i; i>=0; i--) {
+               ldap_pvt_thread_rdwr_rlock(&p->i_kids_rdwr);
+               n = bdb_find_rdn_node(rdns[i], p->i_kids);
+               ldap_pvt_thread_rdwr_runlock(&p->i_kids_rdwr);
+               if (!n) break;
+               p = n;
+       }
+       ldap_pvt_thread_rdwr_runlock(&bdb->bi_tree_rdwr);
+       charray_free(rdns);
+
+       if (n) {
+               *id = n->i_id;
+       } else if (id2) {
+               *id2 = p->i_id;
+       }
+       return n ? 0 : DB_NOTFOUND;
 }
 
-/*
- * dn2entry - look up dn in the cache/indexes and return the corresponding
- * entry.
- */
+int
+bdb_dn2id(
+       BackendDB       *be,
+       DB_TXN *txn,
+       struct berval   *dn,
+       ID *id )
+{
+       return bdb_dn2id_matched(be, txn, dn, id, NULL);
+}
 
-Entry *
-dn2entry_rw(
-    Backend    *be,
-    const char *dn,
-    Entry      **matched,
-    int         rw
-)
+int
+bdb_dn2id_children(
+       BackendDB       *be,
+       DB_TXN *txn,
+       struct berval   *dn )
 {
+       int             rc;
+       struct bdb_info *bdb = (struct bdb_info *) be->be_private;
        ID              id;
-       Entry           *e = NULL;
-       char            *pdn;
+       idNode *n;
 
-       Debug(LDAP_DEBUG_TRACE, "dn2entry_%s: dn: \"%s\"\n",
-               rw ? "w" : "r", dn, 0);
+       rc = bdb_dn2id(be, txn, dn, &id);
+       if (rc != 0)
+               return rc;
 
-       if( matched != NULL ) {
-               /* caller cares about match */
-               *matched = NULL;
-       }
+       ldap_pvt_thread_rdwr_rlock(&bdb->bi_tree_rdwr);
+       n = bdb_find_id_node(id, bdb->bi_tree);
+       ldap_pvt_thread_rdwr_runlock(&bdb->bi_tree_rdwr);
 
-       if ( (id = dn2id( be, dn )) != NOID &&
-               (e = id2entry_rw( be, id, rw )) != NULL )
-       {
-               return( e );
-       }
+       if (!n->i_kids)
+               return DB_NOTFOUND;
+       else
+               return 0;
+}
 
-       if ( id != NOID ) {
-               Debug(LDAP_DEBUG_ANY,
-                       "dn2entry_%s: no entry for valid id (%ld), dn \"%s\"\n",
-                       rw ? "w" : "r", id, dn);
-               /* must have been deleted from underneath us */
-               /* treat as if NOID was found */
+/* Since we don't store IDLs for onelevel or subtree, we have to construct
+ * them on the fly... Perhaps the i_kids tree ought to just be an IDL?
+ */
+static int
+insert_one(
+       idNode *n,
+       ID *ids
+)
+{
+       return bdb_idl_insert(ids, n->i_id);
+}
+
+static int
+insert_sub(
+       idNode *n,
+       ID *ids
+)
+{
+       int rc;
+
+       rc = bdb_idl_insert(ids, n->i_id);
+       if (rc == 0) {
+               ldap_pvt_thread_rdwr_rlock(&n->i_kids_rdwr);
+               rc = avl_apply(n->i_kids, (AVL_APPLY)insert_sub, ids, -1,
+                       AVL_INORDER);
+               ldap_pvt_thread_rdwr_runlock(&n->i_kids_rdwr);
        }
+       return rc;
+}
 
-       /* caller doesn't care about match */
-       if( matched == NULL ) return NULL;
+int
+bdb_dn2idl(
+       BackendDB       *be,
+       struct berval   *dn,
+       int prefix,
+       ID *ids )
+{
+       struct bdb_info *bdb = (struct bdb_info *) be->be_private;
+       int             rc;
+       ID              id;
+       idNode          *n;
 
-       /* entry does not exist - see how much of the dn does exist */
-       /* dn_parent checks returns NULL if dn is suffix */
-       if ( (pdn = dn_parent( be, dn )) != NULL ) {
-               /* get entry with reader lock */
-               if ( (e = dn2entry_r( be, pdn, matched )) != NULL ) {
-                       *matched = e;
-               }
-               free( pdn );
+       if (prefix == DN_SUBTREE_PREFIX && be_issuffix(be, dn->bv_val)) {
+               BDB_IDL_ALL(bdb, ids);
+               return 0;
        }
 
-       return NULL;
+       rc = bdb_dn2id(be, NULL, dn, &id);
+       if (rc) return rc;
+
+       ldap_pvt_thread_rdwr_rlock(&bdb->bi_tree_rdwr);
+       n = bdb_find_id_node(id, bdb->bi_tree);
+       ldap_pvt_thread_rdwr_runlock(&bdb->bi_tree_rdwr);
+
+       ids[0] = 0;
+       ldap_pvt_thread_rdwr_rlock(&n->i_kids_rdwr);
+       if (prefix == DN_ONE_PREFIX) {
+               rc = avl_apply(n->i_kids, (AVL_APPLY)insert_one, ids, -1,
+                       AVL_INORDER);
+       } else {
+               rc = avl_apply(n->i_kids, (AVL_APPLY)insert_sub, ids, -1,
+                       AVL_INORDER);
+       }
+       ldap_pvt_thread_rdwr_runlock(&n->i_kids_rdwr);
+       return rc;
 }
-
-#endif
\ No newline at end of file
+#endif /* BDB_HIER */