Updated for mdb_txn_begin API change.

[openldap] / servers / slapd / back-mdb / tools.c

diff --git a/servers/slapd/back-mdb/tools.c b/servers/slapd/back-mdb/tools.c
index 75a2951f5993418c2fc384309b97793e74eb7364..08dc01d9affd5e1e2d72e10ff38689a97ffcd87a 100644 (file)
--- a/servers/slapd/back-mdb/tools.c
+++ b/servers/slapd/back-mdb/tools.c
@@ -154,7 +154,7 @@ int mdb_tool_entry_close(
                nholes = 0;
                return -1;
        }
-                       
+
        return 0;
 }
 
@@ -168,7 +168,7 @@ mdb_tool_entry_first_x(
        tool_base = base;
        tool_scope = scope;
        tool_filter = f;
-       
+
        return mdb_tool_entry_next( be );
 }
 
@@ -186,7 +186,7 @@ ID mdb_tool_entry_next(
        assert( mdb != NULL );
 
        if ( !txn ) {
-               rc = mdb_txn_begin( mdb->mi_dbenv, MDB_RDONLY, &txn );
+               rc = mdb_txn_begin( mdb->mi_dbenv, NULL, MDB_RDONLY, &txn );
                if ( rc )
                        return NOID;
                rc = mdb_cursor_open( txn, mdb->mi_id2entry, &cursor );
@@ -255,7 +255,7 @@ ID mdb_tool_dn2id_get(
        mdb = (struct mdb_info *) be->be_private;
 
        if ( !txn ) {
-               rc = mdb_txn_begin( mdb->mi_dbenv, (slapMode & SLAP_TOOL_READONLY) != 0 ?
+               rc = mdb_txn_begin( mdb->mi_dbenv, NULL, (slapMode & SLAP_TOOL_READONLY) != 0 ?
                        MDB_RDONLY : 0, &txn );
                if ( rc )
                        return NOID;
@@ -269,7 +269,7 @@ ID mdb_tool_dn2id_get(
        rc = mdb_dn2id( &op, txn, dn, &id, NULL, NULL );
        if ( rc == MDB_NOTFOUND )
                return NOID;
-       
+
        return id;
 }
 
@@ -310,7 +310,7 @@ mdb_tool_entry_get_int( BackendDB *be, ID id, Entry **ep )
                rc = mdb_id2name( &op, txn, &idcursor, id, &dn, &ndn );
                if ( rc  ) {
                        rc = LDAP_OTHER;
-                       mdb_entry_return( e );
+                       mdb_entry_return( &op, e );
                        e = NULL;
                        goto done;
                }
@@ -356,6 +356,7 @@ static int mdb_tool_next_id(
        struct berval *text,
        int hole )
 {
+       struct mdb_info *mdb = (struct mdb_info *) op->o_bd->be_private;
        struct berval dn = e->e_name;
        struct berval ndn = e->e_nname;
        struct berval pdn, npdn, nmatched;
@@ -404,12 +405,13 @@ static int mdb_tool_next_id(
                }
                rc = mdb_dn2id_add( op, tid, pid, e );
                if ( rc ) {
-                       snprintf( text->bv_val, text->bv_len, 
+                       snprintf( text->bv_val, text->bv_len,
                                "dn2id_add failed: %s (%d)",
                                mdb_strerror(rc), rc );
-               Debug( LDAP_DEBUG_ANY,
-                       "=> mdb_tool_next_id: %s\n", text->bv_val, 0, 0 );
+                       Debug( LDAP_DEBUG_ANY,
+                               "=> mdb_tool_next_id: %s\n", text->bv_val, 0, 0 );
                } else if ( hole ) {
+                       MDB_val key, data;
                        if ( nholes == nhmax - 1 ) {
                                if ( holes == hbuf ) {
                                        holes = ch_malloc( nhmax * sizeof(dn_id) * 2 );
@@ -421,6 +423,20 @@ static int mdb_tool_next_id(
                        }
                        ber_dupbv( &holes[nholes].dn, &ndn );
                        holes[nholes++].id = e->e_id;
+                       key.mv_size = sizeof(ID);
+                       key.mv_data = &e->e_id;
+                       data.mv_size = 0;
+                       data.mv_data = NULL;
+                       rc = mdb_put( tid, mdb->mi_id2entry, &key, &data, MDB_NOOVERWRITE );
+                       if ( rc == MDB_KEYEXIST )
+                               rc = 0;
+                       if ( rc ) {
+                               snprintf( text->bv_val, text->bv_len,
+                                       "dummy id2entry add failed: %s (%d)",
+                                       mdb_strerror(rc), rc );
+                               Debug( LDAP_DEBUG_ANY,
+                                       "=> mdb_tool_next_id: %s\n", text->bv_val, 0, 0 );
+                       }
                }
        } else if ( !hole ) {
                unsigned i, j;
@@ -458,12 +474,12 @@ mdb_tool_index_add(
                IndexRec *ir;
                int i, rc;
                Attribute *a;
-               
+
                ir = mdb_tool_index_rec;
                memset(ir, 0, mdb->bi_nattrs * sizeof( IndexRec ));
 
                for ( a = e->e_attrs; a != NULL; a = a->a_next ) {
-                       rc = mdb_index_recset( mdb, a, a->a_desc->ad_type, 
+                       rc = mdb_index_recset( mdb, a, a->a_desc->ad_type,
                                &a->a_desc->ad_tags, ir );
                        if ( rc )
                                return rc;
@@ -473,7 +489,7 @@ mdb_tool_index_add(
                ldap_pvt_thread_mutex_lock( &mdb_tool_index_mutex );
                /* Wait for all threads to be ready */
                while ( mdb_tool_index_tcount ) {
-                       ldap_pvt_thread_cond_wait( &mdb_tool_index_cond_main, 
+                       ldap_pvt_thread_cond_wait( &mdb_tool_index_cond_main,
                                &mdb_tool_index_mutex );
                }
                for ( i=1; i<slap_tool_thread_max; i++ )
@@ -487,7 +503,7 @@ mdb_tool_index_add(
                ldap_pvt_thread_mutex_lock( &mdb_tool_index_mutex );
                for ( i=1; i<slap_tool_thread_max; i++ ) {
                        if ( mdb_tool_index_threads[i] == LDAP_BUSY ) {
-                               ldap_pvt_thread_cond_wait( &mdb_tool_index_cond_main, 
+                               ldap_pvt_thread_cond_wait( &mdb_tool_index_cond_main,
                                        &mdb_tool_index_mutex );
                                i--;
                                continue;
@@ -529,7 +545,7 @@ ID mdb_tool_entry_put(
        mdb = (struct mdb_info *) be->be_private;
 
        if ( !txn ) {
-       rc = mdb_txn_begin( mdb->mi_dbenv, 0, &txn );
+       rc = mdb_txn_begin( mdb->mi_dbenv, NULL, 0, &txn );
        if( rc != 0 ) {
                snprintf( text->bv_val, text->bv_len,
                        "txn_begin failed: %s (%d)",
@@ -555,9 +571,7 @@ ID mdb_tool_entry_put(
        rc = mdb_tool_index_add( &op, txn, e );
        if( rc != 0 ) {
                snprintf( text->bv_val, text->bv_len,
-                               "index_entry_add failed: %s (%d)",
-                               rc == LDAP_OTHER ? "Internal error" :
-                               mdb_strerror(rc), rc );
+                               "index_entry_add failed: err=%d", rc );
                Debug( LDAP_DEBUG_ANY,
                        "=> " LDAP_XSTRING(mdb_tool_entry_put) ": %s\n",
                        text->bv_val, 0, 0 );
@@ -569,8 +583,7 @@ ID mdb_tool_entry_put(
        rc = mdb_id2entry_add( &op, txn, e );
        if( rc != 0 ) {
                snprintf( text->bv_val, text->bv_len,
-                               "id2entry_add failed: %s (%d)",
-                               mdb_strerror(rc), rc );
+                               "id2entry_add failed: err=%d", rc );
                Debug( LDAP_DEBUG_ANY,
                        "=> " LDAP_XSTRING(mdb_tool_entry_put) ": %s\n",
                        text->bv_val, 0, 0 );
@@ -676,6 +689,22 @@ int mdb_tool_entry_reindex(
                mi->mi_nattrs = i;
        }
 
+       if ( slapMode & SLAP_TRUNCATE_MODE ) {
+               int i;
+               for ( i=0; i < mi->mi_nattrs; i++ ) {
+                       rc = mdb_drop( txn, mi->mi_attrs[i]->ai_dbi, 0 );
+                       if ( rc ) {
+                               Debug( LDAP_DEBUG_ANY,
+                                       LDAP_XSTRING(mdb_tool_entry_reindex)
+                                       ": (Truncate) mdb_drop(%s) failed: %s (%d)\n",
+                                       mi->mi_attrs[i]->ai_desc->ad_type->sat_cname.bv_val,
+                                       mdb_strerror(rc), rc );
+                               return -1;
+                       }
+               }
+               slapMode ^= SLAP_TRUNCATE_MODE;
+       }
+
        e = mdb_tool_entry_get( be, id );
 
        if( e == NULL ) {
@@ -687,7 +716,7 @@ int mdb_tool_entry_reindex(
        }
 
        if ( !txi ) {
-               rc = mdb_txn_begin( mi->mi_dbenv, 0, &txi );
+               rc = mdb_txn_begin( mi->mi_dbenv, NULL, 0, &txi );
                if( rc != 0 ) {
                        Debug( LDAP_DEBUG_ANY,
                                "=> " LDAP_XSTRING(mdb_tool_entry_reindex) ": "
@@ -696,7 +725,7 @@ int mdb_tool_entry_reindex(
                        goto done;
                }
        }
-       
+
        /*
         * just (re)add them for now
         * assume that some other routine (not yet implemented)
@@ -734,8 +763,8 @@ done:
                mdb_txn_abort( txi );
                Debug( LDAP_DEBUG_ANY,
                        "=> " LDAP_XSTRING(mdb_tool_entry_reindex)
-                       ": txn_aborted! %s (%d)\n",
-                       mdb_strerror(rc), rc, 0 );
+                       ": txn_aborted! err=%d\n",
+                       rc, 0, 0 );
                e->e_id = NOID;
                txi = NULL;
        }
@@ -774,7 +803,7 @@ ID mdb_tool_entry_modify(
                mdb_cursor_close( cursor );
                cursor = NULL;
        }
-       rc = mdb_txn_begin( mdb->mi_dbenv, 0, &tid );
+       rc = mdb_txn_begin( mdb->mi_dbenv, NULL, 0, &tid );
        if( rc != 0 ) {
                snprintf( text->bv_val, text->bv_len,
                        "txn_begin failed: %s (%d)",
@@ -794,8 +823,7 @@ ID mdb_tool_entry_modify(
        rc = mdb_id2entry_update( &op, tid, e );
        if( rc != 0 ) {
                snprintf( text->bv_val, text->bv_len,
-                               "id2entry_add failed: %s (%d)",
-                               mdb_strerror(rc), rc );
+                               "id2entry_update failed: err=%d", rc );
                Debug( LDAP_DEBUG_ANY,
                        "=> " LDAP_XSTRING(mdb_tool_entry_modify) ": %s\n",
                        text->bv_val, 0, 0 );