]> git.sur5r.net Git - openldap/blobdiff - servers/slapd/back-bdb/init.c
Unify use of BDB lockers
[openldap] / servers / slapd / back-bdb / init.c
index 8f35fec6e60b83b5569562ac1df7f9c8ae20214f..135f101702bc0a0e9d214116b12d8fdb52b56340 100644 (file)
@@ -2,7 +2,7 @@
 /* $OpenLDAP$ */
 /* This work is part of OpenLDAP Software <http://www.openldap.org/>.
  *
- * Copyright 2000-2006 The OpenLDAP Foundation.
+ * Copyright 2000-2007 The OpenLDAP Foundation.
  * All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
@@ -45,6 +45,7 @@ static int
 bdb_db_init( BackendDB *be )
 {
        struct bdb_info *bdb;
+       int rc;
 
        Debug( LDAP_DEBUG_TRACE,
                LDAP_XSTRING(bdb_db_init) ": Initializing " BDB_UCTYPE " database\n",
@@ -72,6 +73,8 @@ bdb_db_init( BackendDB *be )
 #endif
        ldap_pvt_thread_mutex_init( &bdb->bi_cache.lru_head_mutex );
        ldap_pvt_thread_mutex_init( &bdb->bi_cache.lru_tail_mutex );
+       ldap_pvt_thread_mutex_init( &bdb->bi_cache.c_count_mutex );
+       ldap_pvt_thread_mutex_init( &bdb->bi_cache.c_eifree_mutex );
        ldap_pvt_thread_mutex_init( &bdb->bi_cache.c_dntree.bei_kids_mutex );
        ldap_pvt_thread_rdwr_init ( &bdb->bi_cache.c_rwlock );
        ldap_pvt_thread_rdwr_init( &bdb->bi_idl_tree_rwlock );
@@ -80,7 +83,9 @@ bdb_db_init( BackendDB *be )
        be->be_private = bdb;
        be->be_cf_ocs = be->bd_info->bi_cf_ocs;
 
-       return 0;
+       rc = bdb_monitor_db_init( be );
+
+       return rc;
 }
 
 static int
@@ -95,7 +100,8 @@ bdb_db_open( BackendDB *be )
        u_int32_t flags;
        char path[MAXPATHLEN];
        char *dbhome;
-       int do_recover = 0, do_alock_recover = 0, open_env = 1, got_env = 0;
+       int do_recover = 0, do_alock_recover = 0;
+       int alockt, quick = 0;
 
        if ( be->be_suffix == NULL ) {
                Debug( LDAP_DEBUG_ANY,
@@ -126,11 +132,19 @@ bdb_db_open( BackendDB *be )
        }
 
        /* Perform database use arbitration/recovery logic */
+       alockt = (slapMode & SLAP_TOOL_READONLY) ? ALOCK_LOCKED : ALOCK_UNIQUE;
+       if ( slapMode & SLAP_TOOL_QUICK ) {
+               alockt |= ALOCK_NOSAVE;
+               quick = 1;
+       }
+
        rc = alock_open( &bdb->bi_alock_info, 
                                "slapd", 
-                               bdb->bi_dbenv_home,
-                               slapMode & SLAP_TOOL_READONLY ?
-                               ALOCK_LOCKED : ALOCK_UNIQUE );
+                               bdb->bi_dbenv_home, alockt );
+
+       /* alockt is TRUE if the existing environment was created in Quick mode */
+       alockt = (rc & ALOCK_NOSAVE) ? 1 : 0;
+       rc &= ~ALOCK_NOSAVE;
 
        if( rc == ALOCK_RECOVER ) {
                Debug( LDAP_DEBUG_ANY,
@@ -138,7 +152,7 @@ bdb_db_open( BackendDB *be )
                        " attempting recovery.\n", 
                        0, 0, 0 );
                do_alock_recover = 1;
-               do_recover = 1;
+               do_recover = DB_RECOVER;
        } else if( rc == ALOCK_BUSY ) {
                Debug( LDAP_DEBUG_ANY,
                        "bdb_db_open: database already in use\n", 
@@ -167,7 +181,7 @@ bdb_db_open( BackendDB *be )
                                                "bdb_db_open: DB_CONFIG for suffix %s has changed.\n"
                                                "Performing database recovery to activate new settings.\n",
                                                be->be_suffix[0].bv_val, 0, 0 );
-                                       do_recover = 1;
+                                       do_recover = DB_RECOVER;
                                }
                        }
                }
