/* $NetBSD: dbcache.c,v 1.1.1.6.6.1 2019/08/10 06:17:17 martin Exp $ */ /* dbcache.c - manage cache of open databases */ /* $OpenLDAP$ */ /* This work is part of OpenLDAP Software . * * Copyright 2000-2019 The OpenLDAP Foundation. * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted only as authorized by the OpenLDAP * Public License. * * A copy of this license is available in the file LICENSE in the * top-level directory of the distribution or, alternatively, at * . */ #include __RCSID("$NetBSD: dbcache.c,v 1.1.1.6.6.1 2019/08/10 06:17:17 martin Exp $"); #include "portable.h" #include #include #include #include #include #include #include "slap.h" #include "back-bdb.h" #include "lutil_hash.h" #ifdef BDB_INDEX_USE_HASH /* Pass-thru hash function. Since the indexer is already giving us hash * values as keys, we don't need BDB to re-hash them. */ static u_int32_t bdb_db_hash( DB *db, const void *bytes, u_int32_t length ) { u_int32_t ret = 0; unsigned char *dst = (unsigned char *)&ret; const unsigned char *src = (const unsigned char *)bytes; if ( length > sizeof(u_int32_t) ) length = sizeof(u_int32_t); while ( length ) { *dst++ = *src++; length--; } return ret; } #define BDB_INDEXTYPE DB_HASH #else #define BDB_INDEXTYPE DB_BTREE #endif /* If a configured size is found, return it, otherwise return 0 */ int bdb_db_findsize( struct bdb_info *bdb, struct berval *name ) { struct bdb_db_pgsize *bp; int rc; for ( bp = bdb->bi_pagesizes; bp; bp=bp->bdp_next ) { rc = strncmp( name->bv_val, bp->bdp_name.bv_val, name->bv_len ); if ( !rc ) { if ( name->bv_len == bp->bdp_name.bv_len ) return bp->bdp_size; if ( name->bv_len < bp->bdp_name.bv_len && bp->bdp_name.bv_val[name->bv_len] == '.' ) return bp->bdp_size; } } return 0; } int bdb_db_cache( Backend *be, struct berval *name, DB **dbout ) { int i, flags; int rc; struct bdb_info *bdb = (struct bdb_info *) be->be_private; struct bdb_db_info *db; char *file; *dbout = NULL; for( i=BDB_NDB; i < bdb->bi_ndatabases; i++ ) { if( !ber_bvcmp( &bdb->bi_databases[i]->bdi_name, name) ) { *dbout = bdb->bi_databases[i]->bdi_db; return 0; } } ldap_pvt_thread_mutex_lock( &bdb->bi_database_mutex ); /* check again! may have been added by another thread */ for( i=BDB_NDB; i < bdb->bi_ndatabases; i++ ) { if( !ber_bvcmp( &bdb->bi_databases[i]->bdi_name, name) ) { *dbout = bdb->bi_databases[i]->bdi_db; ldap_pvt_thread_mutex_unlock( &bdb->bi_database_mutex ); return 0; } } if( i >= BDB_INDICES ) { ldap_pvt_thread_mutex_unlock( &bdb->bi_database_mutex ); return -1; } db = (struct bdb_db_info *) ch_calloc(1, sizeof(struct bdb_db_info)); ber_dupbv( &db->bdi_name, name ); rc = db_create( &db->bdi_db, bdb->bi_dbenv, 0 ); if( rc != 0 ) { Debug( LDAP_DEBUG_ANY, "bdb_db_cache: db_create(%s) failed: %s (%d)\n", bdb->bi_dbenv_home, db_strerror(rc), rc ); ldap_pvt_thread_mutex_unlock( &bdb->bi_database_mutex ); ch_free( db ); return rc; } if( !BER_BVISNULL( &bdb->bi_db_crypt_key )) { rc = db->bdi_db->set_flags( db->bdi_db, DB_ENCRYPT ); if ( rc ) { Debug( LDAP_DEBUG_ANY, "bdb_db_cache: db set_flags(DB_ENCRYPT)(%s) failed: %s (%d)\n", bdb->bi_dbenv_home, db_strerror(rc), rc ); ldap_pvt_thread_mutex_unlock( &bdb->bi_database_mutex ); db->bdi_db->close( db->bdi_db, 0 ); ch_free( db ); return rc; } } if( bdb->bi_flags & BDB_CHKSUM ) { rc = db->bdi_db->set_flags( db->bdi_db, DB_CHKSUM ); if ( rc ) { Debug( LDAP_DEBUG_ANY, "bdb_db_cache: db set_flags(DB_CHKSUM)(%s) failed: %s (%d)\n", bdb->bi_dbenv_home, db_strerror(rc), rc ); ldap_pvt_thread_mutex_unlock( &bdb->bi_database_mutex ); db->bdi_db->close( db->bdi_db, 0 ); ch_free( db ); return rc; } } /* If no explicit size set, use the FS default */ flags = bdb_db_findsize( bdb, name ); if ( flags ) rc = db->bdi_db->set_pagesize( db->bdi_db, flags ); #ifdef BDB_INDEX_USE_HASH rc = db->bdi_db->set_h_hash( db->bdi_db, bdb_db_hash ); #endif rc = db->bdi_db->set_flags( db->bdi_db, DB_DUP | DB_DUPSORT ); file = ch_malloc( db->bdi_name.bv_len + sizeof(BDB_SUFFIX) ); strcpy( file, db->bdi_name.bv_val ); strcpy( file+db->bdi_name.bv_len, BDB_SUFFIX ); #ifdef HAVE_EBCDIC __atoe( file ); #endif flags = DB_CREATE | DB_THREAD; #ifdef DB_AUTO_COMMIT if ( !( slapMode & SLAP_TOOL_QUICK )) flags |= DB_AUTO_COMMIT; #endif /* Cannot Truncate when Transactions are in use */ if ( (slapMode & (SLAP_TOOL_QUICK|SLAP_TRUNCATE_MODE)) == (SLAP_TOOL_QUICK|SLAP_TRUNCATE_MODE)) flags |= DB_TRUNCATE; rc = DB_OPEN( db->bdi_db, file, NULL /* name */, BDB_INDEXTYPE, bdb->bi_db_opflags | flags, bdb->bi_dbenv_mode ); ch_free( file ); if( rc != 0 ) { Debug( LDAP_DEBUG_ANY, "bdb_db_cache: db_open(%s) failed: %s (%d)\n", name->bv_val, db_strerror(rc), rc ); ldap_pvt_thread_mutex_unlock( &bdb->bi_database_mutex ); db->bdi_db->close( db->bdi_db, 0 ); ch_free( db ); return rc; } bdb->bi_databases[i] = db; bdb->bi_ndatabases = i+1; *dbout = db->bdi_db; ldap_pvt_thread_mutex_unlock( &bdb->bi_database_mutex ); return 0; }