@@ -180,6 +194,28 @@ bdb_db_open( BackendDB *be )
                        bdb->bi_dbenv_home, errno, be->be_suffix[0].bv_val );
        }
 
+       /* Always let slapcat run, regardless of environment state.
+        * This can be used to cause a cache flush after an unclean
+        * shutdown.
+        */
+       if ( do_recover && ( slapMode & SLAP_TOOL_READONLY )) {
+               Debug( LDAP_DEBUG_ANY,
+                       "bdb_db_open: Recovery skipped in read-only mode. "
+                       "Run manual recovery if errors are encountered.\n",
+                       0, 0, 0 );
+               do_recover = 0;
+               quick = alockt;
+       }
+
+       /* An existing environment in Quick mode has nothing to recover. */
+       if ( alockt && do_recover ) {
+               Debug( LDAP_DEBUG_ANY,
+                       "bdb_db_open: cannot recover, database must be reinitialized.\n", 
+                       0, 0, 0 );
+               rc = -1;
+               goto fail;
+       }
+
        rc = db_env_create( &bdb->bi_dbenv, 0 );
        if( rc != 0 ) {
                Debug( LDAP_DEBUG_ANY,
@@ -188,6 +224,35 @@ bdb_db_open( BackendDB *be )
                goto fail;
        }
 
+#ifdef HAVE_EBCDIC
+       strcpy( path, bdb->bi_dbenv_home );
+       __atoe( path );
+       dbhome = path;
+#else
+       dbhome = bdb->bi_dbenv_home;
+#endif
+
+       /* If existing environment is clean but doesn't support
+        * currently requested modes, remove it.
+        */
+       if ( !do_recover && ( alockt ^ quick )) {
+               rc = bdb->bi_dbenv->remove( bdb->bi_dbenv, dbhome, DB_FORCE );
+               if ( rc ) {
+                       Debug( LDAP_DEBUG_ANY,
+                               "bdb_db_open: dbenv remove failed: %s (%d)\n",
+                               db_strerror(rc), rc, 0 );
+                       bdb->bi_dbenv = NULL;
+                       goto fail;
+               }
+               rc = db_env_create( &bdb->bi_dbenv, 0 );
+               if( rc != 0 ) {
+                       Debug( LDAP_DEBUG_ANY,
+                               "bdb_db_open: db_env_create failed: %s (%d)\n",
+                               db_strerror(rc), rc, 0 );
+                       goto fail;
+               }
+       }
+
        bdb->bi_dbenv->set_errpfx( bdb->bi_dbenv, be->be_suffix[0].bv_val );
        bdb->bi_dbenv->set_errcall( bdb->bi_dbenv, bdb_errcall );
 
@@ -209,189 +274,29 @@ bdb_db_open( BackendDB *be )
 
 #define        BDB_TXN_FLAGS   (DB_INIT_LOCK | DB_INIT_LOG | DB_INIT_TXN)
 
-#ifdef HAVE_EBCDIC
-       strcpy( path, bdb->bi_dbenv_home );
-       __atoe( path );
-       dbhome = path;
-#else
-       dbhome = bdb->bi_dbenv_home;
-#endif
-
        Debug( LDAP_DEBUG_TRACE,
                "bdb_db_open: dbenv_open(%s)\n",
                bdb->bi_dbenv_home, 0, 0);
 
-       /* Check if there is a usable existing environment */
-       flags = DB_JOINENV | DB_THREAD;
-
-       rc = bdb->bi_dbenv->open( bdb->bi_dbenv, dbhome,
-               flags, bdb->bi_dbenv_mode );
-       if( rc == 0 ) {
-               int flags_ok = 0;
-
-               got_env = 1;
-
-               rc = bdb->bi_dbenv->get_open_flags( bdb->bi_dbenv, &flags );
-               if ( rc == 0 ) {
-                       int flag2 = flags & BDB_TXN_FLAGS;
-
-                       /* In quick mode, none of these flags are allowed */
-                       if ( slapMode & SLAP_TOOL_QUICK ) {
-                               if ( !flag2 )
-                                       flags_ok = 1;
-                       } else {
-                       /* In normal mode, all of these flags are required */
-                               if ( flag2 == BDB_TXN_FLAGS )
-                                       flags_ok = 1;
-                       }
-               }
-
-               /* In Quick mode, we cannot Recover... */
-               if ( slapMode & SLAP_TOOL_QUICK ) {
-                       /* If we need to recover but we had no TXNs, just fail */
-                       if ( do_recover && flags_ok ) {
-                               Debug( LDAP_DEBUG_ANY,
-                                       "bdb_db_open: Database cannot be recovered. "
-                                       "Restore from backup!\n", 0, 0, 0);
-                               rc = -1;
-                               goto fail;
-                       }
-                       /* We need to recover, and we had TXN support before:
-                        * Close this env, open a new one with recovery flags.
-                        */
-                       if ( do_recover ) {
-                               bdb->bi_dbenv->close( bdb->bi_dbenv, 0 );
-                               bdb->bi_dbenv = NULL;
-                               rc = db_env_create( &bdb->bi_dbenv, 0 );
-                               if( rc != 0 ) {
-                                       Debug( LDAP_DEBUG_ANY,
-                                               "bdb_db_open: db_env_create failed: %s (%d)\n",
-                                               db_strerror(rc), rc, 0 );
-                                       goto fail;
-                               }
-                               bdb->bi_dbenv->set_errpfx( bdb->bi_dbenv,
-                                       be->be_suffix[0].bv_val );
-                               bdb->bi_dbenv->set_errcall( bdb->bi_dbenv, bdb_errcall );
-                               rc = bdb->bi_dbenv->open( bdb->bi_dbenv, dbhome,
-                                       flags | DB_RECOVER, bdb->bi_dbenv_mode );
-                               if( rc != 0 ) {
-                                       Debug( LDAP_DEBUG_ANY,
-                                               "bdb_db_open: recovery failed: %s (%d)\n",
-                                               db_strerror(rc), rc, 0 );
-                                       goto fail;
-                               }
-                               do_recover = 0;
-                       }
-                       /* Prev environment had TXN support, get rid of it */
-                       if ( !flags_ok ) {
-                               bdb->bi_dbenv->close( bdb->bi_dbenv, 0 );
-                               bdb->bi_dbenv = NULL;
-                               rc = db_env_create( &bdb->bi_dbenv, 0 );
-                               if( rc != 0 ) {
-                                       Debug( LDAP_DEBUG_ANY,
-                                               "bdb_db_open: db_env_create failed: %s (%d)\n",
-                                               db_strerror(rc), rc, 0 );
-                                       goto fail;
-                               }
-                               bdb->bi_dbenv->remove( bdb->bi_dbenv, dbhome, 0 );
-                               bdb->bi_dbenv = NULL;
-                       }
-               /* Normal TXN mode */
-               } else {
-                       /* If we need to recover but we had no TXNs, just fail */
-                       if ( do_recover && !flags_ok ) {
-                               Debug( LDAP_DEBUG_ANY,
-                                       "bdb_db_open: Database cannot be recovered. "
-                                       "Restore from backup!\n", 0, 0, 0);
-                               rc = -1;
-                               goto fail;
-                       }
-                       /* Prev environment had no TXN support, close it */
-                       if ( !flags_ok ) {
-                               bdb->bi_dbenv->close( bdb->bi_dbenv, 0 );
-                               bdb->bi_dbenv = NULL;
-                               do_recover = 1;
-                       }
-               }
-
-               if ( flags_ok && !do_recover ) {
-                       /* This environment is fine, don't reopen it */
-                       open_env = 0;
-               } else {
-                       /* Create a new env that can take the desired settings */
-                       if ( bdb->bi_dbenv != NULL ) {
-                               bdb->bi_dbenv->close( bdb->bi_dbenv, 0 );
-                               bdb->bi_dbenv = NULL;
-                       }
-                       rc = db_env_create( &bdb->bi_dbenv, 0 );
-                       if( rc != 0 ) {
-                               Debug( LDAP_DEBUG_ANY,
-                                       "bdb_db_open: db_env_create failed: %s (%d)\n",
-                                       db_strerror(rc), rc, 0 );
-                               goto fail;
-                       }
-
-                       bdb->bi_dbenv->set_errpfx( bdb->bi_dbenv, be->be_suffix[0].bv_val );
-                       bdb->bi_dbenv->set_errcall( bdb->bi_dbenv, bdb_errcall );
-                       bdb->bi_dbenv->set_lk_detect( bdb->bi_dbenv, bdb->bi_lock_detect );
+       flags = DB_INIT_MPOOL | DB_CREATE | DB_THREAD;
 
-                       /* One long-lived TXN per thread, two TXNs per write op */
-                       bdb->bi_dbenv->set_tx_max( bdb->bi_dbenv, connection_pool_max * 3 );
-
-                       if( bdb->bi_dbenv_xflags != 0 ) {
-                               rc = bdb->bi_dbenv->set_flags( bdb->bi_dbenv,
-                                       bdb->bi_dbenv_xflags, 1);
-                               if( rc != 0 ) {
-                                       Debug( LDAP_DEBUG_ANY,
-                                               "bdb_db_open: dbenv_set_flags failed: %s (%d)\n",
-                                               db_strerror(rc), rc, 0 );
-                                       goto fail;
-                               }
-                       }
-               }
-       }
+       if ( !quick )
+               flags |= BDB_TXN_FLAGS;
 
-       /* If we need to recover but there was no existing environment,
-        * then we assume that someone has already manually recovered using
-        * db_recover. Just ignore it.
-        */
-       if ( do_recover && !got_env ) {
-               do_recover = 0;
-               Debug( LDAP_DEBUG_TRACE,
-                       "bdb_db_open: Recovery needed but environment is missing - "
-                       "assuming recovery was done manually...\n", 0, 0, 0 );
+       /* If a key was set, use shared memory for the BDB environment */
+       if ( bdb->bi_shm_key ) {
+               bdb->bi_dbenv->set_shm_key( bdb->bi_dbenv, bdb->bi_shm_key );
+               flags |= DB_SYSTEM_MEM;
        }
+       rc = bdb->bi_dbenv->open( bdb->bi_dbenv, dbhome,
+                       flags | do_recover, bdb->bi_dbenv_mode );
 
-       if ( open_env ) {
-               flags = DB_INIT_MPOOL | DB_THREAD | DB_CREATE;
-               if ( !( slapMode & SLAP_TOOL_QUICK ))
-                       flags |= BDB_TXN_FLAGS;
-
-               if ( do_recover ) {
-                       if ( slapMode & SLAP_TOOL_READONLY ) {
-                               Debug( LDAP_DEBUG_ANY,
-                                       "bdb_db_open: Recovery skipped in read-only mode. "
-                                       "Run manual recovery if errors are encountered.\n",
-                                       0, 0, 0 );
-                       } else {
-                               flags |= DB_RECOVER;
-                       }
-               }
-
-               /* If a key was set, use shared memory for the BDB environment */
-               if ( bdb->bi_shm_key ) {
-                       bdb->bi_dbenv->set_shm_key( bdb->bi_dbenv, bdb->bi_shm_key );
-                       flags |= DB_SYSTEM_MEM;
-               }
-
-               rc = bdb->bi_dbenv->open( bdb->bi_dbenv, dbhome,
-                       flags, bdb->bi_dbenv_mode );
-               if( rc != 0 ) {
-                       Debug( LDAP_DEBUG_ANY,
-                               "bdb_db_open: dbenv_open failed: %s (%d)\n",
-                               db_strerror(rc), rc, 0 );
-                       goto fail;
-               }
+       if ( rc ) {
+               Debug( LDAP_DEBUG_ANY,
+                       "bdb_db_open: Database cannot be %s, err %d. "
+                       "Restore from backup!\n",
+                               do_recover ? "recovered" : "opened", rc, 0);
+               goto fail;
        }
 
        if ( do_alock_recover && alock_recover (&bdb->bi_alock_info) != 0 ) {
@@ -405,10 +310,8 @@ bdb_db_open( BackendDB *be )
 #ifdef SLAP_ZONE_ALLOC
        if ( bdb->bi_cache.c_maxsize ) {
                bdb->bi_cache.c_zctx = slap_zn_mem_create(
-                                                               SLAP_ZONE_INITSIZE,
-                                                               SLAP_ZONE_MAXSIZE,
-                                                               SLAP_ZONE_DELTA,
-                                                               SLAP_ZONE_SIZE);
+                       SLAP_ZONE_INITSIZE, SLAP_ZONE_MAXSIZE,
+                       SLAP_ZONE_DELTA, SLAP_ZONE_SIZE);
        }
 #endif
 
@@ -420,7 +323,7 @@ bdb_db_open( BackendDB *be )
        flags = DB_THREAD | bdb->bi_db_opflags;
 
 #ifdef DB_AUTO_COMMIT
-       if ( !( slapMode & SLAP_TOOL_QUICK ))
+       if ( !quick )
                flags |= DB_AUTO_COMMIT;
 #endif
 
@@ -520,12 +423,16 @@ bdb_db_open( BackendDB *be )
                goto fail;
        }
 
-       if ( !( slapMode & SLAP_TOOL_QUICK )) {
-               XLOCK_ID(bdb->bi_dbenv, &bdb->bi_cache.c_locker);
+       /* monitor setup */
+       rc = bdb_monitor_db_open( be );
+       if ( rc != 0 ) {
+               goto fail;
        }
 
        bdb->bi_flags |= BDB_IS_OPEN;
 
+       entry_prealloc( bdb->bi_cache.c_maxsize );
+       attr_prealloc( bdb->bi_cache.c_maxsize * 20 );
        return 0;
 
 fail:
@@ -541,6 +448,9 @@ bdb_db_close( BackendDB *be )
        struct bdb_db_info *db;
        bdb_idl_cache_entry_t *entry, *next_entry;
 
+       /* monitor handling */
+       (void)bdb_monitor_db_close( be );
+
        bdb->bi_flags &= ~BDB_IS_OPEN;
 
        ber_bvarray_free( bdb->bi_db_config );
@@ -559,29 +469,23 @@ bdb_db_close( BackendDB *be )
 
        bdb_cache_release_all (&bdb->bi_cache);
 
-       if ( bdb->bi_idl_cache_max_size ) {
+       if ( bdb->bi_idl_cache_size ) {
                avl_free( bdb->bi_idl_tree, NULL );
                bdb->bi_idl_tree = NULL;
                entry = bdb->bi_idl_lru_head;
-               while ( entry != NULL ) {
+               do {
                        next_entry = entry->idl_lru_next;
                        if ( entry->idl )
                                free( entry->idl );
                        free( entry->kstr.bv_val );
                        free( entry );
                        entry = next_entry;
-               }
+               } while ( entry != bdb->bi_idl_lru_head );
                bdb->bi_idl_lru_head = bdb->bi_idl_lru_tail = NULL;
        }
 
        /* close db environment */
        if( bdb->bi_dbenv ) {
-               /* Free cache locker if we enabled locking */
-               if ( !( slapMode & SLAP_TOOL_QUICK )) {
-                       XLOCK_ID_FREE(bdb->bi_dbenv, bdb->bi_cache.c_locker);
-                       bdb->bi_cache.c_locker = 0;
-               }
-
                /* force a checkpoint, but not if we were ReadOnly,
                 * and not in Quick mode since there are no transactions there.
                 */
@@ -619,6 +523,9 @@ bdb_db_destroy( BackendDB *be )
 {
        struct bdb_info *bdb = (struct bdb_info *) be->be_private;
 
+       /* monitor handling */
+       (void)bdb_monitor_db_destroy( be );
+
        if( bdb->bi_dbenv_home ) ch_free( bdb->bi_dbenv_home );
        if( bdb->bi_db_config_path ) ch_free( bdb->bi_db_config_path );
 
@@ -627,6 +534,8 @@ bdb_db_destroy( BackendDB *be )
        ldap_pvt_thread_rdwr_destroy ( &bdb->bi_cache.c_rwlock );
        ldap_pvt_thread_mutex_destroy( &bdb->bi_cache.lru_head_mutex );
        ldap_pvt_thread_mutex_destroy( &bdb->bi_cache.lru_tail_mutex );
+       ldap_pvt_thread_mutex_destroy( &bdb->bi_cache.c_count_mutex );
+       ldap_pvt_thread_mutex_destroy( &bdb->bi_cache.c_eifree_mutex );
        ldap_pvt_thread_mutex_destroy( &bdb->bi_cache.c_dntree.bei_kids_mutex );
 #ifdef BDB_HIER
        ldap_pvt_thread_mutex_destroy( &bdb->bi_modrdns_mutex );
@@ -653,8 +562,13 @@ bdb_back_initialize(
                LDAP_CONTROL_MANAGEDSAIT,
                LDAP_CONTROL_NOOP,
                LDAP_CONTROL_PAGEDRESULTS,
+               LDAP_CONTROL_PRE_READ,
+               LDAP_CONTROL_POST_READ,
                LDAP_CONTROL_SUBENTRIES,
                LDAP_CONTROL_X_PERMISSIVE_MODIFY,
+#ifdef LDAP_X_TXN
+               LDAP_CONTROL_X_TXN_SPEC,
+#endif
                NULL
        